HGCDoublet Class Reference

#include <HGCDoublet.h>

Public Types
using	HGCntuplet = std::vector< unsigned int >

Public Member Functions
int	areAligned (double xi, double yi, double zi, double xo, double yo, double zo, float minCosTheta, float minCosPointing, const GlobalVector &refDir, bool debug=false) const
bool	checkCompatibilityAndTag (std::vector< HGCDoublet > &allDoublets, const std::vector< int > &innerDoublets, const GlobalVector &refDir, float minCosTheta, float minCosPointing=1., bool debug=false)
void	findNtuplets (std::vector< HGCDoublet > &allDoublets, HGCntuplet &tmpNtuplet, int seedIndex, const bool outInDFS, const unsigned int outInHops, const unsigned int maxOutInHops, std::vector< std::pair< unsigned int, unsigned int > > &outInToVisit)
	HGCDoublet (const int innerClusterId, const int outerClusterId, const int doubletId, const std::vector< reco::CaloCluster > *layerClusters, const int seedIndex)
int	innerClusterId () const
double	innerR () const
double	innerX () const
double	innerY () const
double	innerZ () const
int	outerClusterId () const
double	outerR () const
double	outerX () const
double	outerY () const
double	outerZ () const
int	seedIndex () const
void	setVisited (bool visited)
void	tagAsInnerNeighbor (unsigned int otherDoublet)
void	tagAsOuterNeighbor (unsigned int otherDoublet)

Private Attributes
bool	alreadyVisited_
const int	innerClusterId_
std::vector< int >	innerNeighbors_
const double	innerR_
const double	innerX_
const double	innerY_
const double	innerZ_
const std::vector< reco::CaloCluster > *	layerClusters_
const int	outerClusterId_
std::vector< int >	outerNeighbors_
const double	outerR_
const double	outerX_
const double	outerY_
const double	outerZ_
int	seedIndex_
const int	theDoubletId_

Detailed Description

Definition at line 14 of file HGCDoublet.h.

Member Typedef Documentation

HGCntuplet

using HGCDoublet::HGCntuplet = std::vector<unsigned int>

Definition at line 16 of file HGCDoublet.h.

Constructor & Destructor Documentation

HGCDoublet()

HGCDoublet::HGCDoublet ( const int  innerClusterId, const int  outerClusterId, const int  doubletId, const std::vector< reco::CaloCluster > *  layerClusters, const int  seedIndex  )

inline

Definition at line 18 of file HGCDoublet.h.

      : layerClusters_(layerClusters),
        theDoubletId_(doubletId),
        innerClusterId_(innerClusterId),
        outerClusterId_(outerClusterId),
        innerR_((*layerClusters)[innerClusterId].position().r()),
        outerR_((*layerClusters)[outerClusterId].position().r()),
        innerX_((*layerClusters)[innerClusterId].x()),
        outerX_((*layerClusters)[outerClusterId].x()),
        innerY_((*layerClusters)[innerClusterId].y()),
        outerY_((*layerClusters)[outerClusterId].y()),
        innerZ_((*layerClusters)[innerClusterId].z()),
        outerZ_((*layerClusters)[outerClusterId].z()),
        seedIndex_(seedIndex),
        alreadyVisited_(false) {}

Member Function Documentation

areAligned()

int HGCDoublet::areAligned	(	double	xi,
		double	yi,
		double	zi,
		double	xo,
		double	yo,
		double	zo,
		float	minCosTheta,
		float	minCosPointing,
		const GlobalVector &	refDir,
		bool	debug = false
	)		const

Definition at line 64 of file HGCDoublet.cc.

                                             {
  auto dx1 = xo - xi;
  auto dy1 = yo - yi;
  auto dz1 = zo - zi;
 
  auto dx2 = innerX() - xi;
  auto dy2 = innerY() - yi;
  auto dz2 = innerZ() - zi;
 
  // inner product
  auto dot = dx1 * dx2 + dy1 * dy2 + dz1 * dz2;
  auto dotsq = dot * dot;
  // magnitudes
  auto mag1sq = dx1 * dx1 + dy1 * dy1 + dz1 * dz1;
  auto mag2sq = dx2 * dx2 + dy2 * dy2 + dz2 * dz2;
 
  auto minCosTheta_sq = minCosTheta * minCosTheta;
  bool isWithinLimits = (dotsq > minCosTheta_sq * mag1sq * mag2sq);
 
  if (debug) {
    LogDebug("HGCDoublet") << "-- Are Aligned -- dotsq: " << dotsq << " mag1sq: " << mag1sq << " mag2sq: " << mag2sq
                           << "minCosTheta_sq:" << minCosTheta_sq << " isWithinLimits: " << isWithinLimits << std::endl;
  }
 
  // Now check the compatibility with the pointing origin.
  // Pointing origin is a vector obtained by the seed (track extrapolation i.e.)
  // or the direction wrt (0,0,0)
  // The compatibility is checked only for the innermost doublets: the
  // one with the outer doublets comes in by the alignment requirement of
  // the doublets themeselves
 
  const GlobalVector firstDoublet(dx2, dy2, dz2);
  const GlobalVector pointingDir = (seedIndex_ == -1) ? GlobalVector(innerX(), innerY(), innerZ()) : refDir;
 
  auto dot_pointing = pointingDir.dot(firstDoublet);
  auto dot_pointing_sq = dot_pointing * dot_pointing;
  auto mag_pointing_sq = pointingDir.mag2();
  auto minCosPointing_sq = minCosPointing * minCosPointing;
  bool isWithinLimitsPointing = (dot_pointing_sq > minCosPointing_sq * mag_pointing_sq * mag2sq);
  if (debug) {
    LogDebug("HGCDoublet") << "Pointing direction: " << pointingDir << std::endl;
    LogDebug("HGCDoublet") << "-- Are Aligned -- dot_pointing_sq: " << dot_pointing_sq
                           << " mag_pointing_sq: " << mag_pointing_sq << " mag2sq: " << mag2sq
                           << " isWithinLimitsPointing: " << isWithinLimitsPointing << std::endl;
  }
  // by squaring cosTheta and multiplying by the squares of the magnitudes
  // an equivalent comparison is made without the division and square root which are costly FP ops.
  return isWithinLimits && isWithinLimitsPointing;
}

References debug, Vector3DBase< T, FrameTag >::dot(), dot(), innerX(), innerY(), innerZ(), LogDebug, PV3DBase< T, PVType, FrameType >::mag2(), seedIndex_, and protons_cff::xi.

Referenced by checkCompatibilityAndTag().

checkCompatibilityAndTag()

bool HGCDoublet::checkCompatibilityAndTag	(	std::vector< HGCDoublet > &	allDoublets,
		const std::vector< int > &	innerDoublets,
		const GlobalVector &	refDir,
		float	minCosTheta,
		float	minCosPointing = 1.,
		bool	debug = false
	)

Definition at line 3 of file HGCDoublet.cc.

                                                      {
  int nDoublets = innerDoublets.size();
  int constexpr VSIZE = 4;
  int ok[VSIZE];
  double xi[VSIZE];
  double yi[VSIZE];
  double zi[VSIZE];
  int seedi[VSIZE];
  auto xo = outerX();
  auto yo = outerY();
  auto zo = outerZ();
  unsigned int doubletId = theDoubletId_;
 
  auto loop = [&](int i, int vs) {
    for (int j = 0; j < vs; ++j) {
      auto otherDoubletId = innerDoublets[i + j];
      auto &otherDoublet = allDoublets[otherDoubletId];
      xi[j] = otherDoublet.innerX();
      yi[j] = otherDoublet.innerY();
      zi[j] = otherDoublet.innerZ();
      seedi[j] = otherDoublet.seedIndex();
      if (debug) {
        LogDebug("HGCDoublet") << i + j << " is doublet " << otherDoubletId << std::endl;
      }
    }
    for (int j = 0; j < vs; ++j) {
      if (seedi[j] != seedIndex_) {
        ok[j] = 0;
        continue;
      }
      ok[j] = areAligned(xi[j], yi[j], zi[j], xo, yo, zo, minCosTheta, minCosPointing, refDir, debug);
      if (debug) {
        LogDebug("HGCDoublet") << "Are aligned for InnerDoubletId: " << i + j << " is " << ok[j] << std::endl;
      }
    }
    for (int j = 0; j < vs; ++j) {
      auto otherDoubletId = innerDoublets[i + j];
      auto &otherDoublet = allDoublets[otherDoubletId];
      if (ok[j]) {
        otherDoublet.tagAsOuterNeighbor(doubletId);
        allDoublets[doubletId].tagAsInnerNeighbor(otherDoubletId);
      }
    }
  };
  auto lim = VSIZE * (nDoublets / VSIZE);
  for (int i = 0; i < lim; i += VSIZE)
    loop(i, VSIZE);
  loop(lim, nDoublets - lim);
 
  if (debug) {
    LogDebug("HGCDoublet") << "Found " << innerNeighbors_.size() << " compatible doublets out of " << nDoublets
                           << " considered" << std::endl;
  }
  return innerNeighbors_.empty();
}

References areAligned(), debug, mps_fire::i, innerNeighbors_, dqmiolumiharvest::j, LogDebug, heppy_loop::loop, convertSQLiteXML::ok, outerX(), outerY(), outerZ(), seedIndex_, theDoubletId_, and protons_cff::xi.

findNtuplets()

void HGCDoublet::findNtuplets	(	std::vector< HGCDoublet > &	allDoublets,
		HGCntuplet &	tmpNtuplet,
		int	seedIndex,
		const bool	outInDFS,
		const unsigned int	outInHops,
		const unsigned int	maxOutInHops,
		std::vector< std::pair< unsigned int, unsigned int > > &	outInToVisit
	)

Definition at line 123 of file HGCDoublet.cc.

                                                                                             {
  if (!alreadyVisited_ && seedIndex == seedIndex_) {
    alreadyVisited_ = true;
    tmpNtuplet.push_back(theDoubletId_);
    unsigned int numberOfOuterNeighbors = outerNeighbors_.size();
    for (unsigned int i = 0; i < numberOfOuterNeighbors; ++i) {
      allDoublets[outerNeighbors_[i]].findNtuplets(
          allDoublets, tmpNtuplet, seedIndex, outInDFS, outInHops, maxOutInHops, outInToVisit);
    }
    if (outInDFS && outInHops < maxOutInHops) {
      for (auto inN : innerNeighbors_) {
        outInToVisit.emplace_back(inN, outInHops + 1);
      }
    }
  }
}

References alreadyVisited_, mps_fire::i, innerNeighbors_, outerNeighbors_, seedIndex(), seedIndex_, and theDoubletId_.

innerClusterId()

int HGCDoublet::innerClusterId ( ) const

inline

Definition at line 56 of file HGCDoublet.h.

{ return innerClusterId_; }

References innerClusterId_.

innerR()

double HGCDoublet::innerR ( ) const

inline

Definition at line 50 of file HGCDoublet.h.

{ return innerR_; }

References innerR_.

innerX()

double HGCDoublet::innerX ( ) const

inline

Definition at line 38 of file HGCDoublet.h.

{ return innerX_; }

References innerX_.

Referenced by areAligned().

innerY()

double HGCDoublet::innerY ( ) const

inline

Definition at line 42 of file HGCDoublet.h.

{ return innerY_; }

References innerY_.

Referenced by areAligned().

innerZ()

double HGCDoublet::innerZ ( ) const

inline

Definition at line 46 of file HGCDoublet.h.

{ return innerZ_; }

References innerZ_.

Referenced by areAligned().

outerClusterId()

int HGCDoublet::outerClusterId ( ) const

inline

Definition at line 58 of file HGCDoublet.h.

{ return outerClusterId_; }

References outerClusterId_.

outerR()

double HGCDoublet::outerR ( ) const

inline

Definition at line 52 of file HGCDoublet.h.

{ return outerZ_; }

References outerZ_.

outerX()

double HGCDoublet::outerX ( ) const

inline

Definition at line 40 of file HGCDoublet.h.

{ return outerX_; }

References outerX_.

Referenced by checkCompatibilityAndTag().

outerY()

double HGCDoublet::outerY ( ) const

inline

Definition at line 44 of file HGCDoublet.h.

{ return outerY_; }

References outerY_.

Referenced by checkCompatibilityAndTag().

outerZ()

double HGCDoublet::outerZ ( ) const

inline

Definition at line 48 of file HGCDoublet.h.

{ return outerZ_; }

References outerZ_.

Referenced by checkCompatibilityAndTag().

seedIndex()

int HGCDoublet::seedIndex ( ) const

inline

Definition at line 54 of file HGCDoublet.h.

{ return seedIndex_; }

References seedIndex_.

Referenced by findNtuplets().

setVisited()

void HGCDoublet::setVisited ( bool  visited )

inline

Definition at line 90 of file HGCDoublet.h.

{ alreadyVisited_ = visited; }

References alreadyVisited_, and class-composition::visited.

tagAsInnerNeighbor()

void HGCDoublet::tagAsInnerNeighbor ( unsigned int  otherDoublet )

inline

Definition at line 62 of file HGCDoublet.h.

{ innerNeighbors_.push_back(otherDoublet); }

References innerNeighbors_.

tagAsOuterNeighbor()

void HGCDoublet::tagAsOuterNeighbor ( unsigned int  otherDoublet )

inline

Definition at line 60 of file HGCDoublet.h.

{ outerNeighbors_.push_back(otherDoublet); }

References outerNeighbors_.

Member Data Documentation

alreadyVisited_

bool HGCDoublet::alreadyVisited_

private

Definition at line 110 of file HGCDoublet.h.

Referenced by findNtuplets(), and setVisited().

innerClusterId_

const int HGCDoublet::innerClusterId_

private

Definition at line 98 of file HGCDoublet.h.

Referenced by innerClusterId().

innerNeighbors_

std::vector<int> HGCDoublet::innerNeighbors_

private

Definition at line 95 of file HGCDoublet.h.

Referenced by checkCompatibilityAndTag(), findNtuplets(), and tagAsInnerNeighbor().

innerR_

const double HGCDoublet::innerR_

private

Definition at line 101 of file HGCDoublet.h.

Referenced by innerR().

innerX_

const double HGCDoublet::innerX_

private

Definition at line 103 of file HGCDoublet.h.

Referenced by innerX().

innerY_

const double HGCDoublet::innerY_

private

Definition at line 105 of file HGCDoublet.h.

Referenced by innerY().

innerZ_

const double HGCDoublet::innerZ_

private

Definition at line 107 of file HGCDoublet.h.

Referenced by innerZ().

layerClusters_

const std::vector<reco::CaloCluster>* HGCDoublet::layerClusters_

private

Definition at line 93 of file HGCDoublet.h.

outerClusterId_

const int HGCDoublet::outerClusterId_

private

Definition at line 99 of file HGCDoublet.h.

Referenced by outerClusterId().

outerNeighbors_

std::vector<int> HGCDoublet::outerNeighbors_

private

Definition at line 94 of file HGCDoublet.h.

Referenced by findNtuplets(), and tagAsOuterNeighbor().

outerR_

const double HGCDoublet::outerR_

private

Definition at line 102 of file HGCDoublet.h.

outerX_

const double HGCDoublet::outerX_

private

Definition at line 104 of file HGCDoublet.h.

Referenced by outerX().

outerY_

const double HGCDoublet::outerY_

private

Definition at line 106 of file HGCDoublet.h.

Referenced by outerY().

outerZ_

const double HGCDoublet::outerZ_

private

Definition at line 108 of file HGCDoublet.h.

Referenced by outerR(), and outerZ().

seedIndex_

int HGCDoublet::seedIndex_

private

Definition at line 109 of file HGCDoublet.h.

Referenced by areAligned(), checkCompatibilityAndTag(), findNtuplets(), and seedIndex().

theDoubletId_

const int HGCDoublet::theDoubletId_

private

Definition at line 97 of file HGCDoublet.h.

Referenced by checkCompatibilityAndTag(), and findNtuplets().