swu_graph_builder.h

#ifndef GRAPHTOOL_SWUGRAPHBUILDER_H_
#define GRAPHTOOL_SWUGRAPHBUILDER_H_

#include <string>
#include <map>
#include <iostream>
#include <algorithm> // for_each
#include <utility> // pair

#include "grapher.h"
#include "linkler.h"
#include "dependency.h"
#include "swu.h"
#include "graph_builder.h"

class SWUGraphBuilder : public GraphBuilder {
   public:
    typedef Grapher<SWU*,Dependency<SWU,SWU>*,CompareSWUByName, CompareSWUDependency> GraphType;

    SWUGraphBuilder(Linkler* l);
    ~SWUGraphBuilder(){
      
    }

    GraphType& grapher();

    //*******************
    // Graphing functions
    //*******************

    void graph_all();

    void graph_all_dependencies();

    void graph_all_dependents();

    void graph_single(std::string swu_name);

    void graph_recursively(std::string swu);

    void graph_dependencies(std::string swu);

    void graph_dependents(std::string swu);

    Linkler* linkler();

    SWU* get(std::string swu_name); 

    class GraphAdder {
      public:
        GraphAdder(GraphType& g) : grapher_(g) { }

      protected:
        GraphType& grapher_;
    };

    class AddSWU : public GraphAdder {
      public:
        explicit AddSWU(GraphType& g)
                           : GraphAdder(g) {}

        bool operator()(SWU* ptr);
    };

    class AddDependencies : public GraphAdder {
      public:
        explicit AddDependencies(GraphType& g)
                                 : GraphAdder(g) {}

        bool operator()(SWU* ptr);
    };

    class AddDependents : public GraphAdder {
      public:
        explicit AddDependents(GraphType& g)
                               : GraphAdder(g) {}

        bool operator()(SWU* ptr);
    };

    private:
      // The grapher object
      GraphType grapher_;

      // The linkler object used for generating the grapher
      Linkler* linkler_;
};

#endif