leds.cpp

/*
 * This file is part of LynxGradProject.
 *
 * LynxGradProject is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LynxGradProject is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LynxGradProject.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "leds.h"
#include <ChibiOS_AVR.h>

bool LEDs::s_Initialized = false;

LEDs::BlinkType LEDs::s_BlinkType[c_LEDCount] = {BlinkOff, BlinkOff};
unsigned long LEDs::s_BlinkTime[c_LEDCount][4] = {{0, 0, 0, 0}, {0, 0, 0, 0}};
bool LEDs::s_State[c_LEDCount] = {false, false};
byte LEDs::s_BlinkStage[c_LEDCount] = {0, 0};

const byte LEDs::c_nLEDPin[c_LEDCount] = {10, 8};

LEDs::LEDs()
{
    if (s_Initialized == false) {
        s_Initialized = true;

        for (byte index = 0; index < c_LEDCount; index++) {
            pinMode(c_nLEDPin[index], OUTPUT);
            turnOff((LEDColor)index);
        }
    }
}

LEDs::~LEDs()
{

}

void LEDs::setPattern(LEDs::LEDColor color, LEDs::BlinkType blinkType)
{
    if (blinkType == BlinkOff) {
        s_State[color] = false;
    } else if (blinkType == BlinkNone) {
        s_State[color] = true;
    } else if (blinkType == Blink2Short) {
        s_BlinkTime[color][0] = c_Blink2ShortOnTime1;
        s_BlinkTime[color][1] = c_Blink2ShortOffTime1;
        s_BlinkTime[color][2] = c_Blink2ShortOnTime2;
        s_BlinkTime[color][3] = c_Blink2ShortOffTime2;
        s_State[color] = true;
    } else if (blinkType == Blink50DC) {
        s_BlinkTime[color][0] = c_Blink50DCOnTime1;
        s_BlinkTime[color][1] = c_Blink50DCOffTime1;
        s_BlinkTime[color][2] = c_Blink50DCOnTime1;
        s_BlinkTime[color][3] = c_Blink50DCOffTime1;
        s_State[color] = true;
    } else if (blinkType == Blink2ShortStop) {
        s_BlinkTime[color][0] = c_Blink2ShortStopOnTime1;
        s_BlinkTime[color][1] = c_Blink2ShortStopOffTime1;
        s_BlinkTime[color][2] = c_Blink2ShortStopOnTime2;
        s_BlinkTime[color][3] = c_Blink2ShortStopOffTime2;
        s_State[color] = true;
    } else {
        return;
    }

    s_BlinkType[color] = blinkType;
    s_BlinkStage[color] = 0;

    digitalWrite(c_nLEDPin[color], !s_State[color]);
}

void LEDs::turnOn(LEDs::LEDColor color)
{
    setPattern(color, BlinkNone);
}

void LEDs::turnOff(LEDs::LEDColor color)
{
    setPattern(color, BlinkOff);
}

void LEDs::run(LEDs::LEDColor color)
{
    while (1) {
        if (s_BlinkType[color] == BlinkNone) {
            s_State[color] = true;
            digitalWrite(c_nLEDPin[color], !s_State[color]);
            chThdSleepMilliseconds(50);
            continue;
        }
        if (s_BlinkType[color] == BlinkOff) {
            s_State[color] = false;
            digitalWrite(c_nLEDPin[color], !s_State[color]);
            chThdSleepMilliseconds(50);
            continue;
        }

        chThdSleepMilliseconds(s_BlinkTime[color][s_BlinkStage[color]]);
        s_State[color] = !s_State[color];
        s_BlinkStage[color]++;

        if (s_BlinkStage[color] > 3) {
            if (s_BlinkType[color] == Blink2ShortStop) {
                turnOff((LEDColor)color);
            }
            s_BlinkStage[color] = 0;
        }

        digitalWrite(c_nLEDPin[color], !s_State[color]);
    }
}