PFClusterTimeAssigner.cc

Go to the documentation of this file.

#ifndef __PFClusterTimeAssigner__
#define __PFClusterTimeAssigner__

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"

#include "DataFormats/Common/interface/ValueMap.h"


class PFClusterTimeAssigner : public edm::stream::EDProducer<> {
public: 
  PFClusterTimeAssigner(const edm::ParameterSet& conf) {
    const edm::InputTag&  clusters = 
      conf.getParameter<edm::InputTag>("src");    
    clustersTok_ = consumes<reco::PFClusterCollection>( clusters );

    const edm::InputTag& times = 
      conf.getParameter<edm::InputTag>("timeSrc");
    timesTok_ = consumes<edm::ValueMap<float> >( times );
    
    produces<reco::PFClusterCollection>();
  }

  virtual void produce(edm::Event& e, const edm::EventSetup& es);
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:  
  edm::EDGetTokenT<reco::PFClusterCollection> clustersTok_;
  edm::EDGetTokenT<edm::ValueMap<float> > timesTok_;
};

DEFINE_FWK_MODULE(PFClusterTimeAssigner);

void PFClusterTimeAssigner::
produce(edm::Event& e, const edm::EventSetup& es) {
  auto clusters_out = std::make_unique<reco::PFClusterCollection>();
  
  edm::Handle<reco::PFClusterCollection> clustersH;
  e.getByToken(clustersTok_,clustersH);
  edm::Handle<edm::ValueMap<float> > timesH;
  e.getByToken(timesTok_,timesH);

  auto const & clusters = *clustersH;
  auto const & times = *timesH;
  
  clusters_out->reserve(clusters.size());
  clusters_out->insert(clusters_out->end(),
               clusters.begin(),clusters.end());

  //build the EE->PS association
  auto& out = *clusters_out;
  for( unsigned i = 0; i < out.size(); ++i ) {      
    
    edm::Ref<reco::PFClusterCollection> clusterRef(clustersH,i);
    const float time = times[clusterRef];
    out[i].setTime(time);
    
  }

  e.put(std::move(clusters_out));
}

void PFClusterTimeAssigner::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;  
  desc.add<edm::InputTag>("src",edm::InputTag("particleFlowClusterECALUncorrected"));
  desc.add<edm::InputTag>("timeSrc",edm::InputTag("ecalBarrelClusterFastTimer"));
  descriptions.add("particleFlowClusterTimeAssignerDefault",desc);
}

#endif