diff options
Diffstat (limited to 'src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py')
-rwxr-xr-x | src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py | 25 |
1 files changed, 12 insertions, 13 deletions
diff --git a/src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py b/src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py index 727af4fa..fe03be31 100755 --- a/src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py +++ b/src/python/example/alpha_complex_diagram_persistence_from_off_file_example.py @@ -3,7 +3,6 @@ import argparse import errno import os -import matplotlib.pyplot as plot import gudhi """ This file is part of the Gudhi Library - https://gudhi.inria.fr/ - @@ -26,12 +25,12 @@ parser = argparse.ArgumentParser( description="AlphaComplex creation from " "points read in a OFF file.", epilog="Example: " "example/alpha_complex_diagram_persistence_from_off_file_example.py " - "-f ../data/points/tore3D_300.off -a 0.6" + "-f ../data/points/tore3D_300.off" "- Constructs a alpha complex with the " "points from the given OFF file.", ) parser.add_argument("-f", "--file", type=str, required=True) -parser.add_argument("-a", "--max_alpha_square", type=float, default=0.5) +parser.add_argument("-a", "--max_alpha_square", type=float, required=False) parser.add_argument("-b", "--band", type=float, default=0.0) parser.add_argument( "--no-diagram", @@ -48,23 +47,23 @@ with open(args.file, "r") as f: print("##############################################################") print("AlphaComplex creation from points read in a OFF file") - message = "AlphaComplex with max_edge_length=" + repr(args.max_alpha_square) - print(message) - alpha_complex = gudhi.AlphaComplex(off_file=args.file) - simplex_tree = alpha_complex.create_simplex_tree( - max_alpha_square=args.max_alpha_square - ) + if args.max_alpha_square is not None: + print("with max_edge_length=", args.max_alpha_square) + simplex_tree = alpha_complex.create_simplex_tree( + max_alpha_square=args.max_alpha_square + ) + else: + simplex_tree = alpha_complex.create_simplex_tree() - message = "Number of simplices=" + repr(simplex_tree.num_simplices()) - print(message) + print("Number of simplices=", simplex_tree.num_simplices()) diag = simplex_tree.persistence() - print("betti_numbers()=") - print(simplex_tree.betti_numbers()) + print("betti_numbers()=", simplex_tree.betti_numbers()) if args.no_diagram == False: + import matplotlib.pyplot as plot gudhi.plot_persistence_diagram(diag, band=args.band) plot.show() else: |