Class: DecisionAgent::Dmn::TreeDotGenerator

Inherits:

Object

• Object
• DecisionAgent::Dmn::TreeDotGenerator

Defined in:

lib/decision_agent/dmn/visualizer.rb

Overview

Generates DOT format for decision trees (for Graphviz)

Instance Method Summary

• #generate ⇒ Object
• #initialize(decision_tree) ⇒ TreeDotGenerator constructor

  A new instance of TreeDotGenerator.

Constructor Details

#initialize(decision_tree) ⇒ TreeDotGenerator

Returns a new instance of TreeDotGenerator.

# File 'lib/decision_agent/dmn/visualizer.rb', line 202

def initialize(decision_tree)
  @tree = decision_tree
end

Instance Method Details

#generate ⇒ Object

# File 'lib/decision_agent/dmn/visualizer.rb', line 206

def generate
  dot = ["digraph decision_tree {"]
  dot << "  graph [rankdir=TB, splines=ortho];"
  dot << "  node [shape=box, style=rounded];"

  generate_nodes(@tree.root, dot)
  generate_edges(@tree.root, dot)

  dot << "}"
  dot.join("\n")
end