Andrea Vicari / Mbed OS BLELedScanner

Write a proximity-aware Mbed-based LED scanner!

source/main.cpp@5:2d307782cb69, 2018-11-30 (annotated)

Committer:
vicara
Date:
Fri Nov 30 18:34:23 2018 +0000
Revision:
5:2d307782cb69
Parent:
4:28c05d426fc1 
fix

Who changed what in which revision?

UserRevisionLine numberNew contents of line
vicara 3:f6bf91b5f492 1 /* mbed Microcontroller Library
vicara 3:f6bf91b5f492 2 * Copyright (c) 2006-2015 ARM Limited
vicara 3:f6bf91b5f492 3 *
vicara 3:f6bf91b5f492 4 * Licensed under the Apache License, Version 2.0 (the "License");
vicara 3:f6bf91b5f492 5 * you may not use this file except in compliance with the License.
vicara 3:f6bf91b5f492 6 * You may obtain a copy of the License at
vicara 3:f6bf91b5f492 7 *
vicara 3:f6bf91b5f492 8 * http://www.apache.org/licenses/LICENSE-2.0
vicara 3:f6bf91b5f492 9 *
vicara 3:f6bf91b5f492 10 * Unless required by applicable law or agreed to in writing, software
vicara 3:f6bf91b5f492 11 * distributed under the License is distributed on an "AS IS" BASIS,
vicara 3:f6bf91b5f492 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
vicara 3:f6bf91b5f492 13 * See the License for the specific language governing permissions and
vicara 3:f6bf91b5f492 14 * limitations under the License.
vicara 3:f6bf91b5f492 15 */
vicara 3:f6bf91b5f492 16
vicara 3:f6bf91b5f492 17 #include <events/mbed_events.h>
vicara 3:f6bf91b5f492 18 #include <mbed.h>
vicara 3:f6bf91b5f492 19 #include "ble/BLE.h"
vicara 3:f6bf91b5f492 20 #include "ble/DiscoveredCharacteristic.h"
vicara 3:f6bf91b5f492 21 #include "ble/DiscoveredService.h"
vicara 3:f6bf91b5f492 22
vicara 3:f6bf91b5f492 23 DigitalOut alivenessLED(LED1, 1);
vicara 3:f6bf91b5f492 24 static DiscoveredCharacteristic ledCharacteristic;
vicara 3:f6bf91b5f492 25 static bool triggerLedCharacteristic;
vicara 3:f6bf91b5f492 26 static const char PEER_NAME[] = "LED_RED";
vicara 3:f6bf91b5f492 27
vicara 3:f6bf91b5f492 28 DigitalOut led1(LED1);
vicara 3:f6bf91b5f492 29 DigitalOut led2(LED2);
vicara 3:f6bf91b5f492 30 DigitalOut led3(LED3);
vicara 3:f6bf91b5f492 31
vicara 5:2d307782cb69 32 static bool led1_on = false;
vicara 5:2d307782cb69 33 static bool led2_on = false;
vicara 5:2d307782cb69 34 static bool led3_on = false;
vicara 3:f6bf91b5f492 35
vicara 5:2d307782cb69 36 void blink_led_check(void) {
vicara 5:2d307782cb69 37 if(led1_on){
vicara 5:2d307782cb69 38 led1 = !led1;
vicara 5:2d307782cb69 39 led2 = 0;
vicara 5:2d307782cb69 40 led3 = 0;
vicara 5:2d307782cb69 41 }else if(led2_on){
vicara 5:2d307782cb69 42 led1 = 0;
vicara 5:2d307782cb69 43 led2 = !led2;
vicara 5:2d307782cb69 44 led3 = 0;
vicara 5:2d307782cb69 45 }else if(led3_on){
vicara 5:2d307782cb69 46 led1 = 0;
vicara 5:2d307782cb69 47 led2 = 0;
vicara 5:2d307782cb69 48 led3 = !led3;
vicara 5:2d307782cb69 49 }
vicara 3:f6bf91b5f492 50 }
vicara 3:f6bf91b5f492 51
vicara 5:2d307782cb69 52 static EventQueue eventQueue(/* event count */ 16 * EVENTS_EVENT_SIZE);
vicara 5:2d307782cb69 53
vicara 3:f6bf91b5f492 54 void advertisementCallback(const Gap::AdvertisementCallbackParams_t *params) {
vicara 3:f6bf91b5f492 55 // parse the advertising payload, looking for data type COMPLETE_LOCAL_NAME
vicara 3:f6bf91b5f492 56 // The advertising payload is a collection of key/value records where
vicara 3:f6bf91b5f492 57 // byte 0: length of the record excluding this byte
vicara 3:f6bf91b5f492 58 // byte 1: The key, it is the type of the data
vicara 3:f6bf91b5f492 59 // byte [2..N] The value. N is equal to byte0 - 1
vicara 5:2d307782cb69 60
vicara 5:2d307782cb69 61 printf("Device(s) discovered: %d \n", &params->advertisingDataLen);
vicara 5:2d307782cb69 62
vicara 5:2d307782cb69 63
vicara 3:f6bf91b5f492 64 for (uint8_t i = 0; i < params->advertisingDataLen; ++i) {
vicara 3:f6bf91b5f492 65
vicara 3:f6bf91b5f492 66 const uint8_t record_length = params->advertisingData[i];
vicara 3:f6bf91b5f492 67 if (record_length == 0) {
vicara 3:f6bf91b5f492 68 continue;
vicara 3:f6bf91b5f492 69 }
vicara 3:f6bf91b5f492 70 const uint8_t type = params->advertisingData[i + 1];
vicara 3:f6bf91b5f492 71 const uint8_t* value = params->advertisingData + i + 2;
vicara 3:f6bf91b5f492 72 const uint8_t value_length = record_length - 1;
vicara 3:f6bf91b5f492 73
vicara 4:28c05d426fc1 74 // printf("Device discovered: %d \n", &value);
vicara 3:f6bf91b5f492 75
vicara 3:f6bf91b5f492 76 if(type == GapAdvertisingData::COMPLETE_LOCAL_NAME) {
vicara 3:f6bf91b5f492 77 if ((value_length == sizeof(PEER_NAME)) && (memcmp(value, PEER_NAME, value_length) == 0)) {
vicara 4:28c05d426fc1 78 printf("Distance from %d is %d\n", &PEER_NAME, params->rssi);
vicara 5:2d307782cb69 79 int distance = params->rssi;
vicara 5:2d307782cb69 80 if(distance >= -63){
vicara 5:2d307782cb69 81 led1_on = true;
vicara 5:2d307782cb69 82 led2_on = false;
vicara 5:2d307782cb69 83 led3_on = false;
vicara 5:2d307782cb69 84 }else if(distance >= -85){
vicara 5:2d307782cb69 85 led1_on = false;
vicara 5:2d307782cb69 86 led2_on = true;
vicara 5:2d307782cb69 87 led3_on = false;
vicara 5:2d307782cb69 88 }else{
vicara 5:2d307782cb69 89 led1_on = false;
vicara 5:2d307782cb69 90 led2_on = false;
vicara 5:2d307782cb69 91 led3_on = true;
vicara 3:f6bf91b5f492 92 }
vicara 3:f6bf91b5f492 93 // printf(
vicara 3:f6bf91b5f492 94 // "adv peerAddr[%02x %02x %02x %02x %02x %02x] rssi %d, isScanResponse %u, AdvertisementType %u\r\n",
vicara 3:f6bf91b5f492 95 // params->peerAddr[5], params->peerAddr[4], params->peerAddr[3], params->peerAddr[2],
vicara 3:f6bf91b5f492 96 // params->peerAddr[1], params->peerAddr[0], params->rssi, params->isScanResponse, params->type
vicara 3:f6bf91b5f492 97 // );
vicara 3:f6bf91b5f492 98 // BLE::Instance().gap().connect(params->peerAddr, Gap::ADDR_TYPE_RANDOM_STATIC, NULL, NULL);
vicara 3:f6bf91b5f492 99 break;
vicara 3:f6bf91b5f492 100 }
vicara 3:f6bf91b5f492 101 }
vicara 3:f6bf91b5f492 102
vicara 3:f6bf91b5f492 103 i += record_length;
vicara 3:f6bf91b5f492 104 }
vicara 3:f6bf91b5f492 105 }
vicara 3:f6bf91b5f492 106
vicara 3:f6bf91b5f492 107 void serviceDiscoveryCallback(const DiscoveredService *service) {
vicara 3:f6bf91b5f492 108 if (service->getUUID().shortOrLong() == UUID::UUID_TYPE_SHORT) {
vicara 3:f6bf91b5f492 109 printf("S UUID-%x attrs[%u %u]\r\n", service->getUUID().getShortUUID(), service->getStartHandle(), service->getEndHandle());
vicara 3:f6bf91b5f492 110 } else {
vicara 3:f6bf91b5f492 111 printf("S UUID-");
vicara 3:f6bf91b5f492 112 const uint8_t *longUUIDBytes = service->getUUID().getBaseUUID();
vicara 3:f6bf91b5f492 113 for (unsigned i = 0; i < UUID::LENGTH_OF_LONG_UUID; i++) {
vicara 3:f6bf91b5f492 114 printf("%02x", longUUIDBytes[i]);
vicara 3:f6bf91b5f492 115 }
vicara 3:f6bf91b5f492 116 printf(" attrs[%u %u]\r\n", service->getStartHandle(), service->getEndHandle());
vicara 3:f6bf91b5f492 117 }
vicara 3:f6bf91b5f492 118 }
vicara 3:f6bf91b5f492 119
vicara 3:f6bf91b5f492 120 void updateLedCharacteristic(void) {
vicara 3:f6bf91b5f492 121 if (!BLE::Instance().gattClient().isServiceDiscoveryActive()) {
vicara 3:f6bf91b5f492 122 ledCharacteristic.read();
vicara 3:f6bf91b5f492 123 }
vicara 3:f6bf91b5f492 124 }
vicara 3:f6bf91b5f492 125
vicara 3:f6bf91b5f492 126 void characteristicDiscoveryCallback(const DiscoveredCharacteristic *characteristicP) {
vicara 3:f6bf91b5f492 127 printf(" C UUID-%x valueAttr[%u] props[%x]\r\n", characteristicP->getUUID().getShortUUID(), characteristicP->getValueHandle(), (uint8_t)characteristicP->getProperties().broadcast());
vicara 3:f6bf91b5f492 128 if (characteristicP->getUUID().getShortUUID() == 0xa001) { /* !ALERT! Alter this filter to suit your device. */
vicara 3:f6bf91b5f492 129 ledCharacteristic = *characteristicP;
vicara 3:f6bf91b5f492 130 triggerLedCharacteristic = true;
vicara 3:f6bf91b5f492 131 }
vicara 3:f6bf91b5f492 132 }
vicara 3:f6bf91b5f492 133
vicara 3:f6bf91b5f492 134 void discoveryTerminationCallback(Gap::Handle_t connectionHandle) {
vicara 3:f6bf91b5f492 135 printf("terminated SD for handle %u\r\n", connectionHandle);
vicara 3:f6bf91b5f492 136 if (triggerLedCharacteristic) {
vicara 3:f6bf91b5f492 137 triggerLedCharacteristic = false;
vicara 3:f6bf91b5f492 138 eventQueue.call(updateLedCharacteristic);
vicara 3:f6bf91b5f492 139 }
vicara 3:f6bf91b5f492 140 }
vicara 3:f6bf91b5f492 141
vicara 3:f6bf91b5f492 142 void connectionCallback(const Gap::ConnectionCallbackParams_t *params) {
vicara 3:f6bf91b5f492 143 if (params->role == Gap::CENTRAL) {
vicara 3:f6bf91b5f492 144 BLE &ble = BLE::Instance();
vicara 3:f6bf91b5f492 145 ble.gattClient().onServiceDiscoveryTermination(discoveryTerminationCallback);
vicara 3:f6bf91b5f492 146 ble.gattClient().launchServiceDiscovery(params->handle, serviceDiscoveryCallback, characteristicDiscoveryCallback, 0xa000, 0xa001);
vicara 3:f6bf91b5f492 147 }
vicara 3:f6bf91b5f492 148 }
vicara 3:f6bf91b5f492 149
vicara 3:f6bf91b5f492 150 void triggerToggledWrite(const GattReadCallbackParams *response) {
vicara 3:f6bf91b5f492 151 if (response->handle == ledCharacteristic.getValueHandle()) {
vicara 3:f6bf91b5f492 152 printf("triggerToggledWrite: handle %u, offset %u, len %u\r\n", response->handle, response->offset, response->len);
vicara 3:f6bf91b5f492 153 for (unsigned index = 0; index < response->len; index++) {
vicara 3:f6bf91b5f492 154 printf("%c[%02x]", response->data[index], response->data[index]);
vicara 3:f6bf91b5f492 155 }
vicara 3:f6bf91b5f492 156 printf("\r\n");
vicara 3:f6bf91b5f492 157
vicara 3:f6bf91b5f492 158 uint8_t toggledValue = response->data[0] ^ 0x1;
vicara 3:f6bf91b5f492 159 ledCharacteristic.write(1, &toggledValue);
vicara 3:f6bf91b5f492 160 }
vicara 3:f6bf91b5f492 161 }
vicara 3:f6bf91b5f492 162
vicara 3:f6bf91b5f492 163 void triggerRead(const GattWriteCallbackParams *response) {
vicara 3:f6bf91b5f492 164 if (response->handle == ledCharacteristic.getValueHandle()) {
vicara 3:f6bf91b5f492 165 ledCharacteristic.read();
vicara 3:f6bf91b5f492 166 }
vicara 3:f6bf91b5f492 167 }
vicara 3:f6bf91b5f492 168
vicara 3:f6bf91b5f492 169 void disconnectionCallback(const Gap::DisconnectionCallbackParams_t *) {
vicara 3:f6bf91b5f492 170 printf("disconnected\r\n");
vicara 3:f6bf91b5f492 171 /* Start scanning and try to connect again */
vicara 3:f6bf91b5f492 172 BLE::Instance().gap().startScan(advertisementCallback);
vicara 3:f6bf91b5f492 173 }
vicara 3:f6bf91b5f492 174
vicara 3:f6bf91b5f492 175 void onBleInitError(BLE &ble, ble_error_t error)
vicara 3:f6bf91b5f492 176 {
vicara 3:f6bf91b5f492 177 /* Initialization error handling should go here */
vicara 3:f6bf91b5f492 178 }
vicara 3:f6bf91b5f492 179
vicara 3:f6bf91b5f492 180 void printMacAddress()
vicara 3:f6bf91b5f492 181 {
vicara 3:f6bf91b5f492 182 /* Print out device MAC address to the console*/
vicara 3:f6bf91b5f492 183 Gap::AddressType_t addr_type;
vicara 3:f6bf91b5f492 184 Gap::Address_t address;
vicara 3:f6bf91b5f492 185 BLE::Instance().gap().getAddress(&addr_type, address);
vicara 3:f6bf91b5f492 186 printf("DEVICE MAC ADDRESS: ");
vicara 3:f6bf91b5f492 187 for (int i = 5; i >= 1; i--){
vicara 3:f6bf91b5f492 188 printf("%02x:", address[i]);
vicara 3:f6bf91b5f492 189 }
vicara 3:f6bf91b5f492 190 printf("%02x\r\n", address[0]);
vicara 3:f6bf91b5f492 191 }
vicara 3:f6bf91b5f492 192
vicara 3:f6bf91b5f492 193 void bleInitComplete(BLE::InitializationCompleteCallbackContext *params)
vicara 3:f6bf91b5f492 194 {
vicara 3:f6bf91b5f492 195 BLE& ble = params->ble;
vicara 3:f6bf91b5f492 196 ble_error_t error = params->error;
vicara 3:f6bf91b5f492 197
vicara 3:f6bf91b5f492 198 if (error != BLE_ERROR_NONE) {
vicara 3:f6bf91b5f492 199 /* In case of error, forward the error handling to onBleInitError */
vicara 3:f6bf91b5f492 200 onBleInitError(ble, error);
vicara 3:f6bf91b5f492 201 return;
vicara 3:f6bf91b5f492 202 }
vicara 3:f6bf91b5f492 203
vicara 3:f6bf91b5f492 204 /* Ensure that it is the default instance of BLE */
vicara 3:f6bf91b5f492 205 if (ble.getInstanceID() != BLE::DEFAULT_INSTANCE) {
vicara 3:f6bf91b5f492 206 return;
vicara 3:f6bf91b5f492 207 }
vicara 3:f6bf91b5f492 208
vicara 3:f6bf91b5f492 209 ble.gap().onDisconnection(disconnectionCallback);
vicara 3:f6bf91b5f492 210 ble.gap().onConnection(connectionCallback);
vicara 3:f6bf91b5f492 211
vicara 3:f6bf91b5f492 212 ble.gattClient().onDataRead(triggerToggledWrite);
vicara 3:f6bf91b5f492 213 ble.gattClient().onDataWrite(triggerRead);
vicara 3:f6bf91b5f492 214
vicara 3:f6bf91b5f492 215 // scan interval: 400ms and scan window: 400ms.
vicara 3:f6bf91b5f492 216 // Every 400ms the device will scan for 400ms
vicara 3:f6bf91b5f492 217 // This means that the device will scan continuously.
vicara 3:f6bf91b5f492 218 ble.gap().setScanParams(400, 400);
vicara 3:f6bf91b5f492 219 ble.gap().startScan(advertisementCallback);
vicara 3:f6bf91b5f492 220
vicara 3:f6bf91b5f492 221 printMacAddress();
vicara 3:f6bf91b5f492 222 }
vicara 3:f6bf91b5f492 223
vicara 3:f6bf91b5f492 224 void scheduleBleEventsProcessing(BLE::OnEventsToProcessCallbackContext* context) {
vicara 3:f6bf91b5f492 225 BLE &ble = BLE::Instance();
vicara 3:f6bf91b5f492 226 eventQueue.call(Callback<void()>(&ble, &BLE::processEvents));
vicara 3:f6bf91b5f492 227 }
vicara 3:f6bf91b5f492 228
vicara 3:f6bf91b5f492 229 int main()
vicara 3:f6bf91b5f492 230 {
vicara 3:f6bf91b5f492 231 triggerLedCharacteristic = false;
vicara 5:2d307782cb69 232 eventQueue.call_every(500, blink_led_check);
vicara 3:f6bf91b5f492 233 BLE &ble = BLE::Instance();
vicara 3:f6bf91b5f492 234 ble.onEventsToProcess(scheduleBleEventsProcessing);
vicara 3:f6bf91b5f492 235 ble.init(bleInitComplete);
vicara 3:f6bf91b5f492 236
vicara 3:f6bf91b5f492 237 eventQueue.dispatch_forever();
vicara 3:f6bf91b5f492 238
vicara 3:f6bf91b5f492 239 return 0;
vicara 3:f6bf91b5f492 240 }