cyclesafe/firmware/vest/vest.ino

/*********************************************************************
  This is an example for our nRF52 based Bluefruit LE modules

  Pick one up today in the adafruit shop!

  Adafruit invests time and resources providing this open source code,
  please support Adafruit and open-source hardware by purchasing
  products from Adafruit!

  MIT license, check LICENSE for more information
  All text above, and the splash screen below must be included in
  any redistribution
 *********************************************************************/
#include <bluefruit.h>
#include <Adafruit_NeoPixel.h>

#define LEFT_LED 4
#define RIGHT_LED 5

#define NUM_PIXELS 16
#define MAX_RUN_COUNT 192

enum stripStates {
	OFF,
	STRIP_START,
	STRIP_RUN,
	STRIP_STOP,
};

enum stripStates leftStripState = OFF;
enum stripStates rightStripState = OFF;

// LED Strips
Adafruit_NeoPixel leftStrip = Adafruit_NeoPixel(NUM_PIXELS, LEFT_LED, NEO_GRB + NEO_KHZ800);
Adafruit_NeoPixel rightStrip = Adafruit_NeoPixel(NUM_PIXELS, RIGHT_LED, NEO_GRB + NEO_KHZ800);

// BLE Service
BLEDis  bledis;
BLEUart bleuart;
BLEBas  blebas;

// Software Timer for blinking RED LED
SoftwareTimer softwareTimer;

void setup()
{
	Serial.begin(115200);
	Serial.println("Bluefruit52 BLEUART Example");
	Serial.println("---------------------------\n");

	// Initialize softwareTimer for 1000 ms and start it
	//softwareTimer.begin(1000, blink_timer_callback);
	softwareTimer.begin(50, run_machines_callback);
	softwareTimer.start();

	// Setup the BLE LED to be enabled on CONNECT
	// Note: This is actually the default behaviour, but provided
	// here in case you want to control this LED manually via PIN 19
	Bluefruit.autoConnLed(true);

	// Config the peripheral connection with maximum bandwidth 
	// more SRAM required by SoftDevice
	// Note: All config***() function must be called before begin()
	Bluefruit.configPrphBandwidth(BANDWIDTH_MAX);

	Bluefruit.begin();
	// Set max power. Accepted values are: -40, -30, -20, -16, -12, -8, -4, 0, 4
	Bluefruit.setTxPower(4);
	Bluefruit.setName("Bluefruit52");
	//Bluefruit.setName(getMcuUniqueID()); // useful testing with multiple central connections
	Bluefruit.setConnectCallback(connect_callback);
	Bluefruit.setDisconnectCallback(disconnect_callback);

	// Configure and Start Device Information Service
	bledis.setManufacturer("Adafruit Industries");
	bledis.setModel("Bluefruit Feather52");
	bledis.begin();

	// Configure and Start BLE Uart Service
	bleuart.begin();

	// Start BLE Battery Service
	blebas.begin();
	blebas.write(100);

	// Set up and start advertising
	startAdv();

	Serial.println("Please use Adafruit's Bluefruit LE app to connect in UART mode");
	Serial.println("Once connected, enter character(s) that you wish to send");

	leftStrip.begin();
	rightStrip.begin();
}

void startAdv(void)
{
	// Advertising packet
	Bluefruit.Advertising.addFlags(BLE_GAP_ADV_FLAGS_LE_ONLY_GENERAL_DISC_MODE);
	Bluefruit.Advertising.addTxPower();

	// Include bleuart 128-bit uuid
	Bluefruit.Advertising.addService(bleuart);

	// Secondary Scan Response packet (optional)
	// Since there is no room for 'Name' in Advertising packet
	Bluefruit.ScanResponse.addName();

	/* Start Advertising
	 * - Enable auto advertising if disconnected
	 * - Interval:  fast mode = 20 ms, slow mode = 152.5 ms
	 * - Timeout for fast mode is 30 seconds
	 * - Start(timeout) with timeout = 0 will advertise forever (until connected)
	 * 
	 * For recommended advertising interval
	 * https://developer.apple.com/library/content/qa/qa1931/_index.html   
	 */
	Bluefruit.Advertising.restartOnDisconnect(true);
	Bluefruit.Advertising.setInterval(32, 244);    // in unit of 0.625 ms
	Bluefruit.Advertising.setFastTimeout(30);      // number of seconds in fast mode
	Bluefruit.Advertising.start(0);                // 0 = Don't stop advertising after n seconds  
}

void loop()
{
	// Forward data from HW Serial to BLEUART
	while (Serial.available())
	{
		// Delay to wait for enough input, since we have a limited transmission buffer
		delay(2);

		uint8_t buf[64];
		int count = Serial.readBytes(buf, sizeof(buf));
		bleuart.write( buf, count );
	}

	// Forward from BLEUART to HW Serial
	while ( bleuart.available() )
	{
		uint8_t ch;
		ch = (uint8_t) bleuart.read();

		switch(ch) {
			case 'L':
				leftStripState = STRIP_START;
				rightStripState = STRIP_STOP;
				break;
			case 'F':
				leftStripState = STRIP_START;
				rightStripState = STRIP_START;
				break;
			case 'R':
				leftStripState = STRIP_STOP;
				rightStripState = STRIP_START;
				break;
		}
		Serial.write(ch);
	}

	// Request CPU to enter low-power mode until an event/interrupt occurs
	waitForEvent();
}

void connect_callback(uint16_t conn_handle)
{
	char central_name[32] = { 0 };
	Bluefruit.Gap.getPeerName(conn_handle, central_name, sizeof(central_name));

	Serial.print("Connected to ");
	Serial.println(central_name);
}

void disconnect_callback(uint16_t conn_handle, uint8_t reason)
{
	(void) conn_handle;
	(void) reason;

	Serial.println();
	Serial.println("Disconnected");
}

/**
 * Software Timer callback is invoked via a built-in FreeRTOS thread with
 * minimal stack size. Therefore it should be as simple as possible. If
 * a periodically heavy task is needed, please use Scheduler.startLoop() to
 * create a dedicated task for it.
 * 
 * More information http://www.freertos.org/RTOS-software-timer.html
 */
void blink_timer_callback(TimerHandle_t xTimerID)
{
	(void) xTimerID;
	digitalToggle(LED_RED);
}

/**
 * RTOS Idle callback is automatically invoked by FreeRTOS
 * when there are no active threads. E.g when loop() calls delay() and
 * there is no bluetooth or hw event. This is the ideal place to handle
 * background data.
 * 
 * NOTE: FreeRTOS is configured as tickless idle mode. After this callback
 * is executed, if there is time, freeRTOS kernel will go into low power mode.
 * Therefore waitForEvent() should not be called in this callback.
 * http://www.freertos.org/low-power-tickless-rtos.html
 * 
 * WARNING: This function MUST NOT call any blocking FreeRTOS API 
 * such as delay(), xSemaphoreTake() etc ... for more information
 * http://www.freertos.org/a00016.html
 */
void rtos_idle_callback(void)
{
	// Don't call any other FreeRTOS blocking API()
	// Perform background task(s) here
}

void processStripState(Adafruit_NeoPixel &strip, enum stripStates &stripState, int &count) {
	switch (stripState) {
		case OFF:
			break;

		case STRIP_START:
			count = 0;
			stripState = STRIP_RUN;
			break;

		case STRIP_RUN:
			for (int i = 0; i < NUM_PIXELS; i++) {
				if (i <= count % NUM_PIXELS) {
					strip.setPixelColor(i, strip.Color(150, 60, 0));
				} else {
					strip.setPixelColor(i, strip.Color(0, 0, 0));
				}
			}

			count++;
			if (count > MAX_RUN_COUNT) {
				stripState = STRIP_STOP;
			}
			break;

		case STRIP_STOP:
			for (int i = 0; i < NUM_PIXELS; i++) {
				strip.setPixelColor(i, strip.Color(0, 0, 0));
			}

			count = 0;
			stripState = OFF;
			break;
	}

	strip.show();
}

void run_machines_callback(TimerHandle_t xTimerID)
{
	(void) xTimerID;

	static int leftCount = 0;
	static int rightCount = 0;

	processStripState(leftStrip, leftStripState, leftCount);
	processStripState(rightStrip, rightStripState, rightCount);

}