1 |
/* Usage:
|
2 |
root [0] .L ScanChain.C++
|
3 |
root [1] TFile *_file0 = TFile::Open("merged_ntuple.root")
|
4 |
root [2] TChain *chain = new TChain("Events")
|
5 |
root [3] chain->Add("merged_ntuple.root")
|
6 |
|
7 |
There are several places where one may create CMS2 cms2
|
8 |
It can be done here (in a doAll.C script), i.e.:
|
9 |
|
10 |
root [4] CMS2 cms2
|
11 |
|
12 |
It can be done in the source as is done below, or it can be
|
13 |
ascertained by including CORE/CMS2.cc as is commented out
|
14 |
below. They are all the same, and everything will work so
|
15 |
long as it is created somewhere globally.
|
16 |
|
17 |
root [5] ScanChain(chain)
|
18 |
*/
|
19 |
#include <iostream>
|
20 |
#include <vector>
|
21 |
|
22 |
#include "TChain.h"
|
23 |
#include "TFile.h"
|
24 |
#include "TDirectory.h"
|
25 |
#include "TROOT.h"
|
26 |
|
27 |
#include "CMS2.h"
|
28 |
CMS2 cms2;
|
29 |
/*
|
30 |
#include "CORE/CMS2.cc"
|
31 |
#include "CORE/selections.cc"
|
32 |
#include "CORE/utilities.cc"
|
33 |
*/
|
34 |
|
35 |
using namespace tas;
|
36 |
|
37 |
int ScanChain( TChain* chain, int nEvents = -1, std::string skimFilePrefix="") {
|
38 |
|
39 |
TObjArray *listOfFiles = chain->GetListOfFiles();
|
40 |
|
41 |
unsigned int nEventsChain=0;
|
42 |
if(nEvents==-1)
|
43 |
nEvents = chain->GetEntries();
|
44 |
nEventsChain = nEvents;
|
45 |
unsigned int nEventsTotal = 0;
|
46 |
TDirectory *rootdir = gDirectory->GetDirectory("Rint:");
|
47 |
|
48 |
TH1F *samplehisto = new TH1F("samplehisto", "Example histogram", 200,0,200);
|
49 |
samplehisto->SetDirectory(rootdir);
|
50 |
// file loop
|
51 |
TIter fileIter(listOfFiles);
|
52 |
TFile *currentFile = 0;
|
53 |
while ( currentFile = (TFile*)fileIter.Next() ) {
|
54 |
TFile f(currentFile->GetTitle());
|
55 |
TTree *tree = (TTree*)f.Get("Events");
|
56 |
cms2.Init(tree);
|
57 |
|
58 |
//Event Loop
|
59 |
unsigned int nEvents = tree->GetEntries();
|
60 |
for( unsigned int event = 0; event < nEvents; ++event) {
|
61 |
cms2.GetEntry(event);
|
62 |
++nEventsTotal;
|
63 |
}
|
64 |
}
|
65 |
|
66 |
if ( nEventsChain != nEventsTotal ) {
|
67 |
std::cout << "ERROR: number of events from files is not equal to total number of events" << std::endl;
|
68 |
}
|
69 |
|
70 |
samplehisto->Draw();
|
71 |
return 0;
|
72 |
}
|