main.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@author: EmilianoRavera
@email: emiliano.ravera@uner.edu.ar
"""
# ----------- import packages --------------

from utilities.PublicFuntions import c3d2opensim

from ezc3d import c3d
import matplotlib.pyplot as plt
import pandas as pd
import csv
import gc

import os
import glob
import shutil
from pathlib import Path

# ------------------------------------------

class MOCAPData2OpenSim:
    
    def __init__(self, root_path, path_MOCAP):
        
        self.path_MOCAP = path_MOCAP  # path of c3ds folder
        self.dname = root_path # path of the main folder where the OpenSim model and the ExperimentalData folder will be created



    def convert_c3d2opensim(self):
        # create ExperimentalData folder if required
        output_ExperimentalData = os.path.join(
            self.dname, "ExperimentalData"
        )
        dst = Path(output_ExperimentalData)
        dst.mkdir(parents=True, exist_ok=True)

        walking_files = [
            file
            for file in self.path_MOCAP
            if ("standing" not in Path(file).name.lower()
            and "std" not in Path(file).name.lower())
        ]
        standig_file = [
            file
            for file in self.path_MOCAP
            if ("standing" in Path(file).name.lower()
            or "std" in Path(file).name.lower())
        ]

        # 1) convert c3d file to trc (OpenSim format)
        if standig_file != []:
            TrialType = "Static"
            self.Static_TRCFileName = c3d2opensim(self.dname, standig_file[0], TrialType)
        else:
            print("Standing file not exist")
            return

        if walking_files != []:
            self.Dynamic_TRCFileName = []
            TrialType = "Dynamic"

            for file in walking_files:
                self.Dynamic_TRCFileName.append(c3d2opensim(self.dname, file, TrialType))
        else:
            print("Walking file not exist")
            return
        
        gc.collect()  # Forzar limpieza de memoria


if __name__ == "__main__":
    # -----------------------------------------------------------------------
    # Set to Executables directory if running in Python directly
    abspath = os.path.abspath(__file__)
    dname = os.path.dirname(abspath)
    #

    data_folder = os.path.join(dname, 'Data')
    archivos_c3d = glob.glob(os.path.join(data_folder, '*.c3d'))

    conversor = MOCAPData2OpenSim(
        root_path=dname,
        path_MOCAP=archivos_c3d,
    )
    # RUN convertion of MOCAP data to OpenSim data
    conversor.convert_c3d2opensim()