Add a draw class for the CORRFW (produces a warning, will be fixed

[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronCF.h

diff --git a/PWG3/dielectron/AliDielectronCF.h b/PWG3/dielectron/AliDielectronCF.h
index 272791c982b755e4745ae931ad328736857127f2..609cd42834d9cf694562764ffb205a5eff667cb7 100644 (file)
--- a/PWG3/dielectron/AliDielectronCF.h
+++ b/PWG3/dielectron/AliDielectronCF.h
@@ -27,6 +27,7 @@
 class AliAnalysisCuts;
 class AliAnalysisFilter;
 class AliCFContainer;
+class AliDielectronPair;
 
 class AliDielectronCF : public TNamed {
 public:
@@ -46,26 +47,36 @@ public:
   
   void AddStepMask(UInt_t mask)                  { fStepMasks[fNStepMasks++]=mask; }
   
-  void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins, Double_t min, Double_t max);
+  void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins, Double_t min, Double_t max, Bool_t leg=kFALSE);
 
   void InitialiseContainer(const AliAnalysisFilter& filter);
   
-  void Fill(UInt_t mask, const TObject *particle);
+//   void Fill(UInt_t mask, const TObject *particle);
+  void Fill(UInt_t mask, const AliDielectronPair *particle);
   void FillMC(const TObject *particle);
   
   AliCFContainer* GetContainer() const { return fCfContainer; }
   
-// private:
+private:
   UInt_t          fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
+  UInt_t          fVariablesLeg[AliDielectronVarManager::kNMaxValues]; //configured variables for the legs
   
   Int_t           fNSteps;                     // number of selection steps
+  
   Int_t           fNVars;                      // number of variables
   Int_t           fNBins[kNmaxAddSteps];       // array of numbers ob bins of the vars
   Double_t        fVarLoLimit[kNmaxAddSteps];  // array of the lower limits of the vars
   Double_t        fVarUpLimit[kNmaxAddSteps];  // array of the upper limits of the vars
 
-  Int_t           fNCuts;                      // Number of cuts in the filter concerned
+  Int_t           fNVarsLeg;                      // number of variables for the legs
+  Int_t           fNBinsLeg[kNmaxAddSteps];       // array of numbers ob bins of the vars for the legs
+  Double_t        fVarLoLimitLeg[kNmaxAddSteps];  // array of the lower limits of the vars for the legs
+  Double_t        fVarUpLimitLeg[kNmaxAddSteps];  // array of the upper limits of the vars for the legs
+  
+  Int_t           fNCuts;                         // Number of cuts in the filter concerned
 
+  Double_t        *fValues;                       // Value array for filling the container
+  
   Bool_t fStepForMCtruth;               //create a step for the MC truth
   Bool_t fStepForNoCutsMCmotherPid;     //create a step for before cuts, but with MC truth of the mother
   Bool_t fStepForAfterAllCuts;          //create a step for before cuts, but with MC truth of the mother
@@ -82,7 +93,7 @@ public:
   AliDielectronCF(const AliDielectronCF &c);
   AliDielectronCF &operator=(const AliDielectronCF &c);
   
-  ClassDef(AliDielectronCF,2)  //Dielectron Correction Framework handler
+  ClassDef(AliDielectronCF,3)  //Dielectron Correction Framework handler
 };
 
 #endif