/*
 * 60W PD Powered Automatic Reflow Hoteplate
 * Author: tobychui
 * 
 * Notes: 
 * Temperature sensing is done using 1k 1% resistor and a 10k Thermistor
 * You might need to change the code if you are using other values
 * 
 * Touch Button Mappings
 * Start Btn: P1.4
 * Stop Btn: P1.7
 */
#include <TouchKey.h>

#define TEMP_PIN 11 //P1.1
#define HEATER_PIN 34 //P3.4
#define LED_PREHEAT 15 //P1.5
#define LED_REFLOW 16 //P1.6

//Timing settings. Each cycle is approximately 100ms
#define HOLDSTART_TIME 100 //START Button hold down time in cycles
#define SOAKING_TIME 6000 //SOAKING temperature upkeep time in cycles

//Temperature Settings
#define COOLDOWN_TEMP_ADC 255 //Room temperature
//Preheat temp & power
#define PREHEAT_TEMP_ADC 155 //160 degree C
#define PREHEAT_PWR_PWM 130 //50% power output
//Reflow temp & power
#define REFLOW_TEMP_ADC 123 //220 degree C
#define REFLOW_PWR_PWM 255 //100% power output
//Overheat cutoff temp 
#define CUTOFF_TEMP_ADC 100 //240 degree C

//Func def
bool updateHeaterPowerState();
void updateKeyStates();
int readTemp();


//Runtimes
bool reflowing = false; //Reflow has started
bool reflowComplete = false; //Indicate if the stopping is done because of reflow complete
/*
  Reflow Stages define what is the next state the hotplate should reach
  0: Standby
  1: Preheating
  2: Preheat Holding / Soaking
  3: Reflowing 
  4: Cooling down 
*/
int reflowStages = 0; 
int startCountdown = HOLDSTART_TIME; //Hold for HOLDSTART_TIME to start to prevent accidental touches
int cycleCounter = 0; //Record cycle time up to 10 seconds
int soakingCountdown = SOAKING_TIME;

//Blink state controls
bool blinkState = false; //If blinking this will be switching between true and false
bool blinkYellow = true; //Slow blink yellow if enabled (Standby). Do not enable both yellow and red at the same time
bool blinkRed = false; //Slow blink red if enable (End of Reflow). Do not enable both yellow and red at the same time
bool fastBlinkState = false; 
bool fastblinkRed = false;


//Real-time temperature control targets
int targetTempADC = 255; //Target ADC reading to reach, 255 = room temperature
int targetPwrPWM = 0; //Target PWM cycle for heater, 255 = full power, 0 = off
int offset = 1; //Allow +-offset for the ADC reading before reheating / cooling

void setup() {
  //Enable the touch buttons
  //Enable all 6 channels:TIN2(P1.4),TIN5(P1.7)
  TouchKey_begin((1 << 2) | (1 << 5));

  //Set LED pins to output
  pinMode(LED_PREHEAT, OUTPUT);
  pinMode(LED_REFLOW, OUTPUT);
  digitalWrite(LED_PREHEAT, LOW);
  digitalWrite(LED_REFLOW, LOW);

  //Set temp sense pin to input
  pinMode(TEMP_PIN, INPUT);

  //Set mosfet control pins to output and disable it
  pinMode(HEATER_PIN, OUTPUT);
  digitalWrite(HEATER_PIN, LOW);
}


void loop() {
  //Check key status
  updateKeyStates();

  //Main reflowing logics
  //Add delay to the end of each conditions so each loop cycle is approximate 100ms
  if (reflowing){
    bool tempReached = updateHeaterPowerState();
    if (tempReached){
      //This stage temperature reached. Move to next stage
      if (reflowStages == 1){
        //Preheat stage completed. Enter soaking stage
        USBSerial_println("Preheat temperature reached. Soaking started");
        reflowStages = 2; //Set reflow stage to soaking
        soakingCountdown = SOAKING_TIME;
        fastblinkRed = true;
      }else if (reflowStages == 2){
        //Wait for the soaking to complete and enter reflow stage
        soakingCountdown--;
        if (soakingCountdown <= 0){
          //Soaking completed. Enter reflow stage
          USBSerial_println("Soaking time ended. Reflow started");

          //Set Reflow LED to high
          fastblinkRed = false;
          digitalWrite(LED_REFLOW, HIGH);

          //Set to reflow temperature
          targetTempADC = REFLOW_TEMP_ADC; //Set the target temp to reflow
          targetPwrPWM = REFLOW_PWR_PWM; //Set power rating to reflow
          
          //Update the reflow stage to 3
          reflowStages = 3;
        }
      }else if (reflowStages == 3){
        //Reflowing target temperature reached. Start cooldown and shut down heater
        reflowStages = 4;
        blinkYellow = false;
        blinkRed = true;
        targetTempADC = COOLDOWN_TEMP_ADC;
        targetPwrPWM = 0;

        //Reflow ended. Wait until stop being press to exit this state
      }
    }


  delay(1);
  }else{
    //Standby mode
    digitalWrite(HEATER_PIN, LOW);
    blinkYellow = true;
    delay(100);
  }

  //FastBlink handler, execute every 500ms
  if (cycleCounter%5 == 0){
    if (fastblinkRed){
      digitalWrite(LED_REFLOW, HIGH);
    }else{
      digitalWrite(LED_REFLOW, LOW);
    }
    fastBlinkState = !fastBlinkState;
  }

  //Blink handler, execute every 2 seconds
  if (cycleCounter%20 == 0){
    if (blinkYellow){
      if (blinkState){
        digitalWrite(LED_PREHEAT, HIGH);
      }else{
        digitalWrite(LED_PREHEAT, LOW);
      }
    }else if (blinkRed){
      if (blinkState){
        digitalWrite(LED_REFLOW, HIGH);
      }else{
        digitalWrite(LED_REFLOW, LOW);
      }
    }else{
      digitalWrite(LED_REFLOW, LOW);
      digitalWrite(LED_PREHEAT, LOW);
    }
    blinkState = !blinkState;
  }

  //Cycle printout
  //USBSerial_println("cycle");
  cycleCounter++;
  if (cycleCounter > 600){
    cycleCounter = 0;
  }
}