egammaTools Namespace Reference

Classes
struct	DNNConfiguration
class	EgammaDNNHelper
struct	ScalerConfiguration

Typedefs
typedef std::function< uint(const std::map< std::string, float > &)>	ModelSelector

Functions
double	ecalEta (const math::XYZVector &momentum, const math::XYZPoint &vertex)
double	ecalPhi (const MagneticField &magField, const math::XYZVector &momentum, const math::XYZPoint &vertex, const int charge)
void	localEcalClusterCoordsEB (const reco::CaloCluster &bclus, const CaloGeometry &geom, float &etacry, float &phicry, int &ieta, int &iphi, float &thetatilt, float &phitilt)
void	localEcalClusterCoordsEE (const reco::CaloCluster &bclus, const CaloGeometry &geom, float &xcry, float &ycry, int &ix, int &iy, float &thetatilt, float &phitilt)
template<class Candidate >
void	validateEgammaCandidate (Candidate const &candidate)
void	validateGsfElectron (reco::GsfElectron const &electron)

Typedef Documentation

typedef std::function<uint(const std::map<std::string, float>&)> egammaTools::ModelSelector

Definition at line 39 of file EgammaDNNHelper.h.

Function Documentation

double egammaTools::ecalEta	(	const math::XYZVector &	momentum,
		const math::XYZPoint &	vertex
	)

Definition at line 71 of file ECALPositionCalculator.cc.

References ETA, etaBarrelEndcap, log, Geom::pi(), R_ECAL, funct::tan(), theta(), and Z_Endcap.

                                                                            {
    // Get kinematic variables

    float etaParticle = momentum.eta();
    float vZ = vertex.z();
    float vRho = vertex.Rho();

    if (etaParticle != 0.0) {
      float theta = 0.0;
      float zEcal = (R_ECAL - vRho) * sinh(etaParticle) + vZ;

      if (zEcal != 0.0)
        theta = atan(R_ECAL / zEcal);
      if (theta < 0.0)
        theta = theta + Geom::pi();

      float ETA = -log(tan(0.5 * theta));

      if (fabs(ETA) > etaBarrelEndcap) {
        float Zend = Z_Endcap;
        if (etaParticle < 0.0)
          Zend = -Zend;
        float Zlen = Zend - vZ;
        float RR = Zlen / sinh(etaParticle);
        theta = atan((RR + vRho) / Zend);
        if (theta < 0.0)
          theta = theta + Geom::pi();
        ETA = -log(tan(0.5 * theta));
      }
      return ETA;

    } else {
      edm::LogWarning("") << "[EcalPositionFromTrack::etaTransformation] Warning: "
                          << "Eta equals to zero, not correcting";
      return etaParticle;
    }
  }

double egammaTools::ecalPhi	(	const MagneticField &	magField,
		const math::XYZVector &	momentum,
		const math::XYZPoint &	vertex,
		const int	charge
	)

Definition at line 16 of file ECALPositionCalculator.cc.

References RecoTauCleanerPlugins::charge, etaBarrelEndcap, MagneticField::inTesla(), Geom::pi(), Geom::twoPi(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by MuonMETAlgo::correctMETforMuon().

                                   {
    // Get kinematic variables

    float ptParticle = momentum.Rho();
    float etaParticle = momentum.eta();
    float phiParticle = momentum.phi();
    float vRho = vertex.Rho();

    // Magnetic field

    const float RBARM = 1.357;  // was 1.31 , updated on 16122003
    const float ZENDM = 3.186;  // was 3.15 , updated on 16122003

    float rbend = RBARM - (vRho / 100.0);  //Assumed vRho in cm
    float bend = 0.3 * magField.inTesla(GlobalPoint(0., 0., 0.)).z() * rbend / 2.0;
    float phi = 0.0;

    if (fabs(etaParticle) <= etaBarrelEndcap) {
      if (fabs(bend / ptParticle) <= 1.) {
        phi = phiParticle - asin(bend / ptParticle) * charge;
        if (phi > Geom::pi())
          phi = phi - Geom::twoPi();
        if (phi < -Geom::pi())
          phi = phi + Geom::twoPi();
      } else {
        edm::LogWarning("") << "[EcalPositionFromTrack::phiTransformation] Warning: "
                            << "Too low Pt, giving up";
        return phiParticle;
      }

    }  // end if in the barrel

    if (fabs(etaParticle) > etaBarrelEndcap) {
      float rHit = 0.0;
      rHit = ZENDM / sinh(fabs(etaParticle));
      if (fabs(((rHit - (vRho / 100.0)) / rbend) * bend / ptParticle) <= 1.0) {
        phi = phiParticle - asin(((rHit - (vRho / 100.0)) / rbend) * bend / ptParticle) * charge;
        if (phi > Geom::pi())
          phi = phi - Geom::twoPi();
        if (phi < -Geom::pi())
          phi = phi + Geom::twoPi();
      } else {
        edm::LogWarning("") << "[EcalPositionFromTrack::phiTransformation] Warning: "
                            << "Too low Pt, giving up";
        return phiParticle;
      }

    }  // end if in the endcap

    return phi;
  }

void egammaTools::localEcalClusterCoordsEB	(	const reco::CaloCluster &	bclus,
		const CaloGeometry &	geom,
		float &	etacry,
		float &	phicry,
		int &	ieta,
		int &	iphi,
		float &	thetatilt,
		float &	phitilt
	)

Definition at line 13 of file EcalClusterLocal.cc.

References cms::cuda::assert(), funct::cos(), reco::deltaR(), HLT_FULL_cff::depth, runTauDisplay::dr, DetId::Ecal, EcalBarrel, reco::CaloCluster::energy(), PV3DBase< T, PVType, FrameType >::eta(), first, relativeConstraints::geom, TruncatedPyramid::getPhiAxis(), TruncatedPyramid::getPosition(), CaloGeometry::getSubdetectorGeometry(), TruncatedPyramid::getThetaAxis(), reco::CaloCluster::hitsAndFractions(), EBDetId::ieta(), EBDetId::iphi(), log, M_PI, colinearityKinematic::Phi, PV3DBase< T, PVType, FrameType >::phi(), reco::CaloCluster::position(), PV3DBase< T, PVType, FrameType >::theta(), and X0.

Referenced by EGEnergyCorrector::CorrectedEnergyWithError(), EGEnergyCorrector::CorrectedEnergyWithErrorV3(), EcalRegressionData::fill(), EGRegressionModifierV3::getSeedCrysCoord(), SuperClusterHelper::localCoordinates(), EGRegressionModifierV2::modifyObject(), and EGRegressionModifierV1::modifyObject().

                                                {
    assert(bclus.hitsAndFractions().at(0).first.subdetId() == EcalBarrel);

    const CaloSubdetectorGeometry *geom =
        caloGeometry.getSubdetectorGeometry(DetId::Ecal, EcalBarrel);  //EcalBarrel = 1

    const math::XYZPoint &position_ = bclus.position();
    double Theta = -position_.theta() + 0.5 * M_PI;
    double Eta = position_.eta();
    double Phi = TVector2::Phi_mpi_pi(position_.phi());

    //Calculate expected depth of the maximum shower from energy (like in PositionCalc::Calculate_Location()):
    // The parameters X0 and T0 are hardcoded here because these values were used to calculate the corrections:
    const float X0 = 0.89;
    const float T0 = 7.4;
    double depth = X0 * (T0 + log(bclus.energy()));

    //find max energy crystal
    std::vector<std::pair<DetId, float> > crystals_vector = bclus.hitsAndFractions();
    float drmin = 999.;
    EBDetId crystalseed;
    //printf("starting loop over crystals, etot = %5f:\n",bclus.energy());
    for (unsigned int icry = 0; icry != crystals_vector.size(); ++icry) {
      EBDetId crystal(crystals_vector[icry].first);

      auto cell = geom->getGeometry(crystal);
      const TruncatedPyramid *cpyr = dynamic_cast<const TruncatedPyramid *>(cell.get());
      GlobalPoint center_pos = cpyr->getPosition(depth);
      double EtaCentr = center_pos.eta();
      double PhiCentr = TVector2::Phi_mpi_pi(center_pos.phi());

      float dr = reco::deltaR(Eta, Phi, EtaCentr, PhiCentr);
      if (dr < drmin) {
        drmin = dr;
        crystalseed = crystal;
      }
    }

    ieta = crystalseed.ieta();
    iphi = crystalseed.iphi();

    // Get center cell position from shower depth
    auto cell = geom->getGeometry(crystalseed);
    const TruncatedPyramid *cpyr = dynamic_cast<const TruncatedPyramid *>(cell.get());

    thetatilt = cpyr->getThetaAxis();
    phitilt = cpyr->getPhiAxis();

    GlobalPoint center_pos = cpyr->getPosition(depth);

    double PhiCentr = TVector2::Phi_mpi_pi(center_pos.phi());
    double PhiWidth = (M_PI / 180.);
    phicry = (TVector2::Phi_mpi_pi(Phi - PhiCentr)) / PhiWidth;
    //Some flips to take into account ECAL barrel symmetries:
    if (ieta < 0)
      phicry *= -1.;

    double ThetaCentr = -center_pos.theta() + 0.5 * M_PI;
    double ThetaWidth = (M_PI / 180.) * std::cos(ThetaCentr);
    etacry = (Theta - ThetaCentr) / ThetaWidth;
    //flip to take into account ECAL barrel symmetries:
    if (ieta < 0)
      etacry *= -1.;

    return;
  }

void egammaTools::localEcalClusterCoordsEE	(	const reco::CaloCluster &	bclus,
		const CaloGeometry &	geom,
		float &	xcry,
		float &	ycry,
		int &	ix,
		int &	iy,
		float &	thetatilt,
		float &	phitilt
	)

Definition at line 87 of file EcalClusterLocal.cc.

References funct::abs(), cms::cuda::assert(), reco::deltaR(), HLT_FULL_cff::depth, runTauDisplay::dr, DetId::Ecal, EcalEndcap, reco::CaloCluster::energy(), PV3DBase< T, PVType, FrameType >::eta(), reco::CaloCluster::eta(), first, relativeConstraints::geom, TruncatedPyramid::getPhiAxis(), TruncatedPyramid::getPosition(), CaloGeometry::getSubdetectorGeometry(), TruncatedPyramid::getThetaAxis(), reco::CaloCluster::hitsAndFractions(), EEDetId::ix(), EEDetId::iy(), log, colinearityKinematic::Phi, PV3DBase< T, PVType, FrameType >::phi(), reco::CaloCluster::position(), X, PV3DBase< T, PVType, FrameType >::x(), X0, BeamSpotPI::Y, and PV3DBase< T, PVType, FrameType >::y().

Referenced by EcalRegressionData::fill(), EGRegressionModifierV3::getSeedCrysCoord(), SuperClusterHelper::localCoordinates(), EGRegressionModifierV2::modifyObject(), and EGRegressionModifierV1::modifyObject().

                                                {
    assert(bclus.hitsAndFractions().at(0).first.subdetId() == EcalEndcap);

    const CaloSubdetectorGeometry *geom =
        caloGeometry.getSubdetectorGeometry(DetId::Ecal, EcalEndcap);  //EcalBarrel = 1

    const math::XYZPoint &position_ = bclus.position();
    //double Theta = -position_.theta()+0.5*M_PI;
    double Eta = position_.eta();
    double Phi = TVector2::Phi_mpi_pi(position_.phi());
    double X = position_.x();
    double Y = position_.y();

    //Calculate expected depth of the maximum shower from energy (like in PositionCalc::Calculate_Location()):
    // The parameters X0 and T0 are hardcoded here because these values were used to calculate the corrections:
    const float X0 = 0.89;
    float T0 = 1.2;
    //different T0 value if outside of preshower coverage
    if (std::abs(bclus.eta()) < 1.653)
      T0 = 3.1;

    double depth = X0 * (T0 + log(bclus.energy()));

    //find max energy crystal
    std::vector<std::pair<DetId, float> > crystals_vector = bclus.hitsAndFractions();
    float drmin = 999.;
    EEDetId crystalseed;
    //printf("starting loop over crystals, etot = %5f:\n",bclus.energy());
    for (unsigned int icry = 0; icry != crystals_vector.size(); ++icry) {
      EEDetId crystal(crystals_vector[icry].first);

      auto cell = geom->getGeometry(crystal);
      const TruncatedPyramid *cpyr = dynamic_cast<const TruncatedPyramid *>(cell.get());
      GlobalPoint center_pos = cpyr->getPosition(depth);
      double EtaCentr = center_pos.eta();
      double PhiCentr = TVector2::Phi_mpi_pi(center_pos.phi());

      float dr = reco::deltaR(Eta, Phi, EtaCentr, PhiCentr);
      if (dr < drmin) {
        drmin = dr;
        crystalseed = crystal;
      }
    }

    ix = crystalseed.ix();
    iy = crystalseed.iy();

    // Get center cell position from shower depth
    auto cell = geom->getGeometry(crystalseed);
    const TruncatedPyramid *cpyr = dynamic_cast<const TruncatedPyramid *>(cell.get());

    thetatilt = cpyr->getThetaAxis();
    phitilt = cpyr->getPhiAxis();

    GlobalPoint center_pos = cpyr->getPosition(depth);

    double XCentr = center_pos.x();
    double XWidth = 2.59;
    xcry = (X - XCentr) / XWidth;

    double YCentr = center_pos.y();
    double YWidth = 2.59;
    ycry = (Y - YCentr) / YWidth;

    return;
  }

template<class Candidate >

void egammaTools::validateEgammaCandidate

(

Candidate const &

candidate

)

Definition at line 11 of file validateEgammaCandidate.h.

References validateGsfElectron().

Referenced by MVAValueMapProducer< ParticleType >::produce().

                                                           {
    if constexpr (std::is_same<Candidate, reco::GsfElectron>()) {
      validateGsfElectron(candidate);
    }
  }

void egammaTools::validateGsfElectron

(

reco::GsfElectron const &

electron

)

Definition at line 5 of file validateEgammaCandidate.cc.

References reco::GsfElectron::convVtxFitProb(), and Exception.

Referenced by validateEgammaCandidate().

                                                                     {
  if (electron.convVtxFitProb() > 1.0f) {
    throw cms::Exception("EgammaError")
        << "invalid value " << electron.convVtxFitProb() << " for reco::GsfElectron.conversionRejection_.vtxFitProb\n"
        << "It should be either beween 0.0f and 1.0f or negative in case no matching conversion was found.\n"
        << "Probably you need to update the electron collection with the EG9X105XObjectUpdateModifier plugin:\n"
        << "see PhysicsTools/NanoAOD/python/electrons_cff.py for an example.\n";
  }
}