summaryrefslogtreecommitdiff
path: root/src/Alpha_complex/example/Weighted_alpha_complex_3d_from_points.cpp
blob: 68f72f0a63014da152d6a02a51791b50b2286cf1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <gudhi/Alpha_complex_3d.h>
// to construct a simplex_tree from alpha complex
#include <gudhi/Simplex_tree.h>

#include <iostream>
#include <string>
#include <vector>
#include <limits>  // for numeric limits

using Weighted_alpha_complex_3d =
    Gudhi::alpha_complex::Alpha_complex_3d<Gudhi::alpha_complex::complexity::EXACT, true, false>;
using Point = Weighted_alpha_complex_3d::Point_3;
using Weighted_point = Weighted_alpha_complex_3d::Triangulation_3::Weighted_point;
using Vector_of_weighted_points = std::vector<Weighted_point>;
using Vector_of_weights = std::vector<Weighted_alpha_complex_3d::Alpha_shape_3::FT>;

int main(int argc, char **argv) {
  if (argc != 1) {
    std::cerr << "Error: Number of arguments (" << argc << ") is not correct\n";
    std::cerr << "Usage: " << (argv[0] - 1) << " \n";
    exit(-1);  // ----- >>
  }

  // ----------------------------------------------------------------------------
  // Init of a list of points and weights from a small molecule
  // ----------------------------------------------------------------------------
  Vector_of_weighted_points weighted_points;
  weighted_points.push_back(Weighted_point(Point(1, -1, -1), 4.));
  weighted_points.push_back(Weighted_point(Point(-1, 1, -1), 4.));
  weighted_points.push_back(Weighted_point(Point(-1, -1, 1), 4.));
  weighted_points.push_back(Weighted_point(Point(1, 1, 1), 4.));
  weighted_points.push_back(Weighted_point(Point(2, 2, 2), 1.));

  // ----------------------------------------------------------------------------
  // Init of an alpha complex from the list of points
  // ----------------------------------------------------------------------------
  Weighted_alpha_complex_3d alpha_complex_from_points(weighted_points);

  Gudhi::Simplex_tree<> simplex;
  if (alpha_complex_from_points.create_complex(simplex)) {
    // ----------------------------------------------------------------------------
    // Display information about the alpha complex
    // ----------------------------------------------------------------------------
    std::cout << "Alpha complex is of dimension " << simplex.dimension() << " - " << simplex.num_simplices()
              << " simplices - " << simplex.num_vertices() << " vertices." << std::endl;

    std::cout << "Iterator on alpha complex simplices in the filtration order, with [filtration value]:" << std::endl;
    for (auto f_simplex : simplex.filtration_simplex_range()) {
      std::cout << "   ( ";
      for (auto vertex : simplex.simplex_vertex_range(f_simplex)) {
        std::cout << vertex << " ";
      }
      std::cout << ") -> "
                << "[" << simplex.filtration(f_simplex) << "] ";
      std::cout << std::endl;
    }
  }
  return 0;
}