class PointSelection (QMainWindow) :
def __init__ (self) :
QMainWindow.__init__(self)
# points
self._points = []
self._id_gen = IdSetGenerator()
#X self.version = 2
self._view = PixmapStackView()
#QGraphicsScene
self._scene = QGraphicsScene()
#QGraphicsPixmapItem
self._item = QGraphicsPixmapItem()
self._item.setTransformationMode(Qt.SmoothTransformation)
self._item.setFlag(self._item.ItemIsSelectable,False)
self._scene.addItem(self._item)
#QGraphicsView
self._widget = ScalableGraphicsView(self._scene)
self.setCentralWidget(self._widget)
################### mouse handling ###################
self._widget.setMouseTracking(True)
self._last_mouse_x = 0
self._last_mouse_y = 0
self._last_slice = 0
QObject.connect(self._widget,
SIGNAL("mouse_press"),
self.mouse_pressed)
QObject.connect(self._widget,
SIGNAL("mouse_move"),
self.mouse_moved)
QObject.connect(self,
SIGNAL("mouse_moved"),
self.coordinates)
################### menubar ###################
self.menu = self.menuBar()
self.menu_file = self.menu.addMenu('File')
# Import Points
self._load_points = self.menu_file.addAction('Import Points')
QObject.connect(self._load_points,
SIGNAL("triggered(bool)"),
self.load_points)
# Save points
self._save_points = self.menu_file.addAction('Save Points')
QObject.connect(self._save_points,
SIGNAL("triggered(bool)"),
self.save_points)
################### toolbar ###################
self._toolbar = self.addToolBar("tools")
self._toolgroup = QActionGroup(self)
# QWidgetAction : "Rotation Left"
self._action_left = self._toolbar.addAction("left rotation")
self._action_left.setIcon(QIcon(":/image/rotate_left.png") )
QObject.connect(self._action_left,
SIGNAL("triggered(bool)"),
self.rotate_left)
# QWidgetAction : "Rotation Right"
self._action_right = self._toolbar.addAction("right rotation")
self._action_right.setIcon(QIcon(":/image/rotate_right.png") )
QObject.connect(self._action_right,
SIGNAL("triggered(bool)"),
self.rotate_right)
# QWidgetAction : "Add point"
self._action_add = self._toolbar.addAction("Add point")
self._action_add.setCheckable(True)
self._toolgroup.addAction(self._action_add)
self._action_add.setIcon(QIcon(":/image/add.png") )
# QWidgetAction : "Delete point"
self._action_delete = self._toolbar.addAction("Delete point")
self._action_delete.setCheckable(True)
self._toolgroup.addAction(self._action_delete)
self._action_delete.setIcon(QIcon(":/image/delete.png") )
################### palette ###################
self._palette_select = QComboBox()
self._toolbar.addWidget(self._palette_select)
for palname in palette_names :
self._palette_select.addItem(palname)
QObject.connect(self._palette_select,
SIGNAL("currentIndexChanged(int)"),
self.palette_name_changed)
################### slider ###################
self._bot_toolbar = QToolBar("slider")
self._img_slider = QSlider(Qt.Horizontal)
self._img_slider.setRange(0,self._view.nb_slices() - 1)
QObject.connect(self._img_slider,
SIGNAL("valueChanged(int)"),
self.slice_changed)
self._bot_toolbar.addWidget(self._img_slider)
self.addToolBar(Qt.BottomToolBarArea,self._bot_toolbar)
################### statusbar ###################
self._lab_coord = QLabel("coords:")
self._lab_xcoord = QLabel("% 4d" % 0)
self._lab_ycoord = QLabel("% 4d" % 0)
self._lab_zcoord = QLabel("% 4d" % 0)
self._lab_intens = QLabel("intens: None")
self.statusbar = self.statusBar()
self.statusbar.addPermanentWidget(self._lab_coord)
self.statusbar.addPermanentWidget(self._lab_xcoord)
self.statusbar.addPermanentWidget(self._lab_ycoord)
self.statusbar.addPermanentWidget(self._lab_zcoord)
self.statusbar.addPermanentWidget(self._lab_intens)
##############################################
#
# update GUI
#
##############################################
def update_pix (self) :
"""
"""
pix = self._view.pixmap()
if pix is not None :
self._item.setPixmap(pix)
else:
print 'None pixmap'
#update points
ind = self._view.current_slice()
for point in self._points :
if point is not None:
pid,item, x,y,z,textid = point
visible = abs(z - ind) < 5
item.setVisible(visible)
col = QColor.fromHsv( (pid * 10) % 360,255,255)
if visible :
sca = max(0.,(5 - abs(z - ind) ) / 5.)
if z == ind :
pen = QPen(QColor(255,255,255) )
pen.setWidthF(2.)
item.setPen(pen)
textid.setVisible(True)
else :
pen = QPen(QColor(0,0,0) )
pen.setWidthF(2.)
item.setPen(pen)
textid.setVisible(False)
else :
sca = 1.
tr = QTransform()
tr.scale(sca,sca)
item.setTransform(tr)
item.update()
def coordinates (self,(i,j,k)) :
self._last_mouse_x,self._last_mouse_y, self._last_slice = i,j,k
self.fill_infos()
class ImageStackViewerWidget(QtGui.QWidget):
"""
Widget based on openalea.image.gui.slide_viewer.PixmapStackView
"""
valueChanged = QtCore.Signal(object)
stackChanged = QtCore.Signal(object)
def __init__(self):
QtGui.QWidget.__init__(self)
self._im_view = PixmapStackView()
self._label = ScalableLabel()
self._layout = QtGui.QVBoxLayout(self)
self._layout.addWidget(self._label)
self._label.setMouseTracking(True)
self._last_mouse_x = 0
self._last_mouse_y = 0
self.connect(self._label, QtCore.SIGNAL("mouse_press"), self.mouse_pressed)
self.connect(self._label, QtCore.SIGNAL("mouse_move"), self.mouse_pressed)
self.axis = 2
self.inc = 1 # index increment
self._palette_name = None
##############################################
# Qt Control API
##############################################
def setValue(self, img):
if img is self.value():
return
self._im_view.set_image(img)
try:
self.resolution = img.resolution[:]
except AttributeError:
pass
self._im_view._reconstruct_pixmaps()
self.change_axis(0)
self.slice_changed((0 + self._im_view.nb_slices() - 1) / 2)
self.emit_stack_changed()
def value(self):
return self._im_view.image()
##############################################
# slots
##############################################
def emit_stack_changed(self):
img = self.value()
if img is None:
self.stackChanged.emit(None)
else:
args = dict(shape=img.shape, axis=self.axis,
slice=self._im_view.current_slice(),
resolution=self.resolution, inc=self.inc,
palette_name=self._palette_name)
self.stackChanged.emit(args)
def change_axis(self, ind):
if self.axis == ind:
return
try:
res = list(self.resolution)
del res[self.axis]
tr = self._im_view._transform
print res
if tr % 180:
self._label._resolution = res[1], res[0]
else:
self._label.set_resolution(*res[:2])
except AttributeError:
pass
self._im_view._reconstruct_pixmaps(self.axis)
self.emit_stack_changed()
self.update_pix()
#self.fill_infos()
def mouse_pressed(self, event):
self._last_mouse_x = event.x()
self._last_mouse_y = event.y()
def change_palette(self, palette_name):
if palette_name == self._palette_name:
return
self._palette_name = palette_name
self.emit_stack_changed()
img = self._im_view.image()
if img is None:
return
palette = palette_factory(palette_name, img.max())
self._im_view.set_palette(palette, self.axis)
self.update_pix()
def update_pix(self):
pix = self._im_view.pixmap()
if pix is not None:
self._label.setPixmap(pix)
def get_pixel_value_str(self, img, x, y, z):
px = img[x, y, z]
if isinstance(px, np.ndarray):
return str(px)
else:
return "%3d" % px
def slice_changed(self, ind):
img = self.value()
if img is None:
return
if ind >= img.shape[self.axis]:
ind = img.shape[self.axis] - 1
if ind < 0:
ind = 0
if self._im_view.current_slice() == ind:
return
self._im_view.set_current_slice(ind)
self.emit_stack_changed()
self.update_pix()
def snapshot(self):
"""write the current image
"""
pix = self._im_view.pixmap()
if pix is not None:
pix.save("slice%.4d.png" % self.panel.img_slider.value())
def wheelEvent(self, event):
self.inc = event.delta() / 8 / 15
idx = self._im_view.current_slice()
self.slice_changed(idx + self.inc)
def rotate_left(self):
res = self._label._resolution
self._label._resolution = res[1], res[0]
self._im_view.rotate(-1)
self.update_pix()
def rotate_right(self):
res = self._label._resolution
self._label._resolution = res[1], res[0]
self._im_view.rotate(1)
self.update_pix()
class PointSelection(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
# points
self._points = []
self._id_gen = IdSetGenerator()
#X self.version = 2
self._view = PixmapStackView()
#QGraphicsScene
self._scene = QGraphicsScene()
#QGraphicsPixmapItem
self._item = QGraphicsPixmapItem()
self._item.setTransformationMode(Qt.SmoothTransformation)
self._item.setFlag(self._item.ItemIsSelectable, False)
self._scene.addItem(self._item)
#QGraphicsView
self._widget = ScalableGraphicsView(self._scene)
self.setCentralWidget(self._widget)
################### mouse handling ###################
self._widget.setMouseTracking(True)
self._last_mouse_x = 0
self._last_mouse_y = 0
self._last_slice = 0
QObject.connect(self._widget, SIGNAL("mouse_press"),
self.mouse_pressed)
QObject.connect(self._widget, SIGNAL("mouse_move"), self.mouse_moved)
QObject.connect(self, SIGNAL("mouse_moved"), self.coordinates)
################### menubar ###################
self.menu = self.menuBar()
self.menu_file = self.menu.addMenu('File')
# Import Points
self._load_points = self.menu_file.addAction('Import Points')
QObject.connect(self._load_points, SIGNAL("triggered(bool)"),
self.load_points)
# Save points
self._save_points = self.menu_file.addAction('Save Points')
QObject.connect(self._save_points, SIGNAL("triggered(bool)"),
self.save_points)
################### toolbar ###################
self._toolbar = self.addToolBar("tools")
self._toolgroup = QActionGroup(self)
# QWidgetAction : "Rotation Left"
self._action_left = self._toolbar.addAction("left rotation")
self._action_left.setIcon(QIcon(":/image/rotate_left.png"))
QObject.connect(self._action_left, SIGNAL("triggered(bool)"),
self.rotate_left)
# QWidgetAction : "Rotation Right"
self._action_right = self._toolbar.addAction("right rotation")
self._action_right.setIcon(QIcon(":/image/rotate_right.png"))
QObject.connect(self._action_right, SIGNAL("triggered(bool)"),
self.rotate_right)
# QWidgetAction : "Add point"
self._action_add = self._toolbar.addAction("Add point")
self._action_add.setCheckable(True)
self._toolgroup.addAction(self._action_add)
self._action_add.setIcon(QIcon(":/image/add.png"))
# QWidgetAction : "Delete point"
self._action_delete = self._toolbar.addAction("Delete point")
self._action_delete.setCheckable(True)
self._toolgroup.addAction(self._action_delete)
self._action_delete.setIcon(QIcon(":/image/delete.png"))
################### palette ###################
self._palette_select = QComboBox()
self._toolbar.addWidget(self._palette_select)
for palname in palette_names:
self._palette_select.addItem(palname)
QObject.connect(self._palette_select,
SIGNAL("currentIndexChanged(int)"),
self.palette_name_changed)
################### slider ###################
self._bot_toolbar = QToolBar("slider")
self._img_slider = QSlider(Qt.Horizontal)
self._img_slider.setRange(0, self._view.nb_slices() - 1)
QObject.connect(self._img_slider, SIGNAL("valueChanged(int)"),
self.slice_changed)
self._bot_toolbar.addWidget(self._img_slider)
self.addToolBar(Qt.BottomToolBarArea, self._bot_toolbar)
################### statusbar ###################
self._lab_coord = QLabel("coords:")
self._lab_xcoord = QLabel("% 4d" % 0)
self._lab_ycoord = QLabel("% 4d" % 0)
self._lab_zcoord = QLabel("% 4d" % 0)
self._lab_intens = QLabel("intens: None")
self.statusbar = self.statusBar()
self.statusbar.addPermanentWidget(self._lab_coord)
self.statusbar.addPermanentWidget(self._lab_xcoord)
self.statusbar.addPermanentWidget(self._lab_ycoord)
self.statusbar.addPermanentWidget(self._lab_zcoord)
self.statusbar.addPermanentWidget(self._lab_intens)
##############################################
#
# update GUI
#
##############################################
def update_pix(self):
"""
"""
pix = self._view.pixmap()
if pix is not None:
self._item.setPixmap(pix)
else:
print 'None pixmap'
#update points
ind = self._view.current_slice()
for point in self._points:
if point is not None:
pid, item, x, y, z, textid = point
visible = abs(z - ind) < 5
item.setVisible(visible)
col = QColor.fromHsv((pid * 10) % 360, 255, 255)
if visible:
sca = max(0., (5 - abs(z - ind)) / 5.)
if z == ind:
pen = QPen(QColor(255, 255, 255))
pen.setWidthF(2.)
item.setPen(pen)
textid.setVisible(True)
else:
pen = QPen(QColor(0, 0, 0))
pen.setWidthF(2.)
item.setPen(pen)
textid.setVisible(False)
else:
sca = 1.
tr = QTransform()
tr.scale(sca, sca)
item.setTransform(tr)
item.update()
def coordinates(self, (i, j, k)):
self._last_mouse_x, self._last_mouse_y, self._last_slice = i, j, k
self.fill_infos()
