summaryrefslogtreecommitdiff
path: root/python/src/simple_example.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/src/simple_example.py')
-rw-r--r--python/src/simple_example.py42
1 files changed, 19 insertions, 23 deletions
diff --git a/python/src/simple_example.py b/python/src/simple_example.py
index 82cf6be..955e213 100644
--- a/python/src/simple_example.py
+++ b/python/src/simple_example.py
@@ -21,39 +21,35 @@ if __name__ == "__main__":
import phat
- # set the dimension of the cell that each column represents:
- dimensions = [0, 0, 1, 0, 1, 1, 2]
-
# define a boundary matrix with the chosen internal representation
boundary_matrix = phat.boundary_matrix(representation = phat.representations.vector_vector)
- # set the respective columns -- the columns entries have to be sorted
- boundary_matrix.set_dims(dimensions)
- boundary_matrix.set_col(0, [])
- boundary_matrix.set_col(1, [])
- boundary_matrix.set_col(2, [0,1])
- boundary_matrix.set_col(3, [])
- boundary_matrix.set_col(4, [1,3])
- boundary_matrix.set_col(5, [0,3])
- boundary_matrix.set_col(6, [2,4,5])
+ # set the respective columns -- (dimension, boundary) pairs
+ boundary_matrix.columns = [ (0, []),
+ (0, []),
+ (1, [0,1]),
+ (0, []),
+ (1, [1,3]),
+ (1, [0,3]),
+ (2, [2,4,5])]
+
+ # or equivalently, boundary_matrix = phat.boundary_matrix(representation = ..., columns = ...)
+ # would combine the creation of the matrix and the assignment of the columns
# print some information of the boundary matrix:
- print()
- print("The boundary matrix has %d columns:" % boundary_matrix.get_num_cols())
- for col_idx in range(boundary_matrix.get_num_cols()):
- s = "Column %d represents a cell of dimension %d." % (col_idx, boundary_matrix.get_dim(col_idx))
- if (not boundary_matrix.is_empty(col_idx)):
- s = s + " Its boundary consists of the cells " + " ".join([str(c) for c in boundary_matrix.get_col(col_idx)])
+ print("\nThe boundary matrix has %d columns:" % len(boundary_matrix.columns))
+ for col in boundary_matrix.columns:
+ s = "Column %d represents a cell of dimension %d." % (col.index, col.dimension)
+ if (col.boundary):
+ s = s + " Its boundary consists of the cells " + " ".join([str(c) for c in col.boundary])
print(s)
- print("Overall, the boundary matrix has %d entries." % boundary_matrix.get_num_entries())
+ print("Overall, the boundary matrix has %d entries." % len(boundary_matrix))
- pairs = phat.compute_persistence_pairs(boundary_matrix)
+ pairs = boundary_matrix.compute_persistence_pairs()
pairs.sort()
- print()
-
- print("There are %d persistence pairs: " % len(pairs))
+ print("\nThere are %d persistence pairs: " % len(pairs))
for pair in pairs:
print("Birth: %d, Death: %d" % pair)