pat::PATTrackAndVertexUnpacker Class Reference

Inheritance diagram for pat::PATTrackAndVertexUnpacker:

Public Member Functions
	PATTrackAndVertexUnpacker (const edm::ParameterSet &iConfig)
virtual void	produce (edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override
	~PATTrackAndVertexUnpacker ()
Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default
Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducerBase ()
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)

Private Types
typedef std::vector< edm::InputTag >	VInputTag

Private Attributes
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	AdditionalTracks_
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	Cands_
const edm::EDGetTokenT< reco::VertexCollection >	PVs_
const edm::EDGetTokenT< reco::VertexCompositePtrCandidateCollection >	SVs_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
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 28 of file TrackAndVertexUnpacker.cc.

Member Typedef Documentation

typedef std::vector<edm::InputTag> pat::PATTrackAndVertexUnpacker::VInputTag

private

Definition at line 39 of file TrackAndVertexUnpacker.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 53 of file TrackAndVertexUnpacker.cc.

                                                                                   :
  Cands_(consumes< std::vector<pat::PackedCandidate> >(iConfig.getParameter<edm::InputTag>("packedCandidates"))),
  PVs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("slimmedVertices"))),
  SVs_(consumes<reco::VertexCompositePtrCandidateCollection>(iConfig.getParameter<edm::InputTag>("slimmedSecondaryVertices"))),
  AdditionalTracks_(consumes<pat::PackedCandidateCollection>(iConfig.getParameter<edm::InputTag>("additionalTracks")))
{
    produces<reco::TrackCollection>();
    produces<reco::VertexCollection>();
    produces<reco::VertexCollection>("secondary");
}

PATTrackAndVertexUnpacker::~PATTrackAndVertexUnpacker

(

)

Definition at line 65 of file TrackAndVertexUnpacker.cc.

                                                      {
}

Member Function Documentation

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

overridevirtual

Definition at line 69 of file TrackAndVertexUnpacker.cc.

References reco::Vertex::add(), AdditionalTracks_, EnergyCorrector::c, Cands_, pat::PackedCandidate::charge(), reco::CompositePtrCandidate::daughterPtr(), DEFINE_FWK_MODULE, pat::PackedCandidate::fromPV(), edm::Event::getByToken(), mps_fire::i, edm::HandleBase::id(), edm::Ptr< T >::id(), edm::Ptr< T >::key(), eostools::move(), reco::CompositePtrCandidate::numberOfDaughters(), pat::PackedCandidate::numberOfHits(), pat::PackedCandidate::pseudoTrack(), edm::Event::put(), MetAnalyzer::pv(), PVs_, pat::PackedCandidate::PVUsedInFit, alignCSCRings::r, findQualityFiles::rr, SVs_, reco::LeafCandidate::vertex(), reco::VertexCompositePtrCandidate::vertexChi2(), reco::VertexCompositePtrCandidate::vertexCovariance(), and reco::VertexCompositePtrCandidate::vertexNdof().

                                                                                                          {
    using namespace edm; using namespace std; using namespace reco;
    Handle<std::vector<pat::PackedCandidate> > cands;
    iEvent.getByToken(Cands_, cands);
    Handle<VertexCollection> pvs;
    iEvent.getByToken(PVs_, pvs);
    Handle<VertexCompositePtrCandidateCollection> svs;
    iEvent.getByToken(SVs_, svs);
    Handle<std::vector<pat::PackedCandidate> > addTracks;
    iEvent.getByToken(AdditionalTracks_, addTracks);

    auto outTks = std::make_unique<std::vector<reco::Track>>();
    std::map<unsigned int, std::vector<unsigned int> > asso;
    std::map<unsigned int, unsigned int> trackKeys;
    unsigned int j=0;
    for(unsigned int i=0;i<cands->size();i++)   {
        const pat::PackedCandidate & c = (*cands)[i];
        if(c.charge() != 0 && c.numberOfHits()> 0){
            outTks->push_back(c.pseudoTrack());
            for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
                if(c.fromPV(ipv)==pat::PackedCandidate::PVUsedInFit)
                    asso[ipv].push_back(j);
            }
            trackKeys[i]=j;
            j++;
        }   
    }

    int offsetAdd=j;
    for(unsigned int i = 0; i < addTracks->size(); i++) {
        outTks->push_back((*addTracks)[i].pseudoTrack());
        for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
            if((*addTracks)[i].fromPV(ipv)==pat::PackedCandidate::PVUsedInFit)
                asso[ipv].push_back(j);
        }
        j++;
    }
    
    edm::OrphanHandle< std::vector<reco::Track>  > oh = iEvent.put(std::move(outTks));
    
    auto outPv = std::make_unique<std::vector<reco::Vertex>>();
    
    for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
        reco::Vertex  pv = (*pvs)[ipv];
        for(unsigned int i=0;i<asso[ipv].size();i++)
        {
            TrackRef r(oh,asso[ipv][i]);
            TrackBaseRef rr(r);
            pv.add(rr);
        }
        outPv->push_back(pv);
    }
    iEvent.put(std::move(outPv));

        auto outSv = std::make_unique<std::vector<reco::Vertex>>();
    for(size_t i=0;i< svs->size(); i++) {
        const reco::VertexCompositePtrCandidate &sv = (*svs)[i];    
        outSv->push_back(reco::Vertex(sv.vertex(),sv.vertexCovariance(),sv.vertexChi2(),sv.vertexNdof(),0));
        for(size_t j=0;j<sv.numberOfDaughters();j++){
                    TrackRef r;
            if(sv.daughterPtr(j).id() == cands.id()) {
                         r= TrackRef(oh,trackKeys[sv.daughterPtr(j).key()]); // use trackKeys because cand->track has gaps from neutral
            } else {
//              std::cout << "vertex " << i << " using lost Track " << sv.daughterPtr(j).key()  << "  " << offsetAdd+sv.daughterPtr(j).key() << std::endl;  
                                r=TrackRef(oh,offsetAdd+sv.daughterPtr(j).key());  // use directly the key because addTracks is only charged
            }
                    TrackBaseRef rr(r);
            outSv->back().add(rr);

        }   
    }   

       iEvent.put(std::move(outSv),"secondary");

}

Member Data Documentation

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > pat::PATTrackAndVertexUnpacker::AdditionalTracks_

private

Definition at line 44 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > pat::PATTrackAndVertexUnpacker::Cands_

private

Definition at line 41 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::VertexCollection> pat::PATTrackAndVertexUnpacker::PVs_

private

Definition at line 42 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::VertexCompositePtrCandidateCollection> pat::PATTrackAndVertexUnpacker::SVs_

private

Definition at line 43 of file TrackAndVertexUnpacker.cc.

Referenced by produce().