def write_pdf_bootstrap_tree(tree, output_f, hits_dict): def f(node): if not node.Name: return 'black' tip_id=node.Name.split('/')[0] try: if hits_dict[tip_id] < .25: return 'blue' elif hits_dict[tip_id] < .5: return 'green' elif hits_dict[tip_id] < .75: return 'yellow' elif hits_dict[tip_id] <= 1.1: return 'red' return 'black' except: return 'black' t=SquareDendrogram(tree) # Make output size proportional to the tree size. width=8*len(tree.tips()) height=8*len(tree.tips()) if width<700: width=700 if height<700: height=700 t.drawToPDF(output_f, width, height, edge_color_callback=f)