/// -*- tab-width: 4; Mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*-

/*
   This program 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.

   This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef __EXTERNALLED_H__
#define __EXTERNALLED_H__

#include <AP_Common.h>
#include <AP_HAL.h>
#include <AP_Param.h>

#if CONFIG_HAL_BOARD == HAL_BOARD_APM2
 #define EXTERNAL_LED_ARMED   61    // Armed LED - AN7
 #define EXTERNAL_LED_GPS     60    // GPS LED - AN6
 #define EXTERNAL_LED_MOTOR1  58    // Motor1 LED - AN4
 #define EXTERNAL_LED_MOTOR2  62    // Motor2 LED - AN8
#elif CONFIG_HAL_BOARD == HAL_BOARD_APM1
 #define EXTERNAL_LED_GPS     64    // GPS LED - AN10
 #define EXTERNAL_LED_ARMED   65    // Armed LED - AN11
 #define EXTERNAL_LED_MOTOR1  62    // Motor1 LED - AN8
 #define EXTERNAL_LED_MOTOR2  66    // Motor2 LED - AN12
#elif CONFIG_HAL_BOARD == HAL_BOARD_VRBRAIN
 #define EXTERNAL_LED_GPS     28    // GPS LED - AN10
 #define EXTERNAL_LED_ARMED   29    // Armed LED - AN11
 #define EXTERNAL_LED_MOTOR1  30    // Motor1 LED - AN8
 #define EXTERNAL_LED_MOTOR2  31    // Motor2 LED - AN12
#else
 #define EXTERNAL_LED_GPS     0     // pin definitions to allow this lib to build for
 #define EXTERNAL_LED_ARMED   0     // for other boards besides APM1, APM2 even though
 #define EXTERNAL_LED_MOTOR1  0     // they will never be used
 #define EXTERNAL_LED_MOTOR2  0
#endif

class ExternalLED
{
public:
    // constructor
    ExternalLED() : _counter(0), _counter2(0), _pattern(NONE), _pattern_counter(0) {}

    // initialise the LED driver
    void init(void);
    
    // should be called at 50Hz
    void update(void);

private:

    enum LEDPattern {
        NONE = 0,
        FAST_FLASH = 1,
        OSCILLATE = 2
    };

    /// buzzer_flag_type - bitmask of current state and ap_notify states we track
    struct copterleds_flag_type {
        // individual led status
        uint8_t armedled_on         : 1;    // 1 if the armed led is currently on
        uint8_t gpsled_on           : 1;    // 1 if the gps led is currently on
        uint8_t motorled1_on        : 1;    // 1 if motor led #1 is currently on
        uint8_t motorled2_on        : 1;    // 1 if motor led #2 is currently on
    } _flags;

    uint8_t         _counter;           // reduces 50hz calls to 10hz
    uint8_t         _counter2;          // used to control steps of arming and gps leds
    LEDPattern      _pattern;           // current pattern for motor leds
    uint8_t         _pattern_counter;   // used to time on/off of current patter

    // armed_led - set armed light on or off
    void armed_led(bool on_off);

    // gps_led - set gps light on or off
    void gps_led(bool on_off);

    // set_pattern - sets pattern for motor leds
    void set_pattern(LEDPattern pattern_id);

    // motor_led1, motor_led2 - set motor lights on or off
    void motor_led1(bool on_off);
    void motor_led2(bool on_off);
};

#endif // __EXTERNALLED_H__