RFFHitFitter.h

Go to the documentation of this file.

#ifndef RFFHITFITTER_H
#define RFFHITFITTER_H

#include <set>
#include <vector>

#include "GaussianEliminationAlg.h"

namespace hit {

  struct SignalSetComp {
    bool operator()(const std::pair<float, float>& lhs, const std::pair<float, float>& rhs) const
    {
      return lhs.first < rhs.first;
    }
  };

  class RFFHitFitter {

    typedef std::pair<float, float> MeanSigmaPair;

  public:
    RFFHitFitter(float, unsigned int, float, float step = 0.1, float max = 5.0);
    RFFHitFitter(float step = 0.1, float max = 5.0);

    void SetFitterParams(float, unsigned int, float);

    void RunFitter(const std::vector<float>& signal);

    const std::vector<float>& MeanVector() { return fMeanVector; }
    const std::vector<float>& SigmaVector() { return fSigmaVector; }
    const std::vector<float>& MeanErrorVector() { return fMeanErrorVector; }
    const std::vector<float>& SigmaErrorVector() { return fSigmaErrorVector; }
    const std::vector<float>& AmplitudeVector() { return fAmpVector; }
    const std::vector<float>& AmplitudeErrorVector() { return fAmpErrorVector; }
    unsigned int NHits() { return fMeanVector.size(); }

    void ClearResults();

    void PrintResults();

  private:
    float fMeanMatchThreshold;
    unsigned int fMinMergeMultiplicity;
    float fFinalAmpThreshold;

    // Clang: private field is not used (x2):
    //  float fGEAlgStepSize;
    //  float fGEAlgMax;
    util::GaussianEliminationAlg fGEAlg;

    std::vector<float> fMeanVector;
    std::vector<float> fSigmaVector;
    std::vector<float> fMeanErrorVector;
    std::vector<float> fSigmaErrorVector;
    std::vector<float> fAmpVector;
    std::vector<float> fAmpErrorVector;

    std::multiset<MeanSigmaPair, SignalSetComp> fSignalSet;
    std::vector<std::vector<std::multiset<MeanSigmaPair>::iterator>> fMergeVector;

    void CalculateAllMeansAndSigmas(const std::vector<float>& signal);
    void CalculateMergedMeansAndSigmas(std::size_t signal_size);
    void CalculateAmplitudes(const std::vector<float>& signal);
    void CreateMergeVector();

    bool HitsBelowThreshold();
  };

}

#endif