LinearPredictor.h

Go to the documentation of this file.

#ifndef _LinearPredictor_h_
#define _LinearPredictor_h_

#include <vector>

#include <Eigen/Dense>

#ifndef __python_test__
#include <fwk/VModule.h>
#endif

using namespace Eigen;


namespace RdChannelLinearPredictorRFISuppressor {
  
  class LinearPredictor {
  public:
    LinearPredictor(unsigned int numCoeffPerChannel, unsigned int delayLine, unsigned int numChannels, unsigned int optimization = 1);
    void train(std::vector< std::vector<double> > const &channels, double fudgeFactor);
    std::vector< std::vector<double> > predict(std::vector< std::vector<double> > const &channels) const;

  private:
      unsigned int numCoeffPerChannel;
      unsigned int numChannels;
      unsigned int delayLine;
      //unsigned int channels; // unused. LN.
      unsigned int dimCovMatr;
      unsigned int dimPredictorCoefficients;
      unsigned int totalPredictorOffset;
      unsigned int optimization;

      MatrixXd covMatr;
      MatrixXd predictorCoefficients;
  };
}

#endif