CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
PF_PU_FirstVertexTracks Class Reference

#include <PF_PU_FirstVertexTracks.h>

Inheritance diagram for PF_PU_FirstVertexTracks:
edm::EDProducer edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 PF_PU_FirstVertexTracks (const edm::ParameterSet &)
 
 ~PF_PU_FirstVertexTracks () override
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
SerialTaskQueueglobalLuminosityBlocksQueue ()
 
SerialTaskQueueglobalRunsQueue ()
 
ModuleDescription const & moduleDescription () const
 
 ~EDProducer () override
 
- 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 ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
 
 ~ProducerBase () noexcept(false) override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ESProxyIndex const * esGetTokenIndices (edm::Transition iTrans) const
 
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 & itemsToGetFrom (BranchType iType) 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
 
ProductResolverIndexAndSkipBit uncheckedIndexFrom (EDGetToken) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
void updateLookup (eventsetup::ESRecordsToProxyIndices const &)
 
virtual ~EDConsumerBase () noexcept(false)
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 
- Static Public Member Functions inherited from edm::EDProducer
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
static bool wantsGlobalLuminosityBlocks ()
 
static bool wantsGlobalRuns ()
 
static bool wantsStreamLuminosityBlocks ()
 
static bool wantsStreamRuns ()
 

Private Member Functions

void produce (edm::Event &, const edm::EventSetup &) override
 
virtual bool TrackMatch (const reco::Track &, const reco::Track &)
 

Private Attributes

edm::InputTag input_AssociationType_
 
int input_MinQuality_
 
edm::EDGetTokenT< reco::TrackCollectiontoken_generalTracksCollection_
 
edm::EDGetTokenT< TrackToVertexAssMaptoken_TrackToVertexAssMap_
 
edm::EDGetTokenT< reco::VertexCollectiontoken_VertexCollection_
 
edm::EDGetTokenT< VertexToTrackAssMaptoken_VertexToTrackAssMap_
 

Additional Inherited Members

- Public Types inherited from edm::EDProducer
typedef EDProducer ModuleType
 
- Public Types inherited from edm::ProducerBase
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
 
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Protected Member Functions inherited from edm::ProducerBase
ProducesCollector producesCollector ()
 
- 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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes ()
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes (ESInputTag const &tag)
 
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

Definition at line 49 of file PF_PU_FirstVertexTracks.h.

Constructor & Destructor Documentation

PF_PU_FirstVertexTracks::PF_PU_FirstVertexTracks ( const edm::ParameterSet iConfig)
explicit

Definition at line 52 of file PF_PU_FirstVertexTracks.cc.

References gather_cfg::cout, and edm::ParameterSet::getParameter().

52  {
53  //now do what ever other initialization is needed
54 
55  input_AssociationType_ = iConfig.getParameter<edm::InputTag>("AssociationType");
56 
57  token_TrackToVertexAssMap_ = mayConsume<TrackToVertexAssMap>(iConfig.getParameter<InputTag>("AssociationMap"));
58  token_VertexToTrackAssMap_ = mayConsume<VertexToTrackAssMap>(iConfig.getParameter<InputTag>("AssociationMap"));
59 
60  token_generalTracksCollection_ = consumes<TrackCollection>(iConfig.getParameter<InputTag>("TrackCollection"));
61 
62  token_VertexCollection_ = mayConsume<VertexCollection>(iConfig.getParameter<InputTag>("VertexCollection"));
63 
64  input_MinQuality_ = iConfig.getParameter<int>("MinQuality");
65 
66  //register your products
67 
68  if (input_AssociationType_.label() == "TracksToVertex") {
69  produces<TrackCollection>("T2V");
70  } else {
71  if (input_AssociationType_.label() == "VertexToTracks") {
72  produces<TrackCollection>("V2T");
73  } else {
74  if (input_AssociationType_.label() == "Both") {
75  produces<TrackCollection>("T2V");
76  produces<TrackCollection>("V2T");
77  } else {
78  std::cout << "No correct InputTag for AssociationType!" << std::endl;
79  std::cout << "Won't produce any TrackCollection!" << std::endl;
80  }
81  }
82  }
83 }
T getParameter(std::string const &) const
edm::EDGetTokenT< VertexToTrackAssMap > token_VertexToTrackAssMap_
edm::EDGetTokenT< reco::VertexCollection > token_VertexCollection_
std::string const & label() const
Definition: InputTag.h:36
edm::EDGetTokenT< TrackToVertexAssMap > token_TrackToVertexAssMap_
edm::EDGetTokenT< reco::TrackCollection > token_generalTracksCollection_
PF_PU_FirstVertexTracks::~PF_PU_FirstVertexTracks ( )
override

Definition at line 85 of file PF_PU_FirstVertexTracks.cc.

85  {
86  // do anything here that needs to be done at desctruction time
87  // (e.g. close files, deallocate resources etc.)
88 }

Member Function Documentation

void PF_PU_FirstVertexTracks::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 191 of file PF_PU_FirstVertexTracks.cc.

References edm::ConfigurationDescriptions::addDefault(), DEFINE_FWK_MODULE, and edm::ParameterSetDescription::setUnknown().

191  {
192  //The following says we do not know what parameters are allowed so do no validation
193  // Please change this to state exactly what you do use, even if it is no parameters
195  desc.setUnknown();
196  descriptions.addDefault(desc);
197 }
void addDefault(ParameterSetDescription const &psetDescription)
void PF_PU_FirstVertexTracks::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 95 of file PF_PU_FirstVertexTracks.cc.

References edm::AssociationMap< Tag >::begin(), gather_cfg::cout, edm::AssociationMap< Tag >::end(), edm::first(), edm::Event::getByToken(), edm::helpers::KeyVal< K, V >::key, eostools::move(), edm::Event::put(), qcdUeDQM_cfi::quality, edm::second(), findQualityFiles::size, and edm::helpers::KeyVal< K, V >::val.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

95  {
96  unique_ptr<TrackCollection> t2v_firstvertextracks(new TrackCollection());
97  unique_ptr<TrackCollection> v2t_firstvertextracks(new TrackCollection());
98 
99  bool t2vassmap = false;
100  bool v2tassmap = false;
101 
102  //get the input vertex<->general track association map
105 
106  string asstype = input_AssociationType_.label();
107 
108  if ((asstype == "TracksToVertex") || (asstype == "Both")) {
109  if (iEvent.getByToken(token_TrackToVertexAssMap_, t2vAM)) {
110  t2vassmap = true;
111  }
112  }
113 
114  if ((asstype == "VertexToTracks") || (asstype == "Both")) {
115  if (iEvent.getByToken(token_VertexToTrackAssMap_, v2tAM)) {
116  v2tassmap = true;
117  }
118  }
119 
120  if (!t2vassmap && !v2tassmap) {
121  cout << "No input collection could be found" << endl;
122  return;
123  }
124 
125  //get the input track collection
126  Handle<TrackCollection> input_trckcollH;
127  iEvent.getByToken(token_generalTracksCollection_, input_trckcollH);
128 
129  if (t2vassmap) {
130  const TrackQualityPairVector trckcoll = t2vAM->begin()->val;
131 
132  //get the tracks associated to the first vertex and store them in a track collection
133  for (unsigned int trckcoll_ite = 0; trckcoll_ite < trckcoll.size(); trckcoll_ite++) {
134  float quality = trckcoll[trckcoll_ite].second;
135 
136  if (quality >= input_MinQuality_) {
137  TrackRef AMtrkref = trckcoll[trckcoll_ite].first;
138 
139  for (unsigned int index_input_trck = 0; index_input_trck < input_trckcollH->size(); index_input_trck++) {
140  TrackRef input_trackref = TrackRef(input_trckcollH, index_input_trck);
141 
142  if (TrackMatch(*AMtrkref, *input_trackref)) {
143  t2v_firstvertextracks->push_back(*AMtrkref);
144  break;
145  }
146  }
147  }
148  }
149 
150  iEvent.put(std::move(t2v_firstvertextracks), "T2V");
151  }
152 
153  if (v2tassmap) {
154  //get the input vertex collection
155  Handle<VertexCollection> input_vtxcollH;
156  iEvent.getByToken(token_VertexCollection_, input_vtxcollH);
157 
158  VertexRef firstVertexRef(input_vtxcollH, 0);
159 
161 
162  for (v2t_ite = v2tAM->begin(); v2t_ite != v2tAM->end(); v2t_ite++) {
163  TrackRef AMtrkref = v2t_ite->key;
164 
165  for (unsigned int index_input_trck = 0; index_input_trck < input_trckcollH->size(); index_input_trck++) {
166  TrackRef input_trackref = TrackRef(input_trckcollH, index_input_trck);
167 
168  if (TrackMatch(*AMtrkref, *input_trackref)) {
169  for (unsigned v_ite = 0; v_ite < (v2t_ite->val).size(); v_ite++) {
170  VertexRef vtxref = (v2t_ite->val)[v_ite].first;
171  float quality = (v2t_ite->val)[v_ite].second;
172 
173  if ((vtxref == firstVertexRef) && (quality >= input_MinQuality_)) {
174  v2t_firstvertextracks->push_back(*AMtrkref);
175  }
176  }
177  }
178  }
179  }
180 
181  iEvent.put(std::move(v2t_firstvertextracks), "V2T");
182  }
183 }
size
Write out results.
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:131
const_iterator end() const
last iterator over the map (read only)
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:525
edm::EDGetTokenT< VertexToTrackAssMap > token_VertexToTrackAssMap_
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:14
U second(std::pair< T, U > const &p)
edm::EDGetTokenT< reco::VertexCollection > token_VertexCollection_
std::vector< TrackQualityPair > TrackQualityPairVector
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:20
std::string const & label() const
Definition: InputTag.h:36
edm::EDGetTokenT< TrackToVertexAssMap > token_TrackToVertexAssMap_
edm::EDGetTokenT< reco::TrackCollection > token_generalTracksCollection_
const_iterator begin() const
first iterator over the map (read only)
def move(src, dest)
Definition: eostools.py:511
virtual bool TrackMatch(const reco::Track &, const reco::Track &)
bool PF_PU_FirstVertexTracks::TrackMatch ( const reco::Track track1,
const reco::Track track2 
)
privatevirtual

Definition at line 185 of file PF_PU_FirstVertexTracks.cc.

References hltPixelTracks_cff::chi2, PVValHelper::eta, ndof, and AlCaHLTBitMon_ParallelJobs::p.

185  {
186  return ((track1).eta() == (track2).eta() && (track1).phi() == (track2).phi() && (track1).chi2() == (track2).chi2() &&
187  (track1).ndof() == (track2).ndof() && (track1).p() == (track2).p());
188 }

Member Data Documentation

edm::InputTag PF_PU_FirstVertexTracks::input_AssociationType_
private

Definition at line 62 of file PF_PU_FirstVertexTracks.h.

int PF_PU_FirstVertexTracks::input_MinQuality_
private

Definition at line 69 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<reco::TrackCollection> PF_PU_FirstVertexTracks::token_generalTracksCollection_
private

Definition at line 66 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<TrackToVertexAssMap> PF_PU_FirstVertexTracks::token_TrackToVertexAssMap_
private

Definition at line 64 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<reco::VertexCollection> PF_PU_FirstVertexTracks::token_VertexCollection_
private

Definition at line 67 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<VertexToTrackAssMap> PF_PU_FirstVertexTracks::token_VertexToTrackAssMap_
private

Definition at line 65 of file PF_PU_FirstVertexTracks.h.