server.h

Go to the documentation of this file.

/*********************************************************************
* Software License Agreement (BSD License)
*
*  Copyright (c) 2009-2010, Willow Garage, Inc.
*  Copyright (c) 2014, Intermodalics BVBA
*  All rights reserved.
*
*  Redistribution and use in source and binary forms, with or without
*  modification, are permitted provided that the following conditions
*  are met:
*
*   * Redistributions of source code must retain the above copyright
*     notice, this list of conditions and the following disclaimer.
*   * Redistributions in binary form must reproduce the above
*     copyright notice, this list of conditions and the following
*     disclaimer in the documentation and/or other materials provided
*     with the distribution.
*   * Neither the name of the Willow Garage nor the names of its
*     contributors may be used to endorse or promote products derived
*     from this software without specific prior written permission.
*
*  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
*  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
*  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
*  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
*  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
*  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
*  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
*  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
*  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
*  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
*  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
*  POSSIBILITY OF SUCH DAMAGE.
*********************************************************************/

#ifndef RTT_DYNAMIC_RECONFIGURE_SERVER_H
#define RTT_DYNAMIC_RECONFIGURE_SERVER_H

#include <rtt/Service.hpp>
#include <rtt/TaskContext.hpp>
#include <rtt/os/Mutex.hpp>
#include <rtt/Logger.hpp>
#include <rtt/Operation.hpp>
#include <rtt/OperationCaller.hpp>

#include <rtt/internal/DataSources.hpp>
#include <rtt/internal/GlobalEngine.hpp>

#include <ros/ros.h>

#include <dynamic_reconfigure/Reconfigure.h>
#include <dynamic_reconfigure/Config.h>
#include <dynamic_reconfigure/ConfigDescription.h>

namespace rtt_dynamic_reconfigure {

template <class ConfigType> class Server;

template <class ConfigType>
struct Updater {
    virtual bool propertiesFromConfig(ConfigType &config, uint32_t level, RTT::PropertyBag &) { return false; }
    virtual bool configFromProperties(ConfigType &config, const RTT::PropertyBag &) { return false; }
};

template <class ConfigType>
struct dynamic_reconfigure_traits {
    typedef Server<ConfigType> ServerType;

    static void getMin(ConfigType &min, const ServerType *) { min = ConfigType::__getMin__(); }

    static void getMax(ConfigType &max, const ServerType *) { max = ConfigType::__getMax__(); }

    static void getDefault(ConfigType &dflt, const ServerType *) { dflt = ConfigType::__getDefault__(); }

    static dynamic_reconfigure::ConfigDescriptionPtr getDescriptionMessage(const ServerType *) { return dynamic_reconfigure::ConfigDescriptionPtr(new dynamic_reconfigure::ConfigDescription(ConfigType::__getDescriptionMessage__())); }

    static const bool canRefresh = false;

    static void refreshDescription(const ServerType *) {}

    static void toMessage(ConfigType &config, dynamic_reconfigure::Config &message, const ServerType *) { config.__toMessage__(message); }

    static void fromMessage(ConfigType &config, dynamic_reconfigure::Config &message, const ServerType *) { config.__fromMessage__(message); }

    static void clamp(ConfigType &config, const ServerType *) { config.__clamp__(); }

    static RTT::internal::AssignableDataSource<RTT::PropertyBag>::shared_ptr getDataSource(ConfigType &config, const ServerType *server) {
        RTT::internal::AssignableDataSource<RTT::PropertyBag>::shared_ptr ds(new RTT::internal::ValueDataSource<RTT::PropertyBag>());
        if (!server->updater()->propertiesFromConfig(config, ~0, ds->set()))
            ds.reset();
        return ds;
    }
};

namespace {
  struct null_deleter {
      void operator()(const void *) const {}
  };
}

template <class ConfigType>
class Server : public RTT::Service
{
private:
    typedef Updater<ConfigType> UpdaterType;
    typedef dynamic_reconfigure_traits<ConfigType> traits;
    typedef boost::shared_ptr< Server<ConfigType> > shared_ptr;

    RTT::os::MutexRecursive mutex_;
    ros::NodeHandle *node_handle_;
    ros::ServiceServer set_service_;
    ros::Publisher update_pub_;
    ros::Publisher descr_pub_;

    ConfigType config_;
    ConfigType min_;
    ConfigType max_;
    ConfigType default_;

    mutable boost::shared_ptr<UpdaterType> updater_;
    bool initialized_;

    RTT::OperationCaller<bool(const RTT::PropertyBag &source, uint32_t level)> update_callback_;
    RTT::OperationCaller<void(uint32_t level)> notify_callback_;
    RTT::Operation<bool(const RTT::PropertyBag &source, uint32_t level)> update_callback_default_impl_;

public:
    Server(const std::string &name, RTT::TaskContext *owner)
        : RTT::Service(name, owner)
        , node_handle_(0)
        , update_callback_default_impl_("updateProperties", &Server<ConfigType>::updatePropertiesDefaultImpl, this, RTT::OwnThread)
    {
        construct();
    }

    Server(RTT::TaskContext *owner)
        : RTT::Service("reconfigure", owner)
        , node_handle_(0)
        , update_callback_default_impl_("updateProperties", &Server<ConfigType>::updatePropertiesDefaultImpl, this, RTT::OwnThread)
    {
        construct();
    }

    virtual ~Server() {
        shutdown();
    }

    void updateConfig(const ConfigType &config)
    {
        updateConfigInternal(config);
    }

    void getConfigMax(ConfigType &max) const
    {
        max = max_;
    }

    const ConfigType &getConfigMax() const { return max_; }

    ConfigType &getConfigMax() { return max_; }

    void getConfigMin(ConfigType &min) const
    {
        min = min_;
    }

    const ConfigType &getConfigMin() const { return min_; }

    ConfigType &getConfigMin() { return min_; }

    void getConfigDefault(ConfigType &config) const
    {
        config = default_;
    }

    const ConfigType &getConfigDefault() const { return default_; }

    ConfigType &getConfigDefault() { return default_; }

    void setConfigMax(const ConfigType &config)
    {
        max_ = config;
        PublishDescription();
    }

    void setConfigMin(const ConfigType &config)
    {
        min_ = config;
        PublishDescription();
    }

    void setConfigDefault(const ConfigType &config)
    {
        default_ = config;
        PublishDescription();
    }

    dynamic_reconfigure::ConfigDescriptionPtr getDescriptionMessage() {
        dynamic_reconfigure::ConfigDescriptionPtr description_message = traits::getDescriptionMessage(this);

        // Copy over min_ max_ default_
        traits::toMessage(max_, description_message->max, this);
        traits::toMessage(min_, description_message->min, this);
        traits::toMessage(default_, description_message->dflt, this);

        return description_message;
    }

    void advertise(std::string ns = std::string())
    {
        // shutdown publishers/service servers from previous runs
        shutdown();

        // set default namespace
        if (ns.empty()) {
            if (getOwner()->getName() == "Deployer")
                ns = "~";
            else
                ns = "~" + getOwner()->getName();
        }

        // create NodeHandle
        node_handle_ = new ros::NodeHandle(ns);

        // advertise service server and publishers
        set_service_ = node_handle_->advertiseService("set_parameters", &Server<ConfigType>::setConfigCallback, this);
        descr_pub_ = node_handle_->advertise<dynamic_reconfigure::ConfigDescription>("parameter_descriptions", 1, true);
        update_pub_ = node_handle_->advertise<dynamic_reconfigure::Config>("parameter_updates", 1, true);

        // publish update once
        PublishDescription();
        updateConfigInternal(config_);
    }

    void shutdown()
    {
        if (node_handle_) {
            node_handle_->shutdown();
            delete node_handle_;
            node_handle_ = 0;
        }
    }

    bool updated()
    {
        ConfigType new_config = config_;
        if (!updater()->configFromProperties(new_config, *(getOwner()->properties()))) return false;
        updateConfig(new_config);
        return true;
    }

    void refresh()
    {
        RTT::os::MutexLock lock(mutex_);

        // Refresh config description (no-op unless for AutoConfig)
        traits::refreshDescription(this);

        // Grab copys of the data from the config files.  These are declared in the generated config file.
        traits::getMin(min_, this);
        traits::getMax(max_, this);
        traits::getDefault(default_, this);

        // Publish the description
        PublishDescription();

        // Add/replace min/max/default properties
        this->properties()->remove(this->properties()->getProperty("min"));
        this->properties()->remove(this->properties()->getProperty("max"));
        this->properties()->remove(this->properties()->getProperty("default"));
        this->properties()->ownProperty(new RTT::Property<RTT::PropertyBag>("min", "Minimum values as published to dynamic_reconfigure clients", traits::getDataSource(min_, this)));
        this->properties()->ownProperty(new RTT::Property<RTT::PropertyBag>("max", "Maximum values as published to dynamic_reconfigure clients", traits::getDataSource(max_, this)));
        this->properties()->ownProperty(new RTT::Property<RTT::PropertyBag>("default", "Default values as published to dynamic_reconfigure clients", traits::getDataSource(default_, this)));

        // Get initial values from current property settings
        config_ = ConfigType();
        traits::getDefault(config_, this);
        updater()->configFromProperties(config_, *(getOwner()->properties()));
        if (node_handle_)
            config_.__fromServer__(*node_handle_);
        traits::clamp(config_, this);

        // At startup we need to load the configuration with all level bits set (everything has changed).
        RTT::PropertyBag init_config;
        updater()->propertiesFromConfig(config_, ~0, init_config);
        update_callback_(init_config, ~0);
        if (notify_callback_.ready()) notify_callback_(~0);

        updateConfigInternal(config_);
    }

    UpdaterType *updater() const
    {
        if (!updater_) updater_.reset(new UpdaterType());
        return updater_.get();
    }

    void setUpdater(UpdaterType *updater)
    {
        updater_.reset(updater, null_deleter());
    }

    void setUpdateCallback(RTT::OperationInterfacePart *impl)
    {
        update_callback_ = impl;
    }

    void setNotificationCallback(RTT::OperationInterfacePart *impl)
    {
        notify_callback_ = impl;
    }

private:
    void construct()
    {
        this->addOperation("advertise", &Server<ConfigType>::advertise, this)
            .doc("Advertise this dynamic_reconfigure server at the master.")
            .arg("namespace", "The namespace this server should be advertised in. Defaults to ~component.");
        this->addOperation("shutdown", &Server<ConfigType>::shutdown, this)
            .doc("Shutdown this dynamic_reconfigure server.");
        this->addOperation("updated", &Server<ConfigType>::updated, this)
            .doc("Notify the dynamic_reconfigure server that properties have been updated. This will update the GUI.");

        this->addOperation("refresh", &Server<ConfigType>::refresh, this)
            .doc("Rediscover the owner's properties or update advertised min/max/default values. Call this operation after having added properties.");

        // check if owner implements the Updater interface
        UpdaterType *updater = dynamic_cast<UpdaterType *>(getOwner());
        if (updater) setUpdater(updater);

        // check if owner provides the updateProperties operation
        if (getOwner() && getOwner()->provides()->hasOperation("updateProperties")) {
            update_callback_ = getOwner()->provides()->getLocalOperation("updateProperties");
        } else {
            update_callback_ = update_callback_default_impl_.getOperationCaller();
        }

        // check if owner provides the notifyPropertiesUpdate operation
        if (getOwner() && getOwner()->provides()->hasOperation("notifyPropertiesUpdate")) {
            notify_callback_ = getOwner()->provides()->getLocalOperation("notifyPropertiesUpdate");
        }

        // update_callback_ and notify_callback_ are called from the ROS spinner thread -> set GlobalEngine as caller engine
        update_callback_.setCaller(RTT::internal::GlobalEngine::Instance());
        notify_callback_.setCaller(RTT::internal::GlobalEngine::Instance());

        // refresh once
        refresh();
    }

    void PublishDescription() {
        if (!descr_pub_) return;
        descr_pub_.publish(getDescriptionMessage());
    }

    bool setConfigCallback(dynamic_reconfigure::Reconfigure::Request &req,
                           dynamic_reconfigure::Reconfigure::Response &rsp)
    {
        RTT::os::MutexLock lock(mutex_);

        ConfigType new_config = config_;
        traits::fromMessage(new_config, req.config, this);
        traits::clamp(new_config, this);
        uint32_t level = config_.__level__(new_config);

        RTT::PropertyBag bag;
        if (!updater()->propertiesFromConfig(new_config, level, bag)) return false;
        if (!update_callback_.ready() || !update_callback_(bag, level)) return false;
        if (notify_callback_.ready()) notify_callback_(level);

        updateConfigInternal(new_config);
        new_config.__toMessage__(rsp.config);
        return true;
    }

    void updateConfigInternal(const ConfigType &config)
    {
        RTT::os::MutexLock lock(mutex_);
        config_ = config;
        if (node_handle_)
            config_.__toServer__(*node_handle_);
        dynamic_reconfigure::Config msg;
        config_.__toMessage__(msg);

        if (update_pub_)
            update_pub_.publish(msg);
    }

    bool updatePropertiesDefaultImpl(const RTT::PropertyBag &source, uint32_t)
    {
        return RTT::updateProperties(*(getOwner()->properties()), source);
    }
};

template <typename T, typename ValueType>
bool setProperty(const std::string &name, RTT::PropertyBag &bag, ValueType &value)
{
    if (bag.getProperty(name)) {
        RTT::Property<T> *prop = bag.getPropertyType<T>(name);
        if (!prop) {
            RTT::log(RTT::Error) << "Could not assign property '" << name << "': Property exists with a different type." << RTT::endlog();
            return false;
        }

        prop->set() = value;

    } else {
        if (boost::is_same<T,ValueType>::value) {
            bag.addProperty(name, value);
        } else {
            bag.ownProperty(new RTT::Property<T>(name, std::string(), value));
        }
    }

    return true;
}

template <typename T, typename ValueType>
bool getProperty(const std::string &name, const RTT::PropertyBag &bag, ValueType &value)
{
    RTT::Property<T> *prop = bag.getPropertyType<T>(name);
    if (!prop) {
        RTT::log(RTT::Error) << "Could not get property '" << name << "': No such property in the bag." << RTT::endlog();
        return false;
    }

    value = prop->rvalue();
    return true;
}

} // namespace rtt_dynamic_reconfigure

#define RTT_DYNAMIC_RECONFIGURE_SERVICE_PLUGIN(CONFIG, NAME) \
    extern "C" {\
        RTT_EXPORT bool loadRTTPlugin(RTT::TaskContext* tc);  \
        bool loadRTTPlugin(RTT::TaskContext* tc) {    \
            if (tc == 0) return true; \
            RTT::Service::shared_ptr sp( new rtt_dynamic_reconfigure::Server<CONFIG>(NAME, tc ) ); \
            return tc->provides()->addService( sp ); \
        } \
        RTT_EXPORT RTT::Service::shared_ptr createService();  \
        RTT::Service::shared_ptr createService() {    \
            RTT::Service::shared_ptr sp( new rtt_dynamic_reconfigure::Server<CONFIG>( 0 ) ); \
            return sp; \
        } \
        RTT_EXPORT std::string getRTTPluginName(); \
        std::string getRTTPluginName() { \
            return NAME; \
        } \
        RTT_EXPORT std::string getRTTTargetName(); \
        std::string getRTTTargetName() { \
            return OROCOS_TARGET_NAME; \
        } \
    }

#endif // RTT_DYNAMIC_RECONFIGURE_SERVER_H