Adding attribute

Author: Ryan Calme

Diff for: Arduino/Puzzle_Box/Connection.cpp

@@ -1,41 +1,50 @@
 #include "Connection.h"
 
-// Constructors
+// Empty constructor
 Connection::Connection() {
-    left = 255;
-    right = 255;
+  left = 0;
+  right = 0;
+  connected = false;
 }
 
-// Create via provided values
+// Constructor with provided pegs
 Connection::Connection(byte a, byte b) {
-    // Ensure that the smaller of the
-    // two values is the left side
-    if (a < b) {
-        left = a;
-        right = b;
-    }
-    else {
-        left = b;
-        right = a;
-    }
+  // Ensure that the smaller of the
+  // two values is the left side
+  if (a < b) {
+    left = a;
+    right = b;
+    connected = true;
+  } else if (b > a) {
+    left = b;
+    right = a;
+    connected = true;
+  } else {
+    left = a;
+    right = a;
+    connected = false;
+  }
 }
 
 // Overload equals comparator
 boolean Connection::operator==(const Connection &rhs) {
-    return (left==rhs.left && right==rhs.right);
+  return (left==rhs.left && right==rhs.right);
 }
 
 // Returns true if connection is connected
 boolean Connection::isConnected() {
-    return (left != 255 && right != 255);
+  return connected;
 }
 
 // Prints connection to Serial port
 void Connection::print() {
+  if (connected) {
     Serial.print("(");
     Serial.print(left);
     Serial.print(",");
     Serial.print(right);
     Serial.print(")");
+  } else {
+    Serial.print("Not connected");
+  }
 }
-

Diff for: Arduino/Puzzle_Box/Connection.h

@@ -7,15 +7,13 @@
 
   "left", and "right", are the two pegs participating
   in the connection.
-
-  Values of [-1, -1] for left and right, imply no active
-  connection
 */
 
 class Connection {
   private:
     byte left;
     byte right;
+    boolean connected;
 
   public:
     // Constructors