class PlotsMath(QWidget): F_X = Qt.UserRole + 1 FITS_SPECTRUM = Qt.UserRole + 2 def __init__(self, settings, database, project=None): super(PlotsMath, self).__init__() self.ui = Ui_PlotsMath() self.ui.setupUi(self) self.settings = settings self.project=project self.plot = QtCommons.nestWidget(self.ui.plot, QMathPlotWidget()) self.reference_dialog = ReferenceSpectraDialog(database) self.reference_dialog.fits_picked.connect(self.open_fits) self.toolbar = QToolBar('Instrument Response Toolbar') open_btn = QtCommons.addToolbarPopup(self.toolbar, text="Open...", icon_file=':/new_open_20') open_file_action = open_btn.menu().addAction('FITS file') open_btn.menu().addAction('Reference library', self.reference_dialog.show) self.blackbody_menu = blackbody.BlackBodyAction(self.blackbody, open_btn.menu()) if project: save_result = QtCommons.addToolbarPopup(self.toolbar, text='Save', icon_file=':/save_20') save_result.menu().addAction('As File', lambda: QtCommons.save_file('Save Operation Result...', FITS_EXTS, lambda f: self.save(f[0]), project.path)) save_result.menu().addAction('As Instrument Response', self.save_project_instrument_response) open_file_action.triggered.connect(lambda: QtCommons.open_file('Open FITS Spectrum',FITS_EXTS, lambda f: self.open_fits(f[0]), project.path)) else: open_file_action.triggered.connect(lambda: open_file_sticky('Open FITS Spectrum',FITS_EXTS, lambda f: self.open_fits(f[0]), self.settings, CALIBRATED_PROFILE, [RAW_PROFILE])) self.toolbar.addAction(QIcon(':/save_20'), 'Save', lambda: save_file_sticky('Save Operation Result...', 'FITS file (.fit)', lambda f: self.save(f[0]), self.settings, MATH_OPERATION, [CALIBRATED_PROFILE])) self.toolbar.addAction('Set operand', self.set_operand) self.toolbar.addSeparator() self.toolbar.addAction(self.ui.actionZoom) self.ui.actionZoom.triggered.connect(self.start_zoom) self.toolbar.addAction(self.ui.actionReset_Zoom) self.ui.actionReset_Zoom.triggered.connect(self.reset_zoom) self.toolbar.addSeparator() self.operands_model = QStandardItemModel() self.ui.operands_listview.setModel(self.operands_model) remove_btn = QtCommons.addToolbarPopup(self.toolbar, text='Remove...') remove_btn.menu().addAction(self.ui.actionSelectPointsToRemove) remove_btn.menu().addAction("Before point", lambda: spectrum_trim_dialog(self.spectrum, 'before', self.plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) remove_btn.menu().addAction("After point", lambda: spectrum_trim_dialog(self.spectrum, 'after', self.plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) self.ui.clear_operands.clicked.connect(self.operands_model.clear) self.ui.remove_operand.clicked.connect(lambda: self.operands_model.removeRows(self.ui.operands_listview.selectionModel().selectedRows()[0].row(), 1)) self.operands_model.rowsInserted.connect(lambda: self.ui.clear_operands.setEnabled(self.operands_model.rowCount() > 0) ) self.operands_model.rowsRemoved.connect(lambda: self.ui.clear_operands.setEnabled(self.operands_model.rowCount() > 0) ) self.ui.operands_listview.selectionModel().selectionChanged.connect(lambda s, u: self.ui.remove_operand.setEnabled(len(s))) self.ui.actionSelectPointsToRemove.triggered.connect(self.pick_rm_points) self.undo = Undo(None, self.draw) self.undo.add_actions(self.toolbar) self.ui.spline_factor.valueChanged.connect(self.factor_valueChanged) self.ui.spline_degrees.valueChanged.connect(lambda v: self.draw()) self.ui.spline_factor_auto.toggled.connect(lambda v: self.draw()) self.ui.spline_factor_auto.toggled.connect(lambda v: self.ui.spline_factor.setEnabled(not v)) self.ui.execute.clicked.connect(self.execute_operation) self.plot.figure.tight_layout() def blackbody(self, blackbody): self.spectrum = blackbody.spectrum() self.spectrum_name = "Blackbody radiation for {0}".format(blackbody.kelvin) self.undo.set_spectrum(self.spectrum) self.spectrum.normalize_to_max() self.draw() def open_fits(self, filename): fits_file = fits.open(filename) fits_spectrum = FitsSpectrum(fits_file) self.spectrum_name = fits_spectrum.name() self.spectrum = fits_spectrum.spectrum self.undo.set_spectrum(self.spectrum) self.spectrum.normalize_to_max() if self.spectrum.dispersion() <0.4: print("dispersion too high ({}), reducing spectrum resolution".format(self.spectrum.dispersion())) self.spectrum.resample(self.spectrum.dispersion() / 0.4) self.draw() @pyqtSlot(float) def factor_valueChanged(self, f): self.draw() def pick_rm_points(self): self.plot.rm_element('zoom') self.plot.add_span_selector('pick_rm_points', lambda min,max: self.rm_points(min,max+1),direction='horizontal') def start_zoom(self): self.plot.rm_element('pick_rm_points') self.plot.select_zoom() def draw(self): self.ui.spline_degrees_value.setText("{}".format(self.ui.spline_degrees.value())) spline_factor = self.ui.spline_factor.value() if not self.ui.spline_factor_auto.isChecked() else None spline = UnivariateSpline(self.spectrum.wavelengths, self.spectrum.fluxes, k=self.ui.spline_degrees.value(), s=spline_factor) self.f_x = lambda x: spline(x) self.plot.plot(None, self.spectrum.wavelengths, self.spectrum.fluxes, '--', self.spectrum.wavelengths, spline(self.spectrum.wavelengths), '-') self.plot.figure.tight_layout() self.plot.figure.canvas.draw() def rm_points(self, wmin, wmax): self.undo.save_undo() x_min = self.spectrum.wavelength_index(max(self.spectrum.wavelengths[0], wmin)) x_max = self.spectrum.wavelength_index(min(self.spectrum.wavelengths[-1], wmax)) m=(self.spectrum.fluxes[x_max]-self.spectrum.fluxes[x_min])/(x_max-x_min) q = self.spectrum.fluxes[x_min] f = lambda x: x * m + q self.spectrum.fluxes[x_min:x_max] = np.fromfunction(f, self.spectrum.fluxes[x_min:x_max].shape) self.draw() def trim(self, direction): point = QInputDialog.getInt(None, 'Trim curve', 'Enter wavelength for trimming', self.spectrum.wavelengths[0] if direction == 'before' else self.spectrum.wavelengths[-1], self.spectrum.wavelengths[0], self.spectrum.wavelengths[-1]) if not point[1]: return self.undo.save_undo() if direction == 'before': self.spectrum.cut(start=self.spectrum.wavelength_index(point[0])) else: self.spectrum.cut(end=self.spectrum.wavelength_index(point[0])) self.reset_zoom() self.draw() def set_operand(self): item = QStandardItem(self.spectrum_name) item.setData(self.f_x, PlotsMath.F_X) item.setData(self.spectrum, PlotsMath.FITS_SPECTRUM) self.operands_model.appendRow(item) def execute_operation(self): max_wavelengths = lambda operands: np.arange(max([o[0].wavelengths[0] for o in operands]), min([o[0].wavelengths[-1] for o in operands])) datasets = lambda operands, wavelengths: [PlotsMath.__data(o[1], wavelengths) for o in operands] operands = [(self.operands_model.item(a).data(PlotsMath.FITS_SPECTRUM), self.operands_model.item(a).data(PlotsMath.F_X)) for a in np.arange(self.operands_model.rowCount())] def divide(operands): if len(operands) > 2: print("Division supports only 2 operands, truncating") wavelengths = max_wavelengths(operands[0:2]) datas = datasets(operands[0:2], wavelengths) return (wavelengths, datas[0]/datas[1]) def mean(operands): wavelengths = max_wavelengths(operands) mean_data = np.zeros(wavelengths.shape) for data in datasets(operands, wavelengths): mean_data += data return (wavelengths, mean_data/len(wavelengths)) operations = { 0: divide, 1: mean } try: wavelengths, data = operations[self.ui.operation_type.currentIndex()](operands) self.spectrum = Spectrum(data, wavelengths) self.spectrum.normalize_to_max() self.undo.set_spectrum(self.spectrum) self.ui.spline_degrees.setValue(5) self.ui.spline_factor.setValue(0) self.ui.spline_factor_auto.setChecked(False) self.draw() except IndexError: QMessageBox.warning(None, "Error", "Datasets are not compatible. Maybe you need to calibrate better, or use a different reference file") def save_project_instrument_response(self): name = QInputDialog.getText(self, 'Enter Name', 'Enter new instrument response name for saving') if name[1]: self.project.add_file(Project.INSTRUMENT_RESPONSES, lambda f: self.save(f), bare_name=name[0]) def save(self, filename): hdu = fits.PrimaryHDU( PlotsMath.__data(self.f_x, self.spectrum.wavelengths)) #hdu = fits.PrimaryHDU( self.spectrum.fluxes) fits_file = fits.HDUList([hdu]) hdu.header['CRPIX1'] = 1 hdu.header['CRVAL1'] = self.spectrum.wavelengths[0] hdu.header['CDELT1'] = self.spectrum.dispersion() hdu.writeto(filename, clobber=True) def reset_zoom(self): self.plot.reset_zoom(self.spectrum.wavelengths, self.spectrum.fluxes.min(), self.spectrum.fluxes.max()) def __data(f_x, xrange): return np.fromfunction(lambda x: f_x(x+xrange[0]), xrange.shape)
class FinishSpectrum(QWidget): def __init__(self, fits_file, settings, database, project=None): super(FinishSpectrum, self).__init__() self.settings = settings self.ui = Ui_FinishSpectrum() self.ui.setupUi(self) self.profile_line = None self.project = project self.fits_spectrum = FitsSpectrum(fits_file) self.undo = Undo(self.fits_spectrum.spectrum, self.draw) try: fits_file.index_of('ORIGINAL_DATA') except KeyError: hdu = fits.ImageHDU(data = fits_file[0].data, header = fits_file[0].header, name='ORIGINAL_DATA') fits_file.append(hdu) self.fits_spectrum.spectrum.normalize_to_max() self.spectrum = self.fits_spectrum.spectrum self.spectrum_plot = QtCommons.nestWidget(self.ui.plot, QMathPlotWidget()) self.spectrum_plot.mouse_moved.connect(Instances.MainWindow.print_coordinates) self.split_view() self.toolbar = QToolBar('Finish Spectrum Toolbar') if project: instrument_response_action = QtCommons.addToolbarPopup(self.toolbar, "Instrument Response") instrument_response_action.menu().addAction('From FITS file...', lambda: open_file_sticky('Open Instrument Response Profile', FITS_EXTS, lambda f: self.instrument_response(f[0]), settings, MATH_OPERATION, [RAW_PROFILE])) for instrument_response in project.get_instrument_responses(): print("Adding instrument response {}".format(instrument_response)) instrument_response_action.menu().addAction(os.path.basename(instrument_response[1]), lambda: self.instrument_response(instrument_response[1])) else: self.toolbar.addAction('Instrument Response', lambda: open_file_sticky('Open Instrument Response Profile', FITS_EXTS, lambda f: self.instrument_response(f[0]), settings, MATH_OPERATION, [RAW_PROFILE])) self.toolbar.addAction("Zoom", lambda: self.spectrum_plot.select_zoom(self.profile_plot.axes)) self.toolbar.addAction("Reset Zoom", lambda: self.spectrum_plot.reset_zoom(self.spectrum.wavelengths, self.spectrum.fluxes.min(), self.spectrum.fluxes.max(), self.profile_plot.axes)) remove_action = QtCommons.addToolbarPopup(self.toolbar, "Remove") remove_action.menu().addAction("Before point", lambda: spectrum_trim_dialog(self.spectrum, 'before', self.profile_plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) remove_action.menu().addAction("After point", lambda: spectrum_trim_dialog(self.spectrum, 'after', self.profile_plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) self.undo.add_actions(self.toolbar) self.toolbar.addSeparator() self.reference_spectra_dialog = ReferenceSpectraDialog(database, self.fits_spectrum.spectrum) self.reference_spectra_dialog.setup_menu(self.toolbar, self.profile_plot.axes, settings) lines_menu = QtCommons.addToolbarPopup(self.toolbar, "Spectral Lines..") lines_menu.menu().addAction('Lines Database', lambda: self.lines_dialog.show()) lines_menu.menu().addAction('Custom line', self.add_custom_line) labels_action = QtCommons.addToolbarPopup(self.toolbar, "Labels..") self.object_properties = ObjectProperties(fits_file, project=project) labels_action.menu().addAction('Title', self.add_title) if self.object_properties: labels_action.menu().addAction('Information from FITS file', self.add_fits_information_label) labels_action.menu().addAction('Custom', self.add_label) self.object_properties_dialog = ObjectPropertiesDialog(settings, self.object_properties) self.toolbar.addAction("Object properties", self.object_properties_dialog.show) self.labels, self.lines = [], [] for label in self.fits_spectrum.labels(): self.add_label(text=label['text'], coords=label['coords'], type=label['type'], fontsize=label['fontsize']) self.toolbar.addSeparator() if project: self.toolbar.addAction(QIcon(':/image_20'), "Export Image...", lambda: QtCommons.save_file('Export plot to image', 'PNG (*.png);;PDF (*.pdf);;PostScript (*.ps);;SVG (*.svg)', lambda f: self.save_image(f[0]), project.directory_path(Project.EXPORTED_IMAGES))) self.toolbar.addAction(QIcon(':/save_20'), 'Save', self.save_finished_in_project) else: self.toolbar.addAction(QIcon(':/image_20'), "Export Image...", lambda: save_file_sticky('Export plot to image', 'PNG (*.png);;PDF (*.pdf);;PostScript (*.ps);;SVG (*.svg)', lambda f: self.save_image(f[0]), self.settings, EXPORT_IMAGES, [CALIBRATED_PROFILE])) self.toolbar.addAction(QIcon(':/save_20'), 'Save', lambda: save_file_sticky('Save plot...', 'FITS file (.fit)', lambda f: self.__save(f[0]), self.settings, CALIBRATED_PROFILE)) self.lines_dialog = LinesDialog(database, settings, self.spectrum_plot, self.profile_plot.axes) self.lines_dialog.lines.connect(self.add_lines) for line in self.fits_spectrum.lines_labels(): self.lines.append(ReferenceLine(line['text'], line['wavelength'], self.profile_plot.axes, lambda line: self.lines.remove(line), show_wavelength=line['display_wavelength'], fontsize=line['fontsize'], position=line['position'])) def add_custom_line(self): wl = QInputDialog.getDouble(self, "Custom Line", "Enter line wavelength in Å", self.fits_spectrum.spectrum.wavelengths[0],self.fits_spectrum.spectrum.wavelengths[0],self.fits_spectrum.spectrum.wavelengths[-1],3) if not wl[1]: return self.add_lines([{'name': 'Custom Line', 'lambda': wl[0]}]) def add_lines(self, lines): for line in lines: self.lines.append(ReferenceLine(line['name'], line['lambda'], self.profile_plot.axes, lambda line: self.lines.remove(line))) def synthetize_img(wavelengths, fluxes): f_fluxes = lambda f: math.pow(f, 3/5) colors = [wavelength_to_rgb(w/10., f_fluxes(fluxes[i])) for i,w in enumerate(wavelengths)] im_height = 150 colors = np.array(colors*im_height).reshape(im_height,len(colors),4) return colors, im_height def split_view(self): figure = self.spectrum_plot.figure figure.clear() self.gs = gridspec.GridSpec(40,1) self.profile_plot = figure.add_subplot(self.gs[0:-6]) self.synthetize = figure.add_subplot(self.gs[-3:-1], sharex = self.profile_plot) self.synthetize.yaxis.set_visible(False) self.synthetize.xaxis.set_visible(False) self.draw() def draw(self): # self.profile_plot.clear() if self.profile_line: self.profile_line.remove() self.profile_line = self.profile_plot.plot(self.spectrum.wavelengths, self.spectrum.fluxes, color='blue')[0] self.synthetize.axes.set_axis_bgcolor('black') with ThreadPoolExecutor(max_workers=1) as executor: future = executor.submit(FinishSpectrum.synthetize_img, self.spectrum.wavelengths, self.spectrum.fluxes) future.add_done_callback(lambda f: self.synthetize.imshow(f.result()[0], extent=[self.spectrum.wavelengths[0], self.spectrum.wavelengths[-1], 0, f.result()[1]]) ) self.profile_plot.axes.set_xlabel('wavelength (Å)') self.profile_plot.axes.set_ylabel('relative flux') self.profile_plot.axes.xaxis.set_major_locator(MaxNLocator(16)) # TODO: settings for customization? self.profile_plot.axes.xaxis.set_minor_locator(MaxNLocator(200)) self.spectrum_plot.figure.canvas.draw() self.gs.tight_layout(self.spectrum_plot.figure) def instrument_response(self, filename): print("Applying instrument response {}".format(filename)) instrument_response_file = fits.open(filename) instrument_response = FitsSpectrum(instrument_response_file) response = instrument_response.spectrum response.normalize_to_max() range = (max(response.wavelengths[0], self.spectrum.wavelengths[0] ), min(response.wavelengths[-1], self.spectrum.wavelengths[-1])) self.spectrum.cut(self.spectrum.wavelength_index(range[0]), self.spectrum.wavelength_index(range[1])) spline = InterpolatedUnivariateSpline(response.wavelengths, response.fluxes) response_data = [spline(x) for x in self.spectrum.wavelengths] self.spectrum.fluxes /= response_data self.spectrum.normalize_to_max() self.draw() def save_image(self, filename): Notification('Image {} saved in {}'.format(os.path.basename(filename), os.path.dirname(filename)), title='File Saved', type='success', timeout=5) self.spectrum_plot.figure.savefig(filename, bbox_inches='tight', dpi=300) def save_finished_in_project(self): self.project.add_file(Project.FINISHED_PROFILES, self.__save, self.object_properties) def __save(self, filename): self.fits_spectrum.save(filename, spectral_lines = self.lines, labels = self.labels) def add_title(self): title = self.object_properties.name if self.object_properties else 'Title - double click to edit' self.add_label(text=title, coords=(self.spectrum.wavelengths[len(self.spectrum.wavelengths)/2-100], 0.95), fontsize=25, type='lineedit') def add_fits_information_label(self): info_text = "Object Name: {}, type: {}, spectral class: {}\nCoordinates: {}\nDate: {}\nObserver: {}\nEquipment: {}\nPosition: {}".format( self.object_properties.name, self.object_properties.type, self.object_properties.sptype, self.object_properties.printable_coordinates(), self.object_properties.date.toString(), self.object_properties.observer, self.object_properties.equipment, self.object_properties.position ) self.add_label(info_text, type='textbox', coords=(self.spectrum.wavelengths[len(self.spectrum.wavelengths)/4*3], 0.80), fontsize=14) self.profile_plot.figure.canvas.draw() def add_label(self, text=None, type='textbox', coords = None, fontsize = 12, color='black'): if not coords: coords = (self.spectrum.wavelengths[len(self.spectrum.wavelengths)/2], 0.5) self.labels.append((type, MoveableLabel(text=text if text else 'Label - double click to edit', on_dblclick=lambda l: self.edit_label(l, type=type), x=coords[0], y=coords[1], fontsize=fontsize, color=color, axes=self.profile_plot.axes))) self.profile_plot.figure.canvas.draw() def edit_label(self, label, type='lineedit'): def remove_label(self, label, dialog): label.remove() self.labels.remove([l for l in self.labels if l[1] == label][0]) self.profile_plot.figure.canvas.draw() dialog.reject() dialog = QDialog() dialog.setWindowTitle("Edit Label") dialog.setLayout(QVBoxLayout()) font_size = QSpinBox() font_size.setValue(label.get_fontsize()) dialog.layout().addWidget(QLabel("Font Size")) dialog.layout().addWidget(font_size) text_edit = None if type == 'lineedit': text_edit = QLineEdit(label.get_text()) else: text_edit = QTextEdit() text_edit.setPlainText(label.get_text()) dialog.layout().addWidget(QLabel("Text")) dialog.layout().addWidget(text_edit) button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel) button_box.accepted.connect(dialog.accept) button_box.rejected.connect(dialog.reject) remove_button = QPushButton('Remove') remove_button.clicked.connect(lambda: remove_label(self, label, dialog)) dialog.layout().addWidget(remove_button) dialog.layout().addWidget(button_box) if QDialog.Accepted != dialog.exec(): return label.set_text(text_edit.text() if type=='lineedit' else text_edit.toPlainText()) label.set_fontsize(font_size.value()) label.axes.figure.canvas.draw()
class FinishSpectrum(QWidget): def __init__(self, fits_file, settings, database, project=None): super(FinishSpectrum, self).__init__() self.settings = settings self.ui = Ui_FinishSpectrum() self.ui.setupUi(self) self.profile_line = None self.project = project self.fits_spectrum = FitsSpectrum(fits_file) self.undo = Undo(self.fits_spectrum.spectrum, self.draw) try: fits_file.index_of('ORIGINAL_DATA') except KeyError: hdu = fits.ImageHDU(data=fits_file[0].data, header=fits_file[0].header, name='ORIGINAL_DATA') fits_file.append(hdu) self.fits_spectrum.spectrum.normalize_to_max() self.spectrum = self.fits_spectrum.spectrum self.spectrum_plot = QtCommons.nestWidget(self.ui.plot, QMathPlotWidget()) self.spectrum_plot.mouse_moved.connect( Instances.MainWindow.print_coordinates) self.split_view() self.toolbar = QToolBar('Finish Spectrum Toolbar') if project: instrument_response_action = QtCommons.addToolbarPopup( self.toolbar, "Instrument Response") instrument_response_action.menu().addAction( 'From FITS file...', lambda: open_file_sticky( 'Open Instrument Response Profile', FITS_EXTS, lambda f: self.instrument_response(f[ 0]), settings, MATH_OPERATION, [RAW_PROFILE])) for instrument_response in project.get_instrument_responses(): print("Adding instrument response {}".format( instrument_response)) instrument_response_action.menu().addAction( os.path.basename(instrument_response[1]), lambda: self.instrument_response(instrument_response[1])) else: self.toolbar.addAction( 'Instrument Response', lambda: open_file_sticky( 'Open Instrument Response Profile', FITS_EXTS, lambda f: self.instrument_response(f[ 0]), settings, MATH_OPERATION, [RAW_PROFILE])) self.toolbar.addAction( "Zoom", lambda: self.spectrum_plot.select_zoom(self.profile_plot.axes)) self.toolbar.addAction( "Reset Zoom", lambda: self.spectrum_plot.reset_zoom( self.spectrum.wavelengths, self.spectrum.fluxes.min(), self.spectrum.fluxes.max(), self.profile_plot.axes)) remove_action = QtCommons.addToolbarPopup(self.toolbar, "Remove") remove_action.menu().addAction( "Before point", lambda: spectrum_trim_dialog(self.spectrum, 'before', self.profile_plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) remove_action.menu().addAction( "After point", lambda: spectrum_trim_dialog(self.spectrum, 'after', self.profile_plot.axes, lambda: self.draw(), self, before_removal=self.undo.save_undo)) self.undo.add_actions(self.toolbar) self.toolbar.addSeparator() self.reference_spectra_dialog = ReferenceSpectraDialog( database, self.fits_spectrum.spectrum) self.reference_spectra_dialog.setup_menu(self.toolbar, self.profile_plot.axes, settings) lines_menu = QtCommons.addToolbarPopup(self.toolbar, "Spectral Lines..") lines_menu.menu().addAction('Lines Database', lambda: self.lines_dialog.show()) lines_menu.menu().addAction('Custom line', self.add_custom_line) labels_action = QtCommons.addToolbarPopup(self.toolbar, "Labels..") self.object_properties = ObjectProperties(fits_file, project=project) labels_action.menu().addAction('Title', self.add_title) if self.object_properties: labels_action.menu().addAction('Information from FITS file', self.add_fits_information_label) labels_action.menu().addAction('Custom', self.add_label) self.object_properties_dialog = ObjectPropertiesDialog( settings, self.object_properties) self.toolbar.addAction("Object properties", self.object_properties_dialog.show) self.labels, self.lines = [], [] for label in self.fits_spectrum.labels(): self.add_label(text=label['text'], coords=label['coords'], type=label['type'], fontsize=label['fontsize']) self.toolbar.addSeparator() if project: self.toolbar.addAction( QIcon(':/image_20'), "Export Image...", lambda: QtCommons.save_file( 'Export plot to image', 'PNG (*.png);;PDF (*.pdf);;PostScript (*.ps);;SVG (*.svg)', lambda f: self.save_image(f[0]), project.directory_path(Project.EXPORTED_IMAGES))) self.toolbar.addAction(QIcon(':/save_20'), 'Save', self.save_finished_in_project) else: self.toolbar.addAction( QIcon(':/image_20'), "Export Image...", lambda: save_file_sticky( 'Export plot to image', 'PNG (*.png);;PDF (*.pdf);;PostScript (*.ps);;SVG (*.svg)', lambda f: self.save_image(f[0]), self.settings, EXPORT_IMAGES, [CALIBRATED_PROFILE])) self.toolbar.addAction( QIcon(':/save_20'), 'Save', lambda: save_file_sticky( 'Save plot...', 'FITS file (.fit)', lambda f: self.__save( f[0]), self.settings, CALIBRATED_PROFILE)) self.lines_dialog = LinesDialog(database, settings, self.spectrum_plot, self.profile_plot.axes) self.lines_dialog.lines.connect(self.add_lines) for line in self.fits_spectrum.lines_labels(): self.lines.append( ReferenceLine(line['text'], line['wavelength'], self.profile_plot.axes, lambda line: self.lines.remove(line), show_wavelength=line['display_wavelength'], fontsize=line['fontsize'], position=line['position'])) def add_custom_line(self): wl = QInputDialog.getDouble( self, "Custom Line", "Enter line wavelength in Å", self.fits_spectrum.spectrum.wavelengths[0], self.fits_spectrum.spectrum.wavelengths[0], self.fits_spectrum.spectrum.wavelengths[-1], 3) if not wl[1]: return self.add_lines([{'name': 'Custom Line', 'lambda': wl[0]}]) def add_lines(self, lines): for line in lines: self.lines.append( ReferenceLine(line['name'], line['lambda'], self.profile_plot.axes, lambda line: self.lines.remove(line))) def synthetize_img(wavelengths, fluxes): f_fluxes = lambda f: math.pow(f, 3 / 5) colors = [ wavelength_to_rgb(w / 10., f_fluxes(fluxes[i])) for i, w in enumerate(wavelengths) ] im_height = 150 colors = np.array(colors * im_height).reshape(im_height, len(colors), 4) return colors, im_height def split_view(self): figure = self.spectrum_plot.figure figure.clear() self.gs = gridspec.GridSpec(40, 1) self.profile_plot = figure.add_subplot(self.gs[0:-6]) self.synthetize = figure.add_subplot(self.gs[-3:-1], sharex=self.profile_plot) self.synthetize.yaxis.set_visible(False) self.synthetize.xaxis.set_visible(False) self.draw() def draw(self): # self.profile_plot.clear() if self.profile_line: self.profile_line.remove() self.profile_line = self.profile_plot.plot(self.spectrum.wavelengths, self.spectrum.fluxes, color='blue')[0] self.synthetize.axes.set_facecolor('black') with ThreadPoolExecutor(max_workers=1) as executor: future = executor.submit(FinishSpectrum.synthetize_img, self.spectrum.wavelengths, self.spectrum.fluxes) future.add_done_callback(lambda f: self.synthetize.imshow( f.result()[0], extent=[ self.spectrum.wavelengths[0], self.spectrum.wavelengths[ -1], 0, f.result()[1] ])) self.profile_plot.axes.set_xlabel('wavelength (Å)') self.profile_plot.axes.set_ylabel('relative flux') self.profile_plot.axes.xaxis.set_major_locator( MaxNLocator(16)) # TODO: settings for customization? self.profile_plot.axes.xaxis.set_minor_locator(MaxNLocator(200)) self.spectrum_plot.figure.canvas.draw() self.gs.tight_layout(self.spectrum_plot.figure) def instrument_response(self, filename): print("Applying instrument response {}".format(filename)) instrument_response_file = fits.open(filename) instrument_response = FitsSpectrum(instrument_response_file) response = instrument_response.spectrum response.normalize_to_max() range = (max(response.wavelengths[0], self.spectrum.wavelengths[0]), min(response.wavelengths[-1], self.spectrum.wavelengths[-1])) self.spectrum.cut(self.spectrum.wavelength_index(range[0]), self.spectrum.wavelength_index(range[1])) spline = InterpolatedUnivariateSpline(response.wavelengths, response.fluxes) response_data = [spline(x) for x in self.spectrum.wavelengths] self.spectrum.fluxes /= response_data self.spectrum.normalize_to_max() self.draw() def save_image(self, filename): Notification('Image {} saved in {}'.format(os.path.basename(filename), os.path.dirname(filename)), title='File Saved', type='success', timeout=5) self.spectrum_plot.figure.savefig(filename, bbox_inches='tight', dpi=300) def save_finished_in_project(self): self.project.add_file(Project.FINISHED_PROFILES, self.__save, self.object_properties) def __save(self, filename): self.fits_spectrum.save(filename, spectral_lines=self.lines, labels=self.labels) def add_title(self): title = self.object_properties.name if self.object_properties else 'Title - double click to edit' self.add_label( text=title, coords=( self.spectrum.wavelengths[len(self.spectrum.wavelengths) / 2 - 100], 0.95), fontsize=25, type='lineedit') def add_fits_information_label(self): info_text = "Object Name: {}, type: {}, spectral class: {}\nCoordinates: {}\nDate: {}\nObserver: {}\nEquipment: {}\nPosition: {}".format( self.object_properties.name, self.object_properties.type, self.object_properties.sptype, self.object_properties.printable_coordinates(), self.object_properties.date.toString(), self.object_properties.observer, self.object_properties.equipment, self.object_properties.position) self.add_label( info_text, type='textbox', coords=(self.spectrum.wavelengths[len(self.spectrum.wavelengths) / 4 * 3], 0.80), fontsize=14) self.profile_plot.figure.canvas.draw() def add_label(self, text=None, type='textbox', coords=None, fontsize=12, color='black'): if not coords: coords = ( self.spectrum.wavelengths[len(self.spectrum.wavelengths) / 2], 0.5) self.labels.append( (type, MoveableLabel( text=text if text else 'Label - double click to edit', on_dblclick=lambda l: self.edit_label(l, type=type), x=coords[0], y=coords[1], fontsize=fontsize, color=color, axes=self.profile_plot.axes))) self.profile_plot.figure.canvas.draw() def edit_label(self, label, type='lineedit'): def remove_label(self, label, dialog): label.remove() self.labels.remove([l for l in self.labels if l[1] == label][0]) self.profile_plot.figure.canvas.draw() dialog.reject() dialog = QDialog() dialog.setWindowTitle("Edit Label") dialog.setLayout(QVBoxLayout()) font_size = QSpinBox() font_size.setValue(label.get_fontsize()) dialog.layout().addWidget(QLabel("Font Size")) dialog.layout().addWidget(font_size) text_edit = None if type == 'lineedit': text_edit = QLineEdit(label.get_text()) else: text_edit = QTextEdit() text_edit.setPlainText(label.get_text()) dialog.layout().addWidget(QLabel("Text")) dialog.layout().addWidget(text_edit) button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel) button_box.accepted.connect(dialog.accept) button_box.rejected.connect(dialog.reject) remove_button = QPushButton('Remove') remove_button.clicked.connect( lambda: remove_label(self, label, dialog)) dialog.layout().addWidget(remove_button) dialog.layout().addWidget(button_box) if QDialog.Accepted != dialog.exec(): return label.set_text(text_edit.text() if type == 'lineedit' else text_edit.toPlainText()) label.set_fontsize(font_size.value()) label.axes.figure.canvas.draw()