PF_PU_AssoMap Class Reference

#include <CommonTools/RecoUtils/plugins/PFCand_NoPU_WithAM.cc>

Inheritance diagram for PF_PU_AssoMap:

Public Member Functions
	PF_PU_AssoMap (const edm::ParameterSet &)
	~PF_PU_AssoMap ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()
Public Member Functions inherited from PF_PU_AssoMapAlgos
std::auto_ptr < TrackToVertexAssMap >	CreateTrackToVertexMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
std::auto_ptr < VertexToTrackAssMap >	CreateVertexToTrackMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
virtual void	GetInputCollections (edm::Event &, const edm::EventSetup &)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &)
std::auto_ptr < TrackToVertexAssMap >	SortAssociationMap (TrackToVertexAssMap *)

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)

Private Member Functions
virtual void	produce (edm::Event &, const edm::EventSetup &)

Private Attributes
edm::InputTag	input_AssociationType_
edm::EDGetTokenT < reco::TrackCollection >	token_TrackCollection_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
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)
Protected Member Functions inherited from PF_PU_AssoMapAlgos
std::vector< reco::VertexRef > *	CreateVertexVector (edm::Handle< reco::VertexCollection >)
int	DefineQuality (int, int, double)
void	EraseVertex (std::vector< reco::VertexRef > *, reco::VertexRef)
VertexStepPair	FindAssociation (const reco::TrackRef &, std::vector< reco::VertexRef > *, edm::ESHandle< MagneticField >, const edm::EventSetup &, edm::Handle< reco::BeamSpot >, int)

Detailed Description

Description: Produces a map with association between tracks and their particular most probable vertex with a quality of this association

Description: Produces collection of tracks associated to the first vertex based on the pf_pu Association Map

Description: Produces a collection of PFCandidates associated to the first vertex based on the association map

Definition at line 41 of file PF_PU_AssoMap.h.

Constructor & Destructor Documentation

PF_PU_AssoMap::PF_PU_AssoMap ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 26 of file PF_PU_AssoMap.cc.

References gather_cfg::cout, edm::ParameterSet::getParameter(), input_AssociationType_, edm::InputTag::label(), and token_TrackCollection_.

                                                          :PF_PU_AssoMapAlgos(iConfig, consumesCollector())
{

   //now do what ever other initialization is needed

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

    token_TrackCollection_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("TrackCollection"));

   //register your products

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


}

PF_PU_AssoMap::~PF_PU_AssoMap

(

)

Definition at line 57 of file PF_PU_AssoMap.cc.

{

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

}

Member Function Documentation

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

static

Definition at line 102 of file PF_PU_AssoMap.cc.

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

                                                                          {
  //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);
}

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

privatevirtual

Implements edm::EDProducer.

Definition at line 72 of file PF_PU_AssoMap.cc.

References PF_PU_AssoMapAlgos::CreateTrackToVertexMap(), PF_PU_AssoMapAlgos::CreateVertexToTrackMap(), edm::Event::getByToken(), PF_PU_AssoMapAlgos::GetInputCollections(), input_AssociationType_, edm::InputTag::label(), edm::Event::put(), dt_dqm_sourceclient_common_cff::reco, PF_PU_AssoMapAlgos::SortAssociationMap(), and token_TrackCollection_.

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

{

  using namespace edm;
  using namespace std;
  using namespace reco;


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

    string asstype = input_AssociationType_.label();

    PF_PU_AssoMapAlgos::GetInputCollections(iEvent,iSetup);

    if ( ( asstype == "TracksToVertex" ) || ( asstype == "Both" ) ) {
      auto_ptr<TrackToVertexAssMap> Track2Vertex = CreateTrackToVertexMap(trkcollH, iSetup);
      iEvent.put( SortAssociationMap( &(*Track2Vertex) ) );
    }

    if ( ( asstype == "VertexToTracks" ) || ( asstype == "Both" ) ) {
      auto_ptr<VertexToTrackAssMap> Vertex2Track = CreateVertexToTrackMap(trkcollH, iSetup);
      iEvent.put( Vertex2Track );
    }

}

Member Data Documentation

edm::InputTag PF_PU_AssoMap::input_AssociationType_

private

Definition at line 53 of file PF_PU_AssoMap.h.

Referenced by PF_PU_AssoMap(), and produce().

edm::EDGetTokenT<reco::TrackCollection> PF_PU_AssoMap::token_TrackCollection_

private

Definition at line 55 of file PF_PU_AssoMap.h.

Referenced by PF_PU_AssoMap(), and produce().