def __init__(self, parent=None, designMode=False):
TaurusWidget.__init__(self, parent=parent, designMode=designMode)
self._xAttr = None
self._yAttr = None
self._arrayEditor = ArrayEditor()
self.fromFileBT = Qt.QPushButton('Read From File')
self.toFileBT = Qt.QPushButton('Write To File')
self.fromAttrBT = Qt.QPushButton('Read From Device')
self.toAttrBT = Qt.QPushButton('Write To Device')
self.fromAttrBT.setEnabled(False)
self.toAttrBT.setEnabled(False)
layout = Qt.QGridLayout(self)
layout.addWidget(self._arrayEditor, 0, 0, 1, 4)
layout.addWidget(self.fromFileBT, 1, 0)
layout.addWidget(self.toFileBT, 1, 1)
layout.addWidget(self.fromAttrBT, 1, 2)
layout.addWidget(self.toAttrBT, 1, 3)
self.fromFileBT.clicked[()].connect(self.onFromFile)
self.toFileBT.clicked[()].connect(self.onToFile)
self.fromAttrBT.clicked[()].connect(self.onFromAttr)
self.toAttrBT.clicked[()].connect(self.onToAttr)
def __init__(self, parent=None, designMode=False):
TaurusWidget.__init__(self, parent=parent, designMode=designMode)
self._xAttr = None
self._yAttr = None
self._arrayEditor = ArrayEditor()
self.fromFileBT = Qt.QPushButton('Read From File')
self.toFileBT = Qt.QPushButton('Write To File')
self.fromAttrBT = Qt.QPushButton('Read From Device')
self.toAttrBT = Qt.QPushButton('Write To Device')
self.fromAttrBT.setEnabled(False)
self.toAttrBT.setEnabled(False)
layout = Qt.QGridLayout(self)
layout.addWidget(self._arrayEditor, 0, 0, 1, 4)
layout.addWidget(self.fromFileBT, 1, 0)
layout.addWidget(self.toFileBT, 1, 1)
layout.addWidget(self.fromAttrBT, 1, 2)
layout.addWidget(self.toAttrBT, 1, 3)
self.fromFileBT.clicked[()].connect(self.onFromFile)
self.toFileBT.clicked[()].connect(self.onToFile)
self.fromAttrBT.clicked[()].connect(self.onFromAttr)
self.toAttrBT.clicked[()].connect(self.onToAttr)
class TaurusArrayEditor(TaurusWidget):
def __init__(self, parent=None, designMode=False):
TaurusWidget.__init__(self, parent=parent, designMode=designMode)
self._xAttr = None
self._yAttr = None
self._arrayEditor = ArrayEditor()
self.fromFileBT = Qt.QPushButton('Read From File')
self.toFileBT = Qt.QPushButton('Write To File')
self.fromAttrBT = Qt.QPushButton('Read From Device')
self.toAttrBT = Qt.QPushButton('Write To Device')
self.fromAttrBT.setEnabled(False)
self.toAttrBT.setEnabled(False)
layout = Qt.QGridLayout(self)
layout.addWidget(self._arrayEditor, 0, 0, 1, 4)
layout.addWidget(self.fromFileBT, 1, 0)
layout.addWidget(self.toFileBT, 1, 1)
layout.addWidget(self.fromAttrBT, 1, 2)
layout.addWidget(self.toAttrBT, 1, 3)
self.fromFileBT.clicked[()].connect(self.onFromFile)
self.toFileBT.clicked[()].connect(self.onToFile)
self.fromAttrBT.clicked[()].connect(self.onFromAttr)
self.toAttrBT.clicked[()].connect(self.onToAttr)
def arrayEditor(self):
return self._arrayEditor
def setModel(self, model):
'''returns True if a curve could be set from the attribute. Flase otherwise'''
if not model:
self._xAttr = self._yAttr = None
self.fromAttrBT.setEnabled(False)
return
attrs = str(model).split("|")
self._yAttr = taurus.Attribute(attrs[-1])
if self._yAttr is None or len(attrs) not in (1, 2):
self.error('Invalid model for %s' % str(self.__class__))
self.fromAttrBT.setEnabled(False)
return
if len(attrs) == 1:
self._xAttr = None
else:
self._xAttr = taurus.Attribute(attrs[0])
# set the master curve from the attribute(s)
ok = self.onFromAttr(quiet=True)
# enable/disable the attribute I/O buttons
self.fromAttrBT.setEnabled(True)
enableWrite = (self._xAttr is None or
self._xAttr.isWritable()) and self._yAttr.isWritable()
self.toAttrBT.setEnabled(True)
return ok
def onFromFile(self, filename=None, **kwargs):
'''imports Master curve from a two-column ASCII file.
The first colum will be interpreted to be the abcissas.
If filename is not given, a dialog for choosing a file is presented
kwargs can contain keyword arguments to pass to numpy.loadtxt() when reading each file
accepted keywords and their default values are:
{dtype=<type 'float'>, comments='#', delimiter=None, converters=None,
skiprows=0, usecols=None, unpack=False}
see help from numpy.loadtxt for more info on the kwargs'''
if filename is None:
filename = Qt.QFileDialog.getOpenFileName(self,
'Choose input file', '',
'Ascii file (*)')
if not filename:
return False
filename = str(filename)
try:
M = numpy.loadtxt(filename, **kwargs)
if len(M.shape) != 2:
raise Exception()
x = M[:, 0]
y = M[:, 1]
except:
self.error('Invalid input file "%s"' % filename)
Qt.QMessageBox.warning(
self, 'Invalid input file',
'Unknown format in selected file.\n Hint: The file must be two-column ASCII'
)
return
self._arrayEditor.setMaster(x, y)
def onToFile(self):
'''writes the Corrected curve to an ascii file'''
if self._arrayEditor.plot1.exportAscii(curves=['Corrected']):
x, y = self._arrayEditor.getCorrected()
self._arrayEditor.setMaster(x, y, keepCP=True)
def onFromAttr(self, quiet=False):
'''reads the Master curve from the attributes set by model.
'''
if self._yAttr is None:
return False # @TODO: launch a dialog for choosing the x and y attributes
try:
y = numpy.array(self._yAttr.read().rvalue)
if self._xAttr is None:
x = numpy.arange(y.size)
else:
x = numpy.array(self._xAttr.read().rvalue)
except Exception, e:
self.error('Error reading from attribute(s): %s' % (str(e)))
if not quiet:
Qt.QMessageBox.warning(self, 'Error Reading Attribute',
'Cannot read master curve')
return False
if quiet:
should_set = True
else:
answer = Qt.QMessageBox.question(
self, 'Read from attributes?',
'Read Master curve from attributes?',
Qt.QMessageBox.Yes | Qt.QMessageBox.No)
should_set = (answer == Qt.QMessageBox.Yes == answer)
if should_set:
try:
self._arrayEditor.setMaster(x, y)
except ValueError:
self.error('Cannot set master curve from attributes')
if not quiet:
Qt.QMessageBox.warning(
self, 'Error',
'Cannot set master curve from attributes')
return False
return True
class TaurusArrayEditor(TaurusWidget):
def __init__(self, parent=None, designMode=False):
TaurusWidget.__init__(self, parent=parent, designMode=designMode)
self._xAttr = None
self._yAttr = None
self._arrayEditor = ArrayEditor()
self.fromFileBT = Qt.QPushButton('Read From File')
self.toFileBT = Qt.QPushButton('Write To File')
self.fromAttrBT = Qt.QPushButton('Read From Device')
self.toAttrBT = Qt.QPushButton('Write To Device')
self.fromAttrBT.setEnabled(False)
self.toAttrBT.setEnabled(False)
layout = Qt.QGridLayout(self)
layout.addWidget(self._arrayEditor, 0, 0, 1, 4)
layout.addWidget(self.fromFileBT, 1, 0)
layout.addWidget(self.toFileBT, 1, 1)
layout.addWidget(self.fromAttrBT, 1, 2)
layout.addWidget(self.toAttrBT, 1, 3)
self.fromFileBT.clicked[()].connect(self.onFromFile)
self.toFileBT.clicked[()].connect(self.onToFile)
self.fromAttrBT.clicked[()].connect(self.onFromAttr)
self.toAttrBT.clicked[()].connect(self.onToAttr)
def arrayEditor(self):
return self._arrayEditor
def setModel(self, model):
'''returns True if a curve could be set from the attribute. Flase otherwise'''
if not model:
self._xAttr = self._yAttr = None
self.fromAttrBT.setEnabled(False)
return
attrs = str(model).split("|")
self._yAttr = taurus.Attribute(attrs[-1])
if self._yAttr is None or len(attrs) not in (1, 2):
self.error('Invalid model for %s' % str(self.__class__))
self.fromAttrBT.setEnabled(False)
return
if len(attrs) == 1:
self._xAttr = None
else:
self._xAttr = taurus.Attribute(attrs[0])
# set the master curve from the attribute(s)
ok = self.onFromAttr(quiet=True)
# enable/disable the attribute I/O buttons
self.fromAttrBT.setEnabled(True)
enableWrite = (self._xAttr is None or self._xAttr.isWritable()
) and self._yAttr.isWritable()
self.toAttrBT.setEnabled(True)
return ok
def onFromFile(self, filename=None, **kwargs):
'''imports Master curve from a two-column ASCII file.
The first colum will be interpreted to be the abcissas.
If filename is not given, a dialog for choosing a file is presented
kwargs can contain keyword arguments to pass to numpy.loadtxt() when reading each file
accepted keywords and their default values are:
{dtype=<type 'float'>, comments='#', delimiter=None, converters=None,
skiprows=0, usecols=None, unpack=False}
see help from numpy.loadtxt for more info on the kwargs'''
if filename is None:
filename = Qt.QFileDialog.getOpenFileName(
self, 'Choose input file', '', 'Ascii file (*)')
if not filename:
return False
filename = str(filename)
try:
M = numpy.loadtxt(filename, **kwargs)
if len(M.shape) != 2:
raise Exception()
x = M[:, 0]
y = M[:, 1]
except:
self.error('Invalid input file "%s"' % filename)
Qt.QMessageBox.warning(
self, 'Invalid input file', 'Unknown format in selected file.\n Hint: The file must be two-column ASCII')
return
self._arrayEditor.setMaster(x, y)
def onToFile(self):
'''writes the Corrected curve to an ascii file'''
if self._arrayEditor.plot1.exportAscii(curves=['Corrected']):
x, y = self._arrayEditor.getCorrected()
self._arrayEditor.setMaster(x, y, keepCP=True)
def onFromAttr(self, quiet=False):
'''reads the Master curve from the attributes set by model.
'''
if self._yAttr is None:
return False # @TODO: launch a dialog for choosing the x and y attributes
try:
y = numpy.array(self._yAttr.read().rvalue)
if self._xAttr is None:
x = numpy.arange(y.size)
else:
x = numpy.array(self._xAttr.read().rvalue)
except Exception, e:
self.error('Error reading from attribute(s): %s' % (str(e)))
if not quiet:
Qt.QMessageBox.warning(
self, 'Error Reading Attribute', 'Cannot read master curve')
return False
if quiet:
should_set = True
else:
answer = Qt.QMessageBox.question(
self, 'Read from attributes?',
'Read Master curve from attributes?',
Qt.QMessageBox.Yes|Qt.QMessageBox.No
)
should_set = (answer == Qt.QMessageBox.Yes == answer)
if should_set:
try:
self._arrayEditor.setMaster(x, y)
except ValueError:
self.error('Cannot set master curve from attributes')
if not quiet:
Qt.QMessageBox.warning(self, 'Error',
'Cannot set master curve from attributes')
return False
return True
