from st_generated_axial_rot.common.analysis_utils import prepare_db, extract_sub_rot_norm from st_generated_axial_rot.common.json_utils import get_params from st_generated_axial_rot.common.arg_parser import mod_arg_parser import logging from logging.config import fileConfig config_dir = Path( mod_arg_parser( 'Compare HT, ST, GH axial rotation for elevation trials by gender', __package__, __file__)) params = get_params(config_dir / 'parameters.json') if not bool(distutils.util.strtobool(os.getenv('VARS_RETAINED', 'False'))): # ready db db = create_db(params.biplane_vicon_db_dir, BiplaneViconSubject, include_anthro=True) db['age_group'] = db['Age'].map(lambda age: '<35' if age < 40 else '>45') if params.excluded_trials: db = db[~db['Trial_Name'].str.contains('|'.join(params. excluded_trials))] db['Trial'].apply(pre_fetch) # relevant parameters output_path = Path(params.output_dir) # logging fileConfig(config_dir / 'logging.ini', disable_existing_loggers=False) log = logging.getLogger(params.logger_name)
import numpy as np import matplotlib.pyplot as plt from pathlib import Path from st_generated_axial_rot.common.database import create_db, BiplaneViconSubject from st_generated_axial_rot.common.analysis_utils import prepare_db from st_generated_axial_rot.analysis.up_down_identify import extract_up_down_min_max from st_generated_axial_rot.common import plot_utils from st_generated_axial_rot.common.json_utils import get_params from st_generated_axial_rot.common.arg_parser import mod_arg_parser from logging.config import fileConfig import logging config_dir = Path( mod_arg_parser('Check U35_002_SA_t01 filling', __package__, __file__)) params = get_params(config_dir / 'parameters.json') db = create_db(params.biplane_vicon_db_dir, BiplaneViconSubject) db = db.loc[['U35_002_SA_t01']] # logging fileConfig(config_dir / 'logging.ini', disable_existing_loggers=False) log = logging.getLogger(params.logger_name) # compute min and max ht elevation for each subject prepare_db(db, params.torso_def, params.scap_lateral, params.dtheta_fine, params.dtheta_coarse, [params.min_elev, params.max_elev], should_clean=False) (db['up_min_ht'], db['up_max_ht'], db['down_min_ht'], db['down_max_ht']) = zip(