class PlottingWindow(QWidget):
"""
PlottingWindow is the class of the main window containing the
plottng curves
"""
def __init__(self):
QMainWindow.__init__(self)
# init the window for the app
self.setWindowTitle('UCare Blood Pressure Monitor')
#self.resize(640, 480)
#self.statusBar().showMessage('There is nothing now.')
self.setWindowIcon(QIcon('icon/icon.png'))
# add a menu
# add a toolbar
# set the layout box for plotting
vbox = QVBoxLayout()
# initialize the plot_helpers
self.bp_ecg_plot_helper = PlottingHelper(self, BP_ECG_signal_names, 1000)
self.acc_plot_helper = PlottingHelper(self, ACC_signal_names, 100)
for key, curve_plot in self.bp_ecg_plot_helper.curve_plots.items():
vbox.addWidget(curve_plot)
for key, curve_plot in self.acc_plot_helper.curve_plots.items():
vbox.addWidget(curve_plot)
self.setLayout(vbox)
# the dataHolder
self.data_holder = DataHolder()
# the serial port controller
self.data_receiver = DataReceiver()
port = list_ports.comports()
print port
self.serial_settings = {"port":'Com8', "baud":115200, "bytesize":8,\
"parity":"N", "stopbits":1, "timeout":1}
self.connect(self.data_receiver.qtobj, SIGNAL("NewData"),\
self.data_holder.on_receive_data)
self.connect(self.data_holder.qtobj, SIGNAL("UpdateCurve"),\
self.update_curves)
if self.data_receiver.open(self.serial_settings):
print "successfully opened."
sleep(0.1)
self.data_receiver.start()
def update_curves(self):
''' update the curves'''
self.bp_ecg_plot_helper.update_curves(self.data_holder.time_seq, \
self.data_holder.signals, 5)
self.acc_plot_helper.update_curves(self.data_holder.time_seq\
[0:len(self.data_holder.time_seq):10], \
self.data_holder.signals, 5)
def main():
app = QApplication(sys.argv)
main = PlottingWindow()
main.show()
sys.exit(app.exec_())
def __init__(self, parent=None):
''' This is the initializer of the main window
'''
super(MainWindow, self).__init__(parent)
#=======================================
# initial the member field
#=======================================
self.interval_in_seconds = 5
self.q_time_elapsed = QTime(0, 0, 0, 0)
self.timer = QTimer()
self.timer.setSingleShot(False)
self.timer.setInterval(1000)
self.connect(self.timer, SIGNAL("timeout()"), self.updateTheTime)
# Data receiver
self.data_receiver = DataTransfer()
self.serial_settings = {}
# Data holder
self.data_holder = DataHolder()
self.connect(self.data_receiver.qtobj, SIGNAL("NewData"),\
self.data_holder.on_receive_data)
# File helper
self.file_helper = FileHelper()
self.connect(self.data_receiver.qtobj, SIGNAL("NewData"),\
self.file_helper.writeToSignalFile)
# Plot_helpers
self.bp_ecg_plot_helper = PlottingHelper(self, \
BP_ECG_signal_names, 1000)
self.acc_plot_helper = PlottingHelper(self, \
ACC_signal_names, 100)
#=====================================================
# Set central widget
#=====================================================
self.setWindowTitle('UCare BP Monitor')
self.setWindowIcon(QIcon('icon/icon.png'))
# =========================================
# 1: set plot widget
# =========================================
self.plotwidget = QWidget()
self.plotVBox = QVBoxLayout()
# Add plotting widget to the plotVBox
for name in BP_ECG_names:
self.plotVBox.addWidget(self.bp_ecg_plot_helper.curve_plots[name])
label = QLabel(name)
label.setAlignment(Qt.AlignCenter)
self.plotVBox.addWidget(label)
#for key, curve_plot in self.bp_ecg_plot_helper.curve_plots.items():
# self.plotVBox.addWidget(curve_plot)
for key, curve_plot in self.acc_plot_helper.curve_plots.items():
self.plotVBox.addWidget(curve_plot)
label = QLabel(key)
label.setAlignment(Qt.AlignCenter)
self.plotVBox.addWidget(label)
self.plotwidget.setLayout(self.plotVBox)
# =========================================
# 2: set event widget
# =========================================
self.eventWidget = QWidget()
self.eventVBox = QVBoxLayout()
# ============================
# 2-a: set eventButton widget
# ============================
self.eventButtonWidget = EventButtonsWidget()
self.connect(self.eventButtonWidget.qtobj, SIGNAL("NewEvent"), self.addEventLog)
# ============================
# 2-b: set eventLog widget
# ============================
self.eventLogList = QListWidget()
# ============================
# 2-c: add the widgets into eventWidget
# ============================
self.eventVBox.addWidget(self.eventButtonWidget)
self.eventVBox.addWidget(QLabel(u"事件日志"))
self.eventVBox.addWidget(self.eventLogList)
self.eventWidget.setLayout(self.eventVBox)
# =========================================
# 3: add widgets into centralWidget
# =========================================
centralWidget = QWidget()
centralHBox = QHBoxLayout()
centralHBox.addWidget(self.plotwidget)
centralHBox.addWidget(self.eventWidget)
centralWidget.setLayout(centralHBox)
self.setCentralWidget(centralWidget)
self.connect(self.data_holder.qtobj, SIGNAL("UpdateCurve"),\
self.update_curves)
#=======================================
# Set dock windows
#=======================================
#self.
#=======================================
# Set statusbar
#=======================================
self.statusbar = self.statusBar()
self.label_time_elapsed = QLabel()
self.label_time_elapsed.setText(self.q_time_elapsed.toString())
self.statusbar.addPermanentWidget(self.label_time_elapsed)
#=======================================
# Set actions
#=======================================
# Open a file
tip_text = 'Open an existing data file.'
open_file_action = self.create_action('&Open a file', \
self.open_file, QKeySequence.Open, 'icon\open.png',\
tip_text, False)
open_file_action.setEnabled(False)
# Save the file
tip_text = 'Save data into a file.'
save_file_action = self.create_action('&Save into file', \
self.save_file, QKeySequence.Save, 'icon\save.png',\
tip_text, False)
# Bluetooth settings
tip_text = 'Settings of bluetooth. Ctrl+B'
bt_settings_action = self.create_action('&Bluetooth Settings',\
self.bt_settings, 'Ctrl+B', 'icon\settings.png',\
tip_text, False)
# Plotting start
tip_text = 'Start ploting the signals. Ctrl+C'
self.start_action = self.create_action('Start | &Continue plotting.',\
self.start_plot, 'Ctrl+C', 'icon\start.png', \
tip_text, True)
self.start_action.setEnabled(False)
# Plotting pause
tip_text = 'Pause the plotting. Ctrl+P'
self.pause_action = self.create_action('&Pause the plotting.', \
self.pause_plot, 'Ctrl+P', 'icon\pause.png',\
tip_text, True)
self.pause_action.setEnabled(False)
# Plotting stop
tip_text = 'Stop the plotting. Ctrl+E'
self.stop_action = self.create_action('&End the plotting.', \
self.stop_plot, 'Ctrl+E', 'icon\stop.png',\
tip_text, True)
self.stop_action.setEnabled(False)
# Set the plotting actions into a action group
#plot_group = QActionGroup(self)
#self.add_actions(plot_group, (self.start_action, self.pause_action, \
# self.stop_action))
# Scale the plotting
#=============================================
# Set the tool bar
#=============================================
# the file toolbar
file_tool_bar = self.addToolBar("File")
file_tool_bar.setObjectName("FileToolBar")
self.add_actions(file_tool_bar, (open_file_action, \
save_file_action))
# the bluetooth toolbar
bt_tool_bar = self.addToolBar("Bluetooth")
bt_tool_bar.setObjectName("Bluetooth")
self.add_actions(bt_tool_bar, (bt_settings_action, ))
# the plotting toolbar
plot_tool_bar = self.addToolBar("Plotting")
plot_tool_bar.setObjectName("Plotting")
self.add_actions(plot_tool_bar, (self.start_action, self.pause_action, \
self.stop_action, None))
self.scale_spinbox = QSpinBox()
self.scale_spinbox.setRange(1, 50)
self.scale_spinbox.setValue(5)
self.scale_spinbox.setSuffix('s')
scale_label = QLabel('Seconds to plot.')
self.connect(self.scale_spinbox, SIGNAL('valueChanged(int)'), self.scale_changed)
plot_tool_bar.addWidget(scale_label)
plot_tool_bar.addWidget(self.scale_spinbox)
#=============================================
# Set the menu bar
#=============================================
# the file menu
file_menu = self.menuBar().addMenu('&File')
self.add_actions(file_menu, (open_file_action, save_file_action))
# the bluetooth menu
bt_menu = self.menuBar().addMenu('&Bluetooth')
self.add_actions(bt_menu, (bt_settings_action, ))
class MainWindow(QMainWindow):
''' This is the main window of the BPMonitor.
'''
def __init__(self, parent=None):
''' This is the initializer of the main window
'''
super(MainWindow, self).__init__(parent)
#=======================================
# initial the member field
#=======================================
self.interval_in_seconds = 5
self.q_time_elapsed = QTime(0, 0, 0, 0)
self.timer = QTimer()
self.timer.setSingleShot(False)
self.timer.setInterval(1000)
self.connect(self.timer, SIGNAL("timeout()"), self.updateTheTime)
# Data receiver
self.data_receiver = DataTransfer()
self.serial_settings = {}
# Data holder
self.data_holder = DataHolder()
self.connect(self.data_receiver.qtobj, SIGNAL("NewData"),\
self.data_holder.on_receive_data)
# File helper
self.file_helper = FileHelper()
self.connect(self.data_receiver.qtobj, SIGNAL("NewData"),\
self.file_helper.writeToSignalFile)
# Plot_helpers
self.bp_ecg_plot_helper = PlottingHelper(self, \
BP_ECG_signal_names, 1000)
self.acc_plot_helper = PlottingHelper(self, \
ACC_signal_names, 100)
#=====================================================
# Set central widget
#=====================================================
self.setWindowTitle('UCare BP Monitor')
self.setWindowIcon(QIcon('icon/icon.png'))
# =========================================
# 1: set plot widget
# =========================================
self.plotwidget = QWidget()
self.plotVBox = QVBoxLayout()
# Add plotting widget to the plotVBox
for name in BP_ECG_names:
self.plotVBox.addWidget(self.bp_ecg_plot_helper.curve_plots[name])
label = QLabel(name)
label.setAlignment(Qt.AlignCenter)
self.plotVBox.addWidget(label)
#for key, curve_plot in self.bp_ecg_plot_helper.curve_plots.items():
# self.plotVBox.addWidget(curve_plot)
for key, curve_plot in self.acc_plot_helper.curve_plots.items():
self.plotVBox.addWidget(curve_plot)
label = QLabel(key)
label.setAlignment(Qt.AlignCenter)
self.plotVBox.addWidget(label)
self.plotwidget.setLayout(self.plotVBox)
# =========================================
# 2: set event widget
# =========================================
self.eventWidget = QWidget()
self.eventVBox = QVBoxLayout()
# ============================
# 2-a: set eventButton widget
# ============================
self.eventButtonWidget = EventButtonsWidget()
self.connect(self.eventButtonWidget.qtobj, SIGNAL("NewEvent"), self.addEventLog)
# ============================
# 2-b: set eventLog widget
# ============================
self.eventLogList = QListWidget()
# ============================
# 2-c: add the widgets into eventWidget
# ============================
self.eventVBox.addWidget(self.eventButtonWidget)
self.eventVBox.addWidget(QLabel(u"事件日志"))
self.eventVBox.addWidget(self.eventLogList)
self.eventWidget.setLayout(self.eventVBox)
# =========================================
# 3: add widgets into centralWidget
# =========================================
centralWidget = QWidget()
centralHBox = QHBoxLayout()
centralHBox.addWidget(self.plotwidget)
centralHBox.addWidget(self.eventWidget)
centralWidget.setLayout(centralHBox)
self.setCentralWidget(centralWidget)
self.connect(self.data_holder.qtobj, SIGNAL("UpdateCurve"),\
self.update_curves)
#=======================================
# Set dock windows
#=======================================
#self.
#=======================================
# Set statusbar
#=======================================
self.statusbar = self.statusBar()
self.label_time_elapsed = QLabel()
self.label_time_elapsed.setText(self.q_time_elapsed.toString())
self.statusbar.addPermanentWidget(self.label_time_elapsed)
#=======================================
# Set actions
#=======================================
# Open a file
tip_text = 'Open an existing data file.'
open_file_action = self.create_action('&Open a file', \
self.open_file, QKeySequence.Open, 'icon\open.png',\
tip_text, False)
open_file_action.setEnabled(False)
# Save the file
tip_text = 'Save data into a file.'
save_file_action = self.create_action('&Save into file', \
self.save_file, QKeySequence.Save, 'icon\save.png',\
tip_text, False)
# Bluetooth settings
tip_text = 'Settings of bluetooth. Ctrl+B'
bt_settings_action = self.create_action('&Bluetooth Settings',\
self.bt_settings, 'Ctrl+B', 'icon\settings.png',\
tip_text, False)
# Plotting start
tip_text = 'Start ploting the signals. Ctrl+C'
self.start_action = self.create_action('Start | &Continue plotting.',\
self.start_plot, 'Ctrl+C', 'icon\start.png', \
tip_text, True)
self.start_action.setEnabled(False)
# Plotting pause
tip_text = 'Pause the plotting. Ctrl+P'
self.pause_action = self.create_action('&Pause the plotting.', \
self.pause_plot, 'Ctrl+P', 'icon\pause.png',\
tip_text, True)
self.pause_action.setEnabled(False)
# Plotting stop
tip_text = 'Stop the plotting. Ctrl+E'
self.stop_action = self.create_action('&End the plotting.', \
self.stop_plot, 'Ctrl+E', 'icon\stop.png',\
tip_text, True)
self.stop_action.setEnabled(False)
# Set the plotting actions into a action group
#plot_group = QActionGroup(self)
#self.add_actions(plot_group, (self.start_action, self.pause_action, \
# self.stop_action))
# Scale the plotting
#=============================================
# Set the tool bar
#=============================================
# the file toolbar
file_tool_bar = self.addToolBar("File")
file_tool_bar.setObjectName("FileToolBar")
self.add_actions(file_tool_bar, (open_file_action, \
save_file_action))
# the bluetooth toolbar
bt_tool_bar = self.addToolBar("Bluetooth")
bt_tool_bar.setObjectName("Bluetooth")
self.add_actions(bt_tool_bar, (bt_settings_action, ))
# the plotting toolbar
plot_tool_bar = self.addToolBar("Plotting")
plot_tool_bar.setObjectName("Plotting")
self.add_actions(plot_tool_bar, (self.start_action, self.pause_action, \
self.stop_action, None))
self.scale_spinbox = QSpinBox()
self.scale_spinbox.setRange(1, 50)
self.scale_spinbox.setValue(5)
self.scale_spinbox.setSuffix('s')
scale_label = QLabel('Seconds to plot.')
self.connect(self.scale_spinbox, SIGNAL('valueChanged(int)'), self.scale_changed)
plot_tool_bar.addWidget(scale_label)
plot_tool_bar.addWidget(self.scale_spinbox)
#=============================================
# Set the menu bar
#=============================================
# the file menu
file_menu = self.menuBar().addMenu('&File')
self.add_actions(file_menu, (open_file_action, save_file_action))
# the bluetooth menu
bt_menu = self.menuBar().addMenu('&Bluetooth')
self.add_actions(bt_menu, (bt_settings_action, ))
def addEventLog(self, message):
print message
self.eventLogList.addItem(str(message))
self.file_helper.writeToEventFile(str(message))
def updateTheTime(self):
self.q_time_elapsed = self.q_time_elapsed.addSecs(1)
self.label_time_elapsed.setText(self.q_time_elapsed.toString())
#print "Timer: ", self.q_time_elapsed.toString()
def saveFilesAs(self):
''' Save the files to the right directory with the right name after
cellecting data. '''
formats = ["*.uSig", "*.uEvent"]
fname = QFileDialog.getSaveFileName(self, u"请选择数据保存的位置和名称", \
self.file_helper.getFileName(), "Signal & Event Files: (%s)" % " ".join(formats))
if fname:
fname = str(fname)
if '.uSig' in fname:
fname = fname[0:-5]
if '.uEvent' in fname:
fname = fname[0:-7]
print "FileName getted: " + fname
(dirname, name) = os.path.split(fname)
if not os.path.exists(dirname):
os.path.mkdir(dirname)
os.rename(self.file_helper.getFileName()+'.uSig', fname+'.uSig')
os.rename(self.file_helper.getFileName()+'.uEvent', fname+'.uEvent')
else:
print "Warning: Files are not saved."
def add_actions(self, target, actions):
'''helps add actions to the target, which is either a menubar or \
a toolbar.'''
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def update_curves(self):
''' update the curves'''
if self.start_action.isChecked():
self.bp_ecg_plot_helper.update_curves(self.data_holder.time_seq,\
self.data_holder.signals, \
self.interval_in_seconds)
self.acc_plot_helper.update_curves(self.data_holder.time_seq\
[0:len(self.data_holder.time_seq):10],\
self.data_holder.signals, \
self.interval_in_seconds)
def create_action(self, text, slot=None, shortcut=None,
icon=None, tip=None, checkable=False,
signal="triggered()"):
'''This is a action_helper helps to create an action'''
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(icon))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
self.connect(action, SIGNAL(signal), slot)
if checkable:
action.setCheckable(True)
return action
def open_file(self):
''' Open an existing data file'''
print 'MainWindow.open_file() was triggered.'
def save_file(self):
''' Save the data received from bluetooth into a file'''
print 'MainWindow.save_file() was triggered.'
def bt_settings(self):
''' Let user set the parameters of bluetooth connection.'''
print 'MainWindow.bt_settings() was triggered.'
dialog = SerialSettingsDialog(self)
if dialog.exec_():
self.serial_settings = dialog.settings()
print self.serial_settings
if self.data_receiver.open(self.serial_settings):
if self.data_receiver.serial.isOpen():
print 'Open serial successfully'
self.start_action.setEnabled(True)
sleep(0.1)
else:
print "Serial port is not open."
else:
print "Unable to open serial port."
def start_plot(self):
''' Start or restart plotting the signals.'''
#if self.start_action.isChecked():
# print "Already started."
# return
print 'MainWindow.start_plot() was triggered.'
if self.serial_settings == {}:
#Show a dialog here!!!
self.statusBar().showMessage('Set the serial first.')
else:
self.file_helper.openFilesToWrite()
self.eventButtonWidget.startLogging()
self.data_receiver.startTransfer()
self.data_receiver.start()
self.timer.start()
self.startTime = QDateTime.currentMSecsSinceEpoch()
self.stop_action.setEnabled(True)
self.statusBar().showMessage('Start plotting.')
self.start_action.setEnabled(False)
def pause_plot(self):
''' Pause plotting the signals.'''
print 'MainWindow.pause_plot() was triggered.'
if not self.start_action.isChecked():
self.pause_action.setChecked(False)
return
else:
self.data_receiver.pauseTransfer()
self.pause_action.setChecked(True)
self.statusBar().showMessage('Pause plotting.')
def stop_plot(self):
''' Stop plotting the signals.'''
print 'MainWindow.start_plot() was triggered.'
self.data_receiver.pauseTransfer()
self.data_receiver.close()
self.eventButtonWidget.endLogging()
self.timer.stop()
self.file_helper.closeFiles()
self.statusBar().showMessage('Stop plotting.')
self.saveFilesAs()
self.start_action.setEnabled(True)
self.stop_action.setEnabled(False)
def scale_changed(self):
''' Update the scale factor.'''
self.interval_in_seconds = self.scale_spinbox.value()
print ' scale_spinbox triggered. Plot %d seconds.' \
% self.interval_in_seconds
def closeEvent(self, event):
self.file_helper.closeFiles()
self.statusBar().showMessage('Close.')
print "closeEvent."
