MNISTDigit.hpp

Go to the documentation of this file.

#ifndef SRC_TYPES_MNISTDIGIT_HPP_
#define SRC_TYPES_MNISTDIGIT_HPP_

#include <types/Environment.hpp>
#include <importers/MNISTMatrixImporter.hpp>


namespace mic {
namespace environments {

enum class MNISTDigitChannels : std::size_t
{
    Pixels = 0, 
    Goals = 1, 
    Agent = 2, 
    Count = 3 
};


class MNISTDigit : public mic::environments::Environment {
public:
    MNISTDigit(std::string node_name_ = "mnist_digit");

    MNISTDigit (const mic::environments::MNISTDigit & md_);

    virtual ~MNISTDigit();

    mic::environments::MNISTDigit & operator=(const mic::environments::MNISTDigit & md_);

    virtual void initializePropertyDependentVariables();

    virtual void initializeEnvironment();

    mic::types::TensorXfPtr getObservation();

    virtual std::string environmentToString();

    virtual std::string observationToString();

    virtual mic::types::MatrixXfPtr encodeEnvironment();

    virtual mic::types::MatrixXfPtr encodeObservation();

    virtual mic::types::Position2D getAgentPosition();

    virtual bool moveAgentToPosition(mic::types::Position2D pos_);

    virtual float getStateReward(mic::types::Position2D pos_);

    // Makes all versions of polymorphic method isStateAllowed() available.
    using mic::environments::Environment::isStateAllowed;

    virtual bool isStateAllowed(mic::types::Position2D pos_);

    // Makes all versions of polymorphic method isStateTerminal() available.
    using mic::environments::Environment::isStateTerminal;

    virtual bool isStateTerminal(mic::types::Position2D pos_);

    unsigned int optimalPathLength(){
        return optimal_path_length;
    }

protected:

    mic::importers::MNISTMatrixImporter<float> mnist_importer;

    mic::configuration::Property<size_t> sample_number;

    mic::configuration::Property<short> agent_x;

    mic::configuration::Property<short> agent_y;

    mic::configuration::Property<short> goal_x;

    mic::configuration::Property<short> goal_y;

    unsigned int optimal_path_length;


    std::string toString(mic::types::TensorXfPtr env_);

};

} /* namespace environments */
} /* namespace mic */

#endif /* SRC_TYPES_MNISTDIGIT_HPP_ */