#ifndef CGRAPHVIZDIAGRAMER_H
#define CGRAPHVIZDIAGRAMER_H

#include <QMap>
#include <graphviz/gvc.h>
#include "IDiagramer.h"

namespace Diagram {

class CGraphvizDiagramer : public IDiagramer
{
public:
    CGraphvizDiagramer();
    ~CGraphvizDiagramer();
    void resetGraph() override;
    void addNode(std::shared_ptr<Node> pNode) override;
    void addEdge(std::shared_ptr<Edge> pEdge) override;
    void setLayout(LAYOUT_TYPE type) override;
    void setGraphConfig(const QVector<QPair<QVariant, QVariant>>& config) override;
    QImage generateImg() override;
    void setHideSingleClass(bool bFlag) override;
    void setHideOneToOne(bool bFlag) override;

private:
    struct ExtNode
    {
        std::shared_ptr<Node> pNode;
        Agnode_t* pAgnode;
    };

private:
    void __removeHideNode();

private:
    GVC_t* m_pGvc;
    Agraph_t* m_pGraph;
    QMap<QString, ExtNode> m_mapNodes;
    LAYOUT_TYPE m_layoutType;
    bool m_bHideSingleClass;
    bool m_bHideOneToOne;
    QVector<QPair<QVariant, QVariant>> m_graphConfig;
};


}

#endif // CGRAPHVIZDIAGRAMER_H