RosThread.cpp

Go to the documentation of this file.

#include "RosThread.h"
#include <unistd.h>
#include "cvd/thread.h"
#include "ros/ros.h"
#include "ros/callback_queue.h"
#include "tum_ardrone_gui.h"
#include "stdio.h"
#include "std_msgs/Empty.h"


pthread_mutex_t RosThread::send_CS = PTHREAD_MUTEX_INITIALIZER;
RosThread::RosThread()
{
        gui = NULL;
        navdataCount = velCount = dronePoseCount = joyCount = velCount100ms = 0;
        keepRunning = true;
        lastJoyControlSent = ControlCommand(0,0,0,0);
        lastL1Pressed = lastR1Pressed = false;
}

RosThread::~RosThread(void)
{


}

void RosThread::startSystem()
{
        keepRunning = true;
        start();
}

void RosThread::stopSystem()
{
        keepRunning = false;
        join();
}

void RosThread::landCb(std_msgs::EmptyConstPtr)
{
        gui->addLogLine("sent: LAND");
}
void RosThread::toggleStateCb(std_msgs::EmptyConstPtr)
{
        gui->addLogLine("sent: ToggleState");
}
void RosThread::takeoffCb(std_msgs::EmptyConstPtr)
{
        gui->addLogLine("sent: Takeoff");
}

void RosThread::droneposeCb(const tum_ardrone::filter_stateConstPtr statePtr)
{
        dronePoseCount++;
}
void RosThread::velCb(const geometry_msgs::TwistConstPtr vel)
{
    velCount++;
    velCount100ms++;
}
void RosThread::navdataCb(const ardrone_autonomy::NavdataConstPtr navdataPtr)
{

        if(navdataCount%10==0)
        {
                char buf[200];
                snprintf(buf,200,"Motors: %f %f %f %f",
                                (float)navdataPtr->motor1,
                                (float)navdataPtr->motor2,
                                (float)navdataPtr->motor3,
                                (float)navdataPtr->motor4);
                gui->setMotorSpeeds(std::string(buf));
        }
        navdataCount++;
}

void RosThread::joyCb(const sensor_msgs::JoyConstPtr joy_msg)
{
    joyCount++;

    if (joy_msg->axes.size() < 4) {
        ROS_WARN_ONCE("Error: Non-compatible Joystick!");
        return;
    }

    // Avoid crashes if non-ps3 joystick is being used
    short unsigned int actiavte_index = (joy_msg->buttons.size() > 11) ? 11 : 1;

        // if not controlling: start controlling if sth. is pressed (!)
    bool justStartedControlling = false;
        if(gui->currentControlSource != CONTROL_JOY)
        {
                if(             joy_msg->axes[0] > 0.1 ||  joy_msg->axes[0] < -0.1 ||
                                joy_msg->axes[1] > 0.1 ||  joy_msg->axes[1] < -0.1 ||
                                joy_msg->axes[2] > 0.1 ||  joy_msg->axes[2] < -0.1 ||
                                joy_msg->axes[3] > 0.1 ||  joy_msg->axes[3] < -0.1 ||
                joy_msg->buttons.at(actiavte_index))
                {
                        gui->setControlSource(CONTROL_JOY);
                        justStartedControlling = true;
                }
        }

        // are we actually controlling with the Joystick?
        if(justStartedControlling || gui->currentControlSource == CONTROL_JOY)
        {
                ControlCommand c;
                c.yaw = -joy_msg->axes[2];
                c.gaz = joy_msg->axes[3];
                c.roll = -joy_msg->axes[0];
                c.pitch = -joy_msg->axes[1];

                sendControlToDrone(c);
                lastJoyControlSent = c;

        if(!lastL1Pressed && joy_msg->buttons.at(actiavte_index - 1))
                        sendTakeoff();
        if(lastL1Pressed && !joy_msg->buttons.at(actiavte_index - 1))
                        sendLand();

        if(!lastR1Pressed && joy_msg->buttons.at(actiavte_index))
                        sendToggleState();

        }
    lastL1Pressed =joy_msg->buttons.at(actiavte_index - 1);
    lastR1Pressed = joy_msg->buttons.at(actiavte_index);
}


void RosThread::comCb(const std_msgs::StringConstPtr str)
{
        if(str->data.substr(0,2) == "u ")
        {
                if(str->data.substr(0,4) == "u l ")
                        gui->addLogLine(str->data.substr(4,str->data.length()-4));

                else if(str->data.substr(0,4) == "u c ")
                        gui->setAutopilotInfo(str->data.substr(4,str->data.length()-4));

                else if(str->data.substr(0,4) == "u s ")
                        gui->setStateestimationInfo(str->data.substr(4,str->data.length()-4));
        }
}

void RosThread::run()
{
        std::cout << "Starting ROS Thread" << std::endl;

    vel_pub        = nh_.advertise<geometry_msgs::Twist>(nh_.resolveName("cmd_vel"),1);
    vel_sub        = nh_.subscribe(nh_.resolveName("cmd_vel"),50, &RosThread::velCb, this);

    tum_ardrone_pub        = nh_.advertise<std_msgs::String>(nh_.resolveName("tum_ardrone/com"),50);
    tum_ardrone_sub        = nh_.subscribe(nh_.resolveName("tum_ardrone/com"),50, &RosThread::comCb, this);


    dronepose_sub          = nh_.subscribe(nh_.resolveName("ardrone/predictedPose"),50, &RosThread::droneposeCb, this);
    navdata_sub    = nh_.subscribe(nh_.resolveName("ardrone/navdata"),50, &RosThread::navdataCb, this);
    joy_sub        = nh_.subscribe(nh_.resolveName("joy"),50, &RosThread::joyCb, this);

    takeoff_pub    = nh_.advertise<std_msgs::Empty>(nh_.resolveName("ardrone/takeoff"),1);
    land_pub       = nh_.advertise<std_msgs::Empty>(nh_.resolveName("ardrone/land"),1);
    toggleState_pub        = nh_.advertise<std_msgs::Empty>(nh_.resolveName("ardrone/reset"),1);

    takeoff_sub    = nh_.subscribe(nh_.resolveName("ardrone/takeoff"),1, &RosThread::takeoffCb, this);
    land_sub       = nh_.subscribe(nh_.resolveName("ardrone/land"),1, &RosThread::landCb, this);
    toggleState_sub        = nh_.subscribe(nh_.resolveName("ardrone/reset"),1, &RosThread::toggleStateCb, this);

    toggleCam_srv        = nh_.serviceClient<std_srvs::Empty>(nh_.resolveName("ardrone/togglecam"),1);
    flattrim_srv         = nh_.serviceClient<std_srvs::Empty>(nh_.resolveName("ardrone/flattrim"),1);

        ros::Time last = ros::Time::now();
        ros::Time lastHz = ros::Time::now();
        while(keepRunning && nh_.ok())
        {
                // spin for 100ms
                while((ros::Time::now() - last) < ros::Duration(0.1))
                        ros::getGlobalCallbackQueue()->callAvailable(ros::WallDuration(0.1 - (ros::Time::now() - last).toSec()));
                last = ros::Time::now();

                // if nothing on /cmd_vel, repeat!
                if(velCount100ms == 0)
                        switch(gui->currentControlSource)
                        {
                        case CONTROL_AUTO:
                                sendControlToDrone(ControlCommand(0,0,0,0));
                                break;
                        case CONTROL_JOY:
                                sendControlToDrone(lastJoyControlSent);
                                break;
                        case CONTROL_KB:
                                sendControlToDrone(gui->calcKBControl());
                                break;
                        case CONTROL_NONE:
                                sendControlToDrone(ControlCommand(0,0,0,0));
                                break;
                        }
                velCount100ms = 0;

                // if 1s passed: update Hz values
                if((ros::Time::now() - lastHz) > ros::Duration(1.0))
                {
                        gui->setCounts(navdataCount, velCount, dronePoseCount, joyCount);
                        navdataCount = velCount = dronePoseCount = joyCount = 0;
                        lastHz = ros::Time::now();
                }
        }

        gui->closeWindow();
        if(nh_.ok()) ros::shutdown();
        std::cout << "Exiting ROS Thread (ros::shutdown() has been called)" << std::endl;
}


void RosThread::publishCommand(std::string c)
{
        std_msgs::String s;
        s.data = c.c_str();
        pthread_mutex_lock(&send_CS);
        tum_ardrone_pub.publish(s);
        pthread_mutex_unlock(&send_CS);
}

void RosThread::sendControlToDrone(ControlCommand cmd)
{
        // TODO: check converstion (!)
        geometry_msgs::Twist cmdT;
        cmdT.angular.z = -cmd.yaw;
        cmdT.linear.z = cmd.gaz;
        cmdT.linear.x = -cmd.pitch;
        cmdT.linear.y = -cmd.roll;

        cmdT.angular.x = cmdT.angular.y = gui->useHovering ? 0 : 1;


        pthread_mutex_lock(&send_CS);
        vel_pub.publish(cmdT);
        pthread_mutex_unlock(&send_CS);
}

void RosThread::sendLand()
{
        pthread_mutex_lock(&send_CS);
        land_pub.publish(std_msgs::Empty());
        pthread_mutex_unlock(&send_CS);
}
void RosThread::sendTakeoff()
{
        pthread_mutex_lock(&send_CS);
        takeoff_pub.publish(std_msgs::Empty());
        pthread_mutex_unlock(&send_CS);
}
void RosThread::sendToggleState()
{
        pthread_mutex_lock(&send_CS);
        toggleState_pub.publish(std_msgs::Empty());
        pthread_mutex_unlock(&send_CS);
}
void RosThread::sendToggleCam()
{
        pthread_mutex_lock(&send_CS);
        toggleCam_srv.call(toggleCam_srv_srvs);
        pthread_mutex_unlock(&send_CS);
}
void RosThread::sendFlattrim()
{
        pthread_mutex_lock(&send_CS);
        flattrim_srv.call(flattrim_srv_srvs);
        pthread_mutex_unlock(&send_CS);
}