From 34dafa970445bce32692269da622c26782d88b67 Mon Sep 17 00:00:00 2001
From: Paul Nitzke <14367-paulenit@users.noreply.git.rwth-aachen.de>
Date: Wed, 22 Jun 2022 16:52:50 +0200
Subject: [PATCH] Add graph to constructor and store it

* Required for getting the number of colors used in coloring
---
 src/SparseTransform/ConversionManager.cs        | 2 +-
 src/SparseTransform/Convert/ColoredDotWriter.cs | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/SparseTransform/ConversionManager.cs b/src/SparseTransform/ConversionManager.cs
index 4084f6e..30297ec 100644
--- a/src/SparseTransform/ConversionManager.cs
+++ b/src/SparseTransform/ConversionManager.cs
@@ -40,7 +40,7 @@ namespace SparseTransform
             }
             else
                 color.PartialD2Color((BipartiteGraph)graph);
-            ColoredDotWriter writer = new ColoredDotWriter(textColor);
+            ColoredDotWriter writer = new ColoredDotWriter(graph, textColor);
             return writer.Write(graph);
 
         }
diff --git a/src/SparseTransform/Convert/ColoredDotWriter.cs b/src/SparseTransform/Convert/ColoredDotWriter.cs
index be629b6..406fd82 100644
--- a/src/SparseTransform/Convert/ColoredDotWriter.cs
+++ b/src/SparseTransform/Convert/ColoredDotWriter.cs
@@ -7,6 +7,7 @@ namespace SparseTransform.Convert
     /// </summary>
     public class ColoredDotWriter : DotWriter
     {
+        private IGraph _coloredGraph;
         private bool textcolor;
         private string[] dictColor = new string[65];
        
@@ -14,8 +15,9 @@ namespace SparseTransform.Convert
         /// Constructor constructing a color-dictionary for coloring purpose.
         /// </summary>
         /// <param name="textColor">decision whether the labels should be colored or labelled.</param>
-        public ColoredDotWriter(bool textColor)
+        public ColoredDotWriter(IGraph graph, bool textColor)
         {
+            this._coloredGraph = graph;
             this.textcolor = textColor;
 
             for (int i = 0; i <= 3; i++)
-- 
GitLab