module_graph_builder.h

#ifndef GRAPH_BUILDER_H
#define GRAPH_BUILDER_H

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

#include "defines.h"
#include "grapher.h"
#include "linkler.h"
#include "dependency.h"
#include "module.h"
#include "graph_builder.h"

class ModuleGraphBuilder : public GraphBuilder {
   public:
    typedef Grapher<Module*,Dependency<Module,Module>*, CompareModuleByName, CompareModuleDependency> GraphType;

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

    GraphType& grapher();

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

    void graph_all();

    void graph_all_dependencies();

    void graph_all_dependents();

    void graph_single(std::string module);

    void graph_recursively(std::string module);

    void graph_dependencies(std::string module);

    void graph_dependents(std::string module);

    Linkler* linkler();

    Module* get(std::string module_name); 

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

      protected:
        GraphType& grapher_;
    };

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

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

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

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

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

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

    private:
      // The grapher object
      GraphType grapher_;

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

#endif