PamCAL/src/PamRootManager.cxx

#include "PamRootManager.h"

#include <TParticle.h>
#include <TObjArray.h>

ClassImp(PamRootManager)

PamRootManager* PamRootManager::fgInstance = 0;

//_____________________________________________________________________________
PamRootManager::PamRootManager(const char* projectName, FileMode fileMode)
  : TObject()
{
//
  if (fgInstance) {
    Fatal("PamRootManager", "Singleton instance already exists.");
    return;
  }  

  TString fileName(projectName);
  fileName += ".root";

  TString treeTitle(projectName);
  treeTitle += " tree";

  switch (fileMode) {
    case kRead:
      fFile = new TFile(fileName);
      fTree = (TTree*) fFile->Get(projectName);
      break;
      
    case kWrite:  
      fFile = new TFile(fileName, "recreate");
      fTree = new TTree(projectName, treeTitle);
      ;;  
  }

  fgInstance = this;
}

//_____________________________________________________________________________
PamRootManager::PamRootManager()
  : TObject(),
    fFile(0),
    fTree(0) 
{
//   

  if (fgInstance) {
    Fatal("PamRootManager", "Singleton instance already exists.");
    return;
  }  

  fgInstance = this;
}

//_____________________________________________________________________________
PamRootManager::~PamRootManager() 
{
//
  delete fFile;
  fgInstance = 0;
}

//
// static methods
//

//_____________________________________________________________________________
PamRootManager* PamRootManager::Instance()
{
// Returns singleton instance.
// ---

  return fgInstance;
}  

//
// public methods
//

//_____________________________________________________________________________
void  PamRootManager::Register(const char* name, const char* className, 
                                void* objAddress)
{
// Creates a branch of the given name and associates it with
// the given address.
// ---

  if (!fTree->GetBranch(name)) 
    fTree->Branch(name, className, objAddress, 32000, 99);
  else  
    fTree->GetBranch(name)->SetAddress(objAddress);
}

//_____________________________________________________________________________
void  PamRootManager::Fill()
{
// Fills the tree.
// ---

  fTree->Fill();
}  

//_____________________________________________________________________________
void PamRootManager:: WriteAll()
{
// Erites the tree in the file.
// ---

  fTree->Write();
}  

//_____________________________________________________________________________
void  PamRootManager::ReadEvent(Int_t i)
{
// Reads the event data for i-th event for all connected branches.
// ---

  fTree->GetEntry(i);
}
1	#include "PamRootManager.h"
2
3	#include <TParticle.h>
4	#include <TObjArray.h>
5
6	ClassImp(PamRootManager)
7
8	PamRootManager* PamRootManager::fgInstance = 0;
9
10	//_____________________________________________________________________________
11	PamRootManager::PamRootManager(const char* projectName, FileMode fileMode)
12	: TObject()
13	{
14	//
15	if (fgInstance) {
16	Fatal("PamRootManager", "Singleton instance already exists.");
17	return;
18	}
19
20	TString fileName(projectName);
21	fileName += ".root";
22
23	TString treeTitle(projectName);
24	treeTitle += " tree";
25
26	switch (fileMode) {
27	case kRead:
28	fFile = new TFile(fileName);
29	fTree = (TTree*) fFile->Get(projectName);
30	break;
31
32	case kWrite:
33	fFile = new TFile(fileName, "recreate");
34	fTree = new TTree(projectName, treeTitle);
35	;;
36	}
37
38	fgInstance = this;
39	}
40
41	//_____________________________________________________________________________
42	PamRootManager::PamRootManager()
43	: TObject(),
44	fFile(0),
45	fTree(0)
46	{
47	//
48
49	if (fgInstance) {
50	Fatal("PamRootManager", "Singleton instance already exists.");
51	return;
52	}
53
54	fgInstance = this;
55	}
56
57	//_____________________________________________________________________________
58	PamRootManager::~PamRootManager()
59	{
60	//
61	delete fFile;
62	fgInstance = 0;
63	}
64
65	//
66	// static methods
67	//
68
69	//_____________________________________________________________________________
70	PamRootManager* PamRootManager::Instance()
71	{
72	// Returns singleton instance.
73	// ---
74
75	return fgInstance;
76	}
77
78	//
79	// public methods
80	//
81
82	//_____________________________________________________________________________
83	void PamRootManager::Register(const char* name, const char* className,
84	void* objAddress)
85	{
86	// Creates a branch of the given name and associates it with
87	// the given address.
88	// ---
89
90	if (!fTree->GetBranch(name))
91	fTree->Branch(name, className, objAddress, 32000, 99);
92	else
93	fTree->GetBranch(name)->SetAddress(objAddress);
94	}
95
96	//_____________________________________________________________________________
97	void PamRootManager::Fill()
98	{
99	// Fills the tree.
100	// ---
101
102	fTree->Fill();
103	}
104
105	//_____________________________________________________________________________
106	void PamRootManager:: WriteAll()
107	{
108	// Erites the tree in the file.
109	// ---
110
111	fTree->Write();
112	}
113
114	//_____________________________________________________________________________
115	void PamRootManager::ReadEvent(Int_t i)
116	{
117	// Reads the event data for i-th event for all connected branches.
118	// ---
119
120	fTree->GetEntry(i);
121	}