removed uml diagrams

[pmachines.git] / game / items / domino.py

from panda3d.bullet import BulletBoxShape, BulletRigidBodyNode, BulletGhostNode
from game.items.item import Item, StillStrategy


class Domino(Item):

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

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


class UpStrategy(StillStrategy):

    def __init__(self, np, tgt_degrees):
        super().__init__(np)
        self._tgt_degrees = tgt_degrees
        self._np = np

    def win_condition(self):
        return super().win_condition() and abs(self._np.get_r()) <= self._tgt_degrees


class DownStrategy(StillStrategy):

    def __init__(self, np, tgt_degrees):
        super().__init__(np)
        self._tgt_degrees = tgt_degrees
        self._np = np

    def win_condition(self):
        return self._np.get_z() < -6 or super().win_condition() and abs(self._np.get_r()) >= self._tgt_degrees