File content as of revision 3:69ef39823eef:

/*******************************************************************************
* Copyright (c) 2014 IBM Corporation and other Contributors.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors: Sam Danbury
* IBM - Initial Contribution
*******************************************************************************/

#include "stdio.h"
#include "mbed.h"
#include "rtos.h"
#include "C12832.h"
#include "LM75B.h"
#include "MMA7660.h"
#include "EthernetInterface.h"
#include "MQTTSocket.h"
#include "MQTTClient.h"
#include "ConfigFile.h"
#include "Arial12x12.h"

#include <string>
#include <sstream>
#include <algorithm>

using namespace std;

#ifdef TARGET_LPC1768

#warning "Compiling for mbed LPC1768"

LocalFileSystem local("local");
C12832 lcd(p5, p7, p6, p8, p11);
DigitalOut led2(LED2);
PwmOut r (p23);
PwmOut g (p24);
PwmOut b (p25);
MMA7660 MMA(p28, p27);
LM75B sensor(p28, p27);
DigitalIn Down(p12);
DigitalIn Left(p13);
DigitalIn Click(p14);
DigitalIn Up(p15);
DigitalIn Right(p16);
AnalogIn ain1(p19);
AnalogIn ain2(p20);

#elif TARGET_K64F

#warning "Compiling for mbed K64F"

//#define ORGANISATION "<org>";
//#define TYPE "<type>";
//#define ID "<id>";
//#define AUTHMETHOD "<auth-method>";
//#define AUTHTOKEN "<auth-token>";

C12832 lcd(D11, D13, D12, D7, D10);
PwmOut r (D5);
PwmOut g (D8);
PwmOut b (D9);
DigitalOut led2(LED2);
MMA7660 MMA(PTE25, PTE24);
LM75B sensor(PTE25, PTE24);
DigitalIn Up(A2);
DigitalIn Down(A3);
DigitalIn Right(A4);
DigitalIn Left(A5);
DigitalIn Click(D4);
AnalogIn ain1(A0);
AnalogIn ain2(A1);

#else

LocalFileSystem local("local");
C12832 lcd(D11, D13, D12, D7, D10);
DigitalOut led1(LED1);
DigitalOut led2(LED2);
DigitalOut led3(LED3);
MMA7660 MMA(D14, D15);
LM75B sensor(D14,D15);
DigitalIn Up(A2);
DigitalIn Down(A3);
DigitalIn Left(A4);
DigitalIn Right(A5);
DigitalIn Click(D4);
AnalogIn ain1 (A0);
AnalogIn ain2 (A1);

#endif

//Joystick
string joystickPos;
void joystickThread(void const *args);

//Commands
enum command {
    blink
};
command getCommand (std::string const& command);
void messageArrived(MQTT::MessageData& md);

//MQTT
void connect();
void attemptConnect();
int getConnTimeout(int attemptNumber);
void subscribe();

//Config
void parseConfig();
bool quickstartMode = true;
string org = "";
string type = "";
string id = "";
string auth_method = "";
string auth_token = "";
string mac = "";

//LCD menu
bool connected = false;
bool menuActivated = false;
int menu = 0;
void printMenu();

int interval;
string getUUID48();

MQTTSocket ipstack;
MQTT::Client<MQTTSocket, Countdown, 250>* client;

void parseConfig() {
    
    ConfigFile cfg;
    
    char value[30];
    char value1[30];
    char value2[30];
    char value3[30];
    
    if (cfg.read("/local/device.cfg")) {
        quickstartMode = false;
        
        if (cfg.getValue("org", value, sizeof(value))) {
            stringstream ss(value);
            ss >> org;
        }        
        if (cfg.getValue("type", value1, sizeof(value1))) {
            stringstream ss(value1);
            ss >> type;
        }
        if (cfg.getValue("id", value2, sizeof(value2))) {
            stringstream ss(value2);
            ss >> id;
        }
        if (cfg.getValue("auth-token", value3, sizeof(value3))) {
            stringstream ss(value3);
            ss >> auth_token;
        }
        
    } else {
        quickstartMode = true;
        org = "quickstart";
        #ifdef TARGET_K64F
            type = "iotsample-mbed-k64f";
        #else
            type = "iotsample-mbed-lpc1768";
        #endif
        id = mac;
    }
    
    #ifdef TARGET_K64F
        #ifdef ORGANISATION
            quickstartMode = false;
            org = ORGANISATION;
            
            #ifdef TYPE
                type = TYPE;
            #else
                lcd.printf("Type is not defined");
            #endif
            
            #ifdef ID
                id = ID;
            #else
                lcd.printf("ID is not defined");
            #endif
            
            #ifdef AUTHMETHOD
                auth_method = AUTHMETHOD;
            #else
                lcd.printf("Auth method is not defined");
            #endif
            
            #ifdef AUTHTOKEN
                auth_token = AUTHTOKEN;
            #else
                lcd.printf("Auth token is not defined");
            #endif
        #endif
    #endif
}

int main()
{
    //RGB: yellow
    r = 0;
    g = 0;
    b = 1;
    
    lcd.cls();
    lcd.set_font((unsigned char*) Arial12x12);
    lcd.locate(0,0);
    lcd.printf("IBM IoT Cloud");
    lcd.locate(0,16);
    lcd.printf("Connecting");
    
    //Connect to network
    EthernetInterface eth;
    eth.init();
    eth.connect();
    
    //Obtain mac address of mbed
    #ifdef TARGET_K64F
        mac = getUUID48();
    #else
        mac = eth.getMACAddress();
        
        //Remove colons from mac address
        mac.erase(remove(mac.begin(), mac.end(), ':'), mac.end());
    #endif

    //Parse config file if present
    parseConfig();
    
    attemptConnect();
    
    if (!quickstartMode) {
        subscribe();
    }
    
    //Start thread to read data from joystick
    joystickPos = "CENTRE";
    Thread jThd(joystickThread);

    interval = 0;
    int i = 0;

    while(1)
    {
        //Message published every second
        if (i == 100) {
            //MQTT Publish
            MQTT::Message message;
            char* pubTopic = "iot-2/evt/status/fmt/json";
            
            char buf[250];
            sprintf(buf,
                    "{\"d\":{\"myName\":\"IoT mbed\",\"accelX\":%0.4f,\"accelY\":%0.4f,\"accelZ\":%0.4f,\"temp\":%0.4f,\"joystick\":\"%s\",\"potentiometer1\":%0.4f,\"potentiometer2\":%0.4f}}",
                    MMA.x(), MMA.y(), MMA.z(),
                    sensor.temp(),
                    joystickPos,
                    ain1.read(),
                    ain2.read());
            message.qos = MQTT::QOS0;
            message.retained = false;
            message.dup = false;
            message.payload = (void*)buf;
            message.payloadlen = strlen(buf);
            
            int rc = 0;
            if ((rc = client->publish(pubTopic, &message)) != 0) {
                connected = false;
                attemptConnect();   
            }
                
            i = 0;
        }
        
        if (interval == 0) {
            //led2 = 0;   
        } else {
            if (i%(interval)==0) {
                //led2 = !led2;
            }
        }
        
        wait(0.01);
        i++;
        client->yield(1);
    }
}

void attemptConnect() {
    int retryAttempt = 0;
    menuActivated = false;
    
    //RGB: yellow
    r = 0;
    g = 0;
    b = 1;
    
    lcd.cls();
    lcd.locate(0,0);
    lcd.printf("IBM IoT Cloud");
    lcd.locate(0,16);
    lcd.printf("Connecting");
    
    while (!connected) {
        
        int connTimeout = getConnTimeout(++retryAttempt);
        
        connect();
        
        if (!connected) {
            wait(connTimeout);
        } else {
            break;
        }
    }
}

int getConnTimeout(int attemptNumber) {
    if (attemptNumber < 10) {
        return 3; //First 10 attempts try within 3 seconds
    } else if (attemptNumber < 20) {
        return 60; //Next 10 attempts retry after every 1 minute
    } else {
        return 600; //After 20 attempts, retry every 10 minutes
    }
}

void connect() {
    ipstack = MQTTSocket();
    client = new MQTT::Client<MQTTSocket, Countdown, 250>(ipstack);
    
    //TCP Connect
    string ip = org + ".messaging.internetofthings.ibmcloud.com";
    
    char* hostname = new char[ip.length() + 1];
    strcpy(hostname, ip.c_str());
    
    int port = 1883;
    int rc = ipstack.connect(hostname, port);
    if (rc != 0) {
        lcd.printf("TCP connect failed");
    }
    
    //Construct clientId based on config
    string str = string("d:") + org + ":" + type + ":" + id;
    char clientId[str.size()];
    memcpy(clientId, str.c_str(), str.size() + 1);
    
    //MQTT Connect
    MQTTPacket_connectData data = MQTTPacket_connectData_initializer;
    data.MQTTVersion = 3;
    data.clientID.cstring = clientId;
    
    if (!quickstartMode) {
        char* password = new char[auth_token.length() + 1];
        strcpy(password, auth_token.c_str());
        
        data.username.cstring = "use-token-auth";
        data.password.cstring = password;
    }
    
    if ((rc = client->connect(&data)) == 0) {
        connected = true;
        
        //RGB: green
        r = 1;
        g = 0;
        b = 1;
        
        lcd.locate(0,0);        
        lcd.printf("IBM IoT Cloud");
        lcd.locate(0,16);
        lcd.printf("Connected");
        
        wait(2);
        
        lcd.locate(0,0);        
        lcd.printf("IBM IoT Cloud");
        lcd.locate(0,16);
        lcd.printf("Scroll with joystick");
        
        menuActivated = true;
    }
}

void subscribe() {
    char* subTopic = "iot-2/cmd/+/fmt/json";
    int rc = 0;
    if ((rc = client->subscribe(subTopic, MQTT::QOS1, messageArrived)) != 0)
        lcd.printf("rc from MQTT subscribe is %d\n", rc);   
}

void messageArrived(MQTT::MessageData& md) {
    MQTT::Message &message = md.message;
    
    char* topic = new char[md.topicName.lenstring.len + 1];
    sprintf(topic, "%.*s", md.topicName.lenstring.len, md.topicName.lenstring.data);
    
    char* payload = new char[message.payloadlen + 1];
    sprintf(payload, "%.*s", message.payloadlen, (char*)message.payload);
    
    string topicStr = topic;
    string payloadStr = payload;
    
    //Command topic: iot-2/cmd/blink/fmt/json
    string cmd = topicStr.substr(10, topicStr.find("/fmt/") - 10);
    
    switch(getCommand(cmd)) {
        case blink: {   
            string str = payloadStr.substr(8, payloadStr.find("}") - 8);
            int rate = atoi(str.c_str());
            
            if (rate == 0) {
                interval = 0;   
            } else if (rate > 50) {
                interval = 1;   
            } else if (rate > 0) {
                interval = 50/rate;      
            }
            
            break;
        }
        default:
            lcd.printf("Unsupported command: %s\n", cmd);
    }
    
    if (topic) {
        delete[] topic;
    }
    if (payload) {
        delete[] payload;
    }

}

command getCommand (string const& command) {
    if (command == "blink")
        return blink;
}

void joystickThread(void const *args) {
    while (true) {
        
        if (!menuActivated) {
            menu = 0;
        }
        
        if (Down) {
            joystickPos = "DOWN";
            if (menu >= 0 && menu < 3) {
                menu++;
                printMenu();
            }
        } else if (Left) {
            joystickPos = "LEFT";
        } else if (Click) {
            joystickPos = "CLICK";
        } else if (Up) {
            joystickPos = "UP";
            if (menu <= 3 && menu > 0) {
                menu--;
                printMenu();
            }
        } else if (Right) {
            joystickPos = "RIGHT";
        } else {
            joystickPos = "CENTRE";
        }
        wait(0.2);
    }
}

void printMenu() {
    if (menuActivated) {
        lcd.cls();
        lcd.locate(0,0);
    
        switch(menu) {
            case 0:
                lcd.printf("IBM IoT Cloud");
                lcd.locate(0,16);
                lcd.printf("Scroll with joystick");
                break;
            case 1:
                lcd.printf("Go to:");
                lcd.locate(0,16);
                lcd.printf("http://ibm.biz/iotqstart");
                break;
            case 2:
                lcd.printf("Device Identity:");
                lcd.locate(0,16);
                lcd.printf("%s", mac);
                break;
            case 3:
                lcd.printf("Status:");
                lcd.locate(0,16);
                lcd.printf("Connected");
                break;
        }
    } else {
        menu = 0;   
    }
}

string getUUID48 () {
    
    unsigned int UUID_LOC_WORD0 = 0x40048060;
    unsigned int UUID_LOC_WORD1 = 0x4004805C;
 
    // Fetch word 0
    uint32_t Word0 = *(uint32_t *)UUID_LOC_WORD0;
 
    // Fetch word 1
    // we only want bottom 16 bits of word1 (MAC bits 32-47)
    // and bit 9 forced to 1, bit 8 forced to 0
    // Locally administered MAC, reduced conflicts
    // http://en.wikipedia.org/wiki/MAC_address
    uint32_t Word1 = *(uint32_t *)UUID_LOC_WORD1;
    Word1 |= 0x00000200;
    Word1 &= 0x0000FEFF;
 
    string sd;
    char stemp[100] = "";
    snprintf(stemp, 100, "%4X%08X", Word1,Word0); // I use the safer version of sprintf() -- snprintf()
    sd = stemp; // the contents of sd are now "This is a string!" 
    
    return (sd);
}