including switch to set on/off iso-track core removal, cleaning and bug fix

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

diff --git a/PWGDQ/dielectron/AliDielectronCFdraw.h b/PWGDQ/dielectron/AliDielectronCFdraw.h
index e1b806795c89455c50534f5bc33a7d3b80ec96ca..cff3a73bdae338b3b91d9612eefce6a5138483a8 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronCFdraw.h
+++ b/PWGDQ/dielectron/AliDielectronCFdraw.h
@@ -60,6 +60,8 @@ public:
   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(""); }
@@ -103,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