allowing reg_file input to compute_zpos

MouseLand · Feb 6, 2024 · c6e2915 · c6e2915
1 parent 361421f
commit c6e2915
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/suite2p/registration/zalign.py b/suite2p/registration/zalign.py
@@ -122,7 +122,7 @@ def register_stack(Z, ops):
     return Zreg, ops
 
 
-def compute_zpos(Zreg, ops):
+def compute_zpos(Zreg, ops, reg_file=None):
     """ compute z position of frames given z-stack Zreg
 
     Parameters
@@ -156,6 +156,7 @@ def compute_zpos(Zreg, ops):
             :,
         ]
 
+    reg_file = ops["reg_file"] if reg_file is None else reg_file
     nbytes = os.path.getsize(ops["reg_file"])
     nFrames = int(nbytes / (2 * Ly * Lx))