CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Types | Private Attributes | Static Private Attributes
citk::PFIsolationSumProducerForPUPPI Class Reference
Inheritance diagram for citk::PFIsolationSumProducerForPUPPI:
edm::stream::EDProducer<>

Public Member Functions

void beginLuminosityBlock (const edm::LuminosityBlock &, const edm::EventSetup &) final
 
 PFIsolationSumProducerForPUPPI (const edm::ParameterSet &)
 
void produce (edm::Event &, const edm::EventSetup &) final
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
 EDProducer (const EDProducer &)=delete
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginProcessBlocks () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndProcessBlocks () const final
 
bool hasAbilityToProduceInEndRuns () const final
 
const EDProduceroperator= (const EDProducer &)=delete
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 

Private Types

typedef edm::View< reco::CandidateCandView
 
typedef std::vector< std::unique_ptr< IsolationConeDefinitionBase > > IsoTypes
 
typedef std::unordered_map< std::string, int > TypeMap
 

Private Attributes

edm::EDGetTokenT< CandView_isolate_with
 
std::array< IsoTypes, kNPFTypes_isolation_types
 
std::array< std::vector< std::string >, kNPFTypes_product_names
 
edm::EDGetTokenT< CandView_to_isolate
 
const TypeMap _typeMap
 
edm::Handle< edm::ValueMap< float > > puppiValueMap
 
edm::EDGetTokenT< edm::ValueMap< float > > puppiValueMapToken_
 
bool usePUPPINoLepton
 
bool useValueMapForPUPPI = true
 

Static Private Attributes

static constexpr unsigned kNPFTypes = 8
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
using CacheTypes = CacheContexts< T... >
 
using GlobalCache = typename CacheTypes::GlobalCache
 
using HasAbility = AbilityChecker< T... >
 
using InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
 
using LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
 
using LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
 
using LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
 
using RunCache = typename CacheTypes::RunCache
 
using RunContext = RunContextT< RunCache, GlobalCache >
 
using RunSummaryCache = typename CacheTypes::RunSummaryCache
 

Detailed Description

Definition at line 26 of file CITKPFIsolationSumProducerForPUPPI.cc.

Member Typedef Documentation

◆ CandView

Definition at line 41 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ IsoTypes

typedef std::vector<std::unique_ptr<IsolationConeDefinitionBase> > citk::PFIsolationSumProducerForPUPPI::IsoTypes
private

Definition at line 40 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ TypeMap

typedef std::unordered_map<std::string, int> citk::PFIsolationSumProducerForPUPPI::TypeMap
private

Definition at line 39 of file CITKPFIsolationSumProducerForPUPPI.cc.

Constructor & Destructor Documentation

◆ PFIsolationSumProducerForPUPPI()

citk::PFIsolationSumProducerForPUPPI::PFIsolationSumProducerForPUPPI ( const edm::ParameterSet c)

Definition at line 60 of file CITKPFIsolationSumProducerForPUPPI.cc.

61  : _typeMap({{"h+", 1}, {"h0", 5}, {"gamma", 4}, {"electron", 2}, {"muon", 3}, {"HFh", 6}, {"HFgamma", 7}}) {
62  _to_isolate = consumes<CandView>(c.getParameter<edm::InputTag>("srcToIsolate"));
63  _isolate_with = consumes<CandView>(c.getParameter<edm::InputTag>("srcForIsolationCone"));
64  if (!c.getParameter<edm::InputTag>("puppiValueMap").label().empty()) {
65  puppiValueMapToken_ = mayConsume<edm::ValueMap<float>>(
66  c.getParameter<edm::InputTag>("puppiValueMap")); //getting token for puppiValueMap
67  useValueMapForPUPPI = true;
68  } else {
69  useValueMapForPUPPI = false;
70  usePUPPINoLepton = c.getParameter<bool>("usePUPPINoLepton");
71  }
72  const std::vector<edm::ParameterSet>& isoDefs = c.getParameterSetVector("isolationConeDefinitions");
73  for (const auto& isodef : isoDefs) {
74  const std::string& name = isodef.getParameter<std::string>("isolationAlgo");
75  const float coneSize = isodef.getParameter<double>("coneSize");
76  char buf[50];
77  std::sprintf(buf, "DR%.2f", coneSize);
78  std::string coneName(buf);
79  auto decimal = coneName.find('.');
80  if (decimal != std::string::npos)
81  coneName.erase(decimal, 1);
82  const std::string& isotype = isodef.getParameter<std::string>("isolateAgainst");
83  auto theisolator = CITKIsolationConeDefinitionFactory::get()->create(name, isodef);
84  theisolator->setConsumes(consumesCollector());
85  const auto thetype = _typeMap.find(isotype);
86  if (thetype == _typeMap.end()) {
87  throw cms::Exception("InvalidIsolationType") << "Isolation type: " << isotype << " is not available in the "
88  << "list of allowed isolations!.";
89  }
90  const std::string dash("-");
91  std::string pname = isotype + dash + coneName + dash + theisolator->additionalCode();
92  _product_names[thetype->second].emplace_back(pname);
93  produces<edm::ValueMap<float>>(pname);
94  _isolation_types[thetype->second].emplace_back(std::move(theisolator));
95  }
96  }
std::array< std::vector< std::string >, kNPFTypes > _product_names
std::string const & label() const
Definition: InputTag.h:36
edm::EDGetTokenT< edm::ValueMap< float > > puppiValueMapToken_
#define get
def move(src, dest)
Definition: eostools.py:511

Member Function Documentation

◆ beginLuminosityBlock()

void citk::PFIsolationSumProducerForPUPPI::beginLuminosityBlock ( const edm::LuminosityBlock ,
const edm::EventSetup es 
)
final

Definition at line 98 of file CITKPFIsolationSumProducerForPUPPI.cc.

References _isolation_types.

98  {
99  for (const auto& isolators_for_type : _isolation_types) {
100  for (const auto& isolator : isolators_for_type) {
101  isolator->getEventSetupInfo(es);
102  }
103  }
104  }

◆ fillDescriptions()

void citk::PFIsolationSumProducerForPUPPI::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 182 of file CITKPFIsolationSumProducerForPUPPI.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addOptional(), edm::ParameterSetDescription::addVPSet(), ProducerED_cfi::InputTag, egmElectronIsolationPUPPI_cff::isolationConeDefinitions, BPHMonitor_cfi::photons, edm::ParameterSetDescription::setComment(), and AlCaHLTBitMon_QueryRunRegistry::string.

182  {
184  iDesc.setComment("PUPPI isolation sum producer");
185 
186  iDesc.add<edm::InputTag>("srcToIsolate", edm::InputTag("no default"))
187  ->setComment("calculate isolation for this collection");
188  iDesc.add<edm::InputTag>("srcForIsolationCone", edm::InputTag("no default"))
189  ->setComment("collection for the isolation calculation: like particleFlow ");
190  iDesc.add<edm::InputTag>("puppiValueMap", edm::InputTag("puppi"))
191  ->setComment("source for puppi, if left empty weight from packedCandidate is taken");
192 
193  edm::ParameterSetDescription descIsoConeDefinitions;
194  descIsoConeDefinitions.add<std::string>("isolationAlgo", "no default");
195  descIsoConeDefinitions.add<double>("coneSize", 0.3);
196  descIsoConeDefinitions.add<std::string>("isolateAgainst", "no default");
197  descIsoConeDefinitions.add<std::vector<unsigned>>("miniAODVertexCodes", {2, 3});
198  descIsoConeDefinitions.addOptional<double>("VetoConeSizeBarrel", 0.0);
199  descIsoConeDefinitions.addOptional<double>("VetoConeSizeEndcaps", 0.0);
200  descIsoConeDefinitions.addOptional<double>("VetoThreshold", 0.0);
201  descIsoConeDefinitions.addOptional<double>("VetoConeSize", 0.0);
202  descIsoConeDefinitions.addOptional<int>("vertexIndex", 0);
203  descIsoConeDefinitions.addOptional<edm::InputTag>("particleBasedIsolation", edm::InputTag("no default"))
204  ->setComment("map for footprint removal that is used for photons");
205 
206  std::vector<edm::ParameterSet> isolationConeDefinitions;
207  edm::ParameterSet chargedHadrons, neutralHadrons, photons;
208  isolationConeDefinitions.push_back(chargedHadrons);
209  isolationConeDefinitions.push_back(neutralHadrons);
211  iDesc.addVPSet("isolationConeDefinitions", descIsoConeDefinitions, isolationConeDefinitions);
212  iDesc.add<bool>("usePUPPINoLepton", false);
213 
214  descriptions.add("CITKPFIsolationSumProducerForPUPPI", iDesc);
215  }
void setComment(std::string const &value)
ParameterDescriptionBase * add(U const &iLabel, T const &value)
void add(std::string const &label, ParameterSetDescription const &psetDescription)

◆ produce()

void citk::PFIsolationSumProducerForPUPPI::produce ( edm::Event ev,
const edm::EventSetup es 
)
final

Definition at line 106 of file CITKPFIsolationSumProducerForPUPPI.cc.

References _isolate_with, _isolation_types, _product_names, _to_isolate, DummyCfis::c, makeMEIFBenchmarkPlots::ev, edm::helper::Filler< Map >::fill(), mps_fire::i, edm::helper::Filler< Map >::insert(), edm::Ptr< T >::isNull(), dqmiolumiharvest::j, dqmdumpme::k, kNPFTypes, eostools::move(), edm::View< T >::ptrAt(), puppiValueMap, puppiValueMapToken_, pat::PackedCandidate::puppiWeight(), pat::PackedCandidate::puppiWeightNoLep(), edm::View< T >::size(), usePUPPINoLepton, and useValueMapForPUPPI.

106  {
107  typedef std::unique_ptr<edm::ValueMap<float>> product_type;
108  typedef std::vector<float> product_values;
109  edm::Handle<CandView> to_isolate;
110  edm::Handle<CandView> isolate_with;
111  ev.getByToken(_to_isolate, to_isolate);
112  ev.getByToken(_isolate_with, isolate_with);
114  ev.getByToken(puppiValueMapToken_, puppiValueMap);
115 
116  // the list of value vectors indexed as "to_isolate"
117  std::array<std::vector<product_values>, kNPFTypes> the_values;
118  // get extra event info and setup value cache
119  unsigned i = 0;
120  for (const auto& isolators_for_type : _isolation_types) {
121  the_values[i++].resize(isolators_for_type.size());
122  for (const auto& isolator : isolators_for_type) {
123  isolator->getEventInfo(ev);
124  }
125  }
126  reco::PFCandidate helper; // to translate pdg id to type
127  // loop over the candidates we are isolating and fill the values
128  for (size_t c = 0; c < to_isolate->size(); ++c) {
129  auto cand_to_isolate = to_isolate->ptrAt(c);
130  std::array<std::vector<float>, kNPFTypes> cand_values;
131  unsigned k = 0;
132  for (const auto& isolators_for_type : _isolation_types) {
133  cand_values[k].resize(isolators_for_type.size());
134  for (auto& value : cand_values[k])
135  value = 0.0;
136  ++k;
137  }
138  for (size_t ic = 0; ic < isolate_with->size(); ++ic) {
139  auto isocand = isolate_with->ptrAt(ic);
140  edm::Ptr<pat::PackedCandidate> aspackedCandidate;
141  auto isotype = helper.translatePdgIdToType(isocand->pdgId());
142  const auto& isolations = _isolation_types[isotype];
143  for (unsigned i = 0; i < isolations.size(); ++i) {
144  if (isolations[i]->isInIsolationCone(cand_to_isolate, isocand)) {
145  double puppiWeight = 0.;
146  if (!useValueMapForPUPPI && aspackedCandidate.isNull())
147  aspackedCandidate = edm::Ptr<pat::PackedCandidate>(isocand);
149  puppiWeight = aspackedCandidate->puppiWeight(); // if miniAOD, take puppiWeight directly from the object
151  puppiWeight =
152  aspackedCandidate->puppiWeightNoLep(); // if miniAOD, take puppiWeightNoLep directly from the object
153  else
154  puppiWeight = (*puppiValueMap)[isocand]; // if AOD, take puppiWeight from the valueMap
155  if (puppiWeight > 0.)
156  cand_values[isotype][i] +=
157  (isocand->pt()) *
158  puppiWeight; // this is basically the main change to Lindsey's code: scale pt with puppiWeight for candidates with puppiWeight > 0.
159  }
160  }
161  }
162  // add this candidate to isolation value list
163  for (unsigned i = 0; i < kNPFTypes; ++i) {
164  for (unsigned j = 0; j < cand_values[i].size(); ++j) {
165  the_values[i][j].push_back(cand_values[i][j]);
166  }
167  }
168  }
169  // fill and put all products
170  for (unsigned i = 0; i < kNPFTypes; ++i) {
171  for (unsigned j = 0; j < the_values[i].size(); ++j) {
172  product_type the_product(new edm::ValueMap<float>);
173  edm::ValueMap<float>::Filler fillerprod(*the_product);
174  fillerprod.insert(to_isolate, the_values[i][j].begin(), the_values[i][j].end());
175  fillerprod.fill();
176  ev.put(std::move(the_product), _product_names[i][j]);
177  }
178  }
179  }
float puppiWeight() const
float puppiWeightNoLep() const
Weight from full PUPPI.
Definition: helper.py:1
Ptr< value_type > ptrAt(size_type i) const
edm::Handle< edm::ValueMap< float > > puppiValueMap
std::array< std::vector< std::string >, kNPFTypes > _product_names
edm::EDGetTokenT< edm::ValueMap< float > > puppiValueMapToken_
size_type size() const
bool isNull() const
Checks for null.
Definition: Ptr.h:144
Definition: value.py:1
TracksUtilities< TrackerTraits > helper
Particle reconstructed by the particle flow algorithm.
Definition: PFCandidate.h:41
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

◆ _isolate_with

edm::EDGetTokenT<CandView> citk::PFIsolationSumProducerForPUPPI::_isolate_with
private

Definition at line 43 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _isolation_types

std::array<IsoTypes, kNPFTypes> citk::PFIsolationSumProducerForPUPPI::_isolation_types
private

Definition at line 47 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by beginLuminosityBlock(), and produce().

◆ _product_names

std::array<std::vector<std::string>, kNPFTypes> citk::PFIsolationSumProducerForPUPPI::_product_names
private

Definition at line 48 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _to_isolate

edm::EDGetTokenT<CandView> citk::PFIsolationSumProducerForPUPPI::_to_isolate
private

Definition at line 43 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _typeMap

const TypeMap citk::PFIsolationSumProducerForPUPPI::_typeMap
private

Definition at line 42 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ kNPFTypes

constexpr unsigned citk::PFIsolationSumProducerForPUPPI::kNPFTypes = 8
staticprivate

Definition at line 38 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ puppiValueMap

edm::Handle<edm::ValueMap<float> > citk::PFIsolationSumProducerForPUPPI::puppiValueMap
private

Definition at line 45 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ puppiValueMapToken_

edm::EDGetTokenT<edm::ValueMap<float> > citk::PFIsolationSumProducerForPUPPI::puppiValueMapToken_
private

Definition at line 44 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ usePUPPINoLepton

bool citk::PFIsolationSumProducerForPUPPI::usePUPPINoLepton
private
Initial value:
=
false

Definition at line 50 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ useValueMapForPUPPI

bool citk::PFIsolationSumProducerForPUPPI::useValueMapForPUPPI = true
private

Definition at line 49 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().