TreeMod/src/AnaFwkMod.cc

// $Id: AnaFwkMod.cc,v 1.6 2009/03/12 15:37:53 loizides Exp $

#include "MitAna/TreeMod/interface/AnaFwkMod.h"
#include "MitAna/DataUtil/interface/Debug.h"
#include "MitAna/DataTree/interface/Names.h"
#include <TFile.h>
#include <TH1D.h>
#include <TStopwatch.h>
#include <TTree.h>

using namespace mithep;

ClassImp(mithep::AnaFwkMod)

//--------------------------------------------------------------------------------------------------
AnaFwkMod::AnaFwkMod(const char *name, const char *title) : 
  BaseMod(name,title),
  fAllHeadTreeName(Names::gkAllEvtTreeName),
  fAllHeadBrName(Names::gkAllEvtHeaderBrn),
  fSWtotal(0),
  fSWevent(0),
  fAllHeaders(0,Names::gkSkimmedHeaders),
  fAllHeadTree(0),
  fAllEventHeader(0),
  fReload(kFALSE),
  fCurEnt(-2),
  fNEventsSkimmed(0)
{
  // Constructor.
}

//--------------------------------------------------------------------------------------------------
void AnaFwkMod::BeginRun()
{
  // Get HLT tree and set branches if new file was opened. Read next entry in HLT key
  // depending on entry in RunInfo.

  if (fReload) {
    // reset to be (re-)loaded variables
    fReload         = 0;
    fAllHeadTree    = 0;
    fAllEventHeader = 0;
    fCurEnt         = 0;

    // get current file
    TFile *file = GetCurrentFile();
    if (!file) 
      return;

    // get HLT tree
    fAllHeadTree = dynamic_cast<TTree*>(file->Get(fAllHeadTreeName));
    if (!fAllHeadTree) {
      SendError(kWarning, "BeginRun",
                "Can not find tree '%s' in file '%s'", 
                fAllHeadTreeName.Data(),file->GetName());
      return;
    }
  }

  // get HLT trigger name branch
  if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
    fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
  } else {
    SendError(kWarning, "BeginRun",
              "Can not find branch '%s' in tree '%s'", 
              fAllHeadBrName.Data(), fAllHeadTreeName.Data());
    return;
  }
}

//--------------------------------------------------------------------------------------------------
void AnaFwkMod::CopyAllEventHeaders()
{
  // Deal with the headers from all events if needed.

  fAllHeaders.Reset();

  const EventHeader *curev = GetEventHeader();
  if (!curev) {
    SendError(kWarning, "CopyAllEventHeaders", "Can not obtain current event");
    return;
  }

  if (fAllHeadTree) {
    const Int_t nemax = fAllHeadTree->GetEntries();
    if (fCurEnt == nemax) {
      SendError(kAbortEvent, "CopyAllEventHeaders", 
                "End of all events tree reached (%d=%d)", fCurEnt, nemax);
      return;
    }
    fAllHeadTree->GetEntry(fCurEnt++);
    while(fCurEnt<=nemax && fAllEventHeader->Skimmed()) {
      EventHeader *eh = fAllHeaders.AddNew();
      eh->SetRunNum(fAllEventHeader->RunNum());
      eh->SetEvtNum(fAllEventHeader->EvtNum());
      eh->SetLumiSec(fAllEventHeader->LumiSec());
      eh->SetRunEntry(fAllEventHeader->RunEntry());
      eh->SetSkimmed(fAllEventHeader->Skimmed());
      fAllHeadTree->GetEntry(fCurEnt++);
    }
    if ((fAllEventHeader->RunNum()!=curev->RunNum())   ||
        (fAllEventHeader->EvtNum()!=curev->EvtNum())   ||
        (fAllEventHeader->LumiSec()!=curev->LumiSec()) ||
        (fAllEventHeader->RunEntry()!=curev->RunEntry())) {
      SendError(kWarning, "CopyAllEventHeaders", 
                "Event header information for entry %d inconsistent: "
                "%d==%d, %d==%d, %d==%d, %d==%d",
                fCurEnt,
                fAllEventHeader->RunNum(),   curev->RunNum(), 
                fAllEventHeader->EvtNum(),   curev->EvtNum(),
                fAllEventHeader->LumiSec(),  curev->LumiSec(),
                fAllEventHeader->RunEntry(), curev->RunEntry());
      return;
    }

    // read-ahead to check if more events are coming
    if (fCurEnt<nemax) { 
      Int_t testEnt = fCurEnt;
      fAllHeadTree->GetEntry(testEnt++);
      while(testEnt<=nemax && fAllEventHeader->Skimmed())
        fAllHeadTree->GetEntry(testEnt++);
      if (testEnt==nemax+1) { // need to add remaining skimmed events
        fAllHeadTree->GetEntry(fCurEnt++);
        while(fCurEnt<=nemax) {
          EventHeader *eh = fAllHeaders.AddNew();
          eh->SetRunNum(fAllEventHeader->RunNum());
          eh->SetEvtNum(fAllEventHeader->EvtNum());
          eh->SetLumiSec(fAllEventHeader->LumiSec());
          eh->SetRunEntry(fAllEventHeader->RunEntry());
          eh->SetSkimmed(fAllEventHeader->Skimmed());
          fAllHeadTree->GetEntry(fCurEnt++);
        }
        if (fCurEnt != nemax+1) {
          SendError(kAbortEvent, "CopyAllEventHeaders", 
                    "End of all events tree unexpectedly not reached (%d!=%d)", fCurEnt, nemax);
          return;
        }
      }
    }
  }
}

//--------------------------------------------------------------------------------------------------
Bool_t AnaFwkMod::Notify()
{
  // Make sure to get the new "AllEvents" tree when the file changes.

  fReload = kTRUE;
  return kTRUE;
}

//--------------------------------------------------------------------------------------------------
void AnaFwkMod::Process()
{
  // Do event counting and print out timing information.

  
  // get skimmed event headers
  CopyAllEventHeaders();
  fNEventsSkimmed += fAllHeaders.GetEntries();

  // counting events
  IncNEventsProcessed();

  // check if printout should be done
  Bool_t doPrint = 0;

  MDB(kAnalysis, 4) {
    if (GetNEventsProcessed() % 1000  == 0) 
      doPrint = 1;
  } else {
    MDB(kAnalysis, 3) {
      if (GetNEventsProcessed() % 10000  == 0) 
        doPrint = 1;
    } else {
      MDB(kAnalysis, 2) {
        if (GetNEventsProcessed() % 50000  == 0) 
          doPrint = 1;
      } else {
        MDB(kAnalysis, 1) {
          if (GetNEventsProcessed() % 250000 == 0) 
            doPrint = 1;
        } 
      }
    }
  }

  if (doPrint) {
    fSWevent->Stop();
    Info("Process", "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)", 
         GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
         fSWevent->RealTime()/GetNEventsProcessed(), fSWevent->CpuTime()/GetNEventsProcessed());
    fSWevent->Start();
  }  
}

//--------------------------------------------------------------------------------------------------
void AnaFwkMod::SlaveBegin()
{
  // Book our histogram and start the stop watches.

  fSWtotal = new TStopwatch;
  fSWevent = new TStopwatch;

  if (!PublishObj(&fAllHeaders)) {
    SendError(kAbortAnalysis, "SlaveBegin", 
              "Could not publish all event headers with name %s.", fAllHeaders.GetName());
    return;
  }
}

//--------------------------------------------------------------------------------------------------
void AnaFwkMod::SlaveTerminate()
{
  // Fill event histogram and printout timing information.

  RetractObj(fAllHeaders.GetName());

  SaveNEventsProcessed();
  TH1D *hDAllEvents = new TH1D("hDAllEvents","Sum of processed and skimmed events",1,-0.5,0.5);
  hDAllEvents->Fill(0.0,fNEventsSkimmed+GetNEventsProcessed());
  hDAllEvents->SetEntries(fNEventsSkimmed+GetNEventsProcessed());
  AddOutput(hDAllEvents);

  fSWtotal->Stop();
  fSWevent->Stop();

  MDB(kAnalysis, 1)
    Info("SlaveTerminate", "Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)", 
         GetNEventsProcessed(), fSWtotal->RealTime(), fSWtotal->CpuTime(),
         fSWtotal->RealTime()/GetNEventsProcessed(), fSWtotal->CpuTime()/GetNEventsProcessed());

  delete fSWtotal;
  delete fSWevent;
}
Revision:	1.7
Committed:	Thu Mar 12 18:25:35 2009 UTC (16 years, 1 month ago) by loizides
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_008pre2
Changes since 1.6:	+6 -7 lines
Log Message:	Read skimmed headers properly.
#	User	Rev	Content
1	loizides	1.7	// $Id: AnaFwkMod.cc,v 1.6 2009/03/12 15:37:53 loizides Exp $
2	loizides	1.1
3			#include "MitAna/TreeMod/interface/AnaFwkMod.h"
4			#include "MitAna/DataUtil/interface/Debug.h"
5	loizides	1.6	#include "MitAna/DataTree/interface/Names.h"
6	loizides	1.4	#include <TFile.h>
7	loizides	1.1	#include <TH1D.h>
8			#include <TStopwatch.h>
9	loizides	1.4	#include <TTree.h>
10	loizides	1.1
11			using namespace mithep;
12
13			ClassImp(mithep::AnaFwkMod)
14
15			//--------------------------------------------------------------------------------------------------
16			AnaFwkMod::AnaFwkMod(const char name, const char title) :
17			BaseMod(name,title),
18	loizides	1.6	fAllHeadTreeName(Names::gkAllEvtTreeName),
19			fAllHeadBrName(Names::gkAllEvtHeaderBrn),
20	loizides	1.1	fSWtotal(0),
21	loizides	1.4	fSWevent(0),
22	loizides	1.6	fAllHeaders(0,Names::gkSkimmedHeaders),
23	loizides	1.4	fAllHeadTree(0),
24			fAllEventHeader(0),
25			fReload(kFALSE),
26	loizides	1.6	fCurEnt(-2),
27			fNEventsSkimmed(0)
28	loizides	1.1	{
29			// Constructor.
30			}
31
32			//--------------------------------------------------------------------------------------------------
33	loizides	1.4	void AnaFwkMod::BeginRun()
34			{
35			// Get HLT tree and set branches if new file was opened. Read next entry in HLT key
36			// depending on entry in RunInfo.
37
38			if (fReload) {
39			// reset to be (re-)loaded variables
40			fReload = 0;
41			fAllHeadTree = 0;
42			fAllEventHeader = 0;
43			fCurEnt = 0;
44
45			// get current file
46			TFile *file = GetCurrentFile();
47			if (!file)
48			return;
49
50			// get HLT tree
51			fAllHeadTree = dynamic_cast<TTree*>(file->Get(fAllHeadTreeName));
52			if (!fAllHeadTree) {
53	loizides	1.6	SendError(kWarning, "BeginRun",
54			"Can not find tree '%s' in file '%s'",
55			fAllHeadTreeName.Data(),file->GetName());
56	loizides	1.4	return;
57			}
58			}
59
60			// get HLT trigger name branch
61			if (fAllHeadTree->GetBranch(fAllHeadBrName)) {
62			fAllHeadTree->SetBranchAddress(fAllHeadBrName, &fAllEventHeader);
63			} else {
64			SendError(kWarning, "BeginRun",
65	loizides	1.6	"Can not find branch '%s' in tree '%s'",
66	loizides	1.4	fAllHeadBrName.Data(), fAllHeadTreeName.Data());
67			return;
68			}
69			}
70
71			//--------------------------------------------------------------------------------------------------
72			void AnaFwkMod::CopyAllEventHeaders()
73			{
74			// Deal with the headers from all events if needed.
75
76			fAllHeaders.Reset();
77
78			const EventHeader *curev = GetEventHeader();
79			if (!curev) {
80			SendError(kWarning, "CopyAllEventHeaders", "Can not obtain current event");
81			return;
82			}
83
84			if (fAllHeadTree) {
85	loizides	1.6	const Int_t nemax = fAllHeadTree->GetEntries();
86			if (fCurEnt == nemax) {
87			SendError(kAbortEvent, "CopyAllEventHeaders",
88			"End of all events tree reached (%d=%d)", fCurEnt, nemax);
89			return;
90			}
91	loizides	1.4	fAllHeadTree->GetEntry(fCurEnt++);
92	loizides	1.7	while(fCurEnt<=nemax && fAllEventHeader->Skimmed()) {
93	loizides	1.4	EventHeader *eh = fAllHeaders.AddNew();
94			eh->SetRunNum(fAllEventHeader->RunNum());
95			eh->SetEvtNum(fAllEventHeader->EvtNum());
96			eh->SetLumiSec(fAllEventHeader->LumiSec());
97	loizides	1.5	eh->SetRunEntry(fAllEventHeader->RunEntry());
98			eh->SetSkimmed(fAllEventHeader->Skimmed());
99	loizides	1.4	fAllHeadTree->GetEntry(fCurEnt++);
100			}
101			if ((fAllEventHeader->RunNum()!=curev->RunNum()) \|\|
102			(fAllEventHeader->EvtNum()!=curev->EvtNum()) \|\|
103			(fAllEventHeader->LumiSec()!=curev->LumiSec()) \|\|
104			(fAllEventHeader->RunEntry()!=curev->RunEntry())) {
105			SendError(kWarning, "CopyAllEventHeaders",
106	loizides	1.6	"Event header information for entry %d inconsistent: "
107			"%d==%d, %d==%d, %d==%d, %d==%d",
108			fCurEnt,
109	loizides	1.4	fAllEventHeader->RunNum(), curev->RunNum(),
110			fAllEventHeader->EvtNum(), curev->EvtNum(),
111			fAllEventHeader->LumiSec(), curev->LumiSec(),
112			fAllEventHeader->RunEntry(), curev->RunEntry());
113			return;
114			}
115	loizides	1.6
116			// read-ahead to check if more events are coming
117			if (fCurEnt<nemax) {
118			Int_t testEnt = fCurEnt;
119			fAllHeadTree->GetEntry(testEnt++);
120	loizides	1.7	while(testEnt<=nemax && fAllEventHeader->Skimmed())
121	loizides	1.6	fAllHeadTree->GetEntry(testEnt++);
122	loizides	1.7	if (testEnt==nemax+1) { // need to add remaining skimmed events
123	loizides	1.6	fAllHeadTree->GetEntry(fCurEnt++);
124	loizides	1.7	while(fCurEnt<=nemax) {
125	loizides	1.6	EventHeader *eh = fAllHeaders.AddNew();
126			eh->SetRunNum(fAllEventHeader->RunNum());
127			eh->SetEvtNum(fAllEventHeader->EvtNum());
128			eh->SetLumiSec(fAllEventHeader->LumiSec());
129			eh->SetRunEntry(fAllEventHeader->RunEntry());
130			eh->SetSkimmed(fAllEventHeader->Skimmed());
131			fAllHeadTree->GetEntry(fCurEnt++);
132			}
133	loizides	1.7	if (fCurEnt != nemax+1) {
134	loizides	1.6	SendError(kAbortEvent, "CopyAllEventHeaders",
135			"End of all events tree unexpectedly not reached (%d!=%d)", fCurEnt, nemax);
136			return;
137			}
138			}
139			}
140	loizides	1.4	}
141			}
142
143			//--------------------------------------------------------------------------------------------------
144			Bool_t AnaFwkMod::Notify()
145			{
146			// Make sure to get the new "AllEvents" tree when the file changes.
147
148			fReload = kTRUE;
149			return kTRUE;
150			}
151
152			//--------------------------------------------------------------------------------------------------
153	loizides	1.1	void AnaFwkMod::Process()
154			{
155			// Do event counting and print out timing information.
156
157	loizides	1.4
158	loizides	1.6	// get skimmed event headers
159			CopyAllEventHeaders();
160			fNEventsSkimmed += fAllHeaders.GetEntries();
161	loizides	1.4
162	loizides	1.1	// counting events
163	loizides	1.3	IncNEventsProcessed();
164	loizides	1.1
165			// check if printout should be done
166			Bool_t doPrint = 0;
167
168			MDB(kAnalysis, 4) {
169	loizides	1.3	if (GetNEventsProcessed() % 1000 == 0)
170	loizides	1.1	doPrint = 1;
171			} else {
172			MDB(kAnalysis, 3) {
173	loizides	1.3	if (GetNEventsProcessed() % 10000 == 0)
174	loizides	1.1	doPrint = 1;
175			} else {
176			MDB(kAnalysis, 2) {
177	loizides	1.3	if (GetNEventsProcessed() % 50000 == 0)
178	loizides	1.1	doPrint = 1;
179			} else {
180			MDB(kAnalysis, 1) {
181	loizides	1.3	if (GetNEventsProcessed() % 250000 == 0)
182	loizides	1.1	doPrint = 1;
183			}
184			}
185			}
186			}
187
188			if (doPrint) {
189			fSWevent->Stop();
190			Info("Process", "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
191	loizides	1.3	GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
192			fSWevent->RealTime()/GetNEventsProcessed(), fSWevent->CpuTime()/GetNEventsProcessed());
193	loizides	1.1	fSWevent->Start();
194			}
195			}
196
197			//--------------------------------------------------------------------------------------------------
198			void AnaFwkMod::SlaveBegin()
199			{
200			// Book our histogram and start the stop watches.
201
202			fSWtotal = new TStopwatch;
203			fSWevent = new TStopwatch;
204	loizides	1.4
205			if (!PublishObj(&fAllHeaders)) {
206			SendError(kAbortAnalysis, "SlaveBegin",
207			"Could not publish all event headers with name %s.", fAllHeaders.GetName());
208			return;
209			}
210	loizides	1.1	}
211
212			//--------------------------------------------------------------------------------------------------
213			void AnaFwkMod::SlaveTerminate()
214			{
215			// Fill event histogram and printout timing information.
216
217	loizides	1.4	RetractObj(fAllHeaders.GetName());
218
219	loizides	1.3	SaveNEventsProcessed();
220	loizides	1.6	TH1D *hDAllEvents = new TH1D("hDAllEvents","Sum of processed and skimmed events",1,-0.5,0.5);
221			hDAllEvents->Fill(0.0,fNEventsSkimmed+GetNEventsProcessed());
222			hDAllEvents->SetEntries(fNEventsSkimmed+GetNEventsProcessed());
223			AddOutput(hDAllEvents);
224	loizides	1.1
225			fSWtotal->Stop();
226			fSWevent->Stop();
227
228			MDB(kAnalysis, 1)
229			Info("SlaveTerminate", "Events %d -> %.2gs real, %.2gs cpu (%.2gs real, %.2gs cpu per event)",
230	loizides	1.3	GetNEventsProcessed(), fSWtotal->RealTime(), fSWtotal->CpuTime(),
231			fSWtotal->RealTime()/GetNEventsProcessed(), fSWtotal->CpuTime()/GetNEventsProcessed());
232	loizides	1.1
233			delete fSWtotal;
234			delete fSWevent;
235			}