DataUtil/interface/TreeWriter.h

//--------------------------------------------------------------------------------------------------
// $Id: TreeWriter.h,v 1.9 2008/06/24 14:05:03 loizides Exp $
//
// TreeWriter
//
// Class implementing a tree writing infrastructure. The class creates and maintaines
// a standard tree with name "tname" as given in the constructor. The tree is stored in 
// a file or in multiple files if needed, where the file path name and maximum size can be
// specified. Using TreeWriter::AddBranch allows to add various branches to the tree.
// The user has to make sure that (s)he calls TreeWriter::BeginEvent and 
// TreeWriter::EndEvent for every event before and after filling of the branch 
// structures. Note that in addition to the standard tree one can create new
// tree(s) in the same file using the TreeWriter::AddBranchToTree member functions.
//
// Authors: C.Loizides
//--------------------------------------------------------------------------------------------------

#ifndef MITANA_DATAUTIL_TREEWRITER_H
#define MITANA_DATAUTIL_TREEWRITER_H

#include <TNamed.h>
#include <TString.h>
#include <TFile.h>
#include <TTree.h>
#include <TObjArray.h>

namespace mithep 
{
  class MyTree : public TTree 
  {
    public:
      MyTree(const char* name, const char* title, Int_t splitlevel = 99) :
        TTree(name,title, splitlevel), fAutoFill(1) {}
        Bool_t GetAutoFill() const   { return fAutoFill; }
        void   SetAutoFill(Bool_t b) { fAutoFill = b; }
    protected:
      Bool_t fAutoFill; //!=true then fill automatically in TreeWriter (def=1)
  };
  
  class TreeWriter : public TNamed 
  {
    public:
      TreeWriter(const char *tname="MitTree", Bool_t doreset = kFALSE);
      ~TreeWriter();

      void                 AddBranch(const char *name, const char *cname, 
                                     void *obj,  Int_t bsize, Int_t level);
      void                 AddBranch(const char *name, void *obj, Int_t bsize, Int_t level);
      void                 AddBranch(const char *name, const char *cname, 
                                     void *obj, Int_t bsize);
      void                 AddBranch(const char *name, void *obj, Int_t bsize);
      void                 AddBranch(const char *name, const char *cname, 
                                     void *obj);
      void                 AddBranch(const char *name, void *obj);
      void                 AddBranchToTree(const char *tname, const char *name, const char *cname, 
                                           void *obj,  Int_t bsize, Int_t level);
      void                 AddBranchToTree(const char *tname, const char *name, void *obj,  
                                           Int_t bsize, Int_t level);
      void                 AddBranchToTree(const char *tname, const char *name, const char *cname, 
                                           void *obj, Int_t bsize);
      void                 AddBranchToTree(const char *tname, const char *name, void *obj, 
                                           Int_t bsize);
      void                 AddBranchToTree(const char *tname, const char *name, const char *cname, 
                                           void *obj);
      void                 AddBranchToTree(const char *tname, const char *name, void *obj);
      Bool_t               BeginEvent(Bool_t doreset=kFALSE);
      Bool_t               EndEvent(Bool_t doreset=kFALSE);
      const char          *GetBaseURL()                 const {
        return fBaseURL.IsNull() ? "." : fBaseURL; }  
      Int_t                GetCompressLevel()           const { return fCompressLevel; }
      Long64_t             GetEntries(const char *tn=0) const;
      Long64_t             GetFileSize()                const {
        return fFile != 0 ? fFile->GetEND() : 0; }
      const TFile         *GetFile()                    const { return fFile; }
      const char          *GetFileName()                const {
        return Form("%s_%03d.root", GetPrefix(), GetFileNumber()); }
      UShort_t             GetFileNumber()              const { return fFileNumber; }
      const char          *GetFullName()                const {
        return Form("%s/%s", GetBaseURL(), GetFileName()); }
      const char          *GetPrefix()                  const { return fPrefix; }  
      const TTree         *GetTree(const char *tn)      const;
      TTree               *GetTree(const char *tn);
      void                 Print(Option_t *option="")   const;
      void                 SetAutoFill(const char *tn, Bool_t b);
      void                 SetBaseURL(const char *b)          { fBaseURL = b; }
      void                 SetCompressLevel(Int_t l)          { fCompressLevel = l; }
      void                 SetDefaultBrSize(Int_t s)          { fDefBrSize=s; }
      void                 SetDefaultSL(Int_t s)              { fDefSL=s;}
      void                 SetMaxSize(Long64_t s);
      void                 SetPrefix(const char *p)           { fPrefix = p; }
      void                 StoreObject(const TObject *obj);

    protected:
      TString              fBaseURL;            // base url for tree storage
      TString              fPrefix;             // prefix of file name 
      UShort_t             fFileNumber;         // current sequence number
      Int_t                fCompressLevel;      // compression level used for TFile
      Int_t                fDefBrSize;          // default buffer size for branches
      Int_t                fDefSL;              // default split level for branches
      Long64_t             fMaxSize;            // maximum file size for a file [Bytes]
      const Long64_t       fkMinFreeSpace;      // minimum free space required for closing file
      const Long64_t       fkMinAvgSize;        // minimum average entry size 
      Long64_t             fEvtObjNum;          // event object number offset (for TRef)
      Bool_t               fIsInit;             // true if OpenFile() was called
      Bool_t               fDoObjNumReset;      // true if obj. number resets automatically (def=0)
      TFile               *fFile;               // file being written
      TObjArray            fTrees;              // array of tree(s) being filled

      mithep::MyTree      *AddOrGetMyTree(const char *tn);
      mithep::MyTree      *GetMyTree(const char *tn);
      Bool_t               IsInit()                     const { return fIsInit; }
      Bool_t               IsFull()                     const;
      void                 OpenFile();
      void                 CloseFile();

    private:
      const char          *CName(void *obj)             const; 

    ClassDef(TreeWriter,0) // Tree writer class
  };
}
#endif
Revision:	1.10
Committed:	Wed Sep 10 03:33:27 2008 UTC (16 years, 7 months ago) by loizides
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_004
Changes since 1.9:	+3 -3 lines
Log Message:	Cleanup
#	User	Rev	Content
1	loizides	1.1	//--------------------------------------------------------------------------------------------------
2	loizides	1.10	// $Id: TreeWriter.h,v 1.9 2008/06/24 14:05:03 loizides Exp $
3	loizides	1.1	//
4			// TreeWriter
5			//
6	paus	1.5	// Class implementing a tree writing infrastructure. The class creates and maintaines
7	loizides	1.3	// a standard tree with name "tname" as given in the constructor. The tree is stored in
8			// a file or in multiple files if needed, where the file path name and maximum size can be
9	loizides	1.1	// specified. Using TreeWriter::AddBranch allows to add various branches to the tree.
10			// The user has to make sure that (s)he calls TreeWriter::BeginEvent and
11			// TreeWriter::EndEvent for every event before and after filling of the branch
12	loizides	1.3	// structures. Note that in addition to the standard tree one can create new
13			// tree(s) in the same file using the TreeWriter::AddBranchToTree member functions.
14	loizides	1.1	//
15			// Authors: C.Loizides
16			//--------------------------------------------------------------------------------------------------
17
18	loizides	1.10	#ifndef MITANA_DATAUTIL_TREEWRITER_H
19			#define MITANA_DATAUTIL_TREEWRITER_H
20	loizides	1.8
21			#include <TNamed.h>
22			#include <TString.h>
23			#include <TFile.h>
24			#include <TTree.h>
25			#include <TObjArray.h>
26
27	loizides	1.2	namespace mithep
28			{
29	loizides	1.3	class MyTree : public TTree
30			{
31			public:
32			MyTree(const char* name, const char* title, Int_t splitlevel = 99) :
33	paus	1.5	TTree(name,title, splitlevel), fAutoFill(1) {}
34			Bool_t GetAutoFill() const { return fAutoFill; }
35			void SetAutoFill(Bool_t b) { fAutoFill = b; }
36	loizides	1.3	protected:
37			Bool_t fAutoFill; //!=true then fill automatically in TreeWriter (def=1)
38			};
39	paus	1.5
40	loizides	1.1	class TreeWriter : public TNamed
41			{
42			public:
43			TreeWriter(const char *tname="MitTree", Bool_t doreset = kFALSE);
44			~TreeWriter();
45
46	loizides	1.3	void AddBranch(const char name, const char cname,
47			void *obj, Int_t bsize, Int_t level);
48	loizides	1.4	void AddBranch(const char name, void obj, Int_t bsize, Int_t level);
49	loizides	1.3	void AddBranch(const char name, const char cname,
50			void *obj, Int_t bsize);
51	loizides	1.4	void AddBranch(const char name, void obj, Int_t bsize);
52	loizides	1.3	void AddBranch(const char name, const char cname,
53			void *obj);
54	loizides	1.4	void AddBranch(const char name, void obj);
55	loizides	1.3	void AddBranchToTree(const char tname, const char name, const char *cname,
56			void *obj, Int_t bsize, Int_t level);
57	loizides	1.4	void AddBranchToTree(const char tname, const char name, void *obj,
58			Int_t bsize, Int_t level);
59	loizides	1.3	void AddBranchToTree(const char tname, const char name, const char *cname,
60			void *obj, Int_t bsize);
61	loizides	1.4	void AddBranchToTree(const char tname, const char name, void *obj,
62			Int_t bsize);
63	loizides	1.3	void AddBranchToTree(const char tname, const char name, const char *cname,
64			void *obj);
65	loizides	1.4	void AddBranchToTree(const char tname, const char name, void *obj);
66	loizides	1.1	Bool_t BeginEvent(Bool_t doreset=kFALSE);
67			Bool_t EndEvent(Bool_t doreset=kFALSE);
68	paus	1.5	const char *GetBaseURL() const {
69			return fBaseURL.IsNull() ? "." : fBaseURL; }
70	loizides	1.3	Int_t GetCompressLevel() const { return fCompressLevel; }
71			Long64_t GetEntries(const char *tn=0) const;
72	paus	1.5	Long64_t GetFileSize() const {
73			return fFile != 0 ? fFile->GetEND() : 0; }
74	loizides	1.3	const TFile *GetFile() const { return fFile; }
75	paus	1.5	const char *GetFileName() const {
76			return Form("%s_%03d.root", GetPrefix(), GetFileNumber()); }
77	loizides	1.3	UShort_t GetFileNumber() const { return fFileNumber; }
78	paus	1.5	const char *GetFullName() const {
79			return Form("%s/%s", GetBaseURL(), GetFileName()); }
80	loizides	1.3	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	loizides	1.7	void SetMaxSize(Long64_t s);
90	loizides	1.3	void SetPrefix(const char *p) { fPrefix = p; }
91	loizides	1.1	void StoreObject(const TObject *obj);
92
93			protected:
94	paus	1.5	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	loizides	1.3
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.9	ClassDef(TreeWriter,0) // Tree writer class
120	loizides	1.1	};
121	loizides	1.6	}
122			#endif