BaselinePFSCRegression.cc

Go to the documentation of this file.

#include "RecoEgamma/EgammaTools/interface/BaselinePFSCRegression.h"
#include "RecoEgamma/EgammaTools/interface/EcalRegressionData.h"
#include "RecoEcal/EgammaCoreTools/interface/EcalClusterTools.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"

#include "TVector2.h"
#include "DataFormats/Math/interface/deltaR.h"

void BaselinePFSCRegression::update(const edm::EventSetup& es) {
  const CaloTopologyRecord& topofrom_es = es.get<CaloTopologyRecord>();
  if (!topo_record || topofrom_es.cacheIdentifier() != topo_record->cacheIdentifier()) {
    topo_record = &topofrom_es;
    topo_record->get(calotopo);
  }
  const CaloGeometryRecord& geomfrom_es = es.get<CaloGeometryRecord>();
  if (!geom_record || geomfrom_es.cacheIdentifier() != geom_record->cacheIdentifier()) {
    geom_record = &geomfrom_es;
    geom_record->get(calogeom);
  }
}

void BaselinePFSCRegression::set(const reco::SuperCluster& sc, std::vector<float>& vars) const {
  EcalRegressionData regData;
  regData.fill(
      sc, rechitsEB.product(), rechitsEE.product(), calogeom.product(), calotopo.product(), vertices.product());
  regData.fillVec(vars);

  //solely to reproduce old exception behaviour, unnessessary although it likely is
  if (sc.seed()->hitsAndFractions().at(0).first.subdetId() != EcalBarrel &&
      sc.seed()->hitsAndFractions().at(0).first.subdetId() != EcalEndcap) {
    throw cms::Exception("PFECALSuperClusterProducer::calculateRegressedEnergy")
        << "Supercluster seed is either EB nor EE!" << std::endl;
  }
}

void BaselinePFSCRegression::setTokens(const edm::ParameterSet& ps, edm::ConsumesCollector&& cc) {
  const edm::InputTag rceb = ps.getParameter<edm::InputTag>("ecalRecHitsEB");
  const edm::InputTag rcee = ps.getParameter<edm::InputTag>("ecalRecHitsEE");
  const edm::InputTag vtx = ps.getParameter<edm::InputTag>("vertexCollection");
  inputTagEBRecHits_ = cc.consumes<EcalRecHitCollection>(rceb);
  inputTagEERecHits_ = cc.consumes<EcalRecHitCollection>(rcee);
  inputTagVertices_ = cc.consumes<reco::VertexCollection>(vtx);
}

void BaselinePFSCRegression::setEvent(const edm::Event& ev) {
  ev.getByToken(inputTagEBRecHits_, rechitsEB);
  ev.getByToken(inputTagEERecHits_, rechitsEE);
  ev.getByToken(inputTagVertices_, vertices);
}