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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
from __future__ import absolute_import
from __future__ import print_function
import os
import sys
import math
from .. import Tree, CircleFace, RectFace, TextFace, add_face_to_node, TreeStyle
NEWLINE = '\n'
TAB = '\t'
UNDERBAR = '_'
SPACE = ' '
COMMA = ','
_circle_tested = False
def node_layout(annotation):
def _node_layout(node):
global _circle_tested
node.img_style["size"] = 0
if node.is_leaf():
if node.name not in annotation:
print('Got unknown leaf "%s"' % (node.name))
return
size = annotation[node.name]
else:
children = COMMA.join(sorted([leaf.name for leaf in node]))
if children not in annotation:
print('Got unknown node ' + children)
return
size = annotation[children]
dimension = 4*math.sqrt(size)
labeld = {
'text' : "%d" % (size),
'color' : 'white',
'font' : 'Helvetica',
'size' : 8
}
if size % 2:
clabel = dict(labeld)
if size == 35:
if not _circle_tested:
_circle_tested = True
clabel['text'] = 'Ly'
clabel['size'] = 12
thisFace = CircleFace(dimension/2, "steelblue", "circle", label=clabel)
elif size == 43:
clabel['size'] = 6
del clabel['color']
thisFace = CircleFace(dimension/2, "steelblue", "sphere", label=clabel)
else:
thisFace = CircleFace(dimension/2, "steelblue", "sphere", label="%d" % (size))
else:
thisFace = RectFace(dimension, dimension, 'green', 'blue', label=labeld)
thisFace.opacity = 0.7
add_face_to_node(thisFace, node, column=0, position="float")
textF = TextFace(str(size), fsize=12, fgcolor="steelblue")
add_face_to_node(textF, node, column=0, position="aligned")
return _node_layout
TREE = '''
(S032 :0.68575,(S032 473 :0.29028,(S032 476 :0.19027,
(S032 477 :0.21930,(S032 478 :0.15731,(S032 475 :0.08076,
S032 474 :0.08914):0.04624):0.04277):0.01821):0.07911):0.02825,S032 202 :0.31425);
'''
SETS = '''
43 S032_477
42 S032_202
35 S032_476
35 S032_474
33 S032_475
32 S032_478
29 S032_473
25 S032_474,S032_475
20 S032_476,S032_477
20 S032_475,S032_478
20 S032_475,S032_477
20 S032_474,S032_478
20 S032_474,S032_476
19 S032_474,S032_475,S032_478
19 S032_474,S032_475,S032_477
19 S032_475,S032_476
18 S032_474,S032_475,S032_476
18 S032_477,S032_478
17 S032_475,S032_477,S032_478
17 S032_476,S032_478
16 S032_474,S032_475,S032_477,S032_478
16 S032_474,S032_476,S032_478
15 S032_474,S032_475,S032_476,S032_478
15 S032_474,S032_475,S032_476,S032_477
15 S032_473,S032_477
14 S032_473,S032_476
13 S032_474,S032_475,S032_476,S032_477,S032_478
12 S032_202,S032_475
11 S032_473,S032_475,S032_476
11 S032_202,S032_474,S032_475
10 S032_473,S032_474,S032_475,S032_476,S032_478
10 S032_202,S032_474,S032_475,S032_477,S032_478
9 S032_473,S032_476,S032_477
8 S032_473,S032_474,S032_475,S032_476,S032_477,S032_478
8 S032_202,S032_473,S032_475,S032_476
7 S032_202,S032_473,S032_474,S032_475,S032_476,S032_477,S032_478
'''
treesets = SETS.strip().split(NEWLINE)
thesets = dict()
for s in treesets:
num, namelist = s.strip().split(TAB)
thesets[namelist.replace(UNDERBAR, SPACE)] = int(num)
t = Tree(TREE)
ts = TreeStyle()
ts.layout_fn = node_layout(thesets)
ts.show_leaf_name = True
ts.show_branch_length = True
t.show(tree_style=ts)
#t.render('S032.png', tree_style=ts)
|
