class ImageWindow(DataViewer):
LABEL = "Image Plot"
def __init__(self, session, parent=None):
super(ImageWindow, self).__init__(session, parent)
self.central_widget = QtGui.QWidget()
self.central_widget.setContentsMargins(0,0,0,0)
self.setCentralWidget(self.central_widget)
# Define layout
self.vb_layout = QtGui.QVBoxLayout()
self.vb_layout.setContentsMargins(0, 0, 0, 0)
self.central_widget.setLayout(self.vb_layout)
self.toolbar = ImageToolBar()
self.toolbar.setContentsMargins(0, 0, 0, 0)
self.vb_layout.addWidget(self.toolbar)
self.viewer = ImageGraph()
self.vb_layout.addWidget(self.viewer)
self._connect_toolbar()
def add_data(self, data):
# data = np.random.sample(size=100)
# spdata = SpectrumData(data, unit=u.Jy)
self.viewer.set_image(data)
return True
def add_subset(self, subset):
return True
def _connect_toolbar(self):
self.toolbar.atn_insert_roi.triggered.connect(self.viewer.add_roi)
def __init__(self, session, parent=None):
super(ImageWindow, self).__init__(session, parent)
self.central_widget = QtGui.QWidget()
self.central_widget.setContentsMargins(0,0,0,0)
self.setCentralWidget(self.central_widget)
# Define layout
self.vb_layout = QtGui.QVBoxLayout()
self.vb_layout.setContentsMargins(0, 0, 0, 0)
self.central_widget.setLayout(self.vb_layout)
self.toolbar = ImageToolBar()
self.toolbar.setContentsMargins(0, 0, 0, 0)
self.vb_layout.addWidget(self.toolbar)
self.viewer = ImageGraph()
self.vb_layout.addWidget(self.viewer)
self._connect_toolbar()
