class imageWidget(QtGui.QWidget):
def __init__(self, imarr, y1=None, y2=None, nrows=1, nsteps=100, hmin=150, wmin=400,
name=None, cmap='Gray', scale='zscale', contrast=0.1, parent=None):
super(imageWidget, self).__init__(parent)
self.y1 = y1
self.y2 = y2
self.x1 = 0
self.x2 = len(imarr[0])
self.nrows = nrows
self.nsteps = nsteps
# Add FITS display widget with mouse interaction and overplotting
self.imdisplay = ImageDisplay()
self.imdisplay.setMinimumHeight(hmin)
self.imdisplay.setMinimumWidth(wmin)
# Set colormap
self.imdisplay.setColormap(cmap)
# Set scale mode for dynamic range
self.imdisplay.scale = scale
self.imdisplay.contrast = contrast
self.imdisplay.aspect = 'auto'
self.imdisplay.loadImage(imarr)
self.imdisplay.drawImage()
self.y1line, = self.imdisplay.axes.plot([self.x1, self.x2], [self.y1, self.y1], ls='-', color='#00FF00')
self.y2line, = self.imdisplay.axes.plot([self.x1, self.x2], [self.y2, self.y2], ls='-', color='#00FF00')
# Add navigation toolbars for each widget to enable zooming
self.toolbar = NavigationToolbar2QTAgg(self.imdisplay, self)
# set up the information panel
self.infopanel = QtGui.QWidget()
# add the name of the file
self.NameLabel = QtGui.QLabel("Filename:")
self.NameLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.NameValueLabel = QtGui.QLabel("%s" % name)
self.NameValueLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Sunken)
# add the rows that are extracted
self.y1Label = QtGui.QLabel("Y1:")
self.y1Label.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.y1ValueEdit = QtGui.QLineEdit("%6i" % self.y1)
self.y2Label = QtGui.QLabel("Y2:")
self.y2Label.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.y2ValueEdit = QtGui.QLineEdit("%6i" % self.y2)
self.updateButton = QtGui.QPushButton("Update")
self.updateButton.clicked.connect(self.updatesection)
# add the update for automatically updating it
self.nrLabel = QtGui.QLabel("nrows:")
self.nrLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.nrValueEdit = QtGui.QLineEdit("%5i" % self.nrows)
self.nsLabel = QtGui.QLabel("nsteps:")
self.nsLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.nsValueEdit = QtGui.QLineEdit("%6i" % self.nsteps)
self.nextButton = QtGui.QPushButton("Next")
self.nextButton.clicked.connect(self.nextsection)
# set up the info panel layout
infoLayout = QtGui.QGridLayout(self.infopanel)
infoLayout.addWidget(self.NameLabel, 0, 0, 1, 1)
infoLayout.addWidget(self.NameValueLabel, 0, 1, 1, 5)
infoLayout.addWidget(self.y1Label, 1, 0, 1, 1)
infoLayout.addWidget(self.y1ValueEdit, 1, 1, 1, 1)
infoLayout.addWidget(self.y2Label, 1, 2, 1, 1)
infoLayout.addWidget(self.y2ValueEdit, 1, 3, 1, 1)
infoLayout.addWidget(self.updateButton, 1, 4, 1, 1)
infoLayout.addWidget(self.nrLabel, 2, 0, 1, 1)
infoLayout.addWidget(self.nrValueEdit, 2, 1, 1, 1)
infoLayout.addWidget(self.nsLabel, 2, 2, 1, 1)
infoLayout.addWidget(self.nsValueEdit, 2, 3, 1, 1)
infoLayout.addWidget(self.nextButton, 2, 4, 1, 1)
# Set up the layout
mainLayout = QtGui.QVBoxLayout()
mainLayout.addWidget(self.imdisplay)
mainLayout.addWidget(self.toolbar)
mainLayout.addWidget(self.infopanel)
self.setLayout(mainLayout)
def updatesection(self):
self.y1 = int(self.y1ValueEdit.text())
self.y2 = int(self.y2ValueEdit.text())
self.y1line.set_ydata([self.y1, self.y1])
self.y2line.set_ydata([self.y2, self.y2])
self.imdisplay.draw()
self.emit(QtCore.SIGNAL("regionChange(int,int)"), self.y1, self.y2)
def nextsection(self):
self.nrows = int(self.nrValueEdit.text())
self.nsteps = int(self.nsValueEdit.text())
self.y1 = self.y1 + self.nsteps
self.y2 = self.y1 + self.nrows
self.y1ValueEdit.setText('%6i' % self.y1)
self.y2ValueEdit.setText('%6i' % self.y2)
self.updatesection()
class imageWidget(QtGui.QWidget):
def __init__(self,
imarr,
y1=None,
y2=None,
nrows=1,
nsteps=100,
hmin=150,
wmin=400,
name=None,
cmap='Gray',
scale='zscale',
contrast=0.1,
parent=None):
super(imageWidget, self).__init__(parent)
self.y1 = y1
self.y2 = y2
self.x1 = 0
self.x2 = len(imarr[0])
self.nrows = nrows
self.nsteps = nsteps
# Add FITS display widget with mouse interaction and overplotting
self.imdisplay = ImageDisplay()
self.imdisplay.setMinimumHeight(hmin)
self.imdisplay.setMinimumWidth(wmin)
# Set colormap
self.imdisplay.setColormap(cmap)
# Set scale mode for dynamic range
self.imdisplay.scale = scale
self.imdisplay.contrast = contrast
self.imdisplay.aspect = 'auto'
self.imdisplay.loadImage(imarr)
self.imdisplay.drawImage()
self.y1line, = self.imdisplay.axes.plot([self.x1, self.x2],
[self.y1, self.y1],
ls='-',
color='#00FF00')
self.y2line, = self.imdisplay.axes.plot([self.x1, self.x2],
[self.y2, self.y2],
ls='-',
color='#00FF00')
# Add navigation toolbars for each widget to enable zooming
self.toolbar = NavigationToolbar2QTAgg(self.imdisplay, self)
# set up the information panel
self.infopanel = QtGui.QWidget()
# add the name of the file
self.NameLabel = QtGui.QLabel("Filename:")
self.NameLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.NameValueLabel = QtGui.QLabel("%s" % name)
self.NameValueLabel.setFrameStyle(QtGui.QFrame.Panel
| QtGui.QFrame.Sunken)
# add the rows that are extracted
self.y1Label = QtGui.QLabel("Y1:")
self.y1Label.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.y1ValueEdit = QtGui.QLineEdit("%6i" % self.y1)
self.y2Label = QtGui.QLabel("Y2:")
self.y2Label.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.y2ValueEdit = QtGui.QLineEdit("%6i" % self.y2)
self.updateButton = QtGui.QPushButton("Update")
self.updateButton.clicked.connect(self.updatesection)
# add the update for automatically updating it
self.nrLabel = QtGui.QLabel("nrows:")
self.nrLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.nrValueEdit = QtGui.QLineEdit("%5i" % self.nrows)
self.nsLabel = QtGui.QLabel("nsteps:")
self.nsLabel.setFrameStyle(QtGui.QFrame.Panel | QtGui.QFrame.Raised)
self.nsValueEdit = QtGui.QLineEdit("%6i" % self.nsteps)
self.nextButton = QtGui.QPushButton("Next")
self.nextButton.clicked.connect(self.nextsection)
# set up the info panel layout
infoLayout = QtGui.QGridLayout(self.infopanel)
infoLayout.addWidget(self.NameLabel, 0, 0, 1, 1)
infoLayout.addWidget(self.NameValueLabel, 0, 1, 1, 5)
infoLayout.addWidget(self.y1Label, 1, 0, 1, 1)
infoLayout.addWidget(self.y1ValueEdit, 1, 1, 1, 1)
infoLayout.addWidget(self.y2Label, 1, 2, 1, 1)
infoLayout.addWidget(self.y2ValueEdit, 1, 3, 1, 1)
infoLayout.addWidget(self.updateButton, 1, 4, 1, 1)
infoLayout.addWidget(self.nrLabel, 2, 0, 1, 1)
infoLayout.addWidget(self.nrValueEdit, 2, 1, 1, 1)
infoLayout.addWidget(self.nsLabel, 2, 2, 1, 1)
infoLayout.addWidget(self.nsValueEdit, 2, 3, 1, 1)
infoLayout.addWidget(self.nextButton, 2, 4, 1, 1)
# Set up the layout
mainLayout = QtGui.QVBoxLayout()
mainLayout.addWidget(self.imdisplay)
mainLayout.addWidget(self.toolbar)
mainLayout.addWidget(self.infopanel)
self.setLayout(mainLayout)
def updatesection(self):
self.y1 = int(self.y1ValueEdit.text())
self.y2 = int(self.y2ValueEdit.text())
self.y1line.set_ydata([self.y1, self.y1])
self.y2line.set_ydata([self.y2, self.y2])
self.imdisplay.draw()
self.emit(QtCore.SIGNAL("regionChange(int,int)"), self.y1, self.y2)
def nextsection(self):
self.nrows = int(self.nrValueEdit.text())
self.nsteps = int(self.nsValueEdit.text())
self.y1 = self.y1 + self.nsteps
self.y2 = self.y1 + self.nrows
self.y1ValueEdit.setText('%6i' % self.y1)
self.y2ValueEdit.setText('%6i' % self.y2)
self.updatesection()
