Interface class for quarkonia efficiencies (Zaida)

[u/mrichter/AliRoot.git] / PWG3 / AliQuarkoniaEfficiency.h

#ifndef ALIQUARKONIAEFFICENCY_H
#define ALIQUARKONIAEFFICENCY_H

//===================================================================
//  Class AliQuarkoniaEfficiency                               
//
//  This class will provide the quarkonia reconstruction efficiency 
//  in ALICE without acceptance consideration
//  for different resonances :
//    kJpsi
//    kPsiP
//    kUpsilon
//    kUpsilonP
//    kUpsilonPP
//  for some vector mesons :
//    kPhi
//    kOmega
//  different decay channels:
//    kDimuon
//    kDielectron
//  different trigger configurations:
//    kSinglePlusLpt, kSinglePlusHpt, kSinglePlusApt,
//    kSingleMinusLpt, kSingleMinusHpt, kSingleMinusApt,
//    kSingleUndefLpt, kSingleUndefHpt, kSingleUndefApt,
//    kPairUnlikeLpt, kPairUnlikeHpt, kPairUnlikeApt,
//    kPairLikeLpt, kPairLikeHpt, kPairLikeApt
//  different parameterizations:
//    kFlat
//    kCDFscaled
//    kCDFscaledPP
//
//
//  Reconstruction efficiency has been evaluated by means of a flat
//  y and pt distribution of quarkonia in -4 < y < -2.5, 
//  0 < pt < 20 GeV/c. Weights have been used to evaluate the
//  reconstruction efficiency in different parameterizations.
//                                                              
//  Subatech 2006
//===================================================================

#include "TNamed.h"

class TH2F;
class TString;

class AliQuarkoniaEfficiency : public TNamed{

 public:

  enum quarkonia{kJpsi, kPsiP, kUpsilon, kUpsilonP, kUpsilonPP, kOmega, kPhi};
  enum decay{kDimuon, kDielectron};
  enum parameterization{kFlat, kCDFscaled, kCDFscaledPP};
  enum trigger{kSinglePlusLpt, kSinglePlusHpt, kSinglePlusApt,
	       kSingleMinusLpt, kSingleMinusHpt, kSingleMinusApt,
	       kSingleUndefLpt, kSingleUndefHpt, kSingleUndefApt,
	       kPairUnlikeLpt, kPairUnlikeHpt, kPairUnlikeApt,
	       kPairLikeLpt, kPairLikeHpt, kPairLikeApt };


  AliQuarkoniaEfficiency(Int_t quarkoniaResonance=kJpsi, Int_t decayChannel=kDimuon, Int_t simParameterization=kCDFscaledPP);
  virtual ~AliQuarkoniaEfficiency();    
  void   Init(); 

  TH2F*  GetEfficiencyHisto() const;
  void   GetEfficiency(Float_t rap, Float_t pT, Double_t & eff, Double_t & error); 

  inline void  SetEfficiencyFileName(char * efficiencyFileName) { fEfficiencyFileName = efficiencyFileName; }
  inline void  SetQuarkoniaResonance(Int_t quarkoniaResonance = kJpsi) { fQuarkoniaResonance= quarkoniaResonance;}
  inline void  SetDecayChannel(Int_t decayChannel = kDimuon) { fDecayChannel = decayChannel;}
  inline void  SetSimulatedParameterization(Int_t simParameterization = kCDFscaledPP) { 
    fParameterization = simParameterization;
  }
  inline void  SetTrigger(bool trigger = kFALSE, Int_t triggerType = kPairUnlikeApt){ 
    fTrigger = trigger; fTriggerType = triggerType;
  }
  
 private:

  TString        fEfficiencyFileName;      // Name of the efficiency root file
  Int_t          fQuarkoniaResonance;      // Resonance Acceptance
  Int_t          fDecayChannel;            // Studied decay channel
  Int_t          fParameterization;        // Quarkonia simulated parameterization  
  Int_t          fTriggerType;             // Trigger type to be considered  
  Bool_t         fTrigger;                 // Boolean to decide if consider or not trigger
  TH2F *         fEfficiency;              // Efficiency histogram

  ClassDef(AliQuarkoniaEfficiency,1)
    };

#endif