JPlotEvtWeightFactorTriggerEfficiencyRatio.cc

Go to the documentation of this file.

#include <iostream>
#include <string>
 
#include "TH1.h"
#include "TFile.h"
#include "TGraph.h"
#include "TGraph2D.h"
#include "TMultiGraph.h"
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "Jeep/JProperties.hh"
 
#include "JAAnet/JEvtWeightFactorTriggerEfficiencyRatio.hh"
 
#include "JSirene/JVisibleEnergyToolkit.hh"
 
 
/**
 * \file
 * Example program for plotting trigger efficiency ratios.
 *
 * \author bjjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  typedef double (JEvtWeightFactorTriggerEfficiencyRatio::*pFunction1D)(const double) const;
  typedef double (JEvtWeightFactorTriggerEfficiencyRatio::*pFunction2D)(const double,
                                                           const double) const;
  
  
  string                    outputFile;
  
  JEvtWeightFactorTriggerEfficiencyRatio factor;
  
  int                       debug;
  
  try {
        
    JParser<> zap;
    
    zap['o'] = make_field(outputFile);
    zap['@'] = make_field(factor)          = JPARSER::initialised();
    zap['d'] = make_field(debug)           = 1;
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
 
 
  TFile out(outputFile.c_str(), "recreate");  
 
  const JProperties properties = factor.getProperties();
 
  const bool logE   = properties.getValue<bool>("logE");
  const int  option = properties.getValue<int> ("option");
  
  const TH1& hTE  = factor.getHistogram();
 
  const int Nbins = hTE.GetNbinsX();
 
  if (option != (int) JEvtWeightFactorTriggerEfficiencyRatio::EVIS_LEADING_LEPTON_CONTR &&
      option != (int) JEvtWeightFactorTriggerEfficiencyRatio::EVIS_HADRONIC_CONTR) {
 
    pFunction1D mfp;
    
    switch (option) { 
    case (int) JEvtWeightFactorTriggerEfficiencyRatio::ENERGY_INITIAL_STATE :
      mfp = &JEvtWeightFactorTriggerEfficiencyRatio::getTriggerEfficiencyRatio1;
      break;
    case (int) JEvtWeightFactorTriggerEfficiencyRatio::ENERGY_NEUTRINO:
      mfp = &JEvtWeightFactorTriggerEfficiencyRatio::getTriggerEfficiencyRatio2;
      break;
    case (int) JEvtWeightFactorTriggerEfficiencyRatio::EVIS:
      mfp = &JEvtWeightFactorTriggerEfficiencyRatio::getTriggerEfficiencyRatio3;
      break;
    default:
      FATAL("Invalid 1D option " << option);
    }
 
    TGraph g(Nbins);
    g.SetName(hTE.GetName());
  
    for (Int_t i = 0; i < Nbins; ++i) {
      
      const double x = hTE.GetBinCenter(i+1);
      const double y = (factor.*mfp)(logE ? pow(10.0, x) : x);
 
      if (isfinite(y)) {
        g.SetPoint(i, x, y);
      } else {
        WARNING("Trigger efficiency ratio not finite for x = " << SCIENTIFIC(10,3) << x << endl);
      }
    }
 
    g.Write();
    
  } else {
    
    pFunction2D mfp;
 
    switch (option) { 
    case (int) JEvtWeightFactorTriggerEfficiencyRatio::EVIS_LEADING_LEPTON_CONTR:
      mfp = &JEvtWeightFactorTriggerEfficiencyRatio::getTriggerEfficiencyRatio4;
      break;
    case (int) JEvtWeightFactorTriggerEfficiencyRatio::EVIS_HADRONIC_CONTR:
      mfp = &JEvtWeightFactorTriggerEfficiencyRatio::getTriggerEfficiencyRatio5;
      break;
    default:
      FATAL("Invalid 2D option " << option);
    }
 
    const Int_t N = Nbins * Nbins * (Nbins + 1) / 2;
 
    TGraph2D g(N);
    g.SetName(hTE.GetName());
 
    Int_t k = 0;
    
    for (Int_t i = 0; i < Nbins; ++i) {
 
      const double x = hTE.GetBinCenter(i+1);
      
      for (Int_t j = 0; j < i; ++j, ++k) {
        
        const double y = hTE.GetBinCenter(j+1);
        const double z = (logE ?
                          (factor.*mfp)(pow(10.0, x), pow(10.0, y)) :
                          (factor.*mfp)(x, y));
        
        g.SetPoint(k, x, y, z);
      }
    }
 
    g.Write();
  }
  
  out.Write();
  out.Close();
  
  return 0;
}