add methods to use mask image or acq-data for scatter estimation to S…

…IRF-object ScatterEstimator
SyneRBI · Oct 4, 2024 · 4900588 · 4900588
1 parent 6aacac7
commit 4900588
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/xSTIR/pSTIR/STIR.py b/src/xSTIR/pSTIR/STIR.py
@@ -3608,6 +3608,15 @@ def set_attenuation_correction_factors(self, arg):
         assert_validity(arg, AcquisitionData)
         parms.set_parameter(self.handle, self.name, 'setAttenuationCorrectionFactors', arg.handle)
 
+    def set_mask_image(self, image):
+        assert_validity(image, ImageData)
+        parms.set_parameter(self.handle, self.name, 'setMaskImage', image.handle)
+
+    def set_mask_acq_data(self, arg):
+        assert_validity(arg, AcquisitionData)
+        parms.set_parameter(self.handle, self.name, 'setMaskAcqData', arg.handle)
+
+
     def set_input(self, acq_data):
         assert_validity(acq_data, AcquisitionData)
         parms.set_parameter(self.handle, self.name, 'setInput', acq_data.handle)