Python btkWrench 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: WrenchTest.py Project: zifangliu95/BTKCore

 def test_Constructor_default(self):
     test = btk.btkWrench()
     self.assertEqual(test.GetPosition().GetLabel(), "")
     self.assertEqual(test.GetPosition().GetType(), btk.btkPoint.Marker)
     self.assertEqual(test.GetForce().GetLabel(), ".F")
     self.assertEqual(test.GetForce().GetType(), btk.btkPoint.Force)
     self.assertEqual(test.GetMoment().GetLabel(), ".M")
     self.assertEqual(test.GetMoment().GetType(), btk.btkPoint.Moment)

Example #3

File: WrenchTest.py Project: zifangliu95/BTKCore

 def test_Constructor_full(self):
     test = btk.btkWrench("KneeJoint", 10)
     self.assertEqual(test.GetPosition().GetLabel(), "KneeJoint")
     self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
     self.assertEqual(test.GetForce().GetLabel(), "KneeJoint.F")
     self.assertEqual(test.GetForce().GetValues().shape[0], 10)
     self.assertEqual(test.GetMoment().GetLabel(), "KneeJoint.M")
     self.assertEqual(test.GetMoment().GetValues().shape[0], 10)

Example #4

File: WrenchTest.py Project: zifangliu95/BTKCore

 def test_GetComponent(self):
     test = btk.btkWrench()
     self.assertEqual(test.GetComponent(0).GetLabel(), "")
     self.assertEqual(test.GetComponent(0).GetType(), btk.btkPoint.Marker)
     self.assertEqual(test.GetComponent(1).GetLabel(), ".F")
     self.assertEqual(test.GetComponent(1).GetType(), btk.btkPoint.Force)
     self.assertEqual(test.GetComponent(2).GetLabel(), ".M")
     self.assertEqual(test.GetComponent(2).GetType(), btk.btkPoint.Moment)

Example #5

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

 def test_GetComponent(self):
   test = btk.btkWrench()
   self.assertEqual(test.GetComponent(0).GetLabel(), "")
   self.assertEqual(test.GetComponent(0).GetType(), btk.btkPoint.Marker)
   self.assertEqual(test.GetComponent(1).GetLabel(), ".F")
   self.assertEqual(test.GetComponent(1).GetType(), btk.btkPoint.Force)
   self.assertEqual(test.GetComponent(2).GetLabel(), ".M")
   self.assertEqual(test.GetComponent(2).GetType(), btk.btkPoint.Moment)

Example #6

File: WrenchTest.py Project: zifangliu95/BTKCore

 def test_Constructor_fn(self):
     test = btk.btkWrench(10)
     self.assertEqual(test.GetPosition().GetLabel(), "")
     self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
     self.assertEqual(test.GetForce().GetLabel(), ".F")
     self.assertEqual(test.GetForce().GetValues().shape[0], 10)
     self.assertEqual(test.GetMoment().GetLabel(), ".M")
     self.assertEqual(test.GetMoment().GetValues().shape[0], 10)

Example #7

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

 def test_Constructor_default(self):
     test = btk.btkWrench()
     self.assertEqual(test.GetPosition().GetLabel(), "")
     self.assertEqual(test.GetPosition().GetType(), btk.btkPoint.Marker)
     self.assertEqual(test.GetForce().GetLabel(), ".F")
     self.assertEqual(test.GetForce().GetType(), btk.btkPoint.Force)
     self.assertEqual(test.GetMoment().GetLabel(), ".M")
     self.assertEqual(test.GetMoment().GetType(), btk.btkPoint.Moment)

Example #8

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

 def test_Constructor_full(self):
   test = btk.btkWrench("KneeJoint", 10)
   self.assertEqual(test.GetPosition().GetLabel(), "KneeJoint")
   self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
   self.assertEqual(test.GetForce().GetLabel(), "KneeJoint.F")
   self.assertEqual(test.GetForce().GetValues().shape[0], 10)
   self.assertEqual(test.GetMoment().GetLabel(), "KneeJoint.M")
   self.assertEqual(test.GetMoment().GetValues().shape[0], 10)

Example #9

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

 def test_Constructor_fn(self):
   test = btk.btkWrench(10)
   self.assertEqual(test.GetPosition().GetLabel(), "")
   self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
   self.assertEqual(test.GetForce().GetLabel(), ".F")
   self.assertEqual(test.GetForce().GetValues().shape[0], 10)
   self.assertEqual(test.GetMoment().GetLabel(), ".M")
   self.assertEqual(test.GetMoment().GetValues().shape[0], 10)

Example #10

 def test_wrench_case1(self):
     wrench = btk.btkWrench(100)
     wdf = btk.btkWrenchDownsampleFilter()
     wdf.SetInput(wrench)
     wdf.SetUpDownRatio(10)
     wdf.Update()
     self.assertEqual(wdf.GetOutput().GetComponent(0).GetFrameNumber(), 10)
     self.assertEqual(wdf.GetOutput().GetComponent(1).GetFrameNumber(), 10)
     self.assertEqual(wdf.GetOutput().GetComponent(2).GetFrameNumber(), 10)

Example #11

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

 def test_wrench_case1(self):
   wrench = btk.btkWrench(100)
   wdf = btk.btkWrenchDownsampleFilter();
   wdf.SetInput(wrench)
   wdf.SetUpDownRatio(10)
   wdf.Update()
   self.assertEqual(wdf.GetOutput().GetComponent(0).GetFrameNumber(), 10)
   self.assertEqual(wdf.GetOutput().GetComponent(1).GetFrameNumber(), 10)
   self.assertEqual(wdf.GetOutput().GetComponent(2).GetFrameNumber(), 10)

Example #12

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)

Example #13

File: WrenchTest.py Project: zifangliu95/BTKCore

 def test_SetComponent(self):
     test = btk.btkWrench()
     p = btk.btkPoint("KneeJointCenter", 1)
     f = btk.btkPoint("KneeJointForce", 1)
     m = btk.btkPoint("KneeJointMoment", 1)
     test.SetPosition(p)
     test.SetForce(f)
     test.SetMoment(m)
     test.SetFrameNumber(10)
     self.assertEqual(test.GetPosition().GetLabel(), "KneeJointCenter")
     self.assertEqual(test.GetPosition().GetType(), btk.btkPoint.Marker)
     self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
     self.assertEqual(test.GetForce().GetLabel(), "KneeJointForce")
     self.assertEqual(test.GetForce().GetType(), btk.btkPoint.Force)
     self.assertEqual(test.GetForce().GetValues().shape[0], 10)
     self.assertEqual(test.GetMoment().GetLabel(), "KneeJointMoment")
     self.assertEqual(test.GetMoment().GetType(), btk.btkPoint.Moment)
     self.assertEqual(test.GetMoment().GetValues().shape[0], 10)
     self.assertEqual(p.GetType(), btk.btkPoint.Marker)
     self.assertEqual(p.GetValues().shape[0], 10)
     self.assertEqual(f.GetType(), btk.btkPoint.Force)
     self.assertEqual(f.GetValues().shape[0], 10)
     self.assertEqual(m.GetType(), btk.btkPoint.Moment)
     self.assertEqual(m.GetValues().shape[0], 10)

Example #14

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

 def test_SetComponent(self):
   test = btk.btkWrench()
   p = btk.btkPoint("KneeJointCenter",1)
   f = btk.btkPoint("KneeJointForce",1)
   m = btk.btkPoint("KneeJointMoment",1)
   test.SetPosition(p);
   test.SetForce(f);
   test.SetMoment(m);
   test.SetFrameNumber(10);
   self.assertEqual(test.GetPosition().GetLabel(), "KneeJointCenter")
   self.assertEqual(test.GetPosition().GetType(), btk.btkPoint.Marker)
   self.assertEqual(test.GetPosition().GetValues().shape[0], 10)
   self.assertEqual(test.GetForce().GetLabel(), "KneeJointForce")
   self.assertEqual(test.GetForce().GetType(), btk.btkPoint.Force)
   self.assertEqual(test.GetForce().GetValues().shape[0], 10)
   self.assertEqual(test.GetMoment().GetLabel(), "KneeJointMoment")
   self.assertEqual(test.GetMoment().GetType(), btk.btkPoint.Moment)
   self.assertEqual(test.GetMoment().GetValues().shape[0], 10)
   self.assertEqual(p.GetType(), btk.btkPoint.Marker)
   self.assertEqual(p.GetValues().shape[0], 10)
   self.assertEqual(f.GetType(), btk.btkPoint.Force)
   self.assertEqual(f.GetValues().shape[0], 10)
   self.assertEqual(m.GetType(), btk.btkPoint.Moment)
   self.assertEqual(m.GetValues().shape[0], 10)

Example #15

File: WrenchTest.py Project: zifangliu95/BTKCore

 def test_SetFrameNumber(self):
     test = btk.btkWrench(10)
     test.SetFrameNumber(20)
     self.assertEqual(test.GetPosition().GetValues().shape[0], 20)
     self.assertEqual(test.GetForce().GetValues().shape[0], 20)
     self.assertEqual(test.GetMoment().GetValues().shape[0], 20)

Example #16

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

 def test_SetFrameNumber(self):
   test = btk.btkWrench(10)
   test.SetFrameNumber(20);
   self.assertEqual(test.GetPosition().GetValues().shape[0], 20)
   self.assertEqual(test.GetForce().GetValues().shape[0], 20)
   self.assertEqual(test.GetMoment().GetValues().shape[0], 20)