-
Notifications
You must be signed in to change notification settings - Fork 0
/
ObjectTree.h
50 lines (43 loc) · 1.53 KB
/
ObjectTree.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#pragma once
#define OBJECTTREE_H
#include "Reader.h"
class objectTree;
struct evtFlatDimu{
double hpt; double heta; double hphi; TLorentzVector omu1; TLorentzVector omu2; TLorentzVector odmu; double dR1; double dR2; double dPt1; double dPt2; int passMu;
};
struct evtFlatSimu{
double hpt; double heta; double hphi; TLorentzVector omu1; double dR1; double dPt1; int passMu;
};
struct evtFlatSummary{
double nevt; double cent; double hpt; double heta; double hphi; double opt1; double opt2; double oeta1; double oeta2; double ophi1; double ophi2; double om1; double om2; double dipt; double diy; double diphi; double dim; double dr1; double dr2; double dpt1; double dpt2; int passMu; int oniaN;
};
class objectTree {
public :
objectTree(){};
void init( std::string _trig, bool _isDimu, unsigned int dataType);
~objectTree(){};
void setOniaIndex();
void setTag(string tag);
void setEventWideContent( EventData edat );
void parcelEntry(evtFlatDimu parcel);
void parcelEntry(evtFlatSimu parcel);
// void getOniaFriend();
void flush();
void write();
int oniaCount =0;
private :
int evtCount = 0;
std::string trig;
std::string file_tag = "";
bool isDimu;
TFile* fout;
TTree *fullTree, *oniaTree;
int passMu;
int oniaN;
bool passEff;
double nevt, cent, nColl, hpt, heta, hphi, opt1, opt2, oeta1, oeta2, ophi1, ophi2, om1, om2, dipt, diy, diphi, dim, dr1, dr2, dpt1, dpt2;
double tdr1, tdr2, tdpt1, tdpt2;
double hpt1, heta1, hphi1;
double hpt2, heta2, hphi2;
std::vector<evtFlatSummary> eventMatrix;
};