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.9 by loizides, Mon Mar 23 22:15:15 2009 UTC vs.
Revision 1.23 by bendavid, Mon Aug 26 22:43:59 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 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 >  fMCEventInfo(0),
39 >  fMCEventInfoName(Names::gkMCEvtInfoBrn),
40 >  hDTotalMCWeight(0)
41   {
42    // Constructor.
43   }
# Line 49 | 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",
# Line 57 | Line 68 | void AnaFwkMod::BeginRun()
68                  fAllHeadTreeName.Data(),file->GetName());
69        return;
70      }
60 –  }
71  
72 <  // get HLT trigger name 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;
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 91 | 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 119 | 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 167 | Line 178 | void AnaFwkMod::Process()
178    if (fNEventsSkipped<fSkipNEvents) {
179      ++fNEventsSkipped;
180      MDB(kAnalysis, 3) {
181 <      Info("Process", "Skipping (aborting) %d out of %d first events.",
181 >      Info("Process", "Skipping (aborting) %d out of %lld first events.",
182             fNEventsSkipped, fSkipNEvents);
183      }
184      AbortEvent();
# Line 175 | Line 186 | void AnaFwkMod::Process()
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() % 250  == 0)
193 >    if (GetNEventsProcessed() % (fPrintScale)  == 0)
194        doPrint = 1;
195    } else {
196      MDB(kAnalysis, 3) {
197 <      if (GetNEventsProcessed() % 2500  == 0)
197 >      if (GetNEventsProcessed() % (fPrintScale*10)  == 0)
198          doPrint = 1;
199 +      nProcessed = fPrintScale*10;
200      } else {
201        MDB(kAnalysis, 2) {
202 <        if (GetNEventsProcessed() % 10000  == 0)
202 >        if (GetNEventsProcessed() % (fPrintScale*100)  == 0)
203            doPrint = 1;
204 +        nProcessed = fPrintScale*100;
205        } else {
206          MDB(kAnalysis, 1) {
207 <          if (GetNEventsProcessed() % 50000 == 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",
238           "Events %d -> %.2gs real, %.2gs cpu (%.2g real, %.2g cpu per event)",
239           GetNEventsProcessed(), fSWevent->RealTime(), fSWevent->CpuTime(),
240 <         fSWevent->RealTime()/GetNEventsProcessed(),
241 <         fSWevent->CpuTime()/GetNEventsProcessed());
240 >         fSWevent->RealTime()/nProcessed,
241 >         fSWevent->CpuTime()/nProcessed);
242      fSWevent->Start();
243 <  }  
243 >  }
244 >  
245 >    
246   }
247  
248   //--------------------------------------------------------------------------------------------------
# Line 221 | 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 +  
264 +  hNPU = new TH1D("hNPU", "hNPU", 201, -0.5, 200.5);
265 +  AddOutput(hNPU);  
266 +  
267 +  hNPU50ns = new TH3D("hNPU50ns", "hNPU50ns", 201, -0.5, 200.5, 201, -0.5, 200.5, 201, -0.5, 200.5);
268 +  AddOutput(hNPU50ns);  
269 +  
270 +  hNPUTrue = new TH1D("hNPUTrue", "hNPUTrue", 2000, 0.0, 200.0);
271 +  AddOutput(hNPUTrue);  
272 +  
273 +  hDTotalMCWeight = new TH1D("hDTotalMCWeight","hDTotalMCWeight",1,-0.5,0.5);
274 +  AddOutput(hDTotalMCWeight);
275 +  
276   }
277  
278   //--------------------------------------------------------------------------------------------------

Diff Legend

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