StopBabyMaker/JetTree.h at master · indarasuarez/StopBabyMaker · GitHub

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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#ifndef JETTREE_H
#define JETTREE_H

#include <string>
#include <vector>
#include "Math/LorentzVector.h"

// forward declarations
class TTree;

// typedefs
typedef ROOT::Math::LorentzVector<ROOT::Math::PxPyPzE4D<float> > LorentzVector;
typedef std::vector<LorentzVector> vecLorentzVector;
typedef std::vector<float> vecd;
typedef std::vector<bool> vecb;
typedef std::vector<int> veci;
typedef std::vector<std::vector<int> > vecii;

class JetTree
{
public:
    // constructor/destructor
    JetTree ();
    virtual ~JetTree () {}

    void Reset ();
    void SetBranches (TTree* tree);
    void SetAliases (TTree* tree);
    void FillCommon(unsigned int overlep1_idx, unsigned int overlep2_idx);
    void SetJetPtSelection (std::string cone_size, float pt_cut);
    void GetJetPtSelections (std::string cone_size = "");

    // branch objects

    // ak4 PF jets
    int ak4GoodPFJets;
    int ak8GoodPFJets;
    int nGoodGenJets;
    int ak4_nBTags_Med;
    float ak4_HT;

    float ak4_htssm;
    float ak4_htosm;
    float ak4_htratiom;

    vecLorentzVector ak4pfjets_p4;
    LorentzVector ak4pfjets_leadMEDbjet_p4;
    vecd ak4pfjets_qg_disc;
    vecd ak4pfjets_btag_disc;
    vecd ak4pfjets_pu_id;
    veci ak4pfjets_parton_flavor;
    vecb ak4pfjets_loose_puid;
    vecb ak4pfjets_loose_pfid;
    vecb ak4pfjets_medium_pfid;
    vecb ak4pfjets_tight_pfid;
    vecb ak4pfjets_passMEDbtag;
    float ak4pfjets_leadMEDbjet_pt;
    vecd ak4pfjets_MEDbjet_pt;
    LorentzVector ak4pfjets_leadbtag_p4;

    vecd ak4pfjets_chf;
    vecd ak4pfjets_nhf;
    vecd ak4pfjets_cef;
    vecd ak4pfjets_nef;
    veci ak4pfjets_cm;
    veci ak4pfjets_nm;

   //overlaps
    LorentzVector ak4pfjet_overlep1_p4;
    float ak4pfjet_overlep1_btag_disc;
    float ak4pfjet_overlep1_pu_id;
    float ak4pfjet_overlep1_chf;
    float ak4pfjet_overlep1_nhf;
    float ak4pfjet_overlep1_cef;
    float ak4pfjet_overlep1_nef;
    int ak4pfjet_overlep1_cm;
    int ak4pfjet_overlep1_nm;

    LorentzVector ak4pfjet_overlep2_p4;
    float ak4pfjet_overlep2_btag_disc;
    float ak4pfjet_overlep2_pu_id;
    float ak4pfjet_overlep2_chf;
    float ak4pfjet_overlep2_nhf;
    float ak4pfjet_overlep2_cef;
    float ak4pfjet_overlep2_nef;
    int ak4pfjet_overlep2_cm;
    int ak4pfjet_overlep2_nm;

    // ak8 PF jets
    vecLorentzVector ak8pfjets_p4;
    vecd ak8pfjets_tau1;
    vecd ak8pfjets_tau2;
    vecd ak8pfjets_tau3;
    vecd ak8pfjets_top_mass;
    vecd ak8pfjets_pruned_mass;
    vecd ak8pfjets_trimmed_mass;
    vecd ak8pfjets_filtered_mass;
    vecd ak8pfjets_pu_id;
    veci ak8pfjets_parton_flavor;

    // genjets
    vecLorentzVector ak4genjets_p4;

private:
    float m_ak4_pt_cut;
    float m_ak8_pt_cut;
};

#endif