transitionMatrix.h

Go to the documentation of this file.

#ifndef MDPTRANSITIONMATRIX_H
#define MDPTRANSITIONMATRIX_H

#include <memory>
#include <ostream>
#include <vector>

#include "state.h"
#include "action_impl.h"

namespace Utils{class RandomGenerator;}

namespace Mdp
{

class TransitionMatrix
{
public:
    TransitionMatrix(int nbOfStates, int nbOfActions);
    ~TransitionMatrix();
    void set(state_t from, state_t to, action_t action, double proba);
    double get(state_t from, state_t to, action_t action);
    double get(size_t stateAndAction, state_t to);
    void initializeRandomly(std::shared_ptr<Utils::RandomGenerator> gen);
    void print(std::ostream& stream);
private:
    double *getArray();

    double *matrix{nullptr};
    size_t nbOfStates;
    size_t nbOfActions;
};


}

#endif