46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
from .base import Plugin
|
||
|
|
from ..canvastools import RectangleTool
|
||
|
|
from ...viewer.widgets import SaveButtons, Button
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ['Crop']
|
||
|
|
|
||
|
|
|
||
|
|
class Crop(Plugin):
|
||
|
|
name = 'Crop'
|
||
|
|
|
||
|
|
def __init__(self, maxdist=10, **kwargs):
|
||
|
|
super(Crop, self).__init__(**kwargs)
|
||
|
|
self.maxdist = maxdist
|
||
|
|
self.add_widget(SaveButtons())
|
||
|
|
print(self.help())
|
||
|
|
|
||
|
|
def attach(self, image_viewer):
|
||
|
|
super(Crop, self).attach(image_viewer)
|
||
|
|
|
||
|
|
self.rect_tool = RectangleTool(image_viewer,
|
||
|
|
maxdist=self.maxdist,
|
||
|
|
on_enter=self.crop)
|
||
|
|
self.artists.append(self.rect_tool)
|
||
|
|
|
||
|
|
self.reset_button = Button('Reset', self.reset)
|
||
|
|
self.add_widget(self.reset_button)
|
||
|
|
|
||
|
|
def help(self):
|
||
|
|
helpstr = ("Crop tool",
|
||
|
|
"Select rectangular region and press enter to crop.")
|
||
|
|
return '\n'.join(helpstr)
|
||
|
|
|
||
|
|
def crop(self, extents):
|
||
|
|
xmin, xmax, ymin, ymax = extents
|
||
|
|
if xmin == xmax or ymin == ymax:
|
||
|
|
return
|
||
|
|
image = self.image_viewer.image[ymin:ymax+1, xmin:xmax+1]
|
||
|
|
self.image_viewer.image = image
|
||
|
|
self.image_viewer.ax.relim()
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.rect_tool.extents = -10, -10, -10, -10
|
||
|
|
self.image_viewer.image = self.image_viewer.original_image
|
||
|
|
self.image_viewer.ax.relim()
|