openlr/candidate_points_getter.hpp


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

#pragma once

#include "openlr/graph.hpp"
#include "openlr/stats.hpp"

#include "indexer/data_source.hpp"

#include "geometry/point2d.hpp"

#include <cstddef>
#include <functional>
#include <vector>

namespace openlr
{
class CandidatePointsGetter
{
public:
  CandidatePointsGetter(size_t const maxJunctionCandidates, size_t const maxProjectionCandidates,
                        DataSource const & dataSource, Graph & graph)
    : m_maxJunctionCandidates(maxJunctionCandidates)
    , m_maxProjectionCandidates(maxProjectionCandidates)
    , m_dataSource(dataSource)
    , m_graph(graph)
  {
  }

  void GetCandidatePoints(m2::PointD const & p, std::vector<m2::PointD> & candidates)
  {
    FillJunctionPointCandidates(p, candidates);
    EnrichWithProjectionPoints(p, candidates);
  }

private:
  void FillJunctionPointCandidates(m2::PointD const & p, std::vector<m2::PointD> & candidates);
  void EnrichWithProjectionPoints(m2::PointD const & p, std::vector<m2::PointD> & candidates);

  size_t const m_maxJunctionCandidates;
  size_t const m_maxProjectionCandidates;

  DataSource const & m_dataSource;
  Graph & m_graph;
};
}  // namespace openlr