diff options
Diffstat (limited to 'src/cython/example/voronoi_graph_induced_complex.py')
| -rwxr-xr-x | src/cython/example/voronoi_graph_induced_complex.py | 44 |
1 files changed, 28 insertions, 16 deletions
diff --git a/src/cython/example/voronoi_graph_induced_complex.py b/src/cython/example/voronoi_graph_induced_complex.py index 1e62fb60..38be6c92 100755 --- a/src/cython/example/voronoi_graph_induced_complex.py +++ b/src/cython/example/voronoi_graph_induced_complex.py @@ -17,24 +17,31 @@ __author__ = "Vincent Rouvreau" __copyright__ = "Copyright (C) 2018 Inria" __license__ = "MIT" -parser = argparse.ArgumentParser(description='Voronoi GIC ' - 'from points read in a OFF file.', - epilog='Example: ' - 'example/voronoi_graph_induced_complex.py ' - '-f ../data/points/human.off -n 700 -v' - '- Constructs the Voronoi GIC with the ' - 'points from the given OFF file.') +parser = argparse.ArgumentParser( + description="Voronoi GIC " "from points read in a OFF file.", + epilog="Example: " + "example/voronoi_graph_induced_complex.py " + "-f ../data/points/human.off -n 700 -v" + "- Constructs the Voronoi GIC with the " + "points from the given OFF file.", +) parser.add_argument("-f", "--file", type=str, required=True) parser.add_argument("-n", "--subsample-nb-points", type=int, default=100) -parser.add_argument("-v", "--verbose", default=False, action='store_true' , help='Flag for program verbosity') +parser.add_argument( + "-v", + "--verbose", + default=False, + action="store_true", + help="Flag for program verbosity", +) args = parser.parse_args() nerve_complex = gudhi.CoverComplex() nerve_complex.set_verbose(args.verbose) -if (nerve_complex.read_point_cloud(args.file)): - nerve_complex.set_type('GIC') +if nerve_complex.read_point_cloud(args.file): + nerve_complex.set_type("GIC") nerve_complex.set_color_from_coordinate() nerve_complex.set_graph_from_OFF() nerve_complex.set_cover_from_Voronoi(args.subsample_nb_points) @@ -42,12 +49,17 @@ if (nerve_complex.read_point_cloud(args.file)): nerve_complex.plot_off() simplex_tree = nerve_complex.create_simplex_tree() nerve_complex.compute_PD() - if (args.verbose): - print('Iterator on graph induced complex simplices') - result_str = 'Graph induced complex is of dimension ' + \ - repr(simplex_tree.dimension()) + ' - ' + \ - repr(simplex_tree.num_simplices()) + ' simplices - ' + \ - repr(simplex_tree.num_vertices()) + ' vertices.' + if args.verbose: + print("Iterator on graph induced complex simplices") + result_str = ( + "Graph induced complex is of dimension " + + repr(simplex_tree.dimension()) + + " - " + + repr(simplex_tree.num_simplices()) + + " simplices - " + + repr(simplex_tree.num_vertices()) + + " vertices." + ) print(result_str) for filtered_value in simplex_tree.get_filtration(): print(filtered_value[0]) |