def main():
parser = argparse.ArgumentParser(description='CGM2-3 Fitting')
parser.add_argument(
'--proj',
type=str,
help='Moment Projection. Choice : Distal, Proximal, Global')
parser.add_argument('-md',
'--markerDiameter',
type=float,
help='marker diameter')
parser.add_argument('--noIk',
action='store_true',
help='cancel inverse kinematic')
parser.add_argument('-ps',
'--pointSuffix',
type=str,
help='suffix of model outputs')
parser.add_argument('--check',
action='store_true',
help='force model output suffix')
parser.add_argument('-a',
'--accuracy',
type=float,
help='Inverse Kinematics accuracy')
parser.add_argument('-ae',
'--anomalyException',
action='store_true',
help='stop if anomaly detected ')
parser.add_argument('-fi',
'--frameInit',
type=int,
help='first frame to process')
parser.add_argument('-fe',
'--frameEnd',
type=int,
help='last frame to process')
args = parser.parse_args()
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------GLOBAL SETTINGS ------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH +
"CGM2_3-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,
"CGM2_3-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,
"CGM2_3-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm(settings, args)
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm2.3")
momentProjection = argsManager.getMomentProjection()
ik_flag = argsManager.enableIKflag
ikAccuracy = argsManager.getIkAccuracy()
# ----------------------LOADING-------------------------------------------
DATA_PATH, reconstructFilenameLabelledNoExt = NEXUS.GetTrialName()
reconstructFilenameLabelled = reconstructFilenameLabelledNoExt + ".c3d"
LOGGER.logger.info("data Path: " + DATA_PATH)
LOGGER.set_file_handler(DATA_PATH + "pyCGM2-Fitting.log")
LOGGER.logger.info("calibration file: " + reconstructFilenameLabelled)
# --------------------------SUBJECT -----------------------------------
# Notice : Work with ONE subject by session
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.getActiveSubject(NEXUS)
LOGGER.logger.info("Subject name : " + subject)
# --------------------pyCGM2 MODEL ------------------------------
model = files.loadModel(DATA_PATH, subject)
# -------------------------- MP ------------------------------------
# allow alteration of thigh offset
model.mp_computed[
"LeftThighRotationOffset"] = NEXUS.GetSubjectParamDetails(
subject, "LeftThighRotation")[0]
model.mp_computed[
"RightThighRotationOffset"] = NEXUS.GetSubjectParamDetails(
subject, "RightThighRotation")[0]
# --------------------------CHECKING -----------------------------------
# check model
LOGGER.logger.info("loaded model : %s" % (model.version))
if model.version != "CGM2.3":
raise Exception(
"%s-pyCGM2.model file was not calibrated from the CGM2.3 calibration pipeline"
% subject)
# --------------------------SESSION INFOS ------------------------------------
# translators management
translators = files.getTranslators(DATA_PATH, "CGM2_3.translators")
if not translators: translators = settings["Translators"]
# ikweight
ikWeight = files.getIKweightSet(DATA_PATH, "CGM2_3.ikw")
if not ikWeight: ikWeight = settings["Fitting"]["Weight"]
#force plate assignement from Nexus
mfpa = nexusTools.getForcePlateAssignment(NEXUS)
# btkAcquisition
nacf = nexusFilters.NexusConstructAcquisitionFilter(
DATA_PATH, reconstructFilenameLabelledNoExt, subject)
acq = nacf.build()
# --------------------------MODELLING PROCESSING -----------------------
finalAcqGait, detectAnomaly = cgm2_3.fitting(
model,
DATA_PATH,
reconstructFilenameLabelled,
translators,
settings,
ik_flag,
markerDiameter,
pointSuffix,
mfpa,
momentProjection,
forceBtkAcq=acq,
ikAccuracy=ikAccuracy,
anomalyException=args.anomalyException,
frameInit=args.frameInit,
frameEnd=args.frameEnd)
# ----------------------DISPLAY ON VICON-------------------------------
nexusFilters.NexusModelFilter(NEXUS, model, finalAcqGait, subject,
pointSuffix).run()
nexusTools.createGeneralEvents(NEXUS, subject, finalAcqGait,
["Left-FP", "Right-FP"])
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
model, acqStatic = cgm2_1.calibrate(DATA_PATH,
calibrateFilenameLabelled,
translators, required_mp,
optional_mp, leftFlatFoot,
rightFlatFoot, markerDiameter,
hjcMethod, pointSuffix)
# ----------------------SAVE-------------------------------------------
#pyCGM2.model
files.saveModel(model, DATA_PATH, subject)
# save mp
files.saveMp(mpInfo, model, DATA_PATH, mpFilename)
# ----------------------DISPLAY ON VICON-------------------------------
nexusUtils.updateNexusSubjectMp(NEXUS, model, subject)
nexusFilters.NexusModelFilter(NEXUS,
model,
acqStatic,
subject,
pointSuffix,
staticProcessing=True).run()
# ========END of the nexus OPERATION if run from Nexus =========
if args.DEBUG:
NEXUS.SaveTrial(30)
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main():
parser = argparse.ArgumentParser(description='CGM2.4 Calibration')
parser.add_argument('-l','--leftFlatFoot', type=int, help='left flat foot option')
parser.add_argument('-r','--rightFlatFoot',type=int, help='right flat foot option')
parser.add_argument('-hf','--headFlat',type=int, help='head flat option')
parser.add_argument('-md','--markerDiameter', type=float, help='marker diameter')
parser.add_argument('-ps','--pointSuffix', type=str, help='suffix of model outputs')
parser.add_argument('--check', action='store_true', help='force model output suffix')
parser.add_argument('--noIk', action='store_true', help='cancel inverse kinematic')
parser.add_argument('--resetMP', action='store_true', help='reset optional mass parameters')
parser.add_argument('--forceLHJC', nargs='+')
parser.add_argument('--forceRHJC', nargs='+')
parser.add_argument('-ae','--anomalyException', action='store_true', help='stop if anomaly detected ')
args = parser.parse_args()
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------GLOBAL SETTINGS ------------------------------
# ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH + "CGM2_4-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,"CGM2_4-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,"CGM2_4-pyCGM2.settings")
argsManager = CgmArgsManager.argsManager_cgm(settings,args)
leftFlatFoot = argsManager.getLeftFlatFoot()
rightFlatFoot = argsManager.getRightFlatFoot()
headFlat = argsManager.getHeadFlat()
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm2.4")
ik_flag = argsManager.enableIKflag()
hjcMethod = settings["Calibration"]["HJC"]
lhjc = argsManager.forceHjc("left")
rhjc = argsManager.forceHjc("right")
if lhjc is not None:
hjcMethod["Left"] = lhjc
if rhjc is not None:
hjcMethod["Right"] = rhjc
# --------------------------LOADING------------------------------
DATA_PATH, calibrateFilenameLabelledNoExt = NEXUS.GetTrialName()
calibrateFilenameLabelled = calibrateFilenameLabelledNoExt+".c3d"
LOGGER.logger.info( "data Path: "+ DATA_PATH )
LOGGER.set_file_handler(DATA_PATH+"pyCGM2-Calibration.log")
LOGGER.logger.info( "calibration file: "+ calibrateFilenameLabelled)
# --------------------------SUBJECT -----------------------------------
# Notice : Work with ONE subject by session
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.getActiveSubject(NEXUS)
Parameters = NEXUS.GetSubjectParamNames(subject)
required_mp,optional_mp = nexusUtils.getNexusSubjectMp(NEXUS,subject,resetFlag=args.resetMP)
# --------------------------SESSION INFOS -----------------------------
# --------------------------SESSIONS INFOS -----------------------------------
mpInfo,mpFilename = files.getMpFileContent(DATA_PATH,"mp.pyCGM2",subject)
# translators management
translators = files.getTranslators(DATA_PATH,"CGM2_4.translators")
if not translators: translators = settings["Translators"]
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(DATA_PATH,calibrateFilenameLabelledNoExt,subject)
acq = nacf.build()
# --------------------------CONFIG ------------------------------------
model,finalAcqStatic,detectAnomaly = cgm2_4.calibrate(DATA_PATH,calibrateFilenameLabelled,translators,settings,
required_mp,optional_mp,
ik_flag,leftFlatFoot,rightFlatFoot,headFlat,
markerDiameter,
hjcMethod,
pointSuffix,forceBtkAcq=acq, anomalyException=args.anomalyException)
# ----------------------SAVE-------------------------------------------
files.saveModel(model,DATA_PATH,subject)
# save mp
files.saveMp(mpInfo,model,DATA_PATH,mpFilename)
# ----------------------DISPLAY ON VICON-------------------------------
nexusUtils.updateNexusSubjectMp(NEXUS,model,subject)
nexusFilters.NexusModelFilter(NEXUS,
model,finalAcqStatic,subject,
pointSuffix,
staticProcessing=True).run()
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main(args):
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------------GLOBAL SETTINGS ------------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH +
"CGM1_1-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,
"CGM1_1-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,
"CGM1_1-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm(settings, args)
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm1_1")
momentProjection = argsManager.getMomentProjection()
DATA_PATH, reconstructFilenameLabelledNoExt = NEXUS.GetTrialName()
reconstructFilenameLabelled = reconstructFilenameLabelledNoExt + ".c3d"
logging.info("data Path: " + DATA_PATH)
logging.info("calibration file: " + reconstructFilenameLabelled)
# --------------------------SUBJECT ------------------------------------
# Notice : Work with ONE subject by session
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS, subjects)
logging.info("Subject name : " + subject)
# --------------------pyCGM2 MODEL ------------------------------
model = files.loadModel(DATA_PATH, subject)
# --------------------------CHECKING -----------------------------------
# check model is the CGM1
logging.info("loaded model : %s" % (model.version))
if model.version != "CGM1.1":
raise Exception(
"%s-pyCGM2.model file was not calibrated from the CGM1.1 calibration pipeline"
% model.version)
# --------------------------SESSION INFOS ------------------------------------
# translators management
translators = files.getTranslators(DATA_PATH, "CGM1_1.translators")
if not translators: translators = settings["Translators"]
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(
DATA_PATH, reconstructFilenameLabelledNoExt, subject)
acq = nacf.build()
#force plate assignement from Nexus
mfpa = nexusTools.getForcePlateAssignment(NEXUS)
# --------------------------MODELLING PROCESSING -----------------------
acqGait = cgm1_1.fitting(model,
DATA_PATH,
reconstructFilenameLabelled,
translators,
markerDiameter,
pointSuffix,
mfpa,
momentProjection,
forceBtkAcq=acq)
# ----------------------SAVE-------------------------------------------
# Todo: pyCGM2 model : cpickle doesn t work. Incompatibility with Swig. ( see about BTK wrench)
# ----------------------DISPLAY ON VICON-------------------------------
nexusFilters.NexusModelFilter(NEXUS, model, acqGait, subject,
pointSuffix).run()
nexusTools.createGeneralEvents(NEXUS, subject, acqGait,
["Left-FP", "Right-FP"])
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main():
parser = argparse.ArgumentParser(description='CGM2-1 Fitting')
parser.add_argument(
'--proj',
type=str,
help='Moment Projection. Choice : Distal, Proximal, Global')
parser.add_argument('-md',
'--markerDiameter',
type=float,
help='marker diameter')
parser.add_argument('-ps',
'--pointSuffix',
type=str,
help='suffix of model outputs')
parser.add_argument('--check',
action='store_true',
help='force model output suffix')
args = parser.parse_args()
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------------GLOBAL SETTINGS ------------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH +
"CGM2_1-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,
"CGM2_1-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,
"CGM2_1-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm(settings, args)
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm2.1")
momentProjection = argsManager.getMomentProjection()
# --------------------------LOADING ------------------------------------
DATA_PATH, reconstructFilenameLabelledNoExt = NEXUS.GetTrialName()
reconstructFilenameLabelled = reconstructFilenameLabelledNoExt + ".c3d"
logging.info("data Path: " + DATA_PATH)
logging.info("calibration file: " + reconstructFilenameLabelled)
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS, subjects)
logging.info("Subject name : " + subject)
# --------------------pyCGM2 MODEL ------------------------------
model = files.loadModel(DATA_PATH, subject)
# check model
logging.info("loaded model : %s" % (model.version))
if model.version != "CGM2.1":
raise Exception(
"%s-pyCGM2.model file was not calibrated from the CGM2.1 calibration pipeline"
% subject)
# --------------------------SESSION INFOS ------------------------------------
# translators management
translators = files.getTranslators(DATA_PATH, "CGM2_1.translators")
if not translators: translators = settings["Translators"]
#force plate assignement from Nexus
mfpa = nexusTools.getForcePlateAssignment(NEXUS)
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(
DATA_PATH, reconstructFilenameLabelledNoExt, subject)
acq = nacf.build()
# --------------------------MODELLING PROCESSING -----------------------
acqGait = cgm2_1.fitting(model,
DATA_PATH,
reconstructFilenameLabelled,
translators,
markerDiameter,
pointSuffix,
mfpa,
momentProjection,
forceBtkAcq=acq)
# ----------------------DISPLAY ON VICON-------------------------------
nexusFilters.NexusModelFilter(NEXUS, model, acqGait, subject,
pointSuffix).run()
nexusTools.createGeneralEvents(NEXUS, subject, acqGait,
["Left-FP", "Right-FP"])
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main(args):
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------GLOBAL SETTINGS ------------------------------
# ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH +
"CGM2_5-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,
"CGM2_5-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,
"CGM2_5-pyCGM2.settings")
argsManager = CgmArgsManager.argsManager_cgm(settings, args)
leftFlatFoot = argsManager.getLeftFlatFoot()
rightFlatFoot = argsManager.getRightFlatFoot()
headFlat = argsManager.getHeadFlat()
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm2.5")
ik_flag = argsManager.enableIKflag()
hjcMethod = settings["Calibration"]["HJC"]
lhjc = argsManager.forceHjc("left")
rhjc = argsManager.forceHjc("right")
if lhjc is not None:
hjcMethod["Left"] = lhjc
if rhjc is not None:
hjcMethod["Right"] = rhjc
# --------------------------LOADING------------------------------
DATA_PATH, calibrateFilenameLabelledNoExt = NEXUS.GetTrialName()
calibrateFilenameLabelled = calibrateFilenameLabelledNoExt + ".c3d"
logging.info("data Path: " + DATA_PATH)
logging.info("calibration file: " + calibrateFilenameLabelled)
# --------------------------SUBJECT -----------------------------------
# Notice : Work with ONE subject by session
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS, subjects)
Parameters = NEXUS.GetSubjectParamNames(subject)
required_mp, optional_mp = nexusUtils.getNexusSubjectMp(
NEXUS, subject, resetFlag=args.resetMP)
# --------------------------SESSION INFOS -----------------------------
# --------------------------SESSIONS INFOS -----------------------------------
mpInfo, mpFilename = files.getMpFileContent(DATA_PATH, "mp.pyCGM2",
subject)
# translators management
translators = files.getTranslators(DATA_PATH, "CGM2_5.translators")
if not translators: translators = settings["Translators"]
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(
DATA_PATH, calibrateFilenameLabelledNoExt, subject)
acq = nacf.build()
# --------------------------CONFIG ------------------------------------
model, finalAcqStatic = cgm2_5.calibrate(DATA_PATH,
calibrateFilenameLabelled,
translators,
settings,
required_mp,
optional_mp,
ik_flag,
leftFlatFoot,
rightFlatFoot,
headFlat,
markerDiameter,
hjcMethod,
pointSuffix,
forceBtkAcq=acq)
# ----------------------SAVE-------------------------------------------
files.saveModel(model, DATA_PATH, subject)
# save mp
files.saveMp(mpInfo, model, DATA_PATH, mpFilename)
# ----------------------DISPLAY ON VICON-------------------------------
nexusUtils.updateNexusSubjectMp(NEXUS, model, subject)
nexusFilters.NexusModelFilter(NEXUS,
model,
finalAcqStatic,
subject,
pointSuffix,
staticProcessing=True).run()
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
raise Exception(
"%s-pyCGM2.model file was not calibrated from the CGM1.0 calibration pipeline"
% model.version)
# --------------------------SESSION INFOS ------------------------------------
# translators management
translators = files.getTranslators(DATA_PATH, "CGM1.translators")
if not translators: translators = settings["Translators"]
# --------------------------MODELLING PROCESSING -----------------------
acqGait = cgm1.fitting(model, DATA_PATH, reconstructFilenameLabelled,
translators, markerDiameter, pointSuffix, mfpa,
momentProjection)
# ----------------------SAVE-------------------------------------------
# Todo: pyCGM2 model : cpickle doesn t work. Incompatibility with Swig. ( see about BTK wrench)
# ----------------------DISPLAY ON VICON-------------------------------
nexusFilters.NexusModelFilter(NEXUS, model, acqGait, subject,
pointSuffix).run()
nexusTools.createGeneralEvents(NEXUS, subject, acqGait,
["Left-FP", "Right-FP"])
# ========END of the nexus OPERATION if run from Nexus =========
if args.DEBUG:
NEXUS.SaveTrial(30)
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main():
parser = argparse.ArgumentParser(description='CGM1.1 Calibration')
parser.add_argument('-l',
'--leftFlatFoot',
type=int,
help='left flat foot option')
parser.add_argument('-r',
'--rightFlatFoot',
type=int,
help='right flat foot option')
parser.add_argument('-hf', '--headFlat', type=int, help='head flat option')
parser.add_argument('-md',
'--markerDiameter',
type=float,
help='marker diameter')
parser.add_argument('-ps',
'--pointSuffix',
type=str,
help='suffix of model outputs')
parser.add_argument('--check',
action='store_true',
help='force model output suffix')
parser.add_argument('--resetMP',
action='store_true',
help='reset optional mass parameters')
args = parser.parse_args()
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------------GLOBAL SETTINGS ------------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH +
"CGM1_1-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,
"CGM1_1-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,
"CGM1_1-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm1(settings, args)
leftFlatFoot = argsManager.getLeftFlatFoot()
rightFlatFoot = argsManager.getRightFlatFoot()
headFlat = argsManager.getHeadFlat()
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm1_1")
# --------------------------LOADING ------------------------------------
DATA_PATH, calibrateFilenameLabelledNoExt = NEXUS.GetTrialName()
calibrateFilenameLabelled = calibrateFilenameLabelledNoExt + ".c3d"
logging.info("data Path: " + DATA_PATH)
logging.info("calibration file: " + calibrateFilenameLabelled)
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS, subjects)
Parameters = NEXUS.GetSubjectParamNames(subject)
required_mp, optional_mp = nexusUtils.getNexusSubjectMp(
NEXUS, subject, resetFlag=args.resetMP)
# -------------------------- INFOS ------------------------------------
mpInfo, mpFilename = files.getMpFileContent(DATA_PATH, "mp.pyCGM2",
subject)
# translators management
translators = files.getTranslators(DATA_PATH, "CGM1_1.translators")
if not translators: translators = settings["Translators"]
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(
DATA_PATH, calibrateFilenameLabelledNoExt, subject)
acq = nacf.build()
# --------------------------MODELLING PROCESSING -----------------------
model, acqStatic = cgm1_1.calibrate(DATA_PATH,
calibrateFilenameLabelled,
translators,
required_mp,
optional_mp,
leftFlatFoot,
rightFlatFoot,
headFlat,
markerDiameter,
pointSuffix,
forceBtkAcq=acq)
# ----------------------SAVE-------------------------------------------
#pyCGM2.model
files.saveModel(model, DATA_PATH, subject)
# save mp
files.saveMp(mpInfo, model, DATA_PATH, mpFilename)
# ----------------------DISPLAY ON VICON-------------------------------
nexusUtils.updateNexusSubjectMp(NEXUS, model, subject)
nexusFilters.NexusModelFilter(NEXUS,
model,
acqStatic,
subject,
pointSuffix,
staticProcessing=True).run()
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main(args):
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------------GLOBAL SETTINGS ------------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH + "CGM2_2-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,"CGM2_2-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,"CGM2_2-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm(settings,args)
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm2.2")
momentProjection = argsManager.getMomentProjection()
# --------------------------LOADING ------------------------------------
DATA_PATH, reconstructFilenameLabelledNoExt = NEXUS.GetTrialName()
reconstructFilenameLabelled = reconstructFilenameLabelledNoExt+".c3d"
logging.info( "data Path: "+ DATA_PATH )
logging.info( "calibration file: "+ reconstructFilenameLabelled)
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS,subjects)
logging.info( "Subject name : " + subject )
# --------------------pyCGM2 MODEL ------------------------------
model = files.loadModel(DATA_PATH,subject)
# check model
logging.info("loaded model : %s" %(model.version))
if model.version != "CGM2.2":
raise Exception ("%s-pyCGM2.model file was not calibrated from the CGM2.2 calibration pipeline"%subject)
# --------------------------SESSION INFOS ------------------------------------
translators = files.getTranslators(DATA_PATH,"CGM2_2.translators")
if not translators: translators = settings["Translators"]
# ikweight
ikWeight = files.getIKweightSet(DATA_PATH,"CGM2_2.ikw")
if not ikWeight: ikWeight = settings["Fitting"]["Weight"]
#force plate assignement from Nexus
mfpa = nexusTools.getForcePlateAssignment(NEXUS)
nacf = nexusFilters.NexusConstructAcquisitionFilter(DATA_PATH,reconstructFilenameLabelledNoExt,subject)
acq = nacf.build()
# --------------------------MODELLING PROCESSING -----------------------
acqIK = cgm2_2.fitting(model,DATA_PATH, reconstructFilenameLabelled,
translators,settings,
markerDiameter,
pointSuffix,
mfpa,
momentProjection,
forceBtkAcq=acq)
# ----------------------DISPLAY ON VICON-------------------------------
nexusFilters.NexusModelFilter(NEXUS,model,acqIK,subject,pointSuffix).run()
nexusTools.createGeneralEvents(NEXUS,subject,acqIK,["Left-FP","Right-FP"])
# ========END of the nexus OPERATION if run from Nexus =========
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main(args):
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --------------------------GLOBAL SETTINGS ------------------------------------
# global setting ( in user/AppData)
if os.path.isfile(pyCGM2.PYCGM2_APPDATA_PATH + "CGM1-pyCGM2.settings"):
settings = files.openFile(pyCGM2.PYCGM2_APPDATA_PATH,"CGM1-pyCGM2.settings")
else:
settings = files.openFile(pyCGM2.PYCGM2_SETTINGS_FOLDER,"CGM1-pyCGM2.settings")
# --------------------------CONFIG ------------------------------------
argsManager = CgmArgsManager.argsManager_cgm1(settings,args)
leftFlatFoot = argsManager.getLeftFlatFoot()
rightFlatFoot = argsManager.getRightFlatFoot()
headFlat = argsManager.getHeadFlat()
markerDiameter = argsManager.getMarkerDiameter()
pointSuffix = argsManager.getPointSuffix("cgm1")
DATA_PATH, calibrateFilenameLabelledNoExt = NEXUS.GetTrialName()
calibrateFilenameLabelled = calibrateFilenameLabelledNoExt+".c3d"
logging.info( "data Path: "+ DATA_PATH )
logging.info( "calibration file: "+ calibrateFilenameLabelled)
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS,subjects)
Parameters = NEXUS.GetSubjectParamNames(subject)
required_mp,optional_mp = nexusUtils.getNexusSubjectMp(NEXUS,subject,resetFlag=args.resetMP)
# -------------------------- INFOS ------------------------------------
mpInfo,mpFilename = files.getMpFileContent(DATA_PATH,"mp.pyCGM2",subject)
# translators management
translators = files.getTranslators(DATA_PATH,"CGM1.translators")
if not translators: translators = settings["Translators"]
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(DATA_PATH,calibrateFilenameLabelledNoExt,subject)
acq = nacf.build()
# --------------------------MODELLING PROCESSING -----------------------
model,acqStatic = cgm1.calibrate(DATA_PATH,calibrateFilenameLabelled,translators,
required_mp,optional_mp,
leftFlatFoot,rightFlatFoot,headFlat,markerDiameter,
pointSuffix,forceBtkAcq=acq)
# ----------------------SAVE-------------------------------------------
#pyCGM2.model
files.saveModel(model,DATA_PATH,subject)
# save mp
files.saveMp(mpInfo,model,DATA_PATH,mpFilename)
# ----------------------DISPLAY ON VICON-------------------------------
nexusUtils.updateNexusSubjectMp(NEXUS,model,subject)
nexusFilters.NexusModelFilter(NEXUS,
model,acqStatic,subject,
pointSuffix,
staticProcessing=True).run()
