AliPIDResponse related

[u/mrichter/AliRoot.git] / STEER / TTreeStream.h

diff --git a/STEER/TTreeStream.h b/STEER/TTreeStream.h
index c09d48ad8dcce8bf74e09b9013019e82dfd18854..7d9a6a25d685dfcd21c8e06c3c8b2f444a8533e1 100644 (file)
--- a/STEER/TTreeStream.h
+++ b/STEER/TTreeStream.h
@@ -6,6 +6,7 @@
 
 #include "TObject.h"
 #include "TString.h"
+#include "TTree.h"
 class TFile;
 class TObjArray;
 class TTree;
@@ -24,12 +25,11 @@ class TTreeDataElement: public TNamed {
   TTreeDataElement(const TTreeDataElement & tde);
   TTreeDataElement & operator=(const TTreeDataElement & tde);
 
-  TString fName;    // name of the data element
   Char_t  fType;     // type of data element
   TDataType *fDType; //data type pointer 
   TClass    *fClass; //data type pointer
   void * fPointer;  // pointer to element
-  ClassDef(TTreeDataElement,1)
+  ClassDef(TTreeDataElement,2)
 };
 
 class TTreeStream: public TNamed {
@@ -44,6 +44,7 @@ public:
   Int_t CheckIn(TObject *o);
   void BuildTree();
   void Fill();
+  Double_t GetSize(){ return fTree->GetZipBytes();}
   TTreeStream& Endl();
   //
   TTreeStream  &operator<<(Bool_t   &b){CheckIn('B',&b);return *this;}
@@ -60,7 +61,8 @@ public:
   TTreeStream  &operator<<(Float_t   &f){CheckIn('F',&f);return *this;}
   TTreeStream  &operator<<(Double_t  &d){CheckIn('D',&d);return *this;}
   TTreeStream  &operator<<(TObject*o){CheckIn(o);return *this;} 
-  TTreeStream  &operator<<(Char_t *name);
+  TTreeStream  &operator<<(const Char_t *name);
+  TTree * GetTree() const { return fTree;}
  protected:
   //
 
@@ -83,9 +85,11 @@ public:
 class TTreeSRedirector: public TObject { 
 public:
   TTreeSRedirector(const char *fname);
-  ~TTreeSRedirector();
+  virtual ~TTreeSRedirector();
   void Close();
   static void Test();
+  void StoreObject(TObject* object);
+  TFile * GetFile() {return fFile;};
   virtual   TTreeStream  &operator<<(Int_t id);
   virtual   TTreeStream  &operator<<(const char *name);
  private: