decorator_node.h

Go to the documentation of this file.

#ifndef DECORATORNODE_H
#define DECORATORNODE_H

#include "behaviortree_cpp/tree_node.h"

namespace BT
{
class DecoratorNode : public TreeNode
{
  protected:
    TreeNode* child_node_;

  public:

    DecoratorNode(const std::string& name, const NodeConfiguration& config);

    virtual ~DecoratorNode() override = default;

    void setChild(TreeNode* child);

    const TreeNode* child() const;

    TreeNode* child();

    virtual void halt() override;

    void haltChild();

    virtual NodeType type() const override
    {
        return NodeType::DECORATOR;
    }

    NodeStatus executeTick() override;
};

class SimpleDecoratorNode : public DecoratorNode
{
  public:
    typedef std::function<NodeStatus(NodeStatus, TreeNode&)> TickFunctor;

    // You must provide the function to call when tick() is invoked
    SimpleDecoratorNode(const std::string& name, TickFunctor tick_functor, const NodeConfiguration& config);

    ~SimpleDecoratorNode() override = default;

  protected:
    virtual NodeStatus tick() override;

    TickFunctor tick_functor_;
};
}

#endif