target_monitor.cpp

Go to the documentation of this file.

#include <ros/ros.h>
#include <tf2/utils.h>
#include <actionlib/server/simple_action_server.h>
#include <geometry_msgs/Pose.h>
#include <cpswarm_msgs/TaskAllocatedEvent.h>
#include <cpswarm_msgs/TargetPositionEvent.h>
#include <cpswarm_msgs/TargetTracking.h>
#include <cpswarm_msgs/TargetAction.h>
#include "lib/targets.h"

using namespace std;
using namespace ros;

typedef actionlib::SimpleActionServer<cpswarm_msgs::TargetAction> action_server_t;

targets* monitor;

geometry_msgs::Pose pose;

bool pose_valid;

double yaw ()
{
    tf2::Quaternion orientation;
    tf2::fromMsg(pose.orientation, orientation);
    return tf2::getYaw(orientation);
}

geometry_msgs::Quaternion rotate (geometry_msgs::Quaternion rotation)
{
    tf2::Quaternion cps_orientation;
    tf2::fromMsg(pose.orientation, cps_orientation);
    tf2::Quaternion target_rotation;
    tf2::fromMsg(rotation, target_rotation);
    tf2::Quaternion target_orientation = target_rotation * cps_orientation;
    target_orientation.normalize();
    return tf2::toMsg(target_orientation);
}

void tracking_callback (const cpswarm_msgs::TargetTracking::ConstPtr& msg)
{
    // compute distance and direction of target
    double distance = hypot(msg->tf.translation.x, msg->tf.translation.y);
    double direction = yaw() + atan2(msg->tf.translation.y, -msg->tf.translation.x) - M_PI / 2; // x is inverted in tracking camera tf

    // create position event message to update target internally
    cpswarm_msgs::TargetPositionEvent event;
    event.id = msg->id;
    event.pose.pose.position.x = pose.position.x + distance * cos(direction);
    event.pose.pose.position.y = pose.position.y + distance * sin(direction);
    event.pose.pose.orientation = rotate(msg->tf.rotation);
    event.header.stamp = msg->header.stamp;
    monitor->update(event, TARGET_TRACKED);
}

void found_callback (const cpswarm_msgs::TargetPositionEvent::ConstPtr& msg)
{
    monitor->update(*msg, TARGET_KNOWN);
}

void update_callback (const cpswarm_msgs::TargetPositionEvent::ConstPtr& msg)
{
    monitor->update(*msg, TARGET_KNOWN);
}

void assigned_callback (const cpswarm_msgs::TaskAllocatedEvent::ConstPtr& msg)
{
    // create position event message to update target internally
    cpswarm_msgs::TargetPositionEvent event;
    event.id = msg->task_id;
    event.header.stamp = msg->header.stamp;
    monitor->update(event, TARGET_ASSIGNED);
}

void lost_callback (const cpswarm_msgs::TargetPositionEvent::ConstPtr& msg)
{
    monitor->update(*msg, TARGET_LOST);
}

void done_callback (const cpswarm_msgs::TargetPositionEvent::ConstPtr& msg)
{
    monitor->update(*msg, TARGET_DONE);
}

void pose_callback (const geometry_msgs::PoseStamped::ConstPtr& msg)
{
    // valid pose received
    if (msg->header.stamp.isValid())
        pose_valid = true;

    // store new position and orientation in class variables
    pose = msg->pose;

    ROS_DEBUG_THROTTLE(1, "Yaw %.2f", yaw());
    ROS_DEBUG_THROTTLE(1, "Pose [%.2f, %.2f, %.2f]", pose.position.x, pose.position.y, pose.position.z);
}

void set_done(const cpswarm_msgs::TargetGoalConstPtr& goal, action_server_t* as)
{
    // create position event message
    cpswarm_msgs::TargetPositionEvent event;
    event.id = goal->id;
    event.pose = goal->pose;
    event.header = goal->pose.header;

    // update target internally
    monitor->update(event, TARGET_DONE);

    // action server finished successfully
    as->setSucceeded();
}

int main (int argc, char** argv)
{
    // init ros node
    init(argc, argv, "target_monitor");
    NodeHandle nh;

    // read parameters
    bool simulation;
    nh.param(this_node::getName() + "/simulation", simulation, false);
    int queue_size;
    nh.param(this_node::getName() + "/queue_size", queue_size, 10);

    // rate of update loop
    double loop_rate;
    nh.param(this_node::getName() + "/loop_rate", loop_rate, 5.0);
    Rate rate(loop_rate);

    // create target monitor
    monitor = new targets();

    // read target positions from parameter file
    if (simulation) {
        monitor->simulate();
    }

    // no pose received yet
    pose_valid = false;

    // publishers and subscribers
    Subscriber pose_sub = nh.subscribe("pos_provider/pose", queue_size, pose_callback);
    Subscriber tracking_sub = nh.subscribe("target_tracking", queue_size, tracking_callback);
    Subscriber local_assigned_sub = nh.subscribe("cps_selected", queue_size, assigned_callback);
    Subscriber local_done_sub = nh.subscribe("target_done", queue_size, done_callback);
    Subscriber found_sub = nh.subscribe("bridge/events/target_found", queue_size, found_callback);
    Subscriber update_sub = nh.subscribe("bridge/events/target_update", queue_size, update_callback);
    Subscriber assigned_sub = nh.subscribe("bridge/events/cps_selected", queue_size, assigned_callback);
    Subscriber lost_sub = nh.subscribe("bridge/events/target_lost", queue_size, lost_callback);
    Subscriber done_sub = nh.subscribe("bridge/events/target_done", queue_size, done_callback);

    // action servers
    action_server_t set_done_as(nh, "cmd/target_done", boost::bind(&set_done, _1, &set_done_as), false);
    set_done_as.start();

    // init position and yaw
    while (ok() && pose_valid == false) {
        ROS_DEBUG_ONCE("Waiting for valid pose...");
        rate.sleep();
        spinOnce();
    }

    // update information about targets
    while (ok()) {
        monitor->update(pose);
        rate.sleep();
        spinOnce();
    }

    // destroy target monitor
    delete monitor;

    return 0;
}