Changeset 15765 for branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/GrammarEnumeration/GrammarEnumerationAlgorithm.cs

Timestamp:

02/13/18 13:01:08 (6 years ago)

Author:

lkammere

Message:

#2886: Add graphviz output.

File:

• branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/GrammarEnumeration/GrammarEnumerationAlgorithm.cs (modified) (9 diffs)

branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/GrammarEnumeration/GrammarEnumerationAlgorithm.cs

@@ -1 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 using System.Linq;
+using System.Text;
 using System.Threading;
 using HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration.GrammarEnumeration;
@@ -32 +34 @@
     private readonly string MaxTreeSizeParameterName = "Max. Tree Nodes";
     private readonly string GuiUpdateIntervalParameterName = "GUI Update Interval";
-    
+
     public override bool SupportsPause { get { return false; } }
 
@@ -74 +76 @@
     public Grammar Grammar;
 
+    public StringBuilder dotFileBuilder;
+
     #region ctors
     public override IDeepCloneable Clone(Cloner cloner) {
@@ -99 +103 @@
       #region init
       InitResults();
+
+      dotFileBuilder = new StringBuilder();
 
       AllSentences = new Dictionary<int, List<SymbolString>>();
@@ -120 +126 @@
         SymbolString currPhrase = fetchedPhrase.SymbolString;
 
+        if (dotFileBuilder.Length == 0) {
+          dotFileBuilder.AppendLine("digraph searchgraph {");
+          dotFileBuilder.AppendLine(fetchedPhrase.Hash + " [label=\"" + Grammar.PostfixToInfixParser(fetchedPhrase.SymbolString) + "\", shape=doublecircle];");
+        } else {
+          dotFileBuilder.AppendLine(fetchedPhrase.Hash + " [label=\"" + Grammar.PostfixToInfixParser(fetchedPhrase.SymbolString) + "\"];");
+        }
+
         ArchivedPhrases.Add(fetchedPhrase.Hash, currPhrase);
 
@@ -135 +148 @@
             var phraseHash = Grammar.CalcHashCode(newPhrase);
 
+            dotFileBuilder.AppendLine(fetchedPhrase.Hash + " -> " + phraseHash + " [label=\"" + expandedSymbol.StringRepresentation + "&rarr;" + productionAlternative + "\"];");
+
             if (newPhrase.IsSentence()) { // Sentence was generated.
               SaveToAllSentences(phraseHash, newPhrase);
@@ -143 +158 @@
                 DistinctSentences[phraseHash] = newPhrase;
                 EvaluateSentence(newPhrase);
+
+                dotFileBuilder.AppendLine(phraseHash + " [label=\"" + Grammar.PostfixToInfixParser(newPhrase) + "\", style=\"filled\"];");
               }
               UpdateView(AllSentences, DistinctSentences, Expansions);
@@ -152 +169 @@
         }
       }
+
+      dotFileBuilder.AppendLine(Grammar.CalcHashCode(BestTrainingSentence) + " [label=\"" + Grammar.PostfixToInfixParser(BestTrainingSentence) + "\", shape=Mcircle, style=\"filled,bold\"];");
+      dotFileBuilder.AppendLine("}");
+
+      System.IO.File.WriteAllText(Environment.GetFolderPath(System.Environment.SpecialFolder.DesktopDirectory)+ @"\searchgraph.dot", dotFileBuilder.ToString());
 
       UpdateView(AllSentences, DistinctSentences, Expansions, force: true);
@@ -257 +279 @@
     }
     #endregion
- 
+
   }
 }