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 &)
	~PF_PU_FirstVertexTracks () override
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
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
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
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
std::vector< ESProxyIndex > 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
bool	registeredToConsumeMany (TypeID const &, 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::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	wantsInputProcessBlocks ()
static bool	wantsProcessBlocks ()
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::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::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
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 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<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;
      }
    }
  }
}

~PF_PU_FirstVertexTracks()

PF_PU_FirstVertexTracks::~PF_PU_FirstVertexTracks ( )

override

Definition at line 85 of file PF_PU_FirstVertexTracks.cc.

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

Member Function Documentation

fillDescriptions()

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

static

Definition at line 191 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;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

produce()

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

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 95 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(), findQualityFiles::size, 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  )

privatevirtual

Definition at line 185 of file PF_PU_FirstVertexTracks.cc.

References hltPixelTracks_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 62 of file PF_PU_FirstVertexTracks.h.

input_MinQuality_

int PF_PU_FirstVertexTracks::input_MinQuality_

private

Definition at line 69 of file PF_PU_FirstVertexTracks.h.

token_generalTracksCollection_

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

private

Definition at line 66 of file PF_PU_FirstVertexTracks.h.

token_TrackToVertexAssMap_

edm::EDGetTokenT<TrackToVertexAssMap> PF_PU_FirstVertexTracks::token_TrackToVertexAssMap_

private

Definition at line 64 of file PF_PU_FirstVertexTracks.h.

token_VertexCollection_

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

private

Definition at line 67 of file PF_PU_FirstVertexTracks.h.

token_VertexToTrackAssMap_

edm::EDGetTokenT<VertexToTrackAssMap> PF_PU_FirstVertexTracks::token_VertexToTrackAssMap_

private

Definition at line 65 of file PF_PU_FirstVertexTracks.h.