Use BETWEEN for the node selection in the map API call as MySQL seems to
[rails.git] / app / controllers / api_controller.rb
1 class ApiController < ApplicationController
2
3   session :off
4   after_filter :compress_output
5
6   #COUNT is the number of map requests to allow before exiting and starting a new process
7   @@count = COUNT
8
9   # The maximum area you're allowed to request, in square degrees
10   MAX_REQUEST_AREA = 0.25
11
12
13   # Number of GPS trace/trackpoints returned per-page
14   TRACEPOINTS_PER_PAGE = 5000
15   
16   def trackpoints
17     @@count+=1
18     #retrieve the page number
19     page = params['page'].to_i
20     unless page
21         page = 0;
22     end
23
24     unless page >= 0
25         report_error("Page number must be greater than or equal to 0")
26         return
27     end
28
29     offset = page * TRACEPOINTS_PER_PAGE
30
31     # Figure out the bbox
32     bbox = params['bbox']
33     unless bbox and bbox.count(',') == 3
34       report_error("The parameter bbox is required, and must be of the form min_lon,min_lat,max_lon,max_lat")
35       return
36     end
37
38     bbox = bbox.split(',')
39
40     min_lon = bbox[0].to_f
41     min_lat = bbox[1].to_f
42     max_lon = bbox[2].to_f
43     max_lat = bbox[3].to_f
44
45     # check the bbox is sane
46     unless min_lon <= max_lon
47       report_error("The minimum longitude must be less than the maximum longitude, but it wasn't")
48       return
49     end
50     unless min_lat <= max_lat
51       report_error("The minimum latitude must be less than the maximum latitude, but it wasn't")
52       return
53     end
54     unless min_lon >= -180 && min_lat >= -90 && max_lon <= 180 && max_lat <= 90
55       report_error("The latitudes must be between -90 and 90, and longitudes between -180 and 180")
56       return
57     end
58
59     # check the bbox isn't too large
60     requested_area = (max_lat-min_lat)*(max_lon-min_lon)
61     if requested_area > MAX_REQUEST_AREA
62       report_error("The maximum bbox size is " + MAX_REQUEST_AREA.to_s + ", and your request was too large. Either request a smaller area, or use planet.osm")
63       return
64     end
65
66     # integerise
67     min_lat = min_lat * 1000000
68     max_lat = max_lat * 1000000
69     min_lon = min_lon * 1000000
70     max_lon = max_lon * 1000000
71     # get all the points
72     points = Tracepoint.find(:all, :conditions => ['latitude BETWEEN ? AND ? AND longitude BETWEEN ? AND ?', min_lat.to_i, max_lat.to_i, min_lon.to_i, max_lon.to_i], :select => "DISTINCT *", :offset => offset, :limit => TRACEPOINTS_PER_PAGE, :order => "timestamp DESC" )
73
74     doc = XML::Document.new
75     doc.encoding = 'UTF-8'
76     root = XML::Node.new 'gpx'
77     root['version'] = '1.0'
78     root['creator'] = 'OpenStreetMap.org'
79     root['xmlns'] = "http://www.topografix.com/GPX/1/0/"
80     
81     doc.root = root
82
83     track = XML::Node.new 'trk'
84     doc.root << track
85
86     trkseg = XML::Node.new 'trkseg'
87     track << trkseg
88
89     points.each do |point|
90       trkseg << point.to_xml_node()
91     end
92
93     #exit when we have too many requests
94     if @@count > MAX_COUNT
95       render :text => doc.to_s, :content_type => "text/xml"
96       @@count = COUNT
97       exit!
98     end
99
100     render :text => doc.to_s, :content_type => "text/xml"
101
102   end
103
104   def map
105     GC.start
106     @@count+=1
107
108     # Figure out the bbox
109     bbox = params['bbox']
110     unless bbox and bbox.count(',') == 3
111       report_error("The parameter bbox is required, and must be of the form min_lon,min_lat,max_lon,max_lat")
112       return
113     end
114
115     bbox = bbox.split(',')
116
117     min_lon = bbox[0].to_f
118     min_lat = bbox[1].to_f
119     max_lon = bbox[2].to_f
120     max_lat = bbox[3].to_f
121
122     # check the bbox is sane
123     unless min_lon <= max_lon
124       report_error("The minimum longitude must be less than the maximum longitude, but it wasn't")
125       return
126     end
127     unless min_lat <= max_lat
128       report_error("The minimum latitude must be less than the maximum latitude, but it wasn't")
129       return
130     end
131     unless min_lon >= -180 && min_lat >= -90 && max_lon <= 180 && max_lat <= 90
132       report_error("The latitudes must be between -90 and 90, and longitudes between -180 and 180")
133       return
134     end
135
136     # check the bbox isn't too large
137     requested_area = (max_lat-min_lat)*(max_lon-min_lon)
138     if requested_area > MAX_REQUEST_AREA
139       report_error("The maximum bbox size is " + MAX_REQUEST_AREA.to_s + ", and your request was too large. Either request a smaller area, or use planet.osm")
140       return
141     end
142
143     # get all the nodes
144     nodes = Node.find(:all, :conditions => ['latitude BETWEEN ? AND ? AND longitude BETWEEN ? AND ? AND visible = 1', min_lat, max_lat, min_lon, max_lon])
145
146     node_ids = nodes.collect {|node| node.id }
147
148     if node_ids.length > 50_000
149       report_error("You requested too many nodes (limit is 50,000). Either request a smaller area, or use planet.osm")
150       return
151     end
152
153     if node_ids.length == 0
154       render :text => "<osm version='0.4'></osm>", :content_type => "text/xml"
155       return
156     end
157
158     # grab the segments
159     segments = Array.new
160     if node_ids.length > 0
161       node_ids_sql = "(#{node_ids.join(',')})"
162       # get the referenced segments
163       segments = Segment.find_by_sql "select * from current_segments where visible = 1 and (node_a in #{node_ids_sql} or node_b in #{node_ids_sql})"
164     end
165     # see if we have any missing nodes
166     segments_nodes = segments.collect {|segment| segment.node_a }
167     segments_nodes += segments.collect {|segment| segment.node_b }
168
169     segments_nodes.uniq!
170
171     missing_nodes = segments_nodes - node_ids
172
173     # get missing nodes if there are any
174     nodes += Node.find(missing_nodes) if missing_nodes.length > 0
175
176     doc = OSM::API.new.get_xml_doc
177
178     # get ways
179     # find which ways are needed
180     segment_ids = segments.collect {|segment| segment.id }
181     ways = Array.new
182     if segment_ids.length > 0
183       way_segments = WaySegment.find_all_by_segment_id(segment_ids)
184       way_ids = way_segments.collect {|way_segment| way_segment.id }
185       ways = Way.find(way_ids) # NB: doesn't pick up segments, tags from db until accessed via way.way_segments etc.
186
187       # seg_ids = way_segments.collect {|way_segment| way_segment.segment_id }
188
189       list_of_way_segs = ways.collect {|way| way.way_segments}
190       list_of_way_segs.flatten!
191
192       list_of_way_segments = list_of_way_segs.collect { |way_seg| way_seg.segment_id }
193
194         else
195           list_of_way_segments = Array.new
196     end
197
198     # - [0] in case some thing links to segment 0 which doesn't exist. Shouldn't actually ever happen but it does. FIXME: file a ticket for this
199     segments_to_fetch = (list_of_way_segments.uniq - segment_ids) - [0]
200
201     if segments_to_fetch.length > 0
202       segments += Segment.find(segments_to_fetch)
203     end
204
205     # get more nodes
206     #
207
208     segments_nodes = segments.collect {|segment| segment.node_a }
209     segments_nodes += segments.collect {|segment| segment.node_b }
210
211     node_ids_a = nodes.collect {|node| node.id }
212
213     nodes_to_get = segments_nodes - node_ids_a
214     nodes += Node.find(nodes_to_get) if nodes_to_get.length > 0
215
216     visible_nodes = {}
217     user_display_name_cache = {}
218
219     nodes.each do |node|
220       if node.visible?
221         doc.root << node.to_xml_node(user_display_name_cache)
222         visible_nodes[node.id] = node
223       end
224     end
225
226     visible_segments = {}
227
228     segments.each do |segment|
229       if visible_nodes[segment.node_a] and visible_nodes[segment.node_b] and segment.visible?
230         doc.root << segment.to_xml_node(user_display_name_cache) 
231         visible_segments[segment.id] = segment
232       end
233     end
234
235     ways.each do |way|
236       doc.root << way.to_xml_node(visible_segments, user_display_name_cache) if way.visible?
237     end 
238
239     render :text => doc.to_s, :content_type => "text/xml"
240     
241     #exit when we have too many requests
242     if @@count > MAX_COUNT
243       @@count = COUNT
244       
245       exit!
246     end
247   end
248
249   def capabilities
250     doc = OSM::API.new.get_xml_doc
251
252     api = XML::Node.new 'api'
253     version = XML::Node.new 'version'
254     version['minimum'] = '0.4';
255     version['maximum'] = '0.4';
256     api << version
257     area = XML::Node.new 'area'
258     area['maximum'] = MAX_REQUEST_AREA.to_s;
259     api << area
260     
261     doc.root << api
262
263     render :text => doc.to_s, :content_type => "text/xml"
264   end
265 end