UIntMatrixEncoder.hpp

Go to the documentation of this file.

#ifndef SRC_ENCODERS_UINTMATRIXENCODER_HPP_
#define SRC_ENCODERS_UINTMATRIXENCODER_HPP_

#include <encoders/MatrixSDREncoder.hpp>

#include <logger/Log.hpp>

#include <cstring> // for memset
#ifdef _WIN32
#include <string.h>
#endif

namespace mic {
namespace encoders {

template<typename T = float>
class UIntMatrixEncoder: public mic::encoders::MatrixSDREncoder<unsigned int, T> {
public:
    UIntMatrixEncoder(size_t sdr_length_) : MatrixSDREncoder<unsigned int, T>(sdr_length_) {
//      std::cout<<" Hello CharMatrixXfEncoder\n";
    };

    ~UIntMatrixEncoder() { }

    virtual mic::types::MatrixPtr<T> encodeSample(const std::shared_ptr<unsigned int>& sample_){
        // Create returned SDR - a vector, i.e. a matrix with number of columns equal to 1.
        mic::types::MatrixPtr<T> sdr (new mic::types::Matrix<T>(sdr_length,1));
        // Set all zeros.
        sdr->setZero();

        unsigned int index = (*sample_);

        if (index >= sdr_length)
            LOG(LERROR) << "The SDR is too short for proper encoding of "<<index<<"!";
        else
            (*sdr)(index) = 1;
        return sdr;
    }

    virtual std::shared_ptr<unsigned int> decodeSample(const mic::types::MatrixPtr<T>& sdr_){
        // SDR must be in fact a vector, with number of columns equal to 1.
        assert(sdr_->cols() == 1);

        // Find index of max value.
        typename mic::types::Matrix<T>::Index maxRow, maxCol;
        float max_value = sdr_->maxCoeff(&maxRow, &maxCol);
        LOG(LDEBUG) << "(maxRow, maxCol) = max_value [ (" << maxRow << "," << maxCol <<") = " << max_value << "]";

        // Convert to int.
        unsigned int decoded = maxRow;

        return std::make_shared<unsigned int>(decoded);
    }

private:
    using MatrixSDREncoder<unsigned int, T>::sdr_length;
};


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

#endif /* SRC_ENCODERS_UINTMATRIXENCODER_HPP_ */