dcs-retribution/game/sim/gamelooptimer.py
Dan Albert ce4c73098d
Add locking to some UI actions.
This is by no means complete. The bugs that this solves were already in
6.x, but we'd hidden the speed controls for the sim in that release, and
have always said that anything done after pressing "go" the first time
is undefined behavior. This is the first step on making those mid-sim
actions behave correctly.

UI actions such as creating a new package need to be executed between
ticks of the sim. We can either do this synchronously by blocking the UI
until the tick is done executing, acquiring a lock on the sim, executing
the action, then releasing the lock; or asynchronously by queueing
events and letting the sim execute them when it completes the current
tick (or instantly if the sim is paused).

Anything that comes from the new UI (currently just the map) must be
asynchronous because it goes through the REST API, but for the old UI
it's simpler (and because the lock will only be acquired as quickly as
the user can act, shouldn't slow anything down) to do this
synchronously, since it's difficult to use coroutines in Qt.

https://github.com/dcs-liberation/dcs_liberation/issues/1680
2024-11-16 19:09:42 +01:00

62 lines
2.0 KiB
Python

from collections.abc import Iterator
from contextlib import contextmanager
from threading import RLock, Timer
from typing import Callable, Optional
from .simspeedsetting import SimSpeedSetting
class GameLoopTimer:
def __init__(self, callback: Callable[[], None]) -> None:
self.callback = callback
self.simulation_speed = SimSpeedSetting.PAUSED
self._timer: Optional[Timer] = None
# Reentrant to allow a single thread nested use of `locked_pause`.
self._timer_lock = RLock()
def set_speed(self, simulation_speed: SimSpeedSetting) -> None:
with self._timer_lock:
self._set_speed(simulation_speed)
def stop(self) -> None:
with self._timer_lock:
self._stop()
@contextmanager
def locked_pause(self) -> Iterator[None]:
# NB: This must be a locked _pause_ and not a locked speed, because nested use
# of this method is allowed. That's okay if all nested callers set the same
# speed (paused), but not okay if a parent locks a speed and a child locks
# another speed. That's okay though, because we're unlikely to ever want to lock
# any speed but paused.
with self._timer_lock:
old_speed = self.simulation_speed
self._stop()
try:
yield
finally:
self._set_speed(old_speed)
def _set_speed(self, simulation_speed: SimSpeedSetting) -> None:
self._stop()
self.simulation_speed = simulation_speed
self._recreate_timer()
def _stop(self) -> None:
if self._timer is not None:
self._timer.cancel()
def _recreate_timer(self) -> None:
self._stop()
factor = self.simulation_speed.speed_factor
if not factor:
self._timer = None
return None
self._timer = Timer(1 / factor, self._tick)
self._timer.start()
def _tick(self) -> None:
self.callback()
with self._timer_lock:
self._recreate_timer()