#include <Arduino.h>
#define DECODE_NEC
#include <IRremote.hpp>

#define NOTE_B0  31
#define NOTE_C1  33
#define NOTE_CS1 35
#define NOTE_D1  37
#define NOTE_DS1 39
#define NOTE_E1  41
#define NOTE_F1  44
#define NOTE_FS1 46
#define NOTE_G1  49
#define NOTE_GS1 52
#define NOTE_A1  55
#define NOTE_AS1 58
#define NOTE_B1  62
#define NOTE_C2  65
#define NOTE_CS2 69
#define NOTE_D2  73
#define NOTE_DS2 78
#define NOTE_E2  82
#define NOTE_F2  87
#define NOTE_FS2 93
#define NOTE_G2  98
#define NOTE_GS2 104
#define NOTE_A2  110
#define NOTE_AS2 117
#define NOTE_B2  123
#define NOTE_C3  131
#define NOTE_CS3 139
#define NOTE_DB3 139
#define NOTE_D3  147
#define NOTE_DS3 156
#define NOTE_EB3 156
#define NOTE_E3  165
#define NOTE_F3  175
#define NOTE_FS3 185
#define NOTE_G3  196
#define NOTE_GS3 208
#define NOTE_A3  220
#define NOTE_AS3 233
#define NOTE_B3  247
#define NOTE_C4  262
#define NOTE_CS4 277
#define NOTE_D4  294
#define NOTE_DS4 311
#define NOTE_E4  330
#define NOTE_F4  349
#define NOTE_FS4 370
#define NOTE_G4  392
#define NOTE_GS4 415
#define NOTE_A4  440
#define NOTE_AS4 466
#define NOTE_B4  494
#define NOTE_C5  523
#define NOTE_CS5 554
#define NOTE_D5  587
#define NOTE_DS5 622
#define NOTE_E5  659
#define NOTE_F5  698
#define NOTE_FS5 740
#define NOTE_G5  784
#define NOTE_GS5 831
#define NOTE_A5  880
#define NOTE_AS5 932
#define NOTE_B5  988
#define NOTE_C6  1047
#define NOTE_CS6 1109
#define NOTE_D6  1175
#define NOTE_DS6 1245
#define NOTE_E6  1319
#define NOTE_F6  1397
#define NOTE_FS6 1480
#define NOTE_G6  1568
#define NOTE_GS6 1661
#define NOTE_A6  1760
#define NOTE_AS6 1865
#define NOTE_B6  1976
#define NOTE_C7  2093
#define NOTE_CS7 2217
#define NOTE_D7  2349
#define NOTE_DS7 2489
#define NOTE_E7  2637
#define NOTE_F7  2794
#define NOTE_FS7 2960
#define NOTE_G7  3136
#define NOTE_GS7 3322
#define NOTE_A7  3520
#define NOTE_AS7 3729
#define NOTE_B7  3951
#define NOTE_C8  4186
#define NOTE_CS8 4435
#define NOTE_D8  4699
#define NOTE_DS8 4978
#define REST     0

int receiver = 2;
int red = 5;
int green = 7;
int blue = 6;
int greenLight = 8;
int yellowLight = 10;
int redLight = 11;
int buzzer = 9;
//https://github.com/hibit-dev/buzzer/blob/master/src/other/xmas/xmas.ino
void playMelody(int melody[], int durations[], int size) {
  for (int note = 0; note < size; note++) {
    int duration = 1000 / durations[note];
    tone(buzzer, melody[note], duration);
    int pauseBetweenNotes = duration * 1.30;
    delay(pauseBetweenNotes);
    noTone(buzzer);
  }
}
// Xmas
int xmasMelody[] = {NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_G5, NOTE_C5, NOTE_D5, NOTE_E5, NOTE_F5, NOTE_F5, NOTE_F5, NOTE_F5, NOTE_F5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_E5, NOTE_D5, NOTE_D5, NOTE_E5, NOTE_D5, NOTE_G5};
int xmasDurations[] = {8, 8, 4, 8, 8, 4, 8, 8, 8, 8, 2, 8, 8, 8, 8, 8, 8, 8, 16, 16, 8, 8, 8, 8, 4, 4};
// Happy Birthday:
int happyBday[] = {NOTE_C4, NOTE_C4, NOTE_D4, NOTE_C4, NOTE_F4, NOTE_E4, NOTE_C4, NOTE_C4, NOTE_D4, NOTE_C4, NOTE_G4, NOTE_F4, NOTE_C4, NOTE_C4, NOTE_C5, NOTE_A4, NOTE_F4, NOTE_E4, NOTE_D4, NOTE_AS4, NOTE_AS4, NOTE_A4, NOTE_F4, NOTE_G4, NOTE_F4};
int happyBdayDurations[] = {4, 8, 4, 4, 4, 2, 4, 8, 4, 4, 4, 2, 4, 8, 4, 4, 4, 4, 4, 4, 8, 4, 4, 4, 2};


void setup() {
    Serial.begin(115200);
    while (!Serial)
        ; // Wait for Serial to become available. Is optimized away for some cores.
    IrReceiver.begin(receiver, false);
    pinMode(red, OUTPUT);
    pinMode(green, OUTPUT);
    pinMode(blue, OUTPUT);
    pinMode(greenLight, OUTPUT);
    pinMode(yellowLight, OUTPUT);
    pinMode(redLight, OUTPUT);
    pinMode(buzzer, OUTPUT);

}

void loop() {
    if (IrReceiver.decode()) {
        IrReceiver.resume(); // 

        switch(IrReceiver.decodedIRData.command) {
          case 0x47:  // Power
            turnOffLeds();
        break;
        case 0x45:
             rotateTrafficLight(3000, 1000, 3000);
        break;
        case 0x46:
            rotateRGB(500);
        break;
        case 0x44:
            setColor(255,0,0); // Red
        break;
        case 0x40:
            setColor(255,255,0); // Yellow
        break;
        case 0x43:
            setColor(255,0,255);  // Purple
        break;
        case 0x7:
            setColor(0,255,0); // Green
        break;
        case 0x15:
            setColor(70,175,255); // Cyan
        break;
        case 0x9:
            setColor(255,160,0); // Orange
        break;
        case 0x16:
            setColor(0,0,255); // Blue
        break;
        case 0x19:
            setColor(128,0,128);  // Pink
        break;
        case 0xD:
            setColor(255,255,255); // White
        break;
        case 0xC:
            turnOffLeds();
            digitalWrite(greenLight, HIGH); // Green TrafficLight
        break;
        case 0x18:
            turnOffLeds();
            digitalWrite(yellowLight, HIGH); // Yellow TrafficLight
        break;
        case 0x5E:
            turnOffLeds();
            digitalWrite(redLight, HIGH); // Red trafficLight
        break;
        case 0x8:  // Play HappyBday
            turnOffLeds();
            IrReceiver.stopTimer(); 
            playMelody(happyBday, happyBdayDurations, sizeof(happyBdayDurations) / sizeof(int));
            delay(10);
            IrReceiver.restartTimer();
        break;
        case 0x1C:  // Play Xmas
            turnOffLeds();
            IrReceiver.stopTimer();
            playMelody(xmasMelody, xmasDurations, sizeof(xmasDurations) / sizeof(int));
            delay(10);
            IrReceiver.restartTimer();
        break;
        case 0x5A:
            blinkWhite();
        break;
        }
    }
}

void turnOffLeds(){
  // RGB
  digitalWrite(red, LOW);
  digitalWrite(green, LOW);
  digitalWrite(blue, LOW);
  // traffic light
  digitalWrite(redLight, LOW);
  digitalWrite(greenLight, LOW);
  digitalWrite(yellowLight, LOW);
}

void rotateRGB(int gap) {
    turnOffLeds();
    setColor(255,0,0); // Red
    delay(gap);
    setColor(255,255,0); // Yellow
    delay(gap);
    setColor(255,0,255);  // Purple
    delay(gap);
    setColor(0,255,0); // green
    delay(gap);
    setColor(70,175,255); // cyan
    delay(gap);
    setColor(255,160,0); // orange
    delay(gap);
    setColor(0,0,255); // blue
    delay(gap);
    setColor(128,0,128);  // pink
    delay(gap);
    setColor(255,255,255); // white
    delay(gap);
    turnOffLeds();
}

void rotateTrafficLight(int greenPause, int yellowPause, int redPause) {
  turnOffLeds();
  digitalWrite(greenLight, HIGH);
  delay(greenPause);
  turnOffLeds();
  digitalWrite(yellowLight, HIGH);
  delay(yellowPause);
  turnOffLeds();
  digitalWrite(redLight, HIGH);
  delay(redPause);
  turnOffLeds();
}

void blinkWhite() {
  for(int i=0; i<5; i++) {
    turnOffLeds();
    setColor(255, 255, 255);
    delay(500);
    turnOffLeds();
    delay(500);
  }
}

void setColor(int redValue, int greenValue, int blueValue) {
  turnOffLeds();
  analogWrite(red, redValue);
  analogWrite(green, greenValue);
  analogWrite(blue, blueValue);
}