-
Notifications
You must be signed in to change notification settings - Fork 9
/
dot.ex
53 lines (45 loc) · 1.33 KB
/
dot.ex
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
defmodule StatesLanguage.Serializer.Dot do
@moduledoc """
Serializer for `StatesLanguage.Graph` to Graphviz compatible .dot file
"""
alias StatesLanguage.{Edge, Graph, Node}
require Logger
def serialize(%{} = data) do
%Graph{start: start, nodes: nodes, edges: edges} = Graph.serialize(data)
nodes = sort_nodes(nodes, start)
output = parse_nodes("", nodes)
output = parse_edges(output, edges, nodes)
"digraph my_graph {\n" <> output <> "}\n"
end
defp sort_nodes(nodes, start) do
nodes
|> Enum.sort(fn
{^start, _}, _ -> true
_, _ -> false
end)
end
defp parse_nodes(output, nodes) do
nodes
|> Enum.with_index()
|> Enum.reduce(output, fn {{k, %Node{}}, i}, acc ->
acc <> " node#{i} [label=\"#{k}\", shape=box];\n"
end)
end
defp parse_edges(output, edges, nodes) do
Enum.reduce(edges, output, fn %Edge{} = edge, acc ->
source = get_node_index_from_name(edge.source, nodes)
target = get_node_index_from_name(edge.target, nodes)
label =
edge.event
|> Macro.to_string()
|> String.replace("\"", "\\\"")
acc <> ~s( node#{source} -> node#{target} [label="#{label}"];\n)
end)
end
defp get_node_index_from_name(name, nodes) do
Enum.find_index(nodes, fn
{^name, _} -> true
_ -> false
end)
end
end