app/controllers/api_controller.rb

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