Changeset 30 for trunk/EventBenchCore

Timestamp:

04/14/11 16:09:27 (14 years ago)

Author:

sherbold

Message:

+ added Dot output to de.ugoe.cs.eventbench.models.Trie and accessor for the Dot representation of the trie to de.ugoe.cs.eventbench.models.TrieBasedModel?

Location:

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models

Files:

• Trie.java (modified) (2 diffs)
• TrieBasedModel.java (modified) (1 diff)
• TrieNode.java (modified) (2 diffs)

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/Trie.java

@@ -6 +6 @@
 import java.util.Set;
 
+import de.ugoe.cs.util.StringTools;
+
 import edu.uci.ics.jung.graph.DelegateTree;
 import edu.uci.ics.jung.graph.Tree;
 
-public class Trie<T> {
+public class Trie<T> implements IDotCompatible {
         
         private Set<T> knownSymbols;
@@ -148 +150 @@
         }
         
+        public String getDotRepresentation() {
+                StringBuilder stringBuilder = new StringBuilder();
+                stringBuilder.append("digraph model {" + StringTools.ENDLINE);
+                rootNode.appendDotRepresentation(stringBuilder);
+                stringBuilder.append('}' + StringTools.ENDLINE);
+                return stringBuilder.toString();
+        }
+        
         @Override
         public String toString() {

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/TrieBasedModel.java

@@ -73 +73 @@
         }
         
+        public String getTrieDotRepresentation() {
+                return trie.getDotRepresentation();
+        }
+        
         public Tree<TrieVertex, Edge> getTrieGraph() {
                 return trie.getGraph();

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/TrieNode.java

@@ -26 +26 @@
         public TrieNode(T symbol) {
                 if( symbol==null ) {
-                        throw new InvalidParameterException("symbol must not be null.");
+                        throw new InvalidParameterException("symbol must not be null. null is reserved for root node!");
                 }
                 this.symbol = symbol;
@@ -120 +120 @@
                 }               
         }
-
+        
+        void appendDotRepresentation(StringBuilder stringBuilder) {
+                String thisSaneId;
+                if( symbol==null ) {
+                        thisSaneId = "root";
+                } else {
+                        thisSaneId = symbol.toString().replace("\"", "\\\"").replaceAll("[\r\n]","")+"#"+count;
+                }
+                stringBuilder.append(" " + hashCode() + " [label=\""+thisSaneId+"\"];" + StringTools.ENDLINE);
+                for( TrieNode<T> childNode : children ) {
+                        stringBuilder.append(" "+hashCode()+" -> " + childNode.hashCode() + ";" + StringTools.ENDLINE);
+                }
+                for( TrieNode<T> childNode : children ) {
+                        childNode.appendDotRepresentation(stringBuilder);
+                }
+        }
 }