def test_collisions(self):
"""
When objects collide and L{CollisionWorld.debugDrawWorld} is called,
collisions are drawn using the debug drawer.
"""
def objAt(pos):
obj = RigidBody(None, BoxShape(Vector3(1, 1, 1)), 1.0)
xform = Transform()
xform.setOrigin(pos)
obj.setWorldTransform(xform)
return obj
first = objAt(Vector3(-2, 0, 0))
first.applyCentralForce(Vector3(50, 0, 0))
second = objAt(Vector3(2, 0, 0))
second.applyCentralForce(Vector3(-50, 0, 0))
self.world.addRigidBody(first)
self.world.addRigidBody(second)
# Simulate the world enough for them to hit each other
for i in range(100):
self.world.stepSimulation(1.0 / 60.0)
self.world.debugDrawWorld()
self.assertTrue(len(self.recorder.contacts) > 0)
for contact in self.recorder.contacts:
self.assertEquals(len(contact), 11)
def test_fromConstructionInfoWithoutMotionState(self):
"""
When L{RigidBody.fromConstructionInfo} is used to construct a
L{RigidBody} without a L{MotionState}, the C{worldTransform} parameter
is used to specify the body's initial position.
"""
mass = 8
shape = SphereShape(3)
inertia = Vector3(1, 2, 3)
startTransform = Transform()
startTransform.setOrigin(Vector3(5, 6, 7))
linearDamping = 0.3
angularDamping = 0.4
friction = 4.5
restition = 5.4
linearSleepingThreshold = 4.3
angularSleepingThreshold = 3.2
body = RigidBody.fromConstructionInfo(None, shape, mass, inertia,
startTransform, linearDamping,
angularDamping, friction,
restition,
linearSleepingThreshold,
angularSleepingThreshold)
origin = body.getWorldTransform().getOrigin()
self.assertEqual(5, origin.x)
self.assertEqual(6, origin.y)
self.assertEqual(7, origin.z)
def test_cross(self):
"""
L{Vector3.cross} accepts another L{Vector3} and returns a new L{Vector3}
which is perpendicular to the first two.
"""
v1 = Vector3(1, 0, 0)
v2 = Vector3(0, 1, 0)
v3 = v1.cross(v2)
self.assertEqual(v3.x, 0)
self.assertEqual(v3.y, 0)
self.assertEqual(v3.z, 1)
def test_vectorSubtraction(self):
"""
A L{Vector3} instance subtracted from another L{Vector3} instance
results in a new L{Vector3} instance with components equal to the
difference of the component pairs of the inputs.
"""
v1 = Vector3(1, 2, 3)
v2 = Vector3(2, -1, 1)
v3 = v1 - v2
self.assertEqual(v3.x, -1)
self.assertEqual(v3.y, 3)
self.assertEqual(v3.z, 2)
def test_vectorAddition(self):
"""
A L{Vector3} instance added to another L{Vector3} instance results in a
new L{Vector3} instance with components equal to the sum of the
component pairs of the inputs.
"""
v1 = Vector3(1, 2, 3)
v2 = Vector3(2, -1, 1)
v3 = v1 + v2
self.assertEqual(v3.x, 3)
self.assertEqual(v3.y, 1)
self.assertEqual(v3.z, 4)
def test_multiplication(self):
"""
Multiplying one L{Quaternion} by another returns a new L{Quaternion}
giving the product of the two.
"""
a = Quaternion.fromAxisAngle(Vector3(1, 0, 0), pi / 4)
b = Quaternion.fromAxisAngle(Vector3(1, 0, 0), pi / 6)
c = a * b
axis = c.getAxis()
angle = c.getAngle()
self.assertEqual(axis.x, 1)
self.assertEqual(axis.y, 0)
self.assertEqual(axis.z, 0)
self.assertAlmostEqual(angle, pi / 4 + pi / 6, 6)
def test_linearVelocity(self):
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(0, 0, 0))
obj = RigidBody(None, None, 1)
obj.setLinearVelocity(Vector3(1, 2, 3))
world.addRigidBody(obj)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
position = obj.getMotionState().getWorldTransform().getOrigin()
self.assertEquals(position.x, 1)
self.assertEquals(position.y, 2)
self.assertEquals(position.z, 3)
def test_instantiate(self):
"""
L{StaticPlaneShape} is initialized with a L{Vector3} giving its surface
normal and a I{plane constant} XXX TODO what the heck, plane constant?
"""
shape = StaticPlaneShape(Vector3(3, 5, 9), 1)
self.assertTrue(isinstance(shape, StaticPlaneShape))
def test_linearVelocity(self):
body = RigidBody()
body.setLinearVelocity(Vector3(1, 2, 3))
velocity = body.getLinearVelocity()
self.assertEquals(velocity.x, 1)
self.assertEquals(velocity.y, 2)
self.assertEquals(velocity.z, 3)
def test_origin(self):
transform = Transform()
transform.setOrigin(Vector3(1, 2, 3))
origin = transform.getOrigin()
self.assertEqual(origin.x, 1)
self.assertEqual(origin.y, 2)
self.assertEqual(origin.z, 3)
def test_fromAxisAngle(self):
"""
A L{Quaternion} can be constructed from a Vector3 giving an axis and a
scalar giving an angle from that axis.
"""
quat = Quaternion.fromAxisAngle(Vector3(0, 1, 0), 45)
self.assertTrue(isinstance(quat, Quaternion))
def test_warp(self):
self.controller.warp(Vector3(5, 7, 9))
transform = self.controller.ghost.getWorldTransform()
origin = transform.getOrigin()
self.assertEquals(origin.x, 5)
self.assertEquals(origin.y, 7)
self.assertEquals(origin.z, 9)
def test_gravity(self):
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(3, 2, 1))
gravity = world.getGravity()
self.assertEqual(gravity.x, 3)
self.assertEqual(gravity.y, 2)
self.assertEqual(gravity.z, 1)
def test_setIdentity(self):
transform = Transform()
transform.setOrigin(Vector3(2, 3, 4))
transform.setIdentity()
origin = transform.getOrigin()
self.assertEqual(origin.x, 0)
self.assertEqual(origin.y, 0)
self.assertEqual(origin.z, 0)
def test_fromConstructionInfo(self):
mass = 8
motion = DefaultMotionState()
shape = SphereShape(3)
inertia = Vector3(1, 2, 3)
transform = Transform()
transform.setOrigin(Vector3(5, 6, 7))
linearDamping = 0.3
angularDamping = 0.4
friction = 4.5
restition = 5.4
linearSleepingThreshold = 4.3
angularSleepingThreshold = 3.2
body = RigidBody.fromConstructionInfo(motion, shape, mass, inertia,
transform, linearDamping,
angularDamping, friction,
restition,
linearSleepingThreshold,
angularSleepingThreshold)
self.assertTrue(isinstance(body, RigidBody))
self.assertEqual(1. / mass, body.getInvMass())
self.assertTrue(motion is body.getMotionState())
self.assertTrue(shape is body.getCollisionShape())
self.assertAlmostEqual(1. / inertia.x,
body.getInvInertiaDiagLocal().x, 6)
self.assertAlmostEqual(1. / inertia.y,
body.getInvInertiaDiagLocal().y, 6)
self.assertAlmostEqual(1. / inertia.z,
body.getInvInertiaDiagLocal().z, 6)
# The worldTransform is ignored if a MotionState is supplied.
self.assertEqual(0, body.getWorldTransform().getOrigin().x)
self.assertEqual(0, body.getWorldTransform().getOrigin().y)
self.assertEqual(0, body.getWorldTransform().getOrigin().z)
self.assertAlmostEqual(linearDamping, body.getLinearDamping(), 6)
self.assertAlmostEqual(angularDamping, body.getAngularDamping(), 6)
self.assertAlmostEqual(friction, body.getFriction(), 6)
self.assertAlmostEqual(restition, body.getRestitution(), 6)
self.assertAlmostEqual(linearSleepingThreshold,
body.getLinearSleepingThreshold(), 6)
self.assertAlmostEqual(angularSleepingThreshold,
body.getAngularSleepingThreshold(), 6)
def test_applyImpulse(self):
body = RigidBody(mass=1.0)
body.applyImpulse(Vector3(1, 2, 3), Vector3(1, 1, 1))
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(0, 0, 0))
world.addRigidBody(body)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
transform = body.getMotionState().getWorldTransform()
position = transform.getOrigin()
self.assertEqual(position.x, 1.0)
self.assertEqual(position.y, 2.0)
self.assertEqual(position.z, 3.0)
rot = transform.getRotation()
self.assertNotEqual((rot.getX(), rot.getY(), rot.getZ(), rot.getW()),
(0, 0, 0, 1))
def test_worldTransform(self):
obj = CollisionObject()
trans = Transform()
trans.setOrigin(Vector3(3, 5, 7))
obj.setWorldTransform(trans)
origin = obj.getWorldTransform().getOrigin()
self.assertEquals(origin.x, 3)
self.assertEquals(origin.y, 5)
self.assertEquals(origin.z, 7)
def test_worldTransform(self):
xform = Transform()
xform.setOrigin(Vector3(3, 5, 7))
state = DefaultMotionState()
state.setWorldTransform(xform)
xform = state.getWorldTransform()
origin = xform.getOrigin()
self.assertEqual(origin.x, 3)
self.assertEqual(origin.y, 5)
self.assertEqual(origin.z, 7)
def test_cannotInstantiateWithoutHalfExtents(self):
"""
If a Vector3 is not passed as the only argument to the cylinder's
initializer, L{TypeError} is raised.
"""
self.assertRaises(TypeError, CylinderShape)
self.assertRaises(TypeError, CylinderShape, 1)
self.assertRaises(TypeError, CylinderShape, None)
self.assertRaises(TypeError, CylinderShape, "foo")
self.assertRaises(TypeError, CylinderShape, (1, 2, 3))
self.assertRaises(TypeError, CylinderShape, Vector3(1, 2, 3), 5)
def test_getHalfExtentsWithoutMargin(self):
"""
L{CylinderShape.getHalfExtentsWithoutMargin} returns the cylinder's half
extents vector.
"""
halfExtents = Vector3(2, 5, 7)
shape = CylinderShape(halfExtents)
result = shape.getHalfExtentsWithoutMargin()
self.assertEqual(result.x, 1.9600000381469727)
self.assertEqual(result.y, 4.9600000381469727)
self.assertEqual(result.z, 6.9600000381469727)
def test_cycle(self):
world = DiscreteDynamicsWorld()
class Cheat(RigidBody):
pass
body = Cheat(None, BoxShape(Vector3(3, 4, 5)))
body.cycle = world
world.addRigidBody(body)
del body, world
import gc
gc.collect()
def test_lines(self):
"""
Some lines are drawn using the debug drawer when
L{CollisionWorld.debugDrawWorld} is called.
"""
obj = CollisionObject()
shape = BoxShape(Vector3(1, 2, 3))
obj.setCollisionShape(shape)
self.world.addCollisionObject(obj)
self.world.debugDrawWorld()
self.assertTrue(len(self.recorder.lines) > 0)
for line in self.recorder.lines:
self.assertEquals(len(line), 9)
def test_dot(self):
"""
L{Vector3.dot} accepts another L{Vector3} and returns a float which is
the result of the dot product of the two vectors.
"""
self.assertEqual(Vector3(1, 0, 0).dot(Vector3(0, 1, 0)), 0)
self.assertEqual(Vector3(1, 0, 0).dot(Vector3(1, 0, 0)), 1)
self.assertEqual(Vector3(2, 0, 0).dot(Vector3(2, 0, 0)), 4)
def test_normalized(self):
"""
L{Vector3.normalized} returns a new L{Vector3} pointing in the same
direction as the first L{Vector3} but with unit magnitude.
"""
v = Vector3(3, 5, 7)
n = v.normalized()
self.assertEqual(v.x, 3)
self.assertEqual(v.y, 5)
self.assertEqual(v.z, 7)
self.assertAlmostEqual(n.x, 0.3292927, 6)
self.assertAlmostEqual(n.y, 0.5488213, 6)
self.assertAlmostEqual(n.z, 0.7683498, 6)
self.assertAlmostEqual(n.x**2 + n.y**2 + n.z**2, 1.0, 6)
def test_stepSimulation(self):
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(1, 2, 3))
obj = RigidBody(None, None, 1)
world.addRigidBody(obj)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
position = obj.getMotionState().getWorldTransform().getOrigin()
# Unfortunately, there is some error (as compared to physical reality)
# in Bullet's results. My fault? Bullet's fault? I'm not sure.
self.assertEqual(position.x, 0.5 + 0.5 / expectedSteps)
self.assertEqual(position.y, 1.0 + 1.0 / expectedSteps)
self.assertEqual(position.z, 1.5 + 1.5 / expectedSteps)
def test_scalarMultiplication(self):
"""
A L{Vector3} instance multiplied by an integer or a float results in a
new L{Vector3} instance scaled by that amount relative to the original.
"""
v = Vector3(2, 4, 6)
v2 = v * 2
v3 = v * 3.5
self.assertEqual(v.x, 2)
self.assertEqual(v.y, 4)
self.assertEqual(v.z, 6)
self.assertEqual(v2.x, 4)
self.assertEqual(v2.y, 8)
self.assertEqual(v2.z, 12)
self.assertEqual(v3.x, 7)
self.assertEqual(v3.y, 14)
self.assertEqual(v3.z, 21)
def test_getOverlappingPairCache(self):
broadphase = AxisSweep3(Vector3(0, 0, 0), Vector3(1, 1, 1))
cache = broadphase.getOverlappingPairCache()
self.assertTrue(isinstance(cache, OverlappingPairCache))
def objAt(pos):
obj = RigidBody(None, BoxShape(Vector3(1, 1, 1)), 1.0)
xform = Transform()
xform.setOrigin(pos)
obj.setWorldTransform(xform)
return obj
def test_addBoxShape(self):
world = CollisionWorld()
obj = CollisionObject()
shape = BoxShape(Vector3(2, 3, 4))
obj.setCollisionShape(shape)
world.addCollisionObject(obj)
def test_setVelocityForTimeInterval(self):
self.controller.setVelocityForTimeInterval(Vector3(12.0, 0, 0), 6.0)
