QBruteForceTrendFinder.hh

Go to the documentation of this file.

#ifndef _Q_BRUTE_FORCE_TREND_FINDER_HH_
#define _Q_BRUTE_FORCE_TREND_FINDER_HH_

#include <map>
#include <string>
#include <vector>

class QBruteForceTrendFinder {

public:
    class Line {
    public:
        double intercept;
        double slope;
    };
    
    QBruteForceTrendFinder();
    
    virtual ~QBruteForceTrendFinder();
    
    void FindTrends(int minPointsInInterval = 3);
    
    const std::vector<Line>& GetBestFitLines()
    {
        return fBestFitLines;
    }
    
    const std::vector<double>& GetBreakTimes()
    {
        return fBreakTimes;
    }
    
    void RemoveOutliers(const double nMAD = 5.0 * 1.4826);
    
    void SetMinProbability(const double minProbability)
    {
        fMinProbability = minProbability;
    }
    
    void SetPoint(const double time,
                  const double baseline,
                  const double amplitude);
    
protected:
    double GetDeviation(const int start, const int stop);
    double GetDeviation(const int start, const int stop,
                        double& intercept, double& slope);
    double GetDeviation(const std::vector<int>& breakPoints);
    bool Next(std::vector<int>& breakPoints, int minPointsInInterval,
              int numPoints);

    class Point {
    public:
        double time;
        double baseline;
        double amplitude;
        
        bool operator<(const Point& other) const
        {
            return time < other.time;
        }
    };
    
    double* fAmplitudes;
    double* fBaselines;
    std::vector<Line> fBestFitLines;
    std::vector<double> fBreakTimes;
    std::map< int, std::map<int, double> > fDeviationCache;
    double fMinProbability;
    std::vector<Point> fPoints;
    
};

#endif

---