1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
|
#!/usr/bin/python
#
# C++ version Copyright (c) 2006-2007 Erin Catto http://www.gphysics.com
# Python version Copyright (c) 2008 kne / sirkne at gmail dot com
#
# Implemented using the pybox2d SWIG interface for Box2D (pybox2d.googlecode.com)
#
# This software is provided 'as-is', without any express or implied
# warranty. In no event will the authors be held liable for any damages
# arising from the use of this software.
# Permission is granted to anyone to use this software for any purpose,
# including commercial applications, and to alter it and redistribute it
# freely, subject to the following restrictions:
# 1. The origin of this software must not be misrepresented; you must not
# claim that you wrote the original software. If you use this software
# in a product, an acknowledgment in the product documentation would be
# appreciated but is not required.
# 2. Altered source versions must be plainly marked as such, and must not be
# misrepresented as being the original software.
# 3. This notice may not be removed or altered from any source distribution.
from test_main import *
from test_main import fwContactTypes
import math
import pprint
pp = pprint.PrettyPrinter(indent=4)
# Contributed by caspin (C++ version)
class ContactCallbackTest (Framework):
name="ContactCallbackTest"
ball=None
bullet=None
ball_shape=None
_pickle_vars = ['ball', 'bullet', 'ball_shape']
def __init__(self):
super(ContactCallbackTest, self).__init__()
groundBody = self.world.GetGroundBody()
sd=box2d.b2PolygonDef()
sd.friction = 0
sd.vertices = [(10.0, 10.0), (9.0, 7.0), (10.0, 0.0)]
sd.userData = 1
groundBody.CreateShape(sd)
sd.vertices = [(9.0, 7.0), (8.0, 0.0), (10.0, 0.0)]
sd.userData = 2
groundBody.CreateShape(sd)
sd.vertices = [(9.0, 7.0), (8.0, 5.0), (8.0, 0.0)]
sd.userData = 3
groundBody.CreateShape(sd)
sd.vertices = [(8.0, 5.0), (7.0, 4.0), (8.0, 0.0)]
sd.userData = 4
groundBody.CreateShape(sd)
sd.vertices = [(7.0, 4.0), (5.0, 0.0), (8.0, 0.0)]
sd.userData = 5
groundBody.CreateShape(sd)
sd.vertices = [(7.0, 4.0), (5.0, 3.0), (5.0, 0.0)]
sd.userData = 6
groundBody.CreateShape(sd)
sd.vertices = [(5.0, 3.0), (2.0, 2.0), (5.0, 0.0)]
sd.userData = 7
groundBody.CreateShape(sd)
sd.vertices = [(2.0, 2.0), (0.0, 0.0), (5.0, 0.0)]
sd.userData = 8
groundBody.CreateShape(sd)
sd.vertices = [(2.0, 2.0), (-2.0, 2.0), (0.0, 0.0)]
sd.userData = 9
groundBody.CreateShape(sd)
sd.vertices = [(-5.0, 0.0), (0.0, 0.0), (-2.0, 2.0)]
sd.userData = 10
groundBody.CreateShape(sd)
sd.vertices = [(-5.0, 0.0), (-2.0, 2.0), (-5.0, 3.0)]
sd.userData = 11
groundBody.CreateShape(sd)
sd.vertices = [(-5.0, 0.0), (-5.0, 3.0), (-7.0, 4.0)]
sd.userData = 12
groundBody.CreateShape(sd)
sd.vertices = [(-8.0, 0.0), (-5.0, 0.0), (-7.0, 4.0)]
sd.userData = 13
groundBody.CreateShape(sd)
sd.vertices = [(-8.0, 0.0), (-7.0, 4.0), (-8.0, 5.0)]
sd.userData = 14
groundBody.CreateShape(sd)
sd.vertices = [(-8.0, 0.0), (-8.0, 5.0), (-9.0, 7.0)]
sd.userData = 15
groundBody.CreateShape(sd)
sd.vertices = [(-10.0, 0.0), (-8.0, 0.0), (-9.0, 7.0)]
sd.userData = 16
groundBody.CreateShape(sd)
sd.vertices = [(-10.0, 0.0), (-9.0, 7.0), (-10.0, 10.0)]
sd.userData = 17
groundBody.CreateShape(sd)
sd.SetAsBox(.5,6,(10.5,6),0)
groundBody.CreateShape(sd)
sd.SetAsBox(.5,6,(-10.5,6),0)
groundBody.CreateShape(sd)
bd=box2d.b2BodyDef()
bd.position=(9.5,60)
self.ball = self.world.CreateBody( bd )
cd=box2d.b2PolygonDef()
cd.vertexCount = 8
w = 1.0
b = w / (2.0 + math.sqrt(2.0))
s = math.sqrt(2.0) * b
cd.vertices = [( 0.5 * s, 0.0),
( 0.5 * w, b),
( 0.5 * w, b + s),
( 0.5 * s, w),
(-0.5 * s, w),
(-0.5 * w, b + s),
(-0.5 * w, b),
(-0.5 * s, 0.0) ]
cd.density = 1.0
cd.userData = 'BALL'
self.ball_shape = self.ball.CreateShape(cd)
self.ball.SetMassFromShapes()
def Step(self, settings):
strings = []
name_dict = { fwContactTypes.contactAdded : "added",
fwContactTypes.contactRemoved : "removed",
fwContactTypes.contactPersisted : "persisted" }
strings = ["%s: %s, %s (%s)" % (name_dict[point.state], point.shape1.userData, point.shape2.userData, point.id.key)
for point in self.points]
if len(strings) > 15:
strings = strings[:14]
for string in strings:
self.DrawStringCR(string)
super(ContactCallbackTest, self).Step(settings)
if __name__=="__main__":
main(ContactCallbackTest)
|