3 ///////////////////////////////////////////////////////////////////////////////
10 #include "TDirectory.h"
16 class TTreeDataElement: public TNamed {
17 friend class TTreeStream;
19 TTreeDataElement(Char_t type);
20 TTreeDataElement(TDataType* type);
21 TTreeDataElement(TClass* cl);
22 void SetPointer(void* pointer) {fPointer=pointer;}
23 Char_t GetType() const {return fType;}
26 TTreeDataElement(const TTreeDataElement & tde);
27 TTreeDataElement & operator=(const TTreeDataElement & tde);
29 Char_t fType; // type of data element
30 TDataType *fDType; //data type pointer
31 TClass *fClass; //data type pointer
32 void * fPointer; // pointer to element
33 ClassDef(TTreeDataElement,2)
36 class TTreeStream: public TNamed {
37 friend class TTreeSRedirector;
39 TTreeStream(const char *treename, TTree* externalTree=NULL);
43 Int_t CheckIn(Char_t type, void *pointer);
44 //Int_t CheckIn(const char *type, void *pointer);
45 Int_t CheckIn(TObject *o);
48 Double_t GetSize(){ return fTree->GetZipBytes();}
51 TTreeStream &operator<<(Bool_t &b){CheckIn('B',&b);return *this;}
52 TTreeStream &operator<<(Char_t &c){CheckIn('B',&c);return *this;}
53 TTreeStream &operator<<(UChar_t &c){CheckIn('b',&c);return *this;}
54 TTreeStream &operator<<(Short_t &h){CheckIn('S',&h);return *this;}
55 TTreeStream &operator<<(UShort_t &h){CheckIn('s',&h);return *this;}
56 TTreeStream &operator<<(Int_t &i){CheckIn('I',&i);return *this;}
57 TTreeStream &operator<<(UInt_t &i){CheckIn('i',&i);return *this;}
58 TTreeStream &operator<<(Long_t &l){CheckIn('L',&l);return *this;}
59 TTreeStream &operator<<(ULong_t &l){CheckIn('l',&l);return *this;}
60 TTreeStream &operator<<(Long64_t &l){CheckIn('L',&l);return *this;}
61 TTreeStream &operator<<(ULong64_t &l){CheckIn('l',&l);return *this;}
62 TTreeStream &operator<<(Float_t &f){CheckIn('F',&f);return *this;}
63 TTreeStream &operator<<(Double_t &d){CheckIn('D',&d);return *this;}
64 TTreeStream &operator<<(TObject*o){CheckIn(o);return *this;}
65 TTreeStream &operator<<(const Char_t *name);
66 TTree * GetTree() const { return fTree;}
70 TTreeStream(const TTreeStream & ts);
71 TTreeStream & operator=(const TTreeStream & ts);
73 TObjArray *fElements; //array of elements
74 TObjArray *fBranches; //pointers to branches
75 TTree *fTree; //data storage
76 Int_t fCurrentIndex; //index of current element
77 Int_t fId; //identifier of layout
78 TString fNextName; //name for next entry
79 Int_t fNextNameCounter; //next name counter
80 Int_t fStatus; //status of the layout
81 ClassDef(TTreeStream,1)
86 class TTreeSRedirector: public TObject {
88 TTreeSRedirector(const char *fname="", const char * option="new");
89 virtual ~TTreeSRedirector();
93 void StoreObject(TObject* object);
94 TFile * GetFile() {return fDirectory->GetFile();}
95 TDirectory * GetDirectory() {return fDirectory;}
96 virtual TTreeStream &operator<<(Int_t id);
97 virtual TTreeStream &operator<<(const char *name);
98 void SetDirectory(TDirectory *sfile);
99 void SetFile(TFile *sfile) {SetDirectory(sfile);}
100 void SetExternalTree(const char* name, TTree* externalTree);
104 TTreeSRedirector(const TTreeSRedirector & tsr);
105 TTreeSRedirector & operator=(const TTreeSRedirector & tsr);
107 TDirectory* fDirectory; //file
108 Bool_t fDirectoryOwner; //do we own the directory?
109 TObjArray *fDataLayouts; //array of data layouts
110 ClassDef(TTreeSRedirector,2)