multipers 2.3.3b6__cp311-cp311-macosx_11_0_arm64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of multipers might be problematic. Click here for more details.

Files changed (183) hide show
  1. multipers/.dylibs/libc++.1.0.dylib +0 -0
  2. multipers/.dylibs/libtbb.12.16.dylib +0 -0
  3. multipers/__init__.py +33 -0
  4. multipers/_signed_measure_meta.py +453 -0
  5. multipers/_slicer_meta.py +211 -0
  6. multipers/array_api/__init__.py +45 -0
  7. multipers/array_api/numpy.py +41 -0
  8. multipers/array_api/torch.py +58 -0
  9. multipers/data/MOL2.py +458 -0
  10. multipers/data/UCR.py +18 -0
  11. multipers/data/__init__.py +1 -0
  12. multipers/data/graphs.py +466 -0
  13. multipers/data/immuno_regions.py +27 -0
  14. multipers/data/minimal_presentation_to_st_bf.py +0 -0
  15. multipers/data/pytorch2simplextree.py +91 -0
  16. multipers/data/shape3d.py +101 -0
  17. multipers/data/synthetic.py +113 -0
  18. multipers/distances.py +202 -0
  19. multipers/filtration_conversions.pxd +229 -0
  20. multipers/filtration_conversions.pxd.tp +84 -0
  21. multipers/filtrations/__init__.py +18 -0
  22. multipers/filtrations/density.py +574 -0
  23. multipers/filtrations/filtrations.py +361 -0
  24. multipers/filtrations.pxd +224 -0
  25. multipers/function_rips.cpython-311-darwin.so +0 -0
  26. multipers/function_rips.pyx +105 -0
  27. multipers/grids.cpython-311-darwin.so +0 -0
  28. multipers/grids.pyx +433 -0
  29. multipers/gudhi/Persistence_slices_interface.h +132 -0
  30. multipers/gudhi/Simplex_tree_interface.h +239 -0
  31. multipers/gudhi/Simplex_tree_multi_interface.h +551 -0
  32. multipers/gudhi/cubical_to_boundary.h +59 -0
  33. multipers/gudhi/gudhi/Bitmap_cubical_complex.h +450 -0
  34. multipers/gudhi/gudhi/Bitmap_cubical_complex_base.h +1070 -0
  35. multipers/gudhi/gudhi/Bitmap_cubical_complex_periodic_boundary_conditions_base.h +579 -0
  36. multipers/gudhi/gudhi/Debug_utils.h +45 -0
  37. multipers/gudhi/gudhi/Fields/Multi_field.h +484 -0
  38. multipers/gudhi/gudhi/Fields/Multi_field_operators.h +455 -0
  39. multipers/gudhi/gudhi/Fields/Multi_field_shared.h +450 -0
  40. multipers/gudhi/gudhi/Fields/Multi_field_small.h +531 -0
  41. multipers/gudhi/gudhi/Fields/Multi_field_small_operators.h +507 -0
  42. multipers/gudhi/gudhi/Fields/Multi_field_small_shared.h +531 -0
  43. multipers/gudhi/gudhi/Fields/Z2_field.h +355 -0
  44. multipers/gudhi/gudhi/Fields/Z2_field_operators.h +376 -0
  45. multipers/gudhi/gudhi/Fields/Zp_field.h +420 -0
  46. multipers/gudhi/gudhi/Fields/Zp_field_operators.h +400 -0
  47. multipers/gudhi/gudhi/Fields/Zp_field_shared.h +418 -0
  48. multipers/gudhi/gudhi/Flag_complex_edge_collapser.h +337 -0
  49. multipers/gudhi/gudhi/Matrix.h +2107 -0
  50. multipers/gudhi/gudhi/Multi_critical_filtration.h +1038 -0
  51. multipers/gudhi/gudhi/Multi_persistence/Box.h +174 -0
  52. multipers/gudhi/gudhi/Multi_persistence/Line.h +282 -0
  53. multipers/gudhi/gudhi/Off_reader.h +173 -0
  54. multipers/gudhi/gudhi/One_critical_filtration.h +1441 -0
  55. multipers/gudhi/gudhi/Persistence_matrix/Base_matrix.h +769 -0
  56. multipers/gudhi/gudhi/Persistence_matrix/Base_matrix_with_column_compression.h +686 -0
  57. multipers/gudhi/gudhi/Persistence_matrix/Boundary_matrix.h +842 -0
  58. multipers/gudhi/gudhi/Persistence_matrix/Chain_matrix.h +1350 -0
  59. multipers/gudhi/gudhi/Persistence_matrix/Id_to_index_overlay.h +1105 -0
  60. multipers/gudhi/gudhi/Persistence_matrix/Position_to_index_overlay.h +859 -0
  61. multipers/gudhi/gudhi/Persistence_matrix/RU_matrix.h +910 -0
  62. multipers/gudhi/gudhi/Persistence_matrix/allocators/entry_constructors.h +139 -0
  63. multipers/gudhi/gudhi/Persistence_matrix/base_pairing.h +230 -0
  64. multipers/gudhi/gudhi/Persistence_matrix/base_swap.h +211 -0
  65. multipers/gudhi/gudhi/Persistence_matrix/boundary_cell_position_to_id_mapper.h +60 -0
  66. multipers/gudhi/gudhi/Persistence_matrix/boundary_face_position_to_id_mapper.h +60 -0
  67. multipers/gudhi/gudhi/Persistence_matrix/chain_pairing.h +136 -0
  68. multipers/gudhi/gudhi/Persistence_matrix/chain_rep_cycles.h +190 -0
  69. multipers/gudhi/gudhi/Persistence_matrix/chain_vine_swap.h +616 -0
  70. multipers/gudhi/gudhi/Persistence_matrix/columns/chain_column_extra_properties.h +150 -0
  71. multipers/gudhi/gudhi/Persistence_matrix/columns/column_dimension_holder.h +106 -0
  72. multipers/gudhi/gudhi/Persistence_matrix/columns/column_utilities.h +219 -0
  73. multipers/gudhi/gudhi/Persistence_matrix/columns/entry_types.h +327 -0
  74. multipers/gudhi/gudhi/Persistence_matrix/columns/heap_column.h +1140 -0
  75. multipers/gudhi/gudhi/Persistence_matrix/columns/intrusive_list_column.h +934 -0
  76. multipers/gudhi/gudhi/Persistence_matrix/columns/intrusive_set_column.h +934 -0
  77. multipers/gudhi/gudhi/Persistence_matrix/columns/list_column.h +980 -0
  78. multipers/gudhi/gudhi/Persistence_matrix/columns/naive_vector_column.h +1092 -0
  79. multipers/gudhi/gudhi/Persistence_matrix/columns/row_access.h +192 -0
  80. multipers/gudhi/gudhi/Persistence_matrix/columns/set_column.h +921 -0
  81. multipers/gudhi/gudhi/Persistence_matrix/columns/small_vector_column.h +1093 -0
  82. multipers/gudhi/gudhi/Persistence_matrix/columns/unordered_set_column.h +1012 -0
  83. multipers/gudhi/gudhi/Persistence_matrix/columns/vector_column.h +1244 -0
  84. multipers/gudhi/gudhi/Persistence_matrix/matrix_dimension_holders.h +186 -0
  85. multipers/gudhi/gudhi/Persistence_matrix/matrix_row_access.h +164 -0
  86. multipers/gudhi/gudhi/Persistence_matrix/ru_pairing.h +156 -0
  87. multipers/gudhi/gudhi/Persistence_matrix/ru_rep_cycles.h +376 -0
  88. multipers/gudhi/gudhi/Persistence_matrix/ru_vine_swap.h +540 -0
  89. multipers/gudhi/gudhi/Persistent_cohomology/Field_Zp.h +118 -0
  90. multipers/gudhi/gudhi/Persistent_cohomology/Multi_field.h +173 -0
  91. multipers/gudhi/gudhi/Persistent_cohomology/Persistent_cohomology_column.h +128 -0
  92. multipers/gudhi/gudhi/Persistent_cohomology.h +745 -0
  93. multipers/gudhi/gudhi/Points_off_io.h +171 -0
  94. multipers/gudhi/gudhi/Simple_object_pool.h +69 -0
  95. multipers/gudhi/gudhi/Simplex_tree/Simplex_tree_iterators.h +463 -0
  96. multipers/gudhi/gudhi/Simplex_tree/Simplex_tree_node_explicit_storage.h +83 -0
  97. multipers/gudhi/gudhi/Simplex_tree/Simplex_tree_siblings.h +106 -0
  98. multipers/gudhi/gudhi/Simplex_tree/Simplex_tree_star_simplex_iterators.h +277 -0
  99. multipers/gudhi/gudhi/Simplex_tree/hooks_simplex_base.h +62 -0
  100. multipers/gudhi/gudhi/Simplex_tree/indexing_tag.h +27 -0
  101. multipers/gudhi/gudhi/Simplex_tree/serialization_utils.h +62 -0
  102. multipers/gudhi/gudhi/Simplex_tree/simplex_tree_options.h +157 -0
  103. multipers/gudhi/gudhi/Simplex_tree.h +2794 -0
  104. multipers/gudhi/gudhi/Simplex_tree_multi.h +152 -0
  105. multipers/gudhi/gudhi/distance_functions.h +62 -0
  106. multipers/gudhi/gudhi/graph_simplicial_complex.h +104 -0
  107. multipers/gudhi/gudhi/persistence_interval.h +253 -0
  108. multipers/gudhi/gudhi/persistence_matrix_options.h +170 -0
  109. multipers/gudhi/gudhi/reader_utils.h +367 -0
  110. multipers/gudhi/mma_interface_coh.h +256 -0
  111. multipers/gudhi/mma_interface_h0.h +223 -0
  112. multipers/gudhi/mma_interface_matrix.h +293 -0
  113. multipers/gudhi/naive_merge_tree.h +536 -0
  114. multipers/gudhi/scc_io.h +310 -0
  115. multipers/gudhi/truc.h +1403 -0
  116. multipers/io.cpython-311-darwin.so +0 -0
  117. multipers/io.pyx +644 -0
  118. multipers/ml/__init__.py +0 -0
  119. multipers/ml/accuracies.py +90 -0
  120. multipers/ml/invariants_with_persistable.py +79 -0
  121. multipers/ml/kernels.py +176 -0
  122. multipers/ml/mma.py +713 -0
  123. multipers/ml/one.py +472 -0
  124. multipers/ml/point_clouds.py +352 -0
  125. multipers/ml/signed_measures.py +1589 -0
  126. multipers/ml/sliced_wasserstein.py +461 -0
  127. multipers/ml/tools.py +113 -0
  128. multipers/mma_structures.cpython-311-darwin.so +0 -0
  129. multipers/mma_structures.pxd +128 -0
  130. multipers/mma_structures.pyx +2786 -0
  131. multipers/mma_structures.pyx.tp +1094 -0
  132. multipers/multi_parameter_rank_invariant/diff_helpers.h +84 -0
  133. multipers/multi_parameter_rank_invariant/euler_characteristic.h +97 -0
  134. multipers/multi_parameter_rank_invariant/function_rips.h +322 -0
  135. multipers/multi_parameter_rank_invariant/hilbert_function.h +769 -0
  136. multipers/multi_parameter_rank_invariant/persistence_slices.h +148 -0
  137. multipers/multi_parameter_rank_invariant/rank_invariant.h +369 -0
  138. multipers/multiparameter_edge_collapse.py +41 -0
  139. multipers/multiparameter_module_approximation/approximation.h +2330 -0
  140. multipers/multiparameter_module_approximation/combinatory.h +129 -0
  141. multipers/multiparameter_module_approximation/debug.h +107 -0
  142. multipers/multiparameter_module_approximation/euler_curves.h +0 -0
  143. multipers/multiparameter_module_approximation/format_python-cpp.h +286 -0
  144. multipers/multiparameter_module_approximation/heap_column.h +238 -0
  145. multipers/multiparameter_module_approximation/images.h +79 -0
  146. multipers/multiparameter_module_approximation/list_column.h +174 -0
  147. multipers/multiparameter_module_approximation/list_column_2.h +232 -0
  148. multipers/multiparameter_module_approximation/ru_matrix.h +347 -0
  149. multipers/multiparameter_module_approximation/set_column.h +135 -0
  150. multipers/multiparameter_module_approximation/structure_higher_dim_barcode.h +36 -0
  151. multipers/multiparameter_module_approximation/unordered_set_column.h +166 -0
  152. multipers/multiparameter_module_approximation/utilities.h +403 -0
  153. multipers/multiparameter_module_approximation/vector_column.h +223 -0
  154. multipers/multiparameter_module_approximation/vector_matrix.h +331 -0
  155. multipers/multiparameter_module_approximation/vineyards.h +464 -0
  156. multipers/multiparameter_module_approximation/vineyards_trajectories.h +649 -0
  157. multipers/multiparameter_module_approximation.cpython-311-darwin.so +0 -0
  158. multipers/multiparameter_module_approximation.pyx +235 -0
  159. multipers/pickle.py +90 -0
  160. multipers/plots.py +456 -0
  161. multipers/point_measure.cpython-311-darwin.so +0 -0
  162. multipers/point_measure.pyx +395 -0
  163. multipers/simplex_tree_multi.cpython-311-darwin.so +0 -0
  164. multipers/simplex_tree_multi.pxd +134 -0
  165. multipers/simplex_tree_multi.pyx +10840 -0
  166. multipers/simplex_tree_multi.pyx.tp +2009 -0
  167. multipers/slicer.cpython-311-darwin.so +0 -0
  168. multipers/slicer.pxd +3034 -0
  169. multipers/slicer.pxd.tp +234 -0
  170. multipers/slicer.pyx +20481 -0
  171. multipers/slicer.pyx.tp +1088 -0
  172. multipers/tensor/tensor.h +672 -0
  173. multipers/tensor.pxd +13 -0
  174. multipers/test.pyx +44 -0
  175. multipers/tests/__init__.py +62 -0
  176. multipers/torch/__init__.py +1 -0
  177. multipers/torch/diff_grids.py +240 -0
  178. multipers/torch/rips_density.py +310 -0
  179. multipers-2.3.3b6.dist-info/METADATA +128 -0
  180. multipers-2.3.3b6.dist-info/RECORD +183 -0
  181. multipers-2.3.3b6.dist-info/WHEEL +6 -0
  182. multipers-2.3.3b6.dist-info/licenses/LICENSE +21 -0
  183. multipers-2.3.3b6.dist-info/top_level.txt +1 -0
@@ -0,0 +1,148 @@
1
+ #pragma once
2
+
3
+ #include "gudhi/Flag_complex_edge_collapser.h"
4
+ #include <gudhi/Persistent_cohomology.h>
5
+ #include <gudhi/Simplex_tree_multi.h>
6
+ #include <gudhi/Multi_persistence/Box.h>
7
+
8
+ namespace Gudhi {
9
+ namespace multiparameter {
10
+
11
+ struct Simplex_tree_float { // smaller simplextrees
12
+ typedef linear_indexing_tag Indexing_tag;
13
+ typedef std::int32_t Vertex_handle;
14
+ typedef float Filtration_value;
15
+ typedef std::uint32_t Simplex_key;
16
+ static const bool store_key = true;
17
+ static const bool store_filtration = true;
18
+ static const bool contiguous_vertices = false; // TODO OPTIMIZATION : maybe make the simplextree contiguous when
19
+ // calling grid_squeeze ?
20
+ static const bool link_nodes_by_label = true;
21
+ static const bool stable_simplex_handles = false;
22
+ static const bool is_multi_parameter = false;
23
+ };
24
+
25
+ // using Simplex_tree_float = Simplex_tree_options_fast_persistence;
26
+
27
+ using Gudhi::multi_persistence::Box;
28
+ using Simplex_tree_std = Simplex_tree<Simplex_tree_float>;
29
+
30
+ using Barcode = std::vector<std::pair<Simplex_tree_std::Filtration_value, Simplex_tree_std::Filtration_value>>;
31
+
32
+ inline Barcode compute_dgm(Simplex_tree_std &st, int degree) {
33
+ st.initialize_filtration(true);
34
+ constexpr int coeff_field_characteristic = 11;
35
+ constexpr Simplex_tree_std::Filtration_value min_persistence = 0;
36
+ bool persistence_dim_max = st.dimension() == degree;
37
+ Gudhi::persistent_cohomology::Persistent_cohomology<Simplex_tree_std, Gudhi::persistent_cohomology::Field_Zp> pcoh(
38
+ st, persistence_dim_max);
39
+ pcoh.init_coefficients(coeff_field_characteristic);
40
+ pcoh.compute_persistent_cohomology(min_persistence);
41
+ const auto &persistent_pairs = pcoh.intervals_in_dimension(degree);
42
+ if constexpr (false) {
43
+ std::cout << "Number of bars : " << persistent_pairs.size() << "\n";
44
+ }
45
+ return persistent_pairs;
46
+ }
47
+
48
+ template <typename degree_type, class interface_std_like>
49
+ inline std::vector<Barcode> compute_dgms(interface_std_like &st,
50
+ const std::vector<degree_type> &degrees,
51
+ int num_collapses,
52
+ int expansion_dim) {
53
+ std::vector<Barcode> out(degrees.size());
54
+ static_assert(!interface_std_like::Options::is_multi_parameter,
55
+ "Can only compute persistence for 1-parameter simplextrees.");
56
+ const bool verbose = false;
57
+ if (num_collapses > 0) {
58
+ auto collapsed_st = collapse_edges(st, num_collapses);
59
+ return compute_dgms(collapsed_st, degrees, 0, expansion_dim);
60
+ }
61
+ // if (expansion_dim == -1){
62
+ // int max_dim = *std::max_element(degrees.begin(), degrees.end()) +1;
63
+ // st.expansion(max_dim);
64
+ // }
65
+ else if (expansion_dim > 0) {
66
+ st.expansion(expansion_dim);
67
+ }
68
+
69
+ st.initialize_filtration(true); // true is ignore_infinite_values
70
+ constexpr int coeff_field_characteristic = 11;
71
+ constexpr typename interface_std_like::Filtration_value min_persistence = 0;
72
+
73
+ bool persistence_dim_max = false;
74
+ for (auto degree : degrees) {
75
+ if (st.dimension() == degree) {
76
+ persistence_dim_max = true;
77
+ break;
78
+ }
79
+ }
80
+ //
81
+ if constexpr (verbose) {
82
+ std::cout << "Computing dgm of st:\n";
83
+ for (auto &sh : st.filtration_simplex_range()) {
84
+ std::cout << "dim: " << st.dimension(sh) << " vertices: ";
85
+ for (auto v : st.simplex_vertex_range(sh)) {
86
+ std::cout << v << " ";
87
+ }
88
+ std::cout << " filtration: ";
89
+ std::cout << st.filtration(sh) << "\n";
90
+ }
91
+ return out;
92
+ }
93
+
94
+ Gudhi::persistent_cohomology::Persistent_cohomology<interface_std_like, Gudhi::persistent_cohomology::Field_Zp> pcoh(
95
+ st, persistence_dim_max);
96
+ pcoh.init_coefficients(coeff_field_characteristic);
97
+ pcoh.compute_persistent_cohomology(min_persistence);
98
+ for (auto i = 0u; i < degrees.size(); i++) {
99
+ out[i] = pcoh.intervals_in_dimension(degrees[i]);
100
+ }
101
+ return out;
102
+ }
103
+
104
+ // small wrapper
105
+ template <typename degree_type, class interface_std_like>
106
+ inline std::vector<Barcode> compute_dgms(interface_std_like &st,
107
+ const std::vector<degree_type> &degrees,
108
+ int expand_collapse_dim = 0) {
109
+ if (expand_collapse_dim > 0) return compute_dgms(st, degrees, 10, expand_collapse_dim);
110
+ return compute_dgms(st, degrees, 0, 0);
111
+ }
112
+
113
+ // Adapted version from the Simplextree interface
114
+ template <class simplextree_like>
115
+ inline simplextree_like collapse_edges(simplextree_like &st, int num_collapses) {
116
+ using Filtered_edge = std::tuple<typename simplextree_like::Vertex_handle,
117
+ typename simplextree_like::Vertex_handle,
118
+ typename simplextree_like::Filtration_value>;
119
+ std::vector<Filtered_edge> edges;
120
+ for (auto sh : st.skeleton_simplex_range(1)) {
121
+ if (st.dimension(sh) == 1) {
122
+ const auto filtration = st.filtration(sh);
123
+ if (filtration == std::numeric_limits<typename simplextree_like::Filtration_value>::infinity()) continue;
124
+ typename simplextree_like::Simplex_vertex_range rg = st.simplex_vertex_range(sh);
125
+ auto vit = rg.begin();
126
+ typename simplextree_like::Vertex_handle v = *vit;
127
+ typename simplextree_like::Vertex_handle w = *++vit;
128
+ edges.emplace_back(v, w, filtration);
129
+ }
130
+ }
131
+ for (int iteration = 0; iteration < num_collapses; iteration++) {
132
+ auto current_size = edges.size();
133
+ edges = Gudhi::collapse::flag_complex_collapse_edges(std::move(edges));
134
+ if (edges.size() >= current_size) break; // no need to do more
135
+ }
136
+ simplextree_like collapsed_stree;
137
+ // Copy the original 0-skeleton
138
+ for (auto sh : st.skeleton_simplex_range(0)) {
139
+ collapsed_stree.insert_simplex({*(st.simplex_vertex_range(sh).begin())}, st.filtration(sh));
140
+ }
141
+ // Insert remaining edges
142
+ for (auto [x, y, filtration] : edges) {
143
+ collapsed_stree.insert_simplex({x, y}, filtration);
144
+ }
145
+ return collapsed_stree;
146
+ }
147
+
148
+ }} // namespace Gudhi::multiparameter
@@ -0,0 +1,369 @@
1
+ #pragma once
2
+ #include <gudhi/Simplex_tree_multi.h>
3
+ #include "gudhi/truc.h"
4
+ #include "multi_parameter_rank_invariant/persistence_slices.h"
5
+ #include "tensor/tensor.h"
6
+ #include <algorithm>
7
+ #include <cstddef>
8
+ #include <gudhi/One_critical_filtration.h>
9
+ #include <oneapi/tbb/enumerable_thread_specific.h>
10
+ #include <oneapi/tbb/parallel_for.h>
11
+ #include <ostream>
12
+ #include <utility> // std::pair
13
+ #include <vector>
14
+
15
+ namespace Gudhi {
16
+ namespace multiparameter {
17
+ namespace rank_invariant {
18
+ using Index = truc_interface::index_type;
19
+
20
+ // using Elbow = std::vector<std::pair<>>;grid
21
+ template <typename index_type>
22
+ inline void push_in_elbow(index_type &i, index_type &j, const index_type I, const index_type J) {
23
+ if (j < J) {
24
+ j++;
25
+ return;
26
+ }
27
+ if (i < I) {
28
+ i++;
29
+ return;
30
+ }
31
+ j++;
32
+ return;
33
+ }
34
+
35
+ template <typename index_type, typename value_type>
36
+ inline value_type get_slice_rank_filtration(const value_type x,
37
+ const value_type y,
38
+ const index_type I,
39
+ const index_type J) {
40
+ if (x > static_cast<value_type>(I))
41
+ return std::numeric_limits<value_type>::has_infinity ? std::numeric_limits<value_type>::infinity()
42
+ : std::numeric_limits<value_type>::max();
43
+ if (y > static_cast<value_type>(J)) return I + static_cast<index_type>(y);
44
+ return J + static_cast<index_type>(x);
45
+ }
46
+
47
+ template <typename index_type>
48
+ inline std::pair<index_type, index_type> get_coordinates(index_type in_slice_value, index_type I, index_type J) {
49
+ if (in_slice_value <= J) return {0, J};
50
+ if (in_slice_value <= I + J) return {in_slice_value - J, J};
51
+ return {I, in_slice_value - I};
52
+ }
53
+
54
+ template <typename dtype, typename index_type, typename Filtration>
55
+ inline void compute_2d_rank_invariant_of_elbow(
56
+ Simplex_tree<Gudhi::multi_persistence::Simplex_tree_options_multidimensional_filtration<Filtration>> &st_multi,
57
+ Simplex_tree_std &_st_container, // copy of st_multi
58
+ const tensor::static_tensor_view<dtype, index_type> &out, // assumes its a zero tensor
59
+ const index_type I,
60
+ const index_type J,
61
+ const std::vector<index_type> &grid_shape,
62
+ const std::vector<index_type> &degrees,
63
+ const int expand_collapse_max_dim = 0) {
64
+ // const bool verbose = false; // verbose
65
+ using value_type = typename Simplex_tree_std::Filtration_value;
66
+ constexpr const value_type inf = std::numeric_limits<value_type>::infinity();
67
+
68
+ // const auto X = grid_shape[1], Y = grid_shape[2]; // First axis is degree
69
+ const auto Y = grid_shape[2];
70
+ // Fills the filtration in the container
71
+ // TODO : C++23 zip, when Apples clang will stop being obsolete
72
+ auto sh_standard = _st_container.complex_simplex_range().begin();
73
+ auto _end = _st_container.complex_simplex_range().end();
74
+ auto sh_multi = st_multi.complex_simplex_range().begin();
75
+ for (; sh_standard != _end; ++sh_multi, ++sh_standard) {
76
+ const Filtration &multi_filtration = st_multi.filtration(*sh_multi);
77
+ value_type filtration_in_slice;
78
+ if constexpr (Filtration::is_multi_critical) {
79
+ filtration_in_slice = inf;
80
+ for (const auto &stuff : multi_filtration) {
81
+ value_type x = stuff[0];
82
+ value_type y = stuff[1];
83
+ filtration_in_slice = std::min(filtration_in_slice, get_slice_rank_filtration(x, y, I, J));
84
+ }
85
+ } else {
86
+ value_type x = multi_filtration[0];
87
+ value_type y = multi_filtration[1];
88
+ filtration_in_slice = get_slice_rank_filtration(x, y, I, J);
89
+ }
90
+ _st_container.assign_filtration(*sh_standard, filtration_in_slice);
91
+ }
92
+ const std::vector<Barcode> &barcodes = compute_dgms(_st_container, degrees, expand_collapse_max_dim);
93
+ index_type degree_index = 0;
94
+ for (const auto &barcode : barcodes) { // TODO range view cartesian product
95
+ for (const auto &bar : barcode) {
96
+ auto birth = static_cast<index_type>(bar.first);
97
+ auto death = static_cast<index_type>(
98
+ std::min(bar.second,
99
+ static_cast<typename Simplex_tree_std::Filtration_value>(Y + I))); // I,J atteints, pas X ni Y
100
+
101
+ // todo : optimize
102
+ // auto [a,b] = get_coordinates(birth, I,J);
103
+ for (auto intermediate_birth = birth; intermediate_birth < death; intermediate_birth++) {
104
+ for (auto intermediate_death = intermediate_birth; intermediate_death < death; intermediate_death++) {
105
+ auto [i, j] = get_coordinates(intermediate_birth, I, J);
106
+ auto [k, l] = get_coordinates(intermediate_death, I, J);
107
+ if (((i < k || j == J) && (j < l || k == I))) {
108
+ // std::vector<index_type> coordinates_to_remove =
109
+ // {degree_index,i,j,k,l}; out[coordinates_to_remove]++;
110
+ out[{degree_index, i, j, k, l}]++;
111
+ }
112
+ }
113
+ }
114
+ }
115
+ degree_index++;
116
+ }
117
+ }
118
+
119
+ template <typename dtype, typename index_type, typename Filtration>
120
+ inline void compute_2d_rank_invariant(
121
+ Simplex_tree<Gudhi::multi_persistence::Simplex_tree_options_multidimensional_filtration<Filtration>> &st_multi,
122
+ const tensor::static_tensor_view<dtype, index_type> &out, // assumes its a zero tensor
123
+ const std::vector<index_type> &grid_shape,
124
+ const std::vector<index_type> &degrees,
125
+ bool expand_collapse) {
126
+ if (degrees.size() == 0) return;
127
+ assert(st_multi.get_number_of_parameters() == 2);
128
+ Simplex_tree_std st_;
129
+ Gudhi::multi_persistence::flatten(st_, st_multi,
130
+ 0); // copies the st_multi to a standard 1-pers simplextree
131
+ const int max_dim = expand_collapse ? *std::max_element(degrees.begin(), degrees.end()) + 1 : 0;
132
+ index_type X = grid_shape[1];
133
+ index_type Y = grid_shape[2]; // First axis is degree
134
+ tbb::enumerable_thread_specific<Simplex_tree_std> thread_simplex_tree(st_); // initialize with a good simplextree
135
+ tbb::parallel_for(0, X, [&](index_type I) {
136
+ tbb::parallel_for(0, Y, [&](index_type J) {
137
+ auto &st_container = thread_simplex_tree.local();
138
+ compute_2d_rank_invariant_of_elbow<dtype, index_type, Filtration>(
139
+ st_multi, st_container, out, I, J, grid_shape, degrees, max_dim);
140
+ });
141
+ });
142
+ }
143
+
144
+ template <typename Filtration, typename dtype, typename indices_type, typename... Args>
145
+ void compute_rank_invariant_python(
146
+ Simplex_tree<Gudhi::multi_persistence::Simplex_tree_options_multidimensional_filtration<Filtration>> &st_multi,
147
+ dtype *data_ptr,
148
+ const std::vector<indices_type> grid_shape,
149
+ const std::vector<indices_type> degrees,
150
+ indices_type n_jobs,
151
+ bool expand_collapse) {
152
+ if (degrees.size() == 0) return;
153
+ tensor::static_tensor_view<dtype, indices_type> container(data_ptr, grid_shape); // assumes its a zero tensor
154
+
155
+ oneapi::tbb::task_arena arena(n_jobs); // limits the number of threads
156
+ arena.execute([&] {
157
+ compute_2d_rank_invariant<dtype, indices_type, Filtration>(
158
+ st_multi, container, grid_shape, degrees, expand_collapse);
159
+ });
160
+
161
+ return;
162
+ }
163
+
164
+ template <class PersBackend,
165
+ class Structure,
166
+ class MultiFiltration = Gudhi::multi_filtration::One_critical_filtration<float>,
167
+ typename dtype,
168
+ typename index_type>
169
+ inline void compute_2d_rank_invariant_of_elbow(
170
+ typename truc_interface::Truc<PersBackend, Structure, MultiFiltration>::ThreadSafe &slicer, // truc slicer
171
+ const tensor::static_tensor_view<dtype, index_type> &out, // assumes its a zero tensor
172
+ const index_type I,
173
+ const index_type J,
174
+ const std::vector<index_type> &grid_shape,
175
+ const std::vector<index_type> &degrees,
176
+ // std::vector<Index> &order_container, // constant size
177
+ // std::vector<typename MultiFiltration::value_type> &one_persistence, // constant size
178
+ const bool flip_death = false,
179
+ const bool ignore_inf = true) {
180
+ using value_type = typename MultiFiltration::value_type;
181
+ const auto &filtrations_values = slicer.get_filtrations();
182
+ auto num_generators = filtrations_values.size();
183
+ // one_persistence.resize(num_generators); // local variable should be
184
+ // initialized correctly
185
+ const auto Y = grid_shape[2];
186
+ constexpr const bool verbose = false;
187
+ if constexpr (verbose) std::cout << "filtration_in_slice : [ ";
188
+ for (auto i = 0u; i < num_generators; ++i) {
189
+ const auto &f = filtrations_values[i];
190
+ value_type filtration_in_slice = MultiFiltration::Generator::T_inf;
191
+ if constexpr (MultiFiltration::is_multi_critical) {
192
+ for (const auto &stuff : f) {
193
+ value_type x = stuff[0];
194
+ value_type y = stuff[1];
195
+
196
+ filtration_in_slice = std::min(filtration_in_slice, get_slice_rank_filtration(x, y, I, J));
197
+ }
198
+ } else {
199
+ value_type x = f[0];
200
+ value_type y = f[1];
201
+ filtration_in_slice = get_slice_rank_filtration(x, y, I, J);
202
+ }
203
+ if constexpr (verbose) std::cout << filtration_in_slice << ",";
204
+ slicer.get_one_filtration()[i] = filtration_in_slice;
205
+ }
206
+ if constexpr (verbose) std::cout << "\b]" << std::endl;
207
+
208
+ index_type degree_index = 0;
209
+ // order_container.resize(slicer.num_generators()); // local variable should
210
+ // be initialized correctly
211
+ // TODO : use slicer::ThreadSafe instead of maintaining one_pers & order
212
+ // BUG : This will break as soon as slicer interface change
213
+
214
+ using bc_type = typename truc_interface::Truc<PersBackend, Structure, MultiFiltration>::split_barcode;
215
+ bc_type barcodes;
216
+ if constexpr (PersBackend::is_vine) {
217
+ // slicer.set_one_filtration(one_persistence);
218
+ if (I == 0 && J == 0) [[unlikely]] // this is dangerous, assumes it starts at 0 0
219
+ {
220
+ // TODO : This is a good optimization but needs a patch on
221
+ // PersistenceMatrix std::vector<bool>
222
+ // degrees_index(slicer.get_dimensions().back()+1, false); for (const auto
223
+ // &degree : degrees) {
224
+ // if (degree <= slicer.get_dimensions())
225
+ // degrees_index[degree] = true;
226
+ // }
227
+ // slicer.compute_persistence(degrees_index);
228
+ slicer.compute_persistence();
229
+ } else {
230
+ slicer.vineyard_update();
231
+ }
232
+ barcodes = slicer.get_barcode();
233
+ } else {
234
+ slicer.compute_persistence(ignore_inf);
235
+ barcodes = slicer.get_barcode();
236
+ }
237
+
238
+ // note one_pers not necesary when vine, but does the same computation
239
+
240
+ for (auto degree : degrees) {
241
+ // this assumes barcodes degrees starts from 0
242
+ if constexpr (verbose) std::cout << "Adding Barcode of degree " << degree << std::endl;
243
+ if (degree >= static_cast<index_type>(barcodes.size())) continue;
244
+ const auto &barcode = barcodes[degree];
245
+ for (const auto &bar : barcode) {
246
+ if (bar.first > Y + I) continue;
247
+ if constexpr (verbose)
248
+ std::cout << bar.first << " " << bar.second << "checkinf: " << MultiFiltration::Generator::T_inf << " ==? "
249
+ << (bar.first == MultiFiltration::Generator::T_inf) << std::endl;
250
+ auto birth = static_cast<index_type>(bar.first);
251
+ auto death = static_cast<index_type>(
252
+ std::min(bar.second,
253
+ static_cast<typename MultiFiltration::value_type>(Y + I))); // I,J atteints, pas X ni Y
254
+ if constexpr (false) std::cout << "Birth " << birth << " Death " << death << std::endl;
255
+ for (auto intermediate_birth = birth; intermediate_birth < death; intermediate_birth++) {
256
+ for (auto intermediate_death = intermediate_birth; intermediate_death < death; intermediate_death++) {
257
+ auto [i, j] = get_coordinates(intermediate_birth, I, J);
258
+ auto [k, l] = get_coordinates(intermediate_death, I, J);
259
+ if (((i < k || j == J) && (j < l || k == I))) {
260
+ if (flip_death)
261
+ out[{degree_index, i, j, I - 1 - k, J - 1 - l}]++;
262
+ else
263
+ out[{degree_index, i, j, k, l}]++;
264
+ }
265
+ if constexpr (false) std::cout << degree_index << " " << i << " " << j << " " << k << " " << l << std::endl;
266
+ }
267
+ }
268
+ }
269
+ degree_index++;
270
+ }
271
+ };
272
+
273
+ template <class PersBackend,
274
+ class Structure,
275
+ class MultiFiltration = Gudhi::multi_filtration::One_critical_filtration<float>,
276
+ typename dtype,
277
+ typename index_type>
278
+ inline void compute_2d_rank_invariant(
279
+ truc_interface::Truc<PersBackend, Structure, MultiFiltration> &slicer,
280
+ const tensor::static_tensor_view<dtype, index_type> &out, // assumes its a zero tensor
281
+ const std::vector<index_type> &grid_shape,
282
+ const std::vector<index_type> &degrees,
283
+ const bool flip_death,
284
+ const bool ignore_inf) {
285
+ if (degrees.size() == 0) return;
286
+ index_type X = grid_shape[1];
287
+ index_type Y = grid_shape[2]; // First axis is degree
288
+ constexpr const bool verbose = false;
289
+ if constexpr (verbose)
290
+ std::cout << "Shape " << grid_shape[0] << " " << grid_shape[1] << " " << grid_shape[2] << " " << grid_shape[3]
291
+ << " " << grid_shape[4] << std::endl;
292
+
293
+ using ThreadSafe = typename truc_interface::Truc<PersBackend, Structure, MultiFiltration>::ThreadSafe;
294
+ ThreadSafe slicer_thread(slicer);
295
+ tbb::enumerable_thread_specific<ThreadSafe> thread_locals(slicer_thread);
296
+ tbb::parallel_for(0, X, [&](index_type I) {
297
+ tbb::parallel_for(0, Y, [&](index_type J) {
298
+ if constexpr (verbose) std::cout << "Computing elbow " << I << " " << J << "...";
299
+ ThreadSafe &slicer = thread_locals.local();
300
+ compute_2d_rank_invariant_of_elbow<PersBackend, Structure, MultiFiltration, dtype, index_type>(
301
+ slicer, out, I, J, grid_shape, degrees, flip_death, ignore_inf);
302
+ if constexpr (verbose) std::cout << "Done!" << std::endl;
303
+ });
304
+ });
305
+ }
306
+
307
+ template <class PersBackend,
308
+ class Structure,
309
+ class MultiFiltration = Gudhi::multi_filtration::One_critical_filtration<float>,
310
+ typename dtype,
311
+ typename indices_type>
312
+ void compute_rank_invariant_python(truc_interface::Truc<PersBackend, Structure, MultiFiltration> slicer,
313
+ dtype *data_ptr,
314
+ const std::vector<indices_type> grid_shape,
315
+ const std::vector<indices_type> degrees,
316
+ indices_type n_jobs,
317
+ const bool ignore_inf) {
318
+ if (degrees.size() == 0) return;
319
+ tensor::static_tensor_view<dtype, indices_type> container(data_ptr, grid_shape); // assumes its a zero tensor
320
+ if constexpr (false){
321
+ std::cout << "ignore_inf " << ignore_inf << std::endl;
322
+ }
323
+
324
+ oneapi::tbb::task_arena arena(PersBackend::is_vine ? 1 : n_jobs); // limits the number of threads
325
+ arena.execute([&] { compute_2d_rank_invariant(slicer, container, grid_shape, degrees, false, ignore_inf); });
326
+
327
+ return;
328
+ }
329
+
330
+ template <typename PersBackend,
331
+ typename Structure,
332
+ typename MultiFiltration,
333
+ typename dtype = int,
334
+ typename indices_type = int>
335
+ std::pair<std::vector<std::vector<indices_type>>, std::vector<dtype>> compute_rank_signed_measure(
336
+ truc_interface::Truc<PersBackend, Structure, MultiFiltration> slicer,
337
+ dtype *data_ptr,
338
+ const std::vector<indices_type> grid_shape,
339
+ const std::vector<indices_type> degrees,
340
+ indices_type n_jobs,
341
+ bool verbose,
342
+ const bool ignore_inf) {
343
+ if (degrees.size() == 0) return {{}, {}};
344
+ tensor::static_tensor_view<dtype, indices_type> container(data_ptr, grid_shape); // assumes its a zero tensor
345
+ oneapi::tbb::task_arena arena(n_jobs); // limits the number of threads
346
+ constexpr bool flip_death = true;
347
+ arena.execute([&] { compute_2d_rank_invariant(slicer, container, grid_shape, degrees, flip_death, ignore_inf); });
348
+
349
+ if (verbose) {
350
+ std::cout << "Done.\n";
351
+ std::cout << "Computing mobius inversion ..." << std::flush;
352
+ }
353
+
354
+ // for (indices_type axis :
355
+ // std::views::iota(2,st_multi.get_number_of_parameters()+1)) // +1 for the
356
+ // degree in axis 0
357
+ for (std::size_t axis = 0u; axis < slicer.num_parameters() + 1; axis++) container.differentiate(axis);
358
+ if (verbose) {
359
+ std::cout << "Done.\n";
360
+ std::cout << "Sparsifying the measure ..." << std::flush;
361
+ }
362
+ auto raw_signed_measure = container.sparsify({false, false, true, true});
363
+ if (verbose) {
364
+ std::cout << "Done.\n";
365
+ }
366
+ return raw_signed_measure;
367
+ }
368
+
369
+ }}} // namespace Gudhi::multiparameter::rank_invariant
@@ -0,0 +1,41 @@
1
+ from tqdm import tqdm
2
+
3
+
4
+ def _collapse_edge_list(
5
+ edges,
6
+ num: int = 0,
7
+ full: bool = False,
8
+ strong: bool = False,
9
+ progress: bool = False,
10
+ ):
11
+ """
12
+ Given an edge list defining a 1 critical 2 parameter 1 dimensional simplicial complex, simplificates this filtered simplicial complex, using filtration-domination's edge collapser.
13
+ """
14
+ from filtration_domination import (
15
+ remove_filtration_dominated,
16
+ remove_strongly_filtration_dominated,
17
+ )
18
+
19
+ n = len(edges)
20
+ if full:
21
+ num = 100
22
+ with tqdm(
23
+ range(num), total=num, desc="Removing edges", disable=not (progress)
24
+ ) as I:
25
+ for i in I:
26
+ if strong:
27
+ edges = remove_strongly_filtration_dominated(edges) # nogil ?
28
+ else:
29
+ edges = remove_filtration_dominated(edges)
30
+ # Prevents doing useless collapses
31
+ if len(edges) >= n:
32
+ if full and strong:
33
+ strong = False
34
+ n = len(edges)
35
+ # n = edges.size() # len(edges)
36
+ else:
37
+ break
38
+ else:
39
+ n = len(edges)
40
+ # n = edges.size()
41
+ return edges