EgammaTowerExtractor.cc

Go to the documentation of this file.

//*****************************************************************************
// File:      EgammaTowerExtractor.cc
// ----------------------------------------------------------------------------
// OrigAuth:  Matthias Mozer
// Institute: IIHE-VUB
//=============================================================================
//*****************************************************************************
//C++ includes
#include <vector>
#include <functional>
#include <cmath>

//ROOT includes
#include <Math/VectorUtil.h>

//CMSSW includes
#include "RecoEgamma/EgammaIsolationAlgos/plugins/EgammaTowerExtractor.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/CaloTowers/interface/CaloTower.h"

using namespace ROOT::Math::VectorUtil;

using namespace egammaisolation;
using namespace reco::isodeposit;

EgammaTowerExtractor::~EgammaTowerExtractor() {}

reco::IsoDeposit EgammaTowerExtractor::deposit(const edm::Event &iEvent,
                                               const edm::EventSetup &iSetup,
                                               const reco::Candidate &emObject) const {
  edm::Handle<CaloTowerCollection> towercollectionH;
  iEvent.getByToken(caloTowerToken, towercollectionH);

  //Take the SC position
  reco::SuperClusterRef sc = emObject.get<reco::SuperClusterRef>();
  math::XYZPoint caloPosition = sc->position();

  Direction candDir(caloPosition.eta(), caloPosition.phi());
  reco::IsoDeposit deposit(candDir);
  deposit.setVeto(reco::IsoDeposit::Veto(candDir, intRadius_));
  deposit.addCandEnergy(sc->energy() * sin(2 * atan(exp(-sc->eta()))));

  //loop over tracks
  for (CaloTowerCollection::const_iterator trItr = towercollectionH->begin(), trEnd = towercollectionH->end();
       trItr != trEnd;
       ++trItr) {
    double depEt = 0;
    //the hcal can be seperated into different depths
    //currently it is setup to check that the depth is valid in constructor
    //if the depth is not valid it fails gracefully
    //small bug fix, hadEnergyHeInnerLater returns zero for towers which are only depth 1
    //but we want Depth1 isolation to include these so we have to manually check for this
    if (depth_ == AllDepths)
      depEt = trItr->hadEt();
    else if (depth_ == Depth1)
      depEt = trItr->ietaAbs() < 18 || trItr->ietaAbs() > 29
                  ? trItr->hadEt()
                  : trItr->hadEnergyHeInnerLayer() * sin(trItr->p4().theta());
    else if (depth_ == Depth2)
      depEt = trItr->hadEnergyHeOuterLayer() * sin(trItr->p4().theta());

    if (depEt < etLow_)
      continue;

    Direction towerDir(trItr->eta(), trItr->phi());
    double dR2 = candDir.deltaR2(towerDir);

    if (dR2 < extRadius2_) {
      deposit.addDeposit(towerDir, depEt);
    }

  }  //end loop over tracks

  return deposit;
}