def main():
#Load the filenames
dynamic_trial, static_trial, vsk_file, outputfile = getfilenames()
#Load a dynamic trial, static trial, and vsk (subject measurements)
motionData, vskData, staticData = loadData(dynamic_trial, static_trial,
vsk_file)
#Calculate the static offsets
flat_foot = False
calibratedMeasurements = pycgmStatic.getStatic(staticData, vskData,
flat_foot)
#Calculate the dynamic trial
# passing the calibrated subject measurements
motionData = motionData[:500] #temporary, just to speed up the calculation
result = pycgmCalc.calcAngles(motionData,
start=None,
end=None,
vsk=calibratedMeasurements,
splitAnglesAxis=False,
formatData=False)
#Write the results to a csv file, if wanted,
# otherwise could just return the angles/axis to some other function
pycgmIO.writeResult(result, outputfile)
return
def main(argv):
"""
Take in motion data file, time span
returns output file of angles
pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end>
"""
flat_foot = False
global vskdata
start,end = None,None
try:
opts, args = getopt.getopt(argv,"h:i:o:s:e:v:x",["ifile=","ofile=","start=","end=","vskfile=","staticinput="])
except getopt.GetoptError:
print 'pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end>'
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print 'pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end>'
sys.exit()
elif opt in ("-i", "--ifile"):
inputfile = arg
elif opt in ("-o", "--ofile"):
outputfile = arg
elif opt in ("-s", "--start"):
start = int(arg)
elif opt in ("-e", "--end"):
end = int(arg)
elif opt in ("-v","--vskfile"):
inputvsk = arg
elif opt in ("-x","--staticinput"):
staticfile = arg
#TODO -x is not working for input
filename = './'+inputfile
motionData = pycgmIO.loadData(filename)
if len(motionData) == 0 or motionData == None:
print "No Data Loaded"
sys.exit()
if inputvsk != None:
vskdata = pycgmIO.loadVSK(inputvsk)
if vskdata!=None:
vsk = pycgmIO.createVskDataDict(vskdata[0],vskdata[1])
if staticfile != None:
staticData = pycgmIO.loadData(staticfile)
calibratedMeasurements = pycgmStatic.getStatic(staticData,vsk,flat_foot)
result=pycgmCalc.calcAngles(motionData,start=start,end=end,vsk=calibratedMeasurements,splitAnglesAxis=False,formatData=False)
pycgmIO.writeResult(result,outputfile)
sys.exit()
def main(argv):
"""
Take in motion data file, time span
returns output file of angles
pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end> -c <calctype> -sf <singleframe>
"""
start, end, singleframe, procs, inputvsk, staticfile, static, vsk = None, None, None, None, None, None, None, None
flat_foot = False
global vskdata
inputfile = "Jong 10 Min_Working_bck.c3d"
outputfile = "10Min_pycgm"
inputvsk = "Jong.vsk"
staticfile = "Jong Cal 03.c3d"
calctype = 'multi'
procs = 7
totalTime = time.time()
loadStaticTime = 0
loadVskTime = 0
loadDataTime = 0
calculateStaticTime = 0
calculateAnglesTime = 0
savaDataTime = 0
try:
opts, args = getopt.getopt(argv, "h:i:o:s:e:c:f:p:v:x", [
"ifile=", "ofile=", "start=", "end=", "calctype=", "singleframe=",
"nprocs=", "vskfile=", "staticinput="
])
except getopt.GetoptError:
print 'pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end> -c <calctype> -sf <singleframe>'
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print 'pyCGM.py -i <motionFile> -o <outputfile> -s <start> -e <end> -c <calctype> -sf <singleframe> -p <procs>'
sys.exit()
elif opt in ("-i", "--ifile"):
inputfile = arg
elif opt in ("-o", "--ofile"):
outputfile = arg + '.csv'
elif opt in ("-s", "--start"):
start = int(arg)
elif opt in ("-e", "--end"):
end = int(arg)
elif opt in ("-c", "--calctype"):
calctype = arg
elif opt in ("-f", "--singleframe"):
singleframe = arg
elif opt in ("-p", "--nprocs"):
procs = int(arg)
elif opt in ("-v", "--vskfile"):
inputvsk = arg
elif opt in ("-x", "--staticinput"):
staticfile = arg
filename = './' + inputfile
loadDataTime = time.time()
motionData = loadData(filename)
loadDataTime = time.time() - loadDataTime
if len(motionData) == 0:
print "No Data Loaded"
sys.exit()
if inputvsk != None:
loadVskTime = time.time()
vskdata = loadVSK(inputvsk)
if vskdata != None:
vsk = createVskDataDict(vskdata[0], vskdata[1])
loadVskTime = time.time() - loadVskTime
if staticfile != None:
loadStaticTime = time.time()
staticData = loadData(staticfile)
loadStaticTime = time.time() - loadStaticTime
calculateStaticTime = time.time()
calibratedMeasurements = pycgmStatic.getStatic(staticData, vsk,
flat_foot)
calculateStaticTime = time.time() - calculateStaticTime
print calibratedMeasurements
if calctype == 'single':
multi = True
else:
multi = False
calculateAnglesTime = time.time()
result = pycgmCalc.calcAngles(motionData,
start=start,
end=end,
cores=procs,
vsk=calibratedMeasurements,
multiprocessing=True,
nprocs=8,
splitAnglesAxis=False,
formatData=False)
calculateAnglesTime = time.time() - calculateAnglesTime
print "CALCULATION TIME: ", calculateAnglesTime
savaDataTime = time.time()
writeResult(result, outputfile)
savaDataTime = time.time() - savaDataTime
totalTime = time.time() - totalTime
printTimes = True
if printTimes == True:
print "Total time = %.10fs" % (totalTime, )
totalTime = totalTime / 100
print "Total time to load data \t= %.10fs\t%0.4f%%" % (
loadStaticTime + loadVskTime + loadDataTime,
(loadStaticTime + loadVskTime + loadDataTime) / totalTime)
print "\tTime to load VSK = %.10fs\t%0.4f%%" % (
loadVskTime, (loadVskTime) / totalTime)
print "\tTime to load static data = %.10fs\t%0.4f%%" % (
loadStaticTime, (loadStaticTime) / totalTime)
print "\tTime to load data = %.10fs\t%0.4f%%" % (
loadDataTime, (loadDataTime) / totalTime)
print "Total time for calculations \t= %.10fs\t%0.4f%%" % (
calculateStaticTime + calculateAnglesTime,
(calculateStaticTime + calculateAnglesTime) / totalTime)
print "\tTime to calculate static = %.10fs\t%0.4f%%" % (
calculateStaticTime, (calculateStaticTime) / totalTime)
print "\tTime to calculate angles = %.10fs\t%0.4f%%" % (
calculateAnglesTime, (calculateAnglesTime) / totalTime)
print "Total time to save csv \t= %.10fs\t%0.4f%%" % (
savaDataTime, (savaDataTime) / totalTime)
sys.exit()