[53] | 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 |
|
---|