PWG4/PartCorrBase/AliNeutralMesonSelection.h

   1 #ifndef ALINEUTRALMESONSELECTION_H
   2 #define ALINEUTRALMESONSELECTION_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice     */
   5 /* $Id: AliNeutralMesonSelection.h 27413 2008-07-18 13:28:12Z gconesab $ */
   6
   7 //_________________________________________________________________________
   8 // Class that contains methods to select candidate pairs to neutral meson
   9 // 2 main selections, invariant mass around pi0 (also any other mass),
  10 // apperture angle to distinguish from combinatorial.
  11 // There is a 3rd cut based on the gamma correlation on phi or pt.
  12 //-- Author: Gustavo Conesa (INFN-LNF)
  13
  14 // --- ROOT system ---
  15 #include<TObject.h>
  16 #include<TArrayD.h>
  17
  18 class TLorentzVector ;
  19 class TList ;
  20 class TH2F ;
  21
  22 //--- ANALYSIS system ---
  23
  24 class AliNeutralMesonSelection : public TObject {
  25
  26  public:
  27
  28   AliNeutralMesonSelection() ; // default ctor
  29   AliNeutralMesonSelection(const AliNeutralMesonSelection & g) ; // cpy ctor
  30   AliNeutralMesonSelection & operator = (const AliNeutralMesonSelection & g) ;//cpy assignment
  31   virtual ~AliNeutralMesonSelection() ; //virtual dtor
  32
  33   TList * GetCreateOutputObjects();
  34
  35   Double_t GetAngleMaxParam(Int_t i) const {return fAngleMaxParam.At(i) ; }
  36   void SetAngleMaxParam(Int_t i, Double_t par){fAngleMaxParam.AddAt(par,i) ; }
  37
  38   Double_t GetInvMassMaxCut() const {return fInvMassMaxCut ; }
  39   Double_t GetInvMassMinCut() const {return fInvMassMinCut ; }
  40   void SetInvMassCutRange(Double_t invmassmin, Double_t invmassmax)
  41   {fInvMassMaxCut =invmassmax;  fInvMassMinCut =invmassmin;}
  42
  43   Double_t GetMass() const {return fM ; }
  44   void SetMass(Double_t m) { fM =m ; }
  45
  46   Bool_t AreNeutralMesonSelectionHistosKept() const { return fKeepNeutralMesonHistos ; }
  47   void KeepNeutralMesonSelectionHistos(Bool_t keep) { fKeepNeutralMesonHistos = keep ; }
  48
  49   void InitParameters();
  50   Bool_t IsAngleInWindow(const Float_t angle, const Float_t e) const ;
  51   void Print(const Option_t * opt) const;
  52
  53   Bool_t  SelectPair(TLorentzVector particlei,  TLorentzVector particlej)  ;
  54
  55   //Histogrammes setters and getters
  56
  57   virtual void SetHistoERangeAndNBins(Float_t min, Float_t max, Int_t n) {
  58     fHistoNEBins = n ;
  59     fHistoEMax = max ;
  60     fHistoEMin = min ;
  61   }
  62
  63   Int_t   GetHistoNEBins() const { return fHistoNEBins ; }
  64   Float_t GetHistoEMin()   const { return fHistoEMin ; }
  65   Float_t GetHistoEMax()   const { return fHistoEMax ; }
  66
  67   virtual void SetHistoPtRangeAndNBins(Float_t min, Float_t max, Int_t n) {
  68     fHistoNPtBins = n ;
  69     fHistoPtMax = max ;
  70     fHistoPtMin = min ;
  71   }
  72
  73   Int_t   GetHistoNPtBins() const { return fHistoNPtBins ; }
  74   Float_t GetHistoPtMin()   const { return fHistoPtMin ; }
  75   Float_t GetHistoPtMax()   const { return fHistoPtMax ; }
  76
  77   virtual void SetHistoAngleRangeAndNBins(Float_t min, Float_t max, Int_t n) {
  78     fHistoNAngleBins = n ;
  79     fHistoAngleMax = max ;
  80     fHistoAngleMin = min ;
  81   }
  82
  83   Int_t   GetHistoNAngleBins() const { return fHistoNAngleBins ; }
  84   Float_t GetHistoAngleMin()   const { return fHistoAngleMin ; }
  85   Float_t GetHistoAngleMax()   const { return fHistoAngleMax ; }
  86
  87   virtual void SetHistoIMRangeAndNBins(Float_t min, Float_t max, Int_t n) {
  88     fHistoNIMBins = n ;
  89     fHistoIMMax = max ;
  90     fHistoIMMin = min ;
  91   }
  92
  93   Int_t   GetHistoNIMBins() const { return fHistoNIMBins ; }
  94   Float_t GetHistoIMMin()   const { return fHistoIMMin ; }
  95   Float_t GetHistoIMMax()   const { return fHistoIMMax ; }
  96
  97
  98  private:
  99   Double_t fM ; //mass of the neutral meson
 100   Double_t   fInvMassMaxCut ;  // Invariant Mass cut maximum
 101   Double_t   fInvMassMinCut ;  // Invariant Masscut minimun
 102   TArrayD    fAngleMaxParam ; //Max opening angle selection parameters
 103
 104   Bool_t  fKeepNeutralMesonHistos ; // Keep neutral meson selection histograms
 105
 106   //Histograms
 107   TH2F * fhAnglePairNoCut  ;  //Aperture angle of decay photons, no cuts
 108   TH2F * fhAnglePairOpeningAngleCut   ;  //Aperture angle of decay photons, cut on opening angle
 109   TH2F * fhAnglePairAllCut   ;  //Aperture angle of decay photons, all cuts
 110   TH2F * fhInvMassPairNoCut    ;  //Invariant mass of decay photons, no cuts
 111   TH2F * fhInvMassPairOpeningAngleCut  ;  //Invariant mass of decay photons, cut on opening angle
 112   TH2F * fhInvMassPairAllCut   ;  //Invariant mass of decay photons, all cuts
 113
 114   //Histograms binning and range
 115   Int_t   fHistoNEBins ;  //Number of bins in pi0 E axis
 116   Float_t fHistoEMax ;    //Maximum value of pi0 E histogram range
 117   Float_t fHistoEMin ;    //Minimum value of pi0 E histogram range
 118   Int_t   fHistoNPtBins ;  //Number of bins in Pt trigger axis
 119   Float_t fHistoPtMax ;    //Maximum value of Pt trigger histogram range
 120   Float_t fHistoPtMin ;    //Minimum value of Pt trigger histogram range
 121   Int_t   fHistoNAngleBins ; //Number of bins in angle axis
 122   Float_t fHistoAngleMax ;//Maximum value of angle histogram range
 123   Float_t fHistoAngleMin ;//Minimum value of angle histogram range
 124   Int_t   fHistoNIMBins ; //Number of bins in Invariant Mass axis
 125   Float_t fHistoIMMax ;   //Maximum value of Invariant Mass histogram range
 126   Float_t fHistoIMMin ;   //Minimum value of Invariant Mass histogram range
 127
 128   ClassDef(AliNeutralMesonSelection,2)
 129
 130     } ;
 131
 132
 133 #endif //ALINEUTRALMESONSELECTION_H
 134
 135
 136