/Users/jmonk/Physics/ForIA/src/AnalysisTools/TrackDicer.cxx

Go to the documentation of this file.

#include "ForIA/AnalysisTools/TrackDicer.hh"
#include "ForIA/Units.hh"
#include "ForIA/IDataLoader.hh"

#include "TH2D.h"

namespace ForIA{
  
  TrackDicer::TrackDicer(): m_doInitialise(true){
    
  }
  
  void TrackDicer::initialise(Efficiency effType){
    
    if(!m_doInitialise) return;
    
    DataLoaderPtr dl = IDataLoader::create();
    
    dl->open("Root/MinBias/Part7TeV_v20.root");
    
    string plotName = "EfficiencyChargedPrimaryStable";
    
    m_efficiencyPlot = dl->retrieve<TH2D>(plotName);
    
    if(m_efficiencyPlot == 0) throw std::runtime_error("TrackDicer::initialise(): Could not get plot " + plotName);
    
    m_efficiencyType = effType;
    
    string systPlotName = "Eff_Syst_Unc";
    
    if(effType != NOMINAL){
      m_efficiencySystematic = dl->retrieve<TH2D>(systPlotName);
    
      if(m_efficiencySystematic == 0) throw std::runtime_error
        ("TrackDicer::initialise(): Could not get plot " + systPlotName);
    }
    // see http://www.gnu.org/software/gsl/manual/html_node/Random-number-generator-algorithms.html
    m_randomGenerator = gsl_rng_alloc (gsl_rng_taus2);
    
    gsl_rng_set(m_randomGenerator, 14021980);
    
    m_doInitialise = false;
    
    return;
  }

  double TrackDicer::trackEfficiency(IFourMomentumConstPtr track)const{
    
    if(m_doInitialise) throw std::runtime_error
      ("TrackDicer::trackEfficiency: Has not been initialised");
    
    int xbin = m_efficiencyPlot->GetXaxis()->FindBin(track->PT()/GeV);
    int ybin = m_efficiencyPlot->GetYaxis()->FindBin(track->eta());
    
//    std::cout<<track<<", xbin: "<<xbin<<", ybin: "<<ybin<<std::endl;
    
    int bin = m_efficiencyPlot->GetBin(xbin, ybin);
    
    if(m_efficiencyType == NOMINAL){
      return m_efficiencyPlot->GetBinContent(bin);
    }
    
    // for some reason the bin indexing on the systematics plots 
    // is different to that on the nominal values plot
    
    int systXBin = m_efficiencySystematic->GetXaxis()->FindBin(track->PT() / GeV);
    int systYBin = m_efficiencySystematic->GetYaxis()->FindBin(track->eta() / GeV);
    
    int systBin = m_efficiencySystematic->GetBin(systXBin, systYBin);
    
    double syst = m_efficiencySystematic->GetBinContent(systBin);
    
    if(m_efficiencyType == SYST_UP){
      return m_efficiencyPlot->GetBinContent(bin) + syst;
    }else if(m_efficiencyType == SYST_DOWN){
      return m_efficiencyPlot->GetBinContent(bin) - syst;
    }
    
    return 0.;
  }

}