osrm-train-profile/basic.lua at master · envadrouillecorp/osrm-train-profile · GitHub

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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
-- Copyright 2017-2019 Railnova SA <support@railnova.eu>, Nikita Marchant <nikita.marchant@gmail.com>
-- Code under the 2-clause BSD license

api_version = 4

function setup()
  return {
    properties = {
      max_speed_for_map_matching     = 220/3.6, -- speed conversion to m/s
      weight_name                    = 'routability',
      left_hand_driving              = true,
      u_turn_penalty                 = 60 * 2, -- 2 minutes to change cabin
      turn_duration                  = 20,
      continue_straight_at_waypoint  = false,
      max_angle                      = 30,

      secondary_speed                = 30,
      speed                          = 160,
    },

    default_mode              = mode.train,
    default_speed             = 120,
}

end


function ternary ( cond , T , F )
    if cond then return T else return F end
end


function process_node(profile, node, result, relations)
    local railway = node:get_value_by_key("railway")

    -- refuse railway nodes that we cannot go through
    result.barrier = (
        railway == "buffer_stop" or
        railway == "derail"
    )
    result.traffic_lights = false
end

function process_way(profile, way, result, relations)
    local data = {
        railway = way:get_value_by_key("railway"),
        route = way:get_value_by_key("route"),
        service = way:get_value_by_key("service"),
        usage = way:get_value_by_key("usage"),
        maxspeed = way:get_value_by_key("maxspeed"),
        gauge = way:get_value_by_key("gauge"),
    }

    -- Remove everything that is not railway
    if data.route and data.route == 'ferry' then
	    -- ok
    elseif not data.railway then
        return
    -- Remove everything that is not a rail, a turntable, a traverser
    elseif (
        data.railway ~= 'rail' and
        data.railway ~= 'turntable' and
        data.railway ~= 'traverser' and
        data.railway ~= 'light_rail' and
        data.railway ~= 'narrow_gauge' and
        data.railway ~= 'funicular' and
        data.railway ~= 'tram' and
        data.railway ~= 'subway' and
        data.railway  ~= 'ferry'
    ) then
        return
    -- Remove military and tourism rails
    --elseif (
    --    data.usage == "military" or
    --    data.usage == "tourism"
    --) then
    --    return
    -- Keep only most common gauges (and undefined)
    -- uses .find() as some gauges are specified like "1668;1435"
    --elseif (
    --    data.gauge ~= nil and
    --    data.gauge ~= 1000 and not string.find(data.gauge, "1000") and
    --    data.gauge ~= 1435 and not string.find(data.gauge, "1435") and
    --    data.gauge ~= 1520 and not string.find(data.gauge, "1520") and
    --    data.gauge ~= 1524 and not string.find(data.gauge, "1524") and
    --    data.gauge ~= 1600 and not string.find(data.gauge, "1600") and
    --    data.gauge ~= 1668 and not string.find(data.gauge, "1668")
   --) then
   --     return
    end

    local is_secondary = (
        data.service == "siding" or
        data.service == "spur" or
        data.service == "yard" or
        data.usage == "industrial"
    )

    -- by default, use 30km/h for secondary rails, else 160
    local default_speed = ternary(is_secondary, profile.properties.secondary_speed, profile.properties.speed)
    -- but is OSM specifies a maxspeed, use the one from OSM
    local speed = ternary(data.maxspeed, data.maxspeed, default_speed)

   -- fix speed for mph issue
    speed = tostring(speed)
    if speed:find(" mph") or speed:find("mph") then
      speed = speed:gsub(" mph", "")
      speed = speed:gsub("mph", "")
        speed = tonumber (speed)
        if speed == nil then speed = 20 end
	speed = speed * 1.609344
    else
     speed = tonumber (speed)
    end
    -- fix speed for mph issue end


    result.forward_speed = speed
    result.backward_speed = speed
    --
    result.forward_mode = mode.train
    result.backward_mode = mode.train
    --
    result.forward_rate = 1
    result.backward_rate = 1

end

function process_turn(profile, turn)
    -- Refuse truns that have a big angle
    if math.abs(turn.angle) >  profile.properties.max_angle then
        return
    end

    -- If we go backwards, add the penalty to change cabs
    if turn.is_u_turn then
      turn.duration = turn.duration + profile.properties.u_turn_penalty
    end
end

return {
    setup = setup,
    process_way = process_way,
    process_node = process_node,
    process_turn = process_turn
}