use one workflow name and an input argument

Author: nvaytet

packages/essreduce/src/ess/reduce/unwrap/lut.py

@@ -465,26 +465,6 @@ def simulate_chopper_cascade_using_tof(
     return SimulationResults(readings=sim_readings, choppers=choppers)
 
 
-def LookupTableWorkflow():
-    """
-    Create a workflow for computing a wavelength lookup table from a
-    simulation of neutrons propagating through a chopper cascade.
-    """
-    wf = sl.Pipeline(
-        (make_wavelength_lookup_table, simulate_chopper_cascade_using_tof),
-        params={
-            PulsePeriod: 1.0 / sc.scalar(14.0, unit="Hz"),
-            PulseStride: 1,
-            DistanceResolution: sc.scalar(0.1, unit="m"),
-            TimeResolution: sc.scalar(250.0, unit='us'),
-            NumberOfSimulatedNeutrons: 1_000_000,
-            SimulationSeed: None,
-            SimulationFacility: 'ess',
-        },
-    )
-    return wf
-
-
 def _polygon_intersections(polygons: list[np.ndarray], x: np.ndarray) -> np.ndarray:
     # Decompose the polygons into two 1D lines: the upper and lower bounds
     bounds = []
@@ -764,25 +744,50 @@ def make_wavelength_lut_from_polygons(
     )
 
 
-def FastLookupTableWorkflow():
+def LookupTableWorkflow(use_simulation: bool = True):
     """
-    Create a workflow for computing a wavelength lookup table from computing an
-    acceptance diagram for a pulse propagating through a chopper cascade.
+    Create a workflow for computing a wavelength lookup table.
+    If ``use_simulation`` is True, the workflow will compute the lookup table from a
+    simulation of neutrons propagating through a chopper cascade using the ``tof``
+    package.
+    If ``use_simulation`` is False, the workflow will compute the lookup table from
+    the acceptance diagram polygons generated by the ``chopper_cascade`` module.
+
+    Parameters
+    ----------
+    use_simulation:
+        Whether to compute the lookup table from a simulation of neutrons propagating
+        through a chopper cascade using the ``tof`` package, or from the acceptance
+        diagram polygons generated by the ``chopper_cascade`` module.
     """
-    wf = sl.Pipeline(
-        (make_wavelength_lut_from_polygons, compute_frame_sequence),
-        params={
-            PulsePeriod: 1.0 / sc.scalar(14.0, unit="Hz"),
-            PulseStride: 1,
-            DistanceResolution: sc.scalar(0.1, unit="m"),
-            TimeResolution: sc.scalar(250.0, unit='us'),
-            SourceBounds: SourceBounds(
-                time=(sc.scalar(0.0, unit='ms'), sc.scalar(5.0, unit='ms')),
-                wavelength=(
-                    sc.scalar(0.0, unit='angstrom'),
-                    sc.scalar(15.0, unit='angstrom'),
-                ),
-            ),
-        },
-    )
-    return wf
+    default_params = {
+        PulsePeriod: 1.0 / sc.scalar(14.0, unit="Hz"),
+        PulseStride: 1,
+        DistanceResolution: sc.scalar(0.1, unit="m"),
+        TimeResolution: sc.scalar(250.0, unit='us'),
+    }
+
+    if use_simulation:
+        providers = (make_wavelength_lookup_table, simulate_chopper_cascade_using_tof)
+        default_params.update(
+            {
+                NumberOfSimulatedNeutrons: 1_000_000,
+                SimulationSeed: None,
+                SimulationFacility: 'ess',
+            }
+        )
+    else:
+        providers = (make_wavelength_lut_from_polygons, compute_frame_sequence)
+        default_params.update(
+            {
+                SourceBounds: SourceBounds(
+                    time=(sc.scalar(0.0, unit='ms'), sc.scalar(5.0, unit='ms')),
+                    wavelength=(
+                        sc.scalar(0.0, unit='angstrom'),
+                        sc.scalar(15.0, unit='angstrom'),
+                    ),
+                )
+            }
+        )
+
+    return sl.Pipeline(providers, params=default_params)