MadUserAnalysis.cxx

Go to the documentation of this file.

#ifndef maduseranalysis_cxx
#define maduseranalysis_cxx
#include <iostream>
#include "Mad/MadUserAnalysis.h"

using namespace std;

//********************************************************
MadUserAnalysis::MadUserAnalysis(TChain *chainSR,TChain *chainMC,
                                 TChain *chainTH,TChain *chainEM)
{

  if(!chainSR) {
    record = 0;
    strecord = 0;
    emrecord = 0;
    mcrecord = 0;
    threcord = 0;
    Clear();
    whichSource = -1;
    isMC = true;
    isTH = true;
    isEM = true;
    Nentries = -1;
    return;
  }
  
  InitChain(chainSR,chainMC,chainTH,chainEM);
  whichSource = 0;

  user_variable = 0;

}

//********************************************************
MadUserAnalysis::MadUserAnalysis(JobC *j,string path,int entries)
{
  record = 0;
  strecord = 0;
  emrecord = 0;
  mcrecord = 0;
  threcord = 0;
  Clear();
  isMC = true;
  isTH = true;
  isEM = true;
  Nentries = entries;
  whichSource = 1;
  jcPath = path;
  fJC = j;

  user_variable = 0;

}

//********************************************************
MadUserAnalysis::~MadUserAnalysis()
{

}

//*********************************************************
Bool_t MadUserAnalysis::PassTruthSignal(Int_t mcevent)
{
  if(!LoadTruth(mcevent)) return false;
  if(abs(ntpTruth->inu)==14&&ntpTruth->iaction==1) return true;
  return false;
}

//*********************************************************
Bool_t MadUserAnalysis::PassAnalysisCuts(Int_t event)
{
  if(!LoadEvent(event)) return false;
  if(PassBasicCuts(event)) {
    if(PID(event,0)>=1.) return true;
  }
  return false;
}

//******************************************
Bool_t MadUserAnalysis::PassBasicCuts(Int_t event)
{
  //define my own analysis cuts
  if(!LoadEvent(event)) return false;
  Int_t track = -1;
  if(!LoadLargestTrackFromEvent(event,track)) return false;
  if(RecoMuEnergy(track)<100.) return true;
  return false;
}

//******************************************
Float_t MadUserAnalysis::PID(Int_t event,Int_t method){
  if(!LoadEvent(event)) return 0;
  if(method==0) return 1.0;
  return 0;
}

//********************************************************
Float_t MadUserAnalysis::RecoAnalysisEnergy(Int_t event){

  if(!LoadEvent(event)) return 0;
  int largestTrack = -1;
  LoadLargestTrackFromEvent(event,largestTrack);
  float emu = RecoMuEnergy(0,largestTrack);
  float ehad = RecoShwEnergy(event);
  float ereco = emu + ehad;  
  return ereco;

}

//********************************************************
Float_t MadUserAnalysis::GetWeight(Int_t event)
{  
  if(event>=0) return 1.0;
  return 0;
}

//********************************************************
Bool_t MadUserAnalysis::AddUserBranches(TTree *tree)
{
  if(!tree) return false;
  tree->Branch("user_variable",&user_variable,"user_variable/D",32000);
  return true;
}

//********************************************************
void MadUserAnalysis::CallUserFunctions(Int_t event)
{
  user_variable = 0;

  if(!LoadEvent(event)) return;
  user_variable = 0.5;

}

//******************************************
void MadUserAnalysis::MakeMyFile(std::string tag){

  std::string savename = "QEHistos_" + tag + ".root";
  TFile save(savename.c_str(),"RECREATE"); 
  save.cd();
  
  //#declare lots of histos:
  TH1F *NEvent_NC = new TH1F("NEvent_NC",
                              "Number of Reco'd Events - NC",
                              20,0,20);
  NEvent_NC->SetXTitle("Number of Events");
  TH1F *NEvent_CC = new TH1F("NEvent_CC",
                              "Number of Reco'd Events - CC",
                              20,0,20);
  NEvent_CC->SetXTitle("Number of Events");

  for(int i=0;i<Nentries;i++){
    
    if(i%1000==0) std::cout << float(i)*100./float(Nentries) 
                            << "% done" << std::endl;

    if(!this->GetEntry(i)) continue;

    //get which detector this snarl is from
    //int theDetector = ntpHeader->GetVldContext().GetDetector()-1;

    for(int j=0;j<eventSummary->nevent;j++){
      
      if(!LoadEvent(j)) continue;  //does event exist

      //if no good truth candidate, we loose event
      int mcevent = -1;
      if(LoadTruthForRecoTH(j,mcevent)){ 
      
        if(PassTruthSignal()) {
          NEvent_CC->Fill(eventSummary->nevent);
        }
        else {
          NEvent_NC->Fill(eventSummary->nevent);
        }
      }
    }
  }
  //#close file  
  save.Write();
  save.Close();
}

#endif // #ifdef maduseranalysis_cxx

---