FindPeakFastPV.h File Reference

#include <vector>

Go to the source code of this file.

Functions
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)

Function Documentation

FindPeakFastPV()

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  )

inline

Definition at line 13 of file FindPeakFastPV.h.

References cuy::ii.

Referenced by FastPrimaryVertexWithWeightsProducer::produce().

                                                     {
  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;
}