Added servo support

Author: Alexander Akhmetov

arduino/arduino.py

@@ -48,6 +48,16 @@ def analogRead(self, pin):
         self.__sendData(pin)
         return self.__getData()
 
+    def servoAdd(self, pin):
+        self.__sendData('5')
+        self.__sendData(pin)
+        return True
+
+    def servoWrite(self, pin):
+        self.__sendData('6')
+        self.__sendData(pin)
+        return True
+
     def turnOff(self):
         for each_pin in self.__OUTPUT_PINS:
             self.setLow(each_pin)

prototype.pde

@@ -1,51 +1,60 @@
-
-#ifndef SERIAL_RATE
-#define SERIAL_RATE         115200
-#endif
-
-#ifndef SERIAL_TIMEOUT
-#define SERIAL_TIMEOUT      5
-#endif
-
-void setup() {
-    Serial.begin(SERIAL_RATE);
-    Serial.setTimeout(SERIAL_TIMEOUT);
-
-    int cmd = readData();
-    for (int i = 0; i < cmd; i++) {
-        pinMode(readData(), OUTPUT);
-    }
-}
-
-void loop() {
-    switch (readData()) {
-        case 0 :
-            //set digital low
-            digitalWrite(readData(), LOW); break;
-        case 1 :
-            //set digital high
-            digitalWrite(readData(), HIGH); break;
-        case 2 :
-            //get digital value
-            Serial.println(digitalRead(readData())); break;
-        case 3 :
-            // set analog value
-            analogWrite(readData(), readData()); break;
-        case 4 :
-            //read analog value
-            Serial.println(analogRead(readData())); break;
-        case 99:
-            //just dummy to cancel the current read, needed to prevent lock 
-            //when the PC side dropped the "w" that we sent
-            break;
-    }
-}
-
-char readData() {
-    Serial.println("w");
-    while(1) {
-        if(Serial.available() > 0) {
-            return Serial.parseInt();
-        }
-    }
-}
+#include <Servo.h> 
+
+#ifndef SERIAL_RATE
+#define SERIAL_RATE         115200
+#endif
+
+#ifndef SERIAL_TIMEOUT
+#define SERIAL_TIMEOUT      5
+#endif
+
+Servo myservo;
+
+void setup() {
+    Serial.begin(SERIAL_RATE);
+    Serial.setTimeout(SERIAL_TIMEOUT);
+
+    int cmd = readData();
+    for (int i = 0; i < cmd; i++) {
+        pinMode(readData(), OUTPUT);
+    }
+}
+
+void loop() {
+    switch (readData()) {
+        case 0 :
+            //set digital low
+            digitalWrite(readData(), LOW); break;
+        case 1 :
+            //set digital high
+            digitalWrite(readData(), HIGH); break;
+        case 2 :
+            //get digital value
+            Serial.println(digitalRead(readData())); break;
+        case 3 :
+            // set analog value
+            analogWrite(readData(), readData()); break;
+        case 4 :
+            //read analog value
+            Serial.println(analogRead(readData())); break;
+        case 5 :
+            //add servo
+            myservo.attach(readData()); break;
+        case 6 :
+            //write servo position
+            myservo.write(readData()); break;
+        case 99:
+            //just dummy to cancel the current read, needed to prevent lock 
+            //when the PC side dropped the "w" that we sent
+            break;
+    }
+}
+
+char readData() {
+    Serial.println("w");
+    while(1) {
+        if(Serial.available() > 0) {
+            return Serial.parseInt();
+        }
+    }
+}