Improved searching helpers in AliParticleYield

[u/mrichter/AliRoot.git] / PWGLF / ThermalFits / AliParticleYield.h

diff --git a/PWGLF/ThermalFits/AliParticleYield.h b/PWGLF/ThermalFits/AliParticleYield.h
index 633d451d489861d9b3cb12d764ba4cdbf961b729..0808eb86464b9fce972278ba8f5391a8279c642a 100644 (file)
--- a/PWGLF/ThermalFits/AliParticleYield.h
+++ b/PWGLF/ThermalFits/AliParticleYield.h
@@ -9,10 +9,10 @@
 
 #include "TObject.h"
 #include "TString.h"
+#include "TCut.h"
 
 class TClonesArray;
 class TTree;
-
 class AliParticleYield : public TObject
 {
 public:
@@ -53,7 +53,8 @@ public:
   static TTree * ReadFromASCIIFileAsTree(const char * fileName, const char * separators = " \t");
   static void SaveAsASCIIFile(TClonesArray * arr, const char * filename, const char * separator = " ", Int_t colWidth = 7);  
   static void WriteThermusFile(TClonesArray * arr, const char * filename, Int_t colWidth = 10);
-  static TClonesArray * GetEntriesMatchingSelection(TTree * tree, TString selection); 
+  static TClonesArray * GetEntriesMatchingSelection(TTree * tree, TCut selection); 
+  static TTree * GetTreeFromArray(TClonesArray * arr) ;
 
   // Misc helpers
   Bool_t CheckTypeConsistency() const;