HFEMClusterProducer.cc

Go to the documentation of this file.

//Package:    EgammaHFProdcers
// Class  :    HFEMClusterProducer
// Original Author:  Kevin Klapoetke (minnesota)
//
// $Id: HFEMClusterProducer.cc,v 1.2 2007/09/19 Kevin Klapoetke
//

#include <iostream>
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "RecoEgamma/EgammaHFProducers/plugins/HFEMClusterProducer.h"
using namespace reco;
HFEMClusterProducer::HFEMClusterProducer(edm::ParameterSet const& conf)
    : hfreco_(consumes<HFRecHitCollection>(conf.getParameter<edm::InputTag>("hits"))) {
  produces<reco::HFEMClusterShapeCollection>();
  produces<reco::BasicClusterCollection>();
  produces<reco::SuperClusterCollection>();
  produces<reco::HFEMClusterShapeAssociationCollection>();
  algo_.setup(conf.getParameter<double>("minTowerEnergy"),
              conf.getParameter<double>("seedThresholdET"),
              conf.getParameter<double>("maximumSL"),
              conf.getParameter<double>("maximumRenergy"),
              conf.getParameter<bool>("usePMTFlag"),
              conf.getParameter<bool>("usePulseFlag"),
              conf.getParameter<bool>("forcePulseFlagMC"),
              conf.getParameter<int>("correctionType"));
}

void HFEMClusterProducer::produce(edm::Event& e, edm::EventSetup const& iSetup) {
  edm::Handle<HFRecHitCollection> hf_hits;

  e.getByToken(hfreco_, hf_hits);

  edm::ESHandle<CaloGeometry> geometry;
  iSetup.get<CaloGeometryRecord>().get(geometry);

  // create return data
  auto retdata1 = std::make_unique<HFEMClusterShapeCollection>();
  auto retdata2 = std::make_unique<SuperClusterCollection>();

  algo_.isMC(!e.isRealData());

  algo_.clusterize(*hf_hits, *geometry, *retdata1, *retdata2);
  edm::OrphanHandle<reco::SuperClusterCollection> SupHandle;
  edm::OrphanHandle<reco::HFEMClusterShapeCollection> ShapeHandle;

  // put the results
  ShapeHandle = e.put(std::move(retdata1));
  SupHandle = e.put(std::move(retdata2));

  auto retdata3 = std::make_unique<HFEMClusterShapeAssociationCollection>(SupHandle, ShapeHandle);

  for (unsigned int i = 0; i < ShapeHandle->size(); i++) {
    retdata3->insert(edm::Ref<reco::SuperClusterCollection>(SupHandle, i),
                     edm::Ref<reco::HFEMClusterShapeCollection>(ShapeHandle, i));
  }

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