WindowMNISTDigit.hpp

Go to the documentation of this file.

#ifndef SRC_OPENGL_VISUALIZATION_WINDOWMNISTDIGIT_HPP_
#define SRC_OPENGL_VISUALIZATION_WINDOWMNISTDIGIT_HPP_

#include <opengl/visualization/Window.hpp>

// Dependencies on core types.
#include <types/TensorTypes.hpp>
#include <types/Position2D.hpp>
using namespace mic::types;

namespace mic {
namespace opengl {
namespace visualization {

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


class WindowMNISTDigit: public Window {
public:
    WindowMNISTDigit(std::string name_ = "MNISTDigit",
            unsigned int position_x_ = 0, unsigned int position_y_ = 0,
            unsigned int width_ = 512, unsigned int height_ = 512);

    virtual ~WindowMNISTDigit();


    void displayHandler(void);

    void setDigitPointer(mic::types::TensorXfPtr displayed_digit_);

    void setPathPointer(std::shared_ptr<std::vector <mic::types::Position2D> > saccadic_path_);

private:

    mic::types::TensorXfPtr displayed_digit;

    std::shared_ptr<std::vector <mic::types::Position2D> > saccadic_path;

};

} /* namespace visualization */
} /* namespace opengl */
} /* namespace mic */

#endif /* SRC_OPENGL_VISUALIZATION_WINDOWMNISTDIGIT_HPP_ */