/usr/share/pyshared/guiqwt/widgets/base.py is in python-guiqwt 2.3.1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 | # -*- coding: utf-8 -*-
#
# Copyright © 2012 CEA
# Pierre Raybaut
# Licensed under the terms of the CECILL License
# (see guiqwt/__init__.py for details)
"""
base
----
The `base` module provides base objects for internal use of the
`guiqwt.widgets` package.
"""
from PyQt4.QtGui import QWidget, QVBoxLayout, QHBoxLayout, QTabWidget
from PyQt4.QtCore import SIGNAL
from guidata.qthelpers import create_toolbutton
from guidata.configtools import get_icon
# Local imports
from guiqwt.config import _
from guiqwt.histogram import lut_range_threshold
from guiqwt.image import INTERP_LINEAR, TrImageItem
from guiqwt.plot import ImageWidget, ImageDialog
class BaseTransformMixin(object):
"""Base transform widget mixin class (for manipulating TrImageItem objects)
This is to be mixed with a class providing the get_plot method,
like ImageDialog, or BaseTransformWidget (see below)"""
def __init__(self):
self.item = None
self.item_original_state = None
self.item_original_crop = None
self.item_original_transform = None
self.output_array = None
#------Public API----------------------------------------------------------
def add_reset_button(self, layout):
"""Add the standard reset button"""
edit_options_btn = create_toolbutton(self, text=_("Reset"),
icon=get_icon("eraser.png"),
triggered=self.reset,
autoraise=False)
layout.addWidget(edit_options_btn)
layout.addStretch()
def add_apply_button(self, layout):
"""Add the standard apply button"""
apply_btn = create_toolbutton(self, text=_("Apply"),
icon=get_icon("apply.png"),
triggered=self.apply_transformation,
autoraise=False)
layout.addWidget(apply_btn)
layout.addStretch()
def add_buttons_to_layout(self, layout):
"""Add tool buttons to layout"""
self.add_reset_button(layout)
self.add_apply_button(layout)
def set_item(self, item):
"""Set associated item -- must be a TrImageItem object"""
assert isinstance(item, TrImageItem)
self.item = item
self.item_original_state = (item.can_select(),
item.can_move(),
item.can_resize(),
item.can_rotate())
self.item_original_crop = item.get_crop()
self.item_original_transform = item.get_transform()
self.item.set_selectable(True)
self.item.set_movable(True)
self.item.set_resizable(False)
self.item.set_rotatable(True)
item.set_lut_range(lut_range_threshold(item, 256, 2.))
item.set_interpolation(INTERP_LINEAR)
plot = self.get_plot()
plot.add_item(self.item)
# Setting the item as active item (even if the cropping rectangle item
# will also be set as active item just below), for the image tools to
# register this item (contrast, ...):
plot.set_active_item(self.item)
self.item.unselect()
def unset_item(self):
"""Unset the associated item, freeing memory"""
plot = self.get_plot()
plot.del_item(self.item)
self.item = None
def reset(self):
"""Reset crop/transform image settings"""
self.item.set_crop(*self.item_original_crop)
self.item.set_transform(*self.item_original_transform)
self.reset_transformation()
self.apply_transformation()
def reset_transformation(self):
"""Reset transformation"""
raise NotImplementedError
def apply_transformation(self):
"""Apply transformation, e.g. crop or rotate"""
raise NotImplementedError
def compute_transformation(self):
"""Compute transformation, return compute output array"""
raise NotImplementedError
#------Private API---------------------------------------------------------
def restore_original_state(self):
"""Restore item original state"""
select, move, resize, rotate = self.item_original_state
self.item.set_selectable(select)
self.item.set_movable(move)
self.item.set_resizable(resize)
self.item.set_rotatable(rotate)
def accept_changes(self):
"""Computed rotated/cropped array and apply changes to item"""
self.restore_original_state()
self.apply_transformation()
self.output_array = self.compute_transformation()
# Ignoring image position changes
pos_x0, pos_y0, _angle, sx, sy, hf, vf = self.item_original_transform
_pos_x0, _pos_y0, angle, _sx, _sy, hf, vf = self.item.get_transform()
self.item.set_transform(pos_x0, pos_y0, angle, sx, sy, hf, vf)
def reject_changes(self):
"""Restore item original transform settings"""
self.restore_original_state()
self.item.set_crop(*self.item_original_crop)
self.item.set_transform(*self.item_original_transform)
class BaseTransformDialog(ImageDialog):
"""Rotate & Crop Dialog
Rotate and crop a :py:class:`guiqwt.image.TrImageItem` plot item"""
def __init__(self, parent, wintitle=None, options=None, resize_to=None):
if wintitle is None:
wintitle = _("Rotate & Crop")
ImageDialog.__init__(self, wintitle=wintitle, edit=True,
toolbar=False, options=options, parent=parent)
if resize_to is not None:
width, height = resize_to
self.resize(width, height)
self.connect(self, SIGNAL('accepted()'), self.accept_changes)
self.connect(self, SIGNAL('rejected()'), self.reject_changes)
def install_button_layout(self):
"""Reimplemented ImageDialog method"""
self.add_buttons_to_layout(self.button_layout)
super(BaseTransformDialog, self).install_button_layout()
class BaseTransformWidget(QWidget):
"""Base transform widget: see for example rotatecrop.py"""
def __init__(self, parent, options=None):
QWidget.__init__(self, parent=parent)
if options is None:
options = {}
self.imagewidget = ImageWidget(self, **options)
self.imagewidget.register_all_image_tools()
hlayout = QHBoxLayout()
self.add_buttons_to_layout(hlayout)
vlayout = QVBoxLayout()
vlayout.addWidget(self.imagewidget)
vlayout.addLayout(hlayout)
self.setLayout(vlayout)
def get_plot(self):
"""Required for BaseTransformMixin"""
return self.imagewidget.get_plot()
class BaseMultipleTransformWidget(QTabWidget):
"""Base Multiple Transform Widget
Transform several :py:class:`guiqwt.image.TrImageItem` plot items"""
TRANSFORM_WIDGET_CLASS = None
def __init__(self, parent, options=None):
QTabWidget.__init__(self, parent)
self.options = options
self.output_arrays = None
def set_items(self, *items):
"""Set the associated items -- must be a TrImageItem objects"""
for item in items:
self.add_item(item)
def add_item(self, item):
"""Add item to widget"""
widget = self.TRANSFORM_WIDGET_CLASS(self, options=self.options)
widget.set_item(item)
self.addTab(widget, item.title().text())
return widget
def clear_items(self):
"""Clear all items, freeing memory"""
self.items = None
for index in range(self.count()):
self.widget(index).unset_item()
self.clear()
def reset(self):
"""Reset transform image settings"""
for index in range(self.count()):
self.widget(index).reset()
def accept_changes(self):
"""Accept all changes"""
self.output_arrays = []
for index in range(self.count()):
widget = self.widget(index)
widget.accept_changes()
self.output_arrays.append(widget.output_array)
def reject_changes(self):
"""Reject all changes"""
for index in range(self.count()):
self.widget(index).reject_changes()
|