/data/refman/pasoursint/CMSSW_5_2_9/src/RecoJets/JetAssociationProducers/src/JetSignalVertexCompatibility.cc

Go to the documentation of this file.

#include <memory>

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

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/JetTracksAssociation.h"
#include "DataFormats/JetReco/interface/JetFloatAssociation.h"

#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"

#include "RecoJets/JetAssociationAlgorithms/interface/JetSignalVertexCompatibilityAlgo.h"

#include "JetSignalVertexCompatibility.h"

using namespace reco;

JetSignalVertexCompatibility::JetSignalVertexCompatibility(
                                        const edm::ParameterSet &params) :
        algo(params.getParameter<double>("cut"),
             params.getParameter<double>("temperature")),
        jetTracksAssocLabel(params.getParameter<edm::InputTag>("jetTracksAssoc")),
        primaryVerticesLabel(params.getParameter<edm::InputTag>("primaryVertices"))
{
        produces<JetFloatAssociation::Container>();
}

JetSignalVertexCompatibility::~JetSignalVertexCompatibility()
{
}

void JetSignalVertexCompatibility::produce(edm::Event &event,
                                           const edm::EventSetup &es)
{
        edm::ESHandle<TransientTrackBuilder> trackBuilder;
        es.get<TransientTrackRecord>().get("TransientTrackBuilder",
                                           trackBuilder);

        algo.resetEvent(trackBuilder.product());

        edm::Handle<JetTracksAssociationCollection> jetTracksAssoc;
        event.getByLabel(jetTracksAssocLabel, jetTracksAssoc);

        edm::Handle<VertexCollection> primaryVertices;
        event.getByLabel(primaryVerticesLabel, primaryVertices);

        std::auto_ptr<JetFloatAssociation::Container> result(
                new JetFloatAssociation::Container(jetTracksAssoc->keyProduct()));

        for(JetTracksAssociationCollection::const_iterator iter =
                                                jetTracksAssoc->begin();
            iter != jetTracksAssoc->end(); ++iter) {
                if (primaryVertices->empty())
                        (*result)[iter->first] = -1.;

                const TrackRefVector &tracks = iter->second;
                std::vector<float> compatibility =
                        algo.compatibility(*primaryVertices, tracks);

                // the first vertex is the presumed signal vertex
                (*result)[iter->first] = compatibility[0];
        }

        algo.resetEvent(0);

        event.put(result);
}