MatrixSDREncoder.hpp

Go to the documentation of this file.

#ifndef SRC_ENCODERS_MATRIXSDRENCODER_HPP_
#define SRC_ENCODERS_MATRIXSDRENCODER_HPP_

#include <encoders/Encoder.hpp>

#include <types/MatrixTypes.hpp>

namespace mic {
namespace encoders {

template <typename inputDataType, typename outputDataType = float>
class MatrixSDREncoder : public mic::encoders::Encoder <inputDataType, mic::types::Matrix<outputDataType> > {
public:
    MatrixSDREncoder(size_t sdr_length_) : mic::encoders::Encoder<inputDataType, mic::types::Matrix<outputDataType> >(),
        sdr_length(sdr_length_)
    {
//      std::cout<<" Hello MatrixXfSDREncoder\n";
    };

    virtual ~MatrixSDREncoder() { }


    virtual std::shared_ptr<mic::types::Matrix<outputDataType> > encodeBatch(const std::vector<std::shared_ptr<inputDataType> >& batch_) {
        // Create returned matrix.
        std::shared_ptr<mic::types::Matrix<outputDataType> > sdrs (new mic::types::Matrix<outputDataType> (sdr_length, batch_.size()));

        // Encode the samples one by one.
        for (size_t i=0; i < batch_.size(); i++ ) {
            // Encode single sample.
            std::shared_ptr<mic::types::Matrix<outputDataType> > sample_sdr = this->encodeSample(batch_[i]);

            // Set SDR rows.
            sdrs->col(i) = sample_sdr->col(0);
        }//: for

        // Return the matrix containing SDRs.
        return sdrs;
    }


    virtual std::vector<std::shared_ptr<inputDataType> > decodeBatch(const std::shared_ptr<mic::types::Matrix<outputDataType> >& sdrs_) {
        // Create the output vector.
        std::vector<std::shared_ptr<inputDataType> > output;
        std::shared_ptr<mic::types::Matrix<outputDataType> > sample_sdr (new mic::types::Matrix<outputDataType> (sdrs_->rows(), 1));

        // Iterate through columns and decode them one by one.
        for (size_t i=0; i < (size_t)sdrs_->cols(); i++ ) {
            sample_sdr->col(0) = sdrs_->col(i);
            std::shared_ptr<inputDataType> decoded = this->decodeSample(sample_sdr);
            // Add result to vector.
            output.push_back(decoded);
        }//: for

        // Return the decoded vector.
        return output;
    }


    size_t getSDRLength() {
        return sdr_length;
    }

    void setSDRLength(size_t sdr_length_) {
        sdr_length = sdr_length_;
    }

protected:

    size_t sdr_length;
};

} /* namespace encoders */
} /* namespace mic */

#endif /* SRC_ENCODERS_MATRIXSDRENCODER_HPP_ */