biomechzoo 0.4.1__py3-none-any.whl → 0.4.5__py3-none-any.whl

__init__.py

@@ -0,0 +1,33 @@
+"""
+BiomechZoo: A Python toolbox for processing and analyzing human movement data.
+
+This package provides functions for converting, processing, analyzing,
+and visualizing biomechanical data (e.g., motion capture, EMG, kinetics).
+
+Example:
+    from biomechzoo import BiomechZoo
+    from biomechzoo.conversion import c3d2zoo
+
+    zoo = BiomechZoo()
+    zoo.conversion.c3d2zoo('path/to/data')
+"""
+
+# Import main class or entry point
+from .biomechzo import BiomechZoo
+
+# Import commonly used submodules
+from . import conversion
+from . import processing
+from . import plotting
+from . import utils
+
+# Define what gets exposed with "from biomechzoo import *"
+__all__ = [
+    "BiomechZoo",
+    "conversion",
+    "processing",
+    "plotting",
+    "utils",
+]
+
+__version__ = "0.4.4"

biomechzoo/biomechzoo.py

@@ -1,4 +1,6 @@
 import os
+import inspect
+import time
 from biomechzoo.utils.engine import engine  # assumes this returns .zoo files in folder
 from biomechzoo.utils.zload import zload
 from biomechzoo.utils.zsave import zsave
@@ -18,17 +20,28 @@ from biomechzoo.biomech_ops.normalize_data import normalize_data
 from biomechzoo.biomech_ops.phase_angle_data import phase_angle_data
 from biomechzoo.biomech_ops.continuous_relative_phase_data import continuous_relative_phase_data
 
-
 class BiomechZoo:
-    def __init__(self, in_folder, inplace=False, verbose=0):
+    def __init__(self, in_folder, inplace=False, subfolders=None, name_contains=None, verbose=0):
         self.verbose = verbose
         self.in_folder = in_folder
         self.verbose = verbose
-        self.inplace = inplace  # choice to save processed files to new folder
+        self.inplace = inplace               # choice to save processed files to new folder
+        self.subfolders = subfolders         # only run processes on list in subfolder
+        self.name_contains = name_contains   # only run processes on files with name_contains in file name
 
         batchdisp('BiomechZoo initialized', level=1, verbose=verbose)
         batchdisp('verbosity set to: {}'.format(verbose), level=1, verbose=verbose)
-        batchdisp('processing folder set to: {}'.format(self.in_folder), level=1, verbose=verbose)
+        batchdisp('root processing folder set to: {}'.format(self.in_folder), level=1, verbose=verbose)
+        if name_contains is not None:
+            batchdisp('only include files containing name_contains string: {}'.format(self.name_contains), level=1, verbose=verbose)
+        if subfolders is not None:
+            if type(subfolders) is list:
+                batchdisp('only process files in subfolder(s):', level=1, verbose=verbose)
+                for subfolder in self.subfolders:
+                    batchdisp('{}'.format(os.path.join(self.in_folder, subfolder)), level=1, verbose=verbose)
+            else:
+                batchdisp('only process files in subfolder(s): {}'.format(os.path.join(self.in_folder, self.subfolders)), level=1, verbose=verbose)
+
         if inplace:
             batchdisp('Processing mode: overwrite (inplace=True) (each step will be applied to same folder)', level=1, verbose=verbose)
         else:
@@ -50,57 +63,75 @@ class BiomechZoo:
 
     def mvnx2zoo(self, out_folder=None, inplace=False):
         """ Converts all .mvnx files in the folder to .zoo format """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder, extension='.mvnx')
+        fl = engine(in_folder, extension='.mvnx', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('converting mvnx to zoo for {}'.format(f), level=2, verbose=verbose)
             data = mvnx2zoo_data(f)
             f_zoo = f.replace('.mvnx', '.zoo')
             zsave(f_zoo, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
-        batchdisp('mvnx to zoo conversion complete', level=1, verbose=verbose)
-
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} conversion complete for {} files'.format(method_name, len(fl)), level=1, verbose=verbose)
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def c3d2zoo(self, out_folder=None, inplace=None):
         """ Converts all .c3d files in the folder to .zoo format """
+        start_time = time.time()
         from ezc3d import c3d
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder, extension='.c3d')
+        fl = engine(in_folder, extension='.c3d', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('converting c3d to zoo for {}'.format(f), level=2, verbose=verbose)
             c3d_obj = c3d(f)
             data = c3d2zoo_data(c3d_obj)
             f_zoo = f.replace('.c3d', '.zoo')
             zsave(f_zoo, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
-        batchdisp('c3d to zoo conversion complete', level=1, verbose=verbose)
-
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
-    def csv2zoo(self, out_folder=None, inplace=None):
+    def csv2zoo(self, out_folder=None, inplace=None, skip_rows=0):
         """ Converts generic .csv file in the folder to .zoo format """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder, extension='.csv')
+        fl = engine(in_folder, extension='.csv', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('converting csv to zoo for {}'.format(f), level=2, verbose=verbose)
-            data = csv2zoo_data(f)
+            data = csv2zoo_data(f, type='csv', skip_rows=skip_rows)
             f_zoo = f.replace('.csv', '.zoo')
             zsave(f_zoo, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
-        batchdisp('csv to zoo conversion complete', level=1, verbose=verbose)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
+        # Update self.folder after  processing
+        self._update_folder(out_folder, inplace, in_folder)
 
+    def parquet2zoo(self, out_folder=None, inplace=None):
+        """ Converts generic .csv file in the folder to .zoo format """
+        start_time = time.time()
+        verbose = self.verbose
+        in_folder = self.in_folder
+        if inplace is None:
+            inplace = self.inplace
+        fl = engine(in_folder, extension='.parquet', name_contains=self.name_contains, subfolders=self.subfolders)
+        for f in fl:
+            batchdisp('converting parquet to zoo for {}'.format(f), level=2, verbose=verbose)
+            data = csv2zoo_data(f, type='parquet')
+            f_zoo = f.replace('.parquet', '.zoo')
+            zsave(f_zoo, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
@@ -110,67 +141,70 @@ class BiomechZoo:
 
     def phase_angle(self, ch, out_folder=None, inplace=None):
         """ computes phase angles"""
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('computing phase angles for {}'.format(f), level=2, verbose=verbose)
             data = zload(f)
             data = phase_angle_data(data, ch)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('phase angle computation complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def continuous_relative_phase(self, ch_prox, ch_dist, out_folder=None, inplace=None):
         """ computes CRP angles"""
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('computing CRP angles between channel {} (prox) and {} (dist) for {}'.format(ch_prox, ch_dist, f), level=2, verbose=verbose)
             data = zload(f)
             data = continuous_relative_phase_data(data, ch_dist, ch_prox)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('CRP computation complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def split_trial_by_gait_cycle(self, first_event_name, out_folder=None, inplace=None):
         """ split by gait cycle according to event_name"""
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             f_name = os.path.splitext(os.path.basename(f))[0]
-            batchdisp('splitting by gait cycle  for {} by {}'.format(f, first_event_name), level=2, verbose=verbose)
             data = zload(f)
             split_events = get_split_events(data, first_event_name)
             if split_events is None:
                 print('no event {} found, saving original file'.format(first_event_name))
-                zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
+                zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
             else:
                 for i, _ in enumerate(split_events[0:-1]):
                     fl_new = f.replace(f_name, f_name + '_' + str(i + 1))
                     start = split_events[i]
                     end = split_events[i + 1]
+                    batchdisp('splitting by gait cycle from {} to {} for {}'.format(start, end, f), level=2,
+                              verbose=verbose)
                     data_new = split_trial(data, start, end)
-                    zsave(fl_new, data_new, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-
-        batchdisp('splitting by gait cycle complete', level=1, verbose=verbose)
+                    zsave(fl_new, data_new, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
@@ -195,130 +229,137 @@ class BiomechZoo:
     #     self._update_folder(out_folder, inplace, in_folder)
     def renameevent(self, evt, nevt, out_folder=None, inplace=None):
         """ renames event evt to nevt in all zoo files """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('renaming events from {} to {} for {}'.format(evt, nevt ,f), level=2, verbose=verbose)
             data = zload(f)
             data = renameevent_data(data, evt, nevt)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('rename event complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def renamechannnel(self, ch, ch_new, out_folder=None, inplace=None):
         """ renames channels from ch to ch_new in all zoo files """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('renaming channels from {} to {} for {}'.format(ch, ch_new ,f), level=2, verbose=verbose)
             data = zload(f)
             data = renamechannel_data(data, ch, ch_new)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('rename channels complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def removechannel(self, ch, mode='remove', out_folder=None, inplace=None):
         """ removes channels from zoo files """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             batchdisp('removing channels for {}'.format(f), level=2, verbose=verbose)
             data = zload(f)
             data = removechannel_data(data, ch, mode)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('remove channel complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def explodechannel(self, out_folder=None, inplace=None):
         """ explodes all channels in a zoo file """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('removing channels for {}'.format(f), level=2, verbose=verbose)
             data = zload(f)
             data = explodechannel_data(data)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('explode channel complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def normalize(self, nlen=101, out_folder=None, inplace=None):
         """ time normalizes all channels to length nlen """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('normalizing channels to length {} for {}'.format(nlen, f), level=2, verbose=verbose)
             data = zload(f)
             data = normalize_data(data, nlen)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('normalization complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def addevent(self, ch, evt_type, evt_name, out_folder=None, inplace=None):
         """ adds events of type evt_type with name evt_name to channel ch """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('adding event {} to channel {} for {}'.format(evt_type, ch, f), level=2, verbose=verbose)
             data = zload(f)
             data = addevent_data(data, ch, evt_type, evt_name)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('add event complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
 
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
     def partition(self, evt_start, evt_end, out_folder=None, inplace=None):
         """ partitions data between events evt_start and evt_end """
+        start_time = time.time()
         verbose = self.verbose
         in_folder = self.in_folder
         if inplace is None:
             inplace = self.inplace
-
-        fl = engine(in_folder)
+        fl = engine(in_folder, extension='.zoo', name_contains=self.name_contains, subfolders=self.subfolders)
         for f in fl:
             if verbose:
                 batchdisp('partitioning data between events {} and {} for {}'.format(evt_start, evt_end, f), level=2, verbose=verbose)
             data = zload(f)
             data = partition_data(data, evt_start, evt_end)
-            zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        batchdisp('partition complete', level=1, verbose=verbose)
+            zsave(f, data, inplace=inplace, out_folder=out_folder, root_folder=in_folder)
+        method_name = inspect.currentframe().f_code.co_name
+        batchdisp('{} process complete for {} file(s) in {:.2f} secs'.format(method_name, len(fl), time.time() - start_time), level=1, verbose=verbose)
         # Update self.folder after  processing
         self._update_folder(out_folder, inplace, in_folder)
 
@@ -340,8 +381,9 @@ class BiomechZoo:
         #     batchdisp('filtering data in channels {} for {}'.format(ch, f), level=2, verbose=verbose)
         #     data = zload(f)
         #     data = filter_data(data, ch, filt)
-        #     zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder)
-        # batchdisp('filter data complete', level=1, verbose=verbose)
+        #     zsave(f, data, inplace=inplace, root_folder=in_folder, out_folder=out_folder, verbose=verbose)
+        # method_name = inspect.currentframe().f_code.co_name
+        # batchdisp('{} computation complete for {} file(s)'.format(method_name, len(fl)), level=1, verbose=verbose)
         #
         # # Update self.folder after  processing
         # self._update_folder(out_folder, inplace, in_folder)

biomechzoo/conversion/csv2zoo_data.py

@@ -5,25 +5,32 @@ import re
 from biomechzoo.utils.compute_sampling_rate_from_time import compute_sampling_rate_from_time
 
 
-def csv2zoo_data(csv_path, header_len=10):
+def csv2zoo_data(csv_path, type='csv',skip_rows=0, freq=None):
+    # todo: check calculation for sampling rate
 
     # Read header lines until 'endheader'
-    header_lines = []
-    with open(csv_path, 'r') as f:
-        for line in f:
-            header_lines.append(line.strip())
-            if line.strip().lower() == 'endheader':
-                break
-
-    # Parse metadata
-    metadata = _parse_metadata(header_lines)
-
-    # Step 3: Load data
-    df = pd.read_csv(csv_path, skiprows=header_len)
-    time = df.iloc[:, 0].values  # first column is Time
-    data = df.iloc[:, 1:]
-
-    # S Assemble zoo data
+    metadata = {}
+    if type == 'csv' and skip_rows > 0:
+        header_lines = []
+        with open(csv_path, 'r') as f:
+            for line in f:
+                header_lines.append(line.strip())
+                if line.strip().lower() == 'endheader':
+                    break
+        # Parse metadata
+        metadata = _parse_metadata(header_lines)
+
+    if type == 'csv':
+        df = pd.read_csv(csv_path, skiprows=skip_rows)
+    elif type =='parquet':
+        df = pd.read_parquet(csv_path)
+    else:
+        raise ValueError('type must be csv or parquet')
+
+    # Use all columns
+    data = df.iloc[:, 0:]
+
+    # assemble zoo data
     zoo_data = {}
     for ch in data.columns:
         zoo_data[ch] = {
@@ -31,13 +38,26 @@ def csv2zoo_data(csv_path, header_len=10):
             'event': []
         }
 
-    # compute sampling rate
-    fsamp = compute_sampling_rate_from_time(time)
+    # try to find frequency in metadata
+    if freq is None:
+        if 'freq' in metadata:
+            freq = metadata['freq']
+        elif 'sampling_rate' in metadata:
+            freq = metadata['sampling_rate']
+        else:
+            freq = None  # or raise an error
+
+    # now try to calculate freq from a time column
+    if freq is None:
+        time_col = [col for col in df.columns if 'time' in col.lower()]
+        if time_col is not None:
+            time_data = df[time_col].to_numpy()[:,0]
+            freq = compute_sampling_rate_from_time(time_data)
 
     # add metadata
     # todo update zoosystem to match biomechzoo requirements
     zoo_data['zoosystem'] = metadata
-    zoo_data['zoosystem']['Freq'] = fsamp
+    zoo_data['zoosystem']['Freq'] = freq
 
     return zoo_data
 

biomechzoo/processing/explodechannel_data.py

@@ -1,3 +1,5 @@
+import copy
+
 def explodechannel_data(data, channels=None):
     """ Explodes 3D channels (n x 3 arrays) into separate X, Y, Z channels.
 
@@ -10,8 +12,7 @@ def explodechannel_data(data, channels=None):
         data_new (dict): Modified zoo dictionary with exploded channels.
     """
 
-    data_new = data.copy()
-
+    data_new = copy.deepcopy(data)
     # Find default channels if none provided
     if channels is None:
         channels = []

biomechzoo/processing/partition_data.py

@@ -1,19 +1,20 @@
 from biomechzoo.utils.findfield import findfield
 import warnings
+import copy
 
 
 def partition_data(data, evt_start, evt_end):
     """ partition data for all channels between events evt_start and evt_end"""
 
     # extract event values
-    e1, _ = findfield(data,evt_start)
-    e2, _ = findfield(data,evt_end)
+    e1, _ = findfield(data, evt_start)
+    e2, _ = findfield(data, evt_end)
 
-    data_new = data.copy()
+    data_new = copy.deepcopy(data)
     for ch_name, ch_data in data_new.items():
         if ch_name != 'zoosystem':
             try:
-                data_new[ch_name]['line'] = ch_data[e1[0]:e2[0], :]
+                data_new[ch_name]['line'] = ch_data[e1[0]:e2[0],]
             except (IndexError, ValueError) as e:
                 # IndexError: if e1[0]:e2[0] goes beyond the available indices
                 # ValueError: less likely, but may arise with shape mismatches
@@ -29,6 +30,7 @@ def partition_data(data, evt_start, evt_end):
                     continue  # do not change outlier markers
                 else:
                     new_frame = original_frame - e1[0] + 1
+                    print(new_frame)
                     data_new[ch_name]['event'][event_name][0] = new_frame
 
     return data_new
@@ -48,4 +50,4 @@ def _partition_event(event_dict, evt_start, evt_end, arr_len):
     #
     #
     #
-    # return event_dict_new
+    # return event_dict_new

biomechzoo/utils/engine.py

@@ -19,7 +19,7 @@ def engine(root_folder, extension='.zoo', subfolders=None, name_contains=None, v
     Arguments:
         root_folder (str): The root directory path where the search begins.
         extension (str): File extension to search for (e.g., '.zoo', '.c3d'). Default .zoo
-        subfolders (list of str, optional): List of folder names to restrict the search to.
+        subfolders (list or str, optional): List of folder names to restrict the search to.
             Only files inside these folders (or their subfolders) are included.
             If None, search all subfolders.
         name_contains (str, optional): Substring that must be present in the filename
@@ -28,6 +28,11 @@ def engine(root_folder, extension='.zoo', subfolders=None, name_contains=None, v
     Returns:
         list of str: List of full file paths matching the criteria.
     """
+    # check format of subfolder (string or list)
+    if subfolders is not None:
+        if type(subfolders) is str:
+            subfolders = [subfolders]
+
     matched_files = []
 
     subfolders_set = set(subfolders) if subfolders else None

biomechzoo/utils/zsave.py

@@ -2,8 +2,10 @@ from scipy.io import savemat
 import inspect
 import os
 
+from biomechzoo.utils.batchdisp import batchdisp
 
-def zsave(fl, data, inplace=True, out_folder=None, root_folder=None):
+
+def zsave(fl, data, inplace=True, out_folder=None, root_folder=None, verbose=False):
     """
     Save zoo data to .zoo file (MAT format)
 
@@ -28,9 +30,9 @@ def zsave(fl, data, inplace=True, out_folder=None, root_folder=None):
 
     # Determine save path
     if inplace:
-        save_path = fl
+        fl_new = fl
+        out_dir = os.path.dirname(fl)
     else:
-        filename = os.path.basename(fl)
         if out_folder is None:
             out_folder = 'processed'
 
@@ -47,4 +49,6 @@ def zsave(fl, data, inplace=True, out_folder=None, root_folder=None):
 
     # Save the .zoo file
     savemat(fl_new, data)
+    batchdisp('all files saved to ' + out_dir, level=1, verbose=verbose)
+
 

{biomechzoo-0.4.1.dist-info → biomechzoo-0.4.5.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: biomechzoo
-Version: 0.4.1
+Version: 0.4.5
 Summary: Python implementation of the biomechZoo toolbox
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/mcgillmotionlab/biomechzoo
-Requires-Python: >=3.11
+Requires-Python: <3.12,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ezc3d>=1.5.19
@@ -12,6 +12,7 @@ Requires-Dist: matplotlib>=3.10.6
 Requires-Dist: numpy==2.2.6
 Requires-Dist: pandas>=2.3.2
 Requires-Dist: scipy>=1.16.2
+Requires-Dist: pyarrow>=19.0.0
 Dynamic: license-file
 
 # BiomechZoo for Python
@@ -30,3 +31,15 @@ This is a development version of the biomechzoo toolbox for python.
 - We use Numpy 2.2.6 for compatibility with https://pypi.org/project/numba/
 
 See also http://www.github.com/mcgillmotionlab/biomechzoo or http://www.biomechzoo.com for more information
+
+## Developer notes
+
+### Installing a dev environment
+conda create -n biomechzoo-dev python=3.11
+conda activate biomechzoo-dev
+pip install -e ".[dev]"
+
+### import issues
+if using PyCharm: 
+- Right-click on src/.
+- Select Mark Directory as → Sources Root.

{biomechzoo-0.4.1.dist-info → biomechzoo-0.4.5.dist-info}/RECORD

@@ -1,6 +1,7 @@
+__init__.py,sha256=Uy3ykqw4l_lZKiUWSUFPRZpkZajYUfZLBaQLVhKzxdI,772
 biomechzoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biomechzoo/__main__.py,sha256=hSMHN1Rxn2367fSGTLHoOQ4_pocZw0IWI7HFxl-74oY,88
-biomechzoo/biomechzoo.py,sha256=bmP81dLlYPE2NnZ76xK-MVlVyDuKKU08OF58dtzieps,15433
+biomechzoo/biomechzoo.py,sha256=6YpqhUaMH76d9zjNAsnOvyU5zppbCMD1t72FIzA8VyI,20782
 biomechzoo/biomech_ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biomechzoo/biomech_ops/continuous_relative_phase_data.py,sha256=RePbt6zyOI1iv74CWhxSrunIokTFYVfFmFnoW51781E,1300
 biomechzoo/biomech_ops/continuous_relative_phase_line.py,sha256=Fa1LFRuPlmGPLQLvln6HVnJy3zMSm9z5YeooHmTu0lc,1365
@@ -12,7 +13,7 @@ biomechzoo/biomech_ops/phase_angle_data.py,sha256=_ekUBW2v3iC4UawcDL38ZZLYJmQsAm
 biomechzoo/biomech_ops/phase_angle_line.py,sha256=p6osB17_3QQSyKLNojuc6nYhv-k0K6EUUH75EXu8ifc,1391
 biomechzoo/conversion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biomechzoo/conversion/c3d2zoo_data.py,sha256=kmiLQHglchSHty6DrMEl5yTOUi3U4Ai5nHoSGvbF6k8,2660
-biomechzoo/conversion/csv2zoo_data.py,sha256=-c4CXxcKd70HJWWgHRztOxOckvA1XgIDzo8EhgJhcKE,2185
+biomechzoo/conversion/csv2zoo_data.py,sha256=hNslzga9Iy3pnD_cMiHZuPupjAz-Qb1VwGSJ5rm3uDM,2928
 biomechzoo/conversion/mvnx2zoo_data.py,sha256=tlqg8iVmTxQPo9u9T8mghS2eAEtqOF3xB7tTNR0H1mQ,3504
 biomechzoo/conversion/opencap2zoo_data.py,sha256=n4bLsrJI0wTSzG5bgJcmxj1dp2plnUKFNRzcTIbmV1o,608
 biomechzoo/mvn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,8 +25,8 @@ biomechzoo/processing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 biomechzoo/processing/add_channel_data.py,sha256=U1xLLBSnyJeeDWzgmHSxOz1hyguzuuDXxQCQ8IFoZkw,1955
 biomechzoo/processing/addchannel_data.py,sha256=rmnnlMRVkoMlQCR-nRg1jjh-hzMDt37Sx9fAmocrpqA,1953
 biomechzoo/processing/addevent_data.py,sha256=DyTfvjOWZIlsWfnT6XDwvsU1jOzSDf4HhLa1TaKIhwI,1325
-biomechzoo/processing/explodechannel_data.py,sha256=t4gC2ZQD4baECexj0hQYDzOnXV5GHb-z068mtZzh0DQ,1470
-biomechzoo/processing/partition_data.py,sha256=8pMGRVXzYbPa4tuB7K3OOjUCnROpXb2v1gcRRNAbSig,1704
+biomechzoo/processing/explodechannel_data.py,sha256=AC2BOEw1tXcgJ1WuYWSE-yToS-q9XGdgkOHS4D3iUFo,1490
+biomechzoo/processing/partition_data.py,sha256=8miAL8kzOSJtg6j6WetAxRv8whJD8gtDVR9FEpj6mo8,1760
 biomechzoo/processing/removechannel_data.py,sha256=ndZcbWJRDvd7drlahGaq8MseT-rsxiu7pgdMtA1cTlo,1218
 biomechzoo/processing/renamechannel_data.py,sha256=5lEnWJknAwnJYXDlwO6cFe7C8ct5o42tTHW0Y4GeIL4,2210
 biomechzoo/processing/renameevent_data.py,sha256=9w7C_fQOsQ8XbdTr_hrg_iQe51oDczq2Rj7yJLyYG0M,2215
@@ -33,16 +34,16 @@ biomechzoo/processing/split_trial_by_gait_cycle.py,sha256=maMUwumSlFLFc5LHdNdBO9
 biomechzoo/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biomechzoo/utils/batchdisp.py,sha256=Ke7yeZ4cYQvyf8bmgsiLaRElUeQm49TYIYzAcPWan9s,839
 biomechzoo/utils/compute_sampling_rate_from_time.py,sha256=iyHbN734vYe5bXEoAsHScJN7Qw--2quUgII7judSQDk,579
-biomechzoo/utils/engine.py,sha256=WSZmi6DDZHcdOhgsu2LvmLF-Ftxcjteh9TbKDMZIN5k,3240
+biomechzoo/utils/engine.py,sha256=7Ztf1PbjnEn4C6jueJ8AhQG3W6OAWS2Ze5CDiF7Xu_I,3395
 biomechzoo/utils/findfield.py,sha256=i7ewGQOMIiTC06tYFK-JctehHLCippkY9FoXIygx14w,381
 biomechzoo/utils/get_split_events.py,sha256=xNhEuG6Yqsr1bjWIBHLbepfX-fcqcCYIXZzS3eaDDHQ,911
 biomechzoo/utils/split_trial.py,sha256=Fumz_ZukNBNtPauUhCge5EAHkg05dYDhA1_njQw0lHE,886
 biomechzoo/utils/zload.py,sha256=FPT6_-gwaOOqOckjgPRfnKEVKMsmNVIcenmQZF2KOvo,1535
 biomechzoo/utils/zplot.py,sha256=WVA8aCy1Pqy_bo_HXab9AmW-cBd8J8MPX2LAOd6dznU,1512
-biomechzoo/utils/zsave.py,sha256=2DeF6A1R32SnXJXtb8jtPqZAmlSEB0PVi09UGiskzMo,1625
-biomechzoo-0.4.1.dist-info/licenses/LICENSE,sha256=Fsz62nrgRORre3A1wNXUDISaHoostodMvocRPDdXc9w,1076
-biomechzoo-0.4.1.dist-info/METADATA,sha256=4wd_Ax4xnp2kUdrPrQEeprveTWZfv7a7lYfhN76n3NA,1259
-biomechzoo-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-biomechzoo-0.4.1.dist-info/entry_points.txt,sha256=VdryUUiwwvx0WZxrgmMrsyfe5Z1jtyaxdXOi0zWHOqk,41
-biomechzoo-0.4.1.dist-info/top_level.txt,sha256=W148kdiK-6L0Nohbw8UALkAzHJvJLAhFDRykuva2WjY,11
-biomechzoo-0.4.1.dist-info/RECORD,,
+biomechzoo/utils/zsave.py,sha256=wnRNuDxQc8bwCji4UrfoGjYrSZmka4eaDxQ5rMa78pE,1759
+biomechzoo-0.4.5.dist-info/licenses/LICENSE,sha256=Fsz62nrgRORre3A1wNXUDISaHoostodMvocRPDdXc9w,1076
+biomechzoo-0.4.5.dist-info/METADATA,sha256=tKqy3v7eordmf6uD0bCDHqXtoLBGHAu8rDrIjqhxJWs,1553
+biomechzoo-0.4.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+biomechzoo-0.4.5.dist-info/entry_points.txt,sha256=VdryUUiwwvx0WZxrgmMrsyfe5Z1jtyaxdXOi0zWHOqk,41
+biomechzoo-0.4.5.dist-info/top_level.txt,sha256=nJEtuEZ9UPoN3EOR-BJ6myevEu7B5quWsWhaM_YeQpw,20
+biomechzoo-0.4.5.dist-info/RECORD,,

{biomechzoo-0.4.1.dist-info → biomechzoo-0.4.5.dist-info}/top_level.txt

@@ -1 +1,2 @@
+__init__
 biomechzoo