Update myturtle.py

myturtle.py

@@ -1,49 +1,28 @@
-import re
-from PIL import Image
-from matplotlib import pyplot as plt
-import matplotlib as mpl
-import matplotlib.style as mplstyle
 import numpy as np
-
-mpl.use('Agg')
-mpl.rcParams["path.simplify_threshold"] = 0.0
-mpl.rcParams['agg.path.chunksize'] = 10000
-mplstyle.use('fast')
+import cv2
 
 HALF_INF = 63
 INF = 126
 EPS_DIST = 1/20
 EPS_ANGLE = 2.86
+SCALE = 15.5
 
 class Turtle:
-    def __init__(self, ax=None):
-        self.x = 0
-        self.y = 0
+    def __init__(self, canvas_size=(2000, 2000)):
+        self.x = canvas_size[0] // 2
+        self.y = canvas_size[1] // 2
         self.heading = 0
-        if ax is None:
-            self.fig, self.ax = plt.subplots(1, 1, figsize=(20, 20))
-        else:
-            self.ax = ax
-        self.ax.set_xlim(-50, 50)
-        self.ax.set_ylim(-50, 50)
-        self.ax.set_xticklabels([])
-        self.ax.set_yticklabels([])
-        self.ax.set_xticks([])
-        self.ax.set_yticks([])
-        self.ax.spines['top'].set_visible(False)
-        self.ax.spines['bottom'].set_visible(False)
-        self.ax.spines['left'].set_visible(False)
-        self.ax.spines['right'].set_visible(False)
+        self.canvas = np.ones((canvas_size[1], canvas_size[0], 3), dtype=np.uint8) * 255
         self.is_down = True
 
     def forward(self, dist):
+        dist = dist * SCALE
         x0, y0 = self.x, self.y
-        x1 = x0 + dist * np.cos(self.heading)
-        y1 = y0 + dist * np.sin(self.heading)
+        x1 = int(x0 + dist * np.cos(self.heading))
+        y1 = int(y0 - dist * np.sin(self.heading))
         if self.is_down:
-            self.ax.plot([x0, x1], [y0, y1], color='black', linewidth=3)
-        self.x = x1
-        self.y = y1
+            cv2.line(self.canvas, (x0, y0), (x1, y1), (0, 0, 0), 3)
+        self.x, self.y = x1, y1
 
     def left(self, angle):
         self.heading += angle * np.pi / 180
@@ -58,11 +37,9 @@ class Turtle:
         self.is_down = True
 
     def save(self, path):
-        self.fig.canvas.draw()
-        pil_img = Image.frombytes('RGB', self.fig.canvas.get_width_height(), self.fig.canvas.tostring_rgb())
         if path:
-            pil_img.save(path)
-        return pil_img
+            cv2.imwrite(path, self.canvas)
+        return self.canvas
 
     class _TurtleState:
         def __init__(self, turtle):
@@ -135,5 +112,20 @@ if __name__ == "__main__":
 
         right(90)
         forward(50)
-        save("test.png")
-
+        save("test2.png")
+
+    def plot2():
+        for j in range(2):
+            forward(2)
+            left(0.0)
+            for i in range(4):
+                forward(2)
+                left(90)
+            forward(0)
+            left(180.0)
+            forward(2)
+            left(180.0)
+        FINAL_IMAGE = turtle.save("")
+
+    example_plot()
+    # plot2()