summaryrefslogtreecommitdiff
path: root/src/Witness_complex/include/gudhi/Active_witness/Active_witness_iterator.h
blob: 7f48faa42c393b2ee94516b7eea8447706b613d5 (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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*    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) 2016  INRIA (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 ACTIVE_WITNESS_ITERATOR_H_
#define ACTIVE_WITNESS_ITERATOR_H_

#include <boost/iterator/iterator_facade.hpp>
#include <vector>

namespace Gudhi {

namespace witness_complex {

/* \brief Iterator in the nearest landmark list.
 *  \details After the iterator reaches the end of the list,
 *          the list is augmented by a (nearest landmark, distance) pair if possible.
 *          If all the landmarks are present in the list, iterator returns the specific end value
 *          of the corresponding 'Active_witness' object.
 */
  
template< typename Active_witness,
          typename Id_distance_pair,
          typename INS_iterator >
class Active_witness_iterator
  : public boost::iterator_facade< Active_witness_iterator <Active_witness, Id_distance_pair, INS_iterator>
, Id_distance_pair const
, boost::forward_traversal_tag
, Id_distance_pair const> {
  friend class boost::iterator_core_access;
  
  //typedef Active_witness<Id_distance_pair, INS_iterator> Active_witness;
  typedef typename std::list<Id_distance_pair>::iterator Pair_iterator;
  typedef typename Gudhi::witness_complex::Active_witness_iterator<Active_witness, Id_distance_pair, INS_iterator> Iterator;
  
  
  Active_witness *aw_;
  Pair_iterator lh_; // landmark handle
  bool is_end_; // true only if the pointer is end and there are no more neighbors to add

public:
  Active_witness_iterator(Active_witness* aw)
    : aw_(aw), lh_(aw_->nearest_landmark_table_.end()), is_end_(true)
  {
  }

  Active_witness_iterator(Active_witness* aw, const Pair_iterator& lh)
    : aw_(aw), lh_(lh)
  {
    is_end_ = false;
    if (lh_ == aw_->nearest_landmark_table_.end()) {
      if (aw_->iterator_next_ == aw_->iterator_end_)
        is_end_ = true;
      else {
        aw_->nearest_landmark_table_.push_back(*aw_->iterator_next_);
        lh_ = --aw_->nearest_landmark_table_.end();
        ++(aw_->iterator_next_);
      }
    }
  }
  
private :

  Id_distance_pair& dereference() const
  {
    return *lh_;
  }

  bool equal(const Iterator& other) const
  {
    return (is_end_ == other.is_end_) || (lh_ == other.lh_);
  }
  
  void increment()
  {
    // the neighbor search can't be at the end iterator of a list
    GUDHI_CHECK(!is_end_ && lh_ != aw_->nearest_landmark_table_.end(), std::logic_error("Wrong active witness increment."));
    // if the id of the current landmark is the same as the last one

    lh_++;
    if (lh_ == aw_->nearest_landmark_table_.end()) {
      if (aw_->iterator_next_ == aw_->iterator_end_)
        is_end_ = true;
      else {
        aw_->nearest_landmark_table_.push_back(*aw_->iterator_next_);
        lh_ = std::prev(aw_->nearest_landmark_table_.end());
        ++(aw_->iterator_next_);
      }
    }
  }
};

}
}
  
#endif