tpgGraphDotExporter.h

 
#ifndef TPG_GRAPH_DOT_EXPORTER_H
#define TPG_GRAPH_DOT_EXPORTER_H
 
#include <map>
#include <stdexcept>
#include <string>
 
#include "program/line.h"
#include "program/program.h"
#include "tpg/tpgAbstractEngine.h"
#include "tpg/tpgAction.h"
#include "tpg/tpgEdge.h"
#include "tpg/tpgGraph.h"
#include "tpg/tpgTeam.h"
#include "tpg/tpgVertex.h"
 
namespace File {
    class TPGGraphDotExporter : TPG::TPGAbstractEngine
    {
      protected:
        FILE* pFile;
 
        std::string offset;
 
        void printTPGTeam(const TPG::TPGTeam& team);
 
        uint64_t printTPGAction(const TPG::TPGAction& action);
 
        void printTPGEdge(const TPG::TPGEdge& edge);
 
        void printProgram(const Program::Program& program);
 
        void printTPGGraphHeader();
 
        void printTPGGraphFooter();
 
      public:
        TPGGraphDotExporter(const char* filePath, const TPG::TPGGraph& graph)
            : TPG::TPGAbstractEngine(graph), pFile{NULL}, offset{""}
        {
            if ((pFile = fopen(filePath, "w")) == NULL) {
                throw std::runtime_error("Could not open file " +
                                         std::string(filePath));
            }
        };
 
        TPGGraphDotExporter(const TPGGraphDotExporter& other) = delete;
 
        TPGGraphDotExporter& operator=(const TPGGraphDotExporter& other) =
            delete;
 
        ~TPGGraphDotExporter()
        {
            if (pFile != NULL) {
                fclose(pFile);
            }
        }
 
        void setNewFilePath(const char* newFilePath)
        {
            //  Close previous file
            fclose(pFile);
 
            // open new one;
            if ((pFile = fopen(newFilePath, "w")) == NULL) {
                pFile = NULL;
                throw std::runtime_error("Could not open file " +
                                         std::string(newFilePath));
            }
        }
 
        void print();
 
        void printSubGraph(const TPG::TPGVertex* root);
    };
}; // namespace File
 
#endif