#include <PF_PU_FirstVertexTracks.h>

Inheritance diagram for PF_PU_FirstVertexTracks:

Public Member Functions
	PF_PU_FirstVertexTracks (const edm::ParameterSet &)

	~PF_PU_FirstVertexTracks ()

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

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 &)

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

Detailed Description

Definition at line 47 of file PF_PU_FirstVertexTracks.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 53 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 ( )

Definition at line 89 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

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

static

Definition at line 233 of file PF_PU_FirstVertexTracks.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_FirstVertexTracks::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Implements edm::EDProducer.

Definition at line 104 of file PF_PU_FirstVertexTracks.cc.

References gather_cfg::cout, edm::first(), edm::Event::getByToken(), edm::helpers::KeyVal< K, V >::key, edm::Event::put(), edm::second(), findQualityFiles::size, and edm::helpers::KeyVal< K, V >::val.

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

 {
 
     auto_ptr<TrackCollection> t2v_firstvertextracks(new TrackCollection() );
     auto_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( 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( v2t_firstvertextracks, "V2T" );
 
     }
 
 }

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

privatevirtual

Definition at line 218 of file PF_PU_FirstVertexTracks.cc.

References eta(), AlCaHLTBitMon_ParallelJobs::p, and phi.

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

Member Data Documentation

edm::InputTag PF_PU_FirstVertexTracks::input_AssociationType_

private

Definition at line 60 of file PF_PU_FirstVertexTracks.h.

int PF_PU_FirstVertexTracks::input_MinQuality_

private

Definition at line 67 of file PF_PU_FirstVertexTracks.h.

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

private

Definition at line 64 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<TrackToVertexAssMap> PF_PU_FirstVertexTracks::token_TrackToVertexAssMap_

private

Definition at line 62 of file PF_PU_FirstVertexTracks.h.

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

private

Definition at line 65 of file PF_PU_FirstVertexTracks.h.

edm::EDGetTokenT<VertexToTrackAssMap> PF_PU_FirstVertexTracks::token_VertexToTrackAssMap_

private

Definition at line 63 of file PF_PU_FirstVertexTracks.h.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation