pomdp3piles.h

Go to the documentation of this file.

#ifndef _wam_pomdp3piles_h_
#define _wam_pomdp3piles_h_

#include "ros/this_node.h"
#include "ros/names.h"
#include "ros/node_handle.h"
#include "ros/rate.h"
#include <list>

#include "mutex.h"
#include "GenericPolicy.h"

// [publisher subscriber headers]

// [service client headers]
#include "std_srvs/Empty.h"
#include "std_msgs/Int32.h"
#include "estirabot_msgs/belief_summary.h"
#include "estirabot_msgs/belief.h"
#include "estirabot_msgs/obs2action.h"

// [action server client headers]

#define LOGFILE "log.txt"

#define MAXNUMB 5 //maxnumb + 1
#define AZONEMSG 1
#define BZONEMSG 2
#define ANYZONE 0

enum {
    //remember to change the other .h enums (dummy_pomdp)!
    TAKEHIGH,
    TAKELOW,
    CHANGEPILE,
    FOLDLOW,
    SPREAD,
    FLIP,
    SHAKE,
    FOLDHIGH,
    MOVEAWAY,
    DUMMY
};

enum {
    STRAIGHT,
    ISOMETRIC,
    PEG
};

class Pomdp
{
  private:
    CMutex execution_mutex;
    bool execution_flag;
    int actioncount;
    int last_action_;
    std::string pomdpfile_;
    std::vector<int> probablestates;

    ros::NodeHandle node_handle_;

    Policy2piles *myPolicy;
    
    // [publisher attributes]
    ros::Publisher focused_obj_label_publisher;
    std_msgs::Int32 ObjLabel_msg;
    ros::Publisher belief_summary_publisher;
    estirabot_msgs::belief_summary belief_summary_msg;
    ros::Publisher belief_publisher;
    estirabot_msgs::belief belief_msg;

    // [subscriber attributes]
        
    // [service attributes]
    ros::ServiceServer execution_flow_control_server_;
    bool execution_flow_controlCallback(std_srvs::Empty::Request &req, std_srvs::Empty::Response &res);
    ros::ServiceServer obs2action_server_;
    bool obs2actionCallback(estirabot_msgs::obs2action::Request &req, estirabot_msgs::obs2action::Response &res);
    ros::ServiceServer resetPOMDP_server_;
    bool resetPOMDPCallback(std_srvs::Empty::Request &req, std_srvs::Empty::Response &res);
            
    // [client attributes]
            
    // [action server attributes]
            
    // [action client attributes]

    void publish_belief();
  public:
    Pomdp();

    void mainLoop(void);

};
#endif