import numpy as np
import gudhi
import matplotlib.pyplot as plt

# 1. Matriz de Distancia
D = np.array([
    [0., 1., 1., 2., 3., 3.],
    [1., 0., 2., 1., 3., 3.],
    [1., 2., 0., 1., 3., 3.],
    [2., 1., 1., 0., 3., 3.],
    [3., 3., 3., 3., 0., 2.],
    [3., 3., 3., 3., 2., 0.]
])



# 2. Construcción del complejo de Vietoris-Rips con GUDHI
rips_complex = gudhi.RipsComplex(distance_matrix=D, max_edge_length=3.0)
simplex_tree = rips_complex.create_simplex_tree(max_dimension=2)

# 3. Cálculo de la homología persistente
diag=simplex_tree.persistence()

# 4. Graficamos el diagrama de persistencia

gudhi.plot_persistence_diagram(diag)
plt.title("Diagrama de Persistencia (H0 y H1)")
plt.show()