bastd.actor.onscreencountdown

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Defines Actor Type(s)."""
  4
  5from __future__ import annotations
  6
  7from typing import TYPE_CHECKING
  8
  9import ba
 10
 11if TYPE_CHECKING:
 12    from typing import Any, Callable
 13
 14
 15class OnScreenCountdown(ba.Actor):
 16    """A Handy On-Screen Timer.
 17
 18    category: Gameplay Classes
 19
 20    Useful for time-based games that count down to zero.
 21    """
 22
 23    def __init__(self, duration: int, endcall: Callable[[], Any] | None = None):
 24        """Duration is provided in seconds."""
 25        super().__init__()
 26        self._timeremaining = duration
 27        self._ended = False
 28        self._endcall = endcall
 29        self.node = ba.newnode(
 30            'text',
 31            attrs={
 32                'v_attach': 'top',
 33                'h_attach': 'center',
 34                'h_align': 'center',
 35                'color': (1, 1, 0.5, 1),
 36                'flatness': 0.5,
 37                'shadow': 0.5,
 38                'position': (0, -70),
 39                'scale': 1.4,
 40                'text': '',
 41            },
 42        )
 43        self.inputnode = ba.newnode(
 44            'timedisplay',
 45            attrs={
 46                'time2': duration * 1000,
 47                'timemax': duration * 1000,
 48                'timemin': 0,
 49            },
 50        )
 51        self.inputnode.connectattr('output', self.node, 'text')
 52        self._countdownsounds = {
 53            10: ba.getsound('announceTen'),
 54            9: ba.getsound('announceNine'),
 55            8: ba.getsound('announceEight'),
 56            7: ba.getsound('announceSeven'),
 57            6: ba.getsound('announceSix'),
 58            5: ba.getsound('announceFive'),
 59            4: ba.getsound('announceFour'),
 60            3: ba.getsound('announceThree'),
 61            2: ba.getsound('announceTwo'),
 62            1: ba.getsound('announceOne'),
 63        }
 64        self._timer: ba.Timer | None = None
 65
 66    def start(self) -> None:
 67        """Start the timer."""
 68        globalsnode = ba.getactivity().globalsnode
 69        globalsnode.connectattr('time', self.inputnode, 'time1')
 70        self.inputnode.time2 = (
 71            globalsnode.time + (self._timeremaining + 1) * 1000
 72        )
 73        self._timer = ba.Timer(1.0, self._update, repeat=True)
 74
 75    def on_expire(self) -> None:
 76        super().on_expire()
 77
 78        # Release callbacks/refs.
 79        self._endcall = None
 80
 81    def _update(self, forcevalue: int | None = None) -> None:
 82        if forcevalue is not None:
 83            tval = forcevalue
 84        else:
 85            self._timeremaining = max(0, self._timeremaining - 1)
 86            tval = self._timeremaining
 87
 88        # if there's a countdown sound for this time that we
 89        # haven't played yet, play it
 90        if tval == 10:
 91            assert self.node
 92            assert isinstance(self.node.scale, float)
 93            self.node.scale *= 1.2
 94            cmb = ba.newnode('combine', owner=self.node, attrs={'size': 4})
 95            cmb.connectattr('output', self.node, 'color')
 96            ba.animate(cmb, 'input0', {0: 1.0, 0.15: 1.0}, loop=True)
 97            ba.animate(cmb, 'input1', {0: 1.0, 0.15: 0.5}, loop=True)
 98            ba.animate(cmb, 'input2', {0: 0.1, 0.15: 0.0}, loop=True)
 99            cmb.input3 = 1.0
100        if tval <= 10 and not self._ended:
101            ba.playsound(ba.getsound('tick'))
102        if tval in self._countdownsounds:
103            ba.playsound(self._countdownsounds[tval])
104        if tval <= 0 and not self._ended:
105            self._ended = True
106            if self._endcall is not None:
107                self._endcall()