From 191893b5f035acc0f9d5dfb20c2819065a75c42c Mon Sep 17 00:00:00 2001 From: Alex Levenson Date: Thu, 10 Jul 2008 19:01:23 +0000 Subject: Adding more types of joints... --- (limited to 'elements') diff --git a/elements/add_objects.py b/elements/add_objects.py index 72dc21e..6ce7773 100644 --- a/elements/add_objects.py +++ b/elements/add_objects.py @@ -445,8 +445,52 @@ class Add: # Define the body x, y = c = tools_poly.calc_center(vertices_orig_reduced) return self.poly((x,y), vertices, dynamic, density, restitution, friction) - - def joint(self, *args): + + # Alex Levenson's added joint methods: + def distanceJoint(self,b1,b2,p1,p2): + # Distance Joint + p1 = self.parent.to_world(p1) + p2 = self.parent.to_world(p2) + p1x, p1y = p1 + p2x, p2y = p2 + p1x /= self.parent.ppm + p1y /= self.parent.ppm + p2x /= self.parent.ppm + p2y /= self.parent.ppm + p1 = box2d.b2Vec2(p1x, p1y) + p2 = box2d.b2Vec2(p2x, p2y) + jointDef = box2d.b2DistanceJointDef() + jointDef.Initialize(b1, b2, p1, p2) + jointDef.collideConnected = True + self.parent.world.CreateJoint(jointDef) + + def fixedJoint(self, *args): + if len(args) == 2: + # Fixed Joint to the Background, don't assume the center of the body + b1 = self.parent.world.GetGroundBody() + b2 = args[0] + p1 = self.parent.to_world(args[1]) + p1x, p1y = p1 + p1x /= self.parent.ppm + p1y /= self.parent.ppm + p1 = box2d.b2Vec2(p1x, p1y) + + jointDef = box2d.b2RevoluteJointDef() + jointDef.Initialize(b1, b2, p1) + + self.parent.world.CreateJoint(jointDef) + elif len(args) == 1: + # Fixed Joint to the Background, assume the center of the body + b1 = self.parent.world.GetGroundBody() + b2 = args[0] + p1 = b2.GetWorldCenter() + + jointDef = box2d.b2RevoluteJointDef() + jointDef.Initialize(b1, b2, p1) + + self.parent.world.CreateJoint(jointDef) + + def joint(self, *args): print "* Add Joint:", args if len(args) == 4: @@ -476,22 +520,7 @@ class Add: elif len(args) == 3: # Revolute Joint pass - - elif len(args) == 2: - - # Revolute Joint to the Background, don't assume the center of the body - b1 = self.parent.world.GetGroundBody() - b2 = args[0] - p1 = self.parent.to_world(args[1]) - p1x, p1y = p1 - p1x /= self.parent.ppm - p1y /= self.parent.ppm - p1 = box2d.b2Vec2(p1x, p1y) - - jointDef = box2d.b2RevoluteJointDef() - jointDef.Initialize(b1, b2, p1) - - self.parent.world.CreateJoint(jointDef) + elif len(args) == 1: # Revolute Joint to the Background, assume the center of the body b1 = self.parent.world.GetGroundBody() -- cgit v0.9.1