|
10 | 10 | from ..reflectometry.types import ( |
11 | 11 | Beamline, |
12 | 12 | BeamSize, |
13 | | - DetectorData, |
14 | 13 | DetectorRotation, |
15 | 14 | Filename, |
16 | 15 | Measurement, |
17 | 16 | NeXusComponent, |
18 | 17 | NeXusDetectorName, |
19 | 18 | ProtonCurrent, |
20 | 19 | RawChopper, |
| 20 | + RawDetector, |
21 | 21 | RawSampleRotation, |
22 | 22 | RunType, |
23 | 23 | SampleRotation, |
@@ -49,7 +49,7 @@ def load_events( |
49 | 49 | chopper_separation: ChopperSeparation[RunType], |
50 | 50 | sample_size: SampleSize[RunType], |
51 | 51 | beam_size: BeamSize[RunType], |
52 | | -) -> DetectorData[RunType]: |
| 52 | +) -> RawDetector[RunType]: |
53 | 53 | event_data = detector["data"] |
54 | 54 | if 'event_time_zero' in event_data.coords: |
55 | 55 | event_data.bins.coords['event_time_zero'] = sc.bins_like( |
@@ -77,7 +77,7 @@ def load_events( |
77 | 77 | data.coords["chopper_distance"] = chopper_distance |
78 | 78 | data.coords["sample_size"] = sample_size |
79 | 79 | data.coords["beam_size"] = beam_size |
80 | | - return DetectorData[RunType](data) |
| 80 | + return RawDetector[RunType](data) |
81 | 81 |
|
82 | 82 |
|
83 | 83 | def amor_chopper(f: Filename[RunType]) -> RawChopper[RunType]: |
@@ -133,13 +133,13 @@ def load_amor_proton_current( |
133 | 133 | return pc |
134 | 134 |
|
135 | 135 |
|
136 | | -def load_beamline_metadata(filename: Filename[SampleRun]) -> Beamline: |
| 136 | +def load_beamline_metadata(filename: Filename[RunType]) -> Beamline[RunType]: |
137 | 137 | return nexus_workflow.load_beamline_metadata_from_nexus( |
138 | 138 | NeXusFileSpec[SampleRun](filename) |
139 | 139 | ) |
140 | 140 |
|
141 | 141 |
|
142 | | -def load_measurement_metadata(filename: Filename[SampleRun]) -> Measurement: |
| 142 | +def load_measurement_metadata(filename: Filename[RunType]) -> Measurement[RunType]: |
143 | 143 | return nexus_workflow.load_measurement_metadata_from_nexus( |
144 | 144 | NeXusFileSpec[SampleRun](filename) |
145 | 145 | ) |
|
0 commit comments