Python btkWrenchCollectionDownsampleFilter Examples

Example #1

 def test_wrenchCollection_case1(self):
     collection = btk.btkWrenchCollection()
     collection.InsertItem(btk.btkWrench(100))
     collection.InsertItem(btk.btkWrench(200))
     collection.InsertItem(btk.btkWrench(300))
     wcdf = btk.btkWrenchCollectionDownsampleFilter()
     wcdf.SetInput(collection)
     wcdf.SetUpDownRatio(10)
     wcdf.Update()
     self.assertEqual(
         wcdf.GetOutput().GetItem(0).GetComponent(0).GetFrameNumber(), 10)
     self.assertEqual(
         wcdf.GetOutput().GetItem(0).GetComponent(1).GetFrameNumber(), 10)
     self.assertEqual(
         wcdf.GetOutput().GetItem(0).GetComponent(2).GetFrameNumber(), 10)
     self.assertEqual(
         wcdf.GetOutput().GetItem(1).GetComponent(0).GetFrameNumber(), 20)
     self.assertEqual(
         wcdf.GetOutput().GetItem(1).GetComponent(1).GetFrameNumber(), 20)
     self.assertEqual(
         wcdf.GetOutput().GetItem(1).GetComponent(2).GetFrameNumber(), 20)
     self.assertEqual(
         wcdf.GetOutput().GetItem(2).GetComponent(0).GetFrameNumber(), 30)
     self.assertEqual(
         wcdf.GetOutput().GetItem(2).GetComponent(1).GetFrameNumber(), 30)
     self.assertEqual(
         wcdf.GetOutput().GetItem(2).GetComponent(2).GetFrameNumber(), 30)

Example #2

File: VerticalGroundReactionForceGaitEventDetectorTest.py Project: zifangliu95/BTKCore

 def test_PluginC3D_Threshold50(self):
     reader = btk.btkAcquisitionFileReader()
     reader.SetFilename(_TDDConfigure.C3DFilePathIN +
                        "sample09/PlugInC3D.c3d")
     reader.Update()
     pfe = btk.btkForcePlatformsExtractor()
     grwf = btk.btkGroundReactionWrenchFilter()
     dswc = btk.btkWrenchCollectionDownsampleFilter()
     dswc.SetUpDownRatio(reader.GetOutput().GetNumberAnalogSamplePerFrame())
     pfe.SetInput(reader.GetOutput())
     grwf.SetInput(pfe.GetOutput())
     dswc.SetInput(grwf.GetOutput())
     vgrfged = btk.btkVerticalGroundReactionForceGaitEventDetector()
     vgrfged.SetInput(dswc.GetOutput())
     vgrfged.SetThresholdValue(50)
     vgrfged.SetForceplateContextMapping(["Right", "Left"])
     vgrfged.SetAcquisitionInformation(
         reader.GetOutput().GetFirstFrame(),
         reader.GetOutput().GetPointFrequency(), "")
     output = vgrfged.GetOutput()
     output.Update()
     self.assertEqual(output.GetItemNumber(), 4)
     ev = output.GetItem(0)
     self.assertEqual(ev.GetLabel(), "Foot Strike")
     self.assertEqual(ev.GetContext(), "Right")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 1)
     self.assertEqual(ev.GetFrame(), 67)
     self.assertEqual(ev.GetTime(), 67.0 / 60.0)
     ev = output.GetItem(1)
     self.assertEqual(ev.GetLabel(), "Foot Off")
     self.assertEqual(ev.GetContext(), "Right")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 2)
     self.assertEqual(ev.GetFrame(), 114)
     self.assertEqual(ev.GetTime(), 114.0 / 60.0)
     ev = output.GetItem(2)
     self.assertEqual(ev.GetLabel(), "Foot Strike")
     self.assertEqual(ev.GetContext(), "Left")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 1)
     self.assertEqual(ev.GetFrame(), 109)
     self.assertEqual(ev.GetTime(), 109.0 / 60.0)
     ev = output.GetItem(3)
     self.assertEqual(ev.GetLabel(), "Foot Off")
     self.assertEqual(ev.GetContext(), "Left")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 2)
     self.assertEqual(ev.GetFrame(), 148)
     self.assertEqual(ev.GetTime(), 148.0 / 60.0)

Example #3

File: VerticalGroundReactionForceGaitEventDetectorTest.py Project: zifangliu95/BTKCore

 def test_Gait_NoMapping(self):
     reader = btk.btkAcquisitionFileReader()
     reader.SetFilename(_TDDConfigure.C3DFilePathIN + "others/Gait.c3d")
     reader.Update()
     pfe = btk.btkForcePlatformsExtractor()
     grwf = btk.btkGroundReactionWrenchFilter()
     dswc = btk.btkWrenchCollectionDownsampleFilter()
     dswc.SetUpDownRatio(reader.GetOutput().GetNumberAnalogSamplePerFrame())
     pfe.SetInput(reader.GetOutput())
     grwf.SetInput(pfe.GetOutput())
     dswc.SetInput(grwf.GetOutput())
     vgrfged = btk.btkVerticalGroundReactionForceGaitEventDetector()
     vgrfged.SetInput(dswc.GetOutput())
     vgrfged.SetAcquisitionInformation(
         reader.GetOutput().GetFirstFrame(),
         reader.GetOutput().GetPointFrequency(), "")
     output = vgrfged.GetOutput()
     output.Update()
     self.assertEqual(output.GetItemNumber(), 4)
     ev = output.GetItem(0)
     self.assertEqual(ev.GetLabel(), "Foot Strike")
     self.assertEqual(ev.GetContext(), "General")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 1)
     self.assertEqual(ev.GetFrame(), 257)
     self.assertEqual(ev.GetTime(), 2.57)
     ev = output.GetItem(1)
     self.assertEqual(ev.GetLabel(), "Foot Off")
     self.assertEqual(ev.GetContext(), "General")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 2)
     self.assertEqual(ev.GetFrame(), 316)
     self.assertEqual(ev.GetTime(), 3.16)
     ev = output.GetItem(2)
     self.assertEqual(ev.GetLabel(), "Foot Strike")
     self.assertEqual(ev.GetContext(), "General")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 1)
     self.assertEqual(ev.GetFrame(), 209)
     self.assertEqual(ev.GetTime(), 2.09)
     ev = output.GetItem(3)
     self.assertEqual(ev.GetLabel(), "Foot Off")
     self.assertEqual(ev.GetContext(), "General")
     self.assertEqual(
         ev.GetDetectionFlags(),
         btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
     self.assertEqual(ev.GetId(), 2)
     self.assertEqual(ev.GetFrame(), 267)
     self.assertEqual(ev.GetTime(), 2.67)

Example #4

File: VerticalGroundReactionForceGaitEventDetectorTest.py Project: Biomechanical-ToolKit/BTKCore

 def test_PluginC3D_Threshold50(self):
   reader = btk.btkAcquisitionFileReader()
   reader.SetFilename(_TDDConfigure.C3DFilePathIN + "sample09/PlugInC3D.c3d")
   reader.Update()
   pfe = btk.btkForcePlatformsExtractor()
   grwf = btk.btkGroundReactionWrenchFilter()
   dswc = btk.btkWrenchCollectionDownsampleFilter()
   dswc.SetUpDownRatio(reader.GetOutput().GetNumberAnalogSamplePerFrame())
   pfe.SetInput(reader.GetOutput())
   grwf.SetInput(pfe.GetOutput())
   dswc.SetInput(grwf.GetOutput())
   vgrfged = btk.btkVerticalGroundReactionForceGaitEventDetector()
   vgrfged.SetInput(dswc.GetOutput())
   vgrfged.SetThresholdValue(50)
   vgrfged.SetForceplateContextMapping(["Right","Left"])
   vgrfged.SetAcquisitionInformation(reader.GetOutput().GetFirstFrame(), reader.GetOutput().GetPointFrequency(), "")
   output = vgrfged.GetOutput()
   output.Update()
   self.assertEqual(output.GetItemNumber(), 4)
   ev = output.GetItem(0)
   self.assertEqual(ev.GetLabel(), "Foot Strike")
   self.assertEqual(ev.GetContext(), "Right")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 1)
   self.assertEqual(ev.GetFrame(), 67)
   self.assertEqual(ev.GetTime(), 67.0/60.0)
   ev = output.GetItem(1)
   self.assertEqual(ev.GetLabel(), "Foot Off")
   self.assertEqual(ev.GetContext(), "Right")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 2)
   self.assertEqual(ev.GetFrame(), 114)
   self.assertEqual(ev.GetTime(), 114.0/60.0)
   ev = output.GetItem(2)
   self.assertEqual(ev.GetLabel(), "Foot Strike")
   self.assertEqual(ev.GetContext(), "Left")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 1)
   self.assertEqual(ev.GetFrame(), 109)
   self.assertEqual(ev.GetTime(), 109.0/60.0)
   ev = output.GetItem(3)
   self.assertEqual(ev.GetLabel(), "Foot Off")
   self.assertEqual(ev.GetContext(), "Left")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 2)
   self.assertEqual(ev.GetFrame(), 148)
   self.assertEqual(ev.GetTime(), 148.0/60.0)

Example #5

File: VerticalGroundReactionForceGaitEventDetectorTest.py Project: Biomechanical-ToolKit/BTKCore

 def test_Gait_NoMapping(self):
   reader = btk.btkAcquisitionFileReader()
   reader.SetFilename(_TDDConfigure.C3DFilePathIN + "others/Gait.c3d")
   reader.Update()
   pfe = btk.btkForcePlatformsExtractor()
   grwf = btk.btkGroundReactionWrenchFilter()
   dswc = btk.btkWrenchCollectionDownsampleFilter()
   dswc.SetUpDownRatio(reader.GetOutput().GetNumberAnalogSamplePerFrame())
   pfe.SetInput(reader.GetOutput())
   grwf.SetInput(pfe.GetOutput())
   dswc.SetInput(grwf.GetOutput())
   vgrfged = btk.btkVerticalGroundReactionForceGaitEventDetector()
   vgrfged.SetInput(dswc.GetOutput())
   vgrfged.SetAcquisitionInformation(reader.GetOutput().GetFirstFrame(), reader.GetOutput().GetPointFrequency(), "")
   output = vgrfged.GetOutput()
   output.Update()
   self.assertEqual(output.GetItemNumber(), 4)
   ev = output.GetItem(0)
   self.assertEqual(ev.GetLabel(), "Foot Strike")
   self.assertEqual(ev.GetContext(), "General")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 1)
   self.assertEqual(ev.GetFrame(), 257)
   self.assertEqual(ev.GetTime(), 2.57)
   ev = output.GetItem(1)
   self.assertEqual(ev.GetLabel(), "Foot Off")
   self.assertEqual(ev.GetContext(), "General")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 2)
   self.assertEqual(ev.GetFrame(), 316)
   self.assertEqual(ev.GetTime(), 3.16)
   ev = output.GetItem(2)
   self.assertEqual(ev.GetLabel(), "Foot Strike")
   self.assertEqual(ev.GetContext(), "General")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 1)
   self.assertEqual(ev.GetFrame(), 209)
   self.assertEqual(ev.GetTime(), 2.09)
   ev = output.GetItem(3)
   self.assertEqual(ev.GetLabel(), "Foot Off")
   self.assertEqual(ev.GetContext(), "General")
   self.assertEqual(ev.GetDetectionFlags(), btk.btkEvent.Automatic | btk.btkEvent.FromForcePlatform)
   self.assertEqual(ev.GetId(), 2)
   self.assertEqual(ev.GetFrame(), 267)
   self.assertEqual(ev.GetTime(), 2.67)

Example #6

File: DownsampleFilterTest.py Project: Biomechanical-ToolKit/BTKCore

 def test_wrenchCollection_case1(self):
   collection = btk.btkWrenchCollection()
   collection.InsertItem(btk.btkWrench(100))
   collection.InsertItem(btk.btkWrench(200))
   collection.InsertItem(btk.btkWrench(300))
   wcdf = btk.btkWrenchCollectionDownsampleFilter();
   wcdf.SetInput(collection)
   wcdf.SetUpDownRatio(10)
   wcdf.Update()
   self.assertEqual(wcdf.GetOutput().GetItem(0).GetComponent(0).GetFrameNumber(), 10)
   self.assertEqual(wcdf.GetOutput().GetItem(0).GetComponent(1).GetFrameNumber(), 10)
   self.assertEqual(wcdf.GetOutput().GetItem(0).GetComponent(2).GetFrameNumber(), 10)
   self.assertEqual(wcdf.GetOutput().GetItem(1).GetComponent(0).GetFrameNumber(), 20)
   self.assertEqual(wcdf.GetOutput().GetItem(1).GetComponent(1).GetFrameNumber(), 20)
   self.assertEqual(wcdf.GetOutput().GetItem(1).GetComponent(2).GetFrameNumber(), 20)
   self.assertEqual(wcdf.GetOutput().GetItem(2).GetComponent(0).GetFrameNumber(), 30)
   self.assertEqual(wcdf.GetOutput().GetItem(2).GetComponent(1).GetFrameNumber(), 30)
   self.assertEqual(wcdf.GetOutput().GetItem(2).GetComponent(2).GetFrameNumber(), 30)