def _save_select_detect(self, *args, filename=None):
"""Export selected detection."""
channel, method = self._get_current_chan_type()
# Read Table
row_count = self._DetectLocations.rowCount()
sta_ind = [channel, '', 'Time index (s)']
end_ind = [method, '', 'Time index (s)']
duration = ['', '', 'Duration (s)']
stage = ['', '', 'Sleep stage']
for row in np.arange(row_count):
sta_ind.append(str(self._DetectLocations.item(row, 0).text()))
end_ind.append(str(self._DetectLocations.item(row, 1).text()))
duration.append(str(self._DetectLocations.item(row, 2).text()))
stage.append(str(self._DetectLocations.item(row, 3).text()))
# Get file name :
saveas = "locinfo" + '_' + channel + '-' + method
if filename is None:
filename = dialog_save(
self, 'Save ' + method + ' detection', saveas,
"CSV file (*.csv);;Text file"
" (*.txt);;All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
file += '_' + channel + '-' + method
zp = zip(sta_ind, end_ind, duration, stage)
if ext.find('csv') + 1:
write_csv(file + '.csv', zp)
elif ext.find('txt') + 1:
write_txt(file + '.txt', zp)
def _save_select_detect(self, *args, filename=None):
"""Export selected detection."""
channel, method = self._get_current_chan_type()
# Read Table
row_count = self._DetectLocations.rowCount()
sta_ind = [channel, '', 'Time index (s)']
end_ind = [method, '', 'Time index (s)']
duration = ['', '', 'Duration (s)']
stage = ['', '', 'Sleep stage']
for row in np.arange(row_count):
sta_ind.append(str(self._DetectLocations.item(row, 0).text()))
end_ind.append(str(self._DetectLocations.item(row, 1).text()))
duration.append(str(self._DetectLocations.item(row, 2).text()))
stage.append(str(self._DetectLocations.item(row, 3).text()))
# Get file name :
saveas = "locinfo" + '_' + channel + '-' + method
if filename is None:
filename = dialog_save(self, 'Save ' + method + ' detection',
saveas, "CSV file (*.csv);;Text file"
" (*.txt);;All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
file += '_' + channel + '-' + method
zp = zip(sta_ind, end_ind, duration, stage)
if ext.find('csv') + 1:
write_csv(file + '.csv', zp)
elif ext.find('txt') + 1:
write_txt(file + '.txt', zp)
def _save_info_table(self, *args, filename=None):
"""Export stat info."""
# Get file name :
if filename is None:
filename = dialog_save(self, 'Save file', 'statsinfo',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(self._keysInfo, self._valInfo))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(self._keysInfo, self._valInfo))
def _save_info_table(self, *args, filename=None):
"""Export stat info."""
# Get file name :
if filename is None:
filename = dialog_save(
self, 'Save file', 'statsinfo',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(self._keysInfo, self._valInfo))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(self._keysInfo, self._valInfo))
def _save_annotation_table(self, *args, filename=None):
"""Export annotation table."""
# Read Table
row_count = self._AnnotateTable.rowCount()
sta_ind, end_ind, annot = [], [], []
for row in np.arange(row_count):
sta_ind.append(str(self._AnnotateTable.item(row, 0).text()))
end_ind.append(str(self._AnnotateTable.item(row, 1).text()))
annot.append(str(self._AnnotateTable.item(row, 2).text()))
# Get file name :
if filename is None:
filename = dialog_save(self, 'Save annotations', 'annotations',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(sta_ind, end_ind, annot))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(sta_ind, end_ind, annot))
def _save_scoring_table(self, *args, filename=None):
"""Export score info."""
# Read Table
row_count = self._scoreTable.rowCount()
sta_ind, end_ind, stage = [], [], []
for row in np.arange(row_count):
sta_ind.append(str(self._scoreTable.item(row, 0).text()))
end_ind.append(str(self._scoreTable.item(row, 1).text()))
stage.append(str(self._scoreTable.item(row, 2).text()))
# Get file name :
if filename is None:
filename = dialog_save(self, 'Save file', 'scoring_info',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(sta_ind, end_ind, stage))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(sta_ind, end_ind, stage))
def _save_annotation_table(self, *args, filename=None):
"""Export annotation table."""
# Read Table
row_count = self._AnnotateTable.rowCount()
sta_ind, end_ind, annot = [], [], []
for row in np.arange(row_count):
sta_ind.append(str(self._AnnotateTable.item(row, 0).text()))
end_ind.append(str(self._AnnotateTable.item(row, 1).text()))
annot.append(str(self._AnnotateTable.item(row, 2).text()))
# Get file name :
if filename is None:
filename = dialog_save(
self, 'Save annotations', 'annotations',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(sta_ind, end_ind, annot))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(sta_ind, end_ind, annot))
def _save_scoring_table(self, *args, filename=None):
"""Export score info."""
# Read Table
row_count = self._scoreTable.rowCount()
sta_ind, end_ind, stage = [], [], []
for row in np.arange(row_count):
sta_ind.append(str(self._scoreTable.item(row, 0).text()))
end_ind.append(str(self._scoreTable.item(row, 1).text()))
stage.append(str(self._scoreTable.item(row, 2).text()))
# Get file name :
if filename is None:
filename = dialog_save(
self, 'Save file', 'scoring_info',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(sta_ind, end_ind, stage))
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(sta_ind, end_ind, stage))
def _fcn_save_annotations(self, *args, filename=None):
"""Save annotations."""
# Read Table
row_count = self._annot_table.rowCount()
time, amp, signal, text = [], [], [], []
for row in range(row_count):
time.append(self._annot_table.item(row, 0).text())
amp.append(self._annot_table.item(row, 1).text())
signal.append(self._annot_table.item(row, 2).text())
text.append(self._annot_table.item(row, 3).text())
# Get file name :
if filename is None:
filename = dialog_save(self, 'Save annotations', 'annotations',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
kw = {'delimiter': '_'}
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(time, amp, signal, text), **kw)
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(time, amp, signal, text), **kw)
def _fcn_save_annotations(self, *args, filename=None):
"""Save annotations."""
# Read Table
row_count = self._annot_table.rowCount()
time, amp, signal, text = [], [], [], []
for row in range(row_count):
time.append(self._annot_table.item(row, 0).text())
amp.append(self._annot_table.item(row, 1).text())
signal.append(self._annot_table.item(row, 2).text())
text.append(self._annot_table.item(row, 3).text())
# Get file name :
if filename is None:
filename = dialog_save(
self, 'Save annotations', 'annotations',
"CSV file (*.csv);;Text file (*.txt);;"
"All files (*.*)")
if filename:
kw = {'delimiter': '_'}
file, ext = os.path.splitext(filename)
if ext.find('csv') + 1:
write_csv(file + '.csv', zip(time, amp, signal, text), **kw)
elif ext.find('txt') + 1:
write_txt(file + '.txt', zip(time, amp, signal, text), **kw)