home_position.cpp

Go to the documentation of this file.

/*
 * Copyright 2017 Thomas Stastny, Nuno Marques.
 *
 * This file is part of the mavros package and subject to the license terms
 * in the top-level LICENSE file of the mavros repository.
 * https://github.com/mavlink/mavros/tree/master/LICENSE.md
 */
#include <mavros/mavros_plugin.h>
#include <eigen_conversions/eigen_msg.h>

#include <std_srvs/Trigger.h>
#include <mavros_msgs/CommandLong.h>
#include <mavros_msgs/HomePosition.h>


namespace mavros {
namespace std_plugins {
class HomePositionPlugin : public plugin::PluginBase {
public:
        HomePositionPlugin() :
                hp_nh("~home_position"),
                REQUEST_POLL_TIME_DT(REQUEST_POLL_TIME_MS / 1000.0)
        { }

        void initialize(UAS &uas_) override
        {
                PluginBase::initialize(uas_);

                hp_pub = hp_nh.advertise<mavros_msgs::HomePosition>("home", 2, true);
                hp_sub = hp_nh.subscribe("set", 10, &HomePositionPlugin::home_position_cb, this);
                update_srv = hp_nh.advertiseService("req_update", &HomePositionPlugin::req_update_cb, this);

                poll_timer = hp_nh.createTimer(REQUEST_POLL_TIME_DT, &HomePositionPlugin::timeout_cb, this);
                poll_timer.stop();
                enable_connection_cb();
        }

        Subscriptions get_subscriptions() override
        {
                return {
                               make_handler(&HomePositionPlugin::handle_home_position),
                };
        }

private:
        ros::NodeHandle hp_nh;

        ros::Publisher hp_pub;
        ros::Subscriber hp_sub;
        ros::ServiceServer update_srv;

        ros::Timer poll_timer;

        static constexpr int REQUEST_POLL_TIME_MS = 10000;      
        const ros::Duration REQUEST_POLL_TIME_DT;

        bool call_get_home_position(void)
        {
                using mavlink::common::MAV_CMD;

                bool ret = false;

                try {
                        ros::NodeHandle pnh("~");
                        auto client = pnh.serviceClient<mavros_msgs::CommandLong>("cmd/command");

                        mavros_msgs::CommandLong cmd{};

                        cmd.request.command = utils::enum_value(MAV_CMD::GET_HOME_POSITION);

                        ret = client.call(cmd);
                        ret = cmd.response.success;
                }
                catch (ros::InvalidNameException &ex) {
                        ROS_ERROR_NAMED("home_position", "HP: %s", ex.what());
                }

                return ret;
        }

        void handle_home_position(const mavlink::mavlink_message_t *msg, mavlink::common::msg::HOME_POSITION &home_position)
        {
                poll_timer.stop();

                auto hp = boost::make_shared<mavros_msgs::HomePosition>();

                auto pos = ftf::transform_frame_ned_enu(Eigen::Vector3d(home_position.x, home_position.y, home_position.z));
                auto q = ftf::transform_orientation_ned_enu(ftf::mavlink_to_quaternion(home_position.q));
                auto hp_approach_enu = ftf::transform_frame_ned_enu(Eigen::Vector3d(home_position.approach_x, home_position.approach_y, home_position.approach_z));

                hp->header.stamp = ros::Time::now();
                hp->geo.latitude = home_position.latitude / 1E7;                // deg
                hp->geo.longitude = home_position.longitude / 1E7;              // deg
                hp->geo.altitude = home_position.altitude / 1E3 + m_uas->geoid_to_ellipsoid_height(&hp->geo);   // in meters
                tf::quaternionEigenToMsg(q, hp->orientation);
                tf::pointEigenToMsg(pos, hp->position);
                tf::vectorEigenToMsg(hp_approach_enu, hp->approach);

                ROS_DEBUG_NAMED("home_position", "HP: Home lat %f, long %f, alt %f", hp->geo.latitude, hp->geo.longitude, hp->geo.altitude);
                hp_pub.publish(hp);
        }

        void home_position_cb(const mavros_msgs::HomePosition::ConstPtr &req)
        {
                mavlink::common::msg::SET_HOME_POSITION hp {};

                Eigen::Vector3d pos, approach;
                Eigen::Quaterniond q;

                tf::pointMsgToEigen(req->position, pos);
                pos = ftf::transform_frame_enu_ned(pos);

                tf::quaternionMsgToEigen(req->orientation, q);
                q = ftf::transform_orientation_enu_ned(q);

                tf::vectorMsgToEigen(req->approach, approach);
                approach = ftf::transform_frame_enu_ned(approach);

                hp.target_system = m_uas->get_tgt_system();
                ftf::quaternion_to_mavlink(q, hp.q);

                hp.altitude = req->geo.altitude * 1e3 + m_uas->ellipsoid_to_geoid_height(&req->geo);
                // [[[cog:
                // for f, m in (('latitude', '1e7'), ('longitude', '1e7')):
                //     cog.outl("hp.{f} = req->geo.{f} * {m};".format(**locals()))
                // for a, b in (('', 'pos'), ('approach_', 'approach')):
                //     for f in "xyz":
                //         cog.outl("hp.{a}{f} = {b}.{f}();".format(**locals()))
                // ]]]
                hp.latitude = req->geo.latitude * 1e7;
                hp.longitude = req->geo.longitude * 1e7;
                hp.x = pos.x();
                hp.y = pos.y();
                hp.z = pos.z();
                hp.approach_x = approach.x();
                hp.approach_y = approach.y();
                hp.approach_z = approach.z();
                // [[[end]]] (checksum: 9c40c5b3ac06b3b82016b4f07a8e12b2)

                UAS_FCU(m_uas)->send_message_ignore_drop(hp);
        }

        bool req_update_cb(std_srvs::Trigger::Request &req, std_srvs::Trigger::Response &res)
        {
                res.success = call_get_home_position();
                return true;
        }

        void timeout_cb(const ros::TimerEvent &event)
        {
                ROS_INFO_NAMED("home_position", "HP: requesting home position");
                call_get_home_position();
        }

        void connection_cb(bool connected) override
        {
                if (connected)
                        poll_timer.start();
                else
                        poll_timer.stop();
        }
};
}       // namespace std_plugins
}       // namespace mavros

#include <pluginlib/class_list_macros.h>
PLUGINLIB_EXPORT_CLASS(mavros::std_plugins::HomePositionPlugin, mavros::plugin::PluginBase)