imstkTaskGraphVizWriter.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include <memory>
#include <string>

namespace imstk
{
class TaskGraph;

class TaskGraphVizWriter
{
public:
    TaskGraphVizWriter() = default;
    virtual ~TaskGraphVizWriter() = default;

public:
    void setInput(std::shared_ptr<TaskGraph> graph) { this->m_inputGraph = graph; }

    void setFileName(std::string fileName) { this->m_fileName = fileName; }

    void setHighlightCriticalPath(bool highlightCriticalPath) { this->m_highlightCriticalPath = highlightCriticalPath; }

    void setWriteNodeComputeTimesColor(bool writeNodeComputeTimesColor) { this->m_writeNodeComputeTimesColor = writeNodeComputeTimesColor; }

    void setWriteNodeComputeTimesText(bool writeNodeComputeTimesText) { this->m_writeNodeComputeTimesText = writeNodeComputeTimesText; }

    std::shared_ptr<TaskGraph> getInput() const { return m_inputGraph; }
    const std::string& getFileName() const { return m_fileName; }
    bool getHighlightCriticalPath() const { return m_highlightCriticalPath; }
    bool getWriteNodeComputeTimesColor() const { return m_writeNodeComputeTimesColor; }
    bool getWriteNodeComputeTimesText() const { return m_writeNodeComputeTimesText; }

    void write();

private:
    std::shared_ptr<TaskGraph> m_inputGraph = nullptr;
    std::string m_fileName            = "";
    bool m_highlightCriticalPath      = false;
    bool m_writeNodeComputeTimesColor = false;
    bool m_writeNodeComputeTimesText  = false;
};
} // namespace imstk