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.7 by loizides, Thu Mar 12 18:25:35 2009 UTC vs.
Revision 1.24 by bendavid, Tue Aug 27 14:13:15 2013 UTC

#	Line 3 | Line 3
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 17 | Line 19 | AnaFwkMod::AnaFwkMod(const char *name, c
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,Names::gkSkimmedHeaders),
#	Line 24 | Line 28 | AnaFwkMod::AnaFwkMod(const char *name, c
28		fAllEventHeader(0),
29		fReload(kFALSE),
30		fCurEnt(-2),
31	<	fNEventsSkimmed(0)
31	>	fNEventsSkimmed(0),
32	>	fNEventsSkipped(0),
33	>	fPileupInfoName("PileupInfo"),
34	>	fDoPUInfo(kFALSE),
35	>	hNPU(0),
36	>	hNPU50ns(0),
37	>	hNPUTrue(0),
38	>	fMCEventInfo(0),
39	>	fMCEventInfoName(Names::gkMCEvtInfoBrn),
40	>	hDTotalMCWeight(0)
41		{
42		// Constructor.
43		}
#	Line 47 | Line 60 | void AnaFwkMod::BeginRun()
60		if (!file)
61		return;
62
63	<	// get HLT tree
63	>	// get all event header tree
64		fAllHeadTree = dynamic_cast<TTree*>(file->Get(fAllHeadTreeName));
65		if (!fAllHeadTree) {
66		SendError(kWarning, "BeginRun",
67	<	"Can not find tree '%s' in file '%s'",
67	>	"Cannot find tree '%s' in file '%s'",
68		fAllHeadTreeName.Data(),file->GetName());
69		return;
70		}
58	–	}
71
72	<	// get HLT trigger name branch
73	<	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
74	<	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
75	<	} else {
76	<	SendError(kWarning, "BeginRun",
77	<	"Can not find branch '%s' in tree '%s'",
78	<	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
79	<	return;
72	>	// get all event header branch
73	>	if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
74	>	fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
75	>	} else {
76	>	SendError(kWarning, "BeginRun",
77	>	"Cannot find branch '%s' in tree '%s'",
78	>	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
79	>	return;
80	>	}
81		}
82		}
83
#	Line 77 | Line 90 | void AnaFwkMod::CopyAllEventHeaders()
90
91		const EventHeader *curev = GetEventHeader();
92		if (!curev) {
93	<	SendError(kWarning, "CopyAllEventHeaders", "Can not obtain current event");
93	>	SendError(kWarning, "CopyAllEventHeaders", "Cannot obtain current event");
94		return;
95		}
96
#	Line 89 | Line 102 | void AnaFwkMod::CopyAllEventHeaders()
102		return;
103		}
104		fAllHeadTree->GetEntry(fCurEnt++);
105	<	while(fCurEnt<=nemax && fAllEventHeader->Skimmed()) {
105	>	while (fCurEnt <= nemax && fAllEventHeader->Skimmed()) {
106		EventHeader *eh = fAllHeaders.AddNew();
107	<	eh->SetRunNum(fAllEventHeader->RunNum());
108	<	eh->SetEvtNum(fAllEventHeader->EvtNum());
109	<	eh->SetLumiSec(fAllEventHeader->LumiSec());
107	>	eh->SetRunNum  (fAllEventHeader->RunNum());
108	>	eh->SetEvtNum  (fAllEventHeader->EvtNum());
109	>	eh->SetLumiSec (fAllEventHeader->LumiSec());
110		eh->SetRunEntry(fAllEventHeader->RunEntry());
111	<	eh->SetSkimmed(fAllEventHeader->Skimmed());
111	>	eh->SetSkimmed (fAllEventHeader->Skimmed());
112		fAllHeadTree->GetEntry(fCurEnt++);
113		}
114	<	if ((fAllEventHeader->RunNum()!=curev->RunNum())   ||
115	<	(fAllEventHeader->EvtNum()!=curev->EvtNum())   ||
116	<	(fAllEventHeader->LumiSec()!=curev->LumiSec()) ||
117	<	(fAllEventHeader->RunEntry()!=curev->RunEntry())) {
114	>	if ((fAllEventHeader->RunNum()   != curev->RunNum())   ||
115	>	(fAllEventHeader->EvtNum()   != curev->EvtNum())   ||
116	>	(fAllEventHeader->LumiSec()  != curev->LumiSec())  ||
117	>	(fAllEventHeader->RunEntry() != curev->RunEntry())) {
118		SendError(kWarning, "CopyAllEventHeaders",
119		"Event header information for entry %d inconsistent: "
120		"%d==%d, %d==%d, %d==%d, %d==%d",
#	Line 117 | Line 130 | void AnaFwkMod::CopyAllEventHeaders()
130		if (fCurEnt<nemax) {
131		Int_t testEnt = fCurEnt;
132		fAllHeadTree->GetEntry(testEnt++);
133	<	while(testEnt<=nemax && fAllEventHeader->Skimmed())
133	>	while (testEnt<=nemax && fAllEventHeader->Skimmed())
134		fAllHeadTree->GetEntry(testEnt++);
135		if (testEnt==nemax+1) { // need to add remaining skimmed events
136		fAllHeadTree->GetEntry(fCurEnt++);
137		while(fCurEnt<=nemax) {
138		EventHeader *eh = fAllHeaders.AddNew();
139	<	eh->SetRunNum(fAllEventHeader->RunNum());
140	<	eh->SetEvtNum(fAllEventHeader->EvtNum());
141	<	eh->SetLumiSec(fAllEventHeader->LumiSec());
139	>	eh->SetRunNum  (fAllEventHeader->RunNum());
140	>	eh->SetEvtNum  (fAllEventHeader->EvtNum());
141	>	eh->SetLumiSec (fAllEventHeader->LumiSec());
142		eh->SetRunEntry(fAllEventHeader->RunEntry());
143	<	eh->SetSkimmed(fAllEventHeader->Skimmed());
143	>	eh->SetSkimmed (fAllEventHeader->Skimmed());
144		fAllHeadTree->GetEntry(fCurEnt++);
145		}
146		if (fCurEnt != nemax+1) {
#	Line 152 | Line 165 | Bool_t AnaFwkMod::Notify()
165		//--------------------------------------------------------------------------------------------------
166		void AnaFwkMod::Process()
167		{
168	<	// Do event counting and print out timing information.
168	>	// Do event skipping and counting and print out timing information.
169
170	+	// counting processed events
171	+	IncNEventsProcessed();
172
173		// get skimmed event headers
174		CopyAllEventHeaders();
175		fNEventsSkimmed += fAllHeaders.GetEntries();
176
177	<	// counting events
178	<	IncNEventsProcessed();
177	>	// check if events should be skipped
178	>	if (fNEventsSkipped<fSkipNEvents) {
179	>	++fNEventsSkipped;
180	>	MDB(kAnalysis, 3) {
181	>	Info("Process", "Skipping (aborting) %d out of %lld first events.",
182	>	fNEventsSkipped, fSkipNEvents);
183	>	}
184	>	AbortEvent();
185	>	return;
186	>	}
187
188		// check if printout should be done
189	<	Bool_t doPrint = 0;
189	>	Bool_t doPrint    = 0;
190	>	UInt_t nProcessed = fPrintScale;
191
192		MDB(kAnalysis, 4) {
193	<	if (GetNEventsProcessed() % 1000  == 0)
193	>	if (GetNEventsProcessed() % (fPrintScale)  == 0)
194		doPrint = 1;
195		} else {
196		MDB(kAnalysis, 3) {
197	<	if (GetNEventsProcessed() % 10000  == 0)
197	>	if (GetNEventsProcessed() % (fPrintScale*10)  == 0)
198		doPrint = 1;
199	+	nProcessed = fPrintScale*10;
200		} else {
201		MDB(kAnalysis, 2) {
202	<	if (GetNEventsProcessed() % 50000  == 0)
202	>	if (GetNEventsProcessed() % (fPrintScale*100)  == 0)
203		doPrint = 1;
204	+	nProcessed = fPrintScale*100;
205		} else {
206		MDB(kAnalysis, 1) {
207	<	if (GetNEventsProcessed() % 250000 == 0)
207	>	if (GetNEventsProcessed() % (fPrintScale*1000) == 0)
208		doPrint = 1;
209	+	nProcessed = fPrintScale*1000;
210		}
211		}
212		}
213		}
214
215	+	if (GetEventHeader()->IsMC()) {
216	+	LoadBranch(fPileupInfoName);
217	+	Double_t npu[4] = {0.,0.,0.,0.};
218	+	for (UInt_t i=0; i<fPileupInfo->GetEntries(); ++i) {
219	+	const PileupInfo *puinfo = fPileupInfo->At(i);
220	+	if (puinfo->GetBunchCrossing()==0) npu[0]= puinfo->GetPU_NumInteractions();
221	+	else if (puinfo->GetBunchCrossing()==-1) npu[1] = puinfo->GetPU_NumInteractions();
222	+	else if (puinfo->GetBunchCrossing()==1) npu[2] = puinfo->GetPU_NumInteractions();
223	+	if (puinfo->GetBunchCrossing()==0) npu[3]= puinfo->GetPU_NumMean();
224	+	}
225	+	hNPU->Fill(npu[0]);
226	+	hNPU50ns->Fill(npu[0],npu[1],npu[2]);
227	+	hNPUTrue->Fill(npu[3]);
228	+
229	+	LoadBranch(fMCEventInfoName);
230	+	hDTotalMCWeight->Fill(0., fMCEventInfo->Weight());
231	+
232	+	}
233	+
234	+
235		if (doPrint) {
236		fSWevent->Stop();
237	<	Info("Process", "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
237	>	Info("Process",
238	>	"Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
239		GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
240	<	fSWevent->RealTime()/GetNEventsProcessed(), fSWevent->CpuTime()/GetNEventsProcessed());
240	>	fSWevent->RealTime()/nProcessed,
241	>	fSWevent->CpuTime()/nProcessed);
242		fSWevent->Start();
243	<	}
243	>	}
244	>
245	>
246		}
247
248		//--------------------------------------------------------------------------------------------------
#	Line 207 | Line 258 | void AnaFwkMod::SlaveBegin()
258		"Could not publish all event headers with name %s.", fAllHeaders.GetName());
259		return;
260		}
261	+
262	+	ReqBranch(fPileupInfoName, fPileupInfo);
263	+	ReqBranch(fMCEventInfoName, fMCEventInfo);
264	+
265	+	hNPU = new TH1D("hNPU", "hNPU", 201, -0.5, 200.5);
266	+	AddOutput(hNPU);
267	+
268	+	hNPU50ns = new TH3D("hNPU50ns", "hNPU50ns", 201, -0.5, 200.5, 201, -0.5, 200.5, 201, -0.5, 200.5);
269	+	AddOutput(hNPU50ns);
270	+
271	+	hNPUTrue = new TH1D("hNPUTrue", "hNPUTrue", 2000, 0.0, 200.0);
272	+	AddOutput(hNPUTrue);
273	+
274	+	hDTotalMCWeight = new TH1D("hDTotalMCWeight","hDTotalMCWeight",1,-0.5,0.5);
275	+	AddOutput(hDTotalMCWeight);
276	+
277		}
278
279		//--------------------------------------------------------------------------------------------------
#	Line 222 | Line 289 | void AnaFwkMod::SlaveTerminate()
289		hDAllEvents->SetEntries(fNEventsSkimmed+GetNEventsProcessed());
290		AddOutput(hDAllEvents);
291
292	+	TH1D *hDSkippedEvents = new TH1D("hDSkippedEvents","Number of skipped events",1,-0.5,0.5);
293	+	hDSkippedEvents->Fill(0.0,fNEventsSkipped);
294	+	hDSkippedEvents->SetEntries(fNEventsSkipped);
295	+	AddOutput(hDSkippedEvents);
296	+
297		fSWtotal->Stop();
298		fSWevent->Stop();
299
300		MDB(kAnalysis, 1)
301	<	Info("SlaveTerminate", "Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
301	>	Info("SlaveTerminate",
302	>	"Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
303		GetNEventsProcessed(), fSWtotal->RealTime(), fSWtotal->CpuTime(),
304	<	fSWtotal->RealTime()/GetNEventsProcessed(), fSWtotal->CpuTime()/GetNEventsProcessed());
304	>	fSWtotal->RealTime()/GetNEventsProcessed(),
305	>	fSWtotal->CpuTime()/GetNEventsProcessed());
306
307		delete fSWtotal;
308		delete fSWevent;

Diff Legend

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