def exec_image_open_dialog(parent, basedir='', app_name=None,
to_grayscale=True, dtype=None):
"""
Executes an image open dialog box (QFileDialog.getOpenFileName)
* parent: parent widget (None means no parent)
* basedir: base directory ('' means current directory)
* app_name (opt.): application name (used as a title for an eventual
error message box in case something goes wrong when saving image)
* to_grayscale (default=True): convert image to grayscale
Returns (filename, data) tuple if dialog is accepted, None otherwise
"""
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(parent, _("Open"), basedir,
io.iohandler.get_filters('load', dtype=dtype))
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
filename = to_text_string(filename)
try:
data = io.imread(filename, to_grayscale=to_grayscale)
except Exception as msg:
import traceback
traceback.print_exc()
QMessageBox.critical(parent,
_('Error') if app_name is None else app_name,
(_("%s could not be opened:") % osp.basename(filename))+\
"\n"+str(msg))
return
return filename, data
def exec_image_open_dialog(parent,
basedir='',
app_name=None,
to_grayscale=True,
dtype=None):
"""
Executes an image open dialog box (QFileDialog.getOpenFileName)
* parent: parent widget (None means no parent)
* basedir: base directory ('' means current directory)
* app_name (opt.): application name (used as a title for an eventual
error message box in case something goes wrong when saving image)
* to_grayscale (default=True): convert image to grayscale
Returns (filename, data) tuple if dialog is accepted, None otherwise
"""
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(
parent, _("Open"), basedir,
io.iohandler.get_filters('load', dtype=dtype))
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
filename = to_text_string(filename)
try:
data = io.imread(filename, to_grayscale=to_grayscale)
except Exception as msg:
import traceback
traceback.print_exc()
QMessageBox.critical(parent,
_('Error') if app_name is None else app_name,
(_("%s could not be opened:") % osp.basename(filename))+\
"\n"+str(msg))
return
return filename, data
def open_image(self):
"""Open image file"""
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(self, _("Open"), "",
io.iohandler.get_filters('load'))
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
if filename:
self.mainwidget.add_image_from_file(filename)
def open_image(self):
"""Open image file"""
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(self, _("Open"), "",
io.iohandler.get_filters('load'))
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
if filename:
self.mainwidget.add_image_from_file(filename)
def open_csv(self):
"""Open csv file"""
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(self, _("Open a CSV file"), "",
io.iohandler.get_filters('load'))
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
if filename:
self.mainwidget.csvmodel.addCSV(filename)
def preflag(self):
filename = getopenfilename(self, _("Open"), "")
if not str(filename[0]):
return -1
else:
#try:
self.flagname = str(filename[0])
self.data_ref.flag = self.data_ref.get_flag_curve(
self.flagname, self.dssel.section)
self.properties_changed()
def _open_file(self):
saved_in, saved_out, saved_err = sys.stdin, sys.stdout, sys.stderr
sys.stdout = None
filename, _filter = getopenfilename(self, _("Open"), "")
sys.stdin, sys.stdout, sys.stderr = saved_in, saved_out, saved_err
if filename:
extension = splitext(str(filename))[1]
if extension not in ['.fits', '.hdf5', '.hdf', '.h5']:
msgBox = QMessageBox()
msgBox.setText("Wrong file type!")
msgBox.exec_()
return -1
if extension == '.fits':
file_type = "fits"
elif extension == (".hdf5" or ".h5" or ".hdf"):
msgBox = QMessageBox()
msgBox.setText(
"Are data spectrometric or spectropolarimetric?")
msgBox.addButton(QPushButton('Spect'), QMessageBox.NoRole)
msgBox.addButton(QPushButton('Polar'), QMessageBox.YesRole)
ret = msgBox.exec_()
if ret == False:
widg_type = 0
file_type = 'hdf'
elif ret == True:
widg_type = 1
file_type = 'hdf_pola'
self.data = dwdata.DWData()
self.data.open_data(filename, file_type)
if file_type == 'fits':
if self.data.data_type_dict[0][0] == 'simple':
widg_type = 0
elif self.data.data_type_dict[0][0] == 'stokes':
widg_type = 1
if widg_type == 0:
self.setup_central_widget(dwwidg.MainWidget, self.toolbar)
elif widg_type == 1:
self.setup_central_widget(dwwidg.PolaViewerWidget,
self.toolbar)
self.set_flag_toolbar()
self.mainwidget.open_setup(self.data)
self.setWindowTitle(APP_NAME + " - " + filename)
def apply_fit(self):
msgBox = QMessageBox()
msgBox.setText("Apply corrections?")
msgBox.setInformativeText(
"Select to which files will be applied the correction.")
CancelButton = msgBox.addButton("Cancel", msgBox.RejectRole)
SelectButton = msgBox.addButton("Select", msgBox.NoRole)
AllButton = msgBox.addButton("All", msgBox.YesRole)
#msgBox.setDefaultButton(QMessageBox.Cancel)
msgBox.exec_()
if msgBox.clickedButton() == CancelButton:
return -1
elif msgBox.clickedButton() == AllButton:
filelist = None
elif msgBox.clickedButton() == SelectButton:
flret = getopenfilenames(self, _("Select file list"),
self.data_ref.dir_name)[0]
filelist = []
for f in flret:
filelist.append((basename(str(f)), dirname(str(f))))
msgBox = QMessageBox()
msgBox.setText("Select the correction file")
msgBox.addButton(QMessageBox.Ok)
msgBox.exec_()
filename = getopenfilename(
self, _("Correction file"),
self.data_ref.dir_name + "/" + basename(
normpath(self.data_ref.dir_name) + "_fit." +
self.data_ref.files_type))
try:
self.data_ref.applycorr(str(filename[0]), filelist)
except:
msgBox = QMessageBox()
msgBox.setText("Not a correction file!")
msgBox.exec_()
return -1
