l1tpf_calo::Phase1GridBase Class Reference

#include <CaloClusterer.h>

Inheritance diagram for l1tpf_calo::Phase1GridBase:

Public Member Functions
int	find_cell (float eta, float phi) const override
int	ifind_cell (int ieta, int iphi) const
	Phase1GridBase (int nEta, int nPhi, int ietaCoarse, int ietaVeryCoarse, const float *towerEtas)
Public Member Functions inherited from l1tpf_calo::Grid
float	eta (int icell) const
float	etaWidth (int icell) const
int	ieta (int icell) const
int	iphi (int icell) const
int	neighbour (int icell, unsigned int idx) const
float	phi (int icell) const
float	phiWidth (int icell) const
unsigned int	size () const
virtual	~Grid ()

Protected Member Functions
int	imove (int ieta, int iphi, int deta, int dphi)
bool	valid_ieta_iphi (int ieta, int iphi) const
Protected Member Functions inherited from l1tpf_calo::Grid
	Grid (unsigned int size)

Protected Attributes
std::vector< int >	cell_map_
const int	ietaCoarse_
const int	ietaVeryCoarse_
const int	nEta_
const int	nPhi_
const float *	towerEtas_
Protected Attributes inherited from l1tpf_calo::Grid
std::vector< float >	eta_
std::vector< float >	etaWidth_
std::vector< int >	ieta_
std::vector< int >	iphi_
unsigned int	ncells_
std::vector< std::array< int, 8 > >	neighbours_
std::vector< float >	phi_
std::vector< float >	phiWidth_

Detailed Description

Definition at line 51 of file CaloClusterer.h.

Constructor & Destructor Documentation

Phase1GridBase()

l1tpf_calo::Phase1GridBase::Phase1GridBase	(	int	nEta,
		int	nPhi,
		int	ietaCoarse,
		int	ietaVeryCoarse,
		const float *	towerEtas
	)

Definition at line 19 of file CaloClusterer.cc.

References funct::abs(), cms::cuda::assert(), cell_map_, l1tpf_calo::Grid::eta_, l1tpf_calo::Grid::etaWidth_, ntuplemaker::fill, l1tpf_calo::Grid::ieta_, ietaCoarse_, ietaVeryCoarse_, imove(), createfilelist::int, l1tpf_calo::Grid::iphi_, M_PI, l1tpf_calo::Grid::ncells_, l1tpf_calo::Grid::neighbours_, nEta_, nPhi_, l1tpf_calo::Grid::phi_, l1tpf_calo::Grid::phiWidth_, towerEtas_, and valid_ieta_iphi().

    : Grid(2 * ((ietaCoarse - 1) * nPhi + (ietaVeryCoarse - ietaCoarse) * (nPhi / 2) +
                (nEta - ietaVeryCoarse + 1) * (nPhi / 4))),
      nEta_(nEta),
      nPhi_(nPhi),
      ietaCoarse_(ietaCoarse),
      ietaVeryCoarse_(ietaVeryCoarse),
      towerEtas_(towerEtas),
      cell_map_(2 * nEta * nPhi, -1) {
  int icell = 0;
  for (int ie = -nEta_; ie <= nEta_; ++ie) {
    int absie = std::abs(ie);
    for (int iph = 1; iph <= nPhi_; ++iph) {
      if (!valid_ieta_iphi(ie, iph))
        continue;
      ieta_[icell] = ie;
      iphi_[icell] = iph;
      float etaLo = (absie < nEta_ ? towerEtas_[absie - 1] : towerEtas_[absie - 2]);
      float etaHi = (absie < nEta_ ? towerEtas_[absie] : towerEtas_[absie - 1]);
      eta_[icell] = (ie > 0 ? 0.5 : -0.5) * (etaLo + etaHi);
      etaWidth_[icell] = (etaHi - etaLo);
      phiWidth_[icell] = 2 * M_PI / nPhi_;
      if (absie >= ietaVeryCoarse_)
        phiWidth_[icell] *= 4;
      else if (absie >= ietaCoarse_)
        phiWidth_[icell] *= 2;
      phi_[icell] = (iph - 1) * 2 * M_PI / nPhi_ + 0.5 * phiWidth_[icell];
      if (phi_[icell] > M_PI)
        phi_[icell] -= 2 * M_PI;
      std::fill(neighbours_[icell].begin(), neighbours_[icell].end(), -1);
      cell_map_[(ie + nEta_) + 2 * nEta_ * (iph - 1)] = icell;
      icell++;
    }
  }
  assert(unsigned(icell) == ncells_);
  // now link the cells
  for (icell = 0; icell < int(ncells_); ++icell) {
    int ie = ieta_[icell], iph = iphi_[icell];
    int ineigh = 0;
    for (int deta = -1; deta <= +1; ++deta) {
      for (int dphi = -1; dphi <= +1; ++dphi) {
        if (deta == 0 && dphi == 0)
          continue;
        neighbours_[icell][ineigh++] = imove(ie, iph, deta, dphi);
      }
    }
  }
  //for (float teta = 0; teta <= 5.0; teta += 0.02) {
  //    for (float tphi = -M_PI; tphi <= M_PI; tphi += 0.02) {
  //        find_cell(+teta, tphi);
  //        find_cell(-teta, tphi);
  //    }
  //}
}

Member Function Documentation

find_cell()

int l1tpf_calo::Phase1GridBase::find_cell ( float  eta, float  phi  ) const

overridevirtual

Implements l1tpf_calo::Grid.

Definition at line 77 of file CaloClusterer.cc.

References funct::abs(), cms::cuda::assert(), HLT_2024v11_cff::distance, PVValHelper::eta, hcalRecHitTable_cff::ieta, hcalRecHitTable_cff::iphi, pfDeepBoostedJetPreprocessParams_cfi::lower_bound, M_PI, and reco::reduceRange().

                                                                  {
  int ieta =
      (eta != 0) ? std::distance(towerEtas_, std::lower_bound(towerEtas_, towerEtas_ + nEta_, std::abs(eta))) : 1;
  if (ieta == nEta_)
    return -1;  // outside bounds
  assert(ieta > 0 && ieta < nEta_);
  if (ieta > nEta_)
    ieta = nEta_;
  if (eta < 0)
    ieta = -ieta;
  phi = reco::reduceRange(phi);  // [-PI, PI]
  if (phi < 0)                   // then bring to [0, 2*PI]
    phi += 2 * M_PI;
  int iphi = std::floor(phi * nPhi_ / (2 * M_PI));
  if (phi >= 2 * M_PI)
    iphi = nPhi_ - 1;  // fix corner case due to roundings etc
  assert(iphi < nPhi_);
  if (std::abs(ieta) >= ietaVeryCoarse_)
    iphi -= (iphi % 4);
  else if (std::abs(ieta) >= ietaCoarse_)
    iphi -= (iphi % 2);
  iphi += 1;
  //if (!valid_ieta_iphi(ieta,iphi)) {
  //    printf("Error in finding cell for eta %+7.4f phi %+7.4f, got ieta = %+3d iphi %2d which is not valid\n",
  //        eta, phi, ieta, iphi);
  //}
  assert(valid_ieta_iphi(ieta, iphi));
  int icell = ifind_cell(ieta, iphi);
  assert(icell != -1);

  //if (std::abs(eta - eta_[icell]) > 0.501*etaWidth_[icell] || std::abs(deltaPhi(phi, phi_[icell])) > 0.501*phiWidth_[icell]) {
  //    printf("Mismatch in finding cell for eta %+7.4f phi %+7.4f, got ieta = %+3d iphi %2d which has eta %+7.4f +- %.4f phi %+7.4f +- %.4f ; deta = %+7.4f dphi = %+7.4f\n",
  //        eta, phi, ieta, iphi, eta_[icell], etaWidth_[icell], phi_[icell], phiWidth_[icell], eta - eta_[icell], deltaPhi(phi, phi_[icell]));
  //}
  //assert(std::abs(eta - eta_[icell]) <= 0.5*etaWidth_[icell]);
  //assert(std::abs(deltaPhi(phi, phi_[icell])) <= 0.5*phiWidth_[icell]);
  return icell;
}

ifind_cell()

int l1tpf_calo::Phase1GridBase::ifind_cell ( int  ieta, int  iphi  ) const

inline

Definition at line 56 of file CaloClusterer.h.

References cell_map_, l1tpf_calo::Grid::ieta(), l1tpf_calo::Grid::iphi(), and nEta_.

{ return cell_map_[(ieta + nEta_) + 2 * nEta_ * (iphi - 1)]; }

imove()

int l1tpf_calo::Phase1GridBase::imove ( int  ieta, int  iphi, int  deta, int  dphi  )

protected

Definition at line 118 of file CaloClusterer.cc.

References cms::cuda::assert(), hcalRecHitTable_cff::ieta, and hcalRecHitTable_cff::iphi.

Referenced by Phase1GridBase().

                                                                          {
  int ie = ieta, iph = iphi;
  switch (deta) {
    case -1:
      ie = (ie == -nEta_ ? 0 : (ie == +1 ? -1 : ie - 1));
      break;
    case +1:
      ie = (ie == +nEta_ ? 0 : (ie == -1 ? +1 : ie + 1));
      break;
    case 0:
      break;
    default:
      assert(false);
  };
  if (ie == 0)
    return -1;
  switch (dphi) {
    case -1:
      iph = (iph == 1 ? nPhi_ : iph - 1);
      break;
    case +1:
      iph = (iph == nPhi_ ? 1 : iph + 1);
      break;
    case 0:
      break;
    default:
      assert(false);
  };
  if (!valid_ieta_iphi(ie, iph))
    return -1;
  int icell = ifind_cell(ie, iph);
  assert(!(ie == ieta && iph == iphi));
  assert(icell != -1);
  assert(icell != ifind_cell(ieta, iphi));
  return icell;
}

valid_ieta_iphi()

bool l1tpf_calo::Phase1GridBase::valid_ieta_iphi ( int  ieta, int  iphi  ) const

inlineprotected

Definition at line 63 of file CaloClusterer.h.

References funct::abs(), l1tpf_calo::Grid::ieta(), ietaCoarse_, ietaVeryCoarse_, and l1tpf_calo::Grid::iphi().

Referenced by Phase1GridBase().

                                                   {
      if (ieta == 0 || iphi == 0)
        return false;
      if (std::abs(ieta) >= ietaVeryCoarse_ && (iphi % 4 != 1))
        return false;
      if (std::abs(ieta) >= ietaCoarse_ && (iphi % 2 != 1))
        return false;
      return true;
    }

Member Data Documentation

cell_map_

std::vector<int> l1tpf_calo::Phase1GridBase::cell_map_

protected

Definition at line 61 of file CaloClusterer.h.

Referenced by ifind_cell(), and Phase1GridBase().

ietaCoarse_

const int l1tpf_calo::Phase1GridBase::ietaCoarse_

protected

Definition at line 59 of file CaloClusterer.h.

Referenced by Phase1GridBase(), and valid_ieta_iphi().

ietaVeryCoarse_

const int l1tpf_calo::Phase1GridBase::ietaVeryCoarse_

protected

Definition at line 59 of file CaloClusterer.h.

Referenced by Phase1GridBase(), and valid_ieta_iphi().

nEta_

const int l1tpf_calo::Phase1GridBase::nEta_

protected

Definition at line 59 of file CaloClusterer.h.

Referenced by ifind_cell(), and Phase1GridBase().

nPhi_

const int l1tpf_calo::Phase1GridBase::nPhi_

protected

Definition at line 59 of file CaloClusterer.h.

Referenced by Phase1GridBase().

towerEtas_

const float* l1tpf_calo::Phase1GridBase::towerEtas_

protected

Definition at line 60 of file CaloClusterer.h.

Referenced by Phase1GridBase().