def gui_add_channel(self, chname=None):
if not chname:
self.chncnt += 1
chname = "Image%d" % self.chncnt
lbl = QtGui.QLabel('New channel name:')
ent = QtGui.QLineEdit()
ent.setText(chname)
lbl2 = QtGui.QLabel('Workspace:')
cbox = QtHelp.ComboBox()
names = self.ds.get_wsnames()
try:
idx = names.index(self._lastwsname)
except:
idx = 0
for name in names:
cbox.append_text(name)
cbox.setCurrentIndex(idx)
dialog = QtHelp.Dialog(
"Add Channel", 0, [['Cancel', 0], ['Ok', 1]],
lambda w, rsp: self.add_channel_cb(w, rsp, ent, cbox, names))
box = dialog.get_content_area()
layout = QtGui.QVBoxLayout()
box.setLayout(layout)
layout.addWidget(lbl, stretch=0)
layout.addWidget(ent, stretch=0)
layout.addWidget(lbl2, stretch=0)
layout.addWidget(cbox, stretch=0)
dialog.show()
def build_gui(self, container):
sw = QtGui.QScrollArea()
twidget = QtHelp.VBox()
sp = QtGui.QSizePolicy(QtGui.QSizePolicy.MinimumExpanding,
QtGui.QSizePolicy.Fixed)
twidget.setSizePolicy(sp)
vbox1 = twidget.layout()
vbox1.setContentsMargins(4, 4, 4, 4)
vbox1.setSpacing(2)
sw.setWidgetResizable(True)
sw.setWidget(twidget)
msgFont = QtGui.QFont("Sans", 14)
tw = QtGui.QLabel()
tw.setFont(msgFont)
tw.setWordWrap(True)
self.tw = tw
fr = QtHelp.Frame("Instructions")
fr.layout().addWidget(tw, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
fr = QtHelp.Frame("HDU")
captions = [("Num HDUs", 'label'), ("Choose HDU", 'spinbutton')]
w, b = QtHelp.build_info(captions)
self.w.update(b)
self.w.numhdu = b.num_hdus
self.w.hdu = b.choose_hdu
self.w.hdu.valueChanged.connect(self.set_hdu_cb)
fr.layout().addWidget(w, stretch=1, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
fr = QtHelp.Frame("NAXIS")
self.stack = QtHelp.StackedWidget()
self.stack.addWidget(QtGui.QLabel(''))
fr.layout().addWidget(self.stack,
stretch=1,
alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
btns = QtHelp.HBox()
layout = btns.layout()
layout.setSpacing(3)
#btns.set_child_size(15, -1)
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
layout.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignLeft)
container.addWidget(sw, stretch=1)
def build_gui(self, container):
rvbox = container
if not has_webkit:
self.browser = QtGui.QLabel("Please install the python-webkit package to enable this plugin")
else:
self.browser = webkit.QWebView()
sw = QtGui.QScrollArea()
sw.setWidgetResizable(True)
#sw.set_border_width(2)
sw.setWidget(self.browser)
rvbox.addWidget(sw, stretch=1)
sw.show()
self.entry = QtGui.QLineEdit()
rvbox.addWidget(self.entry, stretch=0)
self.entry.returnPressed.connect(self.browse_cb)
if has_webkit:
helpfile = os.path.abspath(os.path.join(moduleHome, "..",
"..", "doc", "manual",
"quickref.html"))
helpurl = "file://%s" % (helpfile)
self.entry.setText(helpurl)
self.browse(helpurl)
def view_database(self, selected_entries=None):
'''
Logic for creating and viewing of a tab with the database table
'''
print "Viewing"
table_headers = [
'id', 'File', 'Observation Time Start', 'Observation Time End',
'Instrument', 'Min Wavelength', 'Max Wavelength', 'Starred'
]
self.table_headers = table_headers
vbox = QtHelp.VBox()
filter_label = QtGui.QLabel("Filter")
self.filter_label = filter_label
filter_select = QtGui.QComboBox()
search_box = QtGui.QLineEdit()
search_box.textChanged.connect(self.query)
self.search_box = search_box
if selected_entries == None:
selected_entries = database
queries = self.get_data_from_selected_entries(selected_entries)
wtable = QtGui.QTableWidget(len(queries), len(table_headers))
for i, row in enumerate(queries):
for j, col in enumerate(row):
item = QtGui.QTableWidgetItem(str(col))
wtable.setItem(i, j, item)
wtable.itemClicked.connect(self.on_table_row_click)
wtable.setHorizontalHeaderLabels(table_headers)
wtable.horizontalHeader().setResizeMode(
QtGui.QHeaderView.ResizeToContents)
self.wtable = wtable
if 'sunpydb' in self.fv.ds.get_tabnames():
self.fv.ds.remove_tab("sunpydb")
filter_select.addItems(self.table_headers[:-1])
self.filter_select = filter_select
vbox.addWidget(filter_label, stretch=0)
vbox.addWidget(filter_select, stretch=0)
vbox.addWidget(search_box)
vbox.addWidget(wtable, stretch=0)
starred_entries_box = QtGui.QCheckBox("Show starred entries only")
self.starred_entries_box = starred_entries_box
starred_entries_box.stateChanged.connect(self.starred_entries_only)
vbox.addWidget(starred_entries_box)
self.fv.ds.add_tab("right", vbox, 1, "SunPyDB", tabname="sunpydb")
self.fv.ds.raise_tab("sunpydb")
def add_error(self, errmsg):
vbox = QtHelp.VBox()
tw = QtGui.QTextEdit()
tw.setReadOnly(True)
tw.setCurrentFont(self.msgFont)
tw.setText(errmsg)
vbox.addWidget(tw, stretch=1)
hbox = QtHelp.HBox()
btn = QtGui.QPushButton("Remove")
btn.clicked.connect(lambda: self.remove_error(vbox))
hbox.addWidget(btn, stretch=0)
# Add the time the error occurred
ts = time.strftime("%m/%d %H:%M:%S", time.localtime())
lbl = QtGui.QLabel(ts)
hbox.addWidget(lbl, stretch=0)
vbox.addWidget(hbox, stretch=0)
layout = self.msgList.layout()
layout.addWidget(vbox,
layout.rowCount(),
0,
alignment=QtCore.Qt.AlignTop)
self.widgetList.append(vbox)
def add_taskbar(self, bnch):
lblname = bnch.lblname
lbl = QtGui.QLabel(lblname)
lbl.setAlignment(QtCore.Qt.AlignHCenter)
lbl.setToolTip("Right click for menu")
lbl.setSizePolicy(QtGui.QSizePolicy.Minimum, QtGui.QSizePolicy.Minimum)
lbl.setFrameStyle(QtGui.QFrame.Box | QtGui.QFrame.Raised)
self.hbox.addWidget(lbl, stretch=0, alignment=QtCore.Qt.AlignLeft)
lbl.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
lname = bnch.pInfo.name.lower()
menu = QtGui.QMenu()
item = QtGui.QAction("Focus", menu)
item.triggered.connect(lambda: self.set_focus(lname))
menu.addAction(item)
item = QtGui.QAction("Unfocus", menu)
item.triggered.connect(lambda: self.clear_focus(lname))
menu.addAction(item)
item = QtGui.QAction("Stop", menu)
item.triggered.connect(lambda: self.deactivate(lname))
menu.addAction(item)
def on_context_menu(point):
menu.exec_(lbl.mapToGlobal(point))
lbl.customContextMenuRequested.connect(on_context_menu)
# better than making a whole new subclass just to get a label to
# respond to a mouse click
lbl.mousePressEvent = lambda event: self.set_focus_cb(event, lname)
bnch.setvals(widget=lbl, label=lbl, menu=menu)
def build_gui(self, container):
if not has_webkit:
self.browser = QtGui.QLabel("Please install the python-webkit package to enable this plugin")
else:
self.browser = QWebView()
sw = QtGui.QScrollArea()
sw.setWidgetResizable(True)
#sw.set_border_width(2)
sw.setWidget(self.browser)
cw = container.get_widget()
cw.layout().addWidget(sw, stretch=1)
sw.show()
self.entry = QtGui.QLineEdit()
cw.layout().addWidget(self.entry, stretch=0)
self.entry.returnPressed.connect(self.browse_cb)
btns = QtHelp.HBox()
layout = btns.layout()
layout.setSpacing(3)
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
layout.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
cw.layout().addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignLeft)
if has_webkit:
helpfile = os.path.abspath(os.path.join(packageHome,
"doc", "help.html"))
helpurl = "file://%s" % (helpfile)
self.browse(helpurl)
def insert_thumbnail(self, imgwin, thumbkey, thumbname, chname, name, path,
thumbpath, metadata, image_loader):
pixmap = QPixmap.fromImage(imgwin)
imglbl = MyLabel()
imglbl.setPixmap(pixmap)
imglbl.thumbs_cb = lambda: self.load_file(thumbkey, chname, name, path,
image_loader)
text = self.query_thumb(thumbkey, name, metadata)
imglbl.setToolTip(text)
widget = QtGui.QWidget()
#vbox = QtGui.QGridLayout()
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(0, 0, 0, 0)
vbox.setSpacing(0)
widget.setLayout(vbox)
namelbl = QtGui.QLabel(thumbname)
namelbl.setAlignment(QtCore.Qt.AlignLeft)
namelbl.setAlignment(QtCore.Qt.AlignHCenter)
## vbox.addWidget(namelbl, 0, 0)
## vbox.addWidget(imglbl, 1, 0)
vbox.addWidget(namelbl, stretch=0)
vbox.addWidget(imglbl, stretch=0)
widget.setSizePolicy(
QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Fixed))
bnch = Bunch.Bunch(widget=widget,
image=imgwin,
layout=vbox,
imglbl=imglbl,
name=name,
imname=name,
chname=chname,
path=path,
thumbpath=thumbpath,
pixmap=pixmap)
with self.thmblock:
self.thumbDict[thumbkey] = bnch
self.thumbList.append(thumbkey)
self.w.thumbs.addWidget(widget, self.thumbRowCount,
self.thumbColCount)
self.thumbColCount = (self.thumbColCount + 1) % self.thumbNumCols
if self.thumbColCount == 0:
self.thumbRowCount += 1
#self.w.thumbs.show()
# force scroll to bottom of thumbs, if checkbox is set
scrollp = self.w.auto_scroll.isChecked()
if scrollp:
self.fv.update_pending()
area = self.w.thumbs_scroll
area.verticalScrollBar().setValue(
area.verticalScrollBar().maximum())
self.logger.debug("added thumb for %s" % (thumbname))
def gui_delete_channel(self):
chinfo = self.get_channelInfo()
chname = chinfo.name
lbl = QtGui.QLabel("Really delete channel '%s' ?" % (chname))
dialog = QtHelp.Dialog(
"Delete Channel", 0, [['Cancel', 0], ['Ok', 1]],
lambda w, rsp: self.delete_channel_cb(w, rsp, chname))
box = dialog.get_content_area()
layout = QtGui.QVBoxLayout()
box.setLayout(layout)
layout.addWidget(lbl, stretch=0)
dialog.show()
def __init__(self, width, height):
readout = QtGui.QLabel('')
readout.resize(width, height)
readout.setStyleSheet(
"QLabel { background-color: #202030; color: lightgreen; }")
readout.setAlignment(QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter)
self.readout = readout
self.maxx = 0
self.maxy = 0
self.maxv = 0
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
fig = Figure()
w = FigureCanvas(fig)
fi = ImageViewCanvas(logger=self.logger)
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_auto_orient(True)
fi.enable_autozoom('on')
#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_set_active(True)
fi.set_figure(fig)
self.fitsimage = fi
fi.get_bindings().enable_all(True)
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))
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, 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)
fi.configure(512, 512)
def insert_thumbnail(self, imgwin, thumbkey, thumbname, chname, name, path,
metadata):
pixmap = QtGui.QPixmap.fromImage(imgwin)
imglbl = MyLabel()
imglbl.setPixmap(pixmap)
imglbl.thumbs_cb = lambda: self.fv.switch_name(chname, name, path=path)
text = self.query_thumb(thumbkey, name, metadata)
imglbl.setToolTip(text)
widget = QtGui.QWidget()
#vbox = QtGui.QGridLayout()
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(0, 0, 0, 0)
vbox.setSpacing(0)
widget.setLayout(vbox)
namelbl = QtGui.QLabel(thumbname)
namelbl.setAlignment(QtCore.Qt.AlignLeft)
namelbl.setAlignment(QtCore.Qt.AlignHCenter)
## vbox.addWidget(namelbl, 0, 0)
## vbox.addWidget(imglbl, 1, 0)
vbox.addWidget(namelbl, stretch=0)
vbox.addWidget(imglbl, stretch=0)
widget.setSizePolicy(
QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Fixed))
#widget.show()
bnch = Bunch.Bunch(widget=widget,
image=imgwin,
layout=vbox,
imglbl=imglbl,
name=name,
chname=chname,
path=path,
pixmap=pixmap)
self.thumbDict[thumbkey] = bnch
self.thumbList.append(thumbkey)
self.w.thumbs.addWidget(widget, self.thumbRowCount, self.thumbColCount)
self.thumbColCount = (self.thumbColCount + 1) % self.thumbNumCols
if self.thumbColCount == 0:
self.thumbRowCount += 1
#self.w.thumbs.show()
# force scroll to bottom of thumbs
## rect = self.w.thumbs_w.geometry()
## x1, y1, x2, y2 = rect.getCoords()
## self.w.thumbs_scroll.ensureVisible(x1, y1)
#self.w.thumbs_scroll.show()
self.logger.debug("added thumb for %s" % (thumbname))
def build_gui(self, container):
#self.msgFont = QtGui.QFont("Fixed", 10)
tw = QtGui.QPlainTextEdit()
tw.setReadOnly(True)
#tw.setCurrentFont(self.msgFont)
tw.setLineWrapMode(QtGui.QPlainTextEdit.NoWrap)
tw.setMaximumBlockCount(self.histlimit)
self.tw = tw
sw = QtGui.QScrollArea()
sw.setWidgetResizable(True)
sw.setWidget(self.tw)
container.addWidget(sw, stretch=1)
sw.show()
hbox = QtHelp.HBox()
lbl = QtGui.QLabel('Level:')
hbox.addWidget(lbl, stretch=0)
combobox = QtHelp.ComboBox()
for (name, level) in self.levels:
combobox.addItem(name)
combobox.setCurrentIndex(1)
combobox.activated.connect(self.set_loglevel_cb)
hbox.addWidget(combobox, stretch=0)
lbl = QtGui.QLabel('History:')
hbox.addWidget(lbl, stretch=0)
spinbox = QtGui.QSpinBox()
spinbox.setRange(100, self.histmax)
spinbox.setSingleStep(10)
spinbox.valueChanged.connect(self.set_history_cb)
hbox.addWidget(spinbox, stretch=0)
btn = QtGui.QPushButton("Clear")
btn.clicked.connect(self.clear)
hbox.addWidget(btn, stretch=0)
container.addWidget(hbox, stretch=0)
def build_gui(self, container):
sw = QtGui.QScrollArea()
twidget = QtHelp.VBox()
sp = QtGui.QSizePolicy(QtGui.QSizePolicy.MinimumExpanding,
QtGui.QSizePolicy.Fixed)
twidget.setSizePolicy(sp)
vbox1 = twidget.layout()
vbox1.setContentsMargins(4, 4, 4, 4)
vbox1.setSpacing(2)
sw.setWidgetResizable(True)
sw.setWidget(twidget)
msgFont = QtGui.QFont("Sans", 14)
tw = QtGui.QLabel()
tw.setFont(msgFont)
tw.setWordWrap(True)
self.tw = tw
fr = QtHelp.Frame("Instructions")
fr.layout().addWidget(tw, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
fr = QtHelp.Frame("Ruler")
captions = (('Units', 'combobox'), )
w, b = QtHelp.build_info(captions)
self.w = b
combobox = b.units
for name in self.unittypes:
combobox.addItem(name)
index = self.unittypes.index(self.units)
combobox.setCurrentIndex(index)
combobox.activated.connect(self.set_units)
fr.layout().addWidget(w, stretch=1, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
btns = QtHelp.HBox()
layout = btns.layout()
layout.setSpacing(3)
#btns.set_child_size(15, -1)
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
layout.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignLeft)
container.addWidget(sw, stretch=1)
def build_gui(self, container):
canvas = CanvasTypes.DrawingCanvas()
canvas.enable_draw(False)
#canvas.set_callback('button-press', self.btndown)
canvas.set_callback('motion', self.cursormotion)
#canvas.set_callback('button-release', self.update)
canvas.add_callback('key-press', self.window_key_press)
canvas.add_callback('key-release', self.window_key_release)
self.canvas = canvas
vbox = QtHelp.VBox()
sp = QtGui.QSizePolicy(QtGui.QSizePolicy.MinimumExpanding,
QtGui.QSizePolicy.Fixed)
vbox.setSizePolicy(sp)
fr = QtHelp.Frame("IRAF")
captions = [
("Control", 'hbox'),
("Channel", 'label'),
]
w, b = QtHelp.build_info(captions)
self.w = b
self.w.mode_d = {}
btn = QtGui.QRadioButton("Ginga")
btn.toggled.connect(lambda w: self.switchMode('ginga'))
self.w.mode_d['ginga'] = btn
self.w.control.layout().addWidget(btn,
stretch=0,
alignment=QtCore.Qt.AlignLeft)
btn = QtGui.QRadioButton("IRAF")
btn.toggled.connect(lambda w: self.switchMode('iraf'))
self.w.mode_d['iraf'] = btn
self.w.control.layout().addWidget(btn,
stretch=0,
alignment=QtCore.Qt.AlignLeft)
fr.layout().addWidget(w, stretch=1, alignment=QtCore.Qt.AlignLeft)
vbox.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
fr = QtHelp.Frame("Frame/Channel")
lbl = QtGui.QLabel("")
self.w.frch = lbl
fr.layout().addWidget(lbl, stretch=1, alignment=QtCore.Qt.AlignLeft)
vbox.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
container.addWidget(vbox, stretch=0, alignment=QtCore.Qt.AlignTop)
def __init__(self, title=None):
super(Frame, self).__init__()
self.widget = QtGui.QFrame()
self.widget.setFrameStyle(QtGui.QFrame.Box | QtGui.QFrame.Raised)
vbox = QtGui.QVBoxLayout()
# because of ridiculous defaults
vbox.setContentsMargins(2, 2, 2, 2)
self.widget.setLayout(vbox)
if title:
lbl = QtGui.QLabel(title)
lbl.setAlignment(QtCore.Qt.AlignHCenter)
vbox.addWidget(lbl, stretch=0)
self.label = lbl
else:
self.label = None
def gui_add_channel(self, chname=None):
if not chname:
self.chncnt += 1
chname = "Image%d" % self.chncnt
lbl = QtGui.QLabel('New channel name:')
ent = QtGui.QLineEdit()
ent.setText(chname)
#ent.set_activates_default(True)
dialog = QtHelp.Dialog("New Channel", 0, [['Cancel', 0], ['Ok', 1]],
lambda w, rsp: self.new_channel_cb(w, rsp, ent))
box = dialog.get_content_area()
layout = QtGui.QVBoxLayout()
box.setLayout(layout)
layout.addWidget(lbl, stretch=0)
layout.addWidget(ent, stretch=0)
dialog.show()
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
self.drawcolors = ['white', 'black', 'red', 'yellow', 'blue', 'green']
fi = FitsImageCanvas(logger, render='widget')
fi.enable_autocuts('on')
fi.enable_zoom('on')
fi.enable_cuts(True)
fi.enable_flip(True)
fi.enable_rotate(True)
fi.enable_draw(True)
fi.set_drawtype('ruler')
fi.set_drawcolor('blue')
fi.set_callback('drag-drop', self.drop_file)
fi.set_callback('motion', self.motion)
fi.set_bg(0.2, 0.2, 0.2)
fi.ui_setActive(True)
self.fitsimage = fi
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):
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.cachedFiles = None
self.scanning = False
#KTL stuff
#Cache KTL keywords
self.trickxpos = ktl.cache('ao', 'TRKRO1XP')
self.trickypos = ktl.cache('ao', 'TRKRO1YP')
self.trickxsize = ktl.cache('ao', 'TRKRO1XS')
self.trickysize = ktl.cache('ao', 'TRKRO1YS')
self.rawfile = ''
self.threadpool = QtCore.QThreadPool()
self.iqcalc = iqcalc.IQCalc(self.logger)
# 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.set_bg(0.2, 0.2, 0.2)
fi.ui_set_active(True)
self.fitsimage = fi
# enable some user interaction
self.bd = fi.get_bindings()
self.bd.enable_all(True)
w = fi.get_widget()
w.resize(512, 512)
# 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)
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
self.readout = QtGui.QLabel("")
hbox.addStretch(1)
hbox.addWidget(self.readout, stretch=0)
self.box_readout = QtGui.QLabel("")
hbox.addStretch(1)
hbox.addWidget(self.box_readout, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
vbox.addWidget(hw, stretch=0)
hbox2 = QtGui.QHBoxLayout()
hbox2.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
self.wstartscan = QtGui.QPushButton("Start Scan")
self.wstartscan.clicked.connect(self.start_scan)
self.wstartscan.setEnabled(True)
self.wstopscan = QtGui.QPushButton("Stop Scan")
self.wstopscan.clicked.connect(self.stop_scan)
self.wstopscan.setEnabled(False)
self.wcut = QtGui.QComboBox()
for name in fi.get_autocut_methods():
self.wcut.addItem(name)
self.wcut.currentIndexChanged.connect(self.cut_change)
self.wcolor = QtGui.QComboBox()
for name in fi.get_color_algorithms():
self.wcolor.addItem(name)
self.wcolor.currentIndexChanged.connect(self.color_change)
wopen = QtGui.QPushButton("Open File")
wopen.clicked.connect(self.open_file)
self.wsky = QtGui.QPushButton("Load Sky")
self.wsky.clicked.connect(self.load_sky)
self.wsky.setEnabled(False)
wquit = QtGui.QPushButton("Quit")
wquit.clicked.connect(self.quit)
fi.set_callback('cursor-changed', self.motion_cb)
fi.add_callback('cursor-down', self.btndown)
hbox2.addStretch(1)
for w in (self.wstartscan, self.wstopscan, self.wcut, self.wcolor,
wopen, self.wsky, wquit):
hbox2.addWidget(w, stretch=0)
hw2 = QtGui.QWidget()
hw2.setLayout(hbox2)
vbox.addWidget(hw2, stretch=0)
vw = QtGui.QWidget()
self.setCentralWidget(vw)
vw.setLayout(vbox)
self.recdc, self.compdc = self.add_canvas()
self.boxtag = "roi-box"
self.picktag = "pick-box"
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 build_gui(self, container):
# Splitter is just to provide a way to size the graph
# to a reasonable size
vpaned = QtGui.QSplitter()
vpaned.setOrientation(QtCore.Qt.Vertical)
# Make the cuts plot
twidget = QtHelp.VBox()
vbox1 = twidget.layout()
vbox1.setContentsMargins(4, 4, 4, 4)
vbox1.setSpacing(2)
msgFont = QtGui.QFont("Sans", 14)
tw = QtGui.QLabel()
tw.setFont(msgFont)
tw.setWordWrap(True)
self.tw = tw
fr = QtHelp.Frame("Instructions")
fr.layout().addWidget(tw, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
self.plot = Plot.Cuts(self.logger)
w = self.plot.get_widget()
vbox1.addWidget(w, stretch=1, alignment=QtCore.Qt.AlignTop)
hbox = QtHelp.HBox()
hbox.setSpacing(4)
# control for selecting a cut
combobox = QtHelp.ComboBox()
for tag in self.tags:
combobox.append_text(tag)
if self.cutstag == None:
combobox.setCurrentIndex(0)
else:
combobox.show_text(self.cutstag)
combobox.activated.connect(self.cut_select_cb)
self.w.cuts = combobox
combobox.setToolTip("Select a cut")
hbox.addWidget(combobox)
btn = QtGui.QPushButton("Delete")
btn.clicked.connect(self.delete_cut_cb)
btn.setToolTip("Delete selected cut")
hbox.addWidget(btn)
btn = QtGui.QPushButton("Delete All")
btn.clicked.connect(self.delete_all)
btn.setToolTip("Clear all cuts")
hbox.addWidget(btn)
## btn = QtGui.CheckBox("Move together")
## #btn.stateChanged.connect(self.movetogether_cb)
## btn.setChecked(self.move_together)
## btn.setToolTip("Move cuts as a group")
## hbox.addWidget(btn)
vbox1.addWidget(hbox, stretch=0, alignment=QtCore.Qt.AlignLeft)
btns = QtHelp.HBox()
layout = btns.layout()
layout.setSpacing(3)
#btns.set_child_size(15, -1)
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
layout.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignLeft)
vpaned.addWidget(twidget)
vpaned.addWidget(QtGui.QLabel(''))
container.addWidget(vpaned, stretch=1)
def build_gui(self, container):
sw = QtGui.QScrollArea()
twidget = QtHelp.VBox()
sp = QtGui.QSizePolicy(QtGui.QSizePolicy.MinimumExpanding,
QtGui.QSizePolicy.Fixed)
twidget.setSizePolicy(sp)
vbox1 = twidget.layout()
vbox1.setContentsMargins(4, 4, 4, 4)
vbox1.setSpacing(2)
sw.setWidgetResizable(True)
sw.setWidget(twidget)
msgFont = QtGui.QFont("Sans", 14)
tw = QtGui.QLabel()
tw.setFont(msgFont)
tw.setWordWrap(True)
self.tw = tw
fr = QtHelp.Frame("Instructions")
fr.layout().addWidget(tw, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
fr = QtHelp.Frame("Drawing")
captions = (('Draw type', 'combobox'), ('Draw color', 'combobox'),
('Clear canvas', 'button'))
w, b = QtHelp.build_info(captions)
self.w = b
combobox = b.draw_type
options = []
index = 0
for name in self.drawtypes:
options.append(name)
combobox.addItem(name)
index += 1
index = self.drawtypes.index(default_drawtype)
combobox.setCurrentIndex(index)
combobox.activated.connect(self.set_drawparams)
self.w.draw_color = b.draw_color
combobox = b.draw_color
options = []
index = 0
self.drawcolors = draw_colors
for name in self.drawcolors:
options.append(name)
combobox.addItem(name)
index += 1
index = self.drawcolors.index(default_drawcolor)
combobox.setCurrentIndex(index)
combobox.activated.connect(self.set_drawparams)
b.clear_canvas.clicked.connect(self.clear_canvas)
fr.layout().addWidget(w, stretch=1, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
btns = QtHelp.HBox()
layout = btns.layout()
layout.setSpacing(3)
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
layout.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignLeft)
container.addWidget(sw, stretch=1)
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)
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, text=''):
super(Label, self).__init__()
self.widget = QtGui.QLabel(text)
def __init__(self, logger):
super(FitsViewer, self).__init__()
self.logger = logger
menubar = self.menuBar()
# create a File pulldown menu, and add it to the menu bar
filemenu = menubar.addMenu("File")
item = QtGui.QAction("Open File", menubar)
item.triggered.connect(self.open_file)
filemenu.addAction(item)
sep = QtGui.QAction(menubar)
sep.setSeparator(True)
filemenu.addAction(sep)
item = QtGui.QAction("Quit", menubar)
item.triggered.connect(self.close)
filemenu.addAction(item)
# Add matplotlib color maps to our built in ones
cmap.add_matplotlib_cmaps()
self.cmaps = cmap.get_names()
self.imaps = imap.get_names()
wd, ht = 500, 500
# Create a Ginga widget
fi = ImageViewCanvas(logger, render='widget')
fi.enable_autocuts('on')
fi.set_autocut_params('zscale')
fi.enable_autozoom('on')
fi.enable_draw(False)
fi.set_callback('drag-drop', self.drop_file_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.show_color_bar(True)
# enable various key and mouse controlled actions
bd = fi.get_bindings()
bd.enable_all(True)
self.cp_tag = 'compass'
# pack widget into layout
gingaw = fi.get_widget()
gingaw.resize(wd, ht)
vbox1 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
layout.addWidget(gingaw, stretch=1)
self.cm = cmap.get_cmap('gray')
self.im = imap.get_imap('ramp')
# color map selection widget
wcmap = QtGui.QComboBox()
for name in self.cmaps:
wcmap.addItem(name)
index = self.cmaps.index('gray')
wcmap.setCurrentIndex(index)
wcmap.activated.connect(self.set_cmap_cb)
self.wcmap = wcmap
# intensity map selection widget
wimap = QtGui.QComboBox()
for name in self.imaps:
wimap.addItem(name)
index = self.imaps.index('ramp')
wimap.setCurrentIndex(index)
wimap.activated.connect(self.set_cmap_cb)
self.wimap = wimap
#wopen = QtGui.QPushButton("Open File")
#wopen.clicked.connect(self.open_file)
# add buttons to layout
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
hbox.addStretch(1)
for w in (wcmap, wimap):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox1.setLayout(layout)
# Create a matplotlib Figure
#self.fig = matplotlib.figure.Figure(figsize=(wd, ht))
self.fig = matplotlib.figure.Figure()
self.canvas = FigureCanvas(self.fig)
vbox2 = QtGui.QWidget()
layout = QtGui.QVBoxLayout()
layout.addWidget(self.canvas, stretch=1)
# Add matplotlib buttons
hbox = QtGui.QHBoxLayout()
hbox.setContentsMargins(QtCore.QMargins(4, 2, 4, 2))
wgetimg = QtGui.QPushButton("Get Data")
wgetimg.clicked.connect(self.get_image)
wgetrgb = QtGui.QPushButton("Get RGB")
wgetrgb.clicked.connect(self.get_rgb_image)
#wquit = QtGui.QPushButton("Quit")
#wquit.clicked.connect(self.close)
hbox.addStretch(1)
for w in (wgetimg, wgetrgb):
hbox.addWidget(w, stretch=0)
hw = QtGui.QWidget()
hw.setLayout(hbox)
layout.addWidget(hw, stretch=0)
vbox2.setLayout(layout)
vbox = QtGui.QVBoxLayout()
vbox.setContentsMargins(QtCore.QMargins(2, 2, 2, 2))
vbox.setSpacing(1)
w = QtGui.QWidget()
layout = QtGui.QHBoxLayout()
layout.addWidget(vbox1, stretch=1.0)
layout.addWidget(vbox2, stretch=1.0)
w.setLayout(layout)
vbox.addWidget(w, stretch=1)
self.readout = QtGui.QLabel("")
vbox.addWidget(self.readout,
stretch=0,
alignment=QtCore.Qt.AlignCenter)
vw = QtGui.QWidget()
vw.setLayout(vbox)
self.setCentralWidget(vw)
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 build_gui(self, container, future=None):
vbox1 = QtHelp.VBox()
msgFont = self.fv.getFont("sansFont", 14)
tw = QtGui.QLabel()
tw.setFont(msgFont)
tw.setWordWrap(True)
self.tw = tw
fr = QtHelp.Frame("Instructions")
fr.addWidget(tw, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignTop)
nb = QtHelp.TabWidget()
nb.setTabPosition(QtGui.QTabWidget.South)
nb.setUsesScrollButtons(True)
self.w.nb = nb
#vbox1.addWidget(nb, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox1.addWidget(nb, stretch=1)
vbox0 = QtHelp.VBox()
hbox = QtHelp.HBox()
hbox.setSpacing(4)
vbox0.addWidget(hbox, stretch=1, alignment=QtCore.Qt.AlignTop)
vbox = QtHelp.VBox()
fr = QtHelp.Frame(" Image Server ")
fr.addWidget(vbox, stretch=1, alignment=QtCore.Qt.AlignTop)
hbox.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignLeft)
captions = (('Server', 'xlabel'),
('@Server', 'combobox'),
('Use DSS channel', 'checkbutton'),
('Get Image', 'button'))
w, b = QtHelp.build_info(captions)
self.w.update(b)
self.w.get_image.clicked.connect(self.getimage_cb)
self.w.use_dss_channel.setChecked(self.use_dss_channel)
self.w.use_dss_channel.stateChanged.connect(self.use_dss_channel_cb)
vbox.addWidget(w, stretch=0, alignment=QtCore.Qt.AlignTop)
self.w.img_params = QtHelp.StackedWidget()
vbox.addWidget(self.w.img_params, stretch=1,
alignment=QtCore.Qt.AlignTop)
combobox = self.w.server
index = 0
self.image_server_options = self.fv.imgsrv.getServerNames(kind='image')
for name in self.image_server_options:
combobox.addItem(name)
index += 1
index = 0
combobox.setCurrentIndex(index)
combobox.activated.connect(self.setup_params_image)
if len(self.image_server_options) > 0:
self.setup_params_image(index, redo=False)
vbox = QtHelp.VBox()
fr = QtHelp.Frame(" Catalog Server ")
fr.addWidget(vbox, stretch=1, alignment=QtCore.Qt.AlignTop)
hbox.addWidget(fr, stretch=0, alignment=QtCore.Qt.AlignLeft)
captions = (('Server', 'xlabel'),
('@Server', 'combobox'),
('Limit stars to area', 'checkbutton'),
('Search', 'button'))
w, self.w2 = QtHelp.build_info(captions)
self.w2.search.clicked.connect(self.getcatalog_cb)
self.w2.limit_stars_to_area.setChecked(self.limit_stars_to_area)
self.w2.limit_stars_to_area.stateChanged.connect(self.limit_area_cb)
vbox.addWidget(w, stretch=0, alignment=QtCore.Qt.AlignTop)
self.w2.cat_params = QtHelp.StackedWidget()
vbox.addWidget(self.w2.cat_params, stretch=1,
alignment=QtCore.Qt.AlignTop)
combobox = self.w2.server
index = 0
self.catalog_server_options = self.fv.imgsrv.getServerNames(kind='catalog')
for name in self.catalog_server_options:
combobox.addItem(name)
index += 1
index = 0
combobox.setCurrentIndex(index)
combobox.activated.connect(self.setup_params_catalog)
if len(self.catalog_server_options) > 0:
self.setup_params_catalog(index, redo=False)
btns = QtHelp.HBox()
btns.setSpacing(5)
btn = QtGui.QRadioButton("Rectangle")
if self.drawtype == 'rectangle':
btn.setChecked(True)
btn.toggled.connect(lambda tf: self.set_drawtype_cb(tf, 'rectangle'))
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
btn = QtGui.QRadioButton("Circle")
if self.drawtype == 'circle':
btn.setChecked(True)
btn.toggled.connect(lambda tf: self.set_drawtype_cb(tf, 'circle'))
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
btn = QtGui.QPushButton("Entire image")
btn.clicked.connect(self.setfromimage)
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox0.addWidget(btns, stretch=0, alignment=QtCore.Qt.AlignTop)
self.w.params = vbox0
sw = QtGui.QScrollArea()
sw.setWidgetResizable(True)
sw.setWidget(vbox0)
nb.addTab(sw, "Params")
vbox = QtHelp.VBox()
self.table = CatalogListing(self.logger, vbox)
hbox = QtHelp.HBox()
adj = QtGui.QScrollBar(QtCore.Qt.Horizontal)
adj.setRange(0, 1000)
adj.setSingleStep(1)
adj.setPageStep(10)
#adj.setMaximum(1000)
adj.setValue(0)
#adj.resize(200, -1)
adj.setTracking(True)
adj.setToolTip("Choose subset of stars plotted")
self.w.plotgrp = adj
adj.valueChanged.connect(self.plot_pct_cb)
hbox.addWidget(adj, stretch=1)
sb = QtGui.QSpinBox()
sb.setRange(10, self.plot_max)
sb.setValue(self.plot_limit)
sb.setSingleStep(10)
adj.setPageStep(100)
sb.setWrapping(False)
self.w.plotnum = sb
sb.setToolTip("Adjust size of subset of stars plotted")
sb.valueChanged.connect(self.plot_limit_cb)
hbox.addWidget(sb, stretch=0)
vbox.addWidget(hbox, stretch=0)
self.w.listing = vbox
nb.addTab(vbox, "Listing")
btns = QtHelp.HBox()
btns.setSpacing(3)
#btns.set_child_size(15, -1)
self.w.buttons = btns
btn = QtGui.QPushButton("Close")
btn.clicked.connect(self.close)
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
if future:
btn = QtGui.QPushButton('Ok')
btn.clicked.connect(lambda w: self.ok())
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
btn = QtGui.QPushButton('Cancel')
btn.clicked.connect(lambda w: self.cancel())
btns.addWidget(btn, stretch=0, alignment=QtCore.Qt.AlignLeft)
vbox1.addWidget(btns, stretch=0)
cw = container.get_widget()
cw.addWidget(vbox1, stretch=1)
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)