Diff: two_player.h

Revision:

3:3ddefff03cb2

Parent:

2:c358ad9aedc4

Child:

4:d8bd8b41468d

--- a/two_player.h	Mon Oct 12 04:23:24 2015 +0000
+++ b/two_player.h	Mon Oct 12 08:20:31 2015 +0000
@@ -1,6 +1,7 @@
 #ifndef TWO_PLAYER_H__
 #define TWO_PLAYER_H__
 
+#include <cmath>
 #include "uLCD_4DGL.h"
 #include "EthernetInterface.h"
 
@@ -22,17 +23,21 @@
 public :
 
     int p1_p2;
+
     enum Color
     {
+        CLS_CMD,
+        BG_COLOR_CMD,
         LINE_CMD,
-        BG_COLOR_CMD,
         CIRCLE_CMD,
-        CLS_CMD,
-        I_AM_P1,
-        I_AM_P2
+        FILLED_CIRCLE_CMD,
+        TRI_CMD,
+        RECT_CMD,
+        FILLED_RECT_CMD,
+        PIX_CMD,
+        PEN_SZ_CMD
     };
 
-
     uLCD_2P(PinName tx, PinName rx, PinName rst, bool player) {
         p1_p2 = player;
         //LCD = new uLCD_4DGL (tx, rx, rst);        
@@ -43,78 +48,116 @@
         buffer_idx = 0;
         memset(buttons, 0, sizeof(buttons));
         
-        sock = new UDPSocket();
-        endpoint = new Endpoint();
-        eth = new EthernetInterface();
-        
         switch (p1_p2) {
             case PLAYER1:           // Client
+                
+                eth = new EthernetInterface();
                 eth->init(PLAYER1_IP, "255.255.255.0", "0.0.0.0"); 
                 eth->connect();
-                sock->init();
-                endpoint->set_address(PLAYER2_IP, SERVER_PORT);
+                
+                sock = new TCPSocketConnection();
+                while(sock->connect(PLAYER2_IP, SERVER_PORT) < 0) {
+                    pc.printf("Trying to connect.\n");
+                    //wait(1);
+                }
+                
                 break;
             case PLAYER2:           // Server
+                eth = new EthernetInterface();
                 eth->init(PLAYER2_IP, "255.255.255.0", "0.0.0.0"); 
                 eth->connect();
-                sock->bind(SERVER_PORT);  
+
+                server = new TCPSocketServer();
+                server->bind(SERVER_PORT);
+                server->listen();
+                sock = new TCPSocketConnection();
+                server->accept(*sock);
+                sock->set_blocking(false, 1500);
                 break;   
         }
+        pc.printf("Ethernet Connected.\n");
     }
     
-    void draw(int CMD, int a, int b, int c, int d, int e, char nArgs){
+    // Yes, this sucks. If you're smart, find a good way to do variable args and show me!
+    // Look into template metaprogramming!
+    
+    void  draw(int CMD) {_draw(CMD, 0,0,0,0,0,0,0, 0); }
+    void  draw(int CMD, int a) { _draw(CMD, a, 0,0,0,0,0,0, 1); }
+    void  draw(int CMD, int a, int b) { _draw(CMD, a, b, 0,0,0,0,0, 2); }
+    void  draw(int CMD, int a, int b, int c) { _draw(CMD, a, b, c, 0,0,0,0, 3); }
+    void  draw(int CMD, int a, int b, int c, int d) { _draw(CMD, a, b, c, d, 0,0,0, 4); }
+    void  draw(int CMD, int a, int b, int c, int d, int e) { _draw(CMD, a, b, c, d, e, 0,0, 5); }
+    void  draw(int CMD, int a, int b, int c, int d, int e, int f) { _draw(CMD, a, b, c, d, e, f, 0, 6); }
+    void  draw(int CMD, int a, int b, int c, int d, int e, int f, int g) { _draw(CMD, a, b, c, d, e, f, g, 7); }
+    
+    void _draw(int CMD, int a, int b, int c, int d, int e, int f, int g, char nArgs){
         
         // Deal with overflows
-        //if(buffer_idx + nArgs > 256) {//flush();}
+        // if(buffer_idx + nArgs > 256) {//flush();}
+        
+        if(nArgs > 7) { 
+            pc.printf("Error in call to _draw(): nArgs > 7 not allowed!\n");
+            return;
+        }
         
         buffer[buffer_idx] = CMD;
         if(nArgs >= 1) buffer[buffer_idx+1] = a;
         if(nArgs >= 2) buffer[buffer_idx+2] = b;
         if(nArgs >= 3) buffer[buffer_idx+3] = c;
         if(nArgs >= 4) buffer[buffer_idx+4] = d;
-        if(nArgs == 5) buffer[buffer_idx+5] = e;
-        // ERROR: nArgs > 5
+        if(nArgs >= 5) buffer[buffer_idx+5] = e;
+        if(nArgs >= 6) buffer[buffer_idx+6] = f;
+        if(nArgs >= 7) buffer[buffer_idx+7] = g;
+        // ERROR: nArgs > 7
         
         
         buffer_idx += nArgs+1;
     }
     
-    void cls(void) { draw(CLS_CMD, 0,0,0,0,0,0); }
-    void background_color(int color) {
-        draw(BG_COLOR_CMD, color, 0,0,0,0,1);
-    }
+   
+    void background_color(int color) { draw(BG_COLOR_CMD, color); }
+    void line(int sx, int sy, int ex, int ey, int color) { draw(LINE_CMD, sx, sy, ex, ey, color); }
+    void circle(int x , int y , int radius, int color) { draw(CIRCLE_CMD, x, y, radius, color); }
+    void filled_circle(int x , int y , int radius, int color) { draw(FILLED_CIRCLE_CMD, x, y, radius, color); }
+    void triangle(int a, int b, int c, int d , int e, int f, int col) { draw(TRI_CMD, a, b, c, d, e, f, col); }
+    void rectangle(int a, int b, int c, int d, int col) { draw(RECT_CMD, a, b, c, d, col); }
+    void filled_rectangle(int a, int b, int c, int d, int col) { draw(FILLED_RECT_CMD, a, b, c, d, col); }
+    void pixel(int a, int b, int col) { draw(PIX_CMD, a, b, col); }
+    void cls(void) { draw(CLS_CMD); }
+    
+    void pen_size(char sz) { draw(PEN_SZ_CMD, sz); }
+    /*void BLIT(int x1, int y1, int x2, int y2, int *colors) {
+        int size = abs((x1-x2)*(y1-y2));
+        // pad to a multiple of 4 and memset buffer...
+        // I'll get to it later.
+    }*/
+
+    // Reads don't need to be done on the slave side. Hopefully both sides match!
+    int  read_pixel(int x, int y) { return LCD->read_pixel(x, y); }
+    
     void set_button_state(int a, int b, int c, int d, int e) { 
         buttons[0] = a; buttons[1] = b; buttons[2] = c; 
         buttons[3] = d; buttons[4] = e;   
     }
+    
     int* get_button_state() {
         return buttons;
     }
-    void line(int sx, int sy, int ex, int ey, int color) {
-        //draw(LINE_CMD, sx, sy, ex, ey, color);
-    }
     
-
     void update() {
         sock->set_blocking(true, 100);
         if(p1_p2 == PLAYER1) {
-            pc.printf("Sending draw data...\n");  
-            sock->sendTo(*endpoint, (char*)buffer, (buffer_idx+1)*sizeof(buffer[0]));
+            sock->send_all((char*)buffer, (buffer_idx+1)*sizeof(buffer[0]));
             buffer_idx = 0;
-            
-            pc.printf("Receiving button data...\n");  
-            int n = sock->receiveFrom(*endpoint, (char*)buttons, sizeof(buttons));
-            if(n < 0) {pc.printf("RECEIVE ERROR.\n");}     
-            pc.printf("Received button data...\n");         
+ 
+            int n = sock->receive((char*)buttons, sizeof(buttons));
+            if(n < 0) {pc.printf("RECEIVE ERROR.\n");}          
         
-        }else if(p1_p2 == PLAYER2) {
-            pc.printf("Receiving draw buffer...\n");      
-            int n = sock->receiveFrom(*endpoint, (char*)buffer, sizeof(buffer));  
+        }else if(p1_p2 == PLAYER2) {    
+            int n = sock->receive((char*)buffer, sizeof(buffer));  
             if(n < 0) {pc.printf("RECEIVE ERROR.\n");}
             buffer[n] = '\0';   
-            pc.printf("Received draw data...\n");  
-            pc.printf("Sending button data...\n");  
-            sock->sendTo(*endpoint, (char*)buttons, sizeof(buttons));          
+            sock->send_all((char*)buttons, sizeof(buttons));          
         }
         
         int i = 0;        
@@ -136,28 +179,30 @@
                     idx += 1;
                     break;
                 case LINE_CMD:
+                    pc.printf("LINE: (%d, %d) - (%d, %d) COLOR: %X\n", buffer[idx], buffer[idx+1], buffer[idx+2], buffer[idx+3], buffer[idx+4]);
+                    idx += 5;
                     break;
                 case CLS_CMD:
                     //LCD->cls();
                     pc.printf("Clear the screen!\n");
-                    idx += 0;
                     break;
-                case I_AM_P1:
-                case I_AM_P2:
+                case CIRCLE_CMD:
+                    //LCD->cls();
+                    pc.printf("CIRCLE: (%d, %d), r=%d\n", buffer[idx], buffer[idx+1], buffer[idx+2]);
+                    idx += 3;
                     break;
                 default:
                     pc.printf("UNKNOWN CMD %d: This could get ugly!\n", cmd);
                     idx += 0;
             }
         }
-        
     }
     
     ~uLCD_2P() {            
         sock->close();
         eth->disconnect();  
         delete sock;
-        delete endpoint;
+        delete server;
         delete eth; 
     }
  
@@ -167,8 +212,8 @@
         int buffer[256];
         int  buffer_idx;
         
-        UDPSocket* sock;
-        Endpoint*  endpoint;
+        TCPSocketServer* server;
+        TCPSocketConnection*  sock;
         EthernetInterface* eth;
         
         uLCD_4DGL* LCD;
@@ -176,4 +221,4 @@
         int buttons[5];
 };
 
-#endif
\ No newline at end of file
+#endif 
\ No newline at end of file