#include <iostream>
#include <string>

#include "TChain.h"
#include "TROOT.h"
#include "TObjectTable.h"

#include "WCSimRootEvent.hh"
#include "WCSimRootGeom.hh"
#include "WCSimPmtInfo.hh"
#include "WCSimLAPPDInfo.hh"
#include "WCSimEnumerations.hh"
#include "WCSimRootOptions.hh"

int main(int argc, char* argv[]){
  //std::string filepattern="/pnfs/annie/persistent/users/moflaher/wcsim/multipmt/tankonly/wcsim_25_04_19_ANNIEp2v6_nodigit_BNB_Water_10k_22-05-17/wcsim_0.*.root";
  std::string filepattern="./wcsim_0.*.root";
  if(argc>1){
    filepattern=argv[1];
  }
  
  std::cout<<"Reading file pattern "<<filepattern<<std::endl;
  TChain* fChain = new TChain("wcsimT");
  Int_t nbytes = fChain->Add(filepattern.c_str());
  std::cout<<"loaded "<<nbytes<<" files"<<std::endl;
  
  WCSimRootEvent* wcsimrootevent = nullptr;
  TBranch* b_wcsimrootevent = nullptr;
  WCSimRootEvent* wcsimrootevent_mrd = nullptr;
  TBranch* b_wcsimrootevent_mrd = nullptr;
  WCSimRootEvent* wcsimrootevent_facc = nullptr;
  TBranch* b_wcsimrootevent_facc = nullptr;
  
  WCSimRootTrigger* tr=nullptr;
  
  int fCurrent=0;
  Long64_t maxentries=2500;
  Long64_t last_entry_of_current_tree=-1;
  int branchok=0;
  
  std::cout<<"starting entry loop"<<std::endl;
  for(int eventi=0; eventi<maxentries; eventi++){
    if(eventi > last_entry_of_current_tree){
      
      if(eventi>0){
        std::cout<<"Abdout to retrieve the first entry of a new file: cleanup"<<std::endl;
        if(wcsimrootevent) delete wcsimrootevent;
        if(wcsimrootevent_mrd) delete wcsimrootevent_mrd;
        if(wcsimrootevent_facc) delete wcsimrootevent_facc;
        wcsimrootevent = nullptr;
        wcsimrootevent_mrd = nullptr;
        wcsimrootevent_facc = nullptr;
        
        std::cout<<"After cleanup, object table is: "<<std::endl;
        gObjectTable->Print();
      }
      
      Long64_t localentry = fChain->LoadTree(eventi);
      if(localentry<0){
        std::cout<<"end of TChain, breaking"<<std::endl;
        break;
      }
      fCurrent=fChain->GetTreeNumber();
      std::cout<<"Loaded file "<<fCurrent<<std::endl;
      if(fCurrent>1){ std::cout<<"ending loop"<<std::endl; break; }
      
      branchok = fChain->GetTree()->SetBranchAddress("wcsimrootevent",&wcsimrootevent, &b_wcsimrootevent);
      if(branchok<0) std::cerr<<"Failed to set branch address for wcsimrootevent"<<std::endl;
      branchok = fChain->GetTree()->SetBranchAddress("wcsimrootevent_mrd",&wcsimrootevent_mrd, &b_wcsimrootevent_mrd);
      if(branchok<0) std::cerr<<"Failed to set branch address for wcsimrootevent_mrd"<<std::endl;
      branchok = fChain->GetTree()->SetBranchAddress("wcsimrootevent_facc",&wcsimrootevent_facc, &b_wcsimrootevent_facc);
      if(branchok<0) std::cerr<<"Failed to set branch address for wcsimrootevent_facc"<<std::endl;
      
      b_wcsimrootevent->SetAutoDelete(true);
      b_wcsimrootevent_mrd->SetAutoDelete(true);
      b_wcsimrootevent_facc->SetAutoDelete(true);
      
      std::cout<<"getting last entry of this tree: ";
      last_entry_of_current_tree = eventi+fChain->GetTree()->GetEntriesFast() - 1;
      std::cout<<last_entry_of_current_tree<<std::endl;
      
    }
    nbytes = fChain->GetEntry(eventi);
    if(nbytes<0){ std::cout<<"no bytes, ending loop"<<std::endl; break; }
    if(wcsimrootevent==nullptr){ std::cout<<"null wcsimrootevent!"<<std::endl; break; }
    tr=wcsimrootevent->GetTrigger(0);
    if(tr==nullptr){ std::cout<<"null trigger 0!"<<std::endl; continue; }
    Int_t numphots = tr->GetNcherenkovhittimes();
    if((eventi%100)==0) std::cout<<"check: event "<<eventi<<" has "<<numphots<<" photons"<<std::endl;
  }
  std::cout<<"Ended event loop, object table is: "<<std::endl;
  gObjectTable->Print();
  
  std::cout<<"deleting chain"<<std::endl;
  delete fChain;
  std::cout<<"final object dump is "<<std::endl;
  gObjectTable->Print();
  
}