Expose refinement minpks as user parameter

jadball · Dec 13, 2024 · 8cc5f3c · 8cc5f3c
1 parent bfcd7d2
commit 8cc5f3c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ImageD11/sinograms/point_by_point.py b/ImageD11/sinograms/point_by_point.py
@@ -628,6 +628,7 @@ def __init__(self,
                  ifrac=None,
                  forref=None,
                  y0=0.0,
+                 min_grain_npks=6
                  ):
         self.dset = dset
         self.phase_name = phase_name
@@ -647,6 +648,7 @@ def __init__(self,
         # refinement parameters
         self.hkl_tol_refine = hkl_tol_refine
         self.hkl_tol_refine_merged = hkl_tol_refine_merged
+        self.min_grain_npks = min_grain_npks
         # geometry stuff
         self.ystep = self.dset.ystep
         self.y0 = y0
@@ -961,7 +963,7 @@ def to_h5(self, filename=None, h5group='PBPRefine'):
 
             # other pars we need for refinement
             pars = ['phase_name', 'hkl_tol_origins', 'hkl_tol_refine', 'hkl_tol_refine_merged', 'fpks', 'ds_tol',
-                    'etacut', 'ifrac', 'forref', 'y0']
+                    'etacut', 'ifrac', 'forref', 'y0', 'min_grain_npks']
 
             for par in pars:
                 try:
@@ -998,7 +1000,7 @@ def from_h5(cls, filename, h5group='PBPRefine'):
                     continue
 
             pars = ['phase_name', 'hkl_tol_origins', 'hkl_tol_refine', 'hkl_tol_refine_merged', 'fpks', 'ds_tol',
-                    'etacut', 'ifrac', 'forref', 'y0']
+                    'etacut', 'ifrac', 'forref', 'y0', 'min_grain_npks']
             pars_dict = {}
             for par in pars:
                 try:
@@ -1169,7 +1171,8 @@ def run_refine(self, points_step_space=None, npoints=None, output_filename=None,
                                                                         pars['chi'],
                                                                         pars['wavelength'],
                                                                         tol=self.hkl_tol_refine,
-                                                                        merge_tol=self.hkl_tol_refine_merged
+                                                                        merge_tol=self.hkl_tol_refine_merged,
+                                                                        min_grain_npks=int(self.min_grain_npks)
                                                                         )
             # now we match how the indexer returns dodgy values
             # mask nan 3x3 entires to identity matrix