#include <Wire.h>
#include <AD770X.h>
#include <LiquidCrystal_I2C.h>
#include <string.h>
AD770X ad7706(2.5);
double R1, R2;
LiquidCrystal_I2C lcd(0x27, 16, 2);
byte buzzerPin = 4;
#define DRDY 9         
#define RelayPin1 6    
#define RelayPin2 5    
#define EnableTop 2    
#define EnableBottom 3
byte LevelTop = 0x00;  
byte LevelBottom = 0x00;
double Temp_Top = 0;     
double Temp_Bottom = 0;  
int Temp1, Temp2, Temp11, Temp12;
byte EnableLed = 0xFF;   
char command_in[6];      
const byte DESIRED_WIDTH = 3;
void setup() {
  Serial.begin(9600);
  lcd.begin();
  lcd.clear();
  lcd.backlight();
  lcd.setCursor(1, 0);
  lcd.print("SOLDERSTATION");
  lcd.setCursor(5, 1);
  lcd.print("v2.1.2");
  tone(buzzerPin, 523);
  delay(200);
  tone(buzzerPin, 659);
  delay(200);
  tone(buzzerPin, 784);
  delay(200);
  noTone(buzzerPin);
  delay(3000);
  lcd.clear();
  pinMode(RelayPin1, OUTPUT);
  pinMode(RelayPin2, OUTPUT);
  pinMode(EnableTop, OUTPUT);
  pinMode(EnableBottom, OUTPUT);
  digitalWrite(EnableTop, LOW);
  digitalWrite(EnableBottom, LOW);
  analogWrite(RelayPin1, LevelTop);
  analogWrite(RelayPin2,LevelBottom);
  ad7706.reset();
  ad7706.init(AD770X::CHN_AIN1);
  ad7706.init(AD770X::CHN_AIN2);
}
String addZeroes(const String& src, byte pad)
{
  if (src.length() >= pad)
    return src;
  String result;
  byte add = pad - src.length();
  while (add--)
    result += "0";
  result += src;
  return result;
}

void loop() {
  while ( !digitalRead(DRDY) );
  R1 =ad7706.readADResult(AD770X::CHN_AIN1);
  double Temp_Bottom = (R1 - 1) /0.00392;
  R2 =ad7706.readADResult(AD770X::CHN_AIN2);
  double Temp_Top = (R2 - 1) / 0.00392 -10;
  int i = 0;
  int buffer[6];
  if (Serial.available()) {
    delay(100);
    while ( Serial.available() &&i < 7) {
      buffer[i++] = Serial.read();
    }
    buffer[i++] = '\0';
  }
  if (i > 0) {
    if (buffer[0] == 0xAA) {
      LevelTop = buffer[1];
      LevelBottom = buffer[2];
      EnableLed = buffer[3];
    }
    Temp1 = int(Temp_Top);
    Temp11 = 0;
    if (Temp1 > 255) {
      Temp11 = Temp1 - 255;
      Temp1 = 255;
    }
    Temp2 = int(Temp_Bottom);
    Temp12 = 0;
    if (Temp2 > 255) {
      Temp12 = Temp2 - 255;
      Temp2 = 255;
    }
    Serial.write(170);
    Serial.write(Temp1);
    Serial.write(Temp11);
    Serial.write(Temp2);
    Serial.write(Temp12);
    Serial.write(255);
  }
  if (LevelTop == 0x00) {
    digitalWrite(EnableTop, LOW);
  }
  if (LevelTop != 0x00) {
    digitalWrite(EnableTop, HIGH);
  }
  if (LevelBottom == 0x00) {
    digitalWrite(EnableBottom,LOW);
  }
  if (LevelBottom != 0x00) {
    digitalWrite(EnableBottom,HIGH);
  }
  analogWrite(RelayPin1, LevelTop);
  analogWrite(RelayPin2,LevelBottom);
  lcd.setCursor(1, 0);
  lcd.print("TH:");
  lcd.setCursor(1, 1);
  lcd.print("BH:");
  lcd.setCursor(8, 0);
  lcd.print("PWR:");
  lcd.setCursor(8, 1);
  lcd.print("PWR:");
  lcd.setCursor(4, 0);
 lcd.print(addZeroes(String(int(Temp_Top)), DESIRED_WIDTH));
  lcd.setCursor(4, 1);
 lcd.print(addZeroes(String(int(Temp_Bottom)), DESIRED_WIDTH));
  lcd.setCursor(12, 0);
 lcd.print(addZeroes(String(int(LevelTop)), DESIRED_WIDTH));
  lcd.setCursor(12, 1);
 lcd.print(addZeroes(String(int(LevelBottom)), DESIRED_WIDTH));
  if (EnableLed == 0x00) {
    lcd.noBacklight();
  }
  if (EnableLed == 0xFF) {
    lcd.backlight();
  }
}