ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitAna/DataUtil/interface/TreeWriter.h
Revision: 1.4
Committed: Thu Jun 5 07:55:55 2008 UTC (16 years, 11 months ago) by loizides
Content type: text/plain
Branch: MAIN
Changes since 1.3: +12 -1 lines
Log Message: 
Also accept TObject pointers and figure out ClassName automatically.

File Contents

# User Rev Content
1 loizides 1.4 // $Id: TreeWriter.h,v 1.3 2008/06/02 08:58:52 loizides Exp $
2 loizides 1.1
3     #ifndef DATATREE_TREEWRITER_H
4     #define DATATREE_TREEWRITER_H
5    
6     #include <TNamed.h>
7     #include <TString.h>
8     #include <TFile.h>
9     #include <TTree.h>
10 loizides 1.3 #include <TObjArray.h>
11 loizides 1.1
12     //--------------------------------------------------------------------------------------------------
13     //
14     // TreeWriter
15     //
16     // Class implementing a tree writing infra structure. The class creates and maintaines
17 loizides 1.3 // a standard tree with name "tname" as given in the constructor. The tree is stored in
18     // a file or in multiple files if needed, where the file path name and maximum size can be
19 loizides 1.1 // specified. Using TreeWriter::AddBranch allows to add various branches to the tree.
20     // The user has to make sure that (s)he calls TreeWriter::BeginEvent and
21     // TreeWriter::EndEvent for every event before and after filling of the branch
22 loizides 1.3 // structures. Note that in addition to the standard tree one can create new
23     // tree(s) in the same file using the TreeWriter::AddBranchToTree member functions.
24 loizides 1.1 //
25     // Authors: C.Loizides
26     //
27     //--------------------------------------------------------------------------------------------------
28    
29 loizides 1.2 namespace mithep
30     {
31 loizides 1.3 class MyTree : public TTree
32     {
33     public:
34     MyTree(const char* name, const char* title, Int_t splitlevel = 99) :
35     TTree(name,title, splitlevel), fAutoFill(1) {}
36     Bool_t GetAutoFill() const { return fAutoFill; }
37     void SetAutoFill(Bool_t b) { fAutoFill = b; }
38     protected:
39     Bool_t fAutoFill; //!=true then fill automatically in TreeWriter (def=1)
40     };
41    
42 loizides 1.1 class TreeWriter : public TNamed
43     {
44     public:
45     TreeWriter(const char *tname="MitTree", Bool_t doreset = kFALSE);
46     ~TreeWriter();
47    
48 loizides 1.3 void AddBranch(const char *name, const char *cname,
49     void *obj, Int_t bsize, Int_t level);
50 loizides 1.4 void AddBranch(const char *name, void *obj, Int_t bsize, Int_t level);
51 loizides 1.3 void AddBranch(const char *name, const char *cname,
52     void *obj, Int_t bsize);
53 loizides 1.4 void AddBranch(const char *name, void *obj, Int_t bsize);
54 loizides 1.3 void AddBranch(const char *name, const char *cname,
55     void *obj);
56 loizides 1.4 void AddBranch(const char *name, void *obj);
57 loizides 1.3 void AddBranchToTree(const char *tname, const char *name, const char *cname,
58     void *obj, Int_t bsize, Int_t level);
59 loizides 1.4 void AddBranchToTree(const char *tname, const char *name, void *obj,
60     Int_t bsize, Int_t level);
61 loizides 1.3 void AddBranchToTree(const char *tname, const char *name, const char *cname,
62     void *obj, Int_t bsize);
63 loizides 1.4 void AddBranchToTree(const char *tname, const char *name, void *obj,
64     Int_t bsize);
65 loizides 1.3 void AddBranchToTree(const char *tname, const char *name, const char *cname,
66     void *obj);
67 loizides 1.4 void AddBranchToTree(const char *tname, const char *name, void *obj);
68 loizides 1.1 Bool_t BeginEvent(Bool_t doreset=kFALSE);
69     Bool_t EndEvent(Bool_t doreset=kFALSE);
70 loizides 1.3 const char *GetBaseURL() const { return fBaseURL.IsNull() ? "." : fBaseURL; }
71     Int_t GetCompressLevel() const { return fCompressLevel; }
72     Long64_t GetEntries(const char *tn=0) const;
73     Long64_t GetFileSize() const { return fFile != 0 ? fFile->GetEND() : 0; }
74     const TFile *GetFile() const { return fFile; }
75     const char *GetFileName() const { return Form("%s_%03d.root",
76     GetPrefix(), GetFileNumber()); }
77     UShort_t GetFileNumber() const { return fFileNumber; }
78     const char *GetFullName() const { return Form("%s/%s",
79     GetBaseURL(), GetFileName()); }
80     const char *GetPrefix() const { return fPrefix; }
81     const TTree *GetTree(const char *tn) const;
82     TTree *GetTree(const char *tn);
83     void Print(Option_t *option="") const;
84     void SetAutoFill(const char *tn, Bool_t b);
85     void SetBaseURL(const char *b) { fBaseURL = b; }
86     void SetCompressLevel(Int_t l) { fCompressLevel = l; }
87     void SetDefaultBrSize(Int_t s) { fDefBrSize=s; }
88     void SetDefaultSL(Int_t s) { fDefSL=s;}
89     void SetMaxSize(Long64_t s) { fMaxSize=s; }
90     void SetPrefix(const char *p) { fPrefix = p; }
91 loizides 1.1 void StoreObject(const TObject *obj);
92    
93     protected:
94 loizides 1.3 TString fBaseURL; //base url for tree storage
95     TString fPrefix; //prefix of file name
96     UShort_t fFileNumber; //current sequence number
97     Int_t fCompressLevel; //compression level used for TFile
98     Int_t fDefBrSize; //default buffer size for branches
99     Int_t fDefSL; //default split level for branches
100     Long64_t fMaxSize; //maximum file size for a file [Bytes]
101     const Long64_t fkMinFreeSpace; //minimum free space required for closing file
102     const Long64_t fkMinAvgSize; //minimum average entry size
103     Long64_t fEvtObjNum; //event object number offset (for TRef)
104     Bool_t fIsInit; //true if OpenFile() was called
105     Bool_t fDoObjNumReset; //true if obj. number resets automatically (def=0)
106     TFile *fFile; //file being written
107     TObjArray fTrees; //array of tree(s) being filled
108    
109     mithep::MyTree *AddOrGetMyTree(const char *tn);
110     mithep::MyTree *GetMyTree(const char *tn);
111     Bool_t IsInit() const { return fIsInit; }
112     Bool_t IsFull() const;
113 loizides 1.1 void OpenFile();
114     void CloseFile();
115    
116 loizides 1.4 private:
117     const char *CName(void *obj) const;
118    
119 loizides 1.1 ClassDef(TreeWriter,0) // Tree writer class
120     };
121    
122     } /*namespace mithep*/
123    
124     #endif /*DATATREE_TREEWRITER_H*/