def importData(self):
"""Import data into the application as waves."""
# Check if the proposed wave name is acceptable
validatedWaveName = Wave.validateWaveName(Util.getWidgetValue(self._ui.waveName))
if not self._app.waves().goodWaveName(validatedWaveName):
badWaveNameMessage = QMessageBox()
badWaveNameMessage.setText("Wave name is not allowed or already in use. Please change the wave name.")
badWaveNameMessage.exec_()
return False
# Get data type and size
uiDataType = Util.getWidgetValue(self._ui.dataType)
uiByteOrder = Util.getWidgetValue(self._ui.byteOrder)
dataType = self.dataTypes[uiDataType]['dtype']
dataTypeChar = self.dataTypes[uiDataType]['char']
numBytes = self.dataTypes[uiDataType]['numbytes']
byteOrder = self.byteOrders[uiByteOrder]
# Load data
fileName = Util.getWidgetValue(self._ui.fileName)
if os.path.isfile(fileName):
data = []
wave = Wave(str(validatedWaveName), dataType)
self._app.waves().addWave(wave)
fh = open(fileName, 'rb')
binDatum = fh.read(numBytes)
while binDatum != "":
try:
datum = struct.unpack(byteOrder + dataTypeChar, binDatum)
wave.push(datum[0])
except:
pass
binDatum = fh.read(numBytes)
# data = array.array(dataTypeChar)
# numDataPoints = os.path.getsize(fileName) / data.itemsize
# fh = open(fileName, 'rb')
# data.fromfile(fh, numDataPoints)
#
# wave = Wave(str(validatedWaveName), dataType)
# wave.replaceData(data)
# self._app.waves().addWave(wave)
# Close window
self.window.hide()
def goodWaveName(self, name):
"""
Determine if name is a good Wave name for this Waves object. Returns False if name is an empty string. Returns True if name is not an empty string and is unique.
"""
name = Wave.validateWaveName(name)
if name == "":
return False
return self.uniqueWaveName(name)
def validateWaveNames(self):
"""
Run Wave.validateWaveName on all potential wave names.
"""
dataTable = self._ui.data
for col in range(dataTable.columnCount()):
if dataTable.item(1, col):
dataTable.setItem(1, col, QTableWidgetItem(Wave.validateWaveName(str(dataTable.item(1, col).text()))))
