Example #1
0
            filtration_steps=args.filtration_steps)

    print "Decomp. took {}s.".format(time.time() - t0)
    print "Number of loops:", dual.number_of_nodes()
    print "Number of tree nodes:", tree.number_of_nodes()

    if args.save != "":
        print "Saving file."
        
        SAVE_FORMAT_VERSION = 5
        sav = {'version':SAVE_FORMAT_VERSION, \
                'leaf':leaf, 'tree':tree, 'dual':dual, \
                'filtration':filtr, 'pruned':pruned, \
                'removed-edges':removed_edges}
        
        storage.save(sav, args.save)
    
    print "Done."

    if args.plot:
        plt.figure()
        plot.draw_leaf(leaf, "Input leaf data")
        plt.figure()
        plot.draw_leaf(pruned, "Pruned leaf data and dual graph")
        plot.draw_dual(dual)
        plt.figure()
        plot.draw_tree(tree)
        plt.figure()
        plot.draw_filtration(filtr)
        plt.show()
Example #2
0
        print "Plotting/saving plots."
        fname = ""
        fname_no_ext = ""
        fext = ".png"
        if args.save_plots != "":
            fname, fext = os.path.splitext(args.save_plots)
            fname_no_ext = fname + "_no_ext"

        print fname, fext

        plt.figure()
        print "Drawing leaf."
        plot.draw_leaf(leaf, fname=fname, fext=fext, dpi=600)
        plt.figure()
        print "Drawing hierarchical trees."
        plot.draw_tree(marked_tree, pos=tree_pos, fname=fname, fext=fext)
        
        plt.figure()
        
        if args.tree_heights:
            pos = tree_pos_h
        else:
            pos = tree_pos

        plot.draw_tree(marked_tree_no_ext, pos=pos, 
                fname=fname_no_ext, fext=fext)

        plt.figure()
        print "Drawing filtration."
        plot.draw_filtration(filt, fname=fname, fext=fext, dpi=600)