block.h

#ifndef SESSTYPE__NODE__BLOCK_H__
#define SESSTYPE__NODE__BLOCK_H__

#ifdef __cplusplus
#include <vector>
#endif

#include "sesstype/msg.h"
#include "sesstype/role.h"
#include "sesstype/node.h"

#ifdef __cplusplus
namespace sesstype {
#endif

#ifdef __cplusplus

template <class BaseNode, class RoleType, class MessageType, class VisitorType>
class BlockNodeTmpl : public BaseNode {
    std::vector<BaseNode *> children_;

  public:
    typedef typename std::vector<BaseNode *> NodeContainer;

    BlockNodeTmpl() : BaseNode(ST_NODE_ROOT), children_() { }

    BlockNodeTmpl(const BlockNodeTmpl &node) : BaseNode(ST_NODE_ROOT), children_()
    {
        for (BaseNode *child : node.children_) {
            children_.push_back(static_cast<BaseNode *>(child->clone()));
        }
    }

    ~BlockNodeTmpl() override
    {
        for (auto node: children_) {
            delete node;
        }
    }

    BlockNodeTmpl *clone() const override
    {
        return new BlockNodeTmpl<BaseNode, RoleType, MessageType, VisitorType>(*this);
    }

    BaseNode *child(unsigned int idx) const
    {
        return children_.at(idx);
    }

    unsigned int num_children() const
    {
        return children_.size();
    }

    void append_child(BaseNode *child)
    {
        children_.push_back(child);
    }

    void set_child(unsigned int idx, BaseNode *child)
    {
        children_.at(idx) = child;
    }

    typename NodeContainer::const_iterator child_begin() const
    {
        return children_.begin();
    }

    typename NodeContainer::const_iterator child_end() const
    {
        return children_.end();
    }

    virtual void accept(VisitorType &v) override;

  protected:
    BlockNodeTmpl(int type) : BaseNode(type), children_() { }
};

using BlockNode = BlockNodeTmpl<Node, Role, MsgSig, util::NodeVisitor>;
#endif // __cplusplus

#ifdef __cplusplus
extern "C" {
#endif

st_node *st_node_append_child(st_node *const parent, st_node *child);

unsigned int st_node_num_children(st_node *const parent);

st_node *st_node_get_child(st_node *const parent, unsigned int index);

#ifdef __cplusplus
} // extern "C"
#endif

#ifdef __cplusplus
} // namespace sesstype
#endif

#endif//SESSTYPE__NODE__BLOCK_H__