def test_temporalEmgPlot(self):
# ----DATA-----
DATA_PATH = pyCGM2.TEST_DATA_PATH+"GaitData\\EMG\\Hånnibøl Lecter-nerve block\\"
gaitTrial = "PRE-gait trial 01.c3d"
restTrial = "PRE-repos.c3d"
DATA_PATH_OUT = pyCGM2.TEST_DATA_PATH_OUT+"GaitData\\EMG\\Hånnibøl Lecter-nerve block\\"
files.createDir(DATA_PATH_OUT)
#--------------------------settings-------------------------------------
if os.path.isfile(DATA_PATH + "emg.settings"):
emgSettings = files.openFile(DATA_PATH,"emg.settings")
logging.warning("[pyCGM2]: emg.settings detected in the data folder")
else:
emgSettings = None
manager = EmgManager.EmgConfigManager(None,localInternalSettings=emgSettings)
manager.contruct()
EMG_LABELS,EMG_MUSCLES,EMG_CONTEXT,NORMAL_ACTIVITIES = manager.getEmgConfiguration()
rectifyBool=True
acq = btkTools.smartReader(DATA_PATH+gaitTrial)
analysis.processEMG_fromBtkAcq(acq, EMG_LABELS,
highPassFrequencies=[20,200],
envelopFrequency=6.0)
openmaTrial = trialTools.convertBtkAcquisition(acq)
plot.plotTemporalEMG(DATA_PATH,gaitTrial, EMG_LABELS,EMG_MUSCLES, EMG_CONTEXT, NORMAL_ACTIVITIES,exportPdf=False,rectify=rectifyBool,openmaTrial=openmaTrial)
def noCroppedC3d(cls):
DATA_PATH = "C:\\Users\\HLS501\\Documents\\VICON DATA\\pyCGM2-Data\\NexusAPI\\BtkAcquisitionCreator\\sample0\\"
acq = btkTools.smartReader(DATA_PATH + "gait_noCropped.c3d")
root = trialTools.convertBtkAcquisition(acq, returnType="Root")
ma.io.write(root, "gait_noCropped_checked.c3d")
def main(args):
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --- acquisition file and path----
DATA_PATH, inputFileNoExt = NEXUS.GetTrialName()
inputFile = inputFileNoExt+".c3d"
#--------------------------settings-------------------------------------
if os.path.isfile(DATA_PATH + "emg.settings"):
emgSettings = files.openFile(DATA_PATH,"emg.settings")
logging.warning("[pyCGM2]: emg.settings detected in the data folder")
else:
emgSettings = None
manager = EmgManager.EmgConfigManager(None,localInternalSettings=emgSettings)
manager.contruct()
# ----------------------INPUTS-------------------------------------------
bandPassFilterFrequencies = manager.BandpassFrequencies#emgSettings["Processing"]["BandpassFrequencies"]
if args.BandpassFrequencies is not None:
if len(args.BandpassFrequencies) != 2:
raise Exception("[pyCGM2] - bad configuration of the bandpass frequencies ... set 2 frequencies only")
else:
bandPassFilterFrequencies = [float(args.BandpassFrequencies[0]),float(args.BandpassFrequencies[1])]
logging.info("Band pass frequency set to %i - %i instead of 20-200Hz",bandPassFilterFrequencies[0],bandPassFilterFrequencies[1])
envelopCutOffFrequency = manager.EnvelopLowpassFrequency#emgSettings["Processing"]["EnvelopLowpassFrequency"]
if args.EnvelopLowpassFrequency is not None:
envelopCutOffFrequency = args.EnvelopLowpassFrequency
logging.info("Cut-off frequency set to %i instead of 6Hz ",envelopCutOffFrequency)
consistencyFlag = True if args.consistency else False
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS,subjects)
# btkAcq builder
nacf = nexusFilters.NexusConstructAcquisitionFilter(DATA_PATH,inputFileNoExt,subject)
acq = nacf.build()
# --------------emg Processing--------------
EMG_LABELS,EMG_MUSCLES,EMG_CONTEXT,NORMAL_ACTIVITIES = manager.getEmgConfiguration()
analysis.processEMG_fromBtkAcq(acq, EMG_LABELS,
highPassFrequencies=bandPassFilterFrequencies,
envelopFrequency=envelopCutOffFrequency) # high pass then low pass for all c3ds
openmaTrial = trialTools.convertBtkAcquisition(acq)
emgAnalysis = analysis.makeEmgAnalysis(DATA_PATH, [inputFile], EMG_LABELS,openmaTrials = [openmaTrial])
if not consistencyFlag:
plot.plotDescriptiveEnvelopEMGpanel(DATA_PATH,emgAnalysis, EMG_LABELS,EMG_MUSCLES,EMG_CONTEXT, NORMAL_ACTIVITIES, normalized=False,exportPdf=True,outputName=inputFile)
else:
plot.plotConsistencyEnvelopEMGpanel(DATA_PATH,emgAnalysis, EMG_LABELS,EMG_MUSCLES,EMG_CONTEXT, NORMAL_ACTIVITIES, normalized=False,exportPdf=True,outputName=inputFile)
else:
raise Exception("NO Nexus connection. Turn on Nexus")
def main():
parser = argparse.ArgumentParser(description='EMG-plot_temporalEMG')
parser.add_argument('-bpf', '--BandpassFrequencies', nargs='+',help='bandpass filter')
parser.add_argument('-ecf','--EnvelopLowpassFrequency', type=int, help='cutoff frequency for emg envelops')
parser.add_argument('-r','--raw', action='store_true', help='rectified data')
args = parser.parse_args()
NEXUS = ViconNexus.ViconNexus()
NEXUS_PYTHON_CONNECTED = NEXUS.Client.IsConnected()
if NEXUS_PYTHON_CONNECTED: # run Operation
# --- acquisition file and path----
DATA_PATH, inputFileNoExt = NEXUS.GetTrialName()
inputFile = inputFileNoExt+".c3d"
#--------------------------settings-------------------------------------
if os.path.isfile(DATA_PATH + "emg.settings"):
emgSettings = files.openFile(DATA_PATH,"emg.settings")
logging.warning("[pyCGM2]: emg.settings detected in the data folder")
else:
emgSettings = None
manager = EmgManager.EmgConfigManager(None,localInternalSettings=emgSettings)
manager.contruct()
# ----------------------INPUTS-------------------------------------------
bandPassFilterFrequencies = manager.BandpassFrequencies#emgSettings["Processing"]["BandpassFrequencies"]
if args.BandpassFrequencies is not None:
if len(args.BandpassFrequencies) != 2:
raise Exception("[pyCGM2] - bad configuration of the bandpass frequencies ... set 2 frequencies only")
else:
bandPassFilterFrequencies = [float(args.BandpassFrequencies[0]),float(args.BandpassFrequencies[1])]
logging.info("Band pass frequency set to %i - %i instead of 20-200Hz",bandPassFilterFrequencies[0],bandPassFilterFrequencies[1])
envelopCutOffFrequency = manager.EnvelopLowpassFrequency#emgSettings["Processing"]["EnvelopLowpassFrequency"]
if args.EnvelopLowpassFrequency is not None:
envelopCutOffFrequency = args.EnvelopLowpassFrequency
logging.info("Cut-off frequency set to %i instead of 6Hz ",envelopCutOffFrequency)
rectifyBool = False if args.raw else True
# --------------------------SUBJECT ------------------------------------
subjects = NEXUS.GetSubjectNames()
subject = nexusTools.checkActivatedSubject(NEXUS,subjects)
# btk Acquisition
nacf = nexusFilters.NexusConstructAcquisitionFilter(DATA_PATH,inputFileNoExt,subject)
acq = nacf.build()
# --------------emg Processing--------------
EMG_LABELS,EMG_MUSCLES,EMG_CONTEXT,NORMAL_ACTIVITIES = manager.getEmgConfiguration()
analysis.processEMG_fromBtkAcq(acq, EMG_LABELS,
highPassFrequencies=bandPassFilterFrequencies,
envelopFrequency=envelopCutOffFrequency) # high pass then low pass for all c3ds
openmaTrial = trialTools.convertBtkAcquisition(acq)
plot.plotTemporalEMG(DATA_PATH,inputFile, EMG_LABELS,EMG_MUSCLES, EMG_CONTEXT, NORMAL_ACTIVITIES,exportPdf=True,rectify=rectifyBool,
openmaTrial=openmaTrial)
else:
raise Exception("NO Nexus connection. Turn on Nexus")