SVDMaxSumAlgorithm.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <vxd/dataobjects/VxdID.h>
#include <svd/dataobjects/SVDShaperDigit.h>
#include <vector>
 
namespace Belle2::SVD {
 
  class SVDMaxSumAlgorithm {
 
  public:
 
 
    explicit SVDMaxSumAlgorithm(const Belle2::SVDShaperDigit::APVFloatSamples samples)
      : m_samples(samples)
    {
      applyAlgorithm();
    };
 
 
    virtual ~SVDMaxSumAlgorithm() {};
 
    int getFirstFrame() {return m_result.first;}
 
    std::vector<float> getSelectedSamples() {return m_result.second;}
 
 
  protected:
 
    Belle2::SVDShaperDigit::APVFloatSamples m_samples; 
    std::pair<int, std::vector<float>> m_result;
 
    void applyAlgorithm()
    {
 
      //Max Sum selection
      if (m_samples.size() < 3)
        B2ERROR("APV25 samples less than 3!?");
 
      std::vector<float> Sum2bin(m_samples.size() - 1, 0);
 
      for (int iBin = 0; iBin < static_cast<int>(m_samples.size()) - 1; ++iBin)
        Sum2bin.at(iBin) = m_samples.at(iBin) + m_samples.at(iBin + 1);
 
      auto itSum = std::max_element(std::begin(Sum2bin), std::end(Sum2bin));
 
      int ctrFrame = std::distance(std::begin(Sum2bin), itSum);
 
      if (ctrFrame == 0) ctrFrame = 1;
 
      std::vector<float> selectedSamples = {m_samples.at(ctrFrame - 1), m_samples.at(ctrFrame), m_samples.at(ctrFrame + 1)};
 
      m_result.first = ctrFrame - 1;
      m_result.second =  selectedSamples;
 
    };
 
  };
 
}