def Grapple(self,pos):
if self.joint:
self.UnGrapple()
return
diff = pos - self.GetPos()
distance,angle = cmath.polar(complex(diff.x,diff.y))
angle = (angle - (math.pi/2) - self.GetAngle())%(math.pi*2)
#0 = pi*2 is straight ahead, pi is behind.
#so 0.75 pi to 1.25 pi is allowed
if not (angle <= self.min_shoot and angle >= self.max_shoot):
return
if distance > self.max_grapple:
return
#We need to determine if this point is in any objects...
#We'll create a small AABB around the point, and get the list of potentially intersecting shapes,
#then test each one to see if the point is inside it
aabb = box2d.b2AABB()
phys_pos = pos*self.physics.scale_factor
#First of all make sure it's not inside us
trans = box2d.b2XForm()
trans.SetIdentity()
p = phys_pos - Point(*self.body.position)
if self.shapeI.TestPoint(trans,tuple(p)):
return
aabb.lowerBound.Set(phys_pos.x-0.1,phys_pos.y-0.1)
aabb.upperBound.Set(phys_pos.x+0.1,phys_pos.y+0.1)
(count,shapes) = self.physics.world.Query(aabb,10)
for shape in shapes:
trans = box2d.b2XForm()
trans.SetIdentity()
p = phys_pos - Point(*shape.GetBody().position)
if shape.TestPoint(trans,tuple(p)):
self.touching = shape
self.contact = p
break
else:
self.touching = None
self.contact = None
if not self.touching:
return
globals.sounds.grapple.play()
#Tell the other body that it's in a joint with us so that
target = self.touching.userData
if target == None:
self.touching = None
self.contact = None
return
target.parent_joint = self
self.child_joint = target
joint = box2d.b2DistanceJointDef()
joint.Initialize(self.body,self.touching.GetBody(),self.body.GetWorldCenter(),tuple(phys_pos))
joint.collideConnected = True
self.joint = self.physics.world.CreateJoint(joint)
self.grappled = True
self.grapple_quad.Enable()
def GetObjectAtPoint(self,pos):
aabb = box2d.b2AABB()
phys_pos = pos*self.scale_factor
aabb.lowerBound.Set(phys_pos.x-0.1,phys_pos.y-0.1)
aabb.upperBound.Set(phys_pos.x+0.1,phys_pos.y+0.1)
(count,shapes) = self.world.Query(aabb,10)
for shape in shapes:
trans = box2d.b2XForm()
trans.SetIdentity()
p = phys_pos - Point(*shape.GetBody().position)
if shape.TestPoint(trans,tuple(p)):
return shape.userData
return None
