NDFinderPeakFinder Class Reference

Classes
struct	DeletionBounds

Public Member Functions
	NDFinderPeakFinder (const PeakFindingParameters &parameters)
void	setNewPlane (c3array &houghSpace)
std::vector< HoughPeak >	findPeaks ()

Private Member Functions
std::array< c3index, 2 >	getSectionBounds (const unsigned short quadrant, const unsigned section)
void	iterateOverSection (const std::array< c3index, 2 > &sectionBounds, std::vector< HoughPeak > &candidatePeaks)
HoughPeak	getSectionPeak (const std::array< c3index, 2 > &sectionBounds)
void	deletePeakSurroundings (const HoughPeak &peak)
void	clearHoughSpaceRow (const DeletionBounds &bounds)

Private Attributes
PeakFindingParameters	m_peakFindingParams
c3array *	m_houghSpace {0}

Detailed Description

Definition at line 52 of file NDFinderPeakFinder.h.

Constructor & Destructor Documentation

NDFinderPeakFinder()

NDFinderPeakFinder ( const PeakFindingParameters & parameters )

inlineexplicit

Definition at line 66 of file NDFinderPeakFinder.h.

: m_peakFindingParams(parameters) {}

Member Function Documentation

clearHoughSpaceRow()

void clearHoughSpaceRow ( const DeletionBounds & bounds )

private

Definition at line 124 of file NDFinderPeakFinder.cc.

{
  for (c3index phiIdx = bounds.phiLowerBound; phiIdx < bounds.phiUpperBound; ++phiIdx) {
    c3index phiIdxMod = (phiIdx + m_peakFindingParams.nPhi) % m_peakFindingParams.nPhi;
    (*m_houghSpace)[bounds.omega][phiIdxMod][bounds.cot] = 0;
  }
}

deletePeakSurroundings()

void deletePeakSurroundings ( const HoughPeak & peak )

private

Definition at line 81 of file NDFinderPeakFinder.cc.

{
  c3index omegaPeak = peak.cell[0];
  c3index phiPeak = peak.cell[1];
 
  c3index omegaLowerBound = std::max<unsigned short>(0, omegaPeak - m_peakFindingParams.omegaTrim);
  c3index omegaUpperBound = std::min<unsigned short>(m_peakFindingParams.nOmega, omegaPeak + m_peakFindingParams.omegaTrim + 1);
 
  for (c3index cotIdx = 0; cotIdx < m_peakFindingParams.nCot; ++cotIdx) {
    for (c3index omegaIdx = omegaLowerBound; omegaIdx < omegaUpperBound; ++omegaIdx) {
      c3index phiCell = phiPeak + omegaPeak - omegaIdx;
      c3index relativePhi = phiCell - phiPeak;
 
      if (relativePhi > 0) {
        DeletionBounds firstBounds = {
          phiCell - m_peakFindingParams.phiTrim,
          static_cast<c3index>(phiCell + m_peakFindingParams.phiTrim + std::floor(2.4 * relativePhi)),
          omegaIdx,
          cotIdx
        };
        clearHoughSpaceRow(firstBounds);
      } else if (relativePhi < 0) {
        DeletionBounds secondBounds = {
          static_cast<c3index>(phiCell - m_peakFindingParams.phiTrim + std::ceil(2.4 * relativePhi)),
          phiCell + m_peakFindingParams.phiTrim + 1,
          omegaIdx,
          cotIdx
        };
        clearHoughSpaceRow(secondBounds);
      } else {
        DeletionBounds thirdBounds = {
          phiCell - m_peakFindingParams.phiTrim,
          phiCell + m_peakFindingParams.phiTrim + 1,
          omegaIdx,
          cotIdx
        };
        clearHoughSpaceRow(thirdBounds);
      }
    }
  }
}

findPeaks()

std::vector< HoughPeak > findPeaks

(

)

Definition at line 19 of file NDFinderPeakFinder.cc.

{
  std::vector<HoughPeak> candidatePeaks;
  c3array houghSpaceBackup = *m_houghSpace;
  for (unsigned short quadrant = 0; quadrant < 4; ++quadrant) {
    for (unsigned short section = 0; section < 4; ++section) {
      std::array<c3index, 2> sectionBounds = getSectionBounds(quadrant, section);
      iterateOverSection(sectionBounds, candidatePeaks);
      if (m_peakFindingParams.iterations > 1) *m_houghSpace = houghSpaceBackup;
    }
  }
  return candidatePeaks;
}

getSectionBounds()

std::array< c3index, 2 > getSectionBounds ( const unsigned short quadrant, const unsigned section )

private

Definition at line 34 of file NDFinderPeakFinder.cc.

{
  c3index quadrantSize = m_peakFindingParams.nPhi / 4;
  c3index quadrantOffset = m_peakFindingParams.nPhi / 8;
  c3index sectionSize = quadrantSize / 4;
 
  c3index quadrantStart = quadrant * quadrantSize + quadrantOffset;
  c3index sectionStart = quadrantStart + section * sectionSize;
  c3index sectionEnd = sectionStart + sectionSize;
 
  return {sectionStart, sectionEnd};
}

getSectionPeak()

HoughPeak getSectionPeak ( const std::array< c3index, 2 > & sectionBounds )

private

Definition at line 59 of file NDFinderPeakFinder.cc.

{
  unsigned int peakValue = 0;
  cell_index peakCell = {0, 0, 0};
  for (c3index omegaIdx = 0; omegaIdx < m_peakFindingParams.nOmega; ++omegaIdx) {
    for (c3index phiIdx = sectionBounds[0]; phiIdx < sectionBounds[1]; ++phiIdx) {
      c3index phiIdxMod = phiIdx % m_peakFindingParams.nPhi;
      for (c3index cotIdx = 0; cotIdx < m_peakFindingParams.nCot; ++cotIdx) {
        if ((*m_houghSpace)[omegaIdx][phiIdxMod][cotIdx] > peakValue) {
          peakValue = (*m_houghSpace)[omegaIdx][phiIdxMod][cotIdx];
          peakCell = {omegaIdx, phiIdxMod, cotIdx};
        }
      }
    }
  }
  HoughPeak peak;
  peak.cell = peakCell;
  peak.weight = peakValue;
  return peak;
}

iterateOverSection()

void iterateOverSection ( const std::array< c3index, 2 > & sectionBounds, std::vector< HoughPeak > & candidatePeaks )

private

Definition at line 48 of file NDFinderPeakFinder.cc.

{
  for (unsigned short _ = 0; _ < m_peakFindingParams.iterations; ++_) {
    HoughPeak sectionPeak = getSectionPeak(sectionBounds);
    if (sectionPeak.weight > 0) candidatePeaks.push_back(sectionPeak);
    if (m_peakFindingParams.iterations > 1) deletePeakSurroundings(sectionPeak);
  }
}

setNewPlane()

void setNewPlane ( c3array & houghSpace )

inline

Definition at line 69 of file NDFinderPeakFinder.h.

{ m_houghSpace = &houghSpace; }

Member Data Documentation

m_houghSpace

c3array* m_houghSpace {0}

private

Definition at line 87 of file NDFinderPeakFinder.h.

{0};

m_peakFindingParams

PeakFindingParameters m_peakFindingParams

private

Definition at line 85 of file NDFinderPeakFinder.h.

---

The documentation for this class was generated from the following files:

• trg/cdc/include/NDFinderPeakFinder.h
• trg/cdc/src/NDFinderPeakFinder.cc