From 68e785f0faf8684d7deb3c94a591fa9ca77ee67c Mon Sep 17 00:00:00 2001 From: Ruben van de Ven Date: Fri, 16 May 2025 09:53:26 +0200 Subject: [PATCH] draw differences from predictions --- trap/stage.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/trap/stage.py b/trap/stage.py index bb1869e..24c4605 100644 --- a/trap/stage.py +++ b/trap/stage.py @@ -60,12 +60,13 @@ class ProceduralChain(): DeltaT = float # delta_t class DiffSegment(): - DRAW_DECAY_SPEED = 5 + DRAW_DECAY_SPEED = 25 def __init__(self, prediction: ProjectedTrack): self.ptrack = prediction self._last_diff_frame_idx = 0 self.finished = False + self.ready = False self.points: List[Coordinate] = [] self._drawn_points = [] @@ -130,9 +131,11 @@ class DiffSegment(): if np.isclose(self._drawn_points[-1], target, atol=.05).all(): # TODO: might want to migrate to np.isclose() if len(self._drawn_points) == len(self.points): + self.ready = True return # done until a new point is added # add new point as drawing head self._drawn_points.append(self._drawn_points[-1]) + self.ready = False x = exponentialDecayRounded(self._drawn_points[-1][0], target[0], self.DRAW_DECAY_SPEED, dt, .05) y = exponentialDecayRounded(self._drawn_points[-1][1], target[1], self.DRAW_DECAY_SPEED, dt, .05)