PF_PU_FirstVertexTracks Class Reference

#include <PF_PU_FirstVertexTracks.h>

Inheritance diagram for PF_PU_FirstVertexTracks:

Public Member Functions
	PF_PU_FirstVertexTracks (const edm::ParameterSet &)
Public Member Functions inherited from edm::global::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
EDProducer &	operator= (const EDProducer &)=delete
bool	wantsGlobalLuminosityBlocks () const final
bool	wantsGlobalRuns () const final
bool	wantsInputProcessBlocks () const final
bool	wantsProcessBlocks () const final
bool	wantsStreamLuminosityBlocks () const final
bool	wantsStreamRuns () const final
Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
	~EDProducerBase () 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
std::vector< bool > const &	recordProvenanceList () 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)
TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...
	~ProducerBase () noexcept(false) override
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () 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
ESResolverIndex const *	esGetTokenIndices (edm::Transition iTrans) const
std::vector< ESResolverIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const
std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (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::array< std::vector< ModuleDescription const *> *, NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const *> const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)
virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
void	produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override
bool	trackMatch (const reco::Track &, const reco::Track &) const

Private Attributes
edm::InputTag	input_AssociationType_
int	input_MinQuality_
edm::EDGetTokenT< reco::TrackCollection >	token_generalTracksCollection_
edm::EDGetTokenT< TrackToVertexAssMap >	token_TrackToVertexAssMap_
edm::EDGetTokenT< reco::VertexCollection >	token_VertexCollection_
edm::EDGetTokenT< VertexToTrackAssMap >	token_VertexToTrackAssMap_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType
Public Types inherited from edm::ProducerBase
template<typename T >
using	BranchAliasSetterT = ProductRegistryHelper::BranchAliasSetterT< T >
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
template<Transition Tr = Transition::Event>
auto	produces (std::string instanceName) noexcept
	declare what type of product will make and with which optional label More...
template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()
template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()
template<Transition Tr = Transition::Event>
auto	produces () noexcept
ProducesCollector	producesCollector ()
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept
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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()
template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)
template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...
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)
void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Definition at line 49 of file PF_PU_FirstVertexTracks.h.

Constructor & Destructor Documentation

PF_PU_FirstVertexTracks()

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().

                                                                               {
  //now do what ever other initialization is needed

  input_AssociationType_ = iConfig.getParameter<edm::InputTag>("AssociationType");

  token_TrackToVertexAssMap_ = mayConsume<TrackToVertexAssMap>(iConfig.getParameter<InputTag>("AssociationMap"));
  token_VertexToTrackAssMap_ = mayConsume<VertexToTrackAssMap>(iConfig.getParameter<InputTag>("AssociationMap"));

  token_generalTracksCollection_ = consumes<TrackCollection>(iConfig.getParameter<InputTag>("TrackCollection"));

  token_VertexCollection_ = mayConsume<VertexCollection>(iConfig.getParameter<InputTag>("VertexCollection"));

  input_MinQuality_ = iConfig.getParameter<int>("MinQuality");

  //register your products

  if (input_AssociationType_.label() == "TracksToVertex") {
    produces<TrackCollection>("T2V");
  } else {
    if (input_AssociationType_.label() == "VertexToTracks") {
      produces<TrackCollection>("V2T");
    } else {
      if (input_AssociationType_.label() == "Both") {
        produces<TrackCollection>("T2V");
        produces<TrackCollection>("V2T");
      } else {
        std::cout << "No correct InputTag for AssociationType!" << std::endl;
        std::cout << "Won't produce any TrackCollection!" << std::endl;
      }
    }
  }
}

Member Function Documentation

fillDescriptions()

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

static

Definition at line 186 of file PF_PU_FirstVertexTracks.cc.

References edm::ConfigurationDescriptions::addDefault(), and submitPVResolutionJobs::desc.

                                                                                         {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;

  using namespace edm;
  desc.add<InputTag>("AssociationType");
  desc.add<InputTag>("AssociationMap");
  desc.add<InputTag>("TrackCollection");
  desc.add<InputTag>("VertexCollection");
  desc.add<int>("MinQuality");

  descriptions.addDefault(desc);
}

produce()

void PF_PU_FirstVertexTracks::produce ( edm::StreamID  , edm::Event &  iEvent, const edm::EventSetup &  iSetup  ) const

overrideprivatevirtual

Implements edm::global::EDProducerBase.

Definition at line 90 of file PF_PU_FirstVertexTracks.cc.

References edm::AssociationMap< Tag >::begin(), gather_cfg::cout, edm::AssociationMap< Tag >::end(), edm::first(), iEvent, edm::helpers::KeyVal< K, V >::key, eostools::move(), quality, edm::second(), and edm::helpers::KeyVal< K, V >::val.

                                                                                                      {
  unique_ptr<TrackCollection> t2v_firstvertextracks(new TrackCollection());
  unique_ptr<TrackCollection> v2t_firstvertextracks(new TrackCollection());

  bool t2vassmap = false;
  bool v2tassmap = false;

  //get the input vertex<->general track association map
  Handle<TrackToVertexAssMap> t2vAM;
  Handle<VertexToTrackAssMap> v2tAM;

  string asstype = input_AssociationType_.label();

  if ((asstype == "TracksToVertex") || (asstype == "Both")) {
    if (iEvent.getByToken(token_TrackToVertexAssMap_, t2vAM)) {
      t2vassmap = true;
    }
  }

  if ((asstype == "VertexToTracks") || (asstype == "Both")) {
    if (iEvent.getByToken(token_VertexToTrackAssMap_, v2tAM)) {
      v2tassmap = true;
    }
  }

  if (!t2vassmap && !v2tassmap) {
    cout << "No input collection could be found" << endl;
    return;
  }

  //get the input track collection
  Handle<TrackCollection> input_trckcollH;
  iEvent.getByToken(token_generalTracksCollection_, input_trckcollH);

  if (t2vassmap) {
    const TrackQualityPairVector trckcoll = t2vAM->begin()->val;

    //get the tracks associated to the first vertex and store them in a track collection
    for (unsigned int trckcoll_ite = 0; trckcoll_ite < trckcoll.size(); trckcoll_ite++) {
      float quality = trckcoll[trckcoll_ite].second;

      if (quality >= input_MinQuality_) {
        TrackRef AMtrkref = trckcoll[trckcoll_ite].first;

        for (unsigned int index_input_trck = 0; index_input_trck < input_trckcollH->size(); index_input_trck++) {
          TrackRef input_trackref = TrackRef(input_trckcollH, index_input_trck);

          if (trackMatch(*AMtrkref, *input_trackref)) {
            t2v_firstvertextracks->push_back(*AMtrkref);
            break;
          }
        }
      }
    }

    iEvent.put(std::move(t2v_firstvertextracks), "T2V");
  }

  if (v2tassmap) {
    //get the input vertex collection
    Handle<VertexCollection> input_vtxcollH;
    iEvent.getByToken(token_VertexCollection_, input_vtxcollH);

    VertexRef firstVertexRef(input_vtxcollH, 0);

    VertexToTrackAssMap::const_iterator v2t_ite;

    for (v2t_ite = v2tAM->begin(); v2t_ite != v2tAM->end(); v2t_ite++) {
      TrackRef AMtrkref = v2t_ite->key;

      for (unsigned int index_input_trck = 0; index_input_trck < input_trckcollH->size(); index_input_trck++) {
        TrackRef input_trackref = TrackRef(input_trckcollH, index_input_trck);

        if (trackMatch(*AMtrkref, *input_trackref)) {
          for (unsigned v_ite = 0; v_ite < (v2t_ite->val).size(); v_ite++) {
            VertexRef vtxref = (v2t_ite->val)[v_ite].first;
            float quality = (v2t_ite->val)[v_ite].second;

            if ((vtxref == firstVertexRef) && (quality >= input_MinQuality_)) {
              v2t_firstvertextracks->push_back(*AMtrkref);
            }
          }
        }
      }
    }

    iEvent.put(std::move(v2t_firstvertextracks), "V2T");
  }
}

trackMatch()

bool PF_PU_FirstVertexTracks::trackMatch ( const reco::Track &  track1, const reco::Track &  track2  ) const

private

Definition at line 180 of file PF_PU_FirstVertexTracks.cc.

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

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

Member Data Documentation

input_AssociationType_

edm::InputTag PF_PU_FirstVertexTracks::input_AssociationType_

private

Definition at line 61 of file PF_PU_FirstVertexTracks.h.

input_MinQuality_

int PF_PU_FirstVertexTracks::input_MinQuality_

private

Definition at line 68 of file PF_PU_FirstVertexTracks.h.

token_generalTracksCollection_

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

private

Definition at line 65 of file PF_PU_FirstVertexTracks.h.

token_TrackToVertexAssMap_

edm::EDGetTokenT<TrackToVertexAssMap> PF_PU_FirstVertexTracks::token_TrackToVertexAssMap_

private

Definition at line 63 of file PF_PU_FirstVertexTracks.h.

token_VertexCollection_

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

private

Definition at line 66 of file PF_PU_FirstVertexTracks.h.

token_VertexToTrackAssMap_

edm::EDGetTokenT<VertexToTrackAssMap> PF_PU_FirstVertexTracks::token_VertexToTrackAssMap_

private

Definition at line 64 of file PF_PU_FirstVertexTracks.h.