Merge pull request #21 from SoonerRobotics/feat-businout

Feature BusInOut

Author: jkleiber

arduino/Arduino.h

@@ -12,6 +12,7 @@ using namespace std;
 
 #define INPUT 0x0
 #define OUTPUT 0x1
+#define INPUT_PULLUP 0x2
 
 #define true 0x1
 #define false 0x0

src/BusInOut.cpp

@@ -0,0 +1,36 @@
+#include "BusInOut.h"
+
+BusInOut::BusInOut(int p0, int p1, int p2, int p3,
+                int p4, int p5, int p6, int p7,
+                int p8, int p9, int p10, int p11,
+                int p12, int p13, int p14, int p15) {
+    
+    int pins[] = {p0, p1, p2, p3, p4, p5, p6, p7, p8, p9, p10, p11, p12, p13, p14, p15};
+    for (int i=0; i<16; i++) {
+        _pin[i] = pins[i];
+    }
+}
+
+void BusInOut::mode(int mode) {
+    for (int i=0; i<16; i++) {
+        if (_pin[i] != -1)
+            pinMode(_pin[i], mode);
+    }
+}
+
+void BusInOut::write(int value) {
+    for (int i=0; i<16; i++) {
+        if (_pin[i] != -1) {
+            digitalWrite(_pin[i], (value >> i) & 1);
+        }
+    }
+}
+
+int BusInOut::read() {
+    int value = 0;
+    for (int i=0; i<16; i++) {
+        if (_pin[i] != -1)
+            value |= digitalRead(_pin[i]) << i;
+    }
+    return value;
+}

src/BusInOut.h

@@ -0,0 +1,24 @@
+#ifndef BusInOut_H
+#define BusInOut_H
+
+#include <Arduino.h>
+
+class BusInOut
+{
+
+    public:
+        BusInOut(int p0, int p1 = -1, int p2 = -1, int p3 = -1,
+             int p4 = -1, int p5 = -1, int p6 = -1, int p7 = -1,
+             int p8 = -1, int p9 = -1, int p10 = -1, int p11 = -1,
+             int p12 = -1, int p13 = -1, int p14 = -1, int p15 = -1);
+
+        void mode(int mode);
+
+        void write(int value);
+        int read();
+
+    private:
+        int _pin[16];
+};
+
+#endif

tests/test_businout.cpp

@@ -0,0 +1,39 @@
+#include <BusInOut.h>
+#include <catch.hpp>
+
+TEST_CASE("Constructors and mode") 
+{
+    BusInOut test(1,2,4);
+    test.mode(INPUT_PULLUP);
+
+    REQUIRE(getModeAtPin(1) == INPUT_PULLUP);
+    REQUIRE(getModeAtPin(2) == INPUT_PULLUP);
+    REQUIRE(getModeAtPin(4) == INPUT_PULLUP);
+}
+
+TEST_CASE("Write test") 
+{
+    BusInOut test(1,2,4,6,8);
+    test.mode(OUTPUT);
+    test.write(0b10110);
+
+    REQUIRE(getValueAtPin(1) == 0);
+    REQUIRE(getValueAtPin(2) == 1);
+    REQUIRE(getValueAtPin(4) == 1);
+    REQUIRE(getValueAtPin(6) == 0);
+    REQUIRE(getValueAtPin(8) == 1);
+}
+
+TEST_CASE("Read test") 
+{
+    BusInOut test(1,2,4,6,8);
+    test.mode(INPUT);
+    
+    setValueAtPin(1, 0);
+    setValueAtPin(2, 1);
+    setValueAtPin(4, 1);
+    setValueAtPin(6, 0);
+    setValueAtPin(8, 1);
+
+    REQUIRE(test.read() == 0b10110);
+}