reward_model.h

Go to the documentation of this file.

#ifndef _REWARD_MODEL_H_
#define _REWARD_MODEL_H_

#include <mdm_library/common_defs.h>



namespace mdm_library
{
class RewardModel
{
public:
    virtual double getReward ( uint32_t state,
                               uint32_t action ) = 0;
};

class RewardMatrix : public RewardModel
{
public:
    RewardMatrix ( MatrixPtr matrix_ptr ) :
        R_ptr_ ( matrix_ptr ) {}

    double getReward ( uint32_t state,
                       uint32_t action )
    {
        return ( *R_ptr_ ) ( state, action );
    }
private:
    MatrixPtr R_ptr_;
};

#ifdef HAVE_MADP

class MADPReward : public RewardModel
{

};

#endif
}

#endif