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

biomechzoo/conversion/c3d2zoo_data.py

@@ -1,3 +1,5 @@
+from biomechzoo.utils.set_zoosystem import set_zoosystem
+
 def c3d2zoo_data(c3d_obj):
     """
     Converts an ezc3d C3D object to zoo format.
@@ -21,7 +23,7 @@ def c3d2zoo_data(c3d_obj):
         video_freq = c3d_obj['parameters']['POINT']['RATE']['value'][0]
         if 'EVENT' in params and 'TIMES' in params['EVENT']:
             times_array = params['EVENT']['TIMES']['value']
-            frames = times_array[1]  # second row = frames (or time, depending on C3D file)
+            frames = times_array[1]  # should be time depending on C3D file
 
             # Extract sides, types, subjects
             contexts = params['EVENT']['CONTEXTS']['value'] if 'CONTEXTS' in params['EVENT'] else ['']
@@ -55,11 +57,12 @@ def c3d2zoo_data(c3d_obj):
 
                     # Place in correct channel
                     if 'SACR' in data:
-                        data['SACR']['event'][key_name] = [frame, 0, 0]
+                        data['SACR']['event'][key_name] = [frame-1, 0, 0]       # remove 1 to follow python
                     else:
-                        data[labels[0]]['event'][key_name] = [frame, 0, 0]
+                        data[labels[0]]['event'][key_name] = [frame-1, 0, 0]    # remove 1 to follow python
 
     # todo add relevant meta data to zoosystem
-    data['zoosystem'] = params['EVENT']
+    data['zoosystem'] = set_zoosystem()
+    data['zoosystem']['Analog']['Freq'] = int(params['ANALOG']['RATE']['value'][0])
 
     return data

biomechzoo/conversion/csv2zoo_data.py

@@ -2,10 +2,11 @@ import pandas as pd
 import os
 import re
 
+from biomechzoo.utils.set_zoosystem import set_zoosystem
 from biomechzoo.utils.compute_sampling_rate_from_time import compute_sampling_rate_from_time
 
 
-def csv2zoo_data(csv_path, type='csv',skip_rows=0, freq=None):
+def csv2zoo_data(csv_path, type='csv', skip_rows=0, freq=None):
     # todo: check calculation for sampling rate
 
     # Read header lines until 'endheader'
@@ -51,13 +52,16 @@ def csv2zoo_data(csv_path, type='csv',skip_rows=0, freq=None):
     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]
+            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'] = freq
+    zoo_data['zoosystem'] = set_zoosystem(csv_path)
+    zoo_data['zoosystem']['Video']['Freq'] = freq
+    zoo_data['zoosystem']['Analog']['Freq'] = 'None'
+    if 'version' in metadata:
+        zoo_data['zoosystem']['collection_system_version'] = metadata['version']
 
     return zoo_data
 

biomechzoo/conversion/mvnx2zoo_data.py

@@ -1,6 +1,7 @@
 import numpy as np
 from biomechzoo.mvn.load_mvnx import load_mvnx
 from biomechzoo.mvn.mvn import JOINTS, SEGMENTS
+from biomechzoo.utils.set_zoosystem import set_zoosystem
 
 def mvnx2zoo_data(fl):
     """ loads mvnx file from xsens"""
@@ -37,10 +38,11 @@ def mvnx2zoo_data(fl):
     data = _get_foot_strike_events(mvnx_file, data)
 
     # add meta information
-    data = _get_meta_info(mvnx_file, data)
+    data = _get_meta_info(fl, mvnx_file, data)
 
     return data
 
+
 def is_valid_for_zoo(val):
     """
     Returns True if the value is valid for a MATLAB-compatible zoo structure.
@@ -53,13 +55,13 @@ def is_valid_for_zoo(val):
         return False
     return True
 
-def _get_meta_info(mvnx_file, data):
+
+def _get_meta_info(fl, mvnx_file, data):
     # todo: add more, see mvnx_file object
-    data['zoosystem'] = {}
-    data['zoosystem']['Video'] = {}
+    data['zoosystem'] = set_zoosystem(fl)
     data['zoosystem']['Video']['Freq'] = int(mvnx_file.frame_rate)
-    data['zoosystem']['Version'] = mvnx_file.version
-    data['zoosystem']['configuration'] = mvnx_file.configuration
+    data['zoosystem']['mvnx_version'] = mvnx_file.version
+    data['zoosystem']['mvnx_configuration'] = mvnx_file.configuration
     data['zoosystem']['recording_date'] = mvnx_file.recording_date
     data['zoosystem']['original_file_name'] = mvnx_file.original_file_name
     data['zoosystem']['frame_count'] = mvnx_file.frame_count

biomechzoo/processing/partition_data.py

@@ -10,11 +10,19 @@ def partition_data(data, evt_start, evt_end):
     e1, _ = findfield(data, evt_start)
     e2, _ = findfield(data, evt_end)
 
+    if e1 is None or e2 is None or len(e1) == 0 or len(e2) == 0:
+        raise ValueError(f"Event not found: evt_start='{evt_start}' returned {e1}, evt_end='{evt_end}' returned {e2}")
+
     data_new = copy.deepcopy(data)
     for ch_name, ch_data in data_new.items():
         if ch_name != 'zoosystem':
+            print(ch_name)
+            line = ch_data['line']
             try:
-                data_new[ch_name]['line'] = ch_data[e1[0]:e2[0],]
+                if line.ndim == 1:
+                    data_new[ch_name]['line'] = line[e1[0]:e2[0]]
+                else:
+                    data_new[ch_name]['line'] = line[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
@@ -24,9 +32,7 @@ def partition_data(data, evt_start, evt_end):
             events = ch_data['event']
             for event_name, value in events.items():
                 original_frame = value[0]
-                if original_frame == 1:
-                    continue  # leave index 1 as is (per MATLAB version)
-                elif original_frame == 999:
+                if original_frame == 999:
                     continue  # do not change outlier markers
                 else:
                     new_frame = original_frame - e1[0] + 1

biomechzoo/utils/engine.py

@@ -22,7 +22,7 @@ def engine(root_folder, extension='.zoo', subfolders=None, name_contains=None, v
         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
+        name_contains (str, or list; optional): Substring that must be present in the filename
             (case-insensitive). If None, no substring filtering is applied.
         verbose (bool, optional): If true, displays additional information to user
     Returns:
@@ -33,10 +33,14 @@ def engine(root_folder, extension='.zoo', subfolders=None, name_contains=None, v
         if type(subfolders) is str:
             subfolders = [subfolders]
 
+    # check format of name_contants (str or list)
+    if name_contains is not None:
+        if type(name_contains) is str:
+            name_contains = [name_contains]
+
     matched_files = []
 
     subfolders_set = set(subfolders) if subfolders else None
-
     for dirpath, _, filenames in os.walk(root_folder):
         if subfolders_set is not None:
             rel_path = os.path.relpath(dirpath, root_folder)
@@ -51,9 +55,16 @@ def engine(root_folder, extension='.zoo', subfolders=None, name_contains=None, v
         for file in filenames:
             if not file.lower().endswith(extension.lower()):
                 continue
-            if name_contains and name_contains.lower() not in file.lower():
-                continue
-            matched_files.append(os.path.join(dirpath, file))
+            full_path = os.path.join(dirpath, file)
+            if name_contains is not None:
+                match = False
+                for name_contain in name_contains:
+                    if name_contain and name_contain.lower() in full_path.lower():  # <-- check full path
+                        match = True
+                        break
+                if not match:
+                    continue
+            matched_files.append(full_path)
 
     if verbose:
         print("Found {} {} files in subfolders named {} with substring {}:".format(len(matched_files), extension, subfolders, name_contains))

biomechzoo/utils/set_zoosystem.py

@@ -0,0 +1,66 @@
+import numpy as np
+from pathlib import Path
+from biomechzoo.utils.version import get_biomechzoo_version
+
+
+def set_zoosystem(fl=None):
+    """
+    Create the 'zoosystem' branch for data being imported to BiomechZoo.
+
+    Args:
+        fl: str
+            Path to the source file (e.g., C3D or CSV).
+
+    Returns:
+        zoosystem: dict
+            Dictionary containing default BiomechZoo system parameters.
+    """
+
+    # Default top-level fields
+    zch = ['Analog', 'Anthro', 'AVR', 'CompInfo', 'SourceFile',
+           'Units', 'Version', 'Video']
+
+    # Initialize top-level dict
+    zoosystem = {key: {} for key in zch}
+
+    # Section-specific defaults
+    section = ['Video', 'Analog']
+    for sec in section:
+        zoosystem[sec]['Channels'] = []
+        zoosystem[sec]['Freq'] = []
+        zoosystem[sec]['Indx'] = []
+        zoosystem[sec]['ORIGINAL_START_FRAME'] = []
+        zoosystem[sec]['ORIGINAL_END_FRAME'] = []
+        zoosystem[sec]['CURRENT_START_FRAME'] = []
+        zoosystem[sec]['CURRENT_END_FRAME'] = []
+
+    # Processing and AVR defaults
+    zoosystem['Processing'] = ''
+    zoosystem['AVR'] = 0
+
+    # Force plates defaults
+    zoosystem['Analog']['FPlates'] = {
+        'CORNERS': [],
+        'NUMUSED': 0,
+        'LOCALORIGIN': [],
+        'LABELS': []
+    }
+
+    # Version and source file
+    zoosystem['Version'] = get_biomechzoo_version()
+    if fl is None:
+        zoosystem['SourceFile'] = 'None'  # ensure string
+    else:
+        zoosystem['SourceFile'] = str(Path(fl))  # ensure string
+
+    # Units defaults
+    zoosystem['Units'] = {
+        'Markers': 'mm',
+        'Angles': 'deg',
+        'Forces': 'N',
+        'Moments': 'Nmm',
+        'Power': 'W/kg',
+        'Scalars': 'mm'
+    }
+
+    return zoosystem

biomechzoo/utils/version.py

@@ -0,0 +1,5 @@
+from importlib.metadata import version as get_version
+
+
+def get_biomechzoo_version() -> str:
+    return get_version("biomechzoo")

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: biomechzoo
-Version: 0.4.5
+Version: 0.4.8
 Summary: Python implementation of the biomechZoo toolbox
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/mcgillmotionlab/biomechzoo

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

@@ -12,9 +12,9 @@ biomechzoo/biomech_ops/normalize_line.py,sha256=KUE8gEkIolA-VDFCdUuaskk-waO8jjJ2
 biomechzoo/biomech_ops/phase_angle_data.py,sha256=_ekUBW2v3iC4UawcDL38ZZLYJmQsAmyqA61Q6_AMtmQ,1435
 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=hNslzga9Iy3pnD_cMiHZuPupjAz-Qb1VwGSJ5rm3uDM,2928
-biomechzoo/conversion/mvnx2zoo_data.py,sha256=tlqg8iVmTxQPo9u9T8mghS2eAEtqOF3xB7tTNR0H1mQ,3504
+biomechzoo/conversion/c3d2zoo_data.py,sha256=gnSCw99Hg5JLdrGezZNVZ6EHqiD6i-mpsLRtWmEa-Mg,2855
+biomechzoo/conversion/csv2zoo_data.py,sha256=9FjlJjCSdT9d5n10Kc74hq5sXBmUdRA8CSIMWlPAk4o,3175
+biomechzoo/conversion/mvnx2zoo_data.py,sha256=O6WuKe_5fZJfd1gUe3oP9rND_i5P4WGEuu2MDvI3ypY,3560
 biomechzoo/conversion/opencap2zoo_data.py,sha256=n4bLsrJI0wTSzG5bgJcmxj1dp2plnUKFNRzcTIbmV1o,608
 biomechzoo/mvn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biomechzoo/mvn/load_mvnx.py,sha256=B4VGuidQ-5G_5E1t5vpU51Nb_Lu85_EOS_FmGZYjfX8,22499
@@ -26,7 +26,7 @@ biomechzoo/processing/add_channel_data.py,sha256=U1xLLBSnyJeeDWzgmHSxOz1hyguzuuD
 biomechzoo/processing/addchannel_data.py,sha256=rmnnlMRVkoMlQCR-nRg1jjh-hzMDt37Sx9fAmocrpqA,1953
 biomechzoo/processing/addevent_data.py,sha256=DyTfvjOWZIlsWfnT6XDwvsU1jOzSDf4HhLa1TaKIhwI,1325
 biomechzoo/processing/explodechannel_data.py,sha256=AC2BOEw1tXcgJ1WuYWSE-yToS-q9XGdgkOHS4D3iUFo,1490
-biomechzoo/processing/partition_data.py,sha256=8miAL8kzOSJtg6j6WetAxRv8whJD8gtDVR9FEpj6mo8,1760
+biomechzoo/processing/partition_data.py,sha256=XF8dSqvHGpqsT-Q4i6qpoOajAT4LYjP-PJ6KbpoCfFc,2018
 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
@@ -34,16 +34,18 @@ 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=7Ztf1PbjnEn4C6jueJ8AhQG3W6OAWS2Ze5CDiF7Xu_I,3395
+biomechzoo/utils/engine.py,sha256=D91ZhevBnoWml_NztZP-1lzFoANy5TyIAnJg0FFZyUc,3859
 biomechzoo/utils/findfield.py,sha256=i7ewGQOMIiTC06tYFK-JctehHLCippkY9FoXIygx14w,381
 biomechzoo/utils/get_split_events.py,sha256=xNhEuG6Yqsr1bjWIBHLbepfX-fcqcCYIXZzS3eaDDHQ,911
+biomechzoo/utils/set_zoosystem.py,sha256=oubEc3fy0x6y-AlqQWL3v7QYJA951jU9CRnlJ9ikwQo,1750
 biomechzoo/utils/split_trial.py,sha256=Fumz_ZukNBNtPauUhCge5EAHkg05dYDhA1_njQw0lHE,886
+biomechzoo/utils/version.py,sha256=JIXDUuOcaJiZv9ruMP6PtWvJBh4sP0D5kAvlqPiZK_I,130
 biomechzoo/utils/zload.py,sha256=FPT6_-gwaOOqOckjgPRfnKEVKMsmNVIcenmQZF2KOvo,1535
 biomechzoo/utils/zplot.py,sha256=WVA8aCy1Pqy_bo_HXab9AmW-cBd8J8MPX2LAOd6dznU,1512
 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.8.dist-info/licenses/LICENSE,sha256=Fsz62nrgRORre3A1wNXUDISaHoostodMvocRPDdXc9w,1076
+biomechzoo-0.4.8.dist-info/METADATA,sha256=WllGPwbLIbr26wyfwv8ZYz_VjlCa9jskutlRuhMFV_A,1553
+biomechzoo-0.4.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+biomechzoo-0.4.8.dist-info/entry_points.txt,sha256=VdryUUiwwvx0WZxrgmMrsyfe5Z1jtyaxdXOi0zWHOqk,41
+biomechzoo-0.4.8.dist-info/top_level.txt,sha256=nJEtuEZ9UPoN3EOR-BJ6myevEu7B5quWsWhaM_YeQpw,20
+biomechzoo-0.4.8.dist-info/RECORD,,