Graphviz.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <string>
#include <unordered_map>
#include <unordered_set>
 
#include "smgl/Graph.hpp"
#include "smgl/Utilities.hpp"
#include "smgl/filesystem.hpp"
 
namespace smgl
{
 
struct ElementStyle {
    std::string align;
    std::string bgcolor;
    std::uint8_t border{1};
    std::string color;
};
 
struct BaseStyle : public ElementStyle {
    std::uint8_t cellborder{1};
    std::uint8_t cellpadding{2};
    std::uint8_t cellspacing{2};
};
 
struct FontStyle {
    std::string color;
    std::string face;
    auto empty() const -> bool { return color.empty() and face.empty(); }
};
 
struct NodeStyle {
    BaseStyle base;
    FontStyle font;
    ElementStyle inputPorts;
    ElementStyle label;
    ElementStyle outputPorts;
};
 
class GraphStyle
{
public:
    GraphStyle();
 
    void setDefaultStyle(const NodeStyle& style);
    auto defaultStyle() -> NodeStyle&;
    auto defaultStyle() const -> const NodeStyle&;
    void setClassStyle(const std::string& name, const NodeStyle& style);
    template <class NodeT>
    void setClassStyle(const NodeStyle& style);
    auto hasClassStyle(const std::string& name) const -> bool;
    template <class NodeT>
    auto hasClassStyle() const -> bool;
    void eraseClassStyle(const std::string& name);
    template <class NodeT>
    void eraseClassStyle() const;
    auto classStyle(const std::string& name) -> NodeStyle&;
    template <class NodeT>
    auto classStyle() -> NodeStyle&;
    auto classStyle(const std::string& name) const -> const NodeStyle&;
    template <class NodeT>
    auto classStyle() const -> const NodeStyle&;
    void setInstanceStyle(const Node::Pointer& node, const NodeStyle& style);
    void setInstanceStyle(const Node* node, const NodeStyle& style);
    auto hasInstanceStyle(const Node::Pointer& node) const -> bool;
    auto hasInstanceStyle(const Node* node) const -> bool;
    void eraseInstanceStyle(const Node::Pointer& node);
    void eraseInstanceStyle(const Node* node);
    auto instanceStyle(const Node::Pointer& node) -> NodeStyle&;
    auto instanceStyle(const Node::Pointer& node) const -> const NodeStyle&;
    auto instanceStyle(const Node* node) -> NodeStyle&;
    auto instanceStyle(const Node* node) const -> const NodeStyle&;
    auto nodeStyle(const Node::Pointer& node) const -> NodeStyle;
    auto nodeStyle(const Node* node) const -> NodeStyle;
    template <typename... Args>
    void setRankMin(const Args&... args);
 
    template <typename... Args>
    void setRankSource(const Args&... args);
 
    template <typename... Args>
    void setRankMax(const Args&... args);
 
    template <typename... Args>
    void setRankSink(const Args&... args);
 
    template <typename... Args>
    auto setRankSame(const Args&... args) -> std::size_t;
 
    template <typename T, typename... Args>
    void appendRankSame(T idx, const Args&... args);
 
    auto rankMin() -> std::unordered_set<Uuid>&;
    auto rankMin() const -> const std::unordered_set<Uuid>&;
 
    auto rankSource() -> std::unordered_set<Uuid>&;
    auto rankSource() const -> const std::unordered_set<Uuid>&;
 
    auto rankMax() -> std::unordered_set<Uuid>&;
    auto rankMax() const -> const std::unordered_set<Uuid>&;
 
    auto rankSink() -> std::unordered_set<Uuid>&;
    auto rankSink() const -> const std::unordered_set<Uuid>&;
 
    auto rankSame() -> std::vector<std::vector<Uuid>>&;
    auto rankSame() const -> const std::vector<std::vector<Uuid>>&;
private:
    NodeStyle defaultStyle_;
    std::unordered_map<std::string, NodeStyle> classStyles_;
    std::unordered_map<Uuid, NodeStyle> instanceStyles_;
    std::unordered_set<Uuid> rankMin_;
    std::unordered_set<Uuid> rankSrc_;
    std::unordered_set<Uuid> rankMax_;
    std::unordered_set<Uuid> rankSink_;
    std::vector<std::vector<Uuid>> rankSame_;
};
 
void WriteDotFile(
    const filesystem::path& path,
    const Graph& g,
    const GraphStyle& style = GraphStyle());
 
} // namespace smgl
 
#include "smgl/GraphvizImpl.hpp"