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 //#############################################################
   7 //#                                                           #
   8 //#             Class AliDielectronCF                         #
   9 //#       Dielectron Correction Framework Manager             #
  10 //#                                                           #
  11 //#  Authors:                                                 #
  12 //#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
  13 //#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
  14 //#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
  15 //#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
  16 //#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
  17 //#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
  18 //#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
  19 //#                                                           #
  20 //#############################################################
  21
  22
  23
  24 #include <TNamed.h>
  25 #include "AliDielectronVarManager.h"
  26
  27 class AliAnalysisCuts;
  28 class AliAnalysisFilter;
  29 class AliCFContainer;
  30
  31 class AliDielectronCF : public TNamed {
  32 public:
  33   enum {kNmaxAddSteps=50};
  34
  35   AliDielectronCF();
  36   AliDielectronCF(const char* name, const char* title);
  37   virtual ~AliDielectronCF();
  38
  39   void SetStepForMCtruth(Bool_t steps=kTRUE)           { fStepForMCtruth=steps;           }
  40   void SetStepForNoCutsMCmotherPid(Bool_t steps=kTRUE) { fStepForNoCutsMCmotherPid=steps; }
  41   void SetStepForAfterAllCuts(Bool_t steps=kTRUE)      { fStepForAfterAllCuts=steps;      }
  42   void SetStepsForEachCut(Bool_t steps=kTRUE)          { fStepsForEachCut=steps;          }
  43   void SetStepsForCutsIncreasing(Bool_t steps=kTRUE)   { fStepsForCutsIncreasing=steps;   }
  44
  45   void SetPdgMother(Int_t pdg) { fPdgMother=pdg; }
  46
  47   void AddStepMask(UInt_t mask)                  { fStepMasks[fNStepMasks++]=mask; }
  48
  49   void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins, Double_t min, Double_t max);
  50
  51   void InitialiseContainer(const AliAnalysisFilter& filter);
  52
  53   void Fill(UInt_t mask, const TObject *particle);
  54   void FillMC(const TObject *particle);
  55
  56   AliCFContainer* GetContainer() const { return fCfContainer; }
  57
  58 // private:
  59   UInt_t          fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
  60
  61   Int_t           fNSteps;                     // number of selection steps
  62   Int_t           fNVars;                      // number of variables
  63   Int_t           fNBins[kNmaxAddSteps];       // array of numbers ob bins of the vars
  64   Double_t        fVarLoLimit[kNmaxAddSteps];  // array of the lower limits of the vars
  65   Double_t        fVarUpLimit[kNmaxAddSteps];  // array of the upper limits of the vars
  66
  67   Int_t           fNCuts;                      // Number of cuts in the filter concerned
  68
  69   Bool_t fStepForMCtruth;               //create a step for the MC truth
  70   Bool_t fStepForNoCutsMCmotherPid;     //create a step for before cuts, but with MC truth of the mother
  71   Bool_t fStepForAfterAllCuts;          //create a step for before cuts, but with MC truth of the mother
  72   Bool_t fStepsForEachCut;              //create steps for each cut?
  73   Bool_t fStepsForCutsIncreasing;       //create steps for increasing cut combinatons?
  74                                         //e.g. cut1&cut2, cut1&cut2&cut3 ...
  75
  76   UInt_t fStepMasks[kNmaxAddSteps];      //steps for additional cut combinatons
  77   UInt_t fNStepMasks;                    //number of configured step masks
  78
  79   Int_t fPdgMother;                      //Pdg code of MCtruth validation
  80   AliCFContainer* fCfContainer;          //the CF container
  81
  82   AliDielectronCF(const AliDielectronCF &c);
  83   AliDielectronCF &operator=(const AliDielectronCF &c);
  84
  85   ClassDef(AliDielectronCF,2)  //Dielectron Correction Framework handler
  86 };
  87
  88 #endif