Fork me on GitHub

source: git/modules/JetPileUpSubtractor.cc@ 0e6b14e

ImprovedOutputFile Timing
Last change on this file since 0e6b14e was 341014c, checked in by Pavel Demin <pavel-demin@…>, 6 years ago

apply .clang-format to all .h, .cc and .cpp files

  • Property mode set to 100644
File size: 3.7 KB
RevLine 
[b443089]1/*
2 * Delphes: a framework for fast simulation of a generic collider experiment
3 * Copyright (C) 2012-2014 Universite catholique de Louvain (UCL), Belgium
[1fa50c2]4 *
[b443089]5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
[1fa50c2]9 *
[b443089]10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
[1fa50c2]14 *
[b443089]15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 */
18
[d7d2da3]19/** \class JetPileUpSubtractor
20 *
21 * Subtract pile-up contribution from jets using the fastjet area method
22 *
23 * \author M. Selvaggi - UCL, Louvain-la-Neuve
24 *
25 */
26
27#include "modules/JetPileUpSubtractor.h"
28
29#include "classes/DelphesClasses.h"
30#include "classes/DelphesFactory.h"
31#include "classes/DelphesFormula.h"
32
33#include "ExRootAnalysis/ExRootClassifier.h"
[341014c]34#include "ExRootAnalysis/ExRootFilter.h"
35#include "ExRootAnalysis/ExRootResult.h"
[d7d2da3]36
37#include "TDatabasePDG.h"
[341014c]38#include "TFormula.h"
[d7d2da3]39#include "TLorentzVector.h"
[341014c]40#include "TMath.h"
41#include "TObjArray.h"
42#include "TRandom3.h"
43#include "TString.h"
[d7d2da3]44
45#include <algorithm>
46#include <iostream>
47#include <sstream>
[341014c]48#include <stdexcept>
[d7d2da3]49
50using namespace std;
51
52//------------------------------------------------------------------------------
53
54JetPileUpSubtractor::JetPileUpSubtractor() :
[8336b6e]55 fItJetInputArray(0), fItRhoInputArray(0)
[d7d2da3]56{
57}
58
59//------------------------------------------------------------------------------
60
61JetPileUpSubtractor::~JetPileUpSubtractor()
62{
63}
64
65//------------------------------------------------------------------------------
66
67void JetPileUpSubtractor::Init()
68{
69 fJetPTMin = GetDouble("JetPTMin", 20.0);
70
71 // import input array(s)
72
73 fJetInputArray = ImportArray(GetString("JetInputArray", "FastJetFinder/jets"));
74 fItJetInputArray = fJetInputArray->MakeIterator();
75
76 fRhoInputArray = ImportArray(GetString("RhoInputArray", "Rho/rho"));
[8336b6e]77 fItRhoInputArray = fRhoInputArray->MakeIterator();
[d7d2da3]78
79 // create output array(s)
80
81 fOutputArray = ExportArray(GetString("OutputArray", "jets"));
82}
83
84//------------------------------------------------------------------------------
85
86void JetPileUpSubtractor::Finish()
87{
[8336b6e]88 if(fItRhoInputArray) delete fItRhoInputArray;
[d7d2da3]89 if(fItJetInputArray) delete fItJetInputArray;
90}
91
92//------------------------------------------------------------------------------
93
94void JetPileUpSubtractor::Process()
95{
[e9023b9]96 Candidate *candidate, *object;
[d7d2da3]97 TLorentzVector momentum, area;
[8336b6e]98 Double_t eta = 0.0;
[d7d2da3]99 Double_t rho = 0.0;
100
101 // loop over all input candidates
102 fItJetInputArray->Reset();
[341014c]103 while((candidate = static_cast<Candidate *>(fItJetInputArray->Next())))
[d7d2da3]104 {
105 momentum = candidate->Momentum;
106 area = candidate->Area;
[d4b9697]107 eta = momentum.Eta();
[8336b6e]108
109 // find rho
110 rho = 0.0;
[e9023b9]111 if(fRhoInputArray)
[8336b6e]112 {
[e9023b9]113 fItRhoInputArray->Reset();
[341014c]114 while((object = static_cast<Candidate *>(fItRhoInputArray->Next())))
[8336b6e]115 {
[e9023b9]116 if(eta >= object->Edges[0] && eta < object->Edges[1])
117 {
118 rho = object->Momentum.Pt();
119 }
[8336b6e]120 }
[e9023b9]121 }
[d7d2da3]122
123 // apply pile-up correction
124 if(momentum.Pt() <= rho * area.Pt()) continue;
125
126 momentum -= rho * area;
[e9023b9]127
[d7d2da3]128 if(momentum.Pt() <= fJetPTMin) continue;
129
[341014c]130 candidate = static_cast<Candidate *>(candidate->Clone());
[d7d2da3]131 candidate->Momentum = momentum;
132
133 fOutputArray->Add(candidate);
134 }
135}
136
137//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.