__docformat__ = "reStructuredText"
from collections import namedtuple
from ._chipmunk_cffi import lib, ffi
from .vec2d import Vec2d
from .body import Body
class SpaceDebugColor(namedtuple("SpaceDebugColor", ["r","g","b","a"])):
"""Color tuple used by the debug drawing API.
"""
__slots__ = ()
def as_int(self):
return int(self[0]), int(self[1]), int(self[2]), int(self[3])
def as_float(self):
return self[0]/255., self[1]/255., self[2]/255., self[3]/255.
[docs]class SpaceDebugDrawOptions(object):
"""SpaceDebugDrawOptions configures debug drawing.
If appropriate its usually easy to use the supplied draw implementations
directly: pymunk.pygame_util, pymunk.pyglet_util and pymunk.matplotlib_util.
"""
DRAW_SHAPES = lib.CP_SPACE_DEBUG_DRAW_SHAPES
"""Draw shapes"""
DRAW_CONSTRAINTS = lib.CP_SPACE_DEBUG_DRAW_CONSTRAINTS
"""Draw constraints"""
DRAW_COLLISION_POINTS = lib.CP_SPACE_DEBUG_DRAW_COLLISION_POINTS
"""Draw collision points"""
shape_dynamic_color = SpaceDebugColor(52,152,219,255)
shape_static_color = SpaceDebugColor(149,165,166,255)
shape_kinematic_color = SpaceDebugColor(39,174,96,255)
shape_sleeping_color = SpaceDebugColor(114,148,168,255)
[docs] def __init__(self):
_options = ffi.new("cpSpaceDebugDrawOptions *")
self._options = _options
self.shape_outline_color = (44,62,80,255)
self.constraint_color = (142,68,173,255)
self.collision_point_color = (231,76,60,255)
@ffi.callback("cpSpaceDebugDrawCircleImpl")
def f1(pos, angle, radius, outline_color, fill_color, data):
self.draw_circle(
Vec2d._fromcffi(pos), angle, radius,
self._c(outline_color), self._c(fill_color))
_options.drawCircle = f1
@ffi.callback("cpSpaceDebugDrawSegmentImpl")
def f2(a, b, color, data):
self.draw_segment(
Vec2d._fromcffi(a), Vec2d._fromcffi(b), self._c(color))
_options.drawSegment = f2
@ffi.callback("cpSpaceDebugDrawFatSegmentImpl")
def f3(a, b, radius, outline_color, fill_color, data):
self.draw_fat_segment(
Vec2d._fromcffi(a), Vec2d._fromcffi(b), radius,
self._c(outline_color), self._c(fill_color))
_options.drawFatSegment = f3
@ffi.callback("cpSpaceDebugDrawPolygonImpl")
def f4(count, verts, radius, outline_color, fill_color, data):
vs = []
for i in range(count):
vs.append(Vec2d._fromcffi(verts[i]))
self.draw_polygon(
vs, radius,
self._c(outline_color), self._c(fill_color))
_options.drawPolygon = f4
@ffi.callback("cpSpaceDebugDrawDotImpl")
def f5(size, pos, color, data):
self.draw_dot(size, Vec2d._fromcffi(pos), self._c(color))
_options.drawDot = f5
@ffi.callback("cpSpaceDebugDrawColorForShapeImpl")
def f6(_shape, data):
space = ffi.from_handle(data)
shape = space._get_shape(_shape)
return self.color_for_shape(shape)
_options.colorForShape = f6
self.flags = SpaceDebugDrawOptions.DRAW_SHAPES | \
SpaceDebugDrawOptions.DRAW_CONSTRAINTS | \
SpaceDebugDrawOptions.DRAW_COLLISION_POINTS
self._callbacks = [f1,f2,f3,f4,f5,f6]
def _get_shape_outline_color(self):
print(self._options.shapeOutlineColor)
return self._c(self._options.shapeOutlineColor)
def _set_shape_outline_color(self, c):
self._options.shapeOutlineColor = c
shape_outline_color = property(_get_shape_outline_color,
_set_shape_outline_color)
def _get_constraint_color(self):
return self._c(self._options.constraintColor)
def _set_constraint_color(self, c):
self._options.constraintColor = c
constraint_color = property(_get_constraint_color,
_set_constraint_color)
def _get_collision_point_color(self):
return self._c(self._options.collisionPointColor)
def _set_collision_point_color(self, c):
self._options.collisionPointColor = c
collision_point_color = property(_get_collision_point_color,
_set_collision_point_color)
def __enter__(self):
pass
def __exit__(self, type, value, traceback):
pass
def _c(self, color):
return SpaceDebugColor(color.r, color.g, color.b, color.a)
def _get_flags(self):
return self._options.flags
def _set_flags(self, f):
self._options.flags = f
flags = property(_get_flags, _set_flags)
[docs] def draw_circle(self, *args):
print("draw_circle", args)
[docs] def draw_segment(self, *args):
print("draw_segment", args)
[docs] def draw_fat_segment(self, *args):
print("draw_fat_segment", args)
[docs] def draw_polygon(self, *args):
print("draw_polygon", args)
[docs] def draw_dot(self, *args):
print("draw_dot", args)
[docs] def color_for_shape(self, shape):
if hasattr(shape, "color"):
return shape.color
color = self.shape_dynamic_color
if shape.body != None:
if shape.body.body_type == Body.STATIC:
color = self.shape_static_color
elif shape.body.body_type == Body.KINEMATIC:
color = self.shape_kinematic_color
elif shape.body.is_sleeping:
color = self.shape_sleeping_color
return color