Action2D.hpp

Go to the documentation of this file.

#ifndef SRC_TYPES_Action_HPP_
#define SRC_TYPES_Action_HPP_

#include <random>
#include <iostream>

namespace mic {
namespace types {

enum class NESW : short
{
    North = 0, 
    East, 
    South, 
    West, 
    None, 
    Random, 
    Exit 
};


class Action2DInterface {
public:

    int dx;

    int dy;

    Action2DInterface() : dx(0), dy(0) { };

    friend std::ostream& operator<<(std::ostream& os_, const Action2DInterface& ac_)
    {
        os_ << "[dx,dy]: [" << ac_.dx << ',' << ac_.dy << "]";
        return os_;
    }

protected:
};


template <typename ActionType>
class Action2D : public mic::types::Action2DInterface {
public:
    virtual ~Action2D() { };

    virtual void setAction(ActionType type_) = 0;

    ActionType getType() { return type;}

protected:
    ActionType type;

    Action2D() : Action2DInterface() { };

};

class NESWAction : public mic::types::Action2D<NESW> {
public:
    NESWAction() : Action2D()  { type = NESW::None; };

    NESWAction(NESW type_) : Action2D() { setAction(type_); };

    NESWAction(size_t type_) : Action2D() { setAction((NESW)type_); };

    void setAction(NESW type_) {
        type = type_;
        switch (type_) {
            case NESW::North:   dy =-1; dx =0; break;
            case NESW::East:    dy =0; dx =1; break;
            case NESW::South:   dy =1; dx =0; break;
            case NESW::West:    dy =0; dx =-1; break;
            case NESW::None:
            case NESW::Exit:
            default:    dy =0; dx =0;
        }//: switch
    }

    friend std::ostream& operator<<(std::ostream& os_, const NESWAction& ac_)
    {
        os_ << "[dx,dy]: [" << ac_.dx << ',' << ac_.dy << "]";
        switch (ac_.type) {
        case NESW::North:   os_ << " (North)"; break;
        case NESW::East:    os_ << " (East)"; break;
        case NESW::South:   os_ << " (South)"; break;
        case NESW::West:    os_ << " (West)"; break;
        case NESW::None:    os_ << " (None)"; break;
        case NESW::Random:  os_ << " (Random)"; break;
        case NESW::Exit:    os_ << " (Exit)"; break;
        default: break;
        }//: switch

        return os_;
    }

};


class RandomNESWAction : public mic::types::NESWAction {
public:
    RandomNESWAction() : NESWAction(),
        rng_mt19937_64(rd())
    {
        type = NESW::Random;

        // Initialize uniform integer distribution.
        std::uniform_int_distribution<> index_dist((int)NESW::North, (int)NESW::West);

        // Select a random action.
        setAction((NESW) index_dist(rng_mt19937_64));
    };


private:
    std::random_device rd;

    std::mt19937_64 rng_mt19937_64;

};


class ExitAction : public mic::types::NESWAction {
public:
    ExitAction() : NESWAction() { type = NESW::Exit; };

};


#define A_NORTH mic::types::NESWAction(mic::types::NESW::North)

#define A_EAST mic::types::NESWAction(mic::types::NESW::East)

#define A_SOUTH mic::types::NESWAction(mic::types::NESW::South)

#define A_WEST mic::types::NESWAction(mic::types::NESW::West)

#define A_RANDOM mic::types::RandomNESWAction()

#define A_EXIT mic::types::ExitAction()

#define A_NONE mic::types::NESWAction(mic::types::NESW::None)



} /* namespace types */
} /* namespace mic */

#endif /* SRC_TYPES_Action_HPP_ */