class CheckBoxWidget(AbstractDataSetWidget):
"""
BoolItem widget
"""
def __init__(self, item, parent_layout):
super(CheckBoxWidget, self).__init__(item, parent_layout)
self.checkbox = QCheckBox(self.item.get_prop_value("display", "text"))
self.checkbox.setToolTip(item.get_help())
self.group = self.checkbox
self.store = self.item.get_prop("display", "store", None)
if self.store:
self.checkbox.stateChanged.connect(self.do_store)
def get(self):
"""Override AbstractDataSetWidget method"""
value = self.item.get()
if value:
self.checkbox.setChecked(True)
def set(self):
"""Override AbstractDataSetWidget method"""
self.item.set(self.value())
def value(self):
return self.checkbox.isChecked()
def place_on_grid(self,
layout,
row,
label_column,
widget_column,
row_span=1,
column_span=1):
"""Override AbstractDataSetWidget method"""
if not self.item.get_prop_value("display", "label"):
widget_column = label_column
column_span += 1
else:
self.place_label(layout, row, label_column)
layout.addWidget(self.group, row, widget_column, row_span, column_span)
def do_store(self, state):
self.store.set(self.item.instance, self.item.item, state)
self.parent_layout.refresh_widgets()
class CheckBoxWidget(AbstractDataSetWidget):
"""
BoolItem widget
"""
def __init__(self, item, parent_layout):
super(CheckBoxWidget, self).__init__(item, parent_layout)
self.checkbox = QCheckBox(self.item.get_prop_value("display", "text"))
self.checkbox.setToolTip(item.get_help())
self.group = self.checkbox
self.store = self.item.get_prop("display", "store", None)
if self.store:
self.checkbox.stateChanged.connect(self.do_store)
def get(self):
"""Override AbstractDataSetWidget method"""
value = self.item.get()
if value is not None:
self.checkbox.setChecked(value)
def set(self):
"""Override AbstractDataSetWidget method"""
self.item.set(self.value())
def value(self):
return self.checkbox.isChecked()
def place_on_grid(self, layout, row, label_column, widget_column,
row_span=1, column_span=1):
"""Override AbstractDataSetWidget method"""
if not self.item.get_prop_value("display", "label"):
widget_column = label_column
column_span += 1
else:
self.place_label(layout, row, label_column)
layout.addWidget(self.group, row, widget_column, row_span, column_span)
def do_store(self, state):
self.store.set(self.item.instance, self.item.item, state)
self.parent_layout.refresh_widgets()
class FileUi(QSplitter):
def __init__(self, parent):
super(FileUi, self).__init__(Qt.Vertical, parent)
layoutWidget = QWidget()
layout = QVBoxLayout()
layoutWidget.setLayout(layout)
self.comment = QPlainTextEdit()
self.fileName = None
self.saveTxtCheck = QCheckBox('Save Txt')
self.saveHdfCheck = QCheckBox('Save HDF5')
self.saveTxtBtn = QPushButton('Save Txt')
self.saveHdfBtn = QPushButton('Save HDF5')
#############
# stream group
streamGroup = QGroupBox('Stream data to file:')
streamGroup.setFlat(True)
streamGroupLayout = QHBoxLayout()
streamGroupLayout.addWidget(self.saveTxtCheck)
streamGroupLayout.addWidget(self.saveHdfCheck)
streamGroup.setLayout(streamGroupLayout)
##############
# save static file
saveGroup = QGroupBox('Save now to file:')
saveGroup.setFlat(True)
saveGroupLayout = QHBoxLayout()
saveGroupLayout.addWidget(self.saveTxtBtn)
saveGroupLayout.addWidget(self.saveHdfBtn)
saveGroup.setLayout(saveGroupLayout)
##############
# put layout together
layout.addWidget(QLabel('Comment:'))
layout.addWidget(self.comment)
layout.addWidget(streamGroup)
layout.addWidget(saveGroup)
self.addWidget(layoutWidget)
##############
# connnect stuff for functionality
self.saveTxtCheck.stateChanged.connect(self.__makeFileName)
self.saveHdfCheck.stateChanged.connect(self.__makeFileName)
##############
# thread for streaming data to file
#self.updateCurrPos.connect(self.__updateCurrPos)
self.stream_thread = QThread() # create the QThread
self.stream_thread.start()
self.stream_worker = GenericWorker(self.__streamFile)
self.stream_worker.moveToThread(self.stream_thread)
def __makeFileName(self, state):
if state == 2:
self.fileName = datetime.now().strftime('%Y%m%d-%H%M%S')
self.stream_worker.start.emit()
else:
self.fileName = None
def __streamFile(self):
while self.saveHdfCheck.isChecked():
self.saveHdfBtn.released.emit()
time.sleep(10)
