diff options
Diffstat (limited to 'src/cython/example/alpha_complex_from_file_example.py')
-rwxr-xr-x | src/cython/example/alpha_complex_from_file_example.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/cython/example/alpha_complex_from_file_example.py b/src/cython/example/alpha_complex_from_file_example.py index beb5d101..81c4c0b7 100755 --- a/src/cython/example/alpha_complex_from_file_example.py +++ b/src/cython/example/alpha_complex_from_file_example.py @@ -42,6 +42,8 @@ parser = argparse.ArgumentParser(description='AlphaComplex creation from ' 'points from the given file. File format ' 'is X1, X2, ..., Xn') parser.add_argument('file', type=argparse.FileType('r')) +parser.add_argument('--no-diagram', default=False, action='store_true' , help='Flag for not to display the diagrams') + args = parser.parse_args() points = pandas.read_csv(args.file, header=None) @@ -57,6 +59,6 @@ diag = alpha_complex.persistence(homology_coeff_field=2, min_persistence=0.1) print("betti_numbers()=") print(alpha_complex.betti_numbers()) -gudhi.diagram_persistence(diag) - -gudhi.barcode_persistence(diag) +if args.no_diagram == False: + gudhi.diagram_persistence(diag) + gudhi.barcode_persistence(diag) |