From 8e4c9bddbf801582a15eb57c122b5e74341bc89c Mon Sep 17 00:00:00 2001
From: Walter Bender <walter@sugarlabs.org>
Date: Wed, 22 Sep 2010 21:07:10 +0000
Subject: reset turtle shapes on clean (#2353)

---
(limited to 'TurtleArt')

diff --git a/TurtleArt/tacanvas.py b/TurtleArt/tacanvas.py
index 9975cd2..88a9e5d 100644
--- a/TurtleArt/tacanvas.py
+++ b/TurtleArt/tacanvas.py
@@ -184,6 +184,7 @@ class TurtleGraphics:
             self.tw.active_turtle.set_gray(100)
             self.tw.active_turtle.set_pen_size(5)
             self.tw.active_turtle.set_pen_state(True)
+            self.tw.active_turtle.reset_shapes()
             self.seth(0, share)
             self.setxy(0, 0, share)
             self.tw.active_turtle.hide()
diff --git a/TurtleArt/taturtle.py b/TurtleArt/taturtle.py
index 55cac08..570a728 100644
--- a/TurtleArt/taturtle.py
+++ b/TurtleArt/taturtle.py
@@ -116,6 +116,7 @@ class Turtle:
         self.y = 0
         self.hidden = False
         self.shapes = []
+        self.custom_shapes = False
         self.type = 'turtle'
         self.heading = 0
         self.pen_shade = 50
@@ -168,6 +169,13 @@ class Turtle:
             else:
                 j = int(self.heading + 5) % 360 / (360 / SHAPES)
                 self.shapes[j] = images[0]
+        self.custom_shapes = True
+
+    def reset_shapes(self):
+        """ Reset the shapes to the standard turtle """
+        if self.custom_shapes:
+            self.shapes = generate_turtle_pixbufs(self.colors)
+            self.custom_shapes = False
 
     def set_heading(self, heading):
         """ Set the turtle heading (one shape per 360/SHAPES degrees) """
--
cgit v0.9.1