ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitAna/TreeMod/src/AnaFwkMod.cc
(Generate patch)

Comparing UserCode/MitAna/TreeMod/src/AnaFwkMod.cc (file contents):
Revision 1.10 by loizides, Sat Mar 28 14:39:56 2009 UTC vs.
Revision 1.21 by ceballos, Mon Sep 24 13:22:22 2012 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 18 | Line 20 | AnaFwkMod::AnaFwkMod(const char *name, c
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 26 | Line 29 | AnaFwkMod::AnaFwkMod(const char *name, c
29    fReload(kFALSE),
30    fCurEnt(-2),
31    fNEventsSkimmed(0),
32 <  fNEventsSkipped(0)
32 >  fNEventsSkipped(0),
33 >  fPileupInfoName("PileupInfo"),
34 >  fDoPUInfo(kFALSE),
35 >  hNPU(0),
36 >  hNPU50ns(0),
37 >  hNPUTrue(0)
38   {
39    // Constructor.
40   }
# Line 91 | Line 99 | void AnaFwkMod::CopyAllEventHeaders()
99        return;
100      }
101      fAllHeadTree->GetEntry(fCurEnt++);
102 <    while(fCurEnt<=nemax && fAllEventHeader->Skimmed()) {
102 >    while (fCurEnt <= nemax && fAllEventHeader->Skimmed()) {
103        EventHeader *eh = fAllHeaders.AddNew();
104 <      eh->SetRunNum(fAllEventHeader->RunNum());
105 <      eh->SetEvtNum(fAllEventHeader->EvtNum());
106 <      eh->SetLumiSec(fAllEventHeader->LumiSec());
104 >      eh->SetRunNum  (fAllEventHeader->RunNum());
105 >      eh->SetEvtNum  (fAllEventHeader->EvtNum());
106 >      eh->SetLumiSec (fAllEventHeader->LumiSec());
107        eh->SetRunEntry(fAllEventHeader->RunEntry());
108 <      eh->SetSkimmed(fAllEventHeader->Skimmed());
108 >      eh->SetSkimmed (fAllEventHeader->Skimmed());
109        fAllHeadTree->GetEntry(fCurEnt++);
110      }
111 <    if ((fAllEventHeader->RunNum()!=curev->RunNum())   ||
112 <        (fAllEventHeader->EvtNum()!=curev->EvtNum())   ||
113 <        (fAllEventHeader->LumiSec()!=curev->LumiSec()) ||
114 <        (fAllEventHeader->RunEntry()!=curev->RunEntry())) {
111 >    if ((fAllEventHeader->RunNum()   != curev->RunNum())   ||
112 >        (fAllEventHeader->EvtNum()   != curev->EvtNum())   ||
113 >        (fAllEventHeader->LumiSec()  != curev->LumiSec())  ||
114 >        (fAllEventHeader->RunEntry() != curev->RunEntry())) {
115        SendError(kWarning, "CopyAllEventHeaders",
116                  "Event header information for entry %d inconsistent: "
117                  "%d==%d, %d==%d, %d==%d, %d==%d",
# Line 119 | Line 127 | void AnaFwkMod::CopyAllEventHeaders()
127      if (fCurEnt<nemax) {
128        Int_t testEnt = fCurEnt;
129        fAllHeadTree->GetEntry(testEnt++);
130 <      while(testEnt<=nemax && fAllEventHeader->Skimmed())
130 >      while (testEnt<=nemax && fAllEventHeader->Skimmed())
131          fAllHeadTree->GetEntry(testEnt++);
132        if (testEnt==nemax+1) { // need to add remaining skimmed events
133          fAllHeadTree->GetEntry(fCurEnt++);
134          while(fCurEnt<=nemax) {
135            EventHeader *eh = fAllHeaders.AddNew();
136 <          eh->SetRunNum(fAllEventHeader->RunNum());
137 <          eh->SetEvtNum(fAllEventHeader->EvtNum());
138 <          eh->SetLumiSec(fAllEventHeader->LumiSec());
136 >          eh->SetRunNum  (fAllEventHeader->RunNum());
137 >          eh->SetEvtNum  (fAllEventHeader->EvtNum());
138 >          eh->SetLumiSec (fAllEventHeader->LumiSec());
139            eh->SetRunEntry(fAllEventHeader->RunEntry());
140 <          eh->SetSkimmed(fAllEventHeader->Skimmed());
140 >          eh->SetSkimmed (fAllEventHeader->Skimmed());
141            fAllHeadTree->GetEntry(fCurEnt++);
142          }
143          if (fCurEnt != nemax+1) {
# Line 167 | Line 175 | void AnaFwkMod::Process()
175    if (fNEventsSkipped<fSkipNEvents) {
176      ++fNEventsSkipped;
177      MDB(kAnalysis, 3) {
178 <      Info("Process", "Skipping (aborting) %d out of %d first events.",
178 >      Info("Process", "Skipping (aborting) %d out of %lld first events.",
179             fNEventsSkipped, fSkipNEvents);
180      }
181      AbortEvent();
# Line 175 | Line 183 | void AnaFwkMod::Process()
183    }
184  
185    // check if printout should be done
186 <  Bool_t doPrint = 0;
186 >  Bool_t doPrint    = 0;
187 >  UInt_t nProcessed = fPrintScale;
188  
189    MDB(kAnalysis, 4) {
190 <    if (GetNEventsProcessed() % 250  == 0)
190 >    if (GetNEventsProcessed() % (fPrintScale)  == 0)
191        doPrint = 1;
192    } else {
193      MDB(kAnalysis, 3) {
194 <      if (GetNEventsProcessed() % 2500  == 0)
194 >      if (GetNEventsProcessed() % (fPrintScale*10)  == 0)
195          doPrint = 1;
196 +      nProcessed = fPrintScale*10;
197      } else {
198        MDB(kAnalysis, 2) {
199 <        if (GetNEventsProcessed() % 10000  == 0)
199 >        if (GetNEventsProcessed() % (fPrintScale*100)  == 0)
200            doPrint = 1;
201 +        nProcessed = fPrintScale*100;
202        } else {
203          MDB(kAnalysis, 1) {
204 <          if (GetNEventsProcessed() % 50000 == 0)
204 >          if (GetNEventsProcessed() % (fPrintScale*1000) == 0)
205              doPrint = 1;
206 +          nProcessed = fPrintScale*1000;
207          }
208        }
209      }
210    }
211  
212 +  if (GetEventHeader()->IsMC()) {
213 +    LoadBranch(fPileupInfoName);  
214 +    Double_t npu[4] = {0.,0.,0.,0.};
215 +    for (UInt_t i=0; i<fPileupInfo->GetEntries(); ++i) {
216 +      const PileupInfo *puinfo = fPileupInfo->At(i);
217 +      if (puinfo->GetBunchCrossing()==0) npu[0]= puinfo->GetPU_NumInteractions();
218 +      else if (puinfo->GetBunchCrossing()==-1) npu[1] = puinfo->GetPU_NumInteractions();
219 +      else if (puinfo->GetBunchCrossing()==1) npu[2] = puinfo->GetPU_NumInteractions();
220 +      if (puinfo->GetBunchCrossing()==0) npu[3]= puinfo->GetPU_NumMean();
221 +    }
222 +    hNPU->Fill(npu[0]);
223 +    hNPU50ns->Fill(npu[0],npu[1],npu[2]);
224 +    hNPUTrue->Fill(npu[3]);
225 +  }
226 +
227 +
228    if (doPrint) {
229      fSWevent->Stop();
230      Info("Process",
231           "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
232           GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
233 <         fSWevent->RealTime()/GetNEventsProcessed(),
234 <         fSWevent->CpuTime()/GetNEventsProcessed());
233 >         fSWevent->RealTime()/nProcessed,
234 >         fSWevent->CpuTime()/nProcessed);
235      fSWevent->Start();
236 <  }  
236 >  }
237 >  
238 >    
239   }
240  
241   //--------------------------------------------------------------------------------------------------
# Line 221 | Line 251 | void AnaFwkMod::SlaveBegin()
251                "Could not publish all event headers with name %s.", fAllHeaders.GetName());
252      return;
253    }
254 +  
255 +  ReqBranch(fPileupInfoName, fPileupInfo);
256 +  
257 +  hNPU = new TH1D("hNPU", "hNPU", 201, -0.5, 200.5);
258 +  AddOutput(hNPU);  
259 +  
260 +  hNPU50ns = new TH3D("hNPU50ns", "hNPU50ns", 201, -0.5, 200.5, 201, -0.5, 200.5, 201, -0.5, 200.5);
261 +  AddOutput(hNPU50ns);  
262 +  
263 +  hNPUTrue = new TH1D("hNPUTrue", "hNPUTrue", 2000, 0.0, 2000.0);
264 +  AddOutput(hNPUTrue);  
265 +  
266   }
267  
268   //--------------------------------------------------------------------------------------------------

Diff Legend

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