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
|
#pragma once
#include "routing/edge_estimator.hpp"
#include "routing/fake_ending.hpp"
#include "routing/fake_graph.hpp"
#include "routing/fake_vertex.hpp"
#include "routing/road_graph.hpp"
#include "routing/route_weight.hpp"
#include "routing/segment.hpp"
#include "transit/transit_graph_data.hpp"
#include "transit/transit_types.hpp"
#include <cstdint>
#include <map>
#include <memory>
#include <set>
#include <vector>
namespace routing
{
class IndexGraph;
class TransitGraph final
{
public:
using GateEndings = std::map<transit::OsmId, FakeEnding>;
static bool IsTransitFeature(uint32_t featureId);
static bool IsTransitSegment(Segment const & segment);
TransitGraph(NumMwmId numMwmId, std::shared_ptr<EdgeEstimator> estimator);
Junction const & GetJunction(Segment const & segment, bool front) const;
RouteWeight CalcSegmentWeight(Segment const & segment) const;
RouteWeight GetTransferPenalty(Segment const & from, Segment const & to) const;
void GetTransitEdges(Segment const & segment, bool isOutgoing,
std::vector<SegmentEdge> & edges) const;
std::set<Segment> const & GetFake(Segment const & real) const;
bool FindReal(Segment const & fake, Segment & real) const;
void Fill(transit::GraphData const & transitData, GateEndings const & gateEndings);
bool IsGate(Segment const & segment) const;
bool IsEdge(Segment const & segment) const;
transit::Gate const & GetGate(Segment const & segment) const;
transit::Edge const & GetEdge(Segment const & segment) const;
private:
using StopToSegmentsMap = std::map<transit::StopId, std::set<Segment>>;
Segment GetTransitSegment(uint32_t segmentIdx) const;
Segment GetNewTransitSegment() const;
// Adds gate to fake graph. Also adds gate to temporary stopToBack, stopToFront maps used while
// TransitGraph::Fill.
void AddGate(transit::Gate const & gate, FakeEnding const & ending,
std::map<transit::StopId, Junction> const & stopCoords, bool isEnter,
StopToSegmentsMap & stopToBack, StopToSegmentsMap & stopToFront);
// Adds transit edge to fake graph, returns corresponding transit segment. Also adds gate to
// temporary stopToBack, stopToFront maps used while TransitGraph::Fill.
Segment AddEdge(transit::Edge const & edge,
std::map<transit::StopId, Junction> const & stopCoords,
StopToSegmentsMap & stopToBack, StopToSegmentsMap & stopToFront);
// Adds connections to fake graph.
void AddConnections(StopToSegmentsMap const & connections, StopToSegmentsMap const & stopToBack,
StopToSegmentsMap const & stopToFront, bool isOutgoing);
NumMwmId const m_mwmId = kFakeNumMwmId;
std::shared_ptr<EdgeEstimator> m_estimator;
FakeGraph<Segment, FakeVertex> m_fake;
std::map<Segment, transit::Edge> m_segmentToEdge;
std::map<Segment, transit::Gate> m_segmentToGate;
std::map<transit::LineId, double> m_transferPenalties;
};
void MakeGateEndings(std::vector<transit::Gate> const & gates, NumMwmId mwmId,
IndexGraph & indexGraph, TransitGraph::GateEndings & gateEndings);
} // namespace routing
|
