File content as of revision 22:8d106fd5fa84:

/*
 *
 *  Made by Jurica Resetar @ aconno
 *  aconno.de
 *  All rights reserved.
 *
 */

#include "mbed.h"
#include "ble/BLE.h"
#include "GapAdvertisingData.h"
#include "acd52832_bsp.h"
#include "mma8452.h"
#include "AckService.h"
#include "nrf52_uart.h"
#include "nrf52_digital.h"

#define DEBUG               (0)
#define DEBUG_ACC           (0)
#define PRINT               (0)
#define DEBUG_MAC           (0)
#define DEBUG_CONNECTION    (0)
#define USE_ACC             (0)

#define SLEEP_TIME_S              (9.0)           /* Sleep time (in s)          */
#define ADV_TIMER_TIME_S          (0.5)           /* Advertising time (in s)    */
#define SCAN_TIMER_TIME_S         (0.5)           /* Scanning time (in s)       */
#define FREE_TIME_S               (0.1)           /* Time between end of a scanning and sleep mode */
#define AWAKE_TIME_S              (ADV_TIMER_TIME_S+SCAN_TIMER_TIME_S+FREE_TIME_S)      /* Was 0.15 */
#define SHORT_SLEEP_TIME_S        (0.5)           /* Shorter sleep time (s) */
#define SHORT_SLEEP_TIME_PERIOD_S (10)            /* Time after a last scanned advertisment. In the period, sleep time is SHORT_SLEEP_TIME */
#define MAC_SIZE_B                (6)

/* Static constants for the BLE example */
#define MAX_BLE_PACKET_SIZE (31)
#define MSD_SIZE            (18)
#define MSD_ID              (0xFF)

#define BUZZ_TIME_S         (1)   /* Buzz time in s */
#define ADV_INTERVAL        (200)   /* Advertising interval (in ms) */
#define SCAN_INTERVAL       (SCAN_TIMER_TIME_S)   /* Scan interval (in ms) */
#define SCAN_WINDOW         (SCAN_TIMER_TIME_S)

/* Static constants for the accelerometer */
#define WHO_AM_I            0x0D           /* Type 'read' : This should return the device id of 0x2A */
#define OUT_Z_MSB           0x05           /* Type 'read' : z axis - 8 most significatn bit of a 12 bit sample */
#define I2C_DATA            (p29)
#define I2C_CLK             (p2)
#define INT2_PIN            (p4)
#define BUZZER              (p31)

#if PRINT
    /* Defines for debugging over uart */
    #define TX              (p25)
    #define RX              (p26)
    NRF52_UART uart(TX,RX, Baud9600);
    char printBuffer[30] = {};
#endif



bool shushShush = false;
const static uint16_t ACK_CHARA_UUID = 0xA001;

uint8_t txPower = 4;
uint8_t sleepFlag = false;
uint8_t tempSleepTime = SLEEP_TIME_S;
uint8_t msd[MSD_SIZE] = {0x59, 0x00, 0xE1, 0x61, 0x35, 0xBA, 0xC0, 0xEC, 0x47, 0x2A, 0x98, 0x00, 0xAF, 0x18, 0x43, 0xFF, 0x05, 0x00};
uint8_t myMacAddress[6] = {};          
uint8_t buzzer_flag = 0;

enum RadioState{
    OFF,
    ADVERTISING,
    SCANNING
    };
enum RadioState radioState = OFF;

void TurnBuzzOff(void);
void GoToSleep();
void StartAdvertising();
void startScanning();
void WakeMeUp();

Ticker WakeSleepT;
Ticker turnBuzzOffT;
Ticker sleepChanger;
NRF52_DigitalOut buzzer(BUZZER);
#if USE_ACC
    DigitalOut accPower(p7);
    DigitalOut i2cPower(p5);
    InterruptIn accPulse(INT2_PIN);
    Acc_MMA8452 acc(I2C_DATA, I2C_CLK, MMA8452_ADDRESS);
#endif
BLE &ble = BLE::Instance();
ACKService<4> *ackServicePtr;

#if DEBUG || DEBUG_MAC || DEBUG_CONNECTION
    DigitalOut advLED(p22);         // Red
    DigitalOut scanLED(p23);        // Blue
    DigitalOut connectedLED(p24);   // Green
#endif

#if DEBUG_ACC
    DigitalOut int_led(p22);
#endif

void buzz(uint16_t period_us, uint32_t duration_ms){
    int32_t counter;
    for(counter=0; counter<((duration_ms*1000)/(period_us*1.0)); counter++){
        buzzer.toggle();
        wait_us(period_us/2);
        buzzer.toggle();
        wait_us(period_us/2);
    }
    buzzer = 0;
}

void onConnectionCallback(const Gap::ConnectionCallbackParams_t *params){   
    #if DEBUG_CONNECTION
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = !scanLED;       // Blue
        wait_ms(100);
        scanLED = 1;              // Blue
    #endif
    WakeSleepT.detach();
    sleepFlag = false;
}


/* Restart Advertising on disconnection*/
void disconnectionCallback(const Gap::DisconnectionCallbackParams_t *params){
    #if DEBUG_CONNECTION
        advLED = !advLED;       // RED
        wait_ms(100);
        advLED = !advLED;
        wait_ms(100);
        advLED = !advLED;
        wait_ms(100);
        advLED = !advLED;
        wait_ms(100);
        advLED = 1;
        wait_ms(100);
        advLED = 1;
    #endif
    WakeSleepT.attach(WakeMeUp, FREE_TIME_S);
    sleepFlag = true;
    
}

void onDataWrittenCallback(const GattWriteCallbackParams *params) {
    if(params->handle == ACK_CHARA_UUID || 1){
        // Something is written into AckCharacteristic
        if(params->data[0] == 0xBA)
            if(params->data[1] == 0xBE){
                #if DEBUG_CONNECTION
                    connectedLED = !connectedLED;       // BLUE
                    wait_ms(100);
                    connectedLED = !connectedLED;
                    wait_ms(100);
                    connectedLED = !connectedLED;
                    wait_ms(100);
                    connectedLED = !connectedLED;
                    wait_ms(100);
                    connectedLED = !connectedLED;
                    wait_ms(100);
                    connectedLED = 1;
                    wait_ms(100);
                #endif    
                buzz(900,1000);
                WakeSleepT.detach();
                WakeSleepT.attach(WakeMeUp, FREE_TIME_S);
                /*
                    This function should make advertiser stop
                */
                ble.disconnect(Gap::LOCAL_HOST_TERMINATED_CONNECTION);
                return;
            }
    }
    else{
        // Execute this for wrong data written into characteristic   
    }
}

/**
 * This function is called when the ble initialization process has failed
 */
void onBleInitError(BLE &ble, ble_error_t error){
    /* Avoid compiler warnings */
    (void) ble;
    (void) error;
    /* Initialization error handling should go here */
}

/**
 * Callback triggered when the ble initialization process has finished
 */
void bleInitComplete(BLE::InitializationCompleteCallbackContext *params){
    BLE&        ble   = params->ble;
    ble_error_t error = params->error;

    if (error != BLE_ERROR_NONE) {
        /* In case of error, forward the error handling to onBleInitError */
        onBleInitError(ble, error);
        return;
    }

    /* Ensure that it is the default instance of BLE */
    if(ble.getInstanceID() != BLE::DEFAULT_INSTANCE) {
        return;
    }
    
    uint8_t init_values[4] = {0,0,0,0};
    /* Get my MAC address */
    BLEProtocol::AddressType_t temp_address_type;
    ble.gap().getAddress(&temp_address_type, myMacAddress);
    ackServicePtr = new ACKService<4>(ble, init_values);
    ackServicePtr->updateMacAddress(myMacAddress);    // Update MAC address
    
    ble.gap().onDisconnection(disconnectionCallback);
    ble.gap().onConnection(onConnectionCallback);         // -->> Uncomment these two lines for shush-shush 
    ble.gattServer().onDataWritten(onDataWrittenCallback);
    
    /* setup advertising */
    ble.gap().accumulateAdvertisingPayload(GapAdvertisingData::MANUFACTURER_SPECIFIC_DATA, (uint8_t *)msd, MSD_SIZE);
    ble.gap().setAdvertisingInterval(ADV_INTERVAL);  // --> Has to be at least 100ms!
    //ble.gap().startAdvertising();

}


uint8_t findMSDIndex(const Gap::AdvertisementCallbackParams_t *params){
    uint8_t i=0;
    uint8_t advLen = params->advertisingDataLen;
    uint8_t dataLen;
    
    if((advLen < (MAC_SIZE_B + 2)) || advLen == 0){
        // Empty advertisement or not long enough for MAX
        // +2 for SIZE and MSD ID
        return 0;
    }
    
    do{
        dataLen = params->advertisingData[i];
        i++;
        if(params->advertisingData[i] == MSD_ID) return i;
        else i += (dataLen);
    }while(i<advLen);
    
    return 0;
}

uint8_t CheckMac(const Gap::AdvertisementCallbackParams_t *params, uint8_t *myMacAddress, uint8_t msdOffset){
    int i=0;
    
    /* Get my MAC address */
    BLEProtocol::AddressType_t temp_address_type;
    ble.gap().getAddress(&temp_address_type, myMacAddress);
    
    if(!msdOffset){
        #if DEBUG_MAC
        for(i=0; i<10; i++){
            scanLED = !scanLED; // BLUE
            wait_ms(100);
        }
        #endif
        return 0;     // There's no MSD in BLE advertisement data
    }
    for(i=0; i<6; i++){
        if(params->advertisingData[msdOffset + 3 + i] != myMacAddress[5-i]){      // myMacAddress[0] == 0x91
            #if DEBUG_MAC
                for(i=0; i<10; i++){
                    connectedLED = !connectedLED; // Green
                    wait_ms(100);
                }
            #endif
            return 0;
        }
    }       
    #if DEBUG_MAC
        for(i=0; i<10; i++){
            advLED = !advLED; // RED
            wait_ms(100);
        }
        advLED = 1;
    #endif
    return 1;
}

/**
 *  Function is called when BLE radio discovers any kind of advertisment 
 */
void advertisementCallback(const Gap::AdvertisementCallbackParams_t *params){
    uint8_t msdOffset;
                    
    msdOffset = findMSDIndex(params);       // Should be 1 or 4
    if(msdOffset == 0){
        return;     // There's no MSD in BLE advertisement data
    }
    if ((params->advertisingData[msdOffset]) == MSD_ID){
        // Follows Manufacturer Specific Data
        if ((params->advertisingData[msdOffset+1]) == 0x59){
            if ((params->advertisingData[msdOffset+2]) == 0x00){
                if(CheckMac(params, myMacAddress, msdOffset)){
                    //ble.gap().stopScan();
                    buzz(900,1000);
                    WakeSleepT.detach();
                    WakeSleepT.attach(WakeMeUp, FREE_TIME_S);
                }
            }
        }
    }
}


/* Call this function few minutes (TBD) after a last scanned advertisment */
void changeSleepTime(){
    tempSleepTime = SLEEP_TIME_S;    
    sleepChanger.detach();
}


/**
 *  The function is called when ticker generates interrupt
 */
void TurnBuzzOff(void){
    //buzzer.period(0.00F);
    //buzzer.write(0.0F);
    tempSleepTime = SHORT_SLEEP_TIME_S;
    turnBuzzOffT.detach();
    WakeSleepT.detach();
    sleepChanger.attach(changeSleepTime, SHORT_SLEEP_TIME_PERIOD_S);
    WakeSleepT.attach(WakeMeUp, FREE_TIME_S);
}

void startAdvertising(){
    #if USE_ACC
        i2cPower = 1;
    #endif
    wait_ms(10);
    
    if(shushShush){
        // Do not advertise! Go to sleep
        WakeSleepT.detach();
        ble.gap().stopAdvertising();
        WakeMeUp();
    }
    else{
        ble.gap().startAdvertising();
        //buzzer.period(0.00F);
        //buzzer.write(0.0F);
        #if DEBUG
            advLED = 0;
            scanLED = 1;
        #endif
        WakeSleepT.detach();
        WakeSleepT.attach(WakeMeUp, ADV_TIMER_TIME_S);    // Call the wakeMeUp function
    }
}

void startScanning(){
    ble.gap().stopAdvertising();
    ble.gap().setScanParams(SCAN_INTERVAL, SCAN_WINDOW);
    ble.gap().setScanTimeout(SCAN_TIMER_TIME_S);
    ble.gap().startScan(advertisementCallback);
    //buzzer.period(0.00F);
    //buzzer.write(0.0F);
    #if DEBUG
        advLED = 1;
        scanLED = 0;
    #endif
    WakeSleepT.detach();
    WakeSleepT.attach(WakeMeUp, SCAN_TIMER_TIME_S);
}

void WakeMeUp(){
    sleepFlag = 0;
    switch(radioState){
        case OFF:{
                radioState = ADVERTISING;
                startAdvertising();
                break;
            }
        case ADVERTISING:{
                radioState = SCANNING;
                startScanning();
                break;
            }
        case SCANNING:{
                radioState = OFF;
                WakeSleepT.detach();
                //WakeSleepT.attach(GoToSleep, FREE_TIME_S);
                GoToSleep();
                break;
            }
        default: return;
    }
}

void GoToSleep(){
    //buzzer.period(0.00F);
    //buzzer.write(0.0f);
    WakeSleepT.detach();
    WakeSleepT.attach(WakeMeUp, tempSleepTime);
    ble.gap().stopAdvertising();
    ble.gap().stopScan();
    sleepFlag = 1;
    #if DEBUG
        advLED = 1;
        scanLED = 1;
    #endif
}

#if USE_ACC
    void pulse_handler(void){
        #if DEBUG_ACC
            int_led = !int_led;
        #endif
        //buzzer.period(0.0009F);
        //buzzer.write(0.5f);
    }
#endif

int main(void){   
    #if DEBUG || DEBUG_MAC
        advLED = 1;
        scanLED = 1;
        connectedLED = 1;
    #endif
    
    //buzzer.period(0.0009F);
    //buzzer.write(0.0F);
    #if USE_ACC
        accPower = 1;
        i2cPower = 1;
    #endif
    
    #if PRINT
        int i;
        for(i=0; i<10; i++){
            printBuffer[0] = 'B';
            printBuffer[1] = 'o';
            printBuffer[2] = 'k';
            uart.send(printBuffer, 3);
        wait_ms(100);
        }
    #endif
    
    //WakeSleepT.attach(GoToSleep, AWAKE_TIME_S);
    GoToSleep();
    ble.init(bleInitComplete);
    ble.gap().setTxPower(txPower);
    GapAdvertisingData postavke = GapAdvertisingData();
    
    #if USE_ACC
        // Pulse interrupt detection
        acc.set_register((char)CTRL_REG_4, (char) 0x04);        //  INT_EN_FF_MT Freefall/motion interrupt enabled
        wait_ms(1);
        acc.set_register((char)FF_MT_CFG, (char) 0b01011000);   //ELE, Motion Flag ON, YEFE, X Event Flag Enable
        wait_ms(1);        
        acc.set_register((char)CTRL_REG_5, (char) 0x00);        // INT_EN_FF_MT interrupt is router t0 INT2
        wait_ms(1);
        acc.set_register((char)FF_COUNT, (char) 0x08);          // Set Counter degister value (10ms)
        wait_ms(1);
        acc.set_register((char)FF_MT_THS, (char) 0x90);         // Set TH value for motion detection on 1 G (1/0.063) and DBCNTM = 1 (Increments or clears counter)
        wait_ms(1);
                
        // Setup for the interrupt handler
        accPulse.rise(&pulse_handler);                                                                                //  -------------------------------------
        acc.set_register((char)CTRL_REG_1, (char) 0x01);        // Flow data rate and Active mode           
        wait(1);
    #endif
    
    __enable_irq();
    
    /* SpinWait for initialization to complete. This is necessary because the BLE object is used in the main loop below. */
    while (ble.hasInitialized()  == false){
         /* spin loop */ 
    }
        
    while(true){
        if(sleepFlag){
            //NRF_GPIO->PIN_CNF[31] = 0x00000002;
            #if USE_ACC
                i2cPower = 0;
            #endif
            ble.waitForEvent();
            __WFI();
        }
        else{
            ble.waitForEvent();
        }
    }
}