LedSuit-Control/LEDSUITPROGRAM.ino at master · iPet/LedSuit-Control · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
#include <SPI.h>
#include <ESP8266WiFi.h>
#include <WiFiUdp.h>
#include <Adafruit_NeoPixel.h>
#ifdef __AVR__
  #include <avr/power.h>
#endif

// LED
#define PIN 5
#define PARTS 5
#define EVENTS 5


// SUIT Constants
#define NONE 0
#define LOAD_SCENARIO 1
#define START_SCENARIO 2
#define RUNNING 3

byte ledStatus = NONE;

// Position matrixes
byte right_bottom_arm [][3] = {{0,  1,  2}, {3,  4,  5}, {6, 7, 8}, {9, 10, 11}};
byte right_upper_arm [][3] = {{12,  13,  14}, {15,  16,  17}, {18, 18, 19}};
byte chest [][6] = {{20,  21,  22,  23,  24,  25},{26,  27,  28,  29,  30, 31},{32, 33, 34, 35, 36, 37},{38, 39, 40, 41, 42, 43},{44, 45, 46, 47, 48, 49}, {50, 51, 52, 52, 53, 54}};
byte left_upper_arm [][3] = {{56,  57,  58}, {59,  60, 61}, {62, 62, 63}};
byte left_bottom_arm [][3] = {{64,  65, 66}, {67,  68,  69}, {70, 71, 72}, {73, 74, 75}};

byte timings[PARTS] = {0,0,0,0,0};
byte positions[PARTS] = {0,0,0,0,0};
byte delays[PARTS] = {0,0,0,0,0};

//[PARTS][EVENTS][part_id,mode,r,g,b,delay]
byte example_response_matrix [PARTS][EVENTS][6] = {
  {{1,1,255,255,255,50},{1,1,255,0,255,50},{1,1,0,255,0,50},{1,1,0,0,255,50}},
  {{2,1,255,255,255,50},{2,1,0,255,255,50},{2,1,255,255,255,50}},
  {{3,1,255,255,255,50},{3,1,255,0,255,50},{3,1,0,255,0,50},{3,1,0,255,0,50}},
  {{4,1,255,255,255,50},{4,1,0,255,255,50},{4,1,0,255,255,50},{4,1,255,0,255,50},{4,1,0,255,0,50}},
  {{5,1,255,255,255,50},{5,1,0,255,255,50},{5,1,0,255,255,50},{5,1,255,0,255,50},{5,1,0,255,0,50}}};


// LedStrip Object
Adafruit_NeoPixel strip = Adafruit_NeoPixel(12, PIN, NEO_GRB + NEO_KHZ800);

//WIFI Connect
byte WifiStatus = WL_IDLE_STATUS;
char ssid[] = "";  //  your network SSID (name)
char pass[] = "";       // your network password
int keyIndex = 0;            // your network key Index number (needed only for WEP)

//WebServer
const char* host = "172.31.81.194";

//WIFI Webserver
WiFiServer server(80);

//NTP
unsigned int localPort = 2390;
IPAddress timeServer(95, 46, 198, 21);
const int NTP_PACKET_SIZE = 48;
byte packetBuffer[ NTP_PACKET_SIZE]; //buffer to hold incoming and outgoing packets
unsigned long secsSince1900 = 0;

// A UDP instance to let us send and receive packets over UDP
WiFiUDP Udp;


void setupAccessPoint()
{
  WiFi.mode(WIFI_AP);

  // Do a little work to get a unique-ish name. Append the
  // last two bytes of the MAC (HEX'd) to "Thing-":
  uint8_t mac[WL_MAC_ADDR_LENGTH];
  WiFi.softAPmacAddress(mac);
  String macID = String(mac[WL_MAC_ADDR_LENGTH - 2], HEX) +
                 String(mac[WL_MAC_ADDR_LENGTH - 1], HEX);
  macID.toUpperCase();
  String AP_NameString = "Led Suit  " + macID;

  char AP_NameChar[AP_NameString.length() + 1];
  memset(AP_NameChar, 0, AP_NameString.length() + 1);

  for (int i=0; i<AP_NameString.length(); i++)
    AP_NameChar[i] = AP_NameString.charAt(i);

  WiFi.softAP(AP_NameChar, "spark123");
}

void runWebServer()
{
    // Check if a client has connected
  WiFiClient client = server.available();
  if (!client) {
    return;
  }

  // Read the first line of the request
  String req = client.readStringUntil('\r');
  char* charBuf;
  req.toCharArray(charBuf, req.length());

  Serial.println(req);
  client.flush();

  // Match the request
  int val = -1; // We'll use 'val' to keep track of both the
                // request type (read/set) and value if set.
  if (req.substring(5,10) == "setup")
      val = 0; // Will write LED low
  else  if (req.substring(5,12) == "connect")
    val = 1; // Will write LED high

  // Prepare the response. Start with the common header:
  String s = "HTTP/1.1 200 OK\r\n";
  s += "Content-Type: text/html\r\n\r\n";
  s += "<!DOCTYPE HTML>\r\n<html>\r\n";
  // If we're setting the LED, print out a message saying we did
  if (val == 0)
  {
    s += "<form action=\"/connect\" method=\"get\">";
    s += "ssid:<input type=\"text\" name=\"ssid\"><br>";
    s += "pass:<input type=\"text\" name=\"pass\"><br>";
    s += "<input type=\"submit\" value=\"Submit\">";
    s += "</form>";
    s += "</html>\n";
  }
  else if (val == 1)
  { // If we're reading pins, print out those values:
    s += "Connecting wifi";
    // GET /connect?ssid=rdm&pass=123
    int ssidIndex = req.indexOf("ssid=");
    //  Search for the next comma just after the first
    int passIndex = req.indexOf("&pass=", ssidIndex+1);

    String ssid = req.substring(ssidIndex+1, passIndex);
    String pass = req.substring(passIndex); //

    int isIndex = ssid.indexOf("=");
    ssid = ssid.substring(isIndex+1);

    isIndex = pass.indexOf("=");
    int sndIndex = pass.indexOf(" ");
    pass = pass.substring(isIndex+1, sndIndex);

//    connectWiFi();
  }
  else
  {
    s += "Invalid Request.<br> Try /led/1, /led/0, or /read.";
  }
  s += "</html>\n";

  // Send the response to the client
  client.print(s);
}

// Store NTP-Server seconds from 1900
void setTime(){
  int packetSize = Udp.parsePacket();
  if (packetSize) {
    Serial.println("packet received");
    // We've received a packet, read the data from it
    Udp.read(packetBuffer, NTP_PACKET_SIZE); // read the packet into the buffer

    //the timestamp starts at byte 40 of the received packet and is four bytes,
    // or two words, long. First, esxtract the two words:

    unsigned long highWord = word(packetBuffer[40], packetBuffer[41]);
    unsigned long lowWord = word(packetBuffer[42], packetBuffer[43]);
    // combine the four bytes (two words) into a long integer
    // this is NTP time (seconds since Jan 1 1900):
    secsSince1900 = highWord << 16 | lowWord;
// TODO!! Store NTP time and current millis in two global variables
  }
}

// send an NTP request to the time server at the given address
unsigned long sendNTPpacket(IPAddress& address) {
  // set all bytes in the buffer to 0
  memset(packetBuffer, 0, NTP_PACKET_SIZE);
  // Initialize values needed to form NTP request
  // (see URL above for details on the packets)
  //Serial.println("2");
  packetBuffer[0] = 0b11100011;   // LI, Version, Mode
  packetBuffer[1] = 0;     // Stratum, or type of clock
  packetBuffer[2] = 6;     // Polling Interval
  packetBuffer[3] = 0xEC;  // Peer Clock Precision
  // 8 bytes of zero for Root Delay & Root Dispersion
  packetBuffer[12]  = 49;
  packetBuffer[13]  = 0x4E;
  packetBuffer[14]  = 49;
  packetBuffer[15]  = 52;

  // all NTP fields have been given values, now
  // you can send a packet requesting a timestamp:
  Udp.beginPacket(address, 123); //NTP requests are to port 123
  //Serial.println("4");
  Udp.write(packetBuffer, NTP_PACKET_SIZE);
  //Serial.println("5");
  Udp.endPacket();
  //Serial.println("6");
}

// Connect to WiFi based on SSID and PASS
void connectWiFi(){
  WiFi.mode(WIFI_STA);
  // attempt to connect to Wifi network:
  while (WifiStatus != WL_CONNECTED) {
    Serial.println("Attempting to connect to SSID: ");
   //connect to Wifi network
    WifiStatus = WiFi.begin(ssid, pass);
  }
  if (WifiStatus) {
    Serial.println("Connected to the WIFI");
  }
}

// HTTP /GET request to webserver to receive its status
void getStatus(){
  Serial.print("connecting to: ");
  Serial.println(host);

  // Use WiFiClient class to create TCP connections
  WiFiClient client;
  const int httpPort = 80 ;
  if (!client.connect(host, 80)) {
    Serial.print("webserver port 80 ");
    Serial.println("connection failed");
  }

  // We now create a URI for the request
  String url = "/scenarios";

  // This will send the request to the server
  client.print(String("GET ") + url + " HTTP/1.1\r\n" +
               "Host: " + host + "\r\n" +
               "Connection: close\r\n\r\n");
  delay(10);

  // Read all the lines of the reply from server and print them to Serial
  while(client.available()){
    String line = client.readStringUntil('\r');
    Serial.println(line);
  }

  Serial.println();
  Serial.println("closing connection");
}

// calculate hex to rgb, store in byte array buff
void hexToRGB(String hexstring, byte *buff){
  long number = (long) strtol( &hexstring[1], NULL, 16);
  buff[0]= number >> 16;
  buff[1] = number >> 8 & 0xFF;
  buff[2] = number & 0xFF;
}

// Set color of leds in part
void setAllColor(int id, uint32_t color){
  size_t y = 0;
  size_t x = 0;

  if (id == 1){
    y = sizeof (right_bottom_arm);
    x = sizeof (right_bottom_arm[0]);

    for (int i = 0; i < x; i++){
      for (int ii = 0; ii < y; ii++){
        strip.setPixelColor(right_bottom_arm[ii][i], color);
      }
    }
  }
  else if (id == 2){
    y = sizeof (right_upper_arm);
    x = sizeof (right_upper_arm[0]);

    for (int i = 0; i < x; i++){
      for (int ii = 0; ii < y; ii++){
        strip.setPixelColor(right_upper_arm[ii][i], color);
      }
    }
  }
  else if (id == 3){
    y = sizeof (chest);
    x = sizeof (chest[0]);

    for (int i = 0; i < x; i++){
      for (int ii = 0; ii < y; ii++){
        strip.setPixelColor(chest[ii][i], color);
      }
    }
  }
  else if (id == 4){
    y = sizeof (left_upper_arm);
    x = sizeof (left_upper_arm[0]);

    for (int i = 0; i < x; i++){
      for (int ii = 0; ii < y; ii++){
        strip.setPixelColor(left_upper_arm[ii][i], color);
      }
    }
  }
  else if (id == 5){
     y = sizeof (left_bottom_arm);
    x = sizeof (left_bottom_arm[0]);

    for (int i = 0; i < x; i++){
      for (int ii = 0; ii < y; ii++){
        strip.setPixelColor(left_bottom_arm[ii][i], color);
      }
    }
  }
}

// Parse json from webserver to 2d byte array
void parseScenario(){

}
void setup() {
  // Sync controller with NTP server
  sendNTPpacket(timeServer);
  setTime();

  if (WifiStatus = WL_IDLE_STATUS){
     setupAccessPoint();
     while (WifiStatus != WL_CONNECTED){
      runWebServer();
     }
  }
}

void loop() {
  if (ledStatus = NONE){
    getStatus();
  }
  else if (ledStatus = LOAD_SCENARIO){
    parseScenario();
  }
  else if (ledStatus = RUNNING){
    for (int i = 0; i < PARTS; i++){
      if (timings[i] + (delays[i]*100) < millis()){
        byte modus = example_response_matrix[i][positions[i]][1];
        byte r = example_response_matrix[i][positions[i]][2];
        byte g = example_response_matrix[i][positions[i]][3];
        byte b = example_response_matrix[i][positions[i]][4];
        delays[i] = example_response_matrix[i][positions[i]][5];
        positions[i] = positions[i]+1;

// TODO!! Add more patterns
        if (modus == 1){
          setAllColor(i,strip.Color(r,g,b));
        }
      }
      timings[i] = millis();
    }
  }
}