def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
menubar = self.menuBar()
# create a File pulldown menu, and add it to the menu bar
filemenu = menubar.addMenu("File")
item = QtGui.QAction("Open File", menubar)
item.triggered.connect(self.open_file)
filemenu.addAction(item)
sep = QtGui.QAction(menubar)
sep.setSeparator(True)
filemenu.addAction(sep)
item = QtGui.QAction("Quit", menubar)
item.triggered.connect(self.close)
filemenu.addAction(item)
# Add matplotlib color maps to our built in ones
cmap.add_matplotlib_cmaps()
self.cmaps = cmap.get_names()
self.imaps = imap.get_names()
wd, ht = 500, 500
# Create a Ginga widget
fi = ImageViewCanvas(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(False)
fi.set_callback('drag-drop', self.drop_file)
fi.set_callback('none-move', self.motion)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_setActive(True)
self.fitsimage = fi
# enable various key and mouse controlled actions
bd = fi.get_bindings()
bd.enable_all(True)
self.cp_tag = 'compass'
# pack widget into layout
gingaw = fi.get_widget()
gingaw.resize(wd, ht)
vbox1 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
layout.addWidget(gingaw, stretch=1)
self.cm = cmap.get_cmap('gray')
self.im = imap.get_imap('ramp')
# add color bar
rgbmap = fi.get_rgbmap()
rgbmap.set_hash_size(256)
cbar = ColorBar.ColorBar(self.logger, rgbmap=rgbmap,
link=True)
cbar.resize(-1, 15)
#cbar.show()
self.colorbar = cbar
layout.addWidget(cbar, stretch=0)
settings = fi.get_settings()
settings.getSetting('cuts').add_callback('set',
self.change_range_cb, fi,
self.colorbar)
# color map selection widget
wcmap = QtGui.QComboBox()
for name in self.cmaps:
wcmap.addItem(name)
index = self.cmaps.index('gray')
wcmap.setCurrentIndex(index)
wcmap.activated.connect(self.set_cmap_cb)
self.wcmap = wcmap
# intensity map selection widget
wimap = QtGui.QComboBox()
for name in self.imaps:
wimap.addItem(name)
index = self.imaps.index('ramp')
wimap.setCurrentIndex(index)
wimap.activated.connect(self.set_cmap_cb)
self.wimap = wimap
#wopen = QtGui.QPushButton("Open File")
#wopen.clicked.connect(self.open_file)
# add buttons to layout
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
hbox.addStretch(1)
for w in (wcmap, wimap):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox1.setLayout(layout)
# Create a matplotlib Figure
#self.fig = matplotlib.figure.Figure(figsize=(wd, ht))
self.fig = matplotlib.figure.Figure()
self.canvas = FigureCanvas(self.fig)
vbox2 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
# scrw = QtGui.QScrollArea()
# scrw.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
# scrw.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
# scrw.setWidgetResizable(True)
# layout.addWidget(scrw, stretch=1)
# scrw.setWidget(self.canvas)
layout.addWidget(self.canvas, stretch=1)
# Add matplotlib buttons
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wgetimg = QtGui.QPushButton("Get Data")
wgetimg.clicked.connect(self.get_image)
wgetrgb = QtGui.QPushButton("Get RGB")
wgetrgb.clicked.connect(self.get_rgb_image)
#wquit = QtGui.QPushButton("Quit")
#wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wgetimg, wgetrgb):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox2.setLayout(layout)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
w = QtGui.QWidget()
layout = QtGui.QHBoxLayout()
layout.addWidget(vbox1, stretch=1.0)
layout.addWidget(vbox2, stretch=1.0)
w.setLayout(layout)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
vw = QtGui.QWidget()
vw.setLayout(vbox)
self.setCentralWidget(vw)
class GingaWidget(ImageWidgetBase):
LABEL = "Ginga Viewer"
def __init__(self, session, parent=None):
self.logger = log.get_logger(name='ginga', log_stderr=True)
self.canvas = ImageViewCanvas(self.logger, render='widget')
# prevent widget from grabbing focus
self.canvas.set_follow_focus(False)
self.canvas.enable_overlays(True)
# enable interactive features
bindings = self.canvas.get_bindings()
bindings.enable_all(True)
self.canvas.set_callback('none-move', self.motion_readout)
self.canvas.set_callback('draw-event', self._apply_roi_cb)
self.canvas.set_callback('draw-down', self._clear_roi_cb)
self.canvas.enable_draw(False)
self.canvas.enable_autozoom('off')
self.canvas.set_zoom_algorithm('rate')
self.canvas.set_zoomrate(1.4)
bm = self.canvas.get_bindmap()
bm.add_callback('mode-set', self.mode_set_cb)
self.mode_w = None
self.mode_actns = {}
# Create settings and set defaults
settings = self.canvas.get_settings()
self.settings = settings
settings.getSetting('cuts').add_callback('set', self.cut_levels_cb)
settings.set(autozoom='off', autocuts='override',
autocenter='override')
# make color bar, with color maps shared from ginga canvas
self.colorbar = ColorBar.ColorBar(self.logger)
rgbmap = self.canvas.get_rgbmap()
rgbmap.add_callback('changed', self.rgbmap_cb, self.canvas)
self.colorbar.set_rgbmap(rgbmap)
# make coordinates/value readout
self.readout = Readout.Readout(-1, -1)
self.roi_tag = None
super(GingaWidget, self).__init__(session, parent)
@staticmethod
def _get_default_tools():
return []
def make_client(self):
return GingaClient(self._data, self.canvas, self._container)
def make_central_widget(self):
topw = QWidget()
layout = QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
layout.addWidget(self.canvas.get_widget(), stretch=1)
layout.addWidget(self.colorbar, stretch=0)
layout.addWidget(self.readout.get_widget(), stretch=0)
topw.setLayout(layout)
return topw
def match_colorbar(self, canvas, colorbar):
rgbmap = canvas.get_rgbmap()
loval, hival = canvas.get_cut_levels()
colorbar.set_range(loval, hival, redraw=False)
colorbar.set_rgbmap(rgbmap)
def rgbmap_cb(self, rgbmap, canvas):
self.match_colorbar(canvas, self.colorbar)
def cut_levels_cb(self, setting, tup):
(loval, hival) = tup
self.colorbar.set_range(loval, hival)
def make_toolbar(self):
tb = QToolBar(parent=self)
tb.setIconSize(QSize(25, 25))
tb.layout().setSpacing(1)
tb.setFocusPolicy(Qt.StrongFocus)
agroup = QActionGroup(tb)
agroup.setExclusive(True)
for (mode_text, mode_icon, mode_cb) in self._mouse_modes():
# TODO: add icons similar to the Matplotlib toolbar
action = tb.addAction(mode_icon, mode_text)
action.setCheckable(True)
action.toggled.connect(mode_cb)
agroup.addAction(action)
action = tb.addAction(get_icon('glue_move'), "Pan")
self.mode_actns['pan'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('pan', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'hand_48.png'))
action = tb.addAction(icon, "Free Pan")
self.mode_actns['freepan'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('freepan', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'rotate_48.png'))
action = tb.addAction(icon, "Rotate")
self.mode_actns['rotate'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('rotate', tf))
agroup.addAction(action)
action = tb.addAction(get_icon('glue_contrast'), "Contrast")
self.mode_actns['contrast'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('contrast', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'cuts_48.png'))
action = tb.addAction(icon, "Cuts")
self.mode_actns['cuts'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('cuts', tf))
agroup.addAction(action)
cmap_w = _colormap_mode(self, self.client.set_cmap)
tb.addWidget(cmap_w)
return tb
def _mouse_modes(self):
modes = []
modes.append(("Rectangle", get_icon('glue_square'),
lambda tf: self._set_roi_mode('rectangle', tf)))
modes.append(("Circle", get_icon('glue_circle'),
lambda tf: self._set_roi_mode('circle', tf)))
modes.append(("Polygon", get_icon('glue_lasso'),
lambda tf: self._set_roi_mode('polygon', tf)))
for tool in self._tools:
modes += tool._get_modes(self.canvas)
add_callback(self.client, 'display_data', tool._display_data_hook)
return modes
def _set_roi_mode(self, name, tf):
self.canvas.enable_draw(True)
# XXX need better way of setting draw contexts
self.canvas.draw_context = self
self.canvas.set_drawtype(name, color='cyan', linestyle='dash')
bm = self.canvas.get_bindmap()
bm.set_modifier('draw', modtype='locked')
def _clear_roi_cb(self, canvas, *args):
try:
self.canvas.deleteObjectByTag(self.roi_tag)
except:
pass
def _apply_roi_cb(self, canvas, tag):
if self.canvas.draw_context is not self:
return
self.roi_tag = tag
obj = self.canvas.getObjectByTag(self.roi_tag)
roi = ginga_graphic_to_roi(obj)
# delete outline
self.canvas.deleteObject(obj, redraw=False)
self.apply_roi(roi)
def _tweak_geometry(self):
super(GingaWidget, self)._tweak_geometry()
# rgb mode not supported yet, so hide option
self.ui.monochrome.hide()
self.ui.rgb.hide()
def motion_readout(self, canvas, button, data_x, data_y):
"""This method is called when the user moves the mouse around the Ginga
canvas.
"""
d = self.client.point_details(data_x, data_y)
# Get the value under the data coordinates
try:
# value = fitsimage.get_data(data_x, data_y)
# We report the value across the pixel, even though the coords
# change halfway across the pixel
value = canvas.get_data(int(data_x + 0.5), int(data_y + 0.5))
except Exception:
value = None
x_lbl, y_lbl = d['labels'][0], d['labels'][1]
# x_txt, y_txt = d['world'][0], d['world'][1]
text = "%s %s X=%.2f Y=%.2f Value=%s" % (
x_lbl, y_lbl, data_x, data_y, value)
self.readout.set_text(text)
def mode_cb(self, modname, tf):
"""This method is called when a toggle button in the toolbar is pressed
selecting one of the modes.
"""
bm = self.canvas.get_bindmap()
if not tf:
bm.reset_modifier(self.canvas)
return
bm.set_modifier(modname, modtype='locked')
return True
def mode_set_cb(self, bm, modname, mtype):
"""This method is called when a mode is selected in the viewer widget.
NOTE: it may be called when mode_cb() is not called (for example, when
a keypress initiates a mode); however, the converse is not true: calling
mode_cb() will always result in this method also being called as a result.
This logic is to insure that the toggle buttons are left in a sane state
that reflects the current mode, however it was initiated.
"""
if modname in self.mode_actns:
if self.mode_w and (self.mode_w != self.mode_actns[modname]):
self.mode_w.setChecked(False)
self.mode_w = self.mode_actns[modname]
self.mode_w.setChecked(True)
elif self.mode_w:
# keystroke turned on a mode for which we have no GUI button
# and a GUI button is selected--unselect it
self.mode_w.setChecked(False)
self.mode_w = None
return True
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
menubar = self.menuBar()
# create a File pulldown menu, and add it to the menu bar
filemenu = menubar.addMenu("File")
item = QtGui.QAction("Open File", menubar)
item.triggered.connect(self.open_file)
filemenu.addAction(item)
sep = QtGui.QAction(menubar)
sep.setSeparator(True)
filemenu.addAction(sep)
item = QtGui.QAction("Quit", menubar)
item.triggered.connect(self.close)
filemenu.addAction(item)
# Add matplotlib color maps to our built in ones
cmap.add_matplotlib_cmaps()
self.cmaps = cmap.get_names()
self.imaps = imap.get_names()
wd, ht = 500, 500
# Create a Ginga widget
fi = ImageViewCanvas(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(False)
fi.set_callback('drag-drop', self.drop_file)
fi.set_callback('none-move', self.motion)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_setActive(True)
self.fitsimage = fi
# enable various key and mouse controlled actions
bd = fi.get_bindings()
bd.enable_all(True)
self.cp_tag = 'compass'
# pack widget into layout
gingaw = fi.get_widget()
gingaw.resize(wd, ht)
vbox1 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
layout.addWidget(gingaw, stretch=1)
self.cm = cmap.get_cmap('gray')
self.im = imap.get_imap('ramp')
# add color bar
rgbmap = fi.get_rgbmap()
rgbmap.set_hash_size(256)
cbar = ColorBar.ColorBar(self.logger, rgbmap=rgbmap, link=True)
cbar.resize(-1, 15)
#cbar.show()
self.colorbar = cbar
layout.addWidget(cbar, stretch=0)
settings = fi.get_settings()
settings.getSetting('cuts').add_callback('set', self.change_range_cb,
fi, self.colorbar)
# color map selection widget
wcmap = QtGui.QComboBox()
for name in self.cmaps:
wcmap.addItem(name)
index = self.cmaps.index('gray')
wcmap.setCurrentIndex(index)
wcmap.activated.connect(self.set_cmap_cb)
self.wcmap = wcmap
# intensity map selection widget
wimap = QtGui.QComboBox()
for name in self.imaps:
wimap.addItem(name)
index = self.imaps.index('ramp')
wimap.setCurrentIndex(index)
wimap.activated.connect(self.set_cmap_cb)
self.wimap = wimap
#wopen = QtGui.QPushButton("Open File")
#wopen.clicked.connect(self.open_file)
# add buttons to layout
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
hbox.addStretch(1)
for w in (wcmap, wimap):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox1.setLayout(layout)
# Create a matplotlib Figure
#self.fig = matplotlib.figure.Figure(figsize=(wd, ht))
self.fig = matplotlib.figure.Figure()
self.canvas = FigureCanvas(self.fig)
vbox2 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
# scrw = QtGui.QScrollArea()
# scrw.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
# scrw.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
# scrw.setWidgetResizable(True)
# layout.addWidget(scrw, stretch=1)
# scrw.setWidget(self.canvas)
layout.addWidget(self.canvas, stretch=1)
# Add matplotlib buttons
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wgetimg = QtGui.QPushButton("Get Data")
wgetimg.clicked.connect(self.get_image)
wgetrgb = QtGui.QPushButton("Get RGB")
wgetrgb.clicked.connect(self.get_rgb_image)
#wquit = QtGui.QPushButton("Quit")
#wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wgetimg, wgetrgb):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox2.setLayout(layout)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
w = QtGui.QWidget()
layout = QtGui.QHBoxLayout()
layout.addWidget(vbox1, stretch=1.0)
layout.addWidget(vbox2, stretch=1.0)
w.setLayout(layout)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
vw = QtGui.QWidget()
vw.setLayout(vbox)
self.setCentralWidget(vw)
class GingaWidget(ImageWidgetBase):
LABEL = "Ginga Viewer"
def __init__(self, session, parent=None):
self.logger = log.get_logger(name='ginga', log_stderr=True)
self.canvas = ImageViewCanvas(self.logger, render='widget')
# prevent widget from grabbing focus
self.canvas.set_follow_focus(False)
self.canvas.enable_overlays(True)
# enable interactive features
bindings = self.canvas.get_bindings()
bindings.enable_all(True)
self.canvas.set_callback('none-move', self.motion_readout)
self.canvas.set_callback('draw-event', self._apply_roi_cb)
self.canvas.set_callback('draw-down', self._clear_roi_cb)
self.canvas.enable_draw(False)
self.canvas.enable_autozoom('off')
self.canvas.set_zoom_algorithm('rate')
self.canvas.set_zoomrate(1.4)
bm = self.canvas.get_bindmap()
bm.add_callback('mode-set', self.mode_set_cb)
self.mode_w = None
self.mode_actns = {}
# Create settings and set defaults
settings = self.canvas.get_settings()
self.settings = settings
settings.getSetting('cuts').add_callback('set', self.cut_levels_cb)
settings.set(autozoom='off',
autocuts='override',
autocenter='override')
# make color bar, with color maps shared from ginga canvas
self.colorbar = ColorBar.ColorBar(self.logger)
rgbmap = self.canvas.get_rgbmap()
rgbmap.add_callback('changed', self.rgbmap_cb, self.canvas)
self.colorbar.set_rgbmap(rgbmap)
# make coordinates/value readout
self.readout = Readout.Readout(-1, -1)
self.roi_tag = None
super(GingaWidget, self).__init__(session, parent)
@staticmethod
def _get_default_tools():
return []
def make_client(self):
return GingaClient(self._data, self.canvas, self._container)
def make_central_widget(self):
topw = QWidget()
layout = QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
layout.addWidget(self.canvas.get_widget(), stretch=1)
layout.addWidget(self.colorbar, stretch=0)
layout.addWidget(self.readout.get_widget(), stretch=0)
topw.setLayout(layout)
return topw
def match_colorbar(self, canvas, colorbar):
rgbmap = canvas.get_rgbmap()
loval, hival = canvas.get_cut_levels()
colorbar.set_range(loval, hival, redraw=False)
colorbar.set_rgbmap(rgbmap)
def rgbmap_cb(self, rgbmap, canvas):
self.match_colorbar(canvas, self.colorbar)
def cut_levels_cb(self, setting, tup):
(loval, hival) = tup
self.colorbar.set_range(loval, hival)
def make_toolbar(self):
tb = QToolBar(parent=self)
tb.setIconSize(QSize(25, 25))
tb.layout().setSpacing(1)
tb.setFocusPolicy(Qt.StrongFocus)
agroup = QActionGroup(tb)
agroup.setExclusive(True)
for (mode_text, mode_icon, mode_cb) in self._mouse_modes():
# TODO: add icons similar to the Matplotlib toolbar
action = tb.addAction(mode_icon, mode_text)
action.setCheckable(True)
action.toggled.connect(mode_cb)
agroup.addAction(action)
action = tb.addAction(get_icon('glue_move'), "Pan")
self.mode_actns['pan'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('pan', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'hand_48.png'))
action = tb.addAction(icon, "Free Pan")
self.mode_actns['freepan'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('freepan', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'rotate_48.png'))
action = tb.addAction(icon, "Rotate")
self.mode_actns['rotate'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('rotate', tf))
agroup.addAction(action)
action = tb.addAction(get_icon('glue_contrast'), "Contrast")
self.mode_actns['contrast'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('contrast', tf))
agroup.addAction(action)
icon = QIcon(os.path.join(ginga_icon_dir, 'cuts_48.png'))
action = tb.addAction(icon, "Cuts")
self.mode_actns['cuts'] = action
action.setCheckable(True)
action.toggled.connect(lambda tf: self.mode_cb('cuts', tf))
agroup.addAction(action)
cmap_w = _colormap_mode(self, self.client.set_cmap)
tb.addWidget(cmap_w)
return tb
def _mouse_modes(self):
modes = []
modes.append(("Rectangle", get_icon('glue_square'),
lambda tf: self._set_roi_mode('rectangle', tf)))
modes.append(("Circle", get_icon('glue_circle'),
lambda tf: self._set_roi_mode('circle', tf)))
modes.append(("Polygon", get_icon('glue_lasso'),
lambda tf: self._set_roi_mode('polygon', tf)))
for tool in self._tools:
modes += tool._get_modes(self.canvas)
add_callback(self.client, 'display_data', tool._display_data_hook)
return modes
def _set_roi_mode(self, name, tf):
self.canvas.enable_draw(True)
# XXX need better way of setting draw contexts
self.canvas.draw_context = self
self.canvas.set_drawtype(name, color='cyan', linestyle='dash')
bm = self.canvas.get_bindmap()
bm.set_modifier('draw', modtype='locked')
def _clear_roi_cb(self, canvas, *args):
try:
self.canvas.deleteObjectByTag(self.roi_tag)
except:
pass
def _apply_roi_cb(self, canvas, tag):
if self.canvas.draw_context is not self:
return
self.roi_tag = tag
obj = self.canvas.getObjectByTag(self.roi_tag)
roi = ginga_graphic_to_roi(obj)
# delete outline
self.canvas.deleteObject(obj, redraw=False)
self.apply_roi(roi)
def _tweak_geometry(self):
super(GingaWidget, self)._tweak_geometry()
# rgb mode not supported yet, so hide option
self.ui.monochrome.hide()
self.ui.rgb.hide()
def motion_readout(self, canvas, button, data_x, data_y):
"""This method is called when the user moves the mouse around the Ginga
canvas.
"""
d = self.client.point_details(data_x, data_y)
# Get the value under the data coordinates
try:
# value = fitsimage.get_data(data_x, data_y)
# We report the value across the pixel, even though the coords
# change halfway across the pixel
value = canvas.get_data(int(data_x + 0.5), int(data_y + 0.5))
except Exception:
value = None
x_lbl, y_lbl = d['labels'][0], d['labels'][1]
# x_txt, y_txt = d['world'][0], d['world'][1]
text = "%s %s X=%.2f Y=%.2f Value=%s" % (x_lbl, y_lbl, data_x,
data_y, value)
self.readout.set_text(text)
def mode_cb(self, modname, tf):
"""This method is called when a toggle button in the toolbar is pressed
selecting one of the modes.
"""
bm = self.canvas.get_bindmap()
if not tf:
bm.reset_modifier(self.canvas)
return
bm.set_modifier(modname, modtype='locked')
return True
def mode_set_cb(self, bm, modname, mtype):
"""This method is called when a mode is selected in the viewer widget.
NOTE: it may be called when mode_cb() is not called (for example, when
a keypress initiates a mode); however, the converse is not true: calling
mode_cb() will always result in this method also being called as a result.
This logic is to insure that the toggle buttons are left in a sane state
that reflects the current mode, however it was initiated.
"""
if modname in self.mode_actns:
if self.mode_w and (self.mode_w != self.mode_actns[modname]):
self.mode_w.setChecked(False)
self.mode_w = self.mode_actns[modname]
self.mode_w.setChecked(True)
elif self.mode_w:
# keystroke turned on a mode for which we have no GUI button
# and a GUI button is selected--unselect it
self.mode_w.setChecked(False)
self.mode_w = None
return True
class GingaWidget(ImageWidgetBase):
LABEL = "Ginga Viewer"
_toolbar_cls = BasicToolbar
tools = ['ginga:rectangle', 'ginga:circle', 'ginga:polygon', 'ginga:lasso',
'ginga:xrange', 'ginga:yrange',
'ginga:pan', 'ginga:freepan', 'ginga:rotate',
'ginga:contrast', 'ginga:cuts', 'ginga:dist',
'ginga:colormap', 'ginga:spectrum', 'ginga:slicer']
def __init__(self, session, parent=None):
self.logger = log.get_logger(name='ginga', level=20,
# switch commenting for debugging
null=True, log_stderr=False,
#null=False, log_stderr=True
)
# load binding preferences if available
cfgfile = os.path.join(ginga_home, "bindings.cfg")
bindprefs = SettingGroup(name='bindings', logger=self.logger,
preffile=cfgfile)
bindprefs.load(onError='silent')
bd = ImageViewBindings(self.logger, settings=bindprefs)
# make Ginga viewer
self.viewer = ImageViewCanvas(self.logger, render='widget',
bindings=bd)
self.canvas = self.viewer
# prevent widget from grabbing focus
self.viewer.set_enter_focus(False)
self.viewer.set_desired_size(300, 300)
# enable interactive features
bindings = self.viewer.get_bindings()
bindings.enable_all(True)
self.canvas.add_callback('none-move', self.motion_readout)
self.canvas.register_for_cursor_drawing(self.viewer)
self.canvas.add_callback('draw-event', self._apply_roi_cb)
self.canvas.add_callback('edit-event', self._update_roi_cb)
self.canvas.add_callback('draw-down', self._clear_roi_cb)
self.canvas.enable_draw(False)
self.canvas.enable_edit(False)
self.viewer.enable_autozoom('off')
self.viewer.set_zoom_algorithm('rate')
self.viewer.set_zoomrate(1.4)
self.viewer.set_fg(*colors.lookup_color("#D0F0E0"))
bm = self.viewer.get_bindmap()
bm.add_callback('mode-set', self.mode_set_cb)
self.mode_w = None
self.mode_actns = {}
# Create settings and set defaults
settings = self.viewer.get_settings()
self.settings = settings
settings.getSetting('cuts').add_callback('set', self.cut_levels_cb)
settings.set(autozoom='off', autocuts='override',
autocenter='override')
# make color bar, with color maps shared from ginga canvas
rgbmap = self.viewer.get_rgbmap()
self.colorbar = ColorBar.ColorBar(self.logger)
rgbmap.add_callback('changed', self.rgbmap_cb, self.viewer)
self.colorbar.set_rgbmap(rgbmap)
# make coordinates/value readout
self.readout = Readout.Readout(-1, 20)
self.roi_tag = None
self.opn_obj = None
super(GingaWidget, self).__init__(session, parent)
def make_client(self):
return GingaClient(self._data, self.viewer, self._layer_artist_container)
def make_central_widget(self):
topw = QtWidgets.QWidget()
layout = QtWidgets.QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
layout.addWidget(self.viewer.get_widget(), stretch=1)
cbar_w = self.colorbar.get_widget()
if not isinstance(cbar_w, QtWidgets.QWidget):
# ginga wrapped widget
cbar_w = cbar_w.get_widget()
layout.addWidget(cbar_w, stretch=0)
readout_w = self.readout.get_widget()
if not isinstance(readout_w, QtWidgets.QWidget):
# ginga wrapped widget
readout_w = readout_w.get_widget()
layout.addWidget(readout_w, stretch=0)
topw.setLayout(layout)
return topw
def match_colorbar(self, canvas, colorbar):
rgbmap = self.viewer.get_rgbmap()
loval, hival = self.viewer.get_cut_levels()
colorbar.set_range(loval, hival)
colorbar.set_rgbmap(rgbmap)
def rgbmap_cb(self, rgbmap, canvas):
self.match_colorbar(canvas, self.colorbar)
def cut_levels_cb(self, setting, tup):
(loval, hival) = tup
self.colorbar.set_range(loval, hival)
def _set_roi_mode(self, opn_obj, name, mode, **kwargs):
self.opn_obj = opn_obj
en_draw = (mode == 'draw')
self.canvas.enable_draw(en_draw)
self.canvas.set_draw_mode(mode)
# XXX need better way of setting draw contexts
self.canvas.draw_context = self
self.canvas.set_drawtype(name, **kwargs)
## bm = self.viewer.get_bindmap()
## bm.set_mode('draw', mode_type='locked')
def _clear_roi_cb(self, canvas, *args):
if self.opn_obj is not None:
self.opn_obj.opn_init(self, self.roi_tag)
else:
try:
self.canvas.delete_object_by_tag(self.roi_tag)
except:
pass
def _apply_roi_cb(self, canvas, tag):
if self.canvas.draw_context is not self:
return
self.roi_tag = tag
obj = self.canvas.get_object_by_tag(tag)
if self.opn_obj is None:
# delete outline
self.canvas.delete_object(obj)
self.roi_tag = None
return
self.opn_obj.opn_exec(self, tag, obj)
def _update_roi_cb(self, canvas, obj):
if self.canvas.draw_context is not self:
return
if self.opn_obj is None:
return
self.opn_obj.opn_update(self, obj)
def _tweak_geometry(self):
super(GingaWidget, self)._tweak_geometry()
# rgb mode not supported yet, so hide option
self.ui.monochrome.hide()
self.ui.rgb.hide()
def motion_readout(self, canvas, button, data_x, data_y):
"""This method is called when the user moves the mouse around the Ginga
canvas.
"""
d = self.client.point_details(data_x, data_y)
# Get the value under the data coordinates
try:
# value = fitsimage.get_data(data_x, data_y)
# We report the value across the pixel, even though the coords
# change halfway across the pixel
value = self.viewer.get_data(int(data_x + 0.5), int(data_y + 0.5))
except Exception:
value = None
x_lbl, y_lbl = d['labels'][0], d['labels'][1]
# x_txt, y_txt = d['world'][0], d['world'][1]
text = "%s %s X=%.2f Y=%.2f Value=%s" % (
x_lbl, y_lbl, data_x, data_y, value)
self.readout.set_text(text)
def mode_cb(self, modname, tf):
"""This method is called when a toggle button in the toolbar is pressed
selecting one of the modes.
"""
bm = self.viewer.get_bindmap()
if not tf:
bm.reset_mode(self.viewer)
return
bm.set_mode(modname, mode_type='locked')
return True
def mode_set_cb(self, bm, modname, mtype):
"""This method is called when a mode is selected in the viewer widget.
NOTE: it may be called when mode_cb() is not called (for example, when
a keypress initiates a mode); however, the converse is not true: calling
mode_cb() will always result in this method also being called as a result.
This logic is to insure that the toggle buttons are left in a sane state
that reflects the current mode, however it was initiated.
"""
if modname in self.mode_actns:
if self.mode_w and (self.mode_w != self.mode_actns[modname]):
self.mode_w.setChecked(False)
self.mode_w = self.mode_actns[modname]
self.mode_w.setChecked(True)
elif self.mode_w:
# keystroke turned on a mode for which we have no GUI button
# and a GUI button is selected--unselect it
self.mode_w.setChecked(False)
self.mode_w = None
return True
