LeeT WiFiLamp code and test

Dependencies:   ESP8266_WebServer mbed

Fork of WiFiLamp by Sebastian Schocke

main.cpp

Committer:
sschocke
Date:
2014-12-29
Revision:
13:1e8f27da036a
Parent:
12:fbf950985e1c
Child:
19:7fdccfd5b50b

File content as of revision 13:1e8f27da036a:

#include "mbed.h"
#include "PololuLedStrip.h"
#include "ESP8266_WebServer.h"
#include <string>
//#define DEBUG_WIFI

PololuLedStrip ledStrip(D11);

#define LED_COUNT 8
rgb_color colors[LED_COUNT];

DigitalOut wifiCHPD(D4);
DigitalOut wifiReset(D9);

int wifiOn = 0;

Serial wifiSerial(D8,D2);
Serial pc(USBTX,USBRX);
ESP8266_WebServer server(&wifiSerial);

void rxint(void) {
    server.rxint();
}

void setColor( uint8_t r, uint8_t g, uint8_t b )
{
    for ( int c = 0; c < LED_COUNT; c++ )
    {
        colors[c].red = r;
        colors[c].green = g;
        colors[c].blue = b;
    }
    
    ledStrip.write(colors, LED_COUNT);
}

int main() {
    pc.baud(115200);
    
    pc.printf("WiFi Lamp Test - v0.01 ...\r\n");    
    
    setColor( 25, 0, 0);
    
    wifiCHPD = 0;
    wifiReset = 0;
    wifiSerial.baud(115200);
    wifiSerial.attach(&rxint);
#ifdef DEBUG_WIFI
    server.debugSerial = &pc;
#endif
    wait_ms(1000);
    
    pc.printf("Powering WiFi...\r\n");    
    wifiCHPD = 1;
    wifiReset = 1;
    wait_ms(250);
    pc.printf("Hardware Reset WiFi...\r\n");    
    wifiOn = 1;
    
    pc.printf("Starting Web Server...\r\n");
    server.Initialize();
    pc.printf("Done\r\n");    
    
    setColor( 0, 25, 0);
    wait_ms(500);
    setColor( 0, 0, 0);
    
    while(true) {
        ESP8266_WebRequest* request = server.GetRequest();
        if( request != NULL ) {
            pc.printf("HTTP %s %s\r\n", request->Method.c_str(), request->URI.c_str());
            for( std::map<std::string,std::string>::iterator it = request->Parameters.begin(); it!=request->Parameters.end(); ++it ) {
                pc.printf("HTTP Parameter %s = %s\r\n", it->first.c_str(), it->second.c_str());
            }
            std::string httpReply;
            if( request->URI == "/" ) {
                httpReply = "<html><title>WiFi Lamp</title><body><h1>The WiFi Lamp V0.01 is alive ...</h1><br>Quick colour links:<br><a href='/red'>Red</a><br><a href='/green'>Green</a><br><a href='/blue'>Blue</a><br><a href='/white'>White</a><br></body></html>";
                setColor(0,0,0);
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else if( request->URI == "/red" ) {
                httpReply = "<html><head><title>WiFi Lamp</title></head><body><h1>The WiFi Lamp is now Red - To Turn <a href='/'>Off</a></h1></body></html>";
                setColor(100, 0, 0);
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else if( request->URI == "/green" ) {
                httpReply = "<html><head><title>WiFi Lamp</title></head><body><h1>The WiFi Lamp is now Green -  To Turn <a href='/'>Off</a></h1></body></html>";
                setColor(0, 100, 0);
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else if( request->URI == "/blue" ) {
                httpReply = "<html><head><title>WiFi Lamp</title></head><body><h1>The WiFi Lamp is now Blue - To Turn <a href='/'>Off</a></h1></body></html>";
                setColor(0, 0, 100);
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else if( request->URI == "/white" ) {
                httpReply = "<html><head><title>WiFi Lamp</title></head><body><h1>The WiFi Lamp is now White - To Turn <a href='/'>Off</a></h1></body></html>";
                setColor(100, 100, 100);
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else if( request->URI == "/setcolor" ) {
                int r=0, g=0, b=0;
                
                if(request->Parameters.count("r") > 0) r=atoi(request->Parameters["r"].c_str());
                if(request->Parameters.count("g") > 0) g=atoi(request->Parameters["g"].c_str());
                if(request->Parameters.count("b") > 0) b=atoi(request->Parameters["b"].c_str());
                
                pc.printf( "Set color to (%i, %i, %i)\r\n", r,g,b);
                
                setColor( r,g,b );
                
                httpReply = "<html><head><title>WiFi Lamp</title></head><body>ok</body></html>";
                server.SendReply(request->LinkID, httpReply, mimeHTML);
            } else {
                server.Send404Error(request->LinkID);
            }
            pc.printf("\r\nHTTP Reply Sent\r\n");
            delete request;
        }
        
        wait_ms(10);
    }
}