1 | /*
|
---|
2 | * ---- Delphes ----
|
---|
3 | * A Fast Simulator for general purpose LHC detector
|
---|
4 | * S. Ovyn ~~~~ severine.ovyn@uclouvain.be
|
---|
5 | *
|
---|
6 | * Center for Particle Physics and Phenomenology (CP3)
|
---|
7 | * Universite Catholique de Louvain (UCL)
|
---|
8 | * Louvain-la-Neuve, Belgium
|
---|
9 | * */
|
---|
10 |
|
---|
11 | // \brief Trigger class, and some generic definitions
|
---|
12 |
|
---|
13 | #include "interface/JetUtils.h"
|
---|
14 |
|
---|
15 | using namespace std;
|
---|
16 |
|
---|
17 | JetsUtil::JetsUtil() {
|
---|
18 |
|
---|
19 | switch(JETALGO) {
|
---|
20 | default:
|
---|
21 | case 1: {
|
---|
22 |
|
---|
23 | // set up a CDF midpoint jet definition
|
---|
24 | #ifdef ENABLE_PLUGIN_CDFCONES
|
---|
25 | plugins = new fastjet::CDFJetCluPlugin(SEEDTHRESHOLD,CONERADIUS,C_ADJACENCYCUT,C_MAXITERATIONS,C_IRATCH,OVERLAPTHRESHOLD);
|
---|
26 | jet_def = fastjet::JetDefinition(plugins);
|
---|
27 | #else
|
---|
28 | plugins = NULL;
|
---|
29 | #endif
|
---|
30 | }
|
---|
31 | break;
|
---|
32 |
|
---|
33 | case 2: {
|
---|
34 | // set up a CDF midpoint jet definition
|
---|
35 | #ifdef ENABLE_PLUGIN_CDFCONES
|
---|
36 | plugins = new fastjet::CDFMidPointPlugin (SEEDTHRESHOLD,CONERADIUS,M_CONEAREAFRACTION,M_MAXPAIRSIZE,M_MAXITERATIONS,OVERLAPTHRESHOLD);
|
---|
37 | jet_def = fastjet::JetDefinition(plugins);
|
---|
38 | #else
|
---|
39 | plugins = NULL;
|
---|
40 | #endif
|
---|
41 | }
|
---|
42 | break;
|
---|
43 |
|
---|
44 | case 3: {
|
---|
45 | // set up a siscone jet definition
|
---|
46 | #ifdef ENABLE_PLUGIN_SISCONE
|
---|
47 | plugins = new fastjet::SISConePlugin (CONERADIUS,OVERLAPTHRESHOLD,NPASS, PROTOJET_PTMIN);
|
---|
48 | jet_def = fastjet::JetDefinition(plugins);
|
---|
49 | #else
|
---|
50 | plugins = NULL;
|
---|
51 | #endif
|
---|
52 | }
|
---|
53 | break;
|
---|
54 |
|
---|
55 | case 4: {
|
---|
56 | jet_def = fastjet::JetDefinition(fastjet::kt_algorithm, CONERADIUS);
|
---|
57 | }
|
---|
58 | break;
|
---|
59 |
|
---|
60 | case 5: {
|
---|
61 | jet_def = fastjet::JetDefinition(fastjet::cambridge_algorithm,CONERADIUS);
|
---|
62 | }
|
---|
63 | break;
|
---|
64 |
|
---|
65 | case 6: {
|
---|
66 | jet_def = fastjet::JetDefinition(fastjet::antikt_algorithm,CONERADIUS);
|
---|
67 | }
|
---|
68 | break;
|
---|
69 | }
|
---|
70 |
|
---|
71 | }
|
---|
72 |
|
---|
73 | vector<fastjet::PseudoJet> JetsUtil::RunJets(const vector<fastjet::PseudoJet> & input_particles)
|
---|
74 | {
|
---|
75 | inclusive_jets.clear();
|
---|
76 | sorted_jets.clear();
|
---|
77 |
|
---|
78 | // run the jet clustering with the above jet definition
|
---|
79 | if(input_particles.size()!=0)
|
---|
80 | {
|
---|
81 | fastjet::ClusterSequence clust_seq(input_particles, jet_def);
|
---|
82 | // extract the inclusive jets with pt > 5 GeV
|
---|
83 | double ptmin = 5.0;
|
---|
84 | inclusive_jets = clust_seq.inclusive_jets(ptmin);
|
---|
85 | // sort jets into increasing pt
|
---|
86 | sorted_jets = sorted_by_pt(inclusive_jets);
|
---|
87 | }
|
---|
88 |
|
---|
89 | return sorted_jets;
|
---|
90 | }
|
---|
91 |
|
---|
92 | void JetsUtil::RunJetBtagging(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchJet,const vector<fastjet::PseudoJet> & sorted_jets,const TSimpleArray<TRootGenParticle>& NFCentralQ)
|
---|
93 | {
|
---|
94 | TRootJet *elementJet;
|
---|
95 | TLorentzVector JET;
|
---|
96 | for (unsigned int i = 0; i < sorted_jets.size(); i++) {
|
---|
97 | JET.SetPxPyPzE(sorted_jets[i].px(),sorted_jets[i].py(),sorted_jets[i].pz(),sorted_jets[i].E());
|
---|
98 | if(JET.Pt() > JET_pt)
|
---|
99 | {
|
---|
100 | elementJet = (TRootJet*) branchJet->NewEntry();
|
---|
101 | elementJet->Set(JET);
|
---|
102 | // b-jets
|
---|
103 | bool btag=false;
|
---|
104 | if((fabs(JET.Eta()) < MAX_TRACKER && Btaggedjet(JET, NFCentralQ)))btag=true;
|
---|
105 | elementJet->Btag = btag;
|
---|
106 | }
|
---|
107 | } // for itJet : loop on all jets
|
---|
108 | treeWriter->Fill();
|
---|
109 |
|
---|
110 | }
|
---|
111 |
|
---|
112 | void JetsUtil::RunTauJets(ExRootTreeWriter *treeWriter, ExRootTreeBranch *branchTauJet,const vector<fastjet::PseudoJet> & sorted_jets,const vector<PhysicsTower> & towers, const vector<TLorentzVector> & TrackCentral)
|
---|
113 | {
|
---|
114 | TRootTauJet *elementTauJet;
|
---|
115 | TLorentzVector JET;
|
---|
116 | for (unsigned int i = 0; i < sorted_jets.size(); i++) {
|
---|
117 | JET.SetPxPyPzE(sorted_jets[i].px(),sorted_jets[i].py(),sorted_jets[i].pz(),sorted_jets[i].E());
|
---|
118 | // Tau jet identification : 1! track and electromagnetic collimation
|
---|
119 | if(fabs(JET.Eta()) < (MAX_TRACKER - TAU_CONE_TRACKS)) {
|
---|
120 | double Energie_tau_central = EnergySmallCone(towers,JET.Eta(),JET.Phi());
|
---|
121 | if(
|
---|
122 | ( Energie_tau_central/JET.E() > TAU_EM_COLLIMATION ) &&
|
---|
123 | ( NumTracks(TrackCentral,PT_TRACK_TAU,JET.Eta(),JET.Phi()) == 1 ) &&
|
---|
124 | ( JET.Pt() > TAUJET_pt)
|
---|
125 | ) {
|
---|
126 | elementTauJet = (TRootTauJet*) branchTauJet->NewEntry();
|
---|
127 | elementTauJet->Set(JET);
|
---|
128 | } // if tau jet
|
---|
129 | } // if JET.eta < tracker - tau_cone : Tau jet identification
|
---|
130 |
|
---|
131 | } // for itJet : loop on all jets
|
---|
132 |
|
---|
133 | treeWriter->Fill();
|
---|
134 |
|
---|
135 | }
|
---|
136 |
|
---|