humotion

/*

* This file is part of humotion

*

* Copyright(c) sschulz <AT> techfak.uni-bielefeld.de

* http://opensource.cit-ec.de/projects/humotion

*

* This file may be licensed under the terms of the

* GNU Lesser General Public License Version 3 (the ``LGPL''),

* or (at your option) any later version.

*

* Software distributed under the License is distributed

* on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either

* express or implied. See the LGPL for the specific language

* governing rights and limitations.

*

* You should have received a copy of the LGPL along with this

* program. If not, go to http://www.gnu.org/licenses/lgpl.html

* or write to the Free Software Foundation, Inc.,

* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

*

* The development of this software was supported by the

* Excellence Cluster EXC 277 Cognitive Interaction Technology.

* The Excellence Cluster EXC 277 is a grant of the Deutsche

* Forschungsgemeinschaft (DFG) in the context of the German

* Excellence Initiative.

*/

#include <boost/algorithm/string/classification.hpp>

#include <string>

#include "humotion/server/middleware_ros.h"

using humotion::server::MiddlewareROS;

//! constructor

MiddlewareROS::MiddlewareROS(std::string scope, Controller *c)

    : Middleware(scope, c) {

    ROS_DEBUG_STREAM_NAMED("MiddlewareROS", "will use ros middleware");

    // start ros core?

    if (ros::isInitialized()) {

        // oh another process is doing ros comm, fine, we do not need to call it

        tick_necessary_ = false;

    } else {

        // we have to take care of ros

        ROS_DEBUG_STREAM_NAMED("MiddlewareROS", "no active ros middleware, will call ros::init() "

               "now and we will call tick() periodically!\n");

        std::string node_name = scope;

        node_name.erase(std::remove(node_name.begin(), node_name.end(), '/'), node_name.end());

        ROS_DEBUG_STREAM_NAMED("MiddlewareROS", "registering on ROS as node " << node_name.c_str());

        ros::M_string no_remapping;

        ros::init(no_remapping, node_name);

        tick_necessary_ = true;

    }

    // create node handle

    ros::NodeHandle n;

    ros::NodeHandle pnh("~");

    // set up subscribers

    mouth_target_subscriber_ = n.subscribe(scope + "/humotion/mouth/target", 150,

                                          &MiddlewareROS::incoming_mouth_target ,

                                          this, ros::TransportHints().unreliable());

    gaze_target_subscriber_  = n.subscribe(scope + "/humotion/gaze/target",  150,

                                          &MiddlewareROS::incoming_gaze_target,

                                          this, ros::TransportHints().unreliable());

    // set up dynamic reconfiguration service

    attach_to_reconfiguration_server(pnh);

}

//! callback for incoming dynamic reconfigure requests:

void MiddlewareROS::dynamic_reconfigure_callback(const humotion::humotionConfig &dyn_config,

                                                  uint32_t level) {

    // fetch config

    humotion::server::Config *config = controller_->get_config();

    // saccade detection thresholds

    config->threshold_velocity_eye_saccade = dyn_config.threshold_velocity_eye_saccade;

    config->threshold_angle_neck_saccade = dyn_config.threshold_angle_neck_saccade;

    config->threshold_angle_omr_limit = dyn_config.threshold_angle_omr_limit;

    // neck motion generation configuration

    config->scale_velocity_neck = dyn_config.scale_velocity_neck;

    config->scale_acceleration_neck = dyn_config.scale_acceleration_neck;

    config->limit_velocity_neck = dyn_config.limit_velocity_neck;

    config->limit_acceleration_neck = dyn_config.limit_acceleration_neck;

    // eye motion generation configuration

    config->scale_velocity_eye = dyn_config.scale_velocity_eye;

    config->scale_acceleration_eye = dyn_config.scale_acceleration_eye;

    config->limit_velocity_eye = dyn_config.limit_velocity_eye;

    config->limit_acceleration_eye = dyn_config.limit_acceleration_eye;

    // parameters fo the breathing pattern

    config->breath_period = dyn_config.breath_period;

    config->breath_amplitude = dyn_config.breath_amplitude;

    // parameters for eyelids

    config->eyelids_follow_eyemotion = dyn_config.eyelids_follow_eyemotion;

    // parameters for eye blinking

    config->eyeblink_duration = dyn_config.eyeblink_duration;

    config->eyeblink_periodic_distribution_lower = dyn_config.eyeblink_periodic_distribution_lower;

    config->eyeblink_periodic_distribution_upper = dyn_config.eyeblink_periodic_distribution_upper;

    config->eyeblink_probability_after_saccade = dyn_config.eyeblink_probability_after_saccade;

    config->eyeblink_blocked_time = dyn_config.eyeblink_blocked_time;

}

//! attach to dynamic reconfigure server

void MiddlewareROS::attach_to_reconfiguration_server(ros::NodeHandle priv_nodehandle) {

    ROS_DEBUG_STREAM_NAMED("MiddlewareROS", "connecting to dynamic reconfiguration server");

    ros::NodeHandle reconf_node(priv_nodehandle, "humotion/configuration");

    reconf_server_ = new dynamic_reconfigure::Server<humotion::humotionConfig>(reconf_node);

    reconf_server_->setCallback(boost::bind(

                                    &MiddlewareROS::dynamic_reconfigure_callback, this, _1, _2));

}

//! destructor

MiddlewareROS::~MiddlewareROS() {

}

//! connection ok?

//! \return true if conn is alive

bool MiddlewareROS::ok() {

    return ros::ok();

}

//! do a single tick

void MiddlewareROS::tick() {

    if (tick_necessary_) {

        ros::spinOnce();

    }

}

//! callback to handle incoming mouth  target

void MiddlewareROS::incoming_mouth_target(const humotion::mouth::ConstPtr& msg) {

    // printf("> incoming mouth_target\n");

    MouthState mouth_state;

    mouth_state.position_left   = msg->position.left;

    mouth_state.position_center = msg->position.center;

    mouth_state.position_right  = msg->position.right;

    mouth_state.opening_left   = msg->opening.left;

    mouth_state.opening_center = msg->opening.center;

    mouth_state.opening_right  = msg->opening.right;

    controller_->set_mouth_target(mouth_state);

}

//! callback to handle incoming gaze  target

void MiddlewareROS::incoming_gaze_target(const humotion::gaze::ConstPtr& msg) {

    GazeState gaze_state;

    // printf("> incoming gaze_target [P:%3.1f, T:%3.1f, R:%3.1f] %d = %s\n",

    //          msg->pan, msg->tilt, msg->roll, msg->type,

    //          msg->type==humotion::gaze::GAZETYPE_ABSOLUTE?"ABSOLUTE":"RELATIVE");

    gaze_state.pan  = msg->pan;

    gaze_state.tilt = msg->tilt;

    gaze_state.roll = msg->roll;

    gaze_state.vergence = msg->vergence;

    gaze_state.pan_offset  = msg->pan_offset;

    gaze_state.tilt_offset = msg->tilt_offset;

    gaze_state.roll_offset = msg->roll_offset;

    gaze_state.eyelid_opening_upper = msg->eyelid_opening_upper;

    gaze_state.eyelid_opening_lower = msg->eyelid_opening_lower;

    gaze_state.eyebrow_left = msg->eyebrow_left;

    gaze_state.eyebrow_right = msg->eyebrow_right;

    gaze_state.eyeblink_request_left = msg->eyeblink_request_left;

    gaze_state.eyeblink_request_right = msg->eyeblink_request_right;

    if (msg->gaze_type == humotion::gaze::GAZETYPE_ABSOLUTE) {

        gaze_state.gaze_type = GazeState::GAZETYPE_ABSOLUTE;

    } else {

        gaze_state.gaze_type = GazeState::GAZETYPE_RELATIVE;

    }

    gaze_state.timestamp.set(msg->gaze_timestamp.sec, msg->gaze_timestamp.nsec);

    controller_->set_gaze_target(gaze_state);

}