source: remote_control/puzzlebox_brainstorms_client_interface.py @ 77

Last change on this file since 77 was 77, checked in by sc, 11 years ago

client_interface:

  • updated to handle EEG Headset ThinkGear? connection

qt4_form:

  • updated with Speed indication bar
File size: 7.1 KB
Line 
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3#
4# Puzzlebox - Brainstorms - Client Interface - Qt
5#
6# Copyright Puzzlebox Productions, LLC (2010)
7#
8# This code is released under the GNU Pulic License (GPL) version 2
9# For more information please refer to http://www.gnu.org/copyleft/gpl.html
10#
11# Last Update: 2010.06.29
12#
13#####################################################################
14
15import os, sys
16
17#try:
18        #from PySide import QtCore, QtGui, QtNetwork
19#except:
20        #print "Using PyQt4 module"
21        #from PyQt4 import QtCore, QtGui, QtNetwork
22#else:
23        #print "Using PySide module"
24
25from PyQt4 import QtCore, QtGui, QtNetwork
26#from PySide import QtCore, QtGui, QtNetwork
27
28sys.path.append(os.path.join(os.getcwd(), "interface"))
29from qt4_form import Ui_Form
30
31import simplejson as json
32
33import puzzlebox_brainstorms_configuration as configuration
34import puzzlebox_brainstorms_network_client as brainstorms_client
35import puzzlebox_brainstorms_network_client_thinkgear as thinkgear_client
36#import puzzlebox_logger
37
38#####################################################################
39# Globals
40#####################################################################
41
42DEBUG = 1
43
44THINKGEAR_POWER_THRESHOLDS = configuration.THINKGEAR_POWER_THRESHOLDS
45
46NXT_BLUETOOTH_DEVICE = configuration.NXT_BLUETOOTH_DEVICE
47
48THINKGEAR_SERVER_HOST = configuration.THINKGEAR_SERVER_HOST
49THINKGEAR_SERVER_PORT = configuration.THINKGEAR_SERVER_PORT
50
51#####################################################################
52# Classes
53#####################################################################
54
55class puzzlebox_brainstorms_client_interface(QtGui.QWidget, Ui_Form):
56       
57        def __init__(self, log, DEBUG=DEBUG, parent = None):
58               
59                self.log = log
60                self.DEBUG = DEBUG
61               
62                QtGui.QWidget.__init__(self, parent)
63                self.setupUi(self)
64               
65                self.configureSettings()
66                self.configureNetworkBrainstorms()
67                self.connectWidgets()
68       
69       
70        ##################################################################
71       
72        def configureSettings(self):
73               
74                # LEGO Mindstorms
75               
76                # NXT Bluetooth readout current not available
77                self.textLabelNXTStatus.setText("Status: N/A")
78               
79                # Display communication port for LEGO Mindstorms NXT device
80                self.lineEditNXTPort.setText(NXT_BLUETOOTH_DEVICE)
81                self.lineEditNXTPort.setEnabled(False)
82               
83                # NXT Bluetooth connection control not available
84                self.pushButtonNXTConnect.setEnabled(False)
85               
86               
87                # EEG Headset
88               
89                # Display Host for ThinkGear Connect Socket Server
90                self.lineEditThinkGearHost.setText(THINKGEAR_SERVER_HOST)
91                self.lineEditThinkGearHost.setEnabled(False)
92               
93                # Display Port for ThinkGear Connect Socket Server
94                self.lineEditThinkGearPort.setText('%i' % THINKGEAR_SERVER_PORT)
95                self.lineEditThinkGearPort.setEnabled(False)
96               
97               
98                # Thresholds Removed
99               
100                # Set concentration threshold value
101                #concentration_threshold = \
102                        #self.getMinimumThreshold(THINKGEAR_POWER_THRESHOLDS['concentration'])
103                #self.spinBoxConcentrationThreshold.setValue(concentration_threshold)
104                #self.spinBoxConcentrationThreshold.update()
105               
106                # Concentration threshold value not modifiable
107                #self.spinBoxConcentrationThreshold.setEnabled(False)
108               
109                # Set relaxation threshold value
110                #relaxation_threshold = \
111                        #self.getMinimumThreshold(THINKGEAR_POWER_THRESHOLDS['relaxation'])
112                #self.spinBoxRelaxationThreshold.setValue(relaxation_threshold)
113                #self.spinBoxRelaxationThreshold.update()
114               
115                # Relaxation threshold value not modifiable
116                #self.spinBoxRelaxationThreshold.setEnabled(False)
117       
118       
119        ##################################################################
120       
121        def getMinimumThreshold(self, threshold):
122               
123                '''Return the minimum detection level which results
124                in a non-zero power setting'''
125               
126                minimum = 100
127               
128                threshold_keys = threshold.keys()
129                threshold_keys.sort()
130                threshold_keys.reverse()
131               
132                for key in threshold_keys:
133                       
134                        if ((threshold[key] < minimum) and \
135                                 (threshold[key] > 0)):
136                                minimum = key
137               
138               
139                return(minimum)
140       
141       
142        ##################################################################
143       
144        def configureNetworkBrainstorms(self):
145               
146                self.brainstormsClient = \
147                   brainstorms_client.puzzlebox_brainstorms_network_client(self.log, parent=self)
148       
149       
150        ##################################################################
151       
152        def configureNetworkThinkGear(self):
153               
154                self.thinkgearClient = \
155                   thinkgear_client.puzzlebox_brainstorms_network_client_thinkgear(self.log, parent=self)
156               
157                self.pushButtonThinkGearConnect.setText('Connected')
158                self.pushButtonThinkGearConnect.setEnabled(False)
159       
160       
161        ##################################################################
162       
163        def connectWidgets(self):
164               
165                self.connect(self.pushButtonTurnLeft, QtCore.SIGNAL("pressed()"), \
166                             self.turnLeft)
167                self.connect(self.pushButtonTurnLeft, QtCore.SIGNAL("released()"), \
168                             self.stopMotors)
169               
170                self.connect(self.pushButtonForward, QtCore.SIGNAL("pressed()"), \
171                             self.driveForward)
172                self.connect(self.pushButtonForward, QtCore.SIGNAL("released()"), \
173                             self.stopMotors)
174               
175                self.connect(self.pushButtonTurnRight, QtCore.SIGNAL("pressed()"), \
176                             self.turnRight)
177                self.connect(self.pushButtonTurnRight, QtCore.SIGNAL("released()"), \
178                             self.stopMotors)
179               
180                self.connect(self.pushButtonTurnLeftReverse, QtCore.SIGNAL("pressed()"), \
181                             self.turnLeftInReverse)
182                self.connect(self.pushButtonTurnLeftReverse, QtCore.SIGNAL("released()"), \
183                             self.stopMotors)
184               
185                self.connect(self.pushButtonReverse, QtCore.SIGNAL("pressed()"), \
186                             self.driveReverse)
187                self.connect(self.pushButtonReverse, QtCore.SIGNAL("released()"), \
188                             self.stopMotors)
189               
190                self.connect(self.pushButtonTurnRightReverse, QtCore.SIGNAL("pressed()"), \
191                             self.turnRightInReverse)
192                self.connect(self.pushButtonTurnRightReverse, QtCore.SIGNAL("released()"), \
193                             self.stopMotors)
194               
195               
196                self.connect(self.pushButtonThinkGearConnect, \
197                                  QtCore.SIGNAL("clicked()"), \
198                                  self.configureNetworkThinkGear)
199       
200       
201        ##################################################################
202       
203        def turnLeft(self):
204                self.brainstormsClient.sendCommand('turn_left')
205       
206        def driveForward(self):
207                self.brainstormsClient.sendCommand('drive_forward')
208       
209        def turnRight(self):
210                self.brainstormsClient.sendCommand('turn_right')
211       
212        def turnLeftInReverse(self):
213                self.brainstormsClient.sendCommand('turn_left_in_reverse')
214       
215        def driveReverse(self):
216                self.brainstormsClient.sendCommand('drive_reverse')
217       
218        def turnRightInReverse(self):
219                self.brainstormsClient.sendCommand('turn_right_in_reverse')
220       
221        def stopMotors(self):
222                self.brainstormsClient.sendCommand('stop_motors')
223       
224       
225        ##################################################################
226       
227        def test(self):
228               
229                print "Test Successful"
230
231
232#####################################################################
233# Functions
234#####################################################################
235
236#####################################################################
237# Main
238#####################################################################
239
240if __name__ == '__main__':
241       
242        log = None
243       
244        app = QtGui.QApplication(sys.argv)
245       
246        window = puzzlebox_brainstorms_client_interface(log, DEBUG)
247        window.show()
248       
249        app.exec_()
250
Note: See TracBrowser for help on using the repository browser.