Colony » Colony Scout » Colony Scout OS

#include "WH_Robot.h"

#include "../helper_classes/Order.h"

/** @Brief: warehouse robot constructor **/

WH_Robot::WH_Robot(std::string scoutname):Behavior(scoutname, "WH_Robot")

{

  nav_map = new navigationMap(scoutname);

  curr_task = DEFAULT_TASK;

  name = scoutname;

  reg_failed = 1;

}

WH_Robot::~WH_Robot()

{

  delete(nav_map);

}

Duration WH_Robot::get_worst_case_time(State start_state, State target_state)

{

  return nav_map->get_worst_case_time(start_state, target_state);

}

int WH_Robot::exec_task()

{

  ROS_INFO("WH_robot: Executing a task");

  assert(curr_task != DEFAULT_TASK);

  //TODO: do task

  srand(0xDEADBEEF);

  int error = rand() % 12;

  if(error < 9) //Fail with 1/4 probability

  {

    ROS_INFO("WH_robot: TASK COMPLETE!");

    return TASK_COMPLETED;

  }

  else

  {

    ROS_INFO("WH_robot: TASK FAILED!");

    return TASK_FAILED;

  }

}

void WH_Robot::robot_callback(const std_msgs::String::ConstPtr& msg)

{

  std::cout << "robot callback called." << std::endl;

  if(msg->data.compare(0, 11, "REG_SUCCESS") == 0)

  {

    id = atoi(msg->data.substr(12).c_str());

    reg_failed = 0;

  }

  else if(msg->data.compare(0, 8, "SET_TASK") == 0)

  {

    ROS_INFO("WH_ROBOT: Setting robot task");

    char* string = (char*)msg->data.c_str();

    char* data;

    data = strtok(string, " ");

    int order_id = atoi(data);

    data = strtok(string, " ");

    Address source = atoi(data);

    data = strtok(string, " ");

    Address dest = atoi(data);

    Time start_time = ros::Time::now();

    Path path;

    path.len = 0;

    path.path = NULL;

    Duration est_time(0);

    curr_task = new Order(order_id, source, dest, start_time, path, est_time);

  }

  else

  {

    ROS_INFO("I got a bad message: %s", msg->data.c_str());

  }

}

void WH_Robot::run ()

{

  ROS_INFO("A WH_Robot has been created");

  robot_to_sched = node.advertise<std_msgs::String>("robot_to_sched", 1000);

  sched_to_robot = node.subscribe(name + "_topic", 1000, &WH_Robot::robot_callback, this);

  ROS_INFO("I am a robot. Registering with scheduler...");

  while(reg_failed && ok())

  {

    std_msgs::String msg;

    std::stringstream ss;

    ss << "REGISTER "<<name;

    msg.data = ss.str();

    robot_to_sched.publish(msg);

    spinOnce();

  }

  while(ok())

  {

    ROS_INFO("WH_ROBOT: Running main loop");

      std_msgs::String msg;

      std::stringstream ss;

      ss << "GET_TASK "<<id;

      msg.data = ss.str();

      robot_to_sched.publish(msg);

    while(curr_task == DEFAULT_TASK && ok()){

      spinOnce();

    }

    int error = exec_task();

    if(error == TASK_COMPLETED)

    {

      std_msgs::String msg;

      std::stringstream ss;

      ss << "SUCCESS "<<curr_task->getid();

      msg.data = ss.str();

      robot_to_sched.publish(msg);

    }

    else //error == TASK_FAILED

    {

      std_msgs::String msg;

      std::stringstream ss;

      ss << "FAILED "<<curr_task->getid();

      msg.data = ss.str();

      robot_to_sched.publish(msg);

    }

    delete curr_task;

    curr_task = DEFAULT_TASK;

  }

}

void WH_Robot::set_task(Order order)

{

  curr_task = new Order(order.getid(), order.get_source(), order.get_dest(), 

      order.get_start_time(), order.get_path(), order.get_est_time()); 

  return;

}