CMSSW_4_4_3_patch1/src/PhysicsTools/TagAndProbe/plugins/NearbyCandCountComputer.cc

Go to the documentation of this file.

//
// $Id: NearbyCandCountComputer.cc,v 1.3 2010/10/05 15:05:10 gpetrucc Exp $
//

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

#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Common/interface/View.h"

#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "PhysicsTools/PatUtils/interface/PATDiObjectProxy.h"

#include "CommonTools/Utils/interface/StringCutObjectSelector.h"

class NearbyCandCountComputer : public edm::EDProducer {
    public:
        explicit NearbyCandCountComputer(const edm::ParameterSet & iConfig);
        virtual ~NearbyCandCountComputer() ;

        virtual void produce(edm::Event & iEvent, const edm::EventSetup & iSetup);

    private:
        edm::InputTag probes_;            
        edm::InputTag objects_; 
        double deltaR2_;
        StringCutObjectSelector<reco::Candidate,true> objCut_; // lazy parsing, to allow cutting on variables not in reco::Candidate class
        StringCutObjectSelector<pat::DiObjectProxy,true> pairCut_;
};

NearbyCandCountComputer::NearbyCandCountComputer(const edm::ParameterSet & iConfig) :
    probes_(iConfig.getParameter<edm::InputTag>("probes")),
    objects_(iConfig.getParameter<edm::InputTag>("objects")),
    deltaR2_(std::pow(iConfig.getParameter<double>("deltaR"), 2)),
    objCut_(iConfig.existsAs<std::string>("objectSelection") ? iConfig.getParameter<std::string>("objectSelection") : "", true),
    pairCut_(iConfig.existsAs<std::string>("pairSelection") ? iConfig.getParameter<std::string>("pairSelection") : "", true)
{
    produces<edm::ValueMap<float> >();
}


NearbyCandCountComputer::~NearbyCandCountComputer()
{
}

void 
NearbyCandCountComputer::produce(edm::Event & iEvent, const edm::EventSetup & iSetup) {
    using namespace edm;

    // read input
    Handle<View<reco::Candidate> > probes, objects;
    iEvent.getByLabel(probes_,  probes);
    iEvent.getByLabel(objects_, objects);

    // prepare vector for output    
    std::vector<float> values;
    
    // fill
    View<reco::Candidate>::const_iterator probe, endprobes = probes->end();
    View<reco::Candidate>::const_iterator object, beginobjects = objects->begin(), endobjects = objects->end();
    for (probe = probes->begin(); probe != endprobes; ++probe) {
        float count = 0;
        for (object = beginobjects; object != endobjects; ++object) {
            if ((deltaR2(*probe, *object) >= deltaR2_) &&
                objCut_(*object) && 
                pairCut_(pat::DiObjectProxy(*probe, *object))) {
                    count++;
            }             
        }
        values.push_back(count);
    }

    // convert into ValueMap and store
    std::auto_ptr<ValueMap<float> > valMap(new ValueMap<float>());
    ValueMap<float>::Filler filler(*valMap);
    filler.insert(probes, values.begin(), values.end());
    filler.fill();
    iEvent.put(valMap);
}


#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(NearbyCandCountComputer);