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.12 by loizides, Fri Jun 19 07:39:01 2009 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 <TFile.h>
7		#include <TH1D.h>
8		#include <TStopwatch.h>
#	Line 14 | Line 15 | ClassImp(mithep::AnaFwkMod)
15		//--------------------------------------------------------------------------------------------------
16		AnaFwkMod::AnaFwkMod(const char *name, const char *title) :
17		BaseMod(name,title),
18	+	fAllHeadTreeName(Names::gkAllEvtTreeName),
19	+	fAllHeadBrName(Names::gkAllEvtHeaderBrn),
20	+	fSkipNEvents(0),
21	+	fPrintScale(100),
22		fSWtotal(0),
23		fSWevent(0),
24	<	fAllHeaders(0,"AllEventHeaders"),
24	>	fAllHeaders(0,Names::gkSkimmedHeaders),
25		fAllHeadTree(0),
26		fAllEventHeader(0),
27		fReload(kFALSE),
28	<	fCurEnt(-2)
28	>	fCurEnt(-2),
29	>	fNEventsSkimmed(0),
30	>	fNEventsSkipped(0)
31		{
32		// Constructor.
33		}
#	Line 32 | Line 39 | void AnaFwkMod::BeginRun()
39		// depending on entry in RunInfo.
40
41		if (fReload) {
35	–
42		// reset to be (re-)loaded variables
43		fReload         = 0;
44		fAllHeadTree    = 0;
#	Line 44 | Line 50 | void AnaFwkMod::BeginRun()
50		if (!file)
51		return;
52
53	<	// get HLT tree
53	>	// get all event header tree
54		fAllHeadTree = dynamic_cast<TTree*>(file->Get(fAllHeadTreeName));
55		if (!fAllHeadTree) {
56	+	SendError(kWarning, "BeginRun",
57	+	"Cannot find tree '%s' in file '%s'",
58	+	fAllHeadTreeName.Data(),file->GetName());
59		return;
60		}
52	–	}
61
62	<	// get HLT trigger name branch
63	<	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
64	<	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
65	<	} else {
66	<	SendError(kWarning, "BeginRun",
67	<	"Can not find branch with name %s in tree %s",
68	<	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
69	<	return;
62	>	// get all event header branch
63	>	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
64	>	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
65	>	} else {
66	>	SendError(kWarning, "BeginRun",
67	>	"Cannot find branch '%s' in tree '%s'",
68	>	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
69	>	return;
70	>	}
71		}
72		}
73
#	Line 71 | Line 80 | void AnaFwkMod::CopyAllEventHeaders()
80
81		const EventHeader *curev = GetEventHeader();
82		if (!curev) {
83	<	SendError(kWarning, "CopyAllEventHeaders", "Can not obtain current event");
83	>	SendError(kWarning, "CopyAllEventHeaders", "Cannot obtain current event");
84		return;
85		}
86
87		if (fAllHeadTree) {
88	+	const Int_t nemax = fAllHeadTree->GetEntries();
89	+	if (fCurEnt == nemax) {
90	+	SendError(kAbortEvent, "CopyAllEventHeaders",
91	+	"End of all events tree reached (%d=%d)", fCurEnt, nemax);
92	+	return;
93	+	}
94		fAllHeadTree->GetEntry(fCurEnt++);
95	<	while(fAllEventHeader->IsRemoved()) {
95	>	while(fCurEnt<=nemax && fAllEventHeader->Skimmed()) {
96		EventHeader *eh = fAllHeaders.AddNew();
97		eh->SetRunNum(fAllEventHeader->RunNum());
98		eh->SetEvtNum(fAllEventHeader->EvtNum());
99		eh->SetLumiSec(fAllEventHeader->LumiSec());
100	<	eh->SetIsRemoved(fAllEventHeader->RunEntry());
101	<	eh->SetRunEntry(fAllEventHeader->IsRemoved());
100	>	eh->SetRunEntry(fAllEventHeader->RunEntry());
101	>	eh->SetSkimmed(fAllEventHeader->Skimmed());
102		fAllHeadTree->GetEntry(fCurEnt++);
103		}
104		if ((fAllEventHeader->RunNum()!=curev->RunNum())   ||
#	Line 91 | Line 106 | void AnaFwkMod::CopyAllEventHeaders()
106		(fAllEventHeader->LumiSec()!=curev->LumiSec()) ||
107		(fAllEventHeader->RunEntry()!=curev->RunEntry())) {
108		SendError(kWarning, "CopyAllEventHeaders",
109	<	"Event header information inconsistent: %d==%d, %d==%d, %d==%d, %d==%d",
109	>	"Event header information for entry %d inconsistent: "
110	>	"%d==%d, %d==%d, %d==%d, %d==%d",
111	>	fCurEnt,
112		fAllEventHeader->RunNum(),   curev->RunNum(),
113		fAllEventHeader->EvtNum(),   curev->EvtNum(),
114		fAllEventHeader->LumiSec(),  curev->LumiSec(),
115		fAllEventHeader->RunEntry(), curev->RunEntry());
116		return;
117		}
118	+
119	+	// read-ahead to check if more events are coming
120	+	if (fCurEnt<nemax) {
121	+	Int_t testEnt = fCurEnt;
122	+	fAllHeadTree->GetEntry(testEnt++);
123	+	while(testEnt<=nemax && fAllEventHeader->Skimmed())
124	+	fAllHeadTree->GetEntry(testEnt++);
125	+	if (testEnt==nemax+1) { // need to add remaining skimmed events
126	+	fAllHeadTree->GetEntry(fCurEnt++);
127	+	while(fCurEnt<=nemax) {
128	+	EventHeader *eh = fAllHeaders.AddNew();
129	+	eh->SetRunNum(fAllEventHeader->RunNum());
130	+	eh->SetEvtNum(fAllEventHeader->EvtNum());
131	+	eh->SetLumiSec(fAllEventHeader->LumiSec());
132	+	eh->SetRunEntry(fAllEventHeader->RunEntry());
133	+	eh->SetSkimmed(fAllEventHeader->Skimmed());
134	+	fAllHeadTree->GetEntry(fCurEnt++);
135	+	}
136	+	if (fCurEnt != nemax+1) {
137	+	SendError(kAbortEvent, "CopyAllEventHeaders",
138	+	"End of all events tree unexpectedly not reached (%d!=%d)", fCurEnt, nemax);
139	+	return;
140	+	}
141	+	}
142	+	}
143		}
144		}
145
#	Line 113 | Line 155 | Bool_t AnaFwkMod::Notify()
155		//--------------------------------------------------------------------------------------------------
156		void AnaFwkMod::Process()
157		{
158	<	// Do event counting and print out timing information.
158	>	// Do event skipping and counting and print out timing information.
159
160	<
119	<
120	<	// counting events
160	>	// counting processed events
161		IncNEventsProcessed();
162	+
163	+	// get skimmed event headers
164	+	CopyAllEventHeaders();
165	+	fNEventsSkimmed += fAllHeaders.GetEntries();
166	+
167	+	// check if events should be skipped
168	+	if (fNEventsSkipped<fSkipNEvents) {
169	+	++fNEventsSkipped;
170	+	MDB(kAnalysis, 3) {
171	+	Info("Process", "Skipping (aborting) %d out of %d first events.",
172	+	fNEventsSkipped, fSkipNEvents);
173	+	}
174	+	AbortEvent();
175	+	return;
176	+	}
177
178		// check if printout should be done
179		Bool_t doPrint = 0;
180
181		MDB(kAnalysis, 4) {
182	<	if (GetNEventsProcessed() % 1000  == 0)
182	>	if (GetNEventsProcessed() % (fPrintScale)  == 0)
183		doPrint = 1;
184		} else {
185		MDB(kAnalysis, 3) {
186	<	if (GetNEventsProcessed() % 10000  == 0)
186	>	if (GetNEventsProcessed() % (fPrintScale*10)  == 0)
187		doPrint = 1;
188		} else {
189		MDB(kAnalysis, 2) {
190	<	if (GetNEventsProcessed() % 50000  == 0)
190	>	if (GetNEventsProcessed() % (fPrintScale*100)  == 0)
191		doPrint = 1;
192		} else {
193		MDB(kAnalysis, 1) {
194	<	if (GetNEventsProcessed() % 250000 == 0)
194	>	if (GetNEventsProcessed() % (fPrintScale*1000) == 0)
195		doPrint = 1;
196		}
197		}
#	Line 145 | Line 200 | void AnaFwkMod::Process()
200
201		if (doPrint) {
202		fSWevent->Stop();
203	<	Info("Process", "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
203	>	Info("Process",
204	>	"Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
205		GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
206	<	fSWevent->RealTime()/GetNEventsProcessed(), fSWevent->CpuTime()/GetNEventsProcessed());
206	>	fSWevent->RealTime()/GetNEventsProcessed(),
207	>	fSWevent->CpuTime()/GetNEventsProcessed());
208		fSWevent->Start();
209		}
210		}
#	Line 159 | Line 216 | void AnaFwkMod::SlaveBegin()
216
217		fSWtotal = new TStopwatch;
218		fSWevent = new TStopwatch;
219	+	cout << fSkipNEvents << endl;
220
221		if (!PublishObj(&fAllHeaders)) {
222		SendError(kAbortAnalysis, "SlaveBegin",
#	Line 175 | Line 233 | void AnaFwkMod::SlaveTerminate()
233		RetractObj(fAllHeaders.GetName());
234
235		SaveNEventsProcessed();
236	+	TH1D *hDAllEvents = new TH1D("hDAllEvents","Sum of processed and skimmed events",1,-0.5,0.5);
237	+	hDAllEvents->Fill(0.0,fNEventsSkimmed+GetNEventsProcessed());
238	+	hDAllEvents->SetEntries(fNEventsSkimmed+GetNEventsProcessed());
239	+	AddOutput(hDAllEvents);
240	+
241	+	TH1D *hDSkippedEvents = new TH1D("hDSkippedEvents","Number of skipped events",1,-0.5,0.5);
242	+	hDSkippedEvents->Fill(0.0,fNEventsSkipped);
243	+	hDSkippedEvents->SetEntries(fNEventsSkipped);
244	+	AddOutput(hDSkippedEvents);
245
246		fSWtotal->Stop();
247		fSWevent->Stop();
248
249		MDB(kAnalysis, 1)
250	<	Info("SlaveTerminate", "Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
250	>	Info("SlaveTerminate",
251	>	"Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
252		GetNEventsProcessed(), fSWtotal->RealTime(), fSWtotal->CpuTime(),
253	<	fSWtotal->RealTime()/GetNEventsProcessed(), fSWtotal->CpuTime()/GetNEventsProcessed());
253	>	fSWtotal->RealTime()/GetNEventsProcessed(),
254	>	fSWtotal->CpuTime()/GetNEventsProcessed());
255
256		delete fSWtotal;
257		delete fSWevent;
258		}
190	–

Diff Legend

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