dlite/src/LEDState.c

/* LEDState.c */

#include "mgos_sys_config.h"
#include "mgos_timers.h"
#include "LEDDefinition.h"
#include "LEDState.h"
#include "AnimationConfig.h"
#include "NeoPixel.h"

static LEDStateEngine theLEDStateEngine;
static int ticks = 0;

int getTicks(void) { return ticks; }

void LEDStateEngine_setNumberOfLeds(int numberOfLeds) {
	if (numberOfLeds < 1 || numberOfLeds > MAX_LEDS) {
		LOG(LL_ERROR, ("Invalid number of LEDs %d", numberOfLeds));
		numberOfLeds = MAX_LEDS;
	}
	NeoPixel_release();
	NeoPixel_create(theLEDStateEngine.pin, numberOfLeds, NeoPixel_colorOrder_RGB);
	theLEDStateEngine.numberOfLeds = numberOfLeds;
	LOG(LL_INFO, ("LEDStateEngine initialized: numLEDs=%d, MAX_LEDS=%d, pin=%d", numberOfLeds, MAX_LEDS, theLEDStateEngine.pin));
}

void LEDStateEngine_init(int ledPin, int numberOfLeds) {
	theLEDStateEngine.comment = "";
	theLEDStateEngine.pin = ledPin;
	LEDStateEngine_setNumberOfLeds(numberOfLeds);
	for (int i=0; i<numberOfLeds; ++i) {
		LEDState_init(i, LEDDefinition_get(i));
	}
}

void LEDStateEngine_addComment(char *comment) {
	theLEDStateEngine.comment = comment;
}

uint16_t LEDStateEngine_getNumberOfLeds() {
	return theLEDStateEngine.numberOfLeds;
}

static double minTickTime = 99999.99;
static double maxTickTime = 0;

void LEDStateEngine_tick() {
	double startTime = mgos_uptime();
	++ticks;
	// LOG(LL_INFO, ("# %d/%d", theLEDStateEngine.numberOfLeds, ticks));
	for (int i=0; i<theLEDStateEngine.numberOfLeds; ++i) {
		LEDState_tick(i);
	}
	NeoPixel_show();
	double endTime = mgos_uptime();
	double usedTime = endTime - startTime;
	if (usedTime > maxTickTime) maxTickTime = usedTime;
	if (usedTime < minTickTime) minTickTime = usedTime;
}
double LEDStateEngine_getMinTickTime() { return 1000 * minTickTime; }
double LEDStateEngine_getMaxTickTime() { return 1000 * maxTickTime; }

static void updateLedDisplay(LEDState *state) {
	NeoPixel_set(state->index,
		state->currentColor.red,
		state->currentColor.green,
		state->currentColor.blue);
}

void LEDState_tick(int ledNum) {
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	state->currentTick = state->currentTick + 1;
	// LOG(LL_INFO,("LEDState_tick(%d) %d", ledNum, state->currentTick));
	if (state->currentTick == state->nextTick) {
		// perform/initiate next state change
		//LOG(LL_INFO, ("State change for led %d", ledNum));
		state->currentAnimationStep = AnimationConfig_getNextAnimationStepForLED(ledNum, state->currentAnimationStep);
		state->nextTick = state->currentTick + theAnimationConfig.animationStep[state->currentAnimationStep].durationTicks;
		state->mode = theAnimationConfig.animationStep[state->currentAnimationStep].ledMode;
		switch (state->mode) {
			case LEDMode_on:
				state->currentColor = state->ledDefinition->onColor; // state->onColor;
				break;
			case LEDMode_off:
				state->currentColor = offColor;
				break;
			default:
				state->currentColor = dimmedRedColor;
				break;
		}
		updateLedDisplay(state);
	}
	// do animations, if necessary
}

LEDState *LEDStateEngine_getLedState(int n) {
	if (n>=0 && n<theLEDStateEngine.numberOfLeds)
		return &theLEDStateEngine.ledState[n];
	LOG(LL_ERROR, ("*** LEDStateEngine_getLedState -- invalid LED number given: %d", n));
	return &theLEDStateEngine.ledState[0];
}

/* LEDState */
static int verifyLedNumber(int n) {
	if (n < 0 || n >= theLEDStateEngine.numberOfLeds) {
		LOG(LL_ERROR, ("*** Invalid LED number %d, set to zero", n));
		n = 0;
	}
	return n;
}

void LEDState_initDefault(int n, LEDMode defaultMode) {
	n = verifyLedNumber(n);
	LEDState *state = LEDStateEngine_getLedState(n);
	state->index = n;
	state->mode = defaultMode;
	state->currentTick = -10;
	state->nextTick = 0;
	state->ledDefinition = LEDDefinition_get(n);
	state->currentAnimationStep = AnimationConfig_getNextAnimationStepForLED(n, 0);
}

void LEDState_init(int n, LEDDefinition *ledDefinition) {
	n = verifyLedNumber(n);
	LEDState_initDefault(n, LEDMode_off);
	LEDState *state = LEDStateEngine_getLedState(n);
	state->ledDefinition = ledDefinition;
	// state->onColor = ledDefinition->onColor;
	state->onColor.name = ledDefinition->onColor.name;
	state->onColor.red = ledDefinition->onColor.red;
	state->onColor.green = ledDefinition->onColor.green;
	state->onColor.blue = ledDefinition->onColor.blue;
}


int LEDState_getLedRed(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->currentColor.red;
}
int LEDState_getLedGreen(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->currentColor.green;
}
int LEDState_getLedBlue(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->currentColor.blue;
}
char * LEDState_getLedColorName(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->currentColor.name;
}
int LEDState_getCurrentTick(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->currentTick;
}
int LEDState_getNextTick(int ledNum) {
	ledNum = verifyLedNumber(ledNum);
	LEDState *state = LEDStateEngine_getLedState(ledNum);
	return state->nextTick;
}