def setUp(self):
link = 33
# setup an instance of DynParams
self.data = DynParams(link)
# setup data to compare against
self.link = link
# inertia matrix terms
self.xx = var('XX33')
self.xy = var('XY33')
self.xz = var('XZ33')
self.yy = var('YY33')
self.yz = var('YZ33')
self.zz = var('ZZ33')
# mass tensor terms
self.msx = var('MX33')
self.msy = var('MY33')
self.msz = var('MZ33')
# link mass
self.mass = var('M33')
# rotor inertia term
self.ia = var('IA33')
# coulomb friction parameter
self.frc = var('FS33')
# viscous friction parameter
self.frv = var('FV33')
# external forces and moments
self.fx_ext = var('FX33')
self.fy_ext = var('FY33')
self.fz_ext = var('FZ33')
self.mx_ext = var('CX33')
self.my_ext = var('CY33')
self.mz_ext = var('CZ33')
# setup matrix terms to compare against
self.inertia = Matrix([[self.xx, self.xy, self.xz],
[self.xy, self.yy, self.yz],
[self.xz, self.yz, self.zz]])
self.mass_tensor = Matrix([self.msx, self.msy, self.msz])
m_eye = self.mass * eye(3)
ms_skew = tools.skew(self.mass_tensor)
self.spatial_inertia = screw6.Screw6(tl=m_eye,
tr=ms_skew.transpose(),
bl=ms_skew,
br=self.inertia)
self.wrench = screw.Screw(
lin=Matrix([self.fx_ext, self.fy_ext, self.fz_ext]),
ang=Matrix([self.mx_ext, self.my_ext, self.mz_ext]))
# setup params to compare against
self.param_xx = var('XX20')
self.param_yy = var('YY20')
self.param_zz = var('ZZ20')
self.params = {
'xx': self.param_xx,
'yy': self.param_yy,
'zz': self.param_zz
}
self.wrong_param = {'rand': 'some-value'}
def setUp(self):
link = 33
# setup an instance of DynParams
self.data = DynParams(link)
# setup data to compare against
self.link = link
# inertia matrix terms
self.xx = var('XX33')
self.xy = var('XY33')
self.xz = var('XZ33')
self.yy = var('YY33')
self.yz = var('YZ33')
self.zz = var('ZZ33')
# mass tensor terms
self.msx = var('MX33')
self.msy = var('MY33')
self.msz = var('MZ33')
# link mass
self.mass = var('M33')
# rotor inertia term
self.ia = var('IA33')
# coulomb friction parameter
self.frc = var('FS33')
# viscous friction parameter
self.frv = var('FV33')
# external forces and moments
self.fx_ext = var('FX33')
self.fy_ext = var('FY33')
self.fz_ext = var('FZ33')
self.mx_ext = var('CX33')
self.my_ext = var('CY33')
self.mz_ext = var('CZ33')
# setup matrix terms to compare against
self.inertia = Matrix([
[self.xx, self.xy, self.xz],
[self.xy, self.yy, self.yz],
[self.xz, self.yz, self.zz]
])
self.mass_tensor = Matrix([self.msx, self.msy, self.msz])
m_eye = self.mass * eye(3)
ms_skew = tools.skew(self.mass_tensor)
self.spatial_inertia = screw6.Screw6(
tl=m_eye, tr=ms_skew.transpose(),
bl=ms_skew, br=self.inertia
)
self.wrench = screw.Screw(
lin=Matrix([self.fx_ext, self.fy_ext, self.fz_ext]),
ang=Matrix([self.mx_ext, self.my_ext, self.mz_ext])
)
# setup params to compare against
self.param_xx = var('XX20')
self.param_yy = var('YY20')
self.param_zz = var('ZZ20')
self.params = {
'xx': self.param_xx,
'yy': self.param_yy,
'zz': self.param_zz
}
self.wrong_param = {'rand': 'some-value'}
def test_set_to_zero(self):
"""Test set_to_zero()"""
link = 3
data = DynParams(link)
data.set_to_zero()
# check link value remains unchanged
self.assertEqual(data.link, link)
# check other values are zero (just a few)
self.assertEqual(data.xx, 0)
self.assertEqual(data.yy, 0)
self.assertEqual(data.zz, 0)
self.assertEqual(data.msx, 0)
self.assertEqual(data.mass, 0)
self.assertEqual(data.ia, 0)
self.assertEqual(data.frv, 0)
self.assertEqual(data.fx_ext, 0)
self.assertEqual(data.mz_ext, 0)
def test_set_to_zero(self):
"""Test set_to_zero()"""
link = 3
data = DynParams(link)
data.set_to_zero()
# check link value remains unchanged
self.assertEqual(data.link, link)
# check other values are zero (just a few)
self.assertEqual(data.xx, 0)
self.assertEqual(data.yy, 0)
self.assertEqual(data.zz, 0)
self.assertEqual(data.msx, 0)
self.assertEqual(data.mass, 0)
self.assertEqual(data.ia, 0)
self.assertEqual(data.frv, 0)
self.assertEqual(data.fx_ext, 0)
self.assertEqual(data.mz_ext, 0)
def test_init(self):
"""Test constructor"""
# test instance type
self.assertIsInstance(DynParams(self.link), DynParams)
class TestDynParams(unittest.TestCase):
"""Unit test for DynParams class."""
def setUp(self):
link = 33
# setup an instance of DynParams
self.data = DynParams(link)
# setup data to compare against
self.link = link
# inertia matrix terms
self.xx = var('XX33')
self.xy = var('XY33')
self.xz = var('XZ33')
self.yy = var('YY33')
self.yz = var('YZ33')
self.zz = var('ZZ33')
# mass tensor terms
self.msx = var('MX33')
self.msy = var('MY33')
self.msz = var('MZ33')
# link mass
self.mass = var('M33')
# rotor inertia term
self.ia = var('IA33')
# coulomb friction parameter
self.frc = var('FS33')
# viscous friction parameter
self.frv = var('FV33')
# external forces and moments
self.fx_ext = var('FX33')
self.fy_ext = var('FY33')
self.fz_ext = var('FZ33')
self.mx_ext = var('CX33')
self.my_ext = var('CY33')
self.mz_ext = var('CZ33')
# setup matrix terms to compare against
self.inertia = Matrix([[self.xx, self.xy, self.xz],
[self.xy, self.yy, self.yz],
[self.xz, self.yz, self.zz]])
self.mass_tensor = Matrix([self.msx, self.msy, self.msz])
m_eye = self.mass * eye(3)
ms_skew = tools.skew(self.mass_tensor)
self.spatial_inertia = screw6.Screw6(tl=m_eye,
tr=ms_skew.transpose(),
bl=ms_skew,
br=self.inertia)
self.wrench = screw.Screw(
lin=Matrix([self.fx_ext, self.fy_ext, self.fz_ext]),
ang=Matrix([self.mx_ext, self.my_ext, self.mz_ext]))
# setup params to compare against
self.param_xx = var('XX20')
self.param_yy = var('YY20')
self.param_zz = var('ZZ20')
self.params = {
'xx': self.param_xx,
'yy': self.param_yy,
'zz': self.param_zz
}
self.wrong_param = {'rand': 'some-value'}
def test_init(self):
"""Test constructor"""
# test instance type
self.assertIsInstance(DynParams(self.link), DynParams)
def test_inertia(self):
"""Test get of inertia()"""
# test get
self.assertEqual(self.data.inertia, self.inertia)
def test_mass_tensor(self):
"""Test get of mass_tensor()"""
# test get
self.assertEqual(self.data.mass_tensor, self.mass_tensor)
def test_spatial_inertia(self):
"""Test get of spatial_inertia()"""
# test get
self.assertEqual(self.data.spatial_inertia, self.spatial_inertia)
def test_wrench(self):
"""Test get of wrench()"""
# test get
self.assertEqual(self.data.wrench, self.wrench)
def test_force(self):
"""Test get and set of force()"""
# test get
self.assertEqual(self.data.force, self.wrench.lin)
def test_moment(self):
"""Test get of moment()"""
# test get
self.assertEqual(self.data.moment, self.wrench.ang)
def test_update_params(self):
"""Test update_params()"""
# test raise AttributeError
self.assertRaises(AttributeError, self.data.update_params,
self.wrong_param)
# test update values
self.data.update_params(self.params)
self.assertEqual(self.data.xx, self.param_xx)
self.assertEqual(self.data.yy, self.param_yy)
self.assertEqual(self.data.zz, self.param_zz)
def test_set_to_zero(self):
"""Test set_to_zero()"""
link = 3
data = DynParams(link)
data.set_to_zero()
# check link value remains unchanged
self.assertEqual(data.link, link)
# check other values are zero (just a few)
self.assertEqual(data.xx, 0)
self.assertEqual(data.yy, 0)
self.assertEqual(data.zz, 0)
self.assertEqual(data.msx, 0)
self.assertEqual(data.mass, 0)
self.assertEqual(data.ia, 0)
self.assertEqual(data.frv, 0)
self.assertEqual(data.fx_ext, 0)
self.assertEqual(data.mz_ext, 0)
def __init__(
self, name, links=0, joints=0, frames=0,
is_floating=True, structure=tools.TREE, is_mobile=False,
is_symbolic=True
):
"""
Constructor period.
Usage:
"""
"""Name of the robot."""
self.name = name
"""Folder to store the files related to the robot."""
self.directory = filemgr.get_folder_path(name)
"""Total number of links in the robot."""
self.num_links = links
"""Total number of joints in the robot."""
self.num_joints = joints
"""Total number of frames in the robot."""
self.num_frames = frames
"""To indicate if the base is floating or not."""
self.is_floating = is_floating
"""Type of the robot structure - simple, tree, closed-loop"""
self.structure = structure
"""To indicate if the robot is a wheeled mobile robot"""
self.is_mobile = is_mobile
"""To indicate if computation should be symbolic or numeric"""
self.is_symbolic = is_symbolic
# properties dependent on number of links
"""
List to hold the dynamic parameters. The indices of the list
start with 0 and it corresponds to parameters of link 0 (virtual
link of the base).
"""
self.dyns = [DynParams(j) for j in self.link_nums]
# properties dependent on number of joints
"""
To indicate if a joint is rigid or flexible. 0 for rigid and 1
for flexible. The indices of the list start with 0 and
corresponds to a virtual joint of the base. This joint is
usually rigid.
"""
self.etas = [0 for j in self.joint_nums]
"""Joint stiffness usually indicated by k."""
self.stiffness = [0 for j in self.joint_nums]
"""Joint velocities."""
self.qdots = [var('QP{0}'.format(j)) for j in self.joint_nums]
"""Joint accelerations."""
self.qddots = [var('QDP{0}'.format(j)) for j in self.joint_nums]
"""Joint torques."""
self.torques = [var('GAM{0}'.format(j)) for j in self.joint_nums]
# properties dependent on number of frames
"""
List to hold the geometric parameters. NOTE: This might be moved
to a separate function.
The indices of the list start with 0 and the first object
corresponds to parameters of frame 0 (base) wrt its antecedent
(some arbitary reference frame).
"""
self.geos = [GeoParams(j) for j in self.frame_nums]
# properties independent of number of links, joints and frames
"""Gravity vector a 3x1 Matrix."""
self.gravity = Matrix([0, 0, var('G3')])
# the values of properties below would be modified during
# the computation of dynamic models.
"""Base velocity 6x1 column vector - a Screw."""
self.base_vel = Screw()
"""Base acceleration 6x1 column vector - a Screw."""
self.base_accel = Screw()
"""Transformation matrix of base wrt a reference frame at time 0."""
self.base_tmat = eye(4)
# call init methods
self._init_maps()
class TestDynParams(unittest.TestCase):
"""Unit test for DynParams class."""
def setUp(self):
link = 33
# setup an instance of DynParams
self.data = DynParams(link)
# setup data to compare against
self.link = link
# inertia matrix terms
self.xx = var('XX33')
self.xy = var('XY33')
self.xz = var('XZ33')
self.yy = var('YY33')
self.yz = var('YZ33')
self.zz = var('ZZ33')
# mass tensor terms
self.msx = var('MX33')
self.msy = var('MY33')
self.msz = var('MZ33')
# link mass
self.mass = var('M33')
# rotor inertia term
self.ia = var('IA33')
# coulomb friction parameter
self.frc = var('FS33')
# viscous friction parameter
self.frv = var('FV33')
# external forces and moments
self.fx_ext = var('FX33')
self.fy_ext = var('FY33')
self.fz_ext = var('FZ33')
self.mx_ext = var('CX33')
self.my_ext = var('CY33')
self.mz_ext = var('CZ33')
# setup matrix terms to compare against
self.inertia = Matrix([
[self.xx, self.xy, self.xz],
[self.xy, self.yy, self.yz],
[self.xz, self.yz, self.zz]
])
self.mass_tensor = Matrix([self.msx, self.msy, self.msz])
m_eye = self.mass * eye(3)
ms_skew = tools.skew(self.mass_tensor)
self.spatial_inertia = screw6.Screw6(
tl=m_eye, tr=ms_skew.transpose(),
bl=ms_skew, br=self.inertia
)
self.wrench = screw.Screw(
lin=Matrix([self.fx_ext, self.fy_ext, self.fz_ext]),
ang=Matrix([self.mx_ext, self.my_ext, self.mz_ext])
)
# setup params to compare against
self.param_xx = var('XX20')
self.param_yy = var('YY20')
self.param_zz = var('ZZ20')
self.params = {
'xx': self.param_xx,
'yy': self.param_yy,
'zz': self.param_zz
}
self.wrong_param = {'rand': 'some-value'}
def test_init(self):
"""Test constructor"""
# test instance type
self.assertIsInstance(DynParams(self.link), DynParams)
def test_inertia(self):
"""Test get of inertia()"""
# test get
self.assertEqual(self.data.inertia, self.inertia)
def test_mass_tensor(self):
"""Test get of mass_tensor()"""
# test get
self.assertEqual(self.data.mass_tensor, self.mass_tensor)
def test_spatial_inertia(self):
"""Test get of spatial_inertia()"""
# test get
self.assertEqual(self.data.spatial_inertia, self.spatial_inertia)
def test_wrench(self):
"""Test get of wrench()"""
# test get
self.assertEqual(self.data.wrench, self.wrench)
def test_force(self):
"""Test get and set of force()"""
# test get
self.assertEqual(self.data.force, self.wrench.lin)
def test_moment(self):
"""Test get of moment()"""
# test get
self.assertEqual(self.data.moment, self.wrench.ang)
def test_update_params(self):
"""Test update_params()"""
# test raise AttributeError
self.assertRaises(
AttributeError, self.data.update_params, self.wrong_param
)
# test update values
self.data.update_params(self.params)
self.assertEqual(self.data.xx, self.param_xx)
self.assertEqual(self.data.yy, self.param_yy)
self.assertEqual(self.data.zz, self.param_zz)
def test_set_to_zero(self):
"""Test set_to_zero()"""
link = 3
data = DynParams(link)
data.set_to_zero()
# check link value remains unchanged
self.assertEqual(data.link, link)
# check other values are zero (just a few)
self.assertEqual(data.xx, 0)
self.assertEqual(data.yy, 0)
self.assertEqual(data.zz, 0)
self.assertEqual(data.msx, 0)
self.assertEqual(data.mass, 0)
self.assertEqual(data.ia, 0)
self.assertEqual(data.frv, 0)
self.assertEqual(data.fx_ext, 0)
self.assertEqual(data.mz_ext, 0)
