functional tests: editor

[pmachines.git] / pmachines / items / box.py

from panda3d.bullet import BulletBoxShape
from pmachines.items.item import Item, ItemStrategy


class Box(Item):

    def __init__(self, world, plane_node, cb_inst, curr_bottom, repos, json, model_scale=1, mass=1, pos=(0, 0, 0), r=0, count=0, restitution=.5, friction=.8):
        super().__init__(world, plane_node, cb_inst, curr_bottom, repos, 'assets/models/bam/box/box.bam', json, model_scale=model_scale, mass=mass, pos=pos, r=r, count=count, restitution=restitution, friction=friction)

    def _set_shape(self, apply_scale=True):
        self.node.add_shape(BulletBoxShape((.5, .5, .5)))


class HitStrategy(ItemStrategy):

    def __init__(self, hit_by, node, world):
        self._hit_by = hit_by
        self._node = node
        self._world = world

    def win_condition(self):
        for contact in self._world.contact_test(self._node).get_contacts():
            other = contact.get_node1() if contact.get_node0() == self._node else contact.get_node0()
            if other == self._hit_by.node:
                return True