def phylogeny(node):
leaf_color = "#000000"
node.img_style["shape"] = "circle"
if hasattr(node, "evoltype"):
if node.evoltype == 'D':
node.img_style["fgcolor"] = "#FF0000"
node.img_style["hz_line_color"] = "#FF0000"
node.img_style["vt_line_color"] = "#FF0000"
elif node.evoltype == 'S':
node.img_style["fgcolor"] = "#1d176e"
node.img_style["hz_line_color"] = "#1d176e"
node.img_style["vt_line_color"] = "#1d176e"
elif node.evoltype == 'L':
node.img_style["fgcolor"] = "#777777"
node.img_style["vt_line_color"] = "#777777"
node.img_style["hz_line_color"] = "#777777"
node.img_style["hz_line_type"] = 1
node.img_style["vt_line_type"] = 1
leaf_color = "#777777"
if node.is_leaf():
node.img_style["shape"] = "square"
node.img_style["size"] = 4
node.img_style["fgcolor"] = leaf_color
faces.add_face_to_node(
faces.AttrFace("name", "Arial", 11, leaf_color, None), node, 0)
if hasattr(node, "sequence"):
SequenceFace = faces.SequenceFace(node.sequence, "aa", 13)
faces.add_face_to_node(SequenceFace, node, 1, aligned=True)
else:
node.img_style["size"] = 6
def heatmap(node):
square_size = 10
# Extras node info
node.collapsed = False
# Color and style
node.img_style["fgcolor"] = "#3333FF"
node.img_style["size"] = 0
ncols = node.arraytable.matrix.shape[1]
matrix_max = numpy.max(node.arraytable._matrix_max)
matrix_min = numpy.min(node.arraytable._matrix_min)
print matrix_max
matrix_avg = matrix_min + ((matrix_max - matrix_min) / 2)
ProfileFace = faces.ProfileFace(\
matrix_max,\
matrix_min,\
matrix_avg,\
square_size*ncols,\
square_size,\
"heatmap")
ProfileFace.ymargin = 0
if node.is_leaf():
# Set colors
faces.add_face_to_node(ProfileFace, node, 0, aligned=True)
def heatmap(node):
square_size = 10
# Extras node info
node.collapsed = False
# Color and style
node.img_style["fgcolor"] = "#3333FF"
node.img_style["size"] = 0
ncols = node.arraytable.matrix.shape[1]
matrix_max = numpy.max(node.arraytable._matrix_max)
matrix_min = numpy.min(node.arraytable._matrix_min)
print matrix_max
matrix_avg = matrix_min+((matrix_max-matrix_min)/2)
ProfileFace = faces.ProfileFace(\
matrix_max,\
matrix_min,\
matrix_avg,\
square_size*ncols,\
square_size,\
"heatmap")
ProfileFace.ymargin=0
if node.is_leaf():
# Set colors
faces.add_face_to_node(ProfileFace, node, 0, aligned=True)
def phylogeny(node):
leaf_color = "#000000"
node.img_style["shape"] = "circle"
if hasattr(node,"evoltype"):
if node.evoltype == 'D':
node.img_style["fgcolor"] = "#FF0000"
node.img_style["hz_line_color"] = "#FF0000"
node.img_style["vt_line_color"] = "#FF0000"
elif node.evoltype == 'S':
node.img_style["fgcolor"] = "#1d176e"
node.img_style["hz_line_color"] = "#1d176e"
node.img_style["vt_line_color"] = "#1d176e"
elif node.evoltype == 'L':
node.img_style["fgcolor"] = "#777777"
node.img_style["vt_line_color"] = "#777777"
node.img_style["hz_line_color"] = "#777777"
node.img_style["hz_line_type"] = 1
node.img_style["vt_line_type"] = 1
leaf_color = "#777777"
if node.is_leaf():
node.img_style["shape"] = "square"
node.img_style["size"] = 4
node.img_style["fgcolor"] = leaf_color
faces.add_face_to_node( faces.AttrFace("name","Arial",11,leaf_color,None), node, 0 )
if hasattr(node,"sequence"):
SequenceFace = faces.SequenceFace(node.sequence,"aa",13)
faces.add_face_to_node(SequenceFace, node, 1, aligned=True)
else:
node.img_style["size"] = 6
def evol_clean_layout(node):
'''
layout for CodemlTree
'''
if hasattr(node, "collapsed"):
if node.collapsed == 1:
node.img_style["draw_descendants"]= False
leaf_color = "#000000"
node.img_style["size"] = 2
if node.is_leaf():
#if hasattr (node,"highlight"):
# faces.add_face_to_node(faces.AttrFace("name", "Arial", 9, \
# node.highlight,
# None), \
# node, 0 )
#else:
# leaface = faces.AttrFace("name", "Arial", 9, leaf_color, None)
# leaface.margin_right = 10
# faces.add_face_to_node (leaface, node, 0)
if hasattr (node, "sequence"):
seqface = faces.SequenceFace(node.sequence, interactive=True,
codon=node.nt_sequence)
faces.add_face_to_node(seqface, node, 1, aligned=True)
if hasattr (node, 'dN'):
faces.add_face_to_node (faces.TextFace('%.4f'%(node.w), fsize=6,
fgcolor="#7D2D2D"),
node, 0, position="branch-top")
faces.add_face_to_node (faces.TextFace('%.2f/%.2f'% (100*node.dN,
100*node.dS),
fsize=6, fgcolor="#787878"),
node, 0, position="branch-bottom")
if hasattr(node,"extras"):
faces.add_face_to_node( faces.AttrFace("extras", "Arial", 7, \
"#000000", None), node, 2 )
def evol_clean_layout(node):
'''
layout for CodemlTree
'''
if hasattr(node, "collapsed"):
if node.collapsed == 1:
node.img_style["draw_descendants"]= False
leaf_color = "#000000"
node.img_style["size"] = 2
if node.is_leaf():
#if hasattr (node,"highlight"):
# faces.add_face_to_node(faces.AttrFace("name", "Arial", 9, \
# node.highlight,
# None), \
# node, 0 )
#else:
# leaface = faces.AttrFace("name", "Arial", 9, leaf_color, None)
# leaface.margin_right = 10
# faces.add_face_to_node (leaface, node, 0)
if hasattr (node, "sequence"):
seqface = faces.SequenceFace(node.sequence, interactive=True,
codon=node.nt_sequence)
faces.add_face_to_node(seqface, node, 1, aligned=True)
if hasattr (node, 'dN'):
faces.add_face_to_node (faces.TextFace('%.4f'%(node.w), fsize=6,
fgcolor="#7D2D2D"),
node, 0, position="branch-top")
faces.add_face_to_node (faces.TextFace('%.2f/%.2f'% (100*node.dN,
100*node.dS),
fsize=6, fgcolor="#787878"),
node, 0, position="branch-bottom")
if hasattr(node,"extras"):
faces.add_face_to_node( faces.AttrFace("extras", "Arial", 7, \
"#000000", None), node, 2 )
def cluster_bars(node):
# Extras node info
node.collapsed = False
# Color and style
node.img_style["fgcolor"] = "#3333FF"
node.img_style["size"] = 4
if node.is_leaf():
matrix_max = numpy.max(node.arraytable._matrix_max)
matrix_min = numpy.min(node.arraytable._matrix_min)
matrix_avg = matrix_min + ((matrix_max - matrix_min) / 2)
ProfileFace = faces.ProfileFace(\
matrix_max,\
matrix_min,\
matrix_avg,\
200,\
40,\
"bars")
nameFace = faces.AttrFace("name", fsize=6)
faces.add_face_to_node(nameFace, node, 1, aligned=True)
faces.add_face_to_node(ProfileFace, node, 0, aligned=True)
def cluster_bars(node):
# Extras node info
node.collapsed = False
# Color and style
node.img_style["fgcolor"] = "#3333FF"
node.img_style["size"] = 4
if node.is_leaf():
matrix_max = numpy.max(node.arraytable._matrix_max)
matrix_min = numpy.min(node.arraytable._matrix_min)
matrix_avg = matrix_min+((matrix_max-matrix_min)/2)
ProfileFace = faces.ProfileFace(\
matrix_max,\
matrix_min,\
matrix_avg,\
200,\
40,\
"bars")
nameFace = faces.AttrFace("name",fsize=6 )
faces.add_face_to_node(nameFace, node, 1, aligned=True )
faces.add_face_to_node(ProfileFace, node, 0, aligned=True )
def evol_layout(node):
'''
layout for CodemlTree
'''
if hasattr(node, "collapsed"):
if node.collapsed == 1:
node.img_style["draw_descendants"] = False
leaf_color = "#000000"
if not node.is_root() and 'w' in node.features:
node.img_style["shape"] = 'circle'
if (node.w > 900):
node._w = 3
elif (node.w > 100):
node._w = 2.5
elif (node.w > 10):
node._w = 2
elif (node.w > 1):
node._w = 1.5
else:
node._w = node.w
node.img_style["size"] = int((float(node._w)) * 6 + 2)
if node._w == 3:
node.img_style["fgcolor"] = "#c10000"
if node._w == 2.5:
node.img_style["fgcolor"] = "#FF5A02"
if node._w == 2:
node.img_style["fgcolor"] = "#FFA200"
if node._w == 1.5:
node.img_style["fgcolor"] = "#E9BF00"
if node._w < 0.2:
node.img_style["fgcolor"] = "#000000"
if hasattr(node, "extras"):
faces.add_face_to_node( faces.AttrFace("extras", "Arial", 7, \
"#000000", None), node, 2 )
#if node.is_leaf():
if hasattr(node, "sequence"):
seqface = faces.SequenceFace(node.sequence,
interactive=True,
codon=node.nt_sequence)
faces.add_face_to_node(seqface, node, 1, aligned=True)
def evol_layout(node):
'''
layout for CodemlTree
'''
if hasattr(node, "collapsed"):
if node.collapsed == 1:
node.img_style["draw_descendants"]= False
leaf_color = "#000000"
if not node.is_root() and 'w' in node.features:
node.img_style["shape"] = 'circle'
if (node.w > 900):
node._w = 3
elif (node.w > 100):
node._w = 2.5
elif (node.w > 10) :
node._w = 2
elif (node.w > 1):
node._w = 1.5
else:
node._w = node.w
node.img_style["size"] = int ((float(node._w))*6+2)
if node._w == 3 :
node.img_style["fgcolor"] = "#c10000"
if node._w == 2.5 :
node.img_style["fgcolor"] = "#FF5A02"
if node._w == 2 :
node.img_style["fgcolor"] = "#FFA200"
if node._w == 1.5 :
node.img_style["fgcolor"] = "#E9BF00"
if node._w < 0.2 :
node.img_style["fgcolor"] = "#000000"
if hasattr(node,"extras"):
faces.add_face_to_node( faces.AttrFace("extras", "Arial", 7, \
"#000000", None), node, 2 )
#if node.is_leaf():
if hasattr (node, "sequence"):
seqface = faces.SequenceFace(node.sequence, interactive=True,
codon=node.nt_sequence)
faces.add_face_to_node(seqface, node, 1, aligned=True)
def render(root_node, img, hide_root=False):
'''main render function. hide_root option is used when render
trees as Faces
'''
mode = img.mode
orientation = img.orientation
arc_span = img.arc_span
layout_fn = img._layout_handler
parent = _TreeItem()
n2i = parent.n2i # node to items
n2f = parent.n2f # node to faces
parent.bg_layer = _EmptyItem(parent)
parent.tree_layer = _EmptyItem(parent)
parent.float_layer = _EmptyItem(parent)
parent.float_behind_layer = _EmptyItem(parent)
TREE_LAYERS = [parent.bg_layer, parent.tree_layer,
parent.float_layer, parent.float_behind_layer]
parent.bg_layer.setZValue(0)
parent.tree_layer.setZValue(2)
parent.float_behind_layer.setZValue(1)
parent.float_layer.setZValue(3)
# This could be used to handle aligned faces in internal
# nodes.
virtual_leaves = 0
if img.show_branch_length:
bl_face = faces.AttrFace("dist", fsize=8, ftype="Arial", fgcolor="black", formatter = "%0.3g")
if img.show_branch_support:
su_face = faces.AttrFace("support", fsize=8, ftype="Arial", fgcolor="darkred", formatter = "%0.3g")
if img.show_leaf_name:
na_face = faces.AttrFace("name", fsize=10, ftype="Arial", fgcolor="black")
for n in root_node.traverse(is_leaf_fn=_leaf):
set_style(n, layout_fn)
if img.show_branch_length:
faces.add_face_to_node(bl_face, n, 0, position="branch-top")
if not _leaf(n) and img.show_branch_support:
faces.add_face_to_node(su_face, n, 0, position="branch-bottom")
if _leaf(n) and img.show_leaf_name:
faces.add_face_to_node(na_face, n, 0, position="branch-right")
if _leaf(n):# or len(n.img_style["_faces"]["aligned"]):
virtual_leaves += 1
update_node_faces(n, n2f, img)
rot_step = float(arc_span) / virtual_leaves
#rot_step = float(arc_span) / len([n for n in root_node.traverse() if _leaf(n)])
# Calculate optimal branch length
if img._scale is not None:
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
elif img.scale is None:
# create items and calculate node dimensions skipping branch lengths
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
if mode == 'r':
if img.optimal_scale_level == "full":
scales = [(i.widths[1]/n.dist) for n,i in n2i.iteritems() if n.dist]
img._scale = max(scales) if scales else 0.0
else:
farthest, dist = root_node.get_farthest_leaf(topology_only=img.force_topology)
img._scale = img.tree_width / dist if dist else 0.0
update_branch_lengths(root_node, n2i, n2f, img)
else:
img._scale = crender.calculate_optimal_scale(root_node, n2i, rot_step, img)
#print "OPTIMAL circular scale", img._scale
update_branch_lengths(root_node, n2i, n2f, img)
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
else:
# create items and calculate node dimensions CONSIDERING branch lengths
img._scale = img.scale
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
#print "USING scale", img._scale
# Draw node content
for node in root_node.traverse(is_leaf_fn=_leaf):
if node is not root_node or not hide_root:
render_node_content(node, n2i, n2f, img)
# Adjust content to rect or circular layout
mainRect = parent.rect()
if mode == "c":
tree_radius = crender.render_circular(root_node, n2i, rot_step)
mainRect.adjust(-tree_radius, -tree_radius, tree_radius, tree_radius)
else:
iwidth = n2i[root_node].fullRegion.width()
iheight = n2i[root_node].fullRegion.height()
mainRect.adjust(0, 0, iwidth, iheight)
tree_radius = iwidth
# Add extra layers: aligned faces, floating faces, node
# backgrounds, etc. The order by which the following methods are
# called IS IMPORTANT
render_floatings(n2i, n2f, img, parent.float_layer, parent.float_behind_layer)
aligned_region_width = render_aligned_faces(img, mainRect, parent.tree_layer, n2i, n2f)
render_backgrounds(img, mainRect, parent.bg_layer, n2i, n2f)
# rotate if necessary in circular images. flip and adjust if mirror orientation.
adjust_faces_to_tranformations(img, mainRect, n2i, n2f, TREE_LAYERS)
# Rotate main image if necessary
parent.setRect(mainRect)
parent.setPen(QtGui.QPen(QtCore.Qt.NoPen))
if img.rotation:
rect = parent.boundingRect()
x = rect.x() + (rect.width()/2.0)
y = rect.y() + (rect.height()/2.0)
parent.setTransform(QtGui.QTransform().translate(x, y).rotate(img.rotation).translate(-x, -y))
# Creates the main tree item that will act as frame for the whole image
frame = QtGui.QGraphicsRectItem()
parent.setParentItem(frame)
mainRect = parent.mapToScene(mainRect).boundingRect()
mainRect.adjust(-img.margin_left, -img.margin_top, \
img.margin_right, img.margin_bottom)
# Fix negative coordinates, so main item always starts at 0,0
topleft = mainRect.topLeft()
_x = abs(topleft.x()) if topleft.x() < 0 else 0
_y = abs(topleft.y()) if topleft.y() < 0 else 0
if _x or _y:
parent.moveBy(_x, _y)
mainRect.adjust(_x, _y, _x, _y)
# Add extra components and adjust mainRect to them
add_legend(img, mainRect, frame)
add_title(img, mainRect, frame)
add_scale(img, mainRect, frame)
frame.setRect(mainRect)
# Draws a border around the tree
if not img.show_border:
frame.setPen(QtGui.QPen(QtCore.Qt.NoPen))
else:
frame.setPen(QtGui.QPen(QtGui.QColor("black")))
return frame, n2i, n2f
def render(root_node, img, hide_root=False):
'''main render function. hide_root option is used when render
trees as Faces
'''
mode = img.mode
orientation = img.orientation
arc_span = img.arc_span
layout_fn = img._layout_handler
parent = _TreeItem()
n2i = parent.n2i # node to items
n2f = parent.n2f # node to faces
parent.bg_layer = _EmptyItem(parent)
parent.tree_layer = _EmptyItem(parent)
parent.float_layer = _EmptyItem(parent)
parent.float_behind_layer = _EmptyItem(parent)
TREE_LAYERS = [parent.bg_layer, parent.tree_layer,
parent.float_layer, parent.float_behind_layer]
parent.bg_layer.setZValue(0)
parent.tree_layer.setZValue(2)
parent.float_behind_layer.setZValue(1)
parent.float_layer.setZValue(3)
# This could be used to handle aligned faces in internal
# nodes.
virtual_leaves = 0
if img.show_branch_length:
bl_face = faces.AttrFace("dist", fsize=8, ftype="Arial", fgcolor="black", formatter = "%0.3g")
if img.show_branch_support:
su_face = faces.AttrFace("support", fsize=8, ftype="Arial", fgcolor="darkred", formatter = "%0.3g")
if img.show_leaf_name:
na_face = faces.AttrFace("name", fsize=10, ftype="Arial", fgcolor="black")
for n in root_node.traverse(is_leaf_fn=_leaf):
set_style(n, layout_fn)
if img.show_branch_length:
faces.add_face_to_node(bl_face, n, 0, position="branch-top")
if not _leaf(n) and img.show_branch_support:
faces.add_face_to_node(su_face, n, 0, position="branch-bottom")
if _leaf(n) and n.name and img.show_leaf_name:
faces.add_face_to_node(na_face, n, 0, position="branch-right")
if _leaf(n):# or len(n.img_style["_faces"]["aligned"]):
virtual_leaves += 1
update_node_faces(n, n2f, img)
rot_step = float(arc_span) / virtual_leaves
#rot_step = float(arc_span) / len([n for n in root_node.traverse() if _leaf(n)])
# Calculate optimal branch length
if img._scale is not None:
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
elif img.scale is None:
# create items and calculate node dimensions skipping branch lengths
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
if mode == 'r':
if img.optimal_scale_level == "full":
scales = [(i.widths[1]/n.dist) for n,i in n2i.iteritems() if n.dist]
img._scale = max(scales) if scales else 0.0
else:
farthest, dist = root_node.get_farthest_leaf(topology_only=img.force_topology)
img._scale = img.tree_width / dist if dist else 0.0
update_branch_lengths(root_node, n2i, n2f, img)
else:
img._scale = crender.calculate_optimal_scale(root_node, n2i, rot_step, img)
#print "OPTIMAL circular scale", img._scale
update_branch_lengths(root_node, n2i, n2f, img)
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
else:
# create items and calculate node dimensions CONSIDERING branch lengths
img._scale = img.scale
init_items(root_node, parent, n2i, n2f, img, rot_step, hide_root)
#print "USING scale", img._scale
# Draw node content
for node in root_node.traverse(is_leaf_fn=_leaf):
if node is not root_node or not hide_root:
render_node_content(node, n2i, n2f, img)
# Adjust content to rect or circular layout
mainRect = parent.rect()
if mode == "c":
tree_radius = crender.render_circular(root_node, n2i, rot_step)
mainRect.adjust(-tree_radius, -tree_radius, tree_radius, tree_radius)
else:
iwidth = n2i[root_node].fullRegion.width()
iheight = n2i[root_node].fullRegion.height()
mainRect.adjust(0, 0, iwidth, iheight)
tree_radius = iwidth
# Add extra layers: aligned faces, floating faces, node
# backgrounds, etc. The order by which the following methods are
# called IS IMPORTANT
render_floatings(n2i, n2f, img, parent.float_layer, parent.float_behind_layer)
aligned_region_width = render_aligned_faces(img, mainRect, parent.tree_layer, n2i, n2f)
render_backgrounds(img, mainRect, parent.bg_layer, n2i, n2f)
# rotate if necessary in circular images. flip and adjust if mirror orientation.
adjust_faces_to_tranformations(img, mainRect, n2i, n2f, TREE_LAYERS)
# Rotate main image if necessary
parent.setRect(mainRect)
parent.setPen(QtGui.QPen(QtCore.Qt.NoPen))
if img.rotation:
rect = parent.boundingRect()
x = rect.x() + (rect.width()/2.0)
y = rect.y() + (rect.height()/2.0)
parent.setTransform(QtGui.QTransform().translate(x, y).rotate(img.rotation).translate(-x, -y))
# Creates the main tree item that will act as frame for the whole image
frame = QtGui.QGraphicsRectItem()
parent.setParentItem(frame)
mainRect = parent.mapToScene(mainRect).boundingRect()
mainRect.adjust(-img.margin_left, -img.margin_top, \
img.margin_right, img.margin_bottom)
# Fix negative coordinates, so main item always starts at 0,0
topleft = mainRect.topLeft()
_x = abs(topleft.x()) if topleft.x() < 0 else 0
_y = abs(topleft.y()) if topleft.y() < 0 else 0
if _x or _y:
parent.moveBy(_x, _y)
mainRect.adjust(_x, _y, _x, _y)
# Add extra components and adjust mainRect to them
add_legend(img, mainRect, frame)
add_title(img, mainRect, frame)
add_scale(img, mainRect, frame)
frame.setRect(mainRect)
# Draws a border around the tree
if not img.show_border:
frame.setPen(QtGui.QPen(QtCore.Qt.NoPen))
else:
frame.setPen(QtGui.QPen(QtGui.QColor("black")))
return frame, n2i, n2f
def basic(node):
if node.is_leaf():
#node.img_style["size"]=1
#node.img_style["shape"] = "circle"
faces.add_face_to_node(
faces.AttrFace("name", "Arial", 10, "#4f8f0f", None), node, 0)
def basic(node):
if node.is_leaf():
#node.img_style["size"]=1
#node.img_style["shape"] = "circle"
faces.add_face_to_node(faces.AttrFace("name","Arial",10,"#4f8f0f",None), node, 0 )
def render(root_node, img, hide_root=False):
mode = img.mode
orientation = img.orientation
if not img.scale and img.tree_width:
fnode, max_dist = root_node.get_farthest_leaf(topology_only=\
img.force_topology)
if max_dist>0:
img.scale = img.tree_width / max_dist
else:
img.scale = 1
scale = img.scale
arc_span = img.arc_span
last_rotation = img.arc_start
layout_fn = img._layout_handler
parent = _TreeItem()
n2i = parent.n2i # node to items
n2f = parent.n2f # node to faces
parent.bg_layer = _EmptyItem(parent)
parent.tree_layer = _EmptyItem(parent)
parent.float_layer = _EmptyItem(parent)
parent.float_behind_layer = _EmptyItem(parent)
TREE_LAYERS = [parent.bg_layer, parent.tree_layer, parent.float_layer]
parent.bg_layer.setZValue(0)
parent.tree_layer.setZValue(2)
parent.float_behind_layer.setZValue(1)
parent.float_layer.setZValue(3)
visited = set()
to_visit = []
to_visit.append(root_node)
# This could be used to handle aligned faces in internal
# nodes.
virtual_leaves = 0
if img.show_branch_length:
bl_face = faces.AttrFace("dist", fsize=8, ftype="Arial", fgcolor="black", formatter = "%0.3g")
if img.show_branch_support:
su_face = faces.AttrFace("support", fsize=8, ftype="Arial", fgcolor="darkred", formatter = "%0.3g")
if img.show_leaf_name:
na_face = faces.AttrFace("name", fsize=10, ftype="Arial", fgcolor="black")
for n in root_node.traverse():
set_style(n, layout_fn)
if img.show_branch_length:
faces.add_face_to_node(bl_face, n, 0, position="branch-top")
if not _leaf(n) and img.show_branch_support:
faces.add_face_to_node(su_face, n, 0, position="branch-bottom")
if _leaf(n) and img.show_leaf_name:
faces.add_face_to_node(na_face, n, 0, position="branch-right")
if _leaf(n):# or len(n.img_style["_faces"]["aligned"]):
virtual_leaves += 1
rot_step = float(arc_span) / virtual_leaves
#rot_step = float(arc_span) / len([n for n in root_node.traverse() if _leaf(n)])
# ::: Precalculate values :::
depth = 1
while to_visit:
node = to_visit[-1]
finished = True
if node not in n2i:
# Set style according to layout function
item = n2i[node] = _NodeItem(node, parent.tree_layer)
item.setZValue(depth)
depth += 1
if node is root_node and hide_root:
pass
else:
set_node_size(node, n2i, n2f, img)
if not _leaf(node):
# visit children starting from left most to right
# most. Very important!! check all children[-1] and
# children[0]
for c in reversed(node.children):
if c not in visited:
to_visit.append(c)
finished = False
# :: pre-order code here ::
if not finished:
continue
else:
to_visit.pop(-1)
visited.add(node)
# :: Post-order visits. Leaves are already visited here ::
if mode == "c":
if _leaf(node):
crender.init_circular_leaf_item(node, n2i, n2f, last_rotation, rot_step)
last_rotation += rot_step
else:
crender.init_circular_node_item(node, n2i, n2f)
elif mode == "r":
if _leaf(node):
rrender.init_rect_leaf_item(node, n2i, n2f)
else:
rrender.init_rect_node_item(node, n2i, n2f)
if node is not root_node or not hide_root:
render_node_content(node, n2i, n2f, img)
mainRect = parent.rect()
if mode == "c":
tree_radius = crender.render_circular(root_node, n2i, rot_step)
mainRect.adjust( -tree_radius, -tree_radius, tree_radius, tree_radius)
else:
iwidth = n2i[root_node].fullRegion.width()
iheight = n2i[root_node].fullRegion.height()
mainRect.adjust(0, 0, iwidth, iheight)
tree_radius = iwidth
# The order by which the following methods IS IMPORTANT
render_floatings(n2i, n2f, img, parent.float_layer, parent.float_behind_layer)
aligned_region_width = render_aligned_faces(img, mainRect, parent.tree_layer, n2i, n2f)
render_backgrounds(img, mainRect, parent.bg_layer, n2i, n2f)
adjust_faces_to_tranformations(img, mainRect, n2i, n2f, TREE_LAYERS)
parent.setRect(mainRect)
parent.setPen(QtGui.QPen(QtCore.Qt.NoPen))
if img.rotation:
rect = parent.boundingRect()
x = rect.x() + rect.width()/2
y = rect.y() + rect.height()/2
parent.setTransform(QtGui.QTransform().translate(x, y).rotate(img.rotation).translate(-x, -y))
frame = QtGui.QGraphicsRectItem()
parent.setParentItem(frame)
mainRect = parent.mapToScene(mainRect).boundingRect()
mainRect.adjust(-img.margin_left, -img.margin_top, \
img.margin_right, img.margin_bottom)
add_legend(img, mainRect, frame)
add_title(img, mainRect, frame)
add_scale(img, mainRect, frame)
frame.setRect(mainRect)
# Draws a border around the tree
if not img.show_border:
frame.setPen(QtGui.QPen(QtCore.Qt.NoPen))
else:
frame.setPen(QtGui.QPen(QtGui.QColor("black")))
return frame, n2i, n2f
