move to utils_global

Author: chrishalcrow

spikeinterface_gui/backend_panel.py

@@ -4,7 +4,8 @@
 from copy import copy
 
 from .viewlist import possible_class_views
-from .layout_presets import get_layout_description, get_size_bottom_row, get_size_top_row
+from .layout_presets import get_layout_description
+from .utils_global import get_size_bottom_row, get_size_top_row
 
 # Used by views to emit/trigger signals
 class SignalNotifier(param.Parameterized):

spikeinterface_gui/backend_qt.py

@@ -7,7 +7,8 @@
 import weakref
 
 from .viewlist import possible_class_views
-from .layout_presets import get_layout_description, get_size_bottom_row, get_size_top_row
+from .layout_presets import get_layout_description
+from .utils_global import get_size_bottom_row, get_size_top_row
 
 from .utils_qt import qt_style, add_stretch_to_qtoolbar
 

spikeinterface_gui/layout_presets.py

@@ -53,50 +53,6 @@ def get_layout_description(preset_name, layout=None):
             preset_name = 'default'
         return _presets[preset_name]
 
-
-def get_size_top_row(initial_row, initial_col, is_zone_array, original_zone_array):
-    
-    if original_zone_array[initial_row][initial_col] == False:
-        return 0,0
-
-    num_rows = is_zone_array[initial_row][initial_col]*1
-    num_cols = num_rows
-
-    num_rows += (not is_zone_array[1][initial_col])*1
-
-    if num_rows == 1:
-        for zone in is_zone_array[0,1+initial_col:]:
-            if zone == True:
-                break
-            num_cols += 1
-    elif num_rows == 2:
-        for zone1, zone2 in np.transpose(is_zone_array[:,1+initial_col:]):
-            if zone1 == True or zone2 == True:
-                break
-            num_cols += 1
-
-    is_zone_array[initial_row:initial_row+num_rows,initial_col:initial_col+num_cols] = True
-
-    return num_rows, num_cols
-
-def get_size_bottom_row(initial_row, initial_col, is_zone_array, original_zone_array):
-    
-    if original_zone_array[initial_row][initial_col] == False:
-        return 0,0
-    
-    num_rows = is_zone_array[initial_row][initial_col]*1
-    if num_rows == 0:
-        return 0, 0
-    num_cols = num_rows
-
-    for zone in is_zone_array[1,1+initial_col:]:
-        if zone == True:
-            break
-        else:
-            num_cols += 1
-
-    return num_rows, num_cols
-
 default_layout = dict(
     zone1=['curation', 'spikelist'],
     zone2=['unitlist', 'mergelist'],

spikeinterface_gui/utils_global.py

@@ -0,0 +1,44 @@
+import numpy as np
+
+def get_size_top_row(initial_row, initial_col, is_zone_array, original_zone_array):
+    
+    if original_zone_array[initial_row][initial_col] == False:
+        return 0,0
+
+    num_rows = is_zone_array[initial_row][initial_col]*1
+    num_cols = num_rows
+
+    num_rows += (not is_zone_array[1][initial_col])*1
+
+    if num_rows == 1:
+        for zone in is_zone_array[0,1+initial_col:]:
+            if zone == True:
+                break
+            num_cols += 1
+    elif num_rows == 2:
+        for zone1, zone2 in np.transpose(is_zone_array[:,1+initial_col:]):
+            if zone1 == True or zone2 == True:
+                break
+            num_cols += 1
+
+    is_zone_array[initial_row:initial_row+num_rows,initial_col:initial_col+num_cols] = True
+
+    return num_rows, num_cols
+
+def get_size_bottom_row(initial_row, initial_col, is_zone_array, original_zone_array):
+    
+    if original_zone_array[initial_row][initial_col] == False:
+        return 0,0
+    
+    num_rows = is_zone_array[initial_row][initial_col]*1
+    if num_rows == 0:
+        return 0, 0
+    num_cols = num_rows
+
+    for zone in is_zone_array[1,1+initial_col:]:
+        if zone == True:
+            break
+        else:
+            num_cols += 1
+
+    return num_rows, num_cols