def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(Overlays, self).__init__(fv, fitsimage)
self.layertag = 'overlays-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
# TODO: there is some problem with basic "red", at least on Linux
#self.hi_color = 'red'
self.hi_color = 'palevioletred'
self.hi_value = None
self.lo_color = 'blue'
self.lo_value = None
self.opacity = 0.5
self.arrsize = None
self.rgbarr = np.zeros((1, 1, 4), dtype=np.uint8)
self.rgbobj = RGBImage.RGBImage(logger=self.logger,
data_np=self.rgbarr)
self.canvas_img = None
def __init__(self, fv, fitsimage):
super(MultiBars, self).__init__(fv, fitsimage)
self.layertag = 'bars-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
self.canvas_img = None
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(WCSAxes, self).__init__(fv, fitsimage)
self.layertag = 'wcsaxes-canvas'
self.colornames = colors.get_colors()
self.linestyles = ['solid', 'dash']
prefs = self.fv.get_preferences()
self.settings = prefs.create_category('plugin_WCSAxes')
self.settings.add_defaults(linecolor='cyan',
alpha=1,
linestyle='solid',
linewidth=1,
n_ra_lines=10,
n_dec_lines=10,
show_label=True,
fontsize=8,
label_offset=4)
self.settings.load(onError='silent')
linecolor = self.settings.get('linecolor', 'cyan')
alpha = self.settings.get('alpha', 1)
linestyle = self.settings.get('linestyle', 'solid')
linewidth = self.settings.get('linewidth', 1)
num_ra = self.settings.get('n_ra_lines', 10)
num_dec = self.settings.get('n_dec_lines', 10)
show_label = self.settings.get('show_label', True)
fontsize = self.settings.get('fontsize', 8)
txt_off = self.settings.get('label_offset', 4)
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.axes = self.dc.WCSAxes(linewidth=linewidth,
linestyle=linestyle,
color=linecolor,
alpha=alpha,
fontsize=fontsize)
self.axes.num_ra = num_ra
self.axes.num_dec = num_dec
self.axes.show_label = show_label
self.axes.txt_off = txt_off
self.canvas.add(self.axes)
self.gui_up = False
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(WCSAxes, self).__init__(fv, fitsimage)
self.layertag = 'wcsaxes-canvas'
self.colornames = colors.get_colors()
self.linestyles = ['solid', 'dash']
prefs = self.fv.get_preferences()
self.settings = prefs.create_category('plugin_WCSAxes')
self.settings.add_defaults(linecolor='cyan', alpha=1,
linestyle='solid', linewidth=1,
n_ra_lines=10, n_dec_lines=10,
show_label=True, fontsize=8, label_offset=4)
self.settings.load(onError='silent')
linecolor = self.settings.get('linecolor', 'cyan')
alpha = self.settings.get('alpha', 1)
linestyle = self.settings.get('linestyle', 'solid')
linewidth = self.settings.get('linewidth', 1)
num_ra = self.settings.get('n_ra_lines', 10)
num_dec = self.settings.get('n_dec_lines', 10)
show_label = self.settings.get('show_label', True)
fontsize = self.settings.get('fontsize', 8)
txt_off = self.settings.get('label_offset', 4)
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.axes = self.dc.WCSAxes(
linewidth=linewidth, linestyle=linestyle, color=linecolor,
alpha=alpha, fontsize=fontsize)
self.axes.num_ra = num_ra
self.axes.num_dec = num_dec
self.axes.show_label = show_label
self.axes.txt_off = txt_off
self.canvas.add(self.axes)
self.gui_up = False
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(Overlays, self).__init__(fv, fitsimage)
self.layertag = 'overlays-canvas'
self.dc = fv.getDrawClasses()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.setSurface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
self.hi_color = 'red'
self.hi_value = None
self.lo_color = 'blue'
self.lo_value = None
self.opacity = 0.5
self.arrsize = None
self.rgbarr = numpy.zeros((1, 1, 4), dtype=numpy.uint8)
self.rgbobj = RGBImage.RGBImage(self.rgbarr, logger=self.logger)
self.canvas_img = None
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(Overlays, self).__init__(fv, fitsimage)
self.layertag = 'overlays-canvas'
self.dc = fv.getDrawClasses()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.setSurface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
self.hi_color = 'red'
self.hi_value = None
self.lo_color = 'blue'
self.lo_value = None
self.opacity = 0.5
self.arrsize = None
self.rgbarr = numpy.zeros((1, 1, 4), dtype=numpy.uint8)
self.rgbobj = RGBImage.RGBImage(self.rgbarr, logger=self.logger)
self.canvas_img = None
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(Overlays, self).__init__(fv, fitsimage)
self.layertag = 'overlays-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
# TODO: there is some problem with basic "red", at least on Linux
#self.hi_color = 'red'
self.hi_color = 'palevioletred'
self.hi_value = None
self.lo_color = 'blue'
self.lo_value = None
self.opacity = 0.5
self.arrsize = None
self.rgbarr = np.zeros((1, 1, 4), dtype=np.uint8)
self.rgbobj = RGBImage.RGBImage(self.rgbarr, logger=self.logger)
self.canvas_img = None
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
fi = CanvasView(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
#fi.show_pan_mark(True)
#fi.enable_draw(False)
fi.add_callback('drag-drop', self.drop_file_cb)
fi.add_callback('cursor-changed', self.cursor_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
#fi.add(canvas)
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.register_for_cursor_drawing(fi)
canvas.add_callback('draw-event', self.draw_cb)
canvas.set_draw_mode('draw')
canvas.ui_set_active(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
#fi.show_color_bar(True)
fi.show_focus_indicator(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
w = fi.get_widget()
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill, QtGui.QLabel('Alpha:'),
walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
btn1 = QtGui.QRadioButton("Draw")
btn1.setChecked(mode == 'draw')
btn1.toggled.connect(lambda val: self.set_mode_cb('draw', val))
btn1.setToolTip("Choose this to draw on the canvas")
hbox.addWidget(btn1)
btn2 = QtGui.QRadioButton("Edit")
btn2.setChecked(mode == 'edit')
btn2.toggled.connect(lambda val: self.set_mode_cb('edit', val))
btn2.setToolTip("Choose this to edit things on the canvas")
hbox.addWidget(btn2)
btn3 = QtGui.QRadioButton("Pick")
btn3.setChecked(mode == 'pick')
btn3.toggled.connect(lambda val: self.set_mode_cb('pick', val))
btn3.setToolTip("Choose this to pick things on the canvas")
hbox.addWidget(btn3)
hbox.addWidget(QtGui.QLabel(''), stretch=1)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
dest="toolkit",
metavar="NAME",
default='qt',
help="Choose GUI toolkit (gtk|qt)")
log.addlogopts(argprs)
options = argprs.parse_args()
args = options.args
if options.show_timezones:
for timezone in pytz.all_timezones:
print(timezone)
sys.exit(0)
if options.show_colors:
names = colors.get_colors()
for color in names:
print(color)
sys.exit(0)
if options.display:
os.environ['DISPLAY'] = options.display
# Are we debugging this?
if options.debug:
import pdb
pdb.run('main(options, args)')
# Are we profiling this?
elif options.profile:
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
fi = ImageViewCanvas(logger, render="widget")
fi.enable_autocuts("on")
fi.set_autocut_params("zscale")
fi.enable_autozoom("on")
fi.enable_draw(True)
fi.set_drawtype("ruler")
fi.set_drawcolor("blue")
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
bd = fi.get_bindings()
bd.enable_pan(True)
bd.enable_zoom(True)
bd.enable_cuts(True)
bd.enable_flip(True)
bd.enable_rotate(True)
bd.enable_cmap(True)
w = fi.get_widget()
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0, alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index("ruler")
wdrawtype.setCurrentIndex(index)
self.connect(wdrawtype, QtCore.SIGNAL("activated(QString)"), self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index("blue")
wdrawcolor.setCurrentIndex(index)
self.connect(wdrawcolor, QtCore.SIGNAL("activated(QString)"), self.set_drawparams)
self.wdrawcolor = wdrawcolor
wclear = QtGui.QPushButton("Clear Canvas")
self.connect(wclear, QtCore.SIGNAL("clicked()"), self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
self.connect(wopen, QtCore.SIGNAL("clicked()"), self.open_file)
wquit = QtGui.QPushButton("Quit")
self.connect(wquit, QtCore.SIGNAL("clicked()"), self, QtCore.SLOT("close()"))
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
root = Gtk.Window(title="Gtk3 CanvasView Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
self.select = GtkHelp.FileSelection(root)
vbox = Gtk.VBox(spacing=2)
fi = CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
fi.set_callback('drag-drop', self.drop_file_cb)
fi.set_callback('cursor-changed', self.cursor_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.register_for_cursor_drawing(fi)
private_canvas.add(canvas)
canvas.ui_set_active(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
fi.show_color_bar(True)
fi.show_focus_indicator(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, True, True, 1)
self.readout = Gtk.Label("")
vbox.pack_start(self.readout, False, False, 0)
hbox = Gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('rectangle')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('lightblue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = GtkHelp.CheckButton("Fill")
wfill.sconnect('toggled', self.set_drawparams)
self.wfill = wfill
walpha = GtkHelp.SpinButton()
adj = walpha.get_adjustment()
adj.configure(0.0, 0.0, 1.0, 0.1, 0.1, 0)
walpha.set_value(1.0)
walpha.set_digits(1)
walpha.sconnect('value-changed', self.set_drawparams)
self.walpha = walpha
wclear = Gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = Gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = Gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, walpha, Gtk.Label("Alpha:"), wfill,
wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, False, False, 0)
vbox.pack_start(hbox, False, False, 0)
root.add(vbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(True)
fi.set_drawtype('ruler')
fi.set_drawcolor('blue')
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
fi.set_figure(fig)
fi.get_bindings().enable_all(True)
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('ruler')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('blue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
fi = CanvasView(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
#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
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
#fi.add(canvas)
private_canvas = fi.get_canvas()
private_canvas.register_for_cursor_drawing(fi)
private_canvas.add(canvas)
canvas.ui_setActive(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
#fi.show_color_bar(True)
fi.show_focus_indicator(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
w = fi.get_widget()
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
QtGui.QLabel('Alpha:'), walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger, window):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = Viewers.CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
canvas.register_for_cursor_drawing(fi)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# the corner
private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed', lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
## wopen = Widgets.Button("Open File")
## wopen.add_callback('activated', lambda w: self.open_file())
## wquit = Widgets.Button("Quit")
## wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wdrawtype, wdrawcolor, wfill,
Widgets.Label('Alpha:'), walpha, wclear):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated', lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated', lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
btn3 = Widgets.CheckBox("I'm using a trackpad")
btn3.add_callback('activated', lambda w, tf: self.use_trackpad_cb(tf))
hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
# need to put this in an hbox with an expanding label or the
# browser wants to resize the canvas
hbox = Widgets.HBox()
hbox.add_widget(vbox, stretch=0)
hbox.add_widget(Widgets.Label(''), stretch=1)
self.top.set_widget(hbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
fi = CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
#fi.enable_draw(False)
fi.set_callback('drag-drop', self.drop_file_cb)
fi.set_callback('none-move', self.cursor_cb)
fi.ui_set_active(True)
self.fitsimage = fi
# quick hack to get 'u' to invoke hidden camera mode
bm = fi.get_bindmap()
bm.mode_map['u'] = bm.mode_map['mode_camera']
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
#fi.add(canvas)
private_canvas = fi.get_canvas()
private_canvas.register_for_cursor_drawing(fi)
private_canvas.add(canvas)
canvas.ui_set_active(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
#fi.show_color_bar(True)
#fi.show_focus_indicator(True)
# add little mode indicator that shows keyboard modal states
#fi.show_mode_indicator(True, corner='ur')
w = fi.get_widget()
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill, QtGui.QLabel('Alpha:'),
walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
#fi = ImageViewCanvas(logger, render='widget')
fi = CanvasView(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
#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
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
#fi.add(canvas)
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# the corner
private_canvas.add(self.dc.ModeIndicator(corner='ur'))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
w = fi.get_widget()
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
QtGui.QLabel('Alpha:'), walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger=logger)
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_cb)
fi.set_callback('cursor-changed', self.cursor_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
fi.set_figure(fig)
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
fi.get_canvas().add(canvas)
canvas.ui_set_active(True)
canvas.register_for_cursor_drawing(fi)
canvas.add_callback('draw-event', self.draw_cb)
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
QtGui.QLabel('Alpha:'), walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
btn1 = QtGui.QRadioButton("Draw")
btn1.setChecked(mode == 'draw')
btn1.toggled.connect(lambda val: self.set_mode_cb('draw', val))
btn1.setToolTip("Choose this to draw on the canvas")
hbox.addWidget(btn1)
btn2 = QtGui.QRadioButton("Edit")
btn2.setChecked(mode == 'edit')
btn2.toggled.connect(lambda val: self.set_mode_cb('edit', val))
btn2.setToolTip("Choose this to edit things on the canvas")
hbox.addWidget(btn2)
btn3 = QtGui.QRadioButton("Pick")
btn3.setChecked(mode == 'pick')
btn3.toggled.connect(lambda val: self.set_mode_cb('pick', val))
btn3.setToolTip("Choose this to pick things on the canvas")
hbox.addWidget(btn3)
hbox.addWidget(QtGui.QLabel(''), stretch=1)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.select = FileSelection.FileSelection()
root = gtk.Window(gtk.WINDOW_TOPLEVEL)
root.set_title("ImageViewCanvas Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
vbox = gtk.VBox(spacing=2)
fi = ImageViewCanvas(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(True)
fi.set_drawtype('ruler')
fi.set_drawcolor('blue')
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
bd = fi.get_bindings()
bd.enable_pan(True)
bd.enable_zoom(True)
bd.enable_cuts(True)
bd.enable_flip(True)
bd.enable_rotate(True)
bd.enable_cmap(True)
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, fill=True, expand=True)
self.readout = gtk.Label("")
vbox.pack_start(self.readout, fill=True, expand=False)
hbox = gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
self.drawtypes = fi.get_drawtypes()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('ruler')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('blue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wclear = gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, fill=False, expand=False)
vbox.pack_start(hbox, fill=False, expand=False)
root.add(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
from ginga.gw import Widgets, Viewers
self.app = Widgets.Application()
self.app.add_callback('shutdown', self.quit)
self.top = self.app.window("Ginga example2")
self.top.add_callback('closed', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
hbox = Widgets.HBox()
hbox.set_border_width(2)
hbox.set_spacing(4)
v1 = Viewers.CanvasView(logger)
v1.enable_autocuts('on')
v1.set_autocut_params('zscale')
v1.enable_autozoom('on')
v1.set_zoom_algorithm('rate')
v1.set_zoomrate(1.4)
v1.show_pan_mark(True)
v1.set_callback('drag-drop', self.drop_file)
v1.set_callback('none-move', self.motion)
v1.set_bg(0.2, 0.2, 0.2)
v1.ui_setActive(True)
self.viewer1 = v1
self._mi1 = ModeIndicator(v1)
bd = v1.get_bindings()
bd.enable_all(True)
# shared canvas between the two viewers
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
self.canvas = canvas
# Tell viewer1 to use this canvas
v1.set_canvas(canvas)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
hbox.add_widget(v1, stretch=1)
# Add a second viewer viewing the same canvas
v2 = Viewers.CanvasView(logger)
v2.enable_autocuts('on')
v2.set_autocut_params('zscale')
v2.enable_autozoom('on')
v2.set_zoom_algorithm('rate')
v2.set_zoomrate(1.4)
v2.show_pan_mark(True)
v2.set_callback('drag-drop', self.drop_file)
v2.set_callback('none-move', self.motion)
v2.set_bg(0.2, 0.2, 0.2)
v2.ui_setActive(True)
self.viewer2 = v2
self._mi2 = ModeIndicator(v2)
# Tell viewer2 to use this same canvas
v2.set_canvas(canvas)
bd = v2.get_bindings()
bd.enable_all(True)
hbox.add_widget(v2, stretch=1)
vbox.add_widget(hbox, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed', lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
wopen = Widgets.Button("Open File")
wopen.add_callback('activated', lambda w: self.open_file())
wquit = Widgets.Button("Quit")
wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
Widgets.Label('Alpha:'), walpha, wclear, wquit):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
root = gtk.Window(gtk.WINDOW_TOPLEVEL)
root.set_title("ImageViewCanvas Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
self.select = GtkHelp.FileSelection(root)
vbox = gtk.VBox(spacing=2)
fi = ImageViewCanvas(logger)
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
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.add(canvas)
canvas.ui_setActive(True)
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, fill=True, expand=True)
self.readout = gtk.Label("")
vbox.pack_start(self.readout, fill=True, expand=False)
hbox = gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
self.drawtypes = fi.get_drawtypes()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('rectangle')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('lightblue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = GtkHelp.CheckButton("Fill")
wfill.sconnect('toggled', self.set_drawparams)
self.wfill = wfill
walpha = GtkHelp.SpinButton()
adj = walpha.get_adjustment()
adj.configure(0.0, 0.0, 1.0, 0.1, 0.1, 0)
walpha.set_value(1.0)
walpha.set_digits(1)
walpha.sconnect('value-changed', self.set_drawparams)
self.walpha = walpha
wclear = gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, walpha, gtk.Label("Alpha:"), wfill,
wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, fill=False, expand=False)
vbox.pack_start(hbox, fill=False, expand=False)
root.add(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
root = gtk.Window(gtk.WINDOW_TOPLEVEL)
root.set_title("Gtk2 CanvasView Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
self.select = GtkHelp.FileSelection(root)
vbox = gtk.VBox(spacing=2)
fi = CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
fi.set_callback('drag-drop', self.drop_file_cb)
fi.set_callback('cursor-changed', self.cursor_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.register_for_cursor_drawing(fi)
private_canvas.add(canvas)
canvas.ui_set_active(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
#fi.show_color_bar(True)
fi.show_focus_indicator(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, fill=True, expand=True)
self.readout = gtk.Label("")
vbox.pack_start(self.readout, fill=True, expand=False)
hbox = gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('rectangle')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('lightblue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = GtkHelp.CheckButton("Fill")
wfill.sconnect('toggled', self.set_drawparams)
self.wfill = wfill
walpha = GtkHelp.SpinButton()
adj = walpha.get_adjustment()
adj.configure(0.0, 0.0, 1.0, 0.1, 0.1, 0)
walpha.set_value(1.0)
walpha.set_digits(1)
walpha.sconnect('value-changed', self.set_drawparams)
self.walpha = walpha
wclear = gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, walpha, gtk.Label("Alpha:"),
wfill, wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, fill=False, expand=False)
vbox.pack_start(hbox, fill=False, expand=False)
root.add(vbox)
# Please see the file LICENSE.txt for details.
#
import math
import numpy
from collections import namedtuple
from ginga.misc import Callback, Bunch
from ginga import trcalc, colors
from ginga.util.six.moves import map
from . import coordmap
__all__ = ['CanvasObjectBase', 'get_canvas_type', 'get_canvas_types',
'register_canvas_type', 'register_canvas_types']
colors_plus_none = [ None ] + colors.get_colors()
Point = namedtuple('Point', ['x', 'y'])
class EditPoint(Point):
edit_color = 'yellow'
class MovePoint(EditPoint):
edit_color = 'orangered'
class ScalePoint(EditPoint):
edit_color = 'green'
class RotatePoint(EditPoint):
edit_color = 'skyblue'
class CanvasObjectError(Exception):
pass
def __init__(self, logger, window, viewer_class=None):
if viewer_class is None:
viewer_class = EnhancedCanvasView
self.logger = logger
self.url = window.url
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = viewer_class(logger)
fi.url = self.url
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True,
scroll_zoom_acceleration=0.07)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
#canvas.enable_draw(True)
#canvas.enable_edit(True)
#canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
#canvas.register_for_cursor_drawing(fi)
fi.set_canvas(canvas)
## self.drawtypes = canvas.get_drawtypes()
## self.drawtypes.sort()
# add a color bar
private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# the corner
private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.add_widget(Widgets.Label('Zoom sensitivity: '))
slider = Widgets.Slider(orientation='horizontal', dtype=float)
slider.add_callback('value-changed',
lambda w, val: self.adjust_scrolling_accel_cb(val))
slider.set_limits(0.0, 12.0, 0.005)
slider.set_value(8.0)
hbox.add_widget(slider, stretch=1)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
# need to put this in an hbox with an expanding label or the
# browser wants to resize the canvas, distorting it
hbox = Widgets.HBox()
hbox.add_widget(vbox, stretch=0)
hbox.add_widget(Widgets.Label(''), stretch=1)
self.top.set_widget(hbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger)
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
fi.set_figure(fig)
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
DrawingCanvas = fi.getDrawClass('drawingcanvas')
canvas = DrawingCanvas()
canvas.enable_draw(True)
#canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.add(canvas)
canvas.ui_setActive(True)
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill, QtGui.QLabel('Alpha:'),
walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.select = FileSelection.FileSelection()
root = gtk.Window(gtk.WINDOW_TOPLEVEL)
root.set_title("ImageViewCanvas Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
vbox = gtk.VBox(spacing=2)
fi = ImageViewCanvas(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(True)
fi.set_drawtype('ruler')
fi.set_drawcolor('blue')
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
bd = fi.get_bindings()
bd.enable_pan(True)
bd.enable_zoom(True)
bd.enable_cuts(True)
bd.enable_flip(True)
bd.enable_rotate(True)
bd.enable_cmap(True)
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, fill=True, expand=True)
self.readout = gtk.Label("")
vbox.pack_start(self.readout, fill=True, expand=False)
hbox = gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
self.drawtypes = fi.get_drawtypes()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('ruler')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('blue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wclear = gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, fill=False, expand=False)
vbox.pack_start(hbox, fill=False, expand=False)
root.add(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
from ginga.gw import Widgets, Viewers
self.app = Widgets.Application(logger=logger)
#self.app.add_callback('shutdown', self.quit)
self.top = self.app.make_window("Ginga example2")
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
hbox = Widgets.HBox()
hbox.set_border_width(2)
hbox.set_spacing(4)
v1 = Viewers.CanvasView(logger)
v1.enable_autocuts('on')
v1.set_autocut_params('zscale')
v1.enable_autozoom('on')
v1.set_zoom_algorithm('rate')
v1.set_zoomrate(1.4)
v1.show_pan_mark(True)
v1.set_callback('drag-drop', self.drop_file)
v1.set_callback('none-move', self.motion)
v1.set_bg(0.2, 0.2, 0.2)
v1.ui_setActive(True)
v1.set_name('tweedledee')
self.viewer1 = v1
self._mi1 = ModeIndicator(v1)
bd = v1.get_bindings()
bd.enable_all(True)
# shared canvas between the two viewers
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
self.canvas = canvas
# Tell viewer1 to use this canvas
v1.set_canvas(canvas)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
v1.set_desired_size(300, 300)
iw = Viewers.GingaViewerWidget(viewer=v1)
hbox.add_widget(iw, stretch=1)
# Add a second viewer viewing the same canvas
v2 = Viewers.CanvasView(logger)
v2.enable_autocuts('on')
v2.set_autocut_params('zscale')
v2.enable_autozoom('on')
v2.set_zoom_algorithm('rate')
v2.set_zoomrate(1.4)
v2.show_pan_mark(True)
v2.set_callback('drag-drop', self.drop_file)
v2.set_callback('none-move', self.motion)
v2.set_bg(0.2, 0.2, 0.2)
v2.ui_setActive(True)
v1.set_name('tweedledum')
self.viewer2 = v2
self._mi2 = ModeIndicator(v2)
# Tell viewer2 to use this same canvas
v2.set_canvas(canvas)
bd = v2.get_bindings()
bd.enable_all(True)
v2.set_desired_size(300, 300)
iw = Viewers.GingaViewerWidget(viewer=v2)
hbox.add_widget(iw, stretch=1)
vbox.add_widget(hbox, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed',
lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
wopen = Widgets.Button("Open File")
wopen.add_callback('activated', lambda w: self.open_file())
wquit = Widgets.Button("Quit")
wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wopen, wdrawtype, wdrawcolor, wfill, Widgets.Label('Alpha:'),
walpha, wclear, wquit):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
from ginga.gw import Widgets, Viewers, GwHelp
self.app = Widgets.Application(logger=logger)
self.app.add_callback('shutdown', self.quit)
self.top = self.app.make_window("Ginga example2")
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = Viewers.CanvasView(logger=logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# add a color bar
fi.private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# lower left hand corner
fi.private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue', coord='data')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.get_canvas().add(canvas)
canvas.ui_setActive(True)
canvas.register_for_cursor_drawing(fi)
canvas.add_callback('draw-event', self.draw_cb)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
fi.set_desired_size(512, 512)
iw = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(iw, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed', lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
wopen = Widgets.Button("Open File")
wopen.add_callback('activated', lambda w: self.open_file())
wquit = Widgets.Button("Quit")
wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
Widgets.Label('Alpha:'), walpha, wclear, wquit):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated', lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated', lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
btn3 = Widgets.RadioButton("Pick", group=btn1)
btn3.set_state(mode == 'pick')
btn3.add_callback('activated', lambda w, val: self.set_mode_cb('pick', val))
btn3.set_tooltip("Choose this to pick things on the canvas")
hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
self.fs = None
if hasattr(GwHelp, 'FileSelection'):
self.fs = GwHelp.FileSelection(self.top.get_widget())
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger)
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
fi.set_figure(fig)
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
DrawingCanvas = fi.getDrawClass('drawingcanvas')
canvas = DrawingCanvas()
canvas.enable_draw(True)
#canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.add(canvas)
canvas.ui_setActive(True)
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout, stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('rectangle')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = QtGui.QCheckBox("Fill")
wfill.stateChanged.connect(self.set_drawparams)
self.wfill = wfill
walpha = QtGui.QDoubleSpinBox()
walpha.setRange(0.0, 1.0)
walpha.setSingleStep(0.1)
walpha.setValue(1.0)
walpha.valueChanged.connect(self.set_drawparams)
self.walpha = walpha
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
QtGui.QLabel('Alpha:'), walpha, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.set_medfilt(0)
self.set_c2c(0)
# create the ginga viewer and configure it
fi = CanvasView(self.logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_callback('drag-drop', self.drop_file)
fi.add_callback('cursor-changed', self.cursor_cb)
fi.add_callback('cursor-down', self.click_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
# enable some user interaction
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
# canvas.enable_draw(True)
# canvas.enable_edit(True)
# canvas.set_drawtype('circle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
#fi.add(canvas)
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
# canvas.register_for_cursor_drawing(fi)
# canvas.add_callback('draw-event', self.draw_cb)
# canvas.set_draw_mode('draw')
canvas.ui_set_active(True)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
w = fi.get_widget()
w.resize(1000, 1000)
# ---------------------------------------------------------------------
# add scrollbar interface around this viewer
si = ScrolledView(fi)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(si, stretch=1)
# ---------------------------------------------------------------------
# add live cursor readout
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
# ---------------------------------------------------------------------
# add row of interface widgets
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wzoomfit = QtGui.QPushButton("Z0")
wzoomfit.clicked.connect(self.zoom_fit)
wzoomin = QtGui.QPushButton("Z+")
wzoomin.clicked.connect(self.zoom_in)
wzoomout = QtGui.QPushButton("Z-")
wzoomout.clicked.connect(self.zoom_out)
wc2c = QtGui.QCheckBox("Click to Center")
wc2c.stateChanged.connect(self.set_c2c)
self.wc2c = wc2c
wsolve = QtGui.QPushButton("Solve")
wsolve.clicked.connect(self.solve_astrometry)
woverlay = QtGui.QPushButton("Overlay Catalog")
woverlay.clicked.connect(self.overlay_catalog)
wclear = QtGui.QPushButton("Clear Overlays")
wclear.clicked.connect(self.clear_overlays)
wmedfilt = QtGui.QCheckBox("Median Filter")
wmedfilt.stateChanged.connect(self.set_medfilt)
self.wmedfilt = wmedfilt
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
hbox.addStretch(1)
hbox.addWidget(wzoomfit, stretch=0)
hbox.addWidget(wzoomin, stretch=0)
hbox.addWidget(wzoomout, stretch=0)
hbox.addWidget(wc2c, stretch=0)
# hbox.addWidget(wsolve, stretch=0)
hbox.addWidget(woverlay, stretch=0)
hbox.addWidget(wclear, stretch=0)
hbox.addWidget(wmedfilt, stretch=0)
# hbox.addWidget(wopen, stretch=0)
hbox.addWidget(wquit, stretch=0)
# ---------------------------------------------------------------------
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
def __init__(self, logger, window):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_margins(2, 2, 2, 2)
vbox.set_spacing(1)
fi = Viewers.CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
fi.set_callback('drag-drop', self.drop_file_cb)
fi.set_callback('cursor-changed', self.cursor_cb)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
bd = fi.get_bindings()
bd.enable_all(True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True,
scroll_zoom_acceleration=0.07)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.set_surface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_set_active(True)
canvas.register_for_cursor_drawing(fi)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
fi.show_color_bar(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_margins(2, 2, 2, 2)
hbox.set_spacing(4)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed', lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
## wopen = Widgets.Button("Open File")
## wopen.add_callback('activated', lambda w: self.open_file())
## wquit = Widgets.Button("Quit")
## wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wdrawtype, wdrawcolor, wfill,
Widgets.Label('Alpha:'), walpha, wclear):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
hbox.set_spacing(4)
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated', lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated', lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
hbox.add_widget(Widgets.Label('Zoom sensitivity: '))
slider = Widgets.Slider(orientation='horizontal', dtype=float)
slider.add_callback('value-changed',
lambda w, val: self.adjust_scrolling_accel_cb(val))
slider.set_limits(0.0, 12.0, 0.005)
slider.set_value(8.0)
hbox.add_widget(slider, stretch=1)
# hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
# need to put this in an hbox with an expanding label or the
# browser wants to resize the canvas
hbox = Widgets.HBox()
hbox.add_widget(vbox, stretch=0)
hbox.add_widget(Widgets.Label(''), stretch=1)
self.top.set_widget(hbox)
def _short_color_list(self):
"""Color list is too long. Discard variations with numbers."""
return [c for c in colors.get_colors() if not re.search(r'\d', c)]
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
root = Gtk.Window(title="ImageViewCanvas Example")
root.set_border_width(2)
root.connect("delete_event", lambda w, e: quit(w))
self.root = root
self.select = GtkHelp.FileSelection(root)
vbox = Gtk.VBox(spacing=2)
fi = ImageViewCanvas(logger)
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
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = DrawingCanvas()
canvas.enable_draw(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.add(canvas)
canvas.ui_setActive(True)
w = fi.get_widget()
w.set_size_request(512, 512)
vbox.pack_start(w, True, True, 1)
self.readout = Gtk.Label("")
vbox.pack_start(self.readout, False, False, 0)
hbox = Gtk.HBox(spacing=5)
wdrawtype = GtkHelp.combo_box_new_text()
self.drawtypes = fi.get_drawtypes()
index = 0
for name in self.drawtypes:
wdrawtype.insert_text(index, name)
index += 1
index = self.drawtypes.index('rectangle')
wdrawtype.set_active(index)
wdrawtype.connect('changed', self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = GtkHelp.combo_box_new_text()
index = 0
for name in self.drawcolors:
wdrawcolor.insert_text(index, name)
index += 1
index = self.drawcolors.index('lightblue')
wdrawcolor.set_active(index)
wdrawcolor.connect('changed', self.set_drawparams)
self.wdrawcolor = wdrawcolor
wfill = GtkHelp.CheckButton("Fill")
wfill.sconnect('toggled', self.set_drawparams)
self.wfill = wfill
walpha = GtkHelp.SpinButton()
adj = walpha.get_adjustment()
adj.configure(0.0, 0.0, 1.0, 0.1, 0.1, 0)
walpha.set_value(1.0)
walpha.set_digits(1)
walpha.sconnect('value-changed', self.set_drawparams)
self.walpha = walpha
wclear = Gtk.Button("Clear Canvas")
wclear.connect('clicked', self.clear_canvas)
wopen = Gtk.Button("Open File")
wopen.connect('clicked', self.open_file)
wquit = Gtk.Button("Quit")
wquit.connect('clicked', quit)
for w in (wquit, wclear, walpha, Gtk.Label("Alpha:"),
wfill, wdrawcolor, wdrawtype, wopen):
hbox.pack_end(w, False, False, 0)
vbox.pack_start(hbox, False, False, 0)
root.add(vbox)
def __init__(self, fv, fitsimage):
"""
This method is called when the plugin is loaded for the first
time. ``fv`` is a reference to the Ginga (reference viewer) shell
and ``fitsimage`` is a reference to the specific ImageViewCanvas
object associated with the channel on which the plugin is being
invoked.
You need to call the superclass initializer and then do any local
initialization.
"""
super(CSU_initializer, self).__init__(fv, fitsimage)
# Load plugin preferences
prefs = self.fv.get_preferences()
self.settings = prefs.createCategory('plugin_CSU_initializer')
self.settings.setDefaults(
bar_num=1,
move_to_open=False,
overlay_bar_positions_from_csu_bar_state_file=False,
overlay_bar_positions_from_fits_header=False,
overlay_bar_positions_from_analyzed_image=False,
bar_dist=0.0,
)
self.settings.load(onError='silent')
self.layertag = 'bars-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
self.canvas_img = None
self.mfilesel = FileSelection(self.fv.w.root.get_widget())
## Define dimensions and angles relative to the pixels of the image
# self.slit_angle = (4.00-0.22) * np.pi/180.
# pixels = np.array([ (721, 2022), # pixels
# (1068, 1934),
# (984, 1804),
# (1112, 40),
# ])
# physical = np.array([ (179.155, self.bar_to_slit(2)), # mm, slit number
# (133.901, self.bar_to_slit(6)),
# (144.962, self.bar_to_slit(12)),
# (129.033, self.bar_to_slit(92))
# ])
pixels = np.array([
(1026.6847023205248, 31.815757489924671),
(1031.1293065907989, 31.815757489924671),
(1100.0527926274958, 76.568051304306408),
(1104.4723170387663, 76.568051304306408),
(869.79921202733158, 119.71402079180322),
(874.17468615739256, 119.71402079180322),
(790.04504261037619, 163.97941699869187),
(794.38269316256697, 163.97941699869187),
(844.76764696920873, 208.45498973235158),
(849.06840834451555, 208.45498973235158),
(918.16119587182891, 253.46863795483193),
(922.57167115281891, 253.46863795483193),
(667.1708458173706, 296.83477802171569),
(671.58750566149126, 296.83477802171569),
(1210.6743343816352, 342.85304935109269),
(1215.1047501727178, 342.85304935109269),
(1037.1504738673596, 386.56200191364559),
(1041.5376839155629, 386.56200191364559),
(1380.9733624348846, 431.75478066748974),
(1385.3923546613969, 431.75478066748974),
(1392.3137244788115, 476.40898670973735),
(1396.5838727543558, 476.40898670973735),
(701.99737614209846, 518.12290417047029),
(706.31972548163674, 518.12290417047029),
(775.43118955263321, 562.76481942553085),
(779.76336695630744, 562.76481942553085),
(695.39446696825667, 606.9386852721824),
(699.68592870194686, 606.9386852721824),
(1225.8966927438423, 652.79237015375304),
(1230.2681865131638, 652.79237015375304),
(1299.3047613957535, 697.52305237026349),
(1303.6542557465727, 697.52305237026349),
(953.60567493512144, 740.39597570556316),
(957.91890612112604, 740.39597570556316),
(1027.0080928255736, 784.70486151318767),
(1031.3650789520013, 784.70486151318767),
(1241.625753053888, 830.10892664282756),
(1245.9181149708163, 830.10892664282756),
(1266.796600696397, 874.17188807394371),
(1271.1082253968038, 874.17188807394371),
(1404.8881828516335, 919.85774261912377),
(1409.9449171925908, 919.85774261912377),
(1325.0207484270156, 963.32163630950686),
(1329.3681702175545, 963.32163630950686),
(1185.9570564396361, 1007.0164717446025),
(1190.2368155733498, 1007.0164717446025),
(1306.6628878384579, 1051.9073888851103),
(1310.9679069215179, 1051.9073888851103),
(1151.3860791138529, 1095.4860726831637),
(1155.7367238283309, 1095.4860726831637),
(1224.7162502034391, 1140.436681012593),
(1229.0598756552718, 1140.436681012593),
(904.70409145100268, 1183.267412335555),
(908.99297982589781, 1183.267412335555),
(978.00762214758913, 1227.9731804278615),
(982.41054057239705, 1227.9731804278615),
(869.65543493075677, 1271.3564678397893),
(873.95299108698168, 1271.3564678397893),
(942.99396243198464, 1316.2391922602001),
(947.36667894787513, 1316.2391922602001),
(1256.7806430753744, 1361.195495916817),
(1261.0847133245632, 1361.195495916817),
(1330.1305637595844, 1406.3795550431571),
(1334.3960288420271, 1406.3795550431571),
(1060.9423305503171, 1449.3586376395574),
(1065.3182032594575, 1449.3586376395574),
(1108.6465868246237, 1493.9756362677167),
(1112.9382994207679, 1493.9756362677167),
(662.84522896384874, 1536.9734554153649),
(667.12956877347722, 1536.9734554153649),
(712.5287834914659, 1581.2712766110319),
(716.80585127180609, 1581.2712766110319),
(956.48762939159371, 1626.1728182002655),
(960.9581522740466, 1626.1728182002655),
(723.23974640617337, 1670.0165354200499),
(727.67208274341931, 1670.0165354200499),
(1172.3594885486252, 1715.8650599984883),
(1176.8341929555718, 1715.8650599984883),
(1015.7329598422145, 1759.5446833817025),
(1020.1920698607528, 1759.5446833817025),
(935.82358262678224, 1803.5644982617907),
(940.3126440130676, 1803.5644982617907),
(989.98752991018682, 1847.9507718487364),
(994.40511955530712, 1847.9507718487364),
(1278.2218422583971, 1892.8072028048214),
(1282.7070969966558, 1892.8072028048214),
(1351.5377751257745, 1938.5923374638328),
(1355.9221844080257, 1938.5923374638328),
(1171.5812780061251, 1981.4914424153424),
(1176.0817255338613, 1981.4914424153424),
])
physical = np.array([
(139.917, self.bar_to_slit(92)),
(139.41, self.bar_to_slit(91)),
(130.322, self.bar_to_slit(90)),
(129.815, self.bar_to_slit(89)),
(160.334, self.bar_to_slit(88)),
(159.827, self.bar_to_slit(87)),
(170.738, self.bar_to_slit(86)),
(170.231, self.bar_to_slit(85)),
(163.579, self.bar_to_slit(84)),
(163.072, self.bar_to_slit(83)),
(153.983, self.bar_to_slit(82)),
(153.476, self.bar_to_slit(81)),
(186.718, self.bar_to_slit(80)),
(186.211, self.bar_to_slit(79)),
(115.773, self.bar_to_slit(78)),
(115.266, self.bar_to_slit(77)),
(138.413, self.bar_to_slit(76)),
(137.906, self.bar_to_slit(75)),
(93.508, self.bar_to_slit(74)),
(93.001, self.bar_to_slit(73)),
(92.021, self.bar_to_slit(72)),
(91.514, self.bar_to_slit(71)),
(182.097, self.bar_to_slit(70)),
(181.59, self.bar_to_slit(69)),
(172.502, self.bar_to_slit(68)),
(171.995, self.bar_to_slit(67)),
(182.905, self.bar_to_slit(66)),
(182.398, self.bar_to_slit(65)),
(113.665, self.bar_to_slit(64)),
(113.158, self.bar_to_slit(63)),
(104.069, self.bar_to_slit(62)),
(103.562, self.bar_to_slit(61)),
(149.161, self.bar_to_slit(60)),
(148.654, self.bar_to_slit(59)),
(139.566, self.bar_to_slit(58)),
(139.059, self.bar_to_slit(57)),
(111.528, self.bar_to_slit(56)),
(111.021, self.bar_to_slit(55)),
(108.22, self.bar_to_slit(54)),
(107.713, self.bar_to_slit(53)),
(90.189, self.bar_to_slit(52)),
(89.681, self.bar_to_slit(51)),
(100.593, self.bar_to_slit(50)),
(100.086, self.bar_to_slit(49)),
(118.731, self.bar_to_slit(48)),
(118.223, self.bar_to_slit(47)),
(102.94, self.bar_to_slit(46)),
(102.432, self.bar_to_slit(45)),
(123.212, self.bar_to_slit(44)),
(122.704, self.bar_to_slit(43)),
(113.615, self.bar_to_slit(42)),
(113.108, self.bar_to_slit(41)),
(155.354, self.bar_to_slit(40)),
(154.847, self.bar_to_slit(39)),
(145.759, self.bar_to_slit(38)),
(145.251, self.bar_to_slit(37)),
(159.887, self.bar_to_slit(36)),
(159.38, self.bar_to_slit(35)),
(150.292, self.bar_to_slit(34)),
(149.785, self.bar_to_slit(33)),
(109.338, self.bar_to_slit(32)),
(108.83, self.bar_to_slit(31)),
(99.742, self.bar_to_slit(30)),
(99.235, self.bar_to_slit(29)),
(134.842, self.bar_to_slit(28)),
(134.335, self.bar_to_slit(27)),
(128.616, self.bar_to_slit(26)),
(128.109, self.bar_to_slit(25)),
(186.778, self.bar_to_slit(24)),
(186.271, self.bar_to_slit(23)),
(180.272, self.bar_to_slit(22)),
(179.765, self.bar_to_slit(21)),
(148.417, self.bar_to_slit(20)),
(147.91, self.bar_to_slit(19)),
(178.822, self.bar_to_slit(18)),
(178.314, self.bar_to_slit(17)),
(120.197, self.bar_to_slit(16)),
(119.689, self.bar_to_slit(15)),
(140.601, self.bar_to_slit(14)),
(140.094, self.bar_to_slit(13)),
(151.005, self.bar_to_slit(12)),
(150.498, self.bar_to_slit(11)),
(143.947, self.bar_to_slit(10)),
(143.44, self.bar_to_slit(9)),
(106.313, self.bar_to_slit(8)),
(105.806, self.bar_to_slit(7)),
(96.717, self.bar_to_slit(6)),
(96.21, self.bar_to_slit(5)),
(120.202, self.bar_to_slit(4)),
(119.695, self.bar_to_slit(3)),
])
tick = dt.now()
self.fit_transforms(pixels, physical)
tock = dt.now()
elapsed = (tock - tick).total_seconds()
print(' Fitted transforms in {:.3f} s'.format(elapsed))
## Determine slit angle and bar center to center distance in pixels
## from the transformation and the known longslit positions
## in longslit, bar 02 is at 145.472
## in longslit, bar 92 is at 129.480
physical = [[145.472, self.bar_to_slit(2)],
[129.480, self.bar_to_slit(92)]]
pixels = self.physical_to_pixel(physical)
dx = pixels[1][0] - pixels[0][0]
dy = pixels[0][1] - pixels[1][1]
self.slit_angle_pix = np.arctan(dx / dy)
print("Slit Angle on CCD = {:.3f} deg".format(self.slit_angle_pix *
180. / np.pi))
self.slit_height_pix = dy / (self.bar_to_slit(92) -
self.bar_to_slit(2))
print("Slit Height on CCD = {:.3f} pix".format(self.slit_height_pix))
def __init__(self, logger, window):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = Viewers.CanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True,
scroll_zoom_acceleration=0.07)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
canvas.register_for_cursor_drawing(fi)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
# add a color bar
fi.show_color_bar(True)
# add little mode indicator that shows keyboard modal states
fi.show_mode_indicator(True, corner='ur')
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed',
lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
## wopen = Widgets.Button("Open File")
## wopen.add_callback('activated', lambda w: self.open_file())
## wquit = Widgets.Button("Quit")
## wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wdrawtype, wdrawcolor, wfill, Widgets.Label('Alpha:'),
walpha, wclear):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated',
lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated',
lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
hbox.add_widget(Widgets.Label('Zoom sensitivity: '))
slider = Widgets.Slider(orientation='horizontal', dtype=float)
slider.add_callback('value-changed',
lambda w, val: self.adjust_scrolling_accel_cb(val))
slider.set_limits(0.0, 12.0, 0.005)
slider.set_value(8.0)
hbox.add_widget(slider, stretch=1)
# hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
# need to put this in an hbox with an expanding label or the
# browser wants to resize the canvas
hbox = Widgets.HBox()
hbox.add_widget(vbox, stretch=0)
hbox.add_widget(Widgets.Label(''), stretch=1)
self.top.set_widget(hbox)
#
# Drawing.py -- Drawing plugin for Ginga reference viewer
#
# This is open-source software licensed under a BSD license.
# Please see the file LICENSE.txt for details.
#
from ginga import GingaPlugin
from ginga import colors
from ginga.gw import Widgets
from ginga.misc import ParamSet, Bunch
from ginga.util import dp
draw_colors = colors.get_colors()
default_drawtype = 'circle'
default_drawcolor = 'lightblue'
fillkinds = ('circle', 'rectangle', 'polygon', 'triangle', 'righttriangle',
'square', 'ellipse', 'box')
class Drawing(GingaPlugin.LocalPlugin):
"""
Drawing
=======
A plugin for drawing canvas forms (overlaid graphics).
Plugin Type: Local
------------------
Drawing is a local plugin, which means it is associated with a
channel. An instance can be opened for each channel.
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = colors.get_colors()
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(True)
fi.set_drawtype('ruler')
fi.set_drawcolor('blue')
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
fi.set_figure(fig)
fi.get_bindings().enable_all(True)
w.resize(512, 512)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wdrawtype = QtGui.QComboBox()
self.drawtypes = fi.get_drawtypes()
for name in self.drawtypes:
wdrawtype.addItem(name)
index = self.drawtypes.index('ruler')
wdrawtype.setCurrentIndex(index)
wdrawtype.activated.connect(self.set_drawparams)
self.wdrawtype = wdrawtype
wdrawcolor = QtGui.QComboBox()
for name in self.drawcolors:
wdrawcolor.addItem(name)
index = self.drawcolors.index('blue')
wdrawcolor.setCurrentIndex(index)
wdrawcolor.activated.connect(self.set_drawparams)
self.wdrawcolor = wdrawcolor
wclear = QtGui.QPushButton("Clear Canvas")
wclear.clicked.connect(self.clear_canvas)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wopen, wdrawtype, wdrawcolor, wclear, wquit):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
import math
import numpy
from collections import namedtuple
from ginga.misc import Callback, Bunch
from ginga import trcalc, colors
from ginga.util.six.moves import map
from . import coordmap
__all__ = [
'CanvasObjectBase', 'get_canvas_type', 'get_canvas_types',
'register_canvas_type', 'register_canvas_types'
]
colors_plus_none = [None] + colors.get_colors()
Point = namedtuple('Point', ['x', 'y'])
class EditPoint(Point):
edit_color = 'yellow'
class MovePoint(EditPoint):
edit_color = 'orangered'
class ScalePoint(EditPoint):
edit_color = 'green'
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
from ginga.gw import Widgets, Viewers
self.app = Widgets.Application(logger=logger)
#self.app.add_callback('shutdown', self.quit)
self.top = self.app.make_window("Ginga shared canvas example")
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
hbox = Widgets.HBox()
hbox.set_border_width(2)
hbox.set_spacing(4)
v1 = Viewers.CanvasView(logger)
v1.enable_autocuts('on')
v1.set_autocut_params('zscale')
v1.enable_autozoom('on')
v1.set_zoom_algorithm('rate')
v1.set_zoomrate(1.4)
v1.show_pan_mark(True)
v1.set_callback('drag-drop', self.drop_file)
v1.set_callback('none-move', self.motion)
v1.set_bg(0.2, 0.2, 0.2)
v1.ui_set_active(True)
v1.set_name('tweedledee')
self.viewer1 = v1
self._mi1 = ModeIndicator(v1)
bd = v1.get_bindings()
bd.enable_all(True)
# shared canvas between the two viewers
shcanvas = self.dc.DrawingCanvas()
# Tell viewer1 to use this canvas
v1.set_canvas(shcanvas)
v1.set_desired_size(300, 300)
iw = Viewers.GingaViewerWidget(viewer=v1)
hbox.add_widget(iw, stretch=1)
# Add a second viewer viewing the same canvas
v2 = Viewers.CanvasView(logger)
v2.enable_autocuts('on')
v2.set_autocut_params('zscale')
v2.enable_autozoom('on')
v2.set_zoom_algorithm('rate')
v2.set_zoomrate(1.4)
v2.show_pan_mark(True)
v2.set_callback('drag-drop', self.drop_file)
v2.set_callback('none-move', self.motion)
v2.set_bg(0.2, 0.2, 0.2)
v2.ui_set_active(True)
v1.set_name('tweedledum')
self.viewer2 = v2
self._mi2 = ModeIndicator(v2)
# Tell viewer2 to use this same canvas
v2.set_canvas(shcanvas)
bd = v2.get_bindings()
bd.enable_all(True)
v2.set_desired_size(300, 300)
iw = Viewers.GingaViewerWidget(viewer=v2)
hbox.add_widget(iw, stretch=1)
# 2nd canvas as a subcanvas of the shared canvas
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.register_for_cursor_drawing(v1)
canvas.register_for_cursor_drawing(v2)
canvas.set_drawtype('rectangle', color='lightblue')
self.canvas = canvas
shcanvas.add(self.canvas)
shcanvas.ui_set_active(True)
canvas.ui_set_active(True)
canvas.set_surface(v1)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
vbox.add_widget(hbox, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated', lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed', lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
wopen = Widgets.Button("Open File")
wopen.add_callback('activated', lambda w: self.open_file())
wquit = Widgets.Button("Quit")
wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wopen, wdrawtype, wdrawcolor, wfill,
Widgets.Label('Alpha:'), walpha, wclear, wquit):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated', lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated', lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
## btn3 = Widgets.CheckBox("I'm using a trackpad")
## btn3.add_callback('activated', lambda w, tf: self.use_trackpad_cb(tf))
## hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
def _short_color_list(self):
"""Color list is too long. Discard variations with numbers."""
return [c for c in colors.get_colors() if not re.search(r'\d', c)]
def __init__(self, logger, window):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = EnhancedCanvasView(logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
canvas.register_for_cursor_drawing(fi)
fi.set_canvas(canvas)
## self.drawtypes = canvas.get_drawtypes()
## self.drawtypes.sort()
# add a color bar
private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# the corner
private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
btn3 = Widgets.CheckBox("I'm using a trackpad")
btn3.add_callback('activated', lambda w, tf: self.use_trackpad_cb(tf))
hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
#
# Drawing.py -- Drawing plugin for Ginga reference viewer
#
# This is open-source software licensed under a BSD license.
# Please see the file LICENSE.txt for details.
#
from ginga import GingaPlugin
from ginga import colors
from ginga.gw import Widgets
from ginga.misc import ParamSet, Bunch
from ginga.util import dp
draw_colors = colors.get_colors()
default_drawtype = 'circle'
default_drawcolor = 'lightblue'
fillkinds = ('circle', 'rectangle', 'polygon', 'triangle', 'righttriangle',
'square', 'ellipse', 'box')
class Drawing(GingaPlugin.LocalPlugin):
"""Local plugin to draw shapes on canvas."""
def __init__(self, fv, fitsimage):
# superclass defines some variables for us, like logger
super(Drawing, self).__init__(fv, fitsimage)
self.layertag = 'drawing-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
def __init__(self, logger, window, viewer_class=None):
if viewer_class is None:
viewer_class = EnhancedCanvasView
self.logger = logger
self.url = window.url
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
self.top = window
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = viewer_class(logger)
fi.url = self.url
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# so trackpad scrolling can be adjusted
settings = bd.get_settings()
settings.set(scroll_zoom_direct_scale=True)
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
#canvas.enable_draw(True)
#canvas.enable_edit(True)
#canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
private_canvas = fi.get_canvas()
private_canvas.add(canvas)
canvas.ui_setActive(True)
#canvas.register_for_cursor_drawing(fi)
fi.set_canvas(canvas)
## self.drawtypes = canvas.get_drawtypes()
## self.drawtypes.sort()
# add a color bar
private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# the corner
private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
fi.set_desired_size(512, 512)
w = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(w, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
btn3 = Widgets.CheckBox("I'm using a trackpad")
btn3.add_callback('activated', lambda w, tf: self.use_trackpad_cb(tf))
hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
# need to put this in an hbox with an expanding label or the
# browser wants to resize the canvas, distorting it
hbox = Widgets.HBox()
hbox.add_widget(vbox, stretch=0)
hbox.add_widget(Widgets.Label(''), stretch=1)
self.top.set_widget(hbox)
def __init__(self, logger):
self.logger = logger
self.drawcolors = colors.get_colors()
self.dc = get_canvas_types()
from ginga.gw import Widgets, Viewers, GwHelp
self.app = Widgets.Application(logger=logger)
self.app.add_callback('shutdown', self.quit)
self.top = self.app.make_window("Ginga example2")
self.top.add_callback('close', self.closed)
vbox = Widgets.VBox()
vbox.set_border_width(2)
vbox.set_spacing(1)
fi = Viewers.CanvasView(logger=logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.set_zoom_algorithm('rate')
fi.set_zoomrate(1.4)
fi.show_pan_mark(True)
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
bd = fi.get_bindings()
bd.enable_all(True)
# add a color bar
fi.private_canvas.add(self.dc.ColorBar(side='bottom', offset=10))
# add little mode indicator that shows modal states in
# lower left hand corner
fi.private_canvas.add(self.dc.ModeIndicator(corner='ur', fontsize=14))
# little hack necessary to get correct operation of the mode indicator
# in all circumstances
bm = fi.get_bindmap()
bm.add_callback('mode-set', lambda *args: fi.redraw(whence=3))
# canvas that we will draw on
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(True)
canvas.enable_edit(True)
canvas.set_drawtype('rectangle', color='lightblue')
canvas.setSurface(fi)
self.canvas = canvas
# add canvas to view
fi.get_canvas().add(canvas)
canvas.ui_setActive(True)
canvas.register_for_cursor_drawing(fi)
self.drawtypes = canvas.get_drawtypes()
self.drawtypes.sort()
fi.set_desired_size(512, 512)
iw = Viewers.GingaViewerWidget(viewer=fi)
vbox.add_widget(iw, stretch=1)
self.readout = Widgets.Label("")
vbox.add_widget(self.readout, stretch=0)
hbox = Widgets.HBox()
hbox.set_border_width(2)
wdrawtype = Widgets.ComboBox()
for name in self.drawtypes:
wdrawtype.append_text(name)
index = self.drawtypes.index('rectangle')
wdrawtype.set_index(index)
wdrawtype.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawtype = wdrawtype
wdrawcolor = Widgets.ComboBox()
for name in self.drawcolors:
wdrawcolor.append_text(name)
index = self.drawcolors.index('lightblue')
wdrawcolor.set_index(index)
wdrawcolor.add_callback('activated',
lambda w, idx: self.set_drawparams())
self.wdrawcolor = wdrawcolor
wfill = Widgets.CheckBox("Fill")
wfill.add_callback('activated', lambda w, tf: self.set_drawparams())
self.wfill = wfill
walpha = Widgets.SpinBox(dtype=float)
walpha.set_limits(0.0, 1.0, incr_value=0.1)
walpha.set_value(1.0)
walpha.set_decimals(2)
walpha.add_callback('value-changed',
lambda w, val: self.set_drawparams())
self.walpha = walpha
wclear = Widgets.Button("Clear Canvas")
wclear.add_callback('activated', lambda w: self.clear_canvas())
wopen = Widgets.Button("Open File")
wopen.add_callback('activated', lambda w: self.open_file())
wquit = Widgets.Button("Quit")
wquit.add_callback('activated', lambda w: self.quit())
hbox.add_widget(Widgets.Label(''), stretch=1)
for w in (wopen, wdrawtype, wdrawcolor, wfill, Widgets.Label('Alpha:'),
walpha, wclear, wquit):
hbox.add_widget(w, stretch=0)
vbox.add_widget(hbox, stretch=0)
mode = self.canvas.get_draw_mode()
hbox = Widgets.HBox()
btn1 = Widgets.RadioButton("Draw")
btn1.set_state(mode == 'draw')
btn1.add_callback('activated',
lambda w, val: self.set_mode_cb('draw', val))
btn1.set_tooltip("Choose this to draw on the canvas")
hbox.add_widget(btn1)
btn2 = Widgets.RadioButton("Edit", group=btn1)
btn2.set_state(mode == 'edit')
btn2.add_callback('activated',
lambda w, val: self.set_mode_cb('edit', val))
btn2.set_tooltip("Choose this to edit things on the canvas")
hbox.add_widget(btn2)
btn3 = Widgets.CheckBox("I'm using a trackpad")
btn3.add_callback('activated', lambda w, tf: self.use_trackpad_cb(tf))
hbox.add_widget(btn3)
hbox.add_widget(Widgets.Label(''), stretch=1)
vbox.add_widget(hbox, stretch=0)
self.top.set_widget(vbox)
self.fs = None
if hasattr(GwHelp, 'FileSelection'):
self.fs = GwHelp.FileSelection(self.top.get_widget())
def __init__(self, fv, fitsimage):
"""
This method is called when the plugin is loaded for the first
time. ``fv`` is a reference to the Ginga (reference viewer) shell
and ``fitsimage`` is a reference to the specific ImageViewCanvas
object associated with the channel on which the plugin is being
invoked.
You need to call the superclass initializer and then do any local
initialization.
"""
super(CSU_initializer, self).__init__(fv, fitsimage)
# Load plugin preferences
prefs = self.fv.get_preferences()
self.settings = prefs.createCategory('plugin_CSU_initializer')
self.settings.setDefaults(
ibar_num=1,
mbar_num=1,
ebar_num=1,
move_to_open=False,
bar_dest=0.0,
bar_pos=137.0,
)
self.settings.load(onError='silent')
self.instrument_hosts = ['vm-mosfire', 'nuu', 'vm-mosfirebld']
self.hostname = socket.gethostname().split('.')[0].lower()
self.bars_analysis = None
self.state_analysis = None
self.bars_file = None
self.state_file = None
self.bars_header = None
self.state_header = None
self.layertag = 'bars-canvas'
self.dc = fv.get_draw_classes()
canvas = self.dc.DrawingCanvas()
canvas.enable_draw(False)
canvas.set_surface(self.fitsimage)
self.canvas = canvas
self.colornames = colors.get_colors()
self.canvas_img = None
self.mfilesel = FileSelection(self.fv.w.root.get_widget())
## Fit relationship between bar position and pixels
tick = dt.now()
pixels, physical = self.get_data()
self.fit_transforms(pixels, physical)
tock = dt.now()
elapsed = (tock - tick).total_seconds()
# print('Completed fit of transforms in {:.3f} s'.format(elapsed))
## Determine slit angle and bar center to center distance in pixels
## from the transformation and the known longslit positions
## in longslit, bar 02 is at 145.472
## in longslit, bar 92 is at 129.480
physical = [[145.472, self.bar_to_slit(2)],
[129.480, self.bar_to_slit(92)]]
pixels = self.physical_to_pixel(physical)
dx = pixels[1][0] - pixels[0][0]
dy = pixels[0][1] - pixels[1][1]
self.slit_angle_pix = np.arctan(dx / dy)
# print("Slit Angle on CCD = {:.3f} deg".format(self.slit_angle_pix * 180./np.pi))
self.slit_height_pix = dy / (self.bar_to_slit(92) -
self.bar_to_slit(2))