// Button handler
#include "RadioControl.h"
#include "ButtonState.h"

//#define BTN_DEBUG

//////////////////////////////////////////////////////////////////////
//                                                                  //
//              Button Control Variables                            //
//                                                                  //
//////////////////////////////////////////////////////////////////////
byte TuneButtonState = 0;
byte lastTuneButtonState = 0;

// Encoder button control
ButtonState EncButton(ButtonState::LONG_SUPPORT);

//  Sideband select button control
ButtonState SideBandButton(ButtonState::NORMAL);

//  Band Switch button control
ButtonState BandButton(ButtonState::NORMAL);

// VFO select button control
ButtonState VfoButton(ButtonState::NORMAL);


// PTT Control
byte PTTState = 0;
byte lastPTTState = 0;

// CW Control
#define CW_TX_OFF_TIMER    500 // milliseconds before releasing TX after cw tx
unsigned long cw_tx_time = 0;

// Helper function to get sideband from mode
byte GetSidebandFromMode(byte mode_val) {
  switch (mode_val) {
    case L_SSB:
#ifdef CW
    case L_CW:
#endif
      return LSB;
    case U_SSB:
#ifdef CW
    case U_CW:
#endif
      return USB;
    default:
      return USB; // Default fallback
  }
}

//
//********************* Set a new mode *************************************
// new mode select the mode and sets the proper sideband.
//
void SetMode(byte new_mode)
{
  // Save new mode
  mode = new_mode;
  
  // Get sideband directly from mode
  byte new_sideband = GetSidebandFromMode(new_mode);
  
  // Apply the sideband settings (calls SetSB internally)
  SetSB(new_sideband);
  
  // Save the mode for the current VFO
  switch (active_vfo) {
    case VFOA:
      vfoAmode = mode;
      break;
    case VFOB:
      vfoBmode = mode;
      break;
  }
  
  // Update the display to show the active mode
  displayMode(mode);
}

//*********************Check Band ************************************
void SwapBand() {
  
  uint32_t freq;
  byte band_mode;
  
  if (band == BAND20) {          // Switch to 40 Meter Band
    band = BAND40;
    freq = band40Freq;
    // Determine mode from stored sideband (for backward compatibility)
    band_mode = (band40Sideband == LSB) ? L_SSB : U_SSB;
  } else {                        // Switch to 20 Meter Band
    band = BAND20;
    freq = band20Freq;
    // Determine mode from stored sideband (for backward compatibility)
    band_mode = (band20Sideband == LSB) ? L_SSB : U_SSB;
  }

  // Set the active VFO to new frequency
  switch (active_vfo) {
    case VFOA:
      vfoAfreq = freq;
      // Keep the VFO's current mode if it matches the band's sideband
      if (GetSidebandFromMode(vfoAmode) == GetSidebandFromMode(band_mode)) {
        SetMode(vfoAmode);
      } else {
        SetMode(band_mode);
      }
      displayActVFO(vfoAfreq);
      break;
    case VFOB:
      vfoBfreq = freq;
      // Keep the VFO's current mode if it matches the band's sideband
      if (GetSidebandFromMode(vfoBmode) == GetSidebandFromMode(band_mode)) {
        SetMode(vfoBmode);
      } else {
        SetMode(band_mode);
      }
      displayActVFO(vfoBfreq);
      break;
  }

  displayMode(mode);
  startSettingsTimer();
}

void CheckBand() {
  if ( BandButton.CheckButton(digitalRead(BAND_BTN)) == ButtonState::PRESSED )
  {
#ifdef BTN_DEBUG
      ToggleLED();
      String msg = F("CheckBand");
      displayBanner(msg);
#endif      
     SwapBand();
  }
}

//*********************Change Modes *************************************

void ChangeMode() {

  switch (mode) {
#ifndef CW
    case L_SSB:
      SetMode(U_SSB);
      break;
    case U_SSB:
      SetMode(L_SSB);
      break;
#else
    case L_SSB:
      SetMode(U_SSB);
      break;
    case U_SSB:
      SetMode(L_CW);
      break;
    case L_CW:
      SetMode(U_CW);
      break;
    case U_CW:
      SetMode(L_SSB);
      break;
#endif
  }
}


//*********************Set a new sideband ************************************
void SetSB(byte sb) {

  sideband = sb;  // Keep this for hardware compatibility
  bfo = (sb == USB ? USB_BFO : LSB_BFO);

  //
  // Keep track of which SSB is currently selected for current band
  //
  if (band == BAND20) {
    band20Sideband = sideband;
  } else {
    band40Sideband = sideband;
  }

  //
  // Change the clock frequency to the new bfo
  //
  setBFO(bfo);

  //
  // Set the active VFO to adjusted frequency
  //
  switch (active_vfo) {
    case VFOA:
      setVFO(vfoAfreq);
      break;
    case VFOB:
      setVFO(vfoBfreq);
      break;
  }
//  displayDebug("save mode="+String(mode));
  startSettingsTimer();
}

void CheckMode() {
  if ( SideBandButton.CheckButton(digitalRead(SIDEBAND_BTN)) == ButtonState::PRESSED )
    ChangeMode();
}


//********************* Tune Button Handling ************************************
void DoTune() {
  //
  // Temp code - 30 second full duty tone for debugging circuits - cheap tone generator.
  //      Delay(12);
  //      tone(TONE_PIN, NOTE_B5);
  //      Delay(30000);
  //      noTone(TONE_PIN);

  displayTune(true);
  startTx(PTT_TUNE);
  
  for (int i = 0; i < 100; i++) {
    tone(TONE_PIN, NOTE_B5);
    Delay(50);
//    noTone(TONE_PIN); // Uncomment for pulsed output
    Delay(50);
  }
  
  noTone(TONE_PIN);
  stopTx();
  displayTune(false);
}


//*********************Check if Tune Button pressed ****************
void CheckTune() {
  TuneButtonState = digitalRead(TUNE_BTN); // creates a 10 second tuning pulse trani 50% duty cycle and makes TUNE appear on the screen
  if (TuneButtonState != lastTuneButtonState) {
    if (TuneButtonState == LOW) {
      DoTune();
    }
    lastTuneButtonState = TuneButtonState;
    Delay(50);
  }
}

//*********************VFO switch******* VFO A or B ****************
void SwapVFO() {
  
  // Save current VFO mode before switching
  switch (active_vfo) {
    case VFOA:
      vfoAmode = mode;
      break;
    case VFOB:
      vfoBmode = mode;
      break;
  }
  
  // Switch to the other VFO and restore its mode
  if (active_vfo == VFOA) {
    active_vfo = VFOB;
    SetMode(vfoBmode);  // This will set mode, sideband, BFO, etc.
  } else {
    active_vfo = VFOA;
    SetMode(vfoAmode);  // This will set mode, sideband, BFO, etc.
  }

#ifdef BTN_DEBUG
  ToggleLED();
  String msg = F("SwapVFO: active_vfo=");
  msg += active_vfo;
  msg += F(" mode=");
  msg += mode;
  displayBanner(msg);
#endif
 
  //
  // Swap Active/Alternate frequency displays
  // 
  switch (active_vfo) {
    case VFOA:
      displayActVFO(vfoAfreq);
      displayAltVFO(vfoBfreq);
      break;
    case VFOB:
      displayActVFO(vfoBfreq);
      displayAltVFO(vfoAfreq);
      break;
  }
  displayVFOAB(active_vfo);            // Change the A/B indicator
  
  startSettingsTimer();
}


//********************* Check if Swap VFO pressed ****************
void CheckVFO() {
  if ( VfoButton.CheckButton(digitalRead(VFO_BTN)) == ButtonState::PRESSED )
      SwapVFO();
}


// startSplit
// Turn on split mode and update the display
void startSplit() {
  split = true;
  displaySplit(split);
}

// Turnb off split mode and update the display
void stopSplit() {
  split = false;
  displaySplit(split);
}

// startTx
// if split mode is on swap Act and Alt VFO
// Put the rig in to TX by triggering the PTT pin
// Update the display to Tx
void startTx(byte PTT_source) {
  if (split) SwapVFO();
  digitalWrite(PTT,HIGH);
  displayTxRx(TX);
  TxRxState = TX;
  txSource = PTT_source;
}

// stopTx
// Return the rig to Rx by lowering the PTT pin
// If split mode is on swap Act and Alt VFO
// Update the display to Tx
void stopTx() {
  digitalWrite(PTT,LOW);
  if (split) SwapVFO();
  displayTxRx(RX);
  TxRxState = RX;
}

//********************* PTT ****************************

void CheckPTT(){ 

  if ((TxRxState==TX) && (txSource != PTT_MIC)) {
    return;
  }
  
  TxRxState = digitalRead(PTT_SENSE); 

  if(TxRxState != lastTxRxState){

    if(TxRxState == TX){
      startTx(PTT_MIC); 
    } else {
      if (txSource == PTT_MIC) {
        stopTx();
      }
    } 
    lastTxRxState = TxRxState;
    Delay(50);
  }
  
}

#ifdef CW
//********************* CW key down ****************************
void CheckCW() {
  if ( mode != L_CW && mode != U_CW )
    return;

  CwTxRxState = digitalRead(KEY_IN);

  // Keep the PA up between CW keydowns
  if ((TxRxState==TX) && (txSource == PTT_CW) && (CwTxRxState != TX)) {
      if ((millis() - CW_TX_OFF_TIMER) > cw_tx_time) {
//          displayDebug("");
          stopTx();
          digitalWrite(CW_OUT,LOW);
          setCW(false, CW_TONE);
      }
  }

  // Transition states between key up and key down.
  if(CwTxRxState != lastCwTxRxState){
      if (CwTxRxState == TX) {
//          displayDebug("CW TX");
          setCW(true, CW_TONE);
          startTx(PTT_CW);            // key the transmitter
          digitalWrite(CW_OUT,HIGH);  // power the audio amp, switch tune tone as input
          cw_tx_time = millis();      // reset the PA tx timer.
          tone(TONE_PIN, CW_TONE);
      } else {
          noTone(TONE_PIN);
      }
      lastCwTxRxState = CwTxRxState;
      Delay(50);
  }
}
#endif