#include <SevSeg.h>

SevSeg sevseg1;  // Objek untuk seven-segment display pertama (untuk lampu merah 1)
SevSeg sevseg2;  // Objek untuk seven-segment display kedua (untuk lampu merah 2)


// Pin LED
const int red1 = 2;
const int yellow1 = 3;
const int green1 = 4;
const int red2 = 5;
const int yellow2 = 6;
const int green2 = 7;

// Pin Buzzer
const int buzz = 8;

// Durasi not
int noteDurations[] = {
  4, 4, 4, 4, 4, 4, 
  4, 4, 4, 4, 4, 4, 
  4, 4, 4, 4, 4, 4, 4, 
  4, 4, 4, 4, 4, 4 
};


// Definisi nada dalam Hz
#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_D3  147
#define NOTE_DS3 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

#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_D3  147
#define NOTE_DS3 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

// Melodi sederhana
int melody1[] = {

  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 melody2[] = {
  
  // Dart Vader theme (Imperial March) - Star wars 
  // Score available at https://musescore.com/user/202909/scores/1141521
  // The tenor saxophone part was used
  
  NOTE_AS4, NOTE_AS4, NOTE_AS4,8,//1
  NOTE_F5, NOTE_C6,
  NOTE_AS5, NOTE_A5, NOTE_G5, NOTE_F6, NOTE_C6,  
  NOTE_AS5, NOTE_A5, NOTE_G5, NOTE_F6, NOTE_C6,  
  NOTE_AS5, NOTE_A5, NOTE_AS5, NOTE_G5, NOTE_C5, NOTE_C5, NOTE_C5,
  NOTE_F5, NOTE_C6,
  NOTE_AS5
};

bool playMelody1 = true;

void setup() {
  // Atur pin LED dan buzz sebagai output
  pinMode(red1, OUTPUT);
  pinMode(yellow1, OUTPUT);
  pinMode(green1, OUTPUT);
  pinMode(buzz, OUTPUT);
  pinMode(red2, OUTPUT);
  pinMode(yellow2, OUTPUT);
  pinMode(green2, OUTPUT);

  byte numDigits1 = 1;
  byte digitPins1[] = {11};  // Pin digit untuk display 1
  byte segmentPins1[] = {9, 10, 12, 13, A0, A1, A2};  // Pin untuk segmen a-g
  sevseg1.begin(COMMON_ANODE, numDigits1, digitPins1, segmentPins1, true, false, false, true);
  sevseg1.setBrightness(90);

  byte numDigits2 = 1;
  byte digitPins2[] = {11};  // Sama dengan display pertama
  sevseg2.begin(COMMON_ANODE, numDigits2, digitPins2, segmentPins1, true, false, false, true);
  sevseg2.setBrightness(90);
}

void loop() {
  // Traffic Light 1 Hijau, Traffic Light 2 Merah
  digitalWrite(green1, HIGH);
  digitalWrite(red2, HIGH);
  digitalWrite(red1, LOW);
  digitalWrite(yellow1, HIGH);
  digitalWrite(yellow2, HIGH);
  digitalWrite(green2, LOW);

  // Countdown dan mainkan melodi untuk traffic light 1
  if (playMelody1) {
    countdownWithMelody(sevseg1, melody1, sizeof(melody1) / sizeof(int));
  } else {
    countdownWithMelody(sevseg1, melody2, sizeof(melody2) / sizeof(int));
  }

  // Transisi ke kuning untuk Traffic Light 1
  digitalWrite(yellow1, LOW);
  digitalWrite(yellow2, LOW);
  delay(1000); // Delay untuk lampu kuning

  // Traffic Light 1 Merah, Traffic Light 2 Hijau
  digitalWrite(yellow1, HIGH);
  digitalWrite(yellow2, HIGH);
  digitalWrite(red1, HIGH);
  digitalWrite(green1, LOW);
  digitalWrite(green2, HIGH);
  digitalWrite(red2, LOW);

  // Countdown dan mainkan melodi untuk traffic light 2
  if (playMelody1) {
    countdownWithMelody(sevseg2, melody2, sizeof(melody2) / sizeof(int));
  } else {
    countdownWithMelody(sevseg2, melody1, sizeof(melody1) / sizeof(int));
  }

  // Transisi ke kuning untuk Traffic Light 2
  digitalWrite(yellow1, LOW);
  digitalWrite(yellow2, LOW);
  delay(1000); // Delay untuk lampu kuning

  // Traffic Light 2 Merah, Traffic Light 1 Hijau
  digitalWrite(yellow1, HIGH);
  digitalWrite(yellow2, HIGH);
  digitalWrite(red2, HIGH);
  digitalWrite(green1, HIGH);
  digitalWrite(red1, LOW);
  digitalWrite(green2, LOW);
  
  // Tukar melodi untuk siklus berikutnya
  playMelody1 = !playMelody1;

  // Ulangi siklus

}


void countdownWithMelody(SevSeg& sevseg, int melody[], int melodySize) {
  unsigned long countdownStart = millis();
  int countdownDuration = 9000;  // 9 detik
  int noteIndex = 0;
  int remainingTime = 9;

  while (millis() - countdownStart < countdownDuration) {
    unsigned long currentMillis = millis();
    int elapsed = (currentMillis - countdownStart) / 1000;
    int currentTime = 9 - elapsed;

    if (currentTime < remainingTime) {
      remainingTime = currentTime;
      sevseg.setNumber(remainingTime);
      sevseg.refreshDisplay();
    }

    if (noteIndex < melodySize) {
      int noteDuration = 1000 / noteDurations[noteIndex];
      tone(buzz, melody[noteIndex], noteDuration);
      delay(noteDuration);
      noTone(buzz);
      noteIndex++;
      int pauseBetweenNotes = noteDuration * 1.30;
      delay(pauseBetweenNotes);
    } else {
      noteIndex = 0; // Loop melodi jika belum selesai countdown
    }
  }

  noTone(buzz); // Pastikan buzzer dimatikan setelah selesai
}