PATGenJetSlimmer.cc

Go to the documentation of this file.

//
// $Id: PATGenJetSlimmer.cc,v 1.1 2011/03/24 18:45:45 mwlebour Exp $
//

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "DataFormats/Common/interface/Association.h"
#include "DataFormats/Common/interface/RefToPtr.h"
#include "DataFormats/PatCandidates/interface/PackedGenParticle.h"
#include "DataFormats/JetReco/interface/GenJet.h"

namespace pat {

  class PATGenJetSlimmer : public edm::stream::EDProducer<> {
  public:
    explicit PATGenJetSlimmer(const edm::ParameterSet & iConfig);
    virtual ~PATGenJetSlimmer() { }
    
    virtual void produce(edm::Event & iEvent, const edm::EventSetup & iSetup);
    
  private:
    const edm::EDGetTokenT<edm::View<reco::GenJet> > src_;
    const edm::EDGetTokenT<edm::Association<std::vector<pat::PackedGenParticle> > > gp2pgp_;
    
    const StringCutObjectSelector<reco::GenJet> cut_;
    
    const bool clearDaughters_;
    const bool dropSpecific_;
  };

} // namespace

pat::PATGenJetSlimmer::PATGenJetSlimmer(const edm::ParameterSet & iConfig) :
    src_(consumes<edm::View<reco::GenJet> >(iConfig.getParameter<edm::InputTag>("src"))),
    gp2pgp_(consumes<edm::Association<std::vector<pat::PackedGenParticle> > >(iConfig.getParameter<edm::InputTag>("packedGenParticles"))),
    cut_(iConfig.getParameter<std::string>("cut")),
    clearDaughters_(iConfig.getParameter<bool>("clearDaughters")),
    dropSpecific_(iConfig.getParameter<bool>("dropSpecific"))
{
    produces<std::vector<reco::GenJet> >();
}

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

    Handle<View<reco::GenJet> >      src;
    iEvent.getByToken(src_, src);

    auto_ptr<vector<reco::GenJet> >  out(new vector<reco::GenJet>());
    out->reserve(src->size());

    Handle<edm::Association<std::vector<pat::PackedGenParticle> > > gp2pgp;
    iEvent.getByToken(gp2pgp_,gp2pgp);
    

    for (View<reco::GenJet>::const_iterator it = src->begin(), ed = src->end(); it != ed; ++it) {
        if (!cut_(*it)) continue;

        out->push_back(*it);
        reco::GenJet & jet = out->back();

        if (clearDaughters_) {
            jet.clearDaughters();
        }
    else // rekey
    {
                //copy old 
        reco::CompositePtrCandidate::daughters old = jet.daughterPtrVector();
        jet.clearDaughters();
        std::map<unsigned int,reco::CandidatePtr> ptrs;
        for(unsigned int  i=0;i<old.size();i++)
        {
//  if(! ((*gp2pgp)[old[i]]).isNonnull())   {
//      std::cout << "Missing ref for key"  <<  old[i].key() << " pdgid " << old[i]->pdgId() << " st "<<   old[i]->status() <<  " pt " << old[i]->pt() << " eta " << old[i]->eta() << std::endl;
//  }
            ptrs[((*gp2pgp)[old[i]]).key()]=refToPtr((*gp2pgp)[old[i]]);
        }
        for(std::map<unsigned int,reco::CandidatePtr>::iterator itp=ptrs.begin();itp!=ptrs.end();itp++) //iterate on sorted items
        {
            jet.addDaughter(itp->second);
        }


    }
        if (dropSpecific_) {
            jet.setSpecific( reco::GenJet::Specific() );
        }
    }

    iEvent.put(out);
}

#include "FWCore/Framework/interface/MakerMacros.h"
using namespace pat;
DEFINE_FWK_MODULE(PATGenJetSlimmer);