summaryrefslogtreecommitdiff
path: root/src/Witness_complex/include/gudhi/Landmark_choice_by_random_point.h
blob: 215c9e6577d10da307e9786bb7d22e60ac2c0613 (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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/*    This file is part of the Gudhi Library. The Gudhi library
 *    (Geometric Understanding in Higher Dimensions) is a generic C++
 *    library for computational topology.
 *
 *    Author(s):       Siargey Kachanovich
 *
 *    Copyright (C) 2015  INRIA Sophia Antipolis-Méditerranée (France)
 *
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef LANDMARK_CHOICE_BY_RANDOM_POINT_H_
#define LANDMARK_CHOICE_BY_RANDOM_POINT_H_

#include <queue>  // for priority_queue<>
#include <utility>  // for pair<>
#include <vector>
#include <set>

namespace Gudhi {

namespace witness_complex {

/** 
 *  \class Landmark_choice_by_random_point
 *  \brief The class `Landmark_choice_by_random_point` allows to construct the matrix
 *  of closest landmarks per witness by iteratively choosing a random non-chosen witness
 *  as a new landmark. 
 *  \ingroup witness_complex
 */

class Landmark_choice_by_random_point {
 public:
  /** \brief Landmark choice strategy by taking random vertices for landmarks.
   *  \details It chooses nbL distinct landmarks from a random access range `points`
   *  and outputs a matrix {witness}*{closest landmarks} in knn.
   */

  template <typename KNearestNeighbours,
  typename Point_random_access_range>
  Landmark_choice_by_random_point(Point_random_access_range const &points,
                                  int nbL,
                                  KNearestNeighbours &knn) {
    int nbP = points.end() - points.begin();
    assert(nbP >= nbL);
    std::set<int> landmarks;
    int current_number_of_landmarks = 0;  // counter for landmarks

    // TODO(SK) Consider using rand_r(...) instead of rand(...) for improved thread safety
    int chosen_landmark = rand() % nbP;
    for (current_number_of_landmarks = 0; current_number_of_landmarks != nbL; current_number_of_landmarks++) {
      while (landmarks.find(chosen_landmark) != landmarks.end())
        chosen_landmark = rand() % nbP;
      landmarks.insert(chosen_landmark);
    }

    int dim = points.begin()->size();
    typedef std::pair<double, int> dist_i;
    typedef bool (*comp)(dist_i, dist_i);
    knn = KNearestNeighbours(nbP);
    for (int points_i = 0; points_i < nbP; points_i++) {
      std::priority_queue<dist_i, std::vector<dist_i>, comp> l_heap([&](dist_i j1, dist_i j2) {
        return j1.first > j2.first;
      });
      std::set<int>::iterator landmarks_it;
      int landmarks_i = 0;
      for (landmarks_it = landmarks.begin(), landmarks_i = 0; landmarks_it != landmarks.end();
           landmarks_it++, landmarks_i++) {
        dist_i dist = std::make_pair(euclidean_distance(points[points_i], points[*landmarks_it]), landmarks_i);
        l_heap.push(dist);
      }
      for (int i = 0; i < dim + 1; i++) {
        dist_i dist = l_heap.top();
        knn[points_i].push_back(dist.second);
        l_heap.pop();
      }
    }
  }
};

}  // namespace witness_complex

}  // namespace Gudhi

#endif  // LANDMARK_CHOICE_BY_RANDOM_POINT_H_