From a0f4b2757dfde95d172f2f2d6cd9ae163cb5fc0d Mon Sep 17 00:00:00 2001
From: Hugo CHAUVET <hugo.chauvet@synchrotron-soleil.fr>
Date: Mon, 24 Jul 2023 15:48:03 +0200
Subject: [PATCH] Add the option to choose on which filter you can stabilize
 the image (so that you could use the visible one)

---
 src/bactos_functions.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/bactos_functions.py b/src/bactos_functions.py
index ca9f8a2..2200a21 100755
--- a/src/bactos_functions.py
+++ b/src/bactos_functions.py
@@ -1123,7 +1123,7 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
                    stardist_path = None, bact_size=(0.5, 15), prob_thresh=None, 
                    time_to_extract_bacteria=0, build_bacteria_figures=True, 
                    bacteria_extract_mask='TRP', drift_algo='cor', register_filter=True, 
-                   register_stack=True, register_mask=True):
+                   register_stack=True, register_mask=True, drift_filter=0):
     """
     Loop over the given folder name to extract data and export them inside one hdf file (containing all data from each folder)
 
@@ -1178,6 +1178,9 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
     - register_mask, bool:
         Register the two filter based on bacteria mask (default True)
         
+    - drift_filter, bool:
+        The filter on which to register the images in time
+        
     Example:
     --------
 
@@ -1232,7 +1235,7 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
         print('stabilisation')
         imgs = register_images(imgs, gaussian_size=0, move_filter=drug_filter,
                                ref_filter=ctrl_filter, register_filter=register_filter,
-                               register_stack=register_stack, show_info=False, up_factor=10, drift_filter=0,
+                               register_stack=register_stack, show_info=False, up_factor=10, drift_filter=drift_filter,
                                scale_correction=scale_correction, 
                                drift_algo=drift_algo)
     
-- 
GitLab