summaryrefslogtreecommitdiff
path: root/Python code
diff options
context:
space:
mode:
authorKevin Chabowski <kevin@kch42.de>2011-03-22 22:15:22 +0100
committerKevin Chabowski <kevin@kch42.de>2011-03-22 22:15:22 +0100
commit4cdf144c41c7b2d3100251e5a47e35f98cb09c30 (patch)
tree1e32cf43f05be71bea151ee9f4713cc6a7d61053 /Python code
parentaedc42f9ff57eb7fb456ef6fd9e94c099ff84076 (diff)
downloadSerialRGB-LED-Arduino-4cdf144c41c7b2d3100251e5a47e35f98cb09c30.tar.gz
SerialRGB-LED-Arduino-4cdf144c41c7b2d3100251e5a47e35f98cb09c30.tar.bz2
SerialRGB-LED-Arduino-4cdf144c41c7b2d3100251e5a47e35f98cb09c30.zip
Added synchronisation with Arduino.
Diffstat (limited to 'Python code')
-rwxr-xr-xPython code/rgbled.py17
-rw-r--r--Python code/serialrgb.py18
2 files changed, 30 insertions, 5 deletions
diff --git a/Python code/rgbled.py b/Python code/rgbled.py
index 8d699b4..8378577 100755
--- a/Python code/rgbled.py
+++ b/Python code/rgbled.py
@@ -4,7 +4,7 @@
from __future__ import division
import wx, math
from wx.lib.colourchooser.pycolourchooser import PyColourChooser
-from serialrgb import SerialRGB
+from serialrgb import SerialRGB, SyncError
def hsv2rgb(h, s, v):
h *= 360.0
@@ -63,14 +63,23 @@ class rgbled_frame(wx.Frame):
g = colour.Green()
b = colour.Blue()
if isinstance(self.rgbled, SerialRGB):
- self.rgbled.change_color((r, g, b))
+ try:
+ self.rgbled.change_color((r, g, b))
+ except SyncError:
+ wx.MessageDialog(None, "Could not synchronize with Arduino!", "Sync Error", wx.ICON_ERROR | wx.OK).ShowModal()
+ self.Close()
class rgbled_app(wx.App):
def OnInit(self):
self.rgbled = None
portdlg = wx.TextEntryDialog(None, "Serial port:")
if portdlg.ShowModal() == wx.ID_OK:
- self.rgbled = SerialRGB(portdlg.GetValue())
+ try:
+ self.rgbled = SerialRGB(portdlg.GetValue())
+ except SyncError:
+ pass
+ # wx.MessageDialog(None, "Could not synchronize with Arduino!", "Sync Error", wx.ICON_ERROR | wx.OK).ShowModal()
+ # return False
frame = rgbled_frame(self.rgbled)
portdlg.Destroy()
frame.Show()
@@ -78,7 +87,7 @@ class rgbled_app(wx.App):
return True
def OnExit(self):
if isinstance(self.rgbled,SerialRGB):
- self.rgb2hsv.close_connection()
+ self.rgbled.close_connection()
if __name__ == '__main__':
myapp = rgbled_app()
diff --git a/Python code/serialrgb.py b/Python code/serialrgb.py
index 7296e22..8d7c6d8 100644
--- a/Python code/serialrgb.py
+++ b/Python code/serialrgb.py
@@ -1,7 +1,15 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
-import serial
+import serial, time
+
+class SyncError(Exception):
+ """Synchronisation error"""
+ def __init__(self, msg=""):
+ self._msg = str(msg)
+
+ def __str__(self):
+ return msg
class SerialRGB(object):
"""Easy controlling of the RGB-LED / Arduino"""
@@ -16,6 +24,12 @@ class SerialRGB(object):
self.ser = serial.Serial(addr, baud)
except:
raise IOError("Could not connect to Arduino via serial port.")
+ # Sync...
+ while self.ser.inWaiting() < 1:
+ self.ser.write("\x00")
+ time.sleep(.01)
+ if self.ser.read(1) != "1":
+ raise SyncError
def __del__(self):
self.close_connection()
@@ -28,6 +42,8 @@ class SerialRGB(object):
"""
r, g, b = color
self.ser.write(chr(r) + chr(g) + chr(b))
+ if self.ser.read(1) != "1":
+ raise SyncError
def close_connection(self):
"""Closes the connection to the Arduino."""