CompressionMatrixFactory.h

Go to the documentation of this file.

#ifndef OPENANN_COMPRESSION_MATRIX_FACTORY_H_
#define OPENANN_COMPRESSION_MATRIX_FACTORY_H_

#include <Eigen/Core>
#include <vector>

namespace OpenANN
{

class CompressionMatrixFactory
{
public:
  bool compress;
  enum Transformation
  {
    DCT, GAUSSIAN, SPARSE_RANDOM, AVERAGE, EDGE
  } transformation;
  int inputDim;
  int paramDim;

  CompressionMatrixFactory();
  CompressionMatrixFactory(int inputDim, int paramDim,
                           Transformation transformation = DCT);
  void createCompressionMatrix(Eigen::MatrixXd& cm);

private:
  void fillCompressionMatrix(Eigen::MatrixXd& cm);
};

} // namespace OpenANN

#endif // OPENANN_COMPRESSION_MATRIX_FACTORY_H_