#include <Wire.h>
#include <LiquidCrystal_I2C.h>
#include <Keypad.h>
#include <EEPROM.h>
#include "Adafruit_Thermal.h"
#include "SoftwareSerial.h"

#define TX_PIN 10 // Arduino transmit YELLOW WIRE labeled RX on the printer
#define RX_PIN 11 // Arduino receive GREEN WIRE labeled TX on the printer

SoftwareSerial mySerial(RX_PIN, TX_PIN);
Adafruit_Thermal printer(&mySerial);

LiquidCrystal_I2C lcd(0x27, 20, 4);

const int LDR_PIN[] = {A0, A1}; // Gunakan dua pin analog saja
const byte ROWS = 4;
const byte COLS = 4;
char keys[ROWS][COLS] = {
  {'1', '2', '3', 'A'},
  {'4', '5', '6', 'B'},
  {'7', '8', '9', 'C'},
  {'*', '0', '#', 'D'}
};
byte rowPins[ROWS] = {9, 8, 7, 6};
byte colPins[COLS] = {5, 4, 3, 2};
Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);

int calibrationValue = 500;
int currentMenu = 0;

void setup() {
  lcd.init();
  lcd.backlight();
  lcd.clear();
  lcd.setCursor(2, 0);
  lcd.print("HEADLIGHT TESTER");
  lcd.setCursor(3, 2);
  lcd.print("BLANGKONMUTER");
  delay(2000);
  Serial.begin(9600);
  mySerial.begin(9600);
  printer.begin();
  printer.justify('C');
  printer.setSize('S');
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("1. Measurement");
  lcd.setCursor(0, 1);
  lcd.print("2. Calibration");
  EEPROM.get(0, calibrationValue);
}

float measureLight(int pin) {
  int sensorValue = analogRead(pin);
  float voltage = (sensorValue / 1023.0) * 5.0;
  float resistance = (5.0 - voltage) / voltage;
  return calibrationValue / resistance * 100;
}

void displayLightValues() {
  float luxValues[2];
  for (int i = 0; i < 2; i++) {
    luxValues[i] = measureLight(LDR_PIN[i]);
    Serial.print("Sensor ");
    Serial.print(i + 1);
    Serial.print(": ");
    Serial.print(luxValues[i]);
    Serial.println(" Cd");
  }

  float avgLux = (luxValues[0] + luxValues[1]) / 2.0;
  String beamInfo;

  if (luxValues[0] > avgLux) {
    beamInfo = "High Beam";
  } else if (luxValues[1] > avgLux) {
    beamInfo = "Low Beam";
  } else {
    beamInfo = "Beam Info Not Available";
  }

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Avg Brightness:");
  lcd.setCursor(0, 1);
  lcd.print("Cd");
  lcd.setCursor(8, 1);
  lcd.print(avgLux, 0);
  lcd.setCursor(0, 2);
  lcd.print(beamInfo);
  delay(1000);
}

void printValues() {
  float luxValues[2];
  for (int i = 0; i < 2; i++) {
    luxValues[i] = measureLight(LDR_PIN[i]);
    Serial.print("Sensor ");
    Serial.print(i + 1);
    Serial.print(": ");
    Serial.print(luxValues[i]);
    Serial.println(" Cd");
  }

  float avgLux = (luxValues[0] + luxValues[1]) / 2.0;
  String beamInfo;

  if (luxValues[0] > avgLux) {
    beamInfo = "High Beam";
  } else if (luxValues[1] > avgLux) {
    beamInfo = "Low Beam";
  } else {
    beamInfo = "Beam Info Not Available";
  }

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("PRINTING.....");
  delay(2000);
  displayLightValues();
  printer.justify('C');
  printer.setSize('S');
  printer.doubleHeightOn();
  printer.print("BLANGKONMUTER");
  printer.doubleHeightOff();
  printer.println();
  printer.println("HEADLIGHT TESTER");
  printer.println();
  printer.underlineOn();
  printer.println("Measurement Data");
  printer.underlineOff();
  printer.println();
  printer.print("Avg Intensity: " + String(avgLux, 0));
  printer.println(" Cd");
  printer.println("Beam Info: " + beamInfo);
  printer.feed(3);
  printer.sleep();
}

void calibrateSensor() {
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Enter Calibration Lux:");
  char buffer[7];
  int bufferIndex = 0;
  while (1) {
    char key = keypad.getKey();
    if (key) {
      if (key == '#') {
        buffer[bufferIndex] = '\0';
        int luxValue = atoi(buffer);
        calibrationValue = luxValue * 100 / measureLight(LDR_PIN[0]);
        EEPROM.put(0, calibrationValue);
        lcd.clear();
        lcd.setCursor(0, 0);
        lcd.print("Calibration Value:");
        lcd.setCursor(0, 1);
        lcd.print(calibrationValue);
        delay(2000);
        return;
      } else if (key == '*') {
        bufferIndex = 0;
      } else {
        buffer[bufferIndex++] = key;
        lcd.setCursor(bufferIndex - 1, 1);
        lcd.print(key);
      }
    }
  }
}

void loop() {
  char key = keypad.getKey();
  if (key) {
    switch (key) {
      case '1':
        if (currentMenu == 0) {
          currentMenu = 1;
          displayLightValues();
        }
        break;
      case '2':
        if (currentMenu == 0) {
          currentMenu = 2;
          calibrateSensor();
        }
        break;
      case '0':
        if (currentMenu != 0) {
          currentMenu = 0;
          lcd.clear();
          lcd.setCursor(0, 0);
          lcd.print("1. Measurement");
          lcd.setCursor(0, 1);
          lcd.print("2. Calibration");
        }
        break;
      case '3':
        if (currentMenu == 1) {
          printValues();
        }
        break;
    }
  }
}