BoostGraphBuilder.h

Go to the documentation of this file.

#ifndef RDFAnalysis_Utils_BoostGraphBuilder_H
#define RDFAnalysis_Utils_BoostGraphBuilder_H

#include <boost/graph/adjacency_list.hpp>

#include <vector>
#include <functional>
#include <type_traits>

namespace RDFAnalysis { namespace detail {
  template <typename InputNode,
            typename VertexInfo,
            typename ChildItr=typename std::vector<std::decay_t<InputNode>>::const_iterator>
    class BoostGraphBuilder {
      public:
        struct Vertex {
          Vertex() {}
          Vertex(const VertexInfo& info) : info(info) {}
          VertexInfo info;
        }; //> end struct Vertex
        // Type defs
        using input_node_t = InputNode;
        using vertex_info_t = VertexInfo;
        using child_itr_t = ChildItr;
        using graph_t = boost::adjacency_list<
          boost::vecS, boost::vecS, boost::directedS, Vertex>;
        using vert_desc_t = typename boost::graph_traits<graph_t>::vertex_descriptor;
        using prop_map_t = typename boost::property_map<graph_t, vertex_info_t Vertex::*>::type;

        BoostGraphBuilder(
            std::function<ChildItr(InputNode)> childBegin,
            std::function<ChildItr(InputNode)> childEnd);

        virtual ~BoostGraphBuilder() = 0;

        graph_t buildGraph(InputNode root);

      protected:
        virtual VertexInfo info(InputNode input) = 0;

        enum class NodeDecision {
          Write,
          Skip,
          Terminate
        };

        virtual NodeDecision processNode(InputNode input);

        void addToGraph(
            InputNode input,
            const vert_desc_t& parent,
            graph_t& graph);

        std::function<ChildItr(InputNode)> m_childBegin;
        std::function<ChildItr(InputNode)> m_childEnd;

    }; //> end class BoostGraphBuilder<InputNode, ChildBegin, ChildEnd, VertexInfo>
} } //> end namespace RDFAnalysis::detail
#include "RDFAnalysis/Utils/BoostGraphBuilder.icc"

#endif //> !RDFAnalysis_Utils_BoostGraphBuilder_H