Brush button shows stamp

1 files changed, 9 insertions, 2 deletions

diff --git a/toolbox.py b/toolbox.py
index c4903e9..d1a4607 100644
--- a/toolbox.py
+++ b/toolbox.py
@@ -272,7 +272,6 @@ class ToolsToolbarBuilder():
         self._tool_stamp = DrawToolButton('tool-bucket',
             activity.tool_group, _('Stamp'))
         toolbar.insert(self._tool_stamp, -1)
-        toolbar._tool_stamp = self._tool_stamp
         
         is_selected = self._activity.area.is_selected()
         self._tool_stamp.set_sensitive(is_selected)
@@ -301,7 +300,7 @@ class ToolsToolbarBuilder():
             self._TOOL_STAMP_NAME)
         self._tool_marquee_rectangular.connect('clicked', self.set_tool,
             self._TOOL_MARQUEE_RECT_NAME)
-
+    
     def set_tool(self, widget, tool_name):
         """
         Set tool to the Area object. Configures tool's color and size.
@@ -311,6 +310,14 @@ class ToolsToolbarBuilder():
                           necessary in case this method is used in a connect()
             @param tool_name --The name of the selected tool
         """
+        if tool_name == 'stamp':
+            pixbuf = self._activity.area.setupStamp()
+            # Put stamp in ButtonStrokeColor widget:
+            # TODO use signals?
+            self._stroke_color.color_button._pixbuf_stamp = pixbuf
+            self._stroke_color.color_button.set_stamping(True)
+        else:
+            self._stroke_color.color_button.set_stamping(False)
         self.properties['name'] = tool_name
         self._activity.area.set_tool(self.properties)