koperkapel/actors/orientatedsurf.py

   1 """ Actor that takes a surface. """
   2
   3 from pygame.transform import rotate
   4 from pgzero.actor import POS_TOPLEFT, ANCHOR_CENTER
   5
   6 from .surf import SurfActor
   7
   8
   9 class OrientatedSurfActor(SurfActor):
  10     def __init__(self, surf, pos=POS_TOPLEFT, anchor=ANCHOR_CENTER, angle=0,
  11                  **kwargs):
  12         self._handle_unexpected_kwargs(kwargs)
  13
  14         self._angle = angle
  15         self._base_surf = surf
  16         super().__init__(surf, pos, anchor)
  17         self._init_position(pos, anchor, **kwargs)
  18
  19     @property
  20     def angle(self):
  21         return self._angle
  22
  23     @angle.setter
  24     def angle(self, angle):
  25         self._angle = angle
  26         self._surf = self._orig_surf = rotate(self._base_surf, angle)
  27
  28     @property
  29     def surf(self):
  30         """ The original surface displaying the actor. """
  31         return self._orig_surf
  32
  33     @surf.setter
  34     def surf(self, surf):
  35         self._base_surf = surf
  36         self._orig_surf = self._surf = rotate(surf, self._angle)
  37         self._update_pos()