VertexMerger.cc

Go to the documentation of this file.

#include <memory>
#include <set>

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "RecoVertex/VertexTools/interface/VertexDistance3D.h"
#include "RecoVertex/VertexTools/interface/SharedTracks.h"
#include "DataFormats/Candidate/interface/VertexCompositePtrCandidate.h"
#include "RecoVertex/VertexPrimitives/interface/ConvertToFromReco.h"
#include "RecoVertex/VertexPrimitives/interface/VertexState.h"

template <class VTX>
class TemplatedVertexMerger : public edm::stream::EDProducer<> {
    public:
    typedef std::vector<VTX> Product;
    TemplatedVertexMerger(const edm::ParameterSet &params);

    void produce(edm::Event &event, const edm::EventSetup &es) override;

    private:
    bool trackFilter(const reco::TrackRef &track) const;

    edm::EDGetTokenT<Product>           token_secondaryVertex;
    double                  maxFraction;
    double                  minSignificance;
};

template <class VTX>
TemplatedVertexMerger<VTX>::TemplatedVertexMerger(const edm::ParameterSet &params) :
    maxFraction(params.getParameter<double>("maxFraction")),
    minSignificance(params.getParameter<double>("minSignificance"))
{
    token_secondaryVertex = consumes<Product>(params.getParameter<edm::InputTag>("secondaryVertices"));
    produces<Product>();
}

template <class VTX>
void TemplatedVertexMerger<VTX>::produce(edm::Event &event, const edm::EventSetup &es)
{
    using namespace reco;

    edm::Handle<Product> secondaryVertices;
    event.getByToken(token_secondaryVertex, secondaryVertices);

        VertexDistance3D dist;
    auto recoVertices = std::make_unique<Product>();
    for(typename Product::const_iterator sv = secondaryVertices->begin();
        sv != secondaryVertices->end(); ++sv) {
          recoVertices->push_back(*sv);
        }
       for(typename Product::iterator sv = recoVertices->begin();
        sv != recoVertices->end(); ++sv) {

        bool shared=false;
    VertexState s1(RecoVertex::convertPos(sv->position()),RecoVertex::convertError(sv->error()));
        for(typename Product::iterator sv2 = recoVertices->begin();
        sv2 != recoVertices->end(); ++sv2) {
          VertexState s2(RecoVertex::convertPos(sv2->position()),RecoVertex::convertError(sv2->error()));
                  double fr=vertexTools::computeSharedTracks(*sv2, *sv);
        //        std::cout << sv2-recoVertices->begin() << " vs " << sv-recoVertices->begin() << " : " << fr << " "  <<  computeSharedTracks(*sv, *sv2) << " sig " << dist.distance(*sv,*sv2).significance() << std::endl;
          //      std::cout << (fr > maxFraction) << " && " << (dist.distance(*sv,*sv2).significance() < 2)  <<  " && " <<  (sv-sv2!=0)  << " && " <<  (fr >= computeSharedTracks(*sv2, *sv))  << std::endl;
        if (fr > maxFraction && dist.distance(s1,s2).significance() < minSignificance && sv-sv2!=0 
                    && fr >= vertexTools::computeSharedTracks(*sv, *sv2) )
          {
                      shared=true; 
                     // std::cout << "shared " << sv-recoVertices->begin() << " and "  << sv2-recoVertices->begin() << " fractions: " << fr << " , "  << computeSharedTracks(*sv2, *sv) << " sig: " <<  dist.distance(*sv,*sv2).significance() <<  std::endl;
         
                  }
                 

    }
        if(shared) { sv=recoVertices->erase(sv)-1; }
     //    std::cout << "it = " <<  sv-recoVertices->begin() << " new size is: " << recoVertices->size() <<   std::endl;
       }

    event.put(std::move(recoVertices));



}

typedef TemplatedVertexMerger<reco::Vertex> VertexMerger;
typedef TemplatedVertexMerger<reco::VertexCompositePtrCandidate> CandidateVertexMerger;

DEFINE_FWK_MODULE(VertexMerger);
DEFINE_FWK_MODULE(CandidateVertexMerger);