#ifndef SIMPLE_ENV_H
#define SIMPLE_ENV_H

//#include "tdagent.H"
#include <math.h>
#include <boost/numeric/ublas/matrix.hpp>
#include <boost/numeric/ublas/vector.hpp>
#include <boost/numeric/ublas/io.hpp>
#include <boost/numeric/ublas/matrix_proxy.hpp>
#include <boost/numeric/ublas/vector_proxy.hpp>

#define PI 3.1415926535897932384626433832795

namespace ublas = boost::numeric::ublas;
using namespace boost::numeric::ublas;

class simple_env
{
public:
	simple_env();
	~simple_env();
	void create();
	float preform(float u);
	float get_reward();
	ublas::vector<float> cov;
	ublas::vector<float> get_state();
	ublas::vector<float> get_previous_state();
	ublas::vector<float> get_final_state();
	ublas::vector<float> get_cov();

private:
	float sigmoid(float u);
	float reward;
	ublas::vector<float> state;
	ublas::vector<float> prev_state;
	ublas::vector<float> final_state;

	
};

#endif