GridWorld.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_MDP_GRIDWORLD_HEADER_FILE
#define AI_TOOLBOX_MDP_GRIDWORLD_HEADER_FILE
 
#include <cstddef>
#include <array>
 
namespace AIToolbox::MDP {
    namespace GridWorldUtils {
        enum Direction : size_t { UP = 0, RIGHT = 1, DOWN = 2, LEFT = 3, UP_RIGHT = 4, UP_LEFT = 5, DOWN_RIGHT = 6, DOWN_LEFT = 7 };
        static constexpr std::array<Direction, 4> Directions4{{UP, RIGHT, DOWN, LEFT}};
        static constexpr std::array<Direction, 8> Directions8{{UP, RIGHT, DOWN, LEFT, UP_RIGHT, UP_LEFT, DOWN_RIGHT, DOWN_LEFT}};
    }
 
    class GridWorld {
        public:
            using Direction = GridWorldUtils::Direction;
            struct State {
                operator size_t() const;
                unsigned getX() const;
                unsigned getY() const;
 
                bool operator==(const State & other) const;
 
                private:
                    State(int xx, int yy, size_t ss);
                    int x, y;
                    size_t s;
                    friend GridWorld;
            };
 
            GridWorld(unsigned width, unsigned height, bool torus = false);
 
            State getAdjacent(Direction d, State s) const;
 
            State getAdjacent(size_t d, State s) const;
 
            unsigned distance(const State & s1, const State & s2) const;
 
            State operator()(int x, int y) const;
 
            State operator()(const size_t s) const;
 
            unsigned getWidth() const;
 
            unsigned getHeight() const;
 
            bool isTorus() const;
 
            size_t getS() const;
 
        private:
            int boundX(int x) const;
            int boundY(int y) const;
 
            unsigned width_, height_;
            bool isTorus_;
    };
}
 
#endif