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 ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
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 ()
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, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
virtual	~ProducerBase () noexcept(false)
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
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
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
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)

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
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
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(), DEFINE_FWK_MODULE, 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

Definition at line 104 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(), btvTracks_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().

{

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

    }

}

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 HiEvtPlane_cfi::chi2, stringResolutionProvider_cfi::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

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.