def test_DefaultLabelsAndPrefix(self): reader = btk.btkAcquisitionFileReader() reader.SetFilename(_TDDConfigure.C3DFilePathIN + 'sample04/sub_labels.c3d') reader.Update() skvm = btk.btkSeparateKnownVirtualMarkersFilter() skvm.SetInput(reader.GetOutput().GetPoints()) skvm.SetLabelPrefix('Matt:') skvm.Update() # markers points = skvm.GetOutput(0) self.assertEqual(points.GetItemNumber(), 50) inc = 0 # virtual used for axes points = skvm.GetOutput(1) self.assertEqual(points.GetItemNumber(), 36) inc = 0 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:PELO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:PELA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:PELL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:PELP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOP') # other virtual markers points = skvm.GetOutput(2) self.assertEqual(points.GetItemNumber(), 0) # other type of points points = skvm.GetOutput(3) self.assertEqual(points.GetItemNumber(), 28)
def test_Constructor(self): skvm = btk.btkSeparateKnownVirtualMarkersFilter() labels = skvm.GetVirtualReferenceFrames() num = 19 self.assertEqual(labels.size(), num) it = labels.begin() if (labels.size() >= num): # HED self.assertEqual(it.value().Origin, 'HEDO') self.assertEqual(it.value().Axis1, 'HEDA') self.assertEqual(it.value().Axis2, 'HEDL') self.assertEqual(it.value().Axis3, 'HEDP') it.incr() # LCL self.assertEqual(it.value().Origin, 'LCLO') self.assertEqual(it.value().Axis1, 'LCLA') self.assertEqual(it.value().Axis2, 'LCLL') self.assertEqual(it.value().Axis3, 'LCLP') it.incr() # LFE self.assertEqual(it.value().Origin, 'LFEO') self.assertEqual(it.value().Axis1, 'LFEA') self.assertEqual(it.value().Axis2, 'LFEL') self.assertEqual(it.value().Axis3, 'LFEP') it.incr() # LFO self.assertEqual(it.value().Origin, 'LFOO') self.assertEqual(it.value().Axis1, 'LFOA') self.assertEqual(it.value().Axis2, 'LFOL') self.assertEqual(it.value().Axis3, 'LFOP') it.incr() # LHN self.assertEqual(it.value().Origin, 'LHNO') self.assertEqual(it.value().Axis1, 'LHNA') self.assertEqual(it.value().Axis2, 'LHNL') self.assertEqual(it.value().Axis3, 'LHNP') it.incr() # LHU self.assertEqual(it.value().Origin, 'LHUO') self.assertEqual(it.value().Axis1, 'LHUA') self.assertEqual(it.value().Axis2, 'LHUL') self.assertEqual(it.value().Axis3, 'LHUP') it.incr() # LRA self.assertEqual(it.value().Origin, 'LRAO') self.assertEqual(it.value().Axis1, 'LRAA') self.assertEqual(it.value().Axis2, 'LRAL') self.assertEqual(it.value().Axis3, 'LRAP') it.incr() # LTI self.assertEqual(it.value().Origin, 'LTIO') self.assertEqual(it.value().Axis1, 'LTIA') self.assertEqual(it.value().Axis2, 'LTIL') self.assertEqual(it.value().Axis3, 'LTIP') it.incr() # LTO self.assertEqual(it.value().Origin, 'LTOO') self.assertEqual(it.value().Axis1, 'LTOA') self.assertEqual(it.value().Axis2, 'LTOL') self.assertEqual(it.value().Axis3, 'LTOP') it.incr() # PEL self.assertEqual(it.value().Origin, 'PELO') self.assertEqual(it.value().Axis1, 'PELA') self.assertEqual(it.value().Axis2, 'PELL') self.assertEqual(it.value().Axis3, 'PELP') it.incr() # RCL self.assertEqual(it.value().Origin, 'RCLO') self.assertEqual(it.value().Axis1, 'RCLA') self.assertEqual(it.value().Axis2, 'RCLL') self.assertEqual(it.value().Axis3, 'RCLP') it.incr() # RFE self.assertEqual(it.value().Origin, 'RFEO') self.assertEqual(it.value().Axis1, 'RFEA') self.assertEqual(it.value().Axis2, 'RFEL') self.assertEqual(it.value().Axis3, 'RFEP') it.incr() # RFO self.assertEqual(it.value().Origin, 'RFOO') self.assertEqual(it.value().Axis1, 'RFOA') self.assertEqual(it.value().Axis2, 'RFOL') self.assertEqual(it.value().Axis3, 'RFOP') it.incr() # RHN self.assertEqual(it.value().Origin, 'RHNO') self.assertEqual(it.value().Axis1, 'RHNA') self.assertEqual(it.value().Axis2, 'RHNL') self.assertEqual(it.value().Axis3, 'RHNP') it.incr() # RHU self.assertEqual(it.value().Origin, 'RHUO') self.assertEqual(it.value().Axis1, 'RHUA') self.assertEqual(it.value().Axis2, 'RHUL') self.assertEqual(it.value().Axis3, 'RHUP') it.incr() # RRA self.assertEqual(it.value().Origin, 'RRAO') self.assertEqual(it.value().Axis1, 'RRAA') self.assertEqual(it.value().Axis2, 'RRAL') self.assertEqual(it.value().Axis3, 'RRAP') it.incr() # RTI self.assertEqual(it.value().Origin, 'RTIO') self.assertEqual(it.value().Axis1, 'RTIA') self.assertEqual(it.value().Axis2, 'RTIL') self.assertEqual(it.value().Axis3, 'RTIP') it.incr() # RTO self.assertEqual(it.value().Origin, 'RTOO') self.assertEqual(it.value().Axis1, 'RTOA') self.assertEqual(it.value().Axis2, 'RTOL') self.assertEqual(it.value().Axis3, 'RTOP') it.incr() # TRX self.assertEqual(it.value().Origin, 'TRXO') self.assertEqual(it.value().Axis1, 'TRXA') self.assertEqual(it.value().Axis2, 'TRXL') self.assertEqual(it.value().Axis3, 'TRXP') it.incr() labels2 = skvm.GetVirtualMarkers() num = 2 self.assertEqual(labels2.size(), num) it2 = labels2.begin() if (labels2.size() >= num): self.assertEqual(it2.value(), 'CentreOfMass') it2.incr() self.assertEqual(it2.value(), 'CentreOfMassFloor')
def test_FromLabelsLists(self): # virtual markers for frame axes labels = ['LFE', 'LFO', 'LTI', 'LTO', 'RFE', 'RFO', 'RTI', 'RTO'] virtualMarkerLabelsAxes = btk.btkStringAxesList() for i in range(0, len(labels)): label = labels[i] virtualMarkerLabelsAxes.push_back(btk.btkStringAxes(label + 'O', label + 'A', label + 'L', label + 'P')) reader = btk.btkAcquisitionFileReader() reader.SetFilename(_TDDConfigure.C3DFilePathIN + 'sample04/sub_labels.c3d') reader.Update() skvm = btk.btkSeparateKnownVirtualMarkersFilter() skvm.SetInput(reader.GetOutput().GetPoints()) skvm.SetLabelPrefix('Matt:') skvm.AppendVirtualMarker('LKNE') skvm.AppendVirtualMarker('RKNE') skvm.SetVirtualReferenceFrames(virtualMarkerLabelsAxes) skvm.Update() # markers points = skvm.GetOutput(0) self.assertEqual(points.GetItemNumber(), 52) inc = 0 # virtual used for axes points = skvm.GetOutput(1) self.assertEqual(points.GetItemNumber(), 32) inc = 0 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LTOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RTOP') # other virtual markers points = skvm.GetOutput(2) self.assertEqual(points.GetItemNumber(), 2) inc = 0 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:LKNE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'Matt:RKNE') # other type of points points = skvm.GetOutput(3) self.assertEqual(points.GetItemNumber(), 28)
def test_DefaultLabels(self): reader = btk.btkAcquisitionFileReader() reader.SetFilename(_TDDConfigure.C3DFilePathIN + 'sample09/PlugInC3D.c3d') reader.Update() skvm = btk.btkSeparateKnownVirtualMarkersFilter() skvm.SetInput(reader.GetOutput().GetPoints()) skvm.Update() # markers points = skvm.GetOutput(0) self.assertEqual(points.GetItemNumber(), 18) inc = 0 self.assertEqual(points.GetItem(inc).GetLabel(), 'RKNE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTOE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTIB'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RASI'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTHI'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RHEE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LKNE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RANK'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RCLA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTHI'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LASI'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'C7'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTOE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LANK'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'SACR'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LHEE'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LCLA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTIB'); # virtual used for axes points = skvm.GetOutput(1) self.assertEqual(points.GetItemNumber(), 36) inc = 0 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'LTOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'PELO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'PELA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'PELL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'PELP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFEO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFEA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFEL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFEP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RFOP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTIO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTIA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTIL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTIP'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTOO'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTOA'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTOL'); inc +=1 self.assertEqual(points.GetItem(inc).GetLabel(), 'RTOP') # other virtual markers points = skvm.GetOutput(2) self.assertEqual(points.GetItemNumber(), 0) # other type of points points = skvm.GetOutput(3) self.assertEqual(points.GetItemNumber(), 32)