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
54
55
56
57
|
#copyright ben lipkowitz, 2009. distributed under the GNU GPL version 2 or later
#usage:
# cat processes.yaml | sed 's/!.*$//' > processes_notags.yaml #strip tags which confuse yaml
# python taxonomy-graph.py |dot -Grankdir=LR -Tjpg -o 'foo.jpg'
import os
import yaml
import graph
import random
depthcolors = {} #keeping track of colors per each level
def random_color(depth):
'''A colorvalue may be "h,s,v" (hue, saturation, brightness) floating
point numbers between 0 and 1, or an X11 color name such as white black
red green blue yellow magenta cyan or burlywood, or a "#rrggbb" (red,
green, blue, 2 hex characters each) value.'''
hue = random.uniform(0,1)
saturation = random.uniform(0, 0.3)
brightness = 0.9
returnstring = '"%f,%f,%f"' % (hue, saturation, brightness)
# set color to previous color at this depth (if exists)
if depthcolors.has_key(depth):
returnstring = (depthcolors[depth])[0]
return returnstring
# alt functionality: check if there's a color too similar already at this depth.
# similarity means ?
return returnstring
#linnaeus = yaml.load(open('processes_notags.yaml'))['abrasive jet']
linnaeus = yaml.load(open('trans-tech.yaml'))
taxonomy = graph.digraph()
node_id=0 #we need numerical nodes because some terms show up multiple times, like thermal, mechanical, chemical
def walk(treebeard, color, parent_node, depth):
global node_id
children = []
if hasattr(treebeard, 'keys'):
children = treebeard.keys()
for child in children:
node_id += 1
taxonomy.add_node(node_id, [('label', child), ('shape', 'box'),
('fontsize', '24'), ('color', color), ('style', 'filled')])
taxonomy.add_edge(parent_node, node_id)
mycolor = random_color(depth)
if (depthcolors.has_key(depth)): depthcolors[depth].append(mycolor)
else: depthcolors[depth] = [color]
#if hasattr(child, 'keys'):
walk(treebeard[child], mycolor, node_id, depth+1)
taxonomy.add_node(node_id, [('label', 'root')])
walk(linnaeus, 'yellow', node_id, 0)
print graph.readwrite.write_dot_graph(taxonomy, False)
|
