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(
