#include <Arduino.h>
#include <CommandStation.h>
#include <ArduinoTimers.h>

#define DCC_IRQ_MICROSECONDS 29

#define NUM_LOCOS 50

////////////////////////////////////////////////////////////////
// Motor driver selection:
// Comment out all but the two lines that you want to use

// DCCMain* mainTrack = DCC::Create_WSM_SAMCommandStation_Main(NUM_LOCOS);
// DCCService* progTrack = DCC::Create_WSM_SAMCommandStation_Prog();

// DCCMain* mainTrack = DCC::Create_Arduino_L298Shield_Main(NUM_LOCOS);
// DCCService* progTrack = DCC::Create_Arduino_L298Shield_Prog();
  
DCCMain* mainTrack = DCCMain::Create_Pololu_MC33926Shield_Main(NUM_LOCOS);
DCCService* progTrack = DCCService::Create_Pololu_MC33926Shield_Prog();

////////////////////////////////////////////////////////////////

void waveform_IrqHandler() {
  mainTrack->interruptHandler();
  progTrack->interruptHandler();
}

#if defined(ARDUINO_ARCH_SAMD)
void SERCOM4_Handler()
{   
  mainTrack->railcom.getSerial()->IrqHandler();
}
#endif

void setup() {
  mainTrack->hdw.setup();
  mainTrack->railcom.setup();
  progTrack->hdw.setup();

  // TimerA is TCC0 on SAMD21, Timer1 on MEGA2560, and Timer1 on MEGA328
  // We will fire an interrupt every 29us to generate the signal on the track 
  TimerA.initialize();
  TimerA.setPeriod(DCC_IRQ_MICROSECONDS);
  TimerA.attachInterrupt(waveform_IrqHandler);
  TimerA.start();

#if defined (ARDUINO_ARCH_SAMD)
  CommManager::registerInterface(new USBInterface(SerialUSB));     // Register SerialUSB as an interface
  Wire.begin();       // Needed for EEPROM to work
#elif defined(ARDUINO_ARCH_AVR)
  CommManager::registerInterface(new SerialInterface(Serial));        // Register Serial (USB port on mega/uno) as an interface
#endif

  EEStore::init();

  DCCEXParser::init(mainTrack, progTrack);       // Set up the string parser to accept commands from the interfaces
  CommManager::showInitInfo();           
}

void loop() {
  CommManager::update();
  mainTrack->loop();
  progTrack->loop();
}