animation

Author: intendednull

src/animate.py

@@ -4,9 +4,9 @@
 import operator
 import time
 import math
+import random
 from typing import NamedTuple, Callable, TypeVar, Generator
 from enum import Enum
-from dataclasses import dataclass
 from functools import partialmethod
 
 
@@ -36,8 +36,11 @@ def run(self):
             try:
                 move = next(motion)
                 move()
+                self.canvas.update()
             except StopIteration:
                 self._motions.remove(motion)
+        self.canvas.update()
+        # self.canvas.after(10, self.run)
 
     def add(self, motion: Motion):
         self._motions.add(motion.start())
@@ -54,16 +57,12 @@ def running(self):
         return bool(self._motions)
 
 
-@dataclass
 class Motion:
-    """
-
-    """
-    canvas: tk.Canvas
-    id: int
-    end: Coord
-
-    speed: int = 1
+    def __init__(self, canvas: tk.Canvas, id: str, end: Coord, speed: float = 1):
+        self.canvas = canvas
+        self.id = id
+        self.end = end
+        self.speed = speed ** 3
 
     def __iter__(self):
         return self.start()
@@ -81,13 +80,16 @@ def start(self) -> Generator[Callable]:
         """
         The entry point for generating move commands.
         """
-        self.time = time.time()
-        self.start = self.current
-        self.distance = self.start.distance(self.end)
-        self.speed = self.speed ** 3
+        self.reset()
         while self.current != self.end:
+            print(self.current, self.end)
             yield self.move
 
+    def reset(self):
+        self.time = time.time()
+        self.beg = self.current
+        self.distance = self.beg.distance(self.end)
+
     def move(self):
         self.canvas.move(self.id, *self.increment)
         self.canvas.update_idletasks()
@@ -102,13 +104,16 @@ def time(self, val):
 
     @property
     def increment(self):
-        mult = (self.time * self.speed) / self.distance
-        point = (self.end - self.start) * mult + self.start
-
-        if point.distance(self.end) > self.journey:
+        future = self.future
+        if future.distance(self.end) > self.journey:
             return self.end - self.current
         else:
-            return point - self.current
+            return future - self.current
+
+    @property
+    def future(self):
+        mult = (self.time * self.speed) / self.distance
+        return (self.end - self.beg) * mult + self.beg
 
     @property
     def current(self):
@@ -119,6 +124,60 @@ def journey(self):
         return self.current.distance(self.end)
 
 
+class BounceBall(Motion):
+
+    chaos = 3
+
+    def kick(self, direction: Point):
+        self.canvas.update()
+
+        c1, c2 = -self.chaos, self.chaos
+        chaoticx, chaoticy = random.randint(c1, c2), random.randint(c1, c2)
+        self.direction = direction + Coord(chaoticx, chaoticy)
+        self.end = self.direction * self.canvas.winfo_height()
+        self.reset()
+
+    @property
+    def increment(self):
+        bounce = self.get_bounce()
+        if bounce != Coord(0, 0):
+            self.kick(bounce)
+        return self.future - self.current
+
+    def get_bounce(self):
+        x1, y1, x2, y2 = self.canvas.bbox(self.id)
+        bounce = Coord(0, 0)
+        if x1 <= self.bound_x1:
+            # print('x1', x1, self.bound_x1)
+            bounce += Direction.RIGHT
+        if y1 <= self.bound_y1:
+            # print('y1', y1, self.bound_y1)
+            bounce += Direction.DOWN
+        if x2 >= self.bound_x2:
+            # print('x2', x2, self.bound_x2)
+            bounce += Direction.LEFT
+        if y2 >= self.bound_y2:
+            # print('y2', y2, self.bound_y2)
+            bounce += Direction.UP
+        return bounce
+
+    @property
+    def bound_x1(self):
+        return self.canvas.winfo_x()
+
+    @property
+    def bound_y1(self):
+        return self.canvas.winfo_y()
+
+    @property
+    def bound_x2(self):
+        return self.bound_x1 + self.canvas.winfo_width()
+
+    @property
+    def bound_y2(self):
+        return self.bound_y1 + self.canvas.winfo_height()
+
+
 class Coord(NamedTuple):
     """
     Helper class for managing coordinate values.
@@ -149,10 +208,10 @@ class Coord(NamedTuple):
     Operand = TypeVar('Operand', 'Coord', float)
 
     def __apply(self, op: Callable, other: Coord.Operand) -> Coord:
-        if not isinstance(other, self.__class__):
-            other = self.__class__(other, other)
         if isinstance(other, Direction):
             other = other.value
+        elif not isinstance(other, self.__class__):
+            other = self.__class__(other, other)
 
         x = op(self.x, other.x)
         y = op(self.y, other.y)
@@ -183,6 +242,9 @@ def distance(self, other: Coord) -> float:
         diff = other - self
         return math.hypot(*diff)
 
+    def flip(self):
+        return Coord(0, 0) - self
+
     __add__ = partialmethod(__apply, operator.add)
     __sub__ = partialmethod(__apply, operator.sub)
     __mul__ = partialmethod(__apply, operator.mul)