ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitAna/TreeMod/src/AnaFwkMod.cc

Comparing UserCode/MitAna/TreeMod/src/AnaFwkMod.cc (file contents):
Revision 1.4 by loizides, Mon Mar 9 17:07:05 2009 UTC vs.
Revision 1.18 by bendavid, Wed May 30 16:23:54 2012 UTC

#	Line 2 | Line 2
2
3		#include "MitAna/TreeMod/interface/AnaFwkMod.h"
4		#include "MitAna/DataUtil/interface/Debug.h"
5	+	#include "MitAna/DataTree/interface/Names.h"
6	+	#include "MitAna/DataTree/interface/PileupInfo.h"
7		#include <TFile.h>
8		#include <TH1D.h>
9	+	#include <TH3D.h>
10		#include <TStopwatch.h>
11		#include <TTree.h>
12
#	Line 14 | Line 17 | ClassImp(mithep::AnaFwkMod)
17		//--------------------------------------------------------------------------------------------------
18		AnaFwkMod::AnaFwkMod(const char *name, const char *title) :
19		BaseMod(name,title),
20	+	fAllHeadTreeName(Names::gkAllEvtTreeName),
21	+	fAllHeadBrName(Names::gkAllEvtHeaderBrn),
22	+	fSkipNEvents(0),
23	+	fPrintScale(1),
24		fSWtotal(0),
25		fSWevent(0),
26	<	fAllHeaders(0,"AllEventHeaders"),
26	>	fAllHeaders(0,Names::gkSkimmedHeaders),
27		fAllHeadTree(0),
28		fAllEventHeader(0),
29		fReload(kFALSE),
30	<	fCurEnt(-2)
30	>	fCurEnt(-2),
31	>	fNEventsSkimmed(0),
32	>	fNEventsSkipped(0),
33	>	fPileupInfoName("PileupInfo"),
34	>	fDoPUInfo(kFALSE),
35	>	hNPU(0),
36	>	hNPU50ns(0)
37		{
38		// Constructor.
39		}
#	Line 32 | Line 45 | void AnaFwkMod::BeginRun()
45		// depending on entry in RunInfo.
46
47		if (fReload) {
35	–
48		// reset to be (re-)loaded variables
49		fReload         = 0;
50		fAllHeadTree    = 0;
#	Line 44 | Line 56 | void AnaFwkMod::BeginRun()
56		if (!file)
57		return;
58
59	<	// get HLT tree
59	>	// get all event header tree
60		fAllHeadTree = dynamic_cast<TTree*>(file->Get(fAllHeadTreeName));
61		if (!fAllHeadTree) {
62	+	SendError(kWarning, "BeginRun",
63	+	"Cannot find tree '%s' in file '%s'",
64	+	fAllHeadTreeName.Data(),file->GetName());
65		return;
66		}
52	–	}
67
68	<	// get HLT trigger name branch
69	<	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
70	<	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
71	<	} else {
72	<	SendError(kWarning, "BeginRun",
73	<	"Can not find branch with name %s in tree %s",
74	<	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
75	<	return;
68	>	// get all event header branch
69	>	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
70	>	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
71	>	} else {
72	>	SendError(kWarning, "BeginRun",
73	>	"Cannot find branch '%s' in tree '%s'",
74	>	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
75	>	return;
76	>	}
77		}
78		}
79
#	Line 71 | Line 86 | void AnaFwkMod::CopyAllEventHeaders()
86
87		const EventHeader *curev = GetEventHeader();
88		if (!curev) {
89	<	SendError(kWarning, "CopyAllEventHeaders", "Can not obtain current event");
89	>	SendError(kWarning, "CopyAllEventHeaders", "Cannot obtain current event");
90		return;
91		}
92
93		if (fAllHeadTree) {
94	+	const Int_t nemax = fAllHeadTree->GetEntries();
95	+	if (fCurEnt == nemax) {
96	+	SendError(kAbortEvent, "CopyAllEventHeaders",
97	+	"End of all events tree reached (%d=%d)", fCurEnt, nemax);
98	+	return;
99	+	}
100		fAllHeadTree->GetEntry(fCurEnt++);
101	<	while(fAllEventHeader->IsRemoved()) {
101	>	while (fCurEnt <= nemax && fAllEventHeader->Skimmed()) {
102		EventHeader *eh = fAllHeaders.AddNew();
103	<	eh->SetRunNum(fAllEventHeader->RunNum());
104	<	eh->SetEvtNum(fAllEventHeader->EvtNum());
105	<	eh->SetLumiSec(fAllEventHeader->LumiSec());
106	<	eh->SetIsRemoved(fAllEventHeader->RunEntry());
107	<	eh->SetRunEntry(fAllEventHeader->IsRemoved());
103	>	eh->SetRunNum  (fAllEventHeader->RunNum());
104	>	eh->SetEvtNum  (fAllEventHeader->EvtNum());
105	>	eh->SetLumiSec (fAllEventHeader->LumiSec());
106	>	eh->SetRunEntry(fAllEventHeader->RunEntry());
107	>	eh->SetSkimmed (fAllEventHeader->Skimmed());
108		fAllHeadTree->GetEntry(fCurEnt++);
109		}
110	<	if ((fAllEventHeader->RunNum()!=curev->RunNum())   ||
111	<	(fAllEventHeader->EvtNum()!=curev->EvtNum())   ||
112	<	(fAllEventHeader->LumiSec()!=curev->LumiSec()) ||
113	<	(fAllEventHeader->RunEntry()!=curev->RunEntry())) {
110	>	if ((fAllEventHeader->RunNum()   != curev->RunNum())   ||
111	>	(fAllEventHeader->EvtNum()   != curev->EvtNum())   ||
112	>	(fAllEventHeader->LumiSec()  != curev->LumiSec())  ||
113	>	(fAllEventHeader->RunEntry() != curev->RunEntry())) {
114		SendError(kWarning, "CopyAllEventHeaders",
115	<	"Event header information inconsistent: %d==%d, %d==%d, %d==%d, %d==%d",
115	>	"Event header information for entry %d inconsistent: "
116	>	"%d==%d, %d==%d, %d==%d, %d==%d",
117	>	fCurEnt,
118		fAllEventHeader->RunNum(),   curev->RunNum(),
119		fAllEventHeader->EvtNum(),   curev->EvtNum(),
120		fAllEventHeader->LumiSec(),  curev->LumiSec(),
121		fAllEventHeader->RunEntry(), curev->RunEntry());
122		return;
123		}
124	+
125	+	// read-ahead to check if more events are coming
126	+	if (fCurEnt<nemax) {
127	+	Int_t testEnt = fCurEnt;
128	+	fAllHeadTree->GetEntry(testEnt++);
129	+	while (testEnt<=nemax && fAllEventHeader->Skimmed())
130	+	fAllHeadTree->GetEntry(testEnt++);
131	+	if (testEnt==nemax+1) { // need to add remaining skimmed events
132	+	fAllHeadTree->GetEntry(fCurEnt++);
133	+	while(fCurEnt<=nemax) {
134	+	EventHeader *eh = fAllHeaders.AddNew();
135	+	eh->SetRunNum  (fAllEventHeader->RunNum());
136	+	eh->SetEvtNum  (fAllEventHeader->EvtNum());
137	+	eh->SetLumiSec (fAllEventHeader->LumiSec());
138	+	eh->SetRunEntry(fAllEventHeader->RunEntry());
139	+	eh->SetSkimmed (fAllEventHeader->Skimmed());
140	+	fAllHeadTree->GetEntry(fCurEnt++);
141	+	}
142	+	if (fCurEnt != nemax+1) {
143	+	SendError(kAbortEvent, "CopyAllEventHeaders",
144	+	"End of all events tree unexpectedly not reached (%d!=%d)", fCurEnt, nemax);
145	+	return;
146	+	}
147	+	}
148	+	}
149		}
150		}
151
#	Line 113 | Line 161 | Bool_t AnaFwkMod::Notify()
161		//--------------------------------------------------------------------------------------------------
162		void AnaFwkMod::Process()
163		{
164	<	// Do event counting and print out timing information.
164	>	// Do event skipping and counting and print out timing information.
165
166	<
119	<
120	<	// counting events
166	>	// counting processed events
167		IncNEventsProcessed();
168	+
169	+	// get skimmed event headers
170	+	CopyAllEventHeaders();
171	+	fNEventsSkimmed += fAllHeaders.GetEntries();
172	+
173	+	// check if events should be skipped
174	+	if (fNEventsSkipped<fSkipNEvents) {
175	+	++fNEventsSkipped;
176	+	MDB(kAnalysis, 3) {
177	+	Info("Process", "Skipping (aborting) %d out of %lld first events.",
178	+	fNEventsSkipped, fSkipNEvents);
179	+	}
180	+	AbortEvent();
181	+	return;
182	+	}
183
184		// check if printout should be done
185	<	Bool_t doPrint = 0;
185	>	Bool_t doPrint    = 0;
186	>	UInt_t nProcessed = fPrintScale;
187
188		MDB(kAnalysis, 4) {
189	<	if (GetNEventsProcessed() % 1000  == 0)
189	>	if (GetNEventsProcessed() % (fPrintScale)  == 0)
190		doPrint = 1;
191		} else {
192		MDB(kAnalysis, 3) {
193	<	if (GetNEventsProcessed() % 10000  == 0)
193	>	if (GetNEventsProcessed() % (fPrintScale*10)  == 0)
194		doPrint = 1;
195	+	nProcessed = fPrintScale*10;
196		} else {
197		MDB(kAnalysis, 2) {
198	<	if (GetNEventsProcessed() % 50000  == 0)
198	>	if (GetNEventsProcessed() % (fPrintScale*100)  == 0)
199		doPrint = 1;
200	+	nProcessed = fPrintScale*100;
201		} else {
202		MDB(kAnalysis, 1) {
203	<	if (GetNEventsProcessed() % 250000 == 0)
203	>	if (GetNEventsProcessed() % (fPrintScale*1000) == 0)
204		doPrint = 1;
205	+	nProcessed = fPrintScale*1000;
206		}
207		}
208		}
209		}
210
211	+	if (GetEventHeader()->IsMC()) {
212	+	LoadBranch(fPileupInfoName);
213	+	Int_t npu[3] = {0,0,0};
214	+	for (UInt_t i=0; i<fPileupInfo->GetEntries(); ++i) {
215	+	const PileupInfo *puinfo = fPileupInfo->At(i);
216	+	if (puinfo->GetBunchCrossing()==0) npu[0]= puinfo->GetPU_NumInteractions();
217	+	else if (puinfo->GetBunchCrossing()==-1) npu[1] = puinfo->GetPU_NumInteractions();
218	+	else if (puinfo->GetBunchCrossing()==1) npu[2] = puinfo->GetPU_NumInteractions();
219	+	}
220	+
221	+	hNPU->Fill(npu[0]);
222	+	hNPU50ns->Fill(npu[0],npu[1],npu[2]);
223	+	}
224	+
225	+
226		if (doPrint) {
227		fSWevent->Stop();
228	<	Info("Process", "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
228	>	Info("Process",
229	>	"Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
230		GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
231	<	fSWevent->RealTime()/GetNEventsProcessed(), fSWevent->CpuTime()/GetNEventsProcessed());
231	>	fSWevent->RealTime()/nProcessed,
232	>	fSWevent->CpuTime()/nProcessed);
233		fSWevent->Start();
234	<	}
234	>	}
235	>
236	>
237		}
238
239		//--------------------------------------------------------------------------------------------------
#	Line 165 | Line 249 | void AnaFwkMod::SlaveBegin()
249		"Could not publish all event headers with name %s.", fAllHeaders.GetName());
250		return;
251		}
252	+
253	+	ReqBranch(fPileupInfoName, fPileupInfo);
254	+
255	+	hNPU = new TH1D("hNPU", "hNPU", 201, -0.5, 200.5);
256	+	AddOutput(hNPU);
257	+
258	+	hNPU50ns = new TH3D("hNPU50ns", "hNPU50ns", 201, -0.5, 200.5, 201, -0.5, 200.5, 201, -0.5, 200.5);
259	+	AddOutput(hNPU50ns);
260	+
261		}
262
263		//--------------------------------------------------------------------------------------------------
#	Line 175 | Line 268 | void AnaFwkMod::SlaveTerminate()
268		RetractObj(fAllHeaders.GetName());
269
270		SaveNEventsProcessed();
271	+	TH1D *hDAllEvents = new TH1D("hDAllEvents","Sum of processed and skimmed events",1,-0.5,0.5);
272	+	hDAllEvents->Fill(0.0,fNEventsSkimmed+GetNEventsProcessed());
273	+	hDAllEvents->SetEntries(fNEventsSkimmed+GetNEventsProcessed());
274	+	AddOutput(hDAllEvents);
275	+
276	+	TH1D *hDSkippedEvents = new TH1D("hDSkippedEvents","Number of skipped events",1,-0.5,0.5);
277	+	hDSkippedEvents->Fill(0.0,fNEventsSkipped);
278	+	hDSkippedEvents->SetEntries(fNEventsSkipped);
279	+	AddOutput(hDSkippedEvents);
280
281		fSWtotal->Stop();
282		fSWevent->Stop();
283
284		MDB(kAnalysis, 1)
285	<	Info("SlaveTerminate", "Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
285	>	Info("SlaveTerminate",
286	>	"Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
287		GetNEventsProcessed(), fSWtotal->RealTime(), fSWtotal->CpuTime(),
288	<	fSWtotal->RealTime()/GetNEventsProcessed(), fSWtotal->CpuTime()/GetNEventsProcessed());
288	>	fSWtotal->RealTime()/GetNEventsProcessed(),
289	>	fSWtotal->CpuTime()/GetNEventsProcessed());
290
291		delete fSWtotal;
292		delete fSWevent;
293		}
190	–

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines