#include "IUnpackOptionManager.hxx"

//oaEvent
#include <ICOMETLog.hxx>

//C++/STL
#include <iostream>
#include <string>
#include <sstream>

using COMET::IUnpackOptionManager;

//----------------------------------------------------------------------
bool IUnpackOptionManager::SetOption(std::string option, std::string value)
{
  bool returnvalue = 0;
  if (option == "trig"){
    returnvalue = this->SetTriggerMaskFromOption(value);
    COMETLog("Trigger mask set from command line to 0x" << std::hex << fTriggerMask << std::dec);
  }
  return returnvalue;
}

//----------------------------------------------------------------------
void IUnpackOptionManager::Usage()
{
  std::cout << "Options provided by UnpackOptionManager:" << std::endl;
  std::cout << "    -O trig=<F>       Set trigger filter\n"
      << "\tif F is preceded by '0x' it is interpreted as a hexadecimal bit mask.\n"
      << "\tOtherwise we look for alphabetic identifiers\n"
      << "\t  'S': SPILL  triggers\n"
      << "\t  'C': COSMIC triggers\n"
      << "\t  'P': PEDESTAL  triggers\n"
      << "\t  'I': INJECTION (all other calibation) triggers\n"
      << "\teg \"trig=SC\" means process SPILL and CALIB triggers.\n"
      << "\tIf trig is not set all events are processed.\n"
      << "\tEvents with a trigger word of zero (e.g. ODBdump)\n" 
      << "\tare passed in all cases." << std::endl;
}

//----------------------------------------------------------------------
bool IUnpackOptionManager::SetTriggerMaskFromOption(std::string optval){
  UShort_t newMask(0);
  //Advanced option: set as a hexadecimal 
  if (optval.find("0x") == 0){

    std::istringstream ss(optval);
    ss >> std::hex >> newMask;
    Truncate(newMask);
    
    if (newMask != 0){ 
      fTriggerMask = newMask;
      return 1;
    }
    //The mask would not pass any triggers
    return 0;    
  } // if (started with 0x)

  //Basic options:  
  //a combination of SPILL(S), COSMIC(C), PEDESTAL(P) or INJECTION(I)
  //Look for these characters present the optval string
  if ( contains(optval,'S') || contains(optval,'s') ) {
    newMask |= ITriggerFilter::kSpill;
  }
  if ( contains(optval,'C') || contains(optval,'c') ) {
    newMask |= ITriggerFilter::kTFBcosmic | ITriggerFilter::kFGDcosmic;
  }
  if ( contains(optval,'P') || contains(optval,'p') ) {
    newMask |= ITriggerFilter::kPedestal;
  }
  if ( contains(optval,'I') || contains(optval,'i') ) {
    newMask |= (ITriggerFilter::kTPClaser | ITriggerFilter::kLED_P0D 
    | ITriggerFilter::kLED_Ecal | ITriggerFilter::kLED_FGD 
    | ITriggerFilter::kFEE);
  }
  if (newMask != 0){ 
    fTriggerMask = newMask;
    return 1;
  }

  
  //We haven't understood the option string
  return 0;
}

//----------------------------------------------------------------------
COMET::ITriggerFilter IUnpackOptionManager::GetTriggerFilter()
{
  return ITriggerFilter(fTriggerMask);
}