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
 
 ~PFIsolationSumProducerForPUPPI () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginProcessBlocks () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndProcessBlocks () const final
 
bool hasAbilityToProduceInEndRuns () const final
 

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<>
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

Definition at line 26 of file CITKPFIsolationSumProducerForPUPPI.cc.

Member Typedef Documentation

◆ CandView

Definition at line 43 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ IsoTypes

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

Definition at line 42 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ TypeMap

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

Definition at line 41 of file CITKPFIsolationSumProducerForPUPPI.cc.

Constructor & Destructor Documentation

◆ PFIsolationSumProducerForPUPPI()

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

Definition at line 62 of file CITKPFIsolationSumProducerForPUPPI.cc.

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

◆ ~PFIsolationSumProducerForPUPPI()

citk::PFIsolationSumProducerForPUPPI::~PFIsolationSumProducerForPUPPI ( )
inlineoverride

Definition at line 30 of file CITKPFIsolationSumProducerForPUPPI.cc.

30 {}

Member Function Documentation

◆ beginLuminosityBlock()

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

Definition at line 100 of file CITKPFIsolationSumProducerForPUPPI.cc.

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

References _isolation_types.

◆ fillDescriptions()

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

Definition at line 184 of file CITKPFIsolationSumProducerForPUPPI.cc.

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

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

◆ produce()

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

Definition at line 108 of file CITKPFIsolationSumProducerForPUPPI.cc.

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

References _isolate_with, _isolation_types, _product_names, _to_isolate, HltBtagPostValidation_cff::c, mps_fire::end, 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.

Member Data Documentation

◆ _isolate_with

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

Definition at line 45 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _isolation_types

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

Definition at line 49 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 50 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _to_isolate

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

Definition at line 45 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ _typeMap

const TypeMap citk::PFIsolationSumProducerForPUPPI::_typeMap
private

Definition at line 44 of file CITKPFIsolationSumProducerForPUPPI.cc.

◆ kNPFTypes

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

Definition at line 40 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ puppiValueMap

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

Definition at line 47 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ puppiValueMapToken_

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

Definition at line 46 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ usePUPPINoLepton

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

Definition at line 52 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

◆ useValueMapForPUPPI

bool citk::PFIsolationSumProducerForPUPPI::useValueMapForPUPPI = true
private

Definition at line 51 of file CITKPFIsolationSumProducerForPUPPI.cc.

Referenced by produce().

egmElectronIsolationPUPPI_cff.isolationConeDefinitions
isolationConeDefinitions
Definition: egmElectronIsolationPUPPI_cff.py:27
mps_fire.i
i
Definition: mps_fire.py:428
edm::ParameterSetDescription::add
ParameterDescriptionBase * add(U const &iLabel, T const &value)
Definition: ParameterSetDescription.h:95
HLT_FULL_cff.InputTag
InputTag
Definition: HLT_FULL_cff.py:89287
edm::ParameterSetDescription
Definition: ParameterSetDescription.h:52
citk::PFIsolationSumProducerForPUPPI::_product_names
std::array< std::vector< std::string >, kNPFTypes > _product_names
Definition: CITKPFIsolationSumProducerForPUPPI.cc:50
citk::PFIsolationSumProducerForPUPPI::_typeMap
const TypeMap _typeMap
Definition: CITKPFIsolationSumProducerForPUPPI.cc:44
edm::Handle
Definition: AssociativeIterator.h:50
citk::PFIsolationSumProducerForPUPPI::puppiValueMap
edm::Handle< edm::ValueMap< float > > puppiValueMap
Definition: CITKPFIsolationSumProducerForPUPPI.cc:47
edm::InputTag::label
std::string const & label() const
Definition: InputTag.h:36
citk::PFIsolationSumProducerForPUPPI::useValueMapForPUPPI
bool useValueMapForPUPPI
Definition: CITKPFIsolationSumProducerForPUPPI.cc:51
edm::ConfigurationDescriptions::add
void add(std::string const &label, ParameterSetDescription const &psetDescription)
Definition: ConfigurationDescriptions.cc:57
mps_fire.end
end
Definition: mps_fire.py:242
citk::PFIsolationSumProducerForPUPPI::_isolation_types
std::array< IsoTypes, kNPFTypes > _isolation_types
Definition: CITKPFIsolationSumProducerForPUPPI.cc:49
dqmdumpme.k
k
Definition: dqmdumpme.py:60
unpackData-CaloStage2.pname
pname
Definition: unpackData-CaloStage2.py:76
AlCaHLTBitMon_QueryRunRegistry.string
string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
edm::View::size
size_type size() const
citk::PFIsolationSumProducerForPUPPI::usePUPPINoLepton
bool usePUPPINoLepton
Definition: CITKPFIsolationSumProducerForPUPPI.cc:52
edm::ParameterSet
Definition: ParameterSet.h:47
edm::ParameterSetDescription::setComment
void setComment(std::string const &value)
Definition: ParameterSetDescription.cc:33
HLT_FULL_cff.coneSize
coneSize
Definition: HLT_FULL_cff.py:50845
pat::PackedCandidate::puppiWeight
float puppiWeight() const
Definition: PackedCandidate.cc:380
helper
Definition: helper.py:1
value
Definition: value.py:1
citk::PFIsolationSumProducerForPUPPI::_to_isolate
edm::EDGetTokenT< CandView > _to_isolate
Definition: CITKPFIsolationSumProducerForPUPPI.cc:45
BPHMonitor_cfi.photons
photons
Definition: BPHMonitor_cfi.py:91
HltBtagPostValidation_cff.c
c
Definition: HltBtagPostValidation_cff.py:31
get
#define get
visDQMUpload.buf
buf
Definition: visDQMUpload.py:154
edm::Ptr
Definition: AssociationVector.h:31
eostools.move
def move(src, dest)
Definition: eostools.py:511
edm::ValueMap< float >
ev
bool ev
Definition: Hydjet2Hadronizer.cc:95
Exception
Definition: hltDiff.cc:246
citk::PFIsolationSumProducerForPUPPI::kNPFTypes
static constexpr unsigned kNPFTypes
Definition: CITKPFIsolationSumProducerForPUPPI.cc:40
Skims_PA_cff.name
name
Definition: Skims_PA_cff.py:17
reco::PFCandidate
Particle reconstructed by the particle flow algorithm.
Definition: PFCandidate.h:41
edm::helper::Filler
Definition: ValueMap.h:22
citk::PFIsolationSumProducerForPUPPI::puppiValueMapToken_
edm::EDGetTokenT< edm::ValueMap< float > > puppiValueMapToken_
Definition: CITKPFIsolationSumProducerForPUPPI.cc:46
dqmiolumiharvest.j
j
Definition: dqmiolumiharvest.py:66
edm::View::ptrAt
Ptr< value_type > ptrAt(size_type i) const
edm::Ptr::isNull
bool isNull() const
Checks for null.
Definition: Ptr.h:142
edm::InputTag
Definition: InputTag.h:15
citk::PFIsolationSumProducerForPUPPI::_isolate_with
edm::EDGetTokenT< CandView > _isolate_with
Definition: CITKPFIsolationSumProducerForPUPPI.cc:45
pat::PackedCandidate::puppiWeightNoLep
float puppiWeightNoLep() const
Weight from full PUPPI.
Definition: PackedCandidate.cc:384