FindPeakFastPV.h

Go to the documentation of this file.

#ifndef RecoPixelVertexing_FindPeakFastPV_h
#define RecoPixelVertexing_FindPeakFastPV_h

#include <vector>

float FindPeakFastPV(const std::vector<float> &zProjections,
                     const std::vector<float> &zWeights,
                     const float oldVertex,
                     const float m_zClusterWidth,
                     const float m_zClusterSearchArea,
                     const float m_weightCut) {
  float centerWMax = oldVertex;
  if (m_zClusterWidth > 0 && m_zClusterSearchArea > 0) {
    std::vector<float>::const_iterator itCenter = zProjections.begin();
    std::vector<float>::const_iterator itLeftSide = zProjections.begin();
    std::vector<float>::const_iterator itRightSide = zProjections.begin();
    const float zClusterWidth = m_zClusterWidth * 0.5;  //take half zCluster width
    const float zLowerBound = oldVertex - zClusterWidth - m_zClusterSearchArea;
    const float zUpperBound = oldVertex + zClusterWidth + m_zClusterSearchArea;
    float maxW = 0;
    for (; itCenter != zProjections.end(); itCenter++) {
      //loop only on the zProjections within oldVertex +/- (zClusterWidth + m_zClusterSearchArea)
      if ((*itCenter < zLowerBound) || (*itCenter > zUpperBound))
        continue;

      while (itLeftSide != zProjections.end() && (*itCenter - *itLeftSide) > zClusterWidth)
        itLeftSide++;
      while (itRightSide != zProjections.end() && (*itRightSide - *itCenter) < zClusterWidth)
        itRightSide++;
      float nWeighted = 0;
      float centerW = 0;

      for (std::vector<float>::const_iterator ii = itLeftSide; ii != itRightSide; ii++) {
        //loop inside the peak and calculate its weight
        if (zWeights[ii - zProjections.begin()] < m_weightCut)
          continue;
        nWeighted += zWeights[ii - zProjections.begin()];
        centerW += (*ii) * zWeights[ii - zProjections.begin()];
      }
      centerW /= nWeighted;  //calculate the weighted peak center
      if (nWeighted > maxW) {
        maxW = nWeighted;
        centerWMax = centerW;
      }
    }
  }
  return centerWMax;
}

#endif