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 */
6 //#############################################################
8 //# Class AliDielectronCF #
9 //# Dielectron Correction Framework Manager #
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 #
20 //#############################################################
25 #include <TVectorDfwd.h>
27 #include "AliDielectronVarManager.h"
29 class AliAnalysisCuts;
30 class AliAnalysisFilter;
32 class AliDielectronPair;
35 class AliDielectronCF : public TNamed {
37 enum {kNmaxAddSteps=50};
40 AliDielectronCF(const char* name, const char* title);
41 virtual ~AliDielectronCF();
43 void SetStepForMCtruth(Bool_t steps=kTRUE) { fStepForMCtruth=steps; }
44 Bool_t GetStepForMCtruth() const { return fStepForMCtruth; }
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 void SetStepsForMCtruthOnly(Bool_t steps=kTRUE) { fStepsForMCtruthOnly=steps; }
54 void SetPdgMother(Int_t pdg) { fPdgMother=pdg; }
55 void SetSignalsMC(TObjArray* array) {fSignalsMC = array;}
57 void AddStepMask(UInt_t mask) { fStepMasks[fNStepMasks++]=mask; }
59 void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins,
60 Double_t min, Double_t max, Bool_t leg=kFALSE, Bool_t log=kFALSE);
61 void AddVariable(AliDielectronVarManager::ValueTypes type, const char* binLimitStr, Bool_t leg=kFALSE);
62 void AddVariable(AliDielectronVarManager::ValueTypes type, TVectorD *binLimits, Bool_t leg=kFALSE);
64 void InitialiseContainer(const AliAnalysisFilter& filter);
66 Int_t GetNvarsPair() const {return fNVars;}
67 Int_t GetNvarsLeg() const {return fNVarsLeg;}
69 UInt_t GetVariablePair(UInt_t var) const {return (var>=(UInt_t)AliDielectronVarManager::kNMaxValues)? (UInt_t)AliDielectronVarManager::kNMaxValues+1:fVariables[var];}
70 UInt_t GetVariableLeg(UInt_t var) const {return (var>=(UInt_t)AliDielectronVarManager::kNMaxValues)? (UInt_t)AliDielectronVarManager::kNMaxValues+1:fVariablesLeg[var];}
72 // void Fill(UInt_t mask, const TObject *particle);
73 void Fill(UInt_t mask, const AliDielectronPair *particle);
74 void FillMC(const TObject *particle);
75 void FillMC(Int_t label1, Int_t label2, Int_t nSignal);
77 AliCFContainer* GetContainer() const { return fCfContainer; }
80 TBits *fUsedVars; // list of used variables
82 UInt_t fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
83 UInt_t fVariablesLeg[AliDielectronVarManager::kNMaxValues]; //configured variables for the legs
85 Int_t fNSteps; // number of selection steps
87 Int_t fNVars; // number of variables
88 TObjArray *fVarBinLimits; // array of bin limits
90 Int_t fNVarsLeg; // number of variables for the legs
91 TObjArray *fVarBinLimitsLeg; // array of bin limits of the legs
93 Int_t fNCuts; // Number of cuts in the filter concerned
95 Double_t *fValues; //! Value array for filling the container
96 Bool_t *fIsMCTruth; //! Buffer array for MC truth information
98 Bool_t fStepForMCtruth; //create a step for the MC truth
99 Bool_t fStepForNoCutsMCmotherPid; //create a step for before cuts, but with MC truth of the mother
100 Bool_t fStepForAfterAllCuts; //create a step for before cuts, but with MC truth of the mother
101 Bool_t fStepForPreFilter; //create a step after PreFilter application
102 Bool_t fStepsForEachCut; //create steps for each cut?
103 Bool_t fStepsForCutsIncreasing; //create steps for increasing cut combinatons?
104 //e.g. cut1&cut2, cut1&cut2&cut3 ...
105 Bool_t fStepsForSignal; //steps for pure signal
106 Bool_t fStepsForBackground; //steps for pure background
107 Bool_t fStepsForMCtruthOnly; //Switch off all pair steps, allow only MC truth Class
109 UInt_t fStepMasks[kNmaxAddSteps]; //steps for additional cut combinatons
110 UInt_t fNStepMasks; //number of configured step masks
112 Int_t fPdgMother; //Pdg code of MCtruth validation
113 TObjArray* fSignalsMC; //! array of MC signals to be studied
114 AliCFContainer* fCfContainer; //the CF container
116 Bool_t fHasMC; //if MC info is available
117 Int_t fNAddSteps; //number of additional MC related steps per cut step
119 TVectorD* MakeLogBinning(Int_t nbinsX, Double_t xmin, Double_t xmax) const;
120 TVectorD* MakeLinBinning(Int_t nbinsX, Double_t xmin, Double_t xmax) const;
122 AliDielectronCF(const AliDielectronCF &c);
123 AliDielectronCF &operator=(const AliDielectronCF &c);
125 ClassDef(AliDielectronCF,5) //Dielectron Correction Framework handler