Class: DecisionAgent::Dmn::TreeMermaidGenerator

Inherits:
Object
  • Object
show all
Defined in:
lib/decision_agent/dmn/visualizer.rb

Overview

Generates Mermaid diagram syntax for decision trees

Instance Method Summary collapse

Constructor Details

#initialize(decision_tree) ⇒ TreeMermaidGenerator

Returns a new instance of TreeMermaidGenerator.



446
447
448
 
# File 'lib/decision_agent/dmn/visualizer.rb', line 446

def initialize(decision_tree)
  @tree = decision_tree
end

Instance Method Details

#generate ⇒ Object 



450
451
452
453
454
455
 
# File 'lib/decision_agent/dmn/visualizer.rb', line 450

def generate
  mermaid = ["graph TD"]
  generate_nodes(@tree.root, mermaid)
  generate_edges(@tree.root, mermaid)
  mermaid.join("\n")
end