PWG3/dielectron/AliDielectronCF.h

   1 #ifndef ALIDIELECTRONCF_H
   2 #define ALIDIELECTRONCF_H
   3 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //#############################################################
   9 //#                                                           #
  10 //#             Class AliDielectronCF                         #
  11 //#       Dielectron Correction Framework Manager             #
  12 //#                                                           #
  13 //#  Authors:                                                 #
  14 //#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
  15 //#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
  16 //#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
  17 //#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
  18 //#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
  19 //#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
  20 //#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
  21 //#                                                           #
  22 //#############################################################
  23
  24
  25
  26 #include <TNamed.h>
  27 #include <TVectorDfwd.h>
  28 #include "AliDielectronVarManager.h"
  29
  30 class AliAnalysisCuts;
  31 class AliAnalysisFilter;
  32 class AliCFContainer;
  33 class AliDielectronPair;
  34 class TObjArray;
  35
  36 class AliDielectronCF : public TNamed {
  37 public:
  38   enum {kNmaxAddSteps=50};
  39
  40   AliDielectronCF();
  41   AliDielectronCF(const char* name, const char* title);
  42   virtual ~AliDielectronCF();
  43
  44   void SetStepForMCtruth(Bool_t steps=kTRUE)           { fStepForMCtruth=steps;           }
  45   void SetStepForNoCutsMCmotherPid(Bool_t steps=kTRUE) { fStepForNoCutsMCmotherPid=steps; }
  46   void SetStepForAfterAllCuts(Bool_t steps=kTRUE)      { fStepForAfterAllCuts=steps;      }
  47   void SetStepForPreFilter(Bool_t steps=kTRUE)         { fStepForPreFilter=steps;         }
  48   void SetStepsForEachCut(Bool_t steps=kTRUE)          { fStepsForEachCut=steps;          }
  49   void SetStepsForCutsIncreasing(Bool_t steps=kTRUE)   { fStepsForCutsIncreasing=steps;   }
  50   void SetStepsForSignal(Bool_t steps=kTRUE)           { fStepsForSignal=steps;           }
  51   void SetStepsForBackground(Bool_t steps=kTRUE)       { fStepsForBackground=steps;       }
  52
  53   void SetPdgMother(Int_t pdg) { fPdgMother=pdg; }
  54
  55   void AddStepMask(UInt_t mask)                  { fStepMasks[fNStepMasks++]=mask; }
  56
  57   void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins,
  58                    Double_t min, Double_t max, Bool_t leg=kFALSE, Bool_t log=kFALSE);
  59   void AddVariable(AliDielectronVarManager::ValueTypes type, const char* binLimitStr, Bool_t leg=kFALSE);
  60   void AddVariable(AliDielectronVarManager::ValueTypes type, TVectorD *binLimits, Bool_t leg=kFALSE);
  61
  62   void InitialiseContainer(const AliAnalysisFilter& filter);
  63
  64 //   void Fill(UInt_t mask, const TObject *particle);
  65   void Fill(UInt_t mask, const AliDielectronPair *particle);
  66   void FillMC(const TObject *particle);
  67
  68   AliCFContainer* GetContainer() const { return fCfContainer; }
  69
  70 private:
  71   UInt_t          fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
  72   UInt_t          fVariablesLeg[AliDielectronVarManager::kNMaxValues]; //configured variables for the legs
  73
  74   Int_t           fNSteps;                     // number of selection steps
  75
  76   Int_t           fNVars;                      // number of variables
  77   TObjArray      *fVarBinLimits;               // array of bin limits
  78
  79   Int_t           fNVarsLeg;                   // number of variables for the legs
  80   TObjArray      *fVarBinLimitsLeg;            //  array of bin limits of the legs
  81
  82   Int_t           fNCuts;                         // Number of cuts in the filter concerned
  83
  84   Double_t        *fValues;                       //! Value array for filling the container
  85
  86   Bool_t fStepForMCtruth;               //create a step for the MC truth
  87   Bool_t fStepForNoCutsMCmotherPid;     //create a step for before cuts, but with MC truth of the mother
  88   Bool_t fStepForAfterAllCuts;          //create a step for before cuts, but with MC truth of the mother
  89   Bool_t fStepForPreFilter;             //create a step after PreFilter application
  90   Bool_t fStepsForEachCut;              //create steps for each cut?
  91   Bool_t fStepsForCutsIncreasing;       //create steps for increasing cut combinatons?
  92                                         //e.g. cut1&cut2, cut1&cut2&cut3 ...
  93   Bool_t fStepsForSignal;               //steps for pure signal
  94   Bool_t fStepsForBackground;           //steps for pure background
  95
  96   UInt_t fStepMasks[kNmaxAddSteps];      //steps for additional cut combinatons
  97   UInt_t fNStepMasks;                    //number of configured step masks
  98
  99   Int_t fPdgMother;                      //Pdg code of MCtruth validation
 100   AliCFContainer* fCfContainer;          //the CF container
 101
 102   Bool_t fHasMC;                         //if MC info is available
 103   Int_t  fNAddSteps;                     //number of additional MC related steps per cut step
 104
 105   TVectorD* MakeLogBinning(Int_t nbinsX, Double_t xmin, Double_t xmax) const;
 106   TVectorD* MakeLinBinning(Int_t nbinsX, Double_t xmin, Double_t xmax) const;
 107
 108   AliDielectronCF(const AliDielectronCF &c);
 109   AliDielectronCF &operator=(const AliDielectronCF &c);
 110
 111   ClassDef(AliDielectronCF,4)  //Dielectron Correction Framework handler
 112 };
 113
 114 #endif