refactoring dot.cpp

1 files changed, 9 insertions, 5 deletions

diff --git a/src/rtfgen.cpp b/src/rtfgen.cpp
index bb2075b..8139784 100644
--- a/src/rtfgen.cpp
+++ b/src/rtfgen.cpp
@@ -31,6 +31,10 @@
 #include "diagram.h"
 #include "language.h"
 #include "dot.h"
+#include "dotcallgraph.h"
+#include "dotclassgraph.h"
+#include "dotdirdeps.h"
+#include "dotincldepgraph.h"
 #include "version.h"
 #include "pagedef.h"
 #include "rtfstyle.h"
@@ -2497,7 +2501,7 @@ void RTFGenerator::startDotGraph()
   DBG_RTF(t << "{\\comment (startDotGraph)}"    << endl)
 }
 
-void RTFGenerator::endDotGraph(const DotClassGraph &g)
+void RTFGenerator::endDotGraph(DotClassGraph &g)
 {
   newParagraph();
 
@@ -2521,7 +2525,7 @@ void RTFGenerator::startInclDepGraph()
   DBG_RTF(t << "{\\comment (startInclDepGraph)}"    << endl)
 }
 
-void RTFGenerator::endInclDepGraph(const DotInclDepGraph &g)
+void RTFGenerator::endInclDepGraph(DotInclDepGraph &g)
 {
   newParagraph();
 
@@ -2543,7 +2547,7 @@ void RTFGenerator::startGroupCollaboration()
 {
 }
 
-void RTFGenerator::endGroupCollaboration(const DotGroupCollaboration &)
+void RTFGenerator::endGroupCollaboration(DotGroupCollaboration &)
 {
 }
 
@@ -2552,7 +2556,7 @@ void RTFGenerator::startCallGraph()
   DBG_RTF(t << "{\\comment (startCallGraph)}"    << endl)
 }
 
-void RTFGenerator::endCallGraph(const DotCallGraph &g)
+void RTFGenerator::endCallGraph(DotCallGraph &g)
 {
   newParagraph();
 
@@ -2575,7 +2579,7 @@ void RTFGenerator::startDirDepGraph()
   DBG_RTF(t << "{\\comment (startDirDepGraph)}"    << endl)
 }
 
-void RTFGenerator::endDirDepGraph(const DotDirDeps &g)
+void RTFGenerator::endDirDepGraph(DotDirDeps &g)
 {
   newParagraph();