1 files changed, 8 insertions, 6 deletions
diff --git a/src/dot.h b/src/dot.h
index 3a70d8b..20a1f29 100644
--- a/src/dot.h
+++ b/src/dot.h
@@ -26,8 +26,8 @@
 #include <qqueue.h>
 #include <qthread.h>
 #include "sortdict.h"
+#include "classdef.h"
 
-class ClassDef;
 class FileDef;
 class FTextStream;
 class DotNodeList;
@@ -156,7 +156,7 @@ class DotNodeList : public QList<DotNode>
 class DotGfxHierarchyTable
 {
   public:
-    DotGfxHierarchyTable();
+    DotGfxHierarchyTable(const char *prefix="",ClassDef::CompoundType ct=ClassDef::Class);
    ~DotGfxHierarchyTable();
     void writeGraph(FTextStream &t,const char *path, const char *fileName) const;
     void createGraph(DotNode *rootNode,FTextStream &t,const char *path,const char *fileName,int id) const;
@@ -166,10 +166,12 @@ class DotGfxHierarchyTable
     void addHierarchy(DotNode *n,ClassDef *cd,bool hide);
     void addClassList(ClassSDict *cl);
 
-    QList<DotNode> *m_rootNodes; 
-    QDict<DotNode> *m_usedNodes; 
-    int             m_curNodeNumber;
-    DotNodeList    *m_rootSubgraphs;
+    QCString               m_prefix;
+    ClassDef::CompoundType m_classType;
+    QList<DotNode>        *m_rootNodes; 
+    QDict<DotNode>        *m_usedNodes; 
+    int                    m_curNodeNumber;
+    DotNodeList           *m_rootSubgraphs;
 };
 
 /** Representation of a class inheritance or dependency graph */