Delete overlay when deleting plugin.

This commit is contained in:
Tony S Yu
2012-07-21 17:09:29 -05:00
parent fb3f201a2a
commit bd3ee78306
2 changed files with 8 additions and 1 deletions
+4
View File
@@ -18,3 +18,7 @@ class CannyPlugin(Plugin):
def callback(self, *args, **kwargs):
image = canny(*args, **kwargs)
self.image_viewer.overlay = image
def closeEvent(self, event):
self.image_viewer.overlay = None
super(CannyPlugin, self).closeEvent(event)
+4 -1
View File
@@ -117,7 +117,10 @@ class ImageViewer(QtGui.QMainWindow):
@overlay.setter
def overlay(self, image):
self._overlay = image
if self._overlay_plot is None:
if image is None:
self.ax.images.remove(self._overlay_plot)
self._overlay_plot = None
elif self._overlay_plot is None:
self._overlay_plot = self.ax.imshow(image, cmap=self.overlay_cmap)
else:
self._overlay_plot.set_array(image)