Add Gudhi namespace for reader_utils

Add confidence band in persistence_graphical_tools.py
Persistence_diagram returns a plot that is no more showed.


git-svn-id: svn+ssh://scm.gforge.inria.fr/svnroot/gudhi/branches/persistence_diagram_improvement@2582 636b058d-ea47-450e-bf9e-a15bfbe3eedb

Former-commit-id: 4125f4f525057e89c8b0d5d164ea0b9d1df1bd72

3 files changed, 25 insertions, 12 deletions

diff --git a/src/cython/cython/persistence_graphical_tools.py b/src/cython/cython/persistence_graphical_tools.py
index a984633e..70ff6001 100755
--- a/src/cython/cython/persistence_graphical_tools.py
+++ b/src/cython/cython/persistence_graphical_tools.py
@@ -5,7 +5,7 @@ import numpy as np
    (Geometric Understanding in Higher Dimensions) is a generic C++
    library for computational topology.
 
-   Author(s):       Vincent Rouvreau
+   Author(s):       Vincent Rouvreau, Bertrand Michel
 
    Copyright (C) 2016 INRIA
 
@@ -27,11 +27,13 @@ __author__ = "Vincent Rouvreau"
 __copyright__ = "Copyright (C) 2016 INRIA"
 __license__ = "GPL v3"
 
-def __min_birth_max_death(persistence):
+def __min_birth_max_death(persistence, band_boot=0.):
     """This function returns (min_birth, max_death) from the persistence.
 
     :param persistence: The persistence to plot.
     :type persistence: list of tuples(dimension, tuple(birth, death)).
+    :param band_boot: bootstrap band
+    :type band_boot: float.
     :returns: (float, float) -- (min_birth, max_death).
     """
     # Look for minimum birth date and maximum death date for plot optimisation
@@ -45,6 +47,8 @@ def __min_birth_max_death(persistence):
             max_death = float(interval[1][0])
         if float(interval[1][0]) < min_birth:
             min_birth = float(interval[1][0])
+    if band_boot > 0.:
+        max_death += band_boot
     return (min_birth, max_death)
 
 """
@@ -108,16 +112,18 @@ def plot_persistence_barcode(persistence, alpha=0.6):
     plt.axis([axis_start, infinity, 0, ind])
     plt.show()
 
-def plot_persistence_diagram(persistence, alpha=0.6):
-    """This function plots the persistence diagram.
+def plot_persistence_diagram(persistence, alpha=0.6, band_boot=0.):
+    """This function plots the persistence diagram with confidence band.
 
     :param persistence: The persistence to plot.
     :type persistence: list of tuples(dimension, tuple(birth, death)).
     :param alpha: alpha value in [0.0, 1.0] for points and horizontal infinity line (default is 0.6).
     :type alpha: float.
-    :returns: plot -- An diagram plot of persistence.
+    :param band_boot: bootstrap band
+    :type band_boot: float.
+    :returns: plot -- A diagram plot of persistence.
     """
-    (min_birth, max_death) = __min_birth_max_death(persistence)
+    (min_birth, max_death) = __min_birth_max_death(persistence, band_boot)
     ind = 0
     delta = ((max_death - min_birth) / 10.0)
     # Replace infinity values with max_death + delta for diagram to be more
@@ -131,6 +137,9 @@ def plot_persistence_diagram(persistence, alpha=0.6):
     plt.plot(x, x, color='k', linewidth=1.0)
     plt.plot(x, [infinity] * len(x), linewidth=1.0, color='k', alpha=alpha)
     plt.text(axis_start, infinity, r'$\infty$', color='k', alpha=alpha)
+    # bootstrap band
+    if band_boot > 0.:
+        plt.fill_between(x, x, x+band_boot, alpha=alpha, facecolor='red')
 
     # Draw points in loop
     for interval in reversed(persistence):
@@ -149,4 +158,4 @@ def plot_persistence_diagram(persistence, alpha=0.6):
     plt.ylabel('Death')
     # Ends plot on infinity value and starts a little bit before min_birth
     plt.axis([axis_start, infinity, axis_start, infinity + delta])
-    plt.show()
+    return plt
diff --git a/src/cython/doc/persistence_graphical_tools_user.rst b/src/cython/doc/persistence_graphical_tools_user.rst
index cae18323..bc731f12 100644
--- a/src/cython/doc/persistence_graphical_tools_user.rst
+++ b/src/cython/doc/persistence_graphical_tools_user.rst
@@ -51,16 +51,18 @@ This function can display the persistence result as a diagram:
 
     import gudhi
     
-    rips_complex = gudhi.RipsComplex(off_file='tore3D_300.off', max_edge_length=2.0)
+    rips_complex = gudhi.RipsComplex(off_file='tore3D_1307.off', max_edge_length=0.2)
     simplex_tree = rips_complex.create_simplex_tree(max_dimension=3)
     diag = simplex_tree.persistence()
-    gudhi.plot_persistence_diagram(diag)
+    pplot = gudhi.plot_persistence_diagram(diag, band_boot=0.13)
+    pplot.show()
 
 .. plot::
 
     import gudhi
 
-    rips_complex = gudhi.RipsComplex(off_file='tore3D_300.off', max_edge_length=2.0)
+    rips_complex = gudhi.RipsComplex(off_file='tore3D_1307.off', max_edge_length=0.2)
     simplex_tree = rips_complex.create_simplex_tree(max_dimension=3)
     diag = simplex_tree.persistence()
-    gudhi.plot_persistence_diagram(diag)
+    pplot = gudhi.plot_persistence_diagram(diag, band_boot=0.13)
+    pplot.show()
diff --git a/src/cython/example/rips_complex_diagram_persistence_from_off_file_example.py b/src/cython/example/rips_complex_diagram_persistence_from_off_file_example.py
index 4c21b98e..5951eedf 100755
--- a/src/cython/example/rips_complex_diagram_persistence_from_off_file_example.py
+++ b/src/cython/example/rips_complex_diagram_persistence_from_off_file_example.py
@@ -39,6 +39,7 @@ parser = argparse.ArgumentParser(description='RipsComplex creation from '
 parser.add_argument("-f", "--file", type=str, required=True)
 parser.add_argument("-e", "--max_edge_length", type=float, default=0.5)
 parser.add_argument("-d", "--max_dimension", type=int, default=1)
+parser.add_argument("-b", "--band_boot", type=float, default=0.)
 parser.add_argument('--no-diagram', default=False, action='store_true' , help='Flag for not to display the diagrams')
 
 args = parser.parse_args()
@@ -64,7 +65,8 @@ with open(args.file, 'r') as f:
         print(simplex_tree.betti_numbers())
     
         if args.no_diagram == False:
-            gudhi.plot_persistence_diagram(diag)
+            pplot = gudhi.plot_persistence_diagram(diag, band_boot=args.band_boot)
+            pplot.show()
     else:
         print(args.file, "is not a valid OFF file")