///
/// Example program for Trigger Check.
///
/// Missing Spill Checking Program, Original is written by Kendoll.M
/// Beam Data Access was added by Yano, basing on prototype of Beam Accessing Library.

#include "IHandle.hxx"
#include <set>
#include <math.h>
#include <string>
#include <sstream>
#include "IMidasBank.hxx"
#include "IMidasFile.hxx"
#include "IMidasBankProxyRegistry.hxx"
#include "ICOMETRawEvent.hxx"

// COMET software includes
#include "IMidasBankProxy.hxx"
#include "ICOMETRawEvent.hxx"
#include "IRawDataHeader.hxx"

// oaRawEvent includes
//#include "IRunInfoBank.hxx"
#include "IMCMBank.hxx"
//#include "ITriggerBank.hxx"

#include <iostream>
#include <cstdlib>

/// ROOT stuff
#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"

#include "IRawBeamData.hxx"

Int_t Extract_RunNumber(char FileName[], UInt_t *RunNumber, UInt_t *SubRunNumber){
  char ProcessRN[200];
  if (strstr(FileName,".mid.gz")==NULL){
    std::cout << "error" <<std::endl;
    return 0;
  }

  strcpy(ProcessRN,strstr(FileName,".mid.gz")-17);
  //std::cout<<ProcessRN<<std::endl;
  *RunNumber = std::atoi(ProcessRN);

  strcpy(ProcessRN,strstr(ProcessRN,"_")+1);
  //std::cout<<ProcessRN<<std::endl;
  *SubRunNumber = std::atoi(ProcessRN);
  return 1;
};


int main(int argc, char** argv)
{
  UInt_t RunNumber;
  UInt_t SubRunNumber;
  //
  //yano --memo--
  // importing args. arg1 : DBdir(should include .db fies)
  // importing args. arg2 : Version(today, for p02 only.)
  // importing args. arg3 : Beam_summary(beam_summary data)
  //yano --memo--
  //   TString dbdir("./spilldb");
  //   TString version("p02");
  //   TString datadir("/t2k/data/exp/beam/exp/beam_summary");



  //std::cout << " Opening FileName"<< std::endl;
  char FileName[100];
  //	std::cout << argc << std::endl;
  if( argc < 2 ) {
    std::cerr << "Usage : "<< argv[0] << " <Midas File>" << std::endl;
    return(1);
  }
  else{
    strcpy(FileName,argv[1]);
  }	

  if (Extract_RunNumber(FileName,&RunNumber,&SubRunNumber)==0){
    std::cerr << "Something wrong with Filename. It should be .root" << std::endl;
    return 0;
  }
  //std::cout << RunNumber<<std::endl;
  //std::cout << SubRunNumber<<std::endl;

  std::ofstream fout("spilllist.tmp.dat",std::ios::out);


  /*
     if( argc > 1 ) {
     if( argc < 2 ) {
     std::cerr << "Usage : " << argv[0] << " or " << std::endl
     << "        " << argv[0] << " <Midas File>" << std::endl;
     return( 1 );
     }
     std::cout<<argv[1]<<std::endl;
     strcpy(FileName,argv[1]);
     }
     */

  //std::cout <<"Create BeamData"<< std::endl;

  //yano --memo--
  //Creating BeamData instance.
  //yano --memo--
  COMET::IRawBeamData BeamData;

  /// Check that there are correct number of arguments  
  /*
     if(argc != 2){
     std::cerr <<"usage: check-trigger.exe [filename]\n" << std::endl;
     exit(0);
     }
     */

  // Open midas file
  COMETLog("Dumping MIDAS file: " << FileName << "\n");
  COMET::IMidasFile mf;
  bool success = mf.Open(FileName);
  if (!success){
    printf("Cannot open input file \"%s\"\n",FileName);
    exit(0);
  }else
    printf("Opening input file \"%s\"\n",FileName);

  /// Get Run Number
  /*
     char aRunNumberC[128];
     std::string aRunNumberS = argv[1];
     int firstIndex = aRunNumberS.find_last_of("/")+1;
     int lastIndex = aRunNumberS.find(".");
     aRunNumberS = aRunNumberS.substr(firstIndex,lastIndex-firstIndex);
     strcpy(aRunNumberC,aRunNumberS.c_str());
     std::cout << aRunNumberC << std::endl;
     */

  /// ---------------------------------------------------------
  /// Main loop over events.
  bool firstEvent = true;
  UInt_t serNumber,failed;
  UInt_t nBeam=0,nCalib=0;
  UInt_t TrgId=0,nSpill;
  UInt_t missedspill=0;
  /*
     int TrgIdtpc=0;
     int TrgIdfgd=0;
     int nBeamfgd=0,nCalibfgd=0;
     int nBeamtpc=0,nCalibtpc=0;

     int TrgIdsmrd=0;
     int nBeamsmrd=0,nCalibsmrd=0;
     int TrgIdecal=0;
     int nBeamecal=0,nCalibecal=0;
     */
  UInt_t lastnspill=0;
  bool firstspill=true;
  bool Fst_bsd=true;
  Int_t Fst_no=0;
  Int_t Fst_sec=0;
  bool Lst_bsd=true;
  Int_t Lst_no=0;
  Int_t Lst_sec=0;

  Int_t TrgSec=0;
  Int_t HdrSec=0;
  Int_t missedbeamspill=0;
  Int_t spillnum;
  Int_t good_spill_flag=0;
  Int_t bad_spill_flag=0;
  Int_t good_gps_flag;
  Int_t run_type;
  Int_t count_normal_spill = 0;
  Double_t count_normal_pot = 0;
  Int_t count_good_spill = 0;
  Double_t count_good_pot = 0;
  Int_t count_total_spill = 0;
  Double_t count_total_pot = 0;
  Int_t trg_sec[3];


  std::cout<<"-------------------- Check Start ----------------------"<<std::endl;
  while ( COMET::ICOMETRawEvent* re = mf.ReadRawEvent() ) {

    /// Process bank.
    re->PromoteMidasBanks(false);
    //re->ls();
    serNumber = re->GetHeader().GetSeqNo();
    //RunNumber = re->GetHeader().GetRun();
    //SubRunNumber = re->GetContext().GetSubRun();


    if(firstEvent){
      serNumber = -9999; // Sequence number for first event is wrong.
      firstEvent = false;
      continue;
    }

    /* --- M.Kendoll's part--
    /// Trigger Bank
    COMET::IHandle<COMET::ITriggerBank> triggerBank;
    while ( triggerBank = re->GetMidasBank<COMET::ITriggerBank>("OTRI",triggerBank) ) {
    COMET::ITriggerBank& trigger = re->UseMidasBank<COMET::ITriggerBank>("OTRI");
    TrgId  = (trigger.GetTriggerWord()>>48) & 0xffff;
    nSpill = (trigger.GetTriggerWord()>>32) & 0xffff;

    if(TrgId==1){
    if(nSpill!=lastnspill+1&&!firstspill){
    // check for missing spills
    std::cout<<"Missed a spill!"<<std::endl;
    std::cout<<"trigger:"<<trigger.GetTriggerWord()<<std::endl;
    std::cout<<"TrgId: "<<TrgId<<" nSpill: "<<nSpill<<" last spill: "<<lastnspill<<std::endl;
    missedspill++;
    }
    firstspill=false;
    lastnspill = nSpill;
    }
    }      
    */

    /// MCM Bank
    COMET::IHandle<COMET::IMCMBank> mcmBank;
    while ( mcmBank = re->GetMidasBank<COMET::IMCMBank>("OMCM",mcmBank) ) {
      COMET::IMCMBank& mcm = re->UseMidasBank<COMET::IMCMBank>("OMCM");

      TrgId  = (mcm.GetTriggerWord()>>48) & 0xffff;
      nSpill = (mcm.GetTriggerWord()>>32) & 0xffff;
      TrgSec = (Int_t) mcmBank->GetUnixTimeTrig();
      HdrSec = (Int_t) re->GetHeader().GetTimeStamp();
      /*
         if(TrgId==1){
         std::cout << nSpill <<" "<< lastnspill << std::endl;
         }
         */
      if(firstspill){
        lastnspill=nSpill-1;
      }

      if(TrgId==1){
        if((nSpill==0&&lastnspill==65535)&&!firstspill){
          lastnspill=-1;
        }
        if((nSpill!=lastnspill+1&&!firstspill)){
          // check for missing spills
          int nMSpill=0;
          int missing_spill=0;
          while ((nSpill!=lastnspill+1+nMSpill)){
            missing_spill= lastnspill+nMSpill+1;
            if (missing_spill>65535)missing_spill=missing_spill-65536;
            std::cerr<<"Missed a spill!"<<std::endl;
            std::cerr<<"Current trigger:"<<TrgId << " Curret spill No.:"<< nSpill<<std::endl;
            std::cerr<<"Missed spill No.:"<<missing_spill<<std::endl;
            nMSpill++;
            missedspill++;
          }

        }

        //acquire spill information information
        //int ret = BeamData.GetSpill(nSpill, TrgSec, 3);
        int ret = BeamData.GetSpill(nSpill, TrgSec, 3);
        // For Run 31 - 34, time diff between trgsec - BeamData.GetTriggerTime <= 1. 3 is OK. 
        //if( ! ret ) ret = BeamData.GetSpill( nSpill, HdrSec, 5 );
        if(  ret == 1 ) {
          spillnum = (Int_t ) BeamData.GetSpillNumber();
          good_spill_flag= (Int_t )BeamData.GetGoodSpillFlag();
          //good_gps_flag= (Int_t )BeamData.good_gps_flag();
          //run_type = (Int_t ) beam.run_type();
          //checking health
          /*
             double hct[3][5];
             for(int i=0;i<3;i++){
             for(int j=0;j<5;j++){
             hct[i][j]=beam.data().hct[i][j];
             }
             }
             */
          double ct_np[5];
          for(int i=0;i<5;i++){
            ct_np[i]=BeamData.GetProtonsPerSpill(i);
          }
          for(int i=0;i<3;i++){
            trg_sec[i]=BeamData.GetTriggerTime(i);
          }

          if(Fst_bsd==true)
          {
            Fst_no= spillnum;
            Fst_sec= trg_sec[0];
            Fst_bsd=false;
          }

          Lst_no= spillnum;
          Lst_sec= trg_sec[0];

          count_total_spill ++;
          count_total_pot   += ct_np[4];

          /*
             bool bad_cond = 0;

             if( 598833  <= spillnum && spillnum <= 598837 && trg_sec[0] > 1264342865-60 && trg_sec[0] < 1264343146+60) bad_cond = 1 ;
             else if( 598937  <= spillnum && spillnum <= 598941 && trg_sec[0] > 1264344537-60 && trg_sec[0] < 1264344927+60) bad_cond = 1 ;
             else if( 1201242 <= spillnum && spillnum <= 1201270 && trg_sec[0] > 1267200809-60 && trg_sec[0] < 1267335297+60) bad_cond = 1 ;
             else if( 1202461 <= spillnum && spillnum <= 1202533 && trg_sec[0] > 1267200809-60 && trg_sec[0] < 1267335297+60) bad_cond = 1 ;
             else if( 1234283 <= spillnum && spillnum <= 1234304 && trg_sec[0] > 1267200809-60 && trg_sec[0] < 1267335297+60) bad_cond = 1 ;
             else if( 1235436 <= spillnum && spillnum <= 1235450 && trg_sec[0] > 1267200809-60 && trg_sec[0] < 1267335297+60) bad_cond = 1 ;

             bool normal_cond =
             ( (Int_t)run_type == 1  ) &&
             ( hct[0][0] >= 10 ) &&
             ( hct[1][0] >= 10 ) &&
             ( hct[2][0] >= 10 );
             bool beamtrg     =
             ( ct_np[4][0] > 1e11 );
             bool mucenter    =
             ( sqrt( mumon[2] * mumon[2] + mumon[4] * mumon[4] ) < 10 );

             bool muct        =
             ( 31.7 < ( mumon[0] / ct_np[4][0] * 1e9 ) ) &&
             ( ( mumon[0] / ct_np[4][0] * 1e9 ) < 35.1 );
             if(!bad_cond && normal_cond ){
             count_normal_spill ++;
             count_normal_pot   += ct_np[4][0];
             }
             if(!bad_cond && normal_cond && beamtrg && mucenter && muct && good_gps_flag){
             count_good_spill ++;
             count_good_pot   += ct_np[4][0];
             }
             */
          //std::cout << "find a Spill (No.:"<< nSpill << " TriggerTime:"<< TrgSec <<") in Beam Quick Summary" << std::endl;
          fout << RunNumber <<"	"<< SubRunNumber<<"	"<< serNumber<< "	" << nSpill <<"	" <<TrgSec<< "	"<< ct_np[4]<< "	"<<good_spill_flag<<std::endl;
          if (good_spill_flag == 0){
            bad_spill_flag = 1;
          }
        }
        else {
          //std::cout << "Cannot find a Spill (No.:"<< nSpill << " TriggerTime:"<< TrgSec <<") in Beam Quick Summary" << std::endl;
        }

        lastnspill = nSpill;
        firstspill=false;
      }
    }

    /// Only select beam trigger
    if(TrgId==1)nBeam++;
    if(TrgId==2)nCalib++;

    delete re;

    //if(serNumber>100)break;

  }
  //  manager->end();
  double fraction =(double)(missedspill)/(double)(nBeam+missedspill);
  std::cout<<"----------------------- Summary -----------------------"<<std::endl;
  //std::cout<<"From global MCM header"<<std::endl;
  std::cout<<"Beam Trigger: "<<nBeam<<std::endl;
  std::cout<<"Calibration Trigger: "<< nCalib << std::endl;
  std::cout <<"No. of Spills matching Beam Quick Summary: " << count_total_spill << std::endl;
  std::cout <<"POT of Spills matching Beam Quick Summary: " << count_total_pot << std::endl;
  std::cout <<"First Event Spillnum and Triggertime: " << Fst_no << " " << Fst_sec << std::endl;
  std::cout <<"Last Event Spillnum and Triggertime: " << Lst_no << " " << Lst_sec << std::endl;
  std::cout<<"Missed spills: "<< missedspill <<" fractionally "<<fraction<<std::endl;
  fout.close();

  if( bad_spill_flag == 1 ){
  std::ofstream fbad("spilllist.bad.dat",std::ios::out);
  fbad<<"This subrun contain bad spills."<<std::endl;
  fbad.close();
  }
  return 0;

}