PFClusterTimeSelector.cc

Go to the documentation of this file.

#include "RecoParticleFlow/PFClusterProducer/plugins/PFClusterTimeSelector.h"
#include <memory>

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"


using namespace std;
using namespace edm;


PFClusterTimeSelector::PFClusterTimeSelector(const edm::ParameterSet& iConfig):
  clusters_(consumes<reco::PFClusterCollection>(iConfig.getParameter<edm::InputTag>("src")))
{

  std::vector<edm::ParameterSet> cuts = iConfig.getParameter<std::vector<edm::ParameterSet> >("cuts");
  for (const auto& cut :cuts ) {
    CutInfo info;
    info.depth   = cut.getParameter<double>("depth");
    info.minE    = cut.getParameter<double>("minEnergy");
    info.maxE    = cut.getParameter<double>("maxEnergy");
    info.minTime = cut.getParameter<double>("minTime");
    info.maxTime = cut.getParameter<double>("maxTime");
    info.endcap  = cut.getParameter<bool>("endcap");
    cutInfo_.push_back(info);
  }


  produces<reco::PFClusterCollection>();
  produces<reco::PFClusterCollection>("OOT");
}


void PFClusterTimeSelector::produce(edm::Event& iEvent, 
                   const edm::EventSetup& iSetup) {

  edm::Handle<reco::PFClusterCollection> clusters; 
  iEvent.getByToken(clusters_,clusters);
  auto out = std::make_unique<reco::PFClusterCollection>();
  auto outOOT = std::make_unique<reco::PFClusterCollection>();

  for(const auto& cluster : *clusters ) {    
    const double energy = cluster.energy();
    const double time = cluster.time();    
    const double depth = cluster.depth();    
    const PFLayer::Layer  layer = cluster.layer();    
    for (const auto& info : cutInfo_) {
      if (energy<info.minE || energy>info.maxE)
    continue;
      if (depth<0.9*info.depth || depth>1.1*info.depth)
    continue;
      if ((info.endcap && (layer==PFLayer::ECAL_BARREL || layer==PFLayer::HCAL_BARREL1 || layer==PFLayer::HCAL_BARREL2))||
      (((!info.endcap) && (layer==PFLayer::ECAL_ENDCAP || layer==PFLayer::HCAL_ENDCAP))))
    continue;
      
      if (time>info.minTime && time<info.maxTime)
    out->push_back(cluster);
      else
    outOOT->push_back(cluster);

      break;

    }
    
  }    



  iEvent.put(std::move(out));
  iEvent.put(std::move(outOOT),"OOT");

}

PFClusterTimeSelector::~PFClusterTimeSelector() {}

// ------------ method called once each job just before starting event loop  ------------
void 
PFClusterTimeSelector::beginRun(const edm::Run& run,
               const EventSetup& es) {

 
}