Expose tof simulation number of neutron configuration.

Author: YooSunYoung

src/ess/nmx/configurations.py

@@ -98,6 +98,11 @@ class WorkflowConfig(BaseModel):
         "If None, the lookup table will be computed on-the-fly.",
         default=None,
     )
+    tof_simulation_num_neutrons: int = Field(
+        title="Number of Neutrons for TOF Simulation",
+        description="Number of neutrons to simulate for TOF lookup table calculation.",
+        default=1_000_000,
+    )
     tof_simulation_min_wavelength: float = Field(
         title="TOF Simulation Minimum Wavelength",
         description="Minimum wavelength for TOF simulation in Angstrom.",

src/ess/nmx/workflows.py

@@ -83,7 +83,12 @@ def _simulate_fixed_wavelength_tof(
     ----------
     """
     source = tof.Source(
-        facility="ess", neutrons=neutrons, pulses=2, seed=seed, wmax=wmax, wmin=wmin
+        facility="ess",
+        neutrons=neutrons,
+        pulses=1,  # NMX does not use pulse-skipping.
+        seed=seed,
+        wmax=wmax,
+        wmin=wmin,
     )
     nmx_det = tof.Detector(distance=max(ltotal_range), name="detector")
     model = tof.Model(source=source, choppers=[], detectors=[nmx_det])
@@ -290,6 +295,7 @@ def compute_lookup_table(
         wf[TimeOfFlightLookupTableFilename] = workflow_config.tof_lookup_table_file_path
     else:
         wf = patch_workflow_lookup_table_steps(wf=wf)
+        wf[NumberOfSimulatedNeutrons] = workflow_config.tof_simulation_num_neutrons
         wmax = sc.scalar(workflow_config.tof_simulation_max_wavelength, unit='angstrom')
         wmin = sc.scalar(workflow_config.tof_simulation_min_wavelength, unit='angstrom')
         wf[TofSimulationMaxWavelength] = wmax

tests/executable_test.py

@@ -93,6 +93,7 @@ def test_reduction_config() -> None:
         max_time_bin=100_000,
         time_bin_coordinate=TimeBinCoordinate.time_of_flight,
         time_bin_unit=TimeBinUnit.us,
+        tof_simulation_num_neutrons=700_000,
         tof_simulation_max_wavelength=5.0,
         tof_simulation_min_wavelength=1.0,
         tof_simulation_seed=12345,