EgammaEcalExtractor.cc

Go to the documentation of this file.

//*****************************************************************************
// File:      EgammaEcalExtractor.cc
// ----------------------------------------------------------------------------
// OrigAuth:  Gilles De Lentdecker
// Institute: IIHE-ULB
//=============================================================================
//*****************************************************************************

//C++ includes
#include <vector>
#include <functional>

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

//CMSSW includes
#include "RecoEgamma/EgammaIsolationAlgos/plugins/EgammaEcalExtractor.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"

using namespace egammaisolation;
using namespace reco::isodeposit;

EgammaEcalExtractor::~EgammaEcalExtractor() {}

reco::IsoDeposit EgammaEcalExtractor::deposit(const edm::Event &ev,
                                              const edm::EventSetup &evSetup,
                                              const reco::Candidate &candidate) const {
  edm::Handle<reco::SuperClusterCollection> superClusterCollectionH;
  edm::Handle<reco::BasicClusterCollection> basicClusterCollectionH;
  ev.getByToken(superClusterToken_, superClusterCollectionH);
  ev.getByToken(basicClusterToken_, basicClusterCollectionH);

  reco::SuperClusterRef sc = candidate.get<reco::SuperClusterRef>();
  math::XYZPoint position = sc->position();
  // match the photon hybrid supercluster with those with Algo==0 (island)
  double delta1 = 1000.;
  double deltacur = 1000.;
  const reco::SuperCluster *matchedsupercluster = nullptr;
  bool MATCHEDSC = false;

  Direction candDir(position.eta(), position.phi());
  reco::IsoDeposit deposit(candDir);
  deposit.setVeto(reco::IsoDeposit::Veto(candDir, 0));  // no veto is needed for this deposit
  deposit.addCandEnergy(sc->energy() * sin(2 * atan(exp(-sc->eta()))));

  for (reco::SuperClusterCollection::const_iterator scItr = superClusterCollectionH->begin();
       scItr != superClusterCollectionH->end();
       ++scItr) {
    const reco::SuperCluster *supercluster = &(*scItr);

    if (supercluster->seed()->algo() == 0) {
      deltacur = ROOT::Math::VectorUtil::DeltaR(supercluster->position(), position);
      if (deltacur < delta1) {
        delta1 = deltacur;
        matchedsupercluster = supercluster;
        MATCHEDSC = true;
      }
    }
  }

  const reco::BasicCluster *cluster = nullptr;

  //loop over basic clusters
  for (reco::BasicClusterCollection::const_iterator cItr = basicClusterCollectionH->begin();
       cItr != basicClusterCollectionH->end();
       ++cItr) {
    cluster = &(*cItr);
    //    double ebc_bcchi2 = cluster->chi2();
    int ebc_bcalgo = cluster->algo();
    double ebc_bce = cluster->energy();
    double ebc_bceta = cluster->eta();
    double ebc_bcet = ebc_bce * sin(2 * atan(exp(ebc_bceta)));
    double newDelta = 0.;

    if (ebc_bcet > etMin_ && ebc_bcalgo == 0) {
      //      if (ebc_bcchi2 < 30.) {

      if (MATCHEDSC || !scmatch_) {  //skip selection if user wants to fill all superclusters
        bool inSuperCluster = false;

        if (scmatch_) {  // only try the matching if needed
          reco::CaloCluster_iterator theEclust = matchedsupercluster->clustersBegin();
          // loop over the basic clusters of the matched supercluster
          for (; theEclust != matchedsupercluster->clustersEnd(); ++theEclust) {
            if ((**theEclust) == (*cluster))
              inSuperCluster = true;
          }
        }
        if (!inSuperCluster || !scmatch_) {  //skip selection if user wants to fill all superclusters
          newDelta = ROOT::Math::VectorUtil::DeltaR(cluster->position(), position);
          if (newDelta < conesize_) {
            deposit.addDeposit(Direction(cluster->eta(), cluster->phi()), ebc_bcet);
          }
        }
      }
      //      } // matches ebc_bcchi2
    }  // matches ebc_bcet && ebc_bcalgo
  }

  //  std::cout << "Will return ecalIsol = " << ecalIsol << std::endl;
  return deposit;
}