simba-uw-tf-dev 4.6.2__py3-none-any.whl → 4.7.2__py3-none-any.whl

simba/sandbox/get_cpu_pool.py

@@ -0,0 +1,5 @@
+from simba.utils.data import get_cpu_pool, terminate_cpu_pool
+
+
+pool = get_cpu_pool()
+terminate_cpu_pool(pool=pool)

simba/sandbox/proboscis_to_tip.py

@@ -0,0 +1,28 @@
+import os
+from simba.mixins.feature_extraction_mixin import FeatureExtractionMixin
+from simba.mixins.config_reader import ConfigReader
+from simba.utils.read_write import read_df, get_fn_ext, read_video_info, write_df
+
+CONFIG_PATH = r"C:\troubleshooting\srami0619\project_folder\project_config.ini"
+BASE = 'prob_base'
+TIP = 'prob_tip'
+
+
+config = ConfigReader(config_path=CONFIG_PATH, read_video_info=True, create_logger=False)
+animal_cnt = config.animal_cnt
+for file_cnt, file_path in enumerate(config.outlier_corrected_paths):
+    file_name = get_fn_ext(file_path)[1]
+    _, px_per_mm, fps = read_video_info(video_name=file_name, video_info_df=config.video_info_df)
+    data_df = read_df(file_path=file_path)
+    save_path = os.path.join(config.features_dir, f'{file_name}.csv')
+    for animal_id in range(1, animal_cnt+1):
+        print(f'Analysing file {file_name}, animal {animal_id} ({file_cnt + 1}/{len(config.outlier_corrected_paths)})')
+        base_cols = [f'{BASE}_{animal_id}_x', f'{BASE}_{animal_id}_y']
+        tip_cols = [f'{TIP}_{animal_id}_x', f'{TIP}_{animal_id}_y']
+        prob_data, tip_data = data_df[base_cols], data_df[tip_cols]
+        data_df[f'animal_{animal_id}_base_tip_distance'] = FeatureExtractionMixin().keypoint_distances(a=prob_data.values, b=tip_data.values, px_per_mm=px_per_mm, in_centimeters=False)
+    write_df(df=data_df, file_type='csv', save_path=save_path, verbose=True)
+
+
+
+

simba/sandbox/test_directionality.py

@@ -0,0 +1,47 @@
+"""Test directionality_to_nonstatic_target function"""
+import sys
+import numpy as np
+
+print(f"Python: {sys.executable}")
+print("="*60)
+
+try:
+    from simba.data_processors.cuda.geometry import directionality_to_nonstatic_target
+    print("✓ Function imported successfully")
+except Exception as e:
+    print(f"✗ Error importing function: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)
+
+# Create test data
+print("\nCreating test data...")
+left_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+right_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+nose = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+target = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+
+print(f"left_ear shape: {left_ear.shape}, dtype: {left_ear.dtype}")
+print(f"right_ear shape: {right_ear.shape}, dtype: {right_ear.dtype}")
+print(f"nose shape: {nose.shape}, dtype: {nose.dtype}")
+print(f"target shape: {target.shape}, dtype: {target.dtype}")
+
+# Test the function
+print("\nTesting directionality_to_nonstatic_target...")
+try:
+    result = directionality_to_nonstatic_target(
+        left_ear=left_ear,
+        right_ear=right_ear,
+        nose=nose,
+        target=target
+    )
+    print(f"✓ SUCCESS! Result shape: {result.shape}")
+    print(f"Result: {result}")
+except Exception as e:
+    print(f"✗ Error running function: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)
+
+print("\n" + "="*60)
+print("Test passed!")

simba/sandbox/test_nonstatic_directionality.py

@@ -0,0 +1,27 @@
+"""Test directionality_to_nonstatic_target with int64"""
+import numpy as np
+from simba.data_processors.cuda.geometry import directionality_to_nonstatic_target
+
+print("Testing directionality_to_nonstatic_target with int64...")
+
+left_ear = np.random.randint(0, 500, (100, 2)).astype(np.int64)
+right_ear = np.random.randint(0, 500, (100, 2)).astype(np.int64)
+nose = np.random.randint(0, 500, (100, 2)).astype(np.int64)
+target = np.random.randint(0, 500, (100, 2)).astype(np.int64)
+
+print(f"Arrays shape: {left_ear.shape}, dtype: {left_ear.dtype}")
+
+try:
+    result = directionality_to_nonstatic_target(
+        left_ear=left_ear,
+        right_ear=right_ear,
+        nose=nose,
+        target=target,
+        verbose=True
+    )
+    print(f"✓ SUCCESS! Result shape: {result.shape}, dtype: {result.dtype}")
+    print(f"First 5 results:\n{result[:5]}")
+except Exception as e:
+    print(f"✗ Error: {e}")
+    import traceback
+    traceback.print_exc()

simba/sandbox/test_pycharm_cuda.py

@@ -0,0 +1,51 @@
+"""Test script to verify PyCharm CUDA environment setup"""
+import sys
+print(f"Python version: {sys.version}")
+print(f"Python path: {sys.executable}")
+
+try:
+    from numba import cuda
+    print(f"\nNumba CUDA available: {cuda.is_available()}")
+    if cuda.is_available():
+        print(f"CUDA devices: {len(cuda.gpus)}")
+except Exception as e:
+    print(f"\nError importing numba.cuda: {e}")
+
+try:
+    import cupy as cp
+    print(f"CuPy version: {cp.__version__}")
+    # Test CuPy
+    x = cp.array([1, 2, 3, 4, 5])
+    print(f"CuPy test array: {x.get()}")
+except Exception as e:
+    print(f"Error importing cupy: {e}")
+
+try:
+    import numpy as np
+    print(f"NumPy version: {np.__version__}")
+except Exception as e:
+    print(f"Error importing numpy: {e}")
+
+try:
+    import cv2
+    print(f"OpenCV version: {cv2.__version__}")
+except Exception as e:
+    print(f"Error importing cv2: {e}")
+
+# Test SimBA CUDA imports
+try:
+    from simba.data_processors.cuda.geometry import is_inside_rectangle
+    print("\n✓ Successfully imported simba.data_processors.cuda.geometry")
+    
+    # Quick test
+    test_points = np.array([[150, 150], [300, 300], [50, 50]], dtype=np.int32)
+    test_rect = np.array([[100, 100], [400, 400]], dtype=np.int32)
+    result = is_inside_rectangle(x=test_points, y=test_rect)
+    print(f"✓ is_inside_rectangle test passed: {result}")
+except Exception as e:
+    print(f"\n✗ Error importing SimBA CUDA functions: {e}")
+    import traceback
+    traceback.print_exc()
+
+print("\n" + "="*50)
+print("Environment check complete!")

simba/sandbox/test_simba_install.py

@@ -0,0 +1,41 @@
+"""Test SimBA installation in conda environment"""
+import sys
+print(f"Python: {sys.executable}")
+
+try:
+    import simba
+    print("✓ SimBA imported successfully!")
+except Exception as e:
+    print(f"✗ Error importing SimBA: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)
+
+try:
+    from simba.data_processors.cuda.geometry import is_inside_rectangle
+    print("✓ CUDA geometry functions imported!")
+except Exception as e:
+    print(f"✗ Error importing CUDA functions: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)
+
+try:
+    import cupy as cp
+    import numpy as np
+    print(f"✓ CuPy {cp.__version__} available")
+    
+    # Quick test
+    test_points = np.array([[150, 150], [300, 300], [50, 50]], dtype=np.int32)
+    test_rect = np.array([[100, 100], [400, 400]], dtype=np.int32)
+    result = is_inside_rectangle(x=test_points, y=test_rect)
+    print(f"✓ CUDA function test passed: {result}")
+except Exception as e:
+    print(f"✗ Error testing CUDA: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)
+
+print("\n" + "="*50)
+print("All tests passed! SimBA is ready to use.")
+print("="*50)

simba/sandbox/test_static_directionality.py

@@ -0,0 +1,26 @@
+"""Test directionality_to_static_targets function"""
+import numpy as np
+from simba.data_processors.cuda.geometry import directionality_to_static_targets
+
+print("Testing directionality_to_static_targets...")
+
+left_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+right_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+nose = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+target = np.array([250, 250], dtype=np.int32)
+
+print(f"target shape: {target.shape}, dtype: {target.dtype}")
+
+try:
+    result = directionality_to_static_targets(
+        left_ear=left_ear,
+        right_ear=right_ear,
+        nose=nose,
+        target=target
+    )
+    print(f"✓ SUCCESS! Result shape: {result.shape}")
+    print(f"Result:\n{result}")
+except Exception as e:
+    print(f"✗ Error: {e}")
+    import traceback
+    traceback.print_exc()

simba/sandbox/test_static_directionality_2d.py

@@ -0,0 +1,26 @@
+"""Test directionality_to_static_targets with correct 1D target"""
+import numpy as np
+from simba.data_processors.cuda.geometry import directionality_to_static_targets
+
+print("Testing directionality_to_static_targets with 1D target (correct)...")
+
+left_ear = np.random.randint(0, 500, (100, 2)).astype(np.int32)
+right_ear = np.random.randint(0, 500, (100, 2)).astype(np.int32)
+nose = np.random.randint(0, 500, (100, 2)).astype(np.int32)
+target = np.array([250, 250], dtype=np.int32)  # 1D array for static target
+
+print(f"target shape: {target.shape}, dtype: {target.dtype}")
+
+try:
+    result = directionality_to_static_targets(
+        left_ear=left_ear,
+        right_ear=right_ear,
+        nose=nose,
+        target=target
+    )
+    print(f"✓ SUCCESS! Result shape: {result.shape}")
+    print(f"First 5 results:\n{result[:5]}")
+except Exception as e:
+    print(f"✗ Error: {e}")
+    import traceback
+    traceback.print_exc()

simba/sandbox/verify_env.py

@@ -0,0 +1,42 @@
+"""Verify CUDA environment is properly configured"""
+from numba import cuda
+import cupy as cp
+import numba
+
+print("="*60)
+print("CUDA Environment Verification")
+print("="*60)
+print(f"Numba version: {numba.__version__}")
+print(f"CuPy version: {cp.__version__}")
+print(f"CUDA available: {cuda.is_available()}")
+if cuda.is_available():
+    print(f"CUDA devices: {len(cuda.gpus)}")
+    for i, gpu in enumerate(cuda.gpus):
+        print(f"  GPU {i}: {gpu}")
+
+# Test the function
+print("\n" + "="*60)
+print("Testing directionality_to_nonstatic_target...")
+try:
+    from simba.data_processors.cuda.geometry import directionality_to_nonstatic_target
+    import numpy as np
+    
+    left_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+    right_ear = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+    nose = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+    target = np.random.randint(0, 500, (10, 2)).astype(np.int32)
+    
+    result = directionality_to_nonstatic_target(
+        left_ear=left_ear,
+        right_ear=right_ear,
+        nose=nose,
+        target=target
+    )
+    print(f"✓ Function works! Result shape: {result.shape}")
+except Exception as e:
+    print(f"✗ Error: {e}")
+    import traceback
+    traceback.print_exc()
+
+print("\n" + "="*60)
+print("Environment is ready!")

simba/third_party_label_appenders/transform/coco_keypoints_to_yolo.py

@@ -54,15 +54,15 @@ class COCOKeypoints2Yolo:
     :return: None
 
     :example:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\s1\annotations\s1.json", img_dir=r"D:\cvat_annotations\frames\simon", save_dir=r"D:\cvat_annotations\frames\yolo_keypoints", clahe=True)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/s1/annotations/s1.json", img_dir=r"D:/cvat_annotations/frames/simon", save_dir=r"D:/cvat_annotations/frames/yolo_keypoints", clahe=True)
     >>> runner.run()
 
     :example II:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\merged.json", img_dir=r"D:\cvat_annotations\frames", save_dir=r"D:\cvat_annotations\frames\yolo", clahe=False)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/merged.json", img_dir=r"D:/cvat_annotations/frames", save_dir=r"D:/cvat_annotations/frames/yolo", clahe=False)
     >>> runner.run()
 
     :example III:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"E:\netholabs_videos\mosaics\subset\to_annotate\2d_mosaic_batch_1.json", img_dir=r"E:\netholabs_videos\mosaics\subset\to_annotate", save_dir=r"E:\netholabs_videos\mosaics\yolo_mdl", clahe=False)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"E:/netholabs_videos/mosaics/subset/to_annotate/2d_mosaic_batch_1.json", img_dir=r"E:/netholabs_videos/mosaics/subset/to_annotate", save_dir=r"E:/netholabs_videos/mosaics/yolo_mdl", clahe=False)
     >>> runner.run()
 
      :references:

simba/third_party_label_appenders/transform/coco_keypoints_to_yolo_bbox.py

@@ -58,11 +58,11 @@ class COCOKeypoints2YoloBbox:
     :return: None
 
     :example:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\s1\annotations\s1.json", img_dir=r"D:\cvat_annotations\frames\simon", save_dir=r"D:\cvat_annotations\frames\yolo_keypoints", clahe=True)
+    >>> runner = COCOKeypoints2YoloBbox(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/s1/annotations/s1.json", img_dir=r"D:/cvat_annotations/frames/simon", save_dir=r"D:/cvat_annotations/frames/yolo_keypoints", clahe=True)
     >>> runner.run()
 
     :example II:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\merged.json", img_dir=r"D:\cvat_annotations\frames", save_dir=r"D:\cvat_annotations\frames\yolo", clahe=False)
+    >>> runner = COCOKeypoints2YoloBbox(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/merged.json", img_dir=r"D:/cvat_annotations/frames", save_dir=r"D:/cvat_annotations/frames/yolo", clahe=False)
     >>> runner.run()
 
      :references:

simba/third_party_label_appenders/transform/simba_to_yolo.py

@@ -117,12 +117,15 @@ class SimBA2Yolo:
         annotations, timer, body_part_headers = [], SimbaTimer(start=True), []
         for file_cnt, video_name in enumerate(self.data_w_video):
             data = read_df(file_path=self.data_paths[video_name], file_type=self.config.file_type)
+            data.columns = [x.lower() for x in list(data.columns)]
+            bp_header_names = [x.lower() for x in self.config.bp_headers]
             check_valid_dataframe(df=data, source=f'{self.__class__.__name__} {self.data_paths[video_name]}', valid_dtypes=Formats.NUMERIC_DTYPES.value)
             video_path = self.video_paths[video_name]
             check_video_and_data_frm_count_align(video=video_path, data=data, name=self.data_paths[video_name], raise_error=True)
             p_data = data[data.columns[list(data.columns.str.endswith('_p'))]]
             data = data.loc[:, ~data.columns.str.endswith('_p')].reset_index(drop=True)
             data = data.iloc[(p_data[(p_data > self.threshold).all(axis=1)].index)]
+            data = data[[x for x in bp_header_names if not x.endswith('_p')]]
             body_part_headers = data.columns
             data['video'], frm_cnt = video_name, len(data)
             if self.sample_size is None:
@@ -155,7 +158,7 @@ class SimBA2Yolo:
             if frm_idx in train_idx:
                 img_save_path, lbl_save_path = os.path.join(self.img_train_dir, f'{file_name}.png'), os.path.join(self.lbl_train_dir, f'{file_name}.txt')
             else:
-                img_save_path, lbl_save_path = os.path.join(self.img_train_dir, f'{file_name}.png'), os.path.join(self.lb_val_dir, f'{file_name}.txt')
+                img_save_path, lbl_save_path = os.path.join(self.img_val_dir, f'{file_name}.png'), os.path.join(self.lb_val_dir, f'{file_name}.txt')
             img = read_frm_of_video(video_path=vid_path, frame_index=frm_idx, greyscale=self.greyscale, clahe=self.clahe)
             img_h, img_w = img.shape[0], img.shape[1]
             keypoints_with_id = {}
@@ -181,8 +184,8 @@ class SimBA2Yolo:
         timer.stop_timer()
         stdout_success(msg=f'YOLO formated data saved in {self.save_dir} directory', source=self.__class__.__name__, elapsed_time=timer.elapsed_time_str)
 
-
-# SAVE_DIR = r'D:\troubleshooting\mitra\mitra_yolo'
-# CONFIG_PATH = r"C:\troubleshooting\mitra\project_folder\project_config.ini"
-# runner = SimBA2Yolo(config_path=CONFIG_PATH, save_dir=SAVE_DIR, sample_size=10, verbose=True, names=('animal_1',))
+#
+# SAVE_DIR = r'E:\troubleshooting\mitra\yolo_0126\yolo_train_0126'
+# CONFIG_PATH = r"E:\troubleshooting\mitra\project_folder\project_config.ini"
+# runner = SimBA2Yolo(config_path=CONFIG_PATH, save_dir=SAVE_DIR, sample_size=50, verbose=True, names=('animal_1',), threshold=0.5)
 # runner.run()

simba/ui/pop_ups/clf_plot_pop_up.py

@@ -49,8 +49,8 @@ class SklearnVisualizationPopUp(PopUpMixin, ConfigReader):
         pose_palettes = Options.PALETTE_OPTIONS_CATEGORICAL.value + Options.PALETTE_OPTIONS.value
         PopUpMixin.__init__(self, title="VISUALIZE CLASSIFICATION (SKLEARN) RESULTS", icon='photos')
         bp_threshold_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="BODY-PART VISUALIZATION THRESHOLD", icon_name='threshold', icon_link=Links.SKLEARN_PLOTS.value, padx=5, pady=5, relief='solid')
-        self.bp_threshold_lbl = SimBALabel(parent=bp_threshold_frm, txt="Body-parts detected below the set threshold won't be shown in the output videos.", font=Formats.FONT_REGULAR_ITALICS.value)
-        self.bp_threshold_entry = Entry_Box(parent=bp_threshold_frm, fileDescription='BODY-PART PROBABILITY THRESHOLD: ', labelwidth=40, entry_box_width=15, value=0.00, img='green_dice')
+        self.bp_threshold_lbl = SimBALabel(parent=bp_threshold_frm, txt="Body-parts detected below the set threshold won't be shown in the output videos (use 0.0 to see all body-part predictions)", font=Formats.FONT_REGULAR_ITALICS.value)
+        self.bp_threshold_entry = Entry_Box(parent=bp_threshold_frm, fileDescription='BODY-PART PROBABILITY THRESHOLD: ', labelwidth=40, entry_box_width=15, value=0.00, img='green_dice', justify='center')
         self.get_bp_probability_threshold()
 
         bp_threshold_frm.grid(row=0, column=0, sticky=NW)

simba/ui/pop_ups/fsttc_pop_up.py

@@ -7,25 +7,29 @@ from typing import Union
 from simba.data_processors.fsttc_calculator import FSTTCCalculator
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.ui.tkinter_functions import CreateLabelFrameWithIcon, Entry_Box
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box,
+                                        SimbaCheckbox)
 from simba.utils.checks import check_int
-from simba.utils.enums import Defaults, Formats, Keys, Links
+from simba.utils.enums import Formats, Keys, Links
 from simba.utils.errors import CountError
 
 
 class FSTTCPopUp(PopUpMixin, ConfigReader):
-    def __init__(self, config_path: Union[str, os.PathLike]):
+    def __init__(self,
+                 config_path: Union[str, os.PathLike]):
 
-        PopUpMixin.__init__(self, title="FORWARD SPIKE TIME TILING COEFFICIENTS", icon='dependency')
         ConfigReader.__init__(self, config_path=config_path, read_video_info=False)
+        if len(self.clf_names) < 2:
+            raise CountError(msg=f'Cannot compute FSTTC: Needs project with at least 2 classified behaviors, got {len(self.clf_names)}', source=self.__class__.__name__)
+
+        PopUpMixin.__init__(self, title="FORWARD SPIKE TIME TILING COEFFICIENTS", icon='dependency')
         settings_frm = CreateLabelFrameWithIcon( parent=self.main_frm, header="SETTINGS", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.FSTTC.value,)
-        self.time_delta_eb = Entry_Box(settings_frm, "TIME-DELTA", "10", validation="numeric")
-        self.graph_cb_var = BooleanVar(value=False)
-        graph_cb = Checkbutton(settings_frm, font=Formats.FONT_REGULAR.value, text="CREATE GRAPH", variable=self.graph_cb_var)
-        self.join_bouts_within_delta_var = BooleanVar(value=False)
-        join_bouts_within_delta_cb = Checkbutton(settings_frm, text="JOIN BOUTS WITHIN TIME-DELTA", font=Formats.FONT_REGULAR.value, variable=self.join_bouts_within_delta_var,)
-        self.time_delta_at_onset_var = BooleanVar(value=False)
-        time_delta_at_onset_cb = Checkbutton( settings_frm, text="TIME-DELTA AT BOUT START", font=Formats.FONT_REGULAR.value, variable=self.time_delta_at_onset_var)
+        self.time_delta_eb = Entry_Box(parent=settings_frm, fileDescription="TIME-DELTA", labelwidth=25, validation="numeric", entry_box_width=20, img='timer_2')
+
+        graph_cb, self.graph_cb_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='line_chart_blue', txt="CREATE GRAPH")
+        join_bouts_within_delta_cb, self.join_bouts_within_delta_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='join_purple', txt="JOIN BOUTS WITHIN TIME-DELTA")
+        time_delta_at_onset_cb, self.time_delta_at_onset_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='play', txt="TIME-DELTA AT BOUT START")
+
         settings_frm.grid(row=0, column=0, sticky=NW)
         self.time_delta_eb.grid(row=0, column=0, sticky="NW")
         join_bouts_within_delta_cb.grid(row=1, column=0, sticky="NW")
@@ -43,22 +47,20 @@ class FSTTCPopUp(PopUpMixin, ConfigReader):
                 targets.append(behaviour)
 
         if len(targets) < 2:
-            raise CountError(
-                msg="FORWARD SPIKE TIME TILING COEFFICIENTS REQUIRE 2 OR MORE BEHAVIORS.",
-                source=self.__class__.__name__,
-            )
-
-        fsttc_calculator = FSTTCCalculator(
-            config_path=self.config_path,
-            time_window=self.time_delta_eb.entry_get,
-            join_bouts_within_delta=self.join_bouts_within_delta_var.get(),
-            time_delta_at_onset=self.time_delta_at_onset_var.get(),
-            behavior_lst=targets,
-            create_graphs=self.graph_cb_var.get(),
-        )
+            raise CountError(msg="FORWARD SPIKE TIME TILING COEFFICIENTS REQUIRE 2 OR MORE BEHAVIORS.", source=self.__class__.__name__,)
+
+        fsttc_calculator = FSTTCCalculator(config_path=self.config_path,
+                                           time_window=self.time_delta_eb.entry_get,
+                                           join_bouts_within_delta=self.join_bouts_within_delta_var.get(),
+                                           time_delta_at_onset=self.time_delta_at_onset_var.get(),
+                                           behavior_lst=targets,
+                                           create_graphs=self.graph_cb_var.get())
         fsttc_calculator.run()
 
 
-#ExtractAnnotationFramesPopUp(config_path=)
+
 
 #_ = FSTTCPopUp(config_path=r"C:\troubleshooting\multi_animal_dlc_two_c57\project_folder\project_config.ini")
+
+#_ = FSTTCPopUp(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini")
+

simba/ui/pop_ups/gantt_pop_up.py

@@ -13,6 +13,7 @@ from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, SimbaButton,
 from simba.utils.checks import check_if_filepath_list_is_empty
 from simba.utils.enums import Formats, Links, Options
 from simba.utils.errors import NoSpecifiedOutputError
+from simba.utils.lookups import get_fonts
 from simba.utils.read_write import find_files_of_filetypes_in_directory
 
 
@@ -29,7 +30,9 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         check_if_filepath_list_is_empty(filepaths=self.machine_results_paths,error_msg=f"SIMBA ERROR: Zero files found in the {self.machine_results_dir} directory. Create classification results before visualizing gantt charts",)
         palettes = Options.PALETTE_OPTIONS_CATEGORICAL.value + Options.PALETTE_OPTIONS.value
         self.data_paths = find_files_of_filetypes_in_directory(directory=self.machine_results_dir, extensions=[f'.{self.file_type}'], as_dict=True)
-        max_file_name_len = max(len(k) for k in self.data_paths) + 5
+        max_file_name_len, fonts = max(len(k) for k in self.data_paths) + 5, list(get_fonts(sort_alphabetically=True).keys())
+        fonts.insert(0, 'AUTO')
+        default_font = 'Arial' if 'Arial' in fonts else 'AUTO'
         PopUpMixin.__init__(self, config_path=config_path, title="VISUALIZE GANTT PLOTS", icon='gantt_small')
 
         self.style_settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="STYLE SETTINGS", icon_name='settings', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
@@ -39,6 +42,15 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         self.palette_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=palettes, label='COLOR PALETTE: ', label_width=30, dropdown_width=30, value='Set1', img='palette_small')
         self.time_format_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=['SECONDS', 'HH:MM:SS'], label='X-AXIS TIME FORMAT: ', label_width=30, dropdown_width=30, value='SECONDS', img='timer_2')
         self.core_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=list(range(1, self.cpu_cnt+1)), label='CPU CORES: ', label_width=30, dropdown_width=30, value=int(self.cpu_cnt/2), img='cpu_small')
+        self.font_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=fonts, label='FONT: ', label_width=30, dropdown_width=30, value=default_font, img='font')
+
+
+        self.clf_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="BEHAVIORS", icon_name='forest', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
+        self.clf_choices = {}
+        for cnt, clf_name in enumerate(self.clf_names):
+            gantt_frames_cb, self.gantt_frames_var = SimbaCheckbox(parent=self.clf_frm, txt=clf_name, val=True)
+            self.clf_choices[clf_name] = self.gantt_frames_var
+            gantt_frames_cb.grid(row=cnt, column=0, sticky=NW)
 
         self.settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="VISUALIZATION SETTINGS", icon_name='eye', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
         gantt_frames_cb, self.gantt_frames_var = SimbaCheckbox(parent=self.settings_frm, txt='CREATE FRAMES', txt_img='frames', val=False)
@@ -57,18 +69,20 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         self.font_rotation_dropdown.grid(row=2, sticky=NW)
         self.palette_dropdown.grid(row=3, sticky=NW)
         self.time_format_dropdown.grid(row=4, sticky=NW)
-        self.core_dropdown.grid(row=5, sticky=NW)
+        self.font_dropdown.grid(row=5, sticky=NW)
+        self.core_dropdown.grid(row=6, sticky=NW)
 
-        self.settings_frm.grid(row=1, sticky=NW, padx=10, pady=10)
+        self.clf_frm.grid(row=1, sticky=NW, padx=10, pady=10)
+        self.settings_frm.grid(row=2, sticky=NW, padx=10, pady=10)
         gantt_videos_cb.grid(row=0, sticky=NW)
         gantt_frames_cb.grid(row=1, sticky=W)
         gantt_last_frame_cb.grid(row=2, sticky=NW)
 
-        self.run_single_video_frm.grid(row=2, sticky=NW)
+        self.run_single_video_frm.grid(row=3, sticky=NW)
         self.run_single_video_btn.grid(row=0, sticky=NW)
         self.single_video_dropdown.grid(row=0, column=1, sticky=NW)
 
-        self.run_multiple_videos.grid(row=3, sticky=NW)
+        self.run_multiple_videos.grid(row=4, sticky=NW)
         self.run_multiple_video_btn.grid(row=0, sticky=NW)
 
         self.main_frm.mainloop()
@@ -85,8 +99,14 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         video_setting = self.gantt_videos_var.get()
         last_frm_setting = self.gantt_last_frame_var.get()
         palette = self.palette_dropdown.get_value()
+        font = self.font_dropdown.get_value()
+        font = None if font == 'AUTO' else font
         hhmmss = True if self.time_format_dropdown.get_value() == 'HH:MM:SS' else False
-
+        clf_names = []
+        for clf_name, clf_val in self.clf_choices.items():
+            if clf_val.get(): clf_names.append(clf_name)
+        if len(clf_names) < 1:
+            raise NoSpecifiedOutputError(msg="Select AT LEAST one behavior name.")
         if not frame_setting and not video_setting and not last_frm_setting:
             raise NoSpecifiedOutputError(msg="SIMBA ERROR: Please select gantt videos, frames, and/or last frame.")
 
@@ -103,6 +123,8 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
                                                      data_paths=data_paths,
                                                      width=width,
                                                      height=height,
+                                                     font=font,
+                                                     clf_names=clf_names,
                                                      font_size=font_size,
                                                      font_rotation=font_rotation,
                                                      core_cnt=core_cnt,
@@ -116,7 +138,9 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
                                                       last_frm_setting=last_frm_setting,
                                                       data_paths=data_paths,
                                                       width=width,
+                                                      font=font,
                                                       height=height,
+                                                      clf_names=clf_names,
                                                       font_size=font_size,
                                                       font_rotation=font_rotation,
                                                       palette=palette)
@@ -124,6 +148,7 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
 
 
 #_ = GanttPlotPopUp(config_path=r"D:\troubleshooting\maplight_ri\project_folder\project_config.ini")
+#_ = GanttPlotPopUp(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini")
 # _ = GanttPlotPopUp(config_path=r"/Users/simon/Desktop/envs/simba/troubleshooting/two_black_animals_14bp/project_folder/project_config.ini")
 
  # _ = GanttPlotPopUp(config_path=r"C:\troubleshooting\RAT_NOR\project_folder\project_config.ini")