CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init > Class Template Reference

#include <ConstrainedFitCandProducer.h>

Inheritance diagram for ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >:
edm::EDProducer edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 ConstrainedFitCandProducer (const edm::ParameterSet &)
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducer ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFromEvent () const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

void produce (edm::Event &, const edm::EventSetup &)
 

Private Attributes

Fitter fitter_
 
int pdgId_
 
bool setLongLived_
 
bool setMassConstraint_
 
bool setPdgId_
 
edm::EDGetTokenT< InputCollection > srcToken_
 

Additional Inherited Members

- Public Types inherited from edm::EDProducer
typedef EDProducer ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::EDProducer
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

template<typename Fitter, typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
class ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >

Definition at line 20 of file ConstrainedFitCandProducer.h.

Constructor & Destructor Documentation

template<typename Fitter , typename InputCollection , typename OutputCollection , typename Init >
ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::ConstrainedFitCandProducer ( const edm::ParameterSet cfg)
explicit

Definition at line 44 of file ConstrainedFitCandProducer.h.

References egammaCTFFinalFitWithMaterial_cff::alias, spr::find(), runEdmFileComparison::found, edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterNamesForType(), ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::pdgId_, ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setLongLived_, ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setMassConstraint_, ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setPdgId_, and AlCaHLTBitMon_QueryRunRegistry::string.

44  :
45  srcToken_(consumes<InputCollection>(cfg.template getParameter<edm::InputTag>("src"))),
46  setLongLived_(false), setMassConstraint_(false), setPdgId_(false),
47  fitter_(reco::modules::make<Fitter>(cfg)) {
48  produces<OutputCollection>();
49  std::string alias( cfg.getParameter<std::string>("@module_label"));
50  const std::string setLongLived("setLongLived");
51  std::vector<std::string> vBoolParams = cfg.template getParameterNamesForType<bool>();
52  bool found = find(vBoolParams.begin(), vBoolParams.end(), setLongLived) != vBoolParams.end();
53  if(found) setLongLived_ = cfg.template getParameter<bool>("setLongLived");
54  const std::string setMassConstraint("setMassConstraint");
55  found = find(vBoolParams.begin(), vBoolParams.end(), setMassConstraint) != vBoolParams.end();
56  if(found) setMassConstraint_ = cfg.template getParameter<bool>("setMassConstraint");
57  const std::string setPdgId("setPdgId");
58  std::vector<std::string> vIntParams = cfg.getParameterNamesForType<int>();
59  found = find(vIntParams.begin(), vIntParams.end(), setPdgId) != vIntParams.end();
60  if(found) { setPdgId_ = true; pdgId_ = cfg.getParameter<int>("setPdgId"); }
61 }
T getParameter(std::string const &) const
edm::EDGetTokenT< InputCollection > srcToken_
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:20
std::vector< std::string > getParameterNamesForType(bool trackiness=true) const
Definition: ParameterSet.h:194

Member Function Documentation

template<typename Fitter , typename InputCollection , typename OutputCollection , typename Init >
void ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::produce ( edm::Event evt,
const edm::EventSetup es 
)
private

Definition at line 83 of file ConstrainedFitCandProducer.h.

References reco::fitHelper::add(), EnergyCorrector::c, clone(), ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::fitter_, edm::Event::getByToken(), StGenEvtProducer_cfi::init, eostools::move(), ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::pdgId_, edm::Event::put(), ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setLongLived_, ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setMassConstraint_, ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setPdgId_, and ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::srcToken_.

83  {
84  Init::init(fitter_, evt, es);
86  evt.getByToken(srcToken_, cands);
87  auto fitted = std::make_unique<OutputCollection>();
88  fitted->reserve(cands->size());
89  for(typename InputCollection::const_iterator c = cands->begin(); c != cands->end(); ++ c) {
90  auto clone = std::make_unique<reco::VertexCompositeCandidate>(*c);
91  fitter_.set(*clone);
92  if(setLongLived_) clone->setLongLived();
93  if(setMassConstraint_) clone->setMassConstraint();
94  if(setPdgId_) clone->setPdgId(pdgId_);
95  reco::fitHelper::add(fitted.get(), std::move(clone));
96  }
97  evt.put(std::move(fitted));
98 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:457
edm::EDGetTokenT< InputCollection > srcToken_
void add(C *c, std::unique_ptr< reco::VertexCompositeCandidate > t)
TEveGeoShape * clone(const TEveElement *element, TEveElement *parent)
Definition: eve_macros.cc:135
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
Fitter ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::fitter_
private
template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
int ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::pdgId_
private
template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
bool ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setLongLived_
private
template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
bool ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setMassConstraint_
private
template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
bool ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::setPdgId_
private
template<typename Fitter , typename InputCollection = reco::CandidateCollection, typename OutputCollection = InputCollection, typename Init = typename ::reco::modules::EventSetupInit<Fitter>::type>
edm::EDGetTokenT<InputCollection> ConstrainedFitCandProducer< Fitter, InputCollection, OutputCollection, Init >::srcToken_
private