Change Mult binning scheme

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronCFdraw.h

diff --git a/PWGDQ/dielectron/AliDielectronCFdraw.h b/PWGDQ/dielectron/AliDielectronCFdraw.h
index f7799a631b085ce2cdc9d950a1d7799ad067262f..cff3a73bdae338b3b91d9612eefce6a5138483a8 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronCFdraw.h
+++ b/PWGDQ/dielectron/AliDielectronCFdraw.h
@@ -24,6 +24,7 @@
 #include <TNamed.h>
 #include <TVectorD.h>
 #include <AliCFContainer.h>
+#include "AliDielectronVarManager.h"
 
 class TObjArray;
 class TSeqCollection;
@@ -34,6 +35,8 @@ class AliCFContainer;
 
 class AliDielectronCFdraw : public TNamed {
 public:
+  enum ECollectType { kSE=0, kME, kMEOS, kROT, kAll };
+  
   AliDielectronCFdraw();
   AliDielectronCFdraw(const char* name, const char* title);
   AliDielectronCFdraw(AliCFContainer *cont);
@@ -50,9 +53,15 @@ public:
 
   void SetRangeUser(Int_t ivar, Double_t min, Double_t max, const char* slices="");
   void SetRangeUser(const char* varname, Double_t min, Double_t max, const char* slices="");
+  void SetRangeUser(AliDielectronVarManager::ValueTypes type, Double_t min, Double_t max, const char* slices="", Bool_t leg=kFALSE);
 
   void UnsetRangeUser(Int_t ivar, const char* slices="");
   void UnsetRangeUser(const char* varname, const char* slices="");
+  void UnsetRangeUser(AliDielectronVarManager::ValueTypes type, const char* slices="", Bool_t leg=kFALSE);
+
+  TString FindSteps(const char* search="");
+  Int_t   FindStep(const char* search="");
+  Int_t   FindVar(AliDielectronVarManager::ValueTypes type, Bool_t leg=kFALSE);
 
   virtual void Draw(const Option_t* varnames = "") { Draw(varnames,"");}
   virtual void Print(const Option_t*) const { if (fCfContainer) fCfContainer->Print(""); }
@@ -64,7 +73,7 @@ public:
 
   TObjArray* CollectHistosProj(const Option_t* varnames, const char* slices);
   TObjArray* CollectHistosProj(const Int_t vars[3], const char* slices);
-  TObjArray* CollectMinvProj(Int_t slice);
+  TObjArray* CollectMinvProj(Int_t slice, ECollectType collect=kAll, TString var="M");
   TH1* Project(const Int_t vars[3], Int_t slice);
   TH1* Project(const Option_t* var, Int_t slice);
   
@@ -96,5 +105,13 @@ private:
 //
 // Inline functions
 //
+
+inline Int_t AliDielectronCFdraw::FindVar(AliDielectronVarManager::ValueTypes type, Bool_t leg)
+{
+  //
+  // find variable number in CF container
+  //
+  return ( fCfContainer->GetVar(Form("%s%s", leg?"Leg1_":"", AliDielectronVarManager::GetValueName(type))) );
+}
 #endif