now discovering address of mindwave mobile

Author: robintibor

README.md

@@ -6,9 +6,11 @@ Requirements:
 Usage in python:
 
 ```python
+from mindwavemobile.MindwaveDataPointReader import MindwaveDataPointReader
 mindwaveDataPointReader = MindwaveDataPointReader()
 # connect to the mindwave mobile headset...
 mindwaveDataPointReader.start()
 # read one data point, data point types are specified in  MindwaveDataPoints.py'
 dataPoint = mindwaveDataPointReader.readNextDataPoint()
+print dataPoint
 ``` 

example/read_mindwave_mobile.py

@@ -2,15 +2,18 @@
 import bluetooth
 from mindwavemobile.MindwaveDataPoints import RawDataPoint
 from mindwavemobile.MindwaveDataPointReader import MindwaveDataPointReader
-
+import textwrap
 
 if __name__ == '__main__':
     mindwaveDataPointReader = MindwaveDataPointReader()
     mindwaveDataPointReader.start()
-    
-    while(True):
-        dataPoint = mindwaveDataPointReader.readNextDataPoint()
-        if (not dataPoint.__class__ is RawDataPoint):
-            print dataPoint
-
-        
+    if (mindwaveDataPointReader.isConnected()):    
+        while(True):
+            dataPoint = mindwaveDataPointReader.readNextDataPoint()
+            if (not dataPoint.__class__ is RawDataPoint):
+                print dataPoint
+    else:
+        print(textwrap.dedent("""\
+            Exiting because the program could not connect
+            to the Mindwave Mobile device.""").replace("\n", " "))
+        

mindwavemobile/MindwaveDataPointReader.py

@@ -11,7 +11,10 @@ def __init__(self):
 
     def start(self):
         self._mindwaveMobileRawReader.connectToMindWaveMobile()
-        
+
+    def isConnected(self):
+        return self._mindwaveMobileRawReader.isConnected()
+
     def readNextDataPoint(self):
         if (not self._moreDataPointsInQueue()):
             self._putNextDataPointsInQueue()
@@ -76,4 +79,4 @@ def _readDataPointsFromPayload(self, payloadBytes):
 
 
 
-    
+    

mindwavemobile/MindwaveMobileRawReader.py

@@ -1,25 +1,55 @@
 import bluetooth
 import time
+import textwrap
 
 
 class MindwaveMobileRawReader:
     START_OF_PACKET_BYTE = 0xaa;
     def __init__(self):
         self._buffer = [];
         self._bufferPosition = 0;
+        self._isConnected = False;
 
     def connectToMindWaveMobile(self):
-        # connecting via bluetooth RFCOMM
+        # First discover mindwave mobile address, then connect
+        # headset address of me was'9C:B7:0D:72:CD:02';
+        # not sure if it really can be different?
+        # now discovering address because of https://github.com/robintibor/python-mindwave-mobile/issues/4
+        mindwaveMobileAddress = self._findMindwaveMobileAddress()
+        if (mindwaveMobileAddress is not None):            
+            print ("Discovered Mindwave Mobile...")
+            self._connectToAddress(mindwaveMobileAddress)
+        else:
+            self._printErrorDiscoveryMessage()
+        
+    def _findMindwaveMobileAddress(self):
+        nearby_devices = bluetooth.discover_devices(lookup_names = True)
+        for address, name in nearby_devices:
+            if (name == "MindWave Mobile"):
+                return address
+        return None
+        
+    def _connectToAddress(self, mindwaveMobileAddress):
         self.mindwaveMobileSocket = bluetooth.BluetoothSocket(bluetooth.RFCOMM)
-        mindwaveMobileAddress = '9C:B7:0D:72:CD:02';
-        while(True):
+        while (not self._isConnected):
             try:
-                self.mindwaveMobileSocket.connect((mindwaveMobileAddress, 1))
-                return;
+                self.mindwaveMobileSocket.connect(
+                    (mindwaveMobileAddress, 1))
+                self._isConnected = True
             except bluetooth.btcommon.BluetoothError as error:
                 print "Could not connect: ", error, "; Retrying in 5s..."
                 time.sleep(5) 
-    
+           
+
+    def isConnected(self):
+        return self._isConnected
+
+    def _printErrorDiscoveryMessage(self):
+         print(textwrap.dedent("""\
+                    Could not discover Mindwave Mobile. Please make sure the
+                    Mindwave Mobile device is in pairing mode and your computer
+                    has bluetooth enabled.""").replace("\n", " "))
+
     def _readMoreBytesIntoBuffer(self, amountOfBytes):
         newBytes = self._readBytesFromMindwaveMobile(amountOfBytes)
         self._buffer += newBytes
@@ -67,4 +97,4 @@ def clearAlreadyReadBuffer(self):
     def _bufferSize(self):
         return len(self._buffer);
 
-#------------------------------------------------------------------------------ 
+#------------------------------------------------------------------------------