|
1 | 1 | """Defines dataclasses used for the inputs to the compiled RAT code.""" |
2 | 2 |
|
3 | 3 | from dataclasses import dataclass, field |
4 | | -from typing import Union |
5 | | - |
6 | | -from RAT.utils.enums import Parallel, Procedures, Display, BoundHandling, Strategies, Calc, Geometries, Models |
7 | | - |
8 | | - |
9 | | -@dataclass |
10 | | -class Checks: |
11 | | - """Fit parameters for all parameter sets defined in the project.""" |
12 | | - fitParam: list[int] = field(default_factory=list) |
13 | | - fitBackgroundParam: list[int] = field(default_factory=list) |
14 | | - fitScalefactor: list[int] = field(default_factory=list) |
15 | | - fitQzshift: list[int] = field(default_factory=list) |
16 | | - fitBulkIn: list[int] = field(default_factory=list) |
17 | | - fitBulkOut: list[int] = field(default_factory=list) |
18 | | - fitResolutionParam: list[int] = field(default_factory=list) |
19 | | - fitDomainRatio: list[int] = field(default_factory=list) |
20 | | - |
21 | | - |
22 | | -@dataclass |
23 | | -class Limits: |
24 | | - """Parameters limits for all parameter sets defined in the project.""" |
25 | | - param: list[[float, float]] = field(default_factory=list) |
26 | | - backgroundParam: list[[float, float]] = field(default_factory=list) |
27 | | - scalefactor: list[[float, float]] = field(default_factory=list) |
28 | | - qzshift: list[[float, float]] = field(default_factory=list) |
29 | | - bulkIn: list[[float, float]] = field(default_factory=list) |
30 | | - bulkOut: list[[float, float]] = field(default_factory=list) |
31 | | - resolutionParam: list[[float, float]] = field(default_factory=list) |
32 | | - domainRatio: list[[float, float]] = field(default_factory=list) |
33 | | - |
34 | | - |
35 | | -@dataclass |
36 | | -class Priors: |
37 | | - """Priors parameters for all parameter sets defined in the project.""" |
38 | | - param: list[list] = field(default_factory=list) |
39 | | - backgroundParam: list[list] = field(default_factory=list) |
40 | | - scalefactor: list[list] = field(default_factory=list) |
41 | | - qzshift: list[list] = field(default_factory=list) |
42 | | - bulkIn: list[list] = field(default_factory=list) |
43 | | - bulkOut: list[list] = field(default_factory=list) |
44 | | - resolutionParam: list[list] = field(default_factory=list) |
45 | | - domainRatio: list[list] = field(default_factory=list) |
46 | | - priorNames: list[str] = field(default_factory=list) |
47 | | - priorValues: list = field(default_factory=list) |
48 | 4 |
|
| 5 | +from RAT.utils.enums import Parallel, Procedures, Display, BoundHandling, Strategies |
49 | 6 |
|
50 | 7 | @dataclass(frozen=True) |
51 | 8 | class Controls: |
@@ -82,45 +39,3 @@ class Controls: |
82 | 39 | pUnitGamma: float = 0.2 |
83 | 40 | boundHandling: BoundHandling = BoundHandling.Fold |
84 | 41 | adaptPCR: bool = False |
85 | | - # Checks |
86 | | - checks: Checks = Checks() |
87 | | - |
88 | | - |
89 | | -@dataclass(frozen=True) |
90 | | -class Problem: |
91 | | - """The full set of problemDef parameters required for RATMain.""" |
92 | | - # General |
93 | | - TF: str = Calc.NonPolarised |
94 | | - modelType: str = Models.StandardLayers |
95 | | - geometry: str = Geometries.AirSubstrate |
96 | | - useImaginary: bool = False |
97 | | - # Parameter Values |
98 | | - params: list[float] = field(default_factory=list) |
99 | | - bulkIn: list[float] = field(default_factory=list) |
100 | | - bulkOut: list[float] = field(default_factory=list) |
101 | | - qzshifts: list[float] = field(default_factory=list) |
102 | | - scalefactors: list[float] = field(default_factory=list) |
103 | | - domainRatio: list[int] = field(default_factory=list) |
104 | | - backgroundParams: list[float] = field(default_factory=list) |
105 | | - resolutionParams: list[float] = field(default_factory=list) |
106 | | - # Contrast Parameters |
107 | | - contrastBulkIns: list[int] = field(default_factory=list) |
108 | | - contrastBulkOuts: list[int] = field(default_factory=list) |
109 | | - contrastQzshifts: list[int] = field(default_factory=list) |
110 | | - contrastScalefactors: list[int] = field(default_factory=list) |
111 | | - contrastDomainRatios: list[int] = field(default_factory=list) |
112 | | - contrastBackgrounds: list[int] = field(default_factory=list) |
113 | | - contrastBackgroundsActions: list[int] = field(default_factory=list) |
114 | | - contrastResolutions: list[int] = field(default_factory=list) |
115 | | - contrastCustomFiles: list[Union[int, float]] = field(default_factory=list) |
116 | | - resample: list[int] = field(default_factory=list) |
117 | | - dataPresent: list[int] = field(default_factory=list) |
118 | | - oilChiDataPresent: list[int] = field(default_factory=list) |
119 | | - numberOfContrasts: int = 0 |
120 | | - # Other |
121 | | - numberOfLayers: int = 0 |
122 | | - numberOfDomainContrasts: int = 0 |
123 | | - fitParams: list = field(default_factory=list) |
124 | | - otherParams: list = field(default_factory=list) |
125 | | - fitLimits: list = field(default_factory=list) |
126 | | - otherLimits: list = field(default_factory=list) |
0 commit comments