File content as of revision 14:3c3aa4fd7e1a:

#include "mbed.h"
#include "EthernetInterface.h"
#include "NTPClient.h"
#include "RdWebServer.h"
#include "GasUseCounter.h"
#include "Thermometers.h"
#include "VoltAlerter.h"
#include "Watchdog.h"
#include "Logger.h"

// Web and UDB ports 
const int WEBPORT = 80; // Port for web server
const int BROADCAST_PORT = 42853; // Arbitrarily chosen port number

// Main loop delay between data collection passes
const int LOOP_DELAY_IN_MS = 250;

// Debugging and status
RawSerial pc(USBTX, USBRX);
DigitalOut led1(LED1); //ticking (flashes)
DigitalOut led2(LED2); //state of the 1st voltage alerter
DigitalOut led3(LED3); //socket connecting status
DigitalOut led4(LED4); //server status

// Web server
EthernetInterface eth;
UDPSocket sendUDPSocket;
Endpoint broadcastEndpoint;

// Network Time Protocol (NTP)
NTPClient ntp;
const int NTP_REFRESH_INTERVAL_HOURS = 1;

// File system for SD card
SDFileSystem sd(p5, p6, p7, p8, "sd");

// Log file names
const char* gasPulseFileName = "/sd/curPulse.txt";
const char* eventLogFileName = "/sd/log.txt";
const char* dataLogFileBase = "/sd/";

// Logger
Logger logger(eventLogFileName, dataLogFileBase);

// Gas use counter
DigitalIn gasPulsePin(p21);
GasUseCounter gasUseCounter(gasPulseFileName, gasPulsePin, pc);

// Thermometers - DS18B20 OneWire Thermometer connections
const PinName tempSensorPins[] = { p22 };
Thermometers thermometers(sizeof(tempSensorPins)/sizeof(PinName), tempSensorPins, LOOP_DELAY_IN_MS);

// Voltage Sensors / Alerters
const int NUM_VOLT_ALERTERS = 3;
VoltAlerter voltAlerter1(p23);
VoltAlerter voltAlerter2(p24);
VoltAlerter voltAlerter3(p25);

// Watchdog
Watchdog watchdog;

// Broadcast message format
// Data format of the broadcast message - senml - https://tools.ietf.org/html/draft-jennings-senml-08
// {
//     "e": [
//              {"n":"gasCount","v":%d}, 
//              {"n":"gasPulseRateMs","v":%d,"u":"ms"},
//              {"n":"temp_%s","v":%0.1f,"u":"degC"},
//              ...
//              {"n":"pump_%d","v":%d},
//              ...
//          ],
//      "bt": %d
// }
const char broadcastMsgPrefix[] = "{\"e\":[";
const char broadcastMsgGasFormat[] = "{\"n\":\"gasCount\",\"v\":%d,\"u\":\"count\"},{\"n\":\"gasPulseRateMs\",\"v\":%d,\"u\":\"ms\"}";
const char broadcastTemperatureFormat[] = "{\"n\":\"temp_%s\",\"v\":%0.1f,\"u\":\"degC\"}";
const char broadcastVoltAlerterFormat[] = "{\"n\":\"pump_%d\",\"bv\":%d}";
const char broadcastMsgSuffix[] = "],\"bt\":%d}";

// Broadcast message length and buffer
const int broadcastMsgLen = sizeof(broadcastMsgPrefix) + 
            sizeof(broadcastMsgGasFormat) + 
            (sizeof(broadcastTemperatureFormat)*Thermometers::MAX_THERMOMETERS) + 
            (sizeof(broadcastVoltAlerterFormat)*NUM_VOLT_ALERTERS) + 
            sizeof(broadcastMsgSuffix) + 
            60;
char broadcastMsgBuffer[broadcastMsgLen];
    
// Format broadcast message
void GenBroadcastMessage()
{
    // Get temperature values
    TemperatureValue tempValues[Thermometers::MAX_THERMOMETERS];
    int numTempValues = thermometers.GetTemperatureValues(Thermometers::MAX_THERMOMETERS, tempValues, 100);
//    for (int tempIdx = 0; tempIdx < numTempValues; tempIdx++)
//    {
//        printf("Temp: %.1f, Addr: %s, Time: %d\r\n", tempValues[tempIdx].tempInCentigrade, tempValues[tempIdx].address, tempValues[tempIdx].timeStamp);
//    }

    // Format the broadcast message
    time_t timeNow = time(NULL);
    strcpy(broadcastMsgBuffer, broadcastMsgPrefix);
    sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastMsgGasFormat, gasUseCounter.GetCount(), gasUseCounter.GetPulseRateMs());
    strcpy(broadcastMsgBuffer+strlen(broadcastMsgBuffer), ",");
    for (int tempIdx = 0; tempIdx < numTempValues; tempIdx++)
    {
        sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastTemperatureFormat, tempValues[tempIdx].address, tempValues[tempIdx].tempInCentigrade);
        strcpy(broadcastMsgBuffer+strlen(broadcastMsgBuffer), ",");
    }
    sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastVoltAlerterFormat, 1, voltAlerter1.GetState());
    strcpy(broadcastMsgBuffer+strlen(broadcastMsgBuffer), ",");
    sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastVoltAlerterFormat, 2, voltAlerter2.GetState());
    strcpy(broadcastMsgBuffer+strlen(broadcastMsgBuffer), ",");
    sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastVoltAlerterFormat, 3, voltAlerter3.GetState());
    sprintf(broadcastMsgBuffer+strlen(broadcastMsgBuffer), broadcastMsgSuffix, timeNow);
}

// Send broadcast message with current data
void SendInfoBroadcast()
{
    led3 = true;
    
    // Init the sending socket
    sendUDPSocket.init();
    sendUDPSocket.set_broadcasting();
    broadcastEndpoint.set_address("255.255.255.255", BROADCAST_PORT);
        
    // Format the message
    GenBroadcastMessage();
            
    // Send
    int bytesToSend = strlen(broadcastMsgBuffer);
    int rslt = sendUDPSocket.sendTo(broadcastEndpoint, broadcastMsgBuffer, bytesToSend);
    if (rslt == bytesToSend)
    {
        pc.printf("Broadcast (len %d) Sent ok %s\r\n", bytesToSend, broadcastMsgBuffer);
    }
    else if (rslt == -1)
    {
        pc.printf("Broadcast Failed to send %s\r\n", broadcastMsgBuffer);
    }
    else
    {
        pc.printf("Broadcast Didn't send all of %s\r\n", broadcastMsgBuffer);
    }
    
    // Log the data
    logger.LogData(broadcastMsgBuffer);
    
    led3 = false;
}

char* getCurDataCallback(int method, char* cmdStr, char* argStr)
{
    // Format message
    GenBroadcastMessage();
    return broadcastMsgBuffer;
}

char* setGasUseCallback(int method, char* cmdStr, char* argStr)
{
    pc.printf("Setting gas use count %s\r\n", argStr);
    int newGasUse = 0;
    char* eqStr = strchr(argStr, '=');
    if (eqStr == NULL)
        return "SetGasValue FAILED";
    sscanf(eqStr+1, "%d", &newGasUse);
    gasUseCounter.SetCount(newGasUse);
    return "SetGasValue OK";
}

// Create, configure and run the web server
void http_thread(void const* arg)
{
    char* baseWebFolder = "/sd/";
    RdWebServer webServer;
    webServer.addCommand("", RdWebServerCmdDef::CMD_SDORUSBFILE, NULL, "index.htm", false);
    webServer.addCommand("gear-gr.png", RdWebServerCmdDef::CMD_SDORUSBFILE, NULL, NULL, true);
    webServer.addCommand("listfiles", RdWebServerCmdDef::CMD_SDORUSBFILE, NULL, "/", false);
    webServer.addCommand("getcurdata", RdWebServerCmdDef::CMD_CALLBACK, &getCurDataCallback);
    webServer.addCommand("setgascount", RdWebServerCmdDef::CMD_CALLBACK, &setGasUseCallback);
    webServer.init(WEBPORT, &led4, baseWebFolder);
    webServer.run();
}

// Network time protocol (NTP) thread to get time from internet
void ntp_thread(void const* arg)
{
    while (1)
    {
        pc.printf("Trying to update time...\r\n");
        if (ntp.setTime("0.pool.ntp.org") == NTP_OK)
        {
          printf("Set time successfully\r\n");
          time_t ctTime;
          ctTime = time(NULL);
          printf("Time is set to (UTC): %s\r\n", ctime(&ctTime));
        }
        else
        {
          printf("Cannot set from NTP\r\n");
        }
        
        // Refresh time every K hours
        for (int k = 0; k < NTP_REFRESH_INTERVAL_HOURS; k++)
        {
            // 1 hour
            for (int i = 0; i < 60; i++)
            {
                for (int j = 0; j < 60; j++)
                {
                    osDelay(1000);
                }
                pc.printf("%d mins to next NTP time refresh\r\n", (NTP_REFRESH_INTERVAL_HOURS-k-1)*60 + (59-i));
            }
        }
    }
}

// #define TEST_WATCHDOG 1
#ifdef TEST_WATCHDOG
int watchdogTestLoopCount = 0;
#endif

// Main
int main()
{
    pc.baud(115200);
    pc.printf("Gas Monitor V2 - Rob Dobson 2014\r\n");

    // Initialise thermometers
    thermometers.Init();
    
    // Get the current count from the SD Card
    gasUseCounter.Init();
    
    // setup ethernet interface
    eth.init(); //Use DHCP
    eth.connect();
    
    pc.printf("IP Address is %s\r\n", eth.getIPAddress());
    
    // NTP Time setter
    Thread ntpTimeSetter(&ntp_thread);
    
    // Web Server
    Thread httpServer(&http_thread, NULL, osPriorityNormal, (DEFAULT_STACK_SIZE * 3));

    // Store reason for restart
    bool watchdogCausedRestart = watchdog.WatchdogCausedRestart();
    bool restartCauseRecorded = false;
    
    // Setup the watchdog for 10s reset
    watchdog.SetTimeoutSecs(10);
    
    // Time of last broadcast
    time_t timeOfLastBroadcast = time(NULL);
    const int TIME_BETWEEN_BROADCASTS_IN_SECS = 60;
    while(true)
    {
        // Check if we can record the reason for restart (i.e. if time is now set)
        if (!restartCauseRecorded)
        {
            time_t nowTime = time(NULL);
            if (nowTime > 1000000000)
            {
                // Record the reason for restarting in the log file
                if (watchdogCausedRestart)
                    logger.LogEvent("Watchdog Restart");
                else
                    logger.LogEvent("Normal Restart");
                restartCauseRecorded = true;
            }
        }
        
        // Loop delay
        osDelay(LOOP_DELAY_IN_MS);
        
        // Feed the watchdog and show the flashing LED
        led1 = !led1;
        watchdog.Feed();

        // Service gas count
        if (gasUseCounter.Service())
        {
            SendInfoBroadcast();
            timeOfLastBroadcast = time(NULL);
        }
        
        // Service thermometers
        thermometers.Service();
        
        // Check if ready for a broadcast
        if ((time(NULL) - timeOfLastBroadcast) >= TIME_BETWEEN_BROADCASTS_IN_SECS)
        {
            SendInfoBroadcast();
            timeOfLastBroadcast = time(NULL);
        }
        
        // Service volt alerters
        voltAlerter1.Service();
        voltAlerter2.Service();
        voltAlerter3.Service();
        
        // Set LED2 to the state of the first volt alerter
        led2 = voltAlerter1.GetState();
        
#ifdef TEST_WATCHDOG
        // After about 20 seconds of operation we'll hang to test the watchdog
        if (watchdogTestLoopCount++ > 80)
        {
            // This should cause watchdog to kick in and reset
            osDelay(20000);
        }
#endif
    }
}