openstreetmap-website/lib/osm.rb

462 lines
12 KiB
Ruby

# The OSM module provides support functions for OSM.
module OSM
require 'time'
require 'rexml/parsers/sax2parser'
require 'rexml/text'
require 'xml/libxml'
require 'digest/md5'
require 'RMagick'
# The base class for API Errors.
class APIError < RuntimeError
def render_opts { :text => "", :status => :internal_server_error } end
end
# Raised when an API object is not found.
class APINotFoundError < APIError
end
# Raised when a precondition to an API action fails sanity check.
class APIPreconditionFailedError < APIError
def render_opts { :text => "", :status => :precondition_failed } end
end
# Raised when to delete an already-deleted object.
class APIAlreadyDeletedError < APIError
def render_opts { :text => "", :status => :gone } end
end
# Raised when the provided version is not equal to the latest in the db.
class APIVersionMismatchError < APIError
def initialize(provided, latest)
@provided, @latest = provided, latest
end
attr_reader :provided, :latest
def render_opts
{ :text => "Version mismatch: Provided " + ex.provided.to_s +
", server had: " + ex.latest.to_s, :status => :bad_request }
end
end
# Helper methods for going to/from mercator and lat/lng.
class Mercator
include Math
#init me with your bounding box and the size of your image
def initialize(min_lat, min_lon, max_lat, max_lon, width, height)
xsize = xsheet(max_lon) - xsheet(min_lon)
ysize = ysheet(max_lat) - ysheet(min_lat)
xscale = xsize / width
yscale = ysize / height
scale = [xscale, yscale].max
xpad = width * scale - xsize
ypad = height * scale - ysize
@width = width
@height = height
@tx = xsheet(min_lon) - xpad / 2
@ty = ysheet(min_lat) - ypad / 2
@bx = xsheet(max_lon) + xpad / 2
@by = ysheet(max_lat) + ypad / 2
end
#the following two functions will give you the x/y on the entire sheet
def ysheet(lat)
log(tan(PI / 4 + (lat * PI / 180 / 2))) / (PI / 180)
end
def xsheet(lon)
lon
end
#and these two will give you the right points on your image. all the constants can be reduced to speed things up. FIXME
def y(lat)
return @height - ((ysheet(lat) - @ty) / (@by - @ty) * @height)
end
def x(lon)
return ((xsheet(lon) - @tx) / (@bx - @tx) * @width)
end
end
# This piece of magic reads a GPX with SAX and spits out
# lat/lng and stuff
#
# This would print every latitude value:
#
# gpx = OSM::GPXImporter.new('somefile.gpx')
# gpx.points {|p| puts p['latitude']}
class GPXImporter
# FIXME swap REXML for libXML
attr_reader :possible_points
attr_reader :actual_points
attr_reader :tracksegs
def initialize(file)
@file = file
end
def points
@possible_points = 0
@actual_points = 0
@tracksegs = 0
lat = -1
lon = -1
ele = -1
date = DateTime.now();
gotlatlon = false
gotele = false
gotdate = false
@file.rewind
parser = REXML::Parsers::SAX2Parser.new(@file)
parser.listen( :start_element, %w{ trkpt }) do |uri,localname,qname,attributes|
lat = attributes['lat'].to_f
lon = attributes['lon'].to_f
gotlatlon = true
gotele = false
gotdate = false
@possible_points += 1
end
parser.listen( :characters, %w{ ele } ) do |text|
ele = text
gotele = true
end
parser.listen( :characters, %w{ time } ) do |text|
if text && text != ''
begin
date = DateTime.parse(text)
gotdate = true
rescue
end
end
end
parser.listen( :end_element, %w{ trkseg } ) do |uri, localname, qname|
@tracksegs += 1
end
parser.listen( :end_element, %w{ trkpt } ) do |uri,localname,qname|
if gotlatlon && gotdate
ele = '0' unless gotele
if lat < 90 && lat > -90 && lon > -180 && lon < 180
@actual_points += 1
yield Hash['latitude' => lat, 'longitude' => lon, 'timestamp' => date, 'altitude' => ele, 'segment' => @tracksegs]
end
end
gotlatlon = false
gotele = false
gotdate = false
end
parser.parse
end
def get_picture(min_lat, min_lon, max_lat, max_lon, num_points)
#puts "getting picfor bbox #{min_lat},#{min_lon} - #{max_lat},#{max_lon}"
frames = 10
width = 250
height = 250
proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height)
linegc = Magick::Draw.new
linegc.stroke_linejoin('miter')
linegc.stroke_width(1)
linegc.stroke('#BBBBBB')
linegc.fill('#BBBBBB')
highlightgc = Magick::Draw.new
highlightgc.stroke_linejoin('miter')
highlightgc.stroke_width(3)
highlightgc.stroke('#000000')
highlightgc.fill('#000000')
images = []
frames.times do
image = Magick::Image.new(width, height) do |image|
image.background_color = 'white'
image.format = 'GIF'
end
images << image
end
oldpx = 0.0
oldpy = 0.0
first = true
m = 0
mm = 0
points do |p|
px = proj.x(p['longitude'])
py = proj.y(p['latitude'])
if m > 0
frames.times do |n|
if n == mm
gc = highlightgc.dup
else
gc = linegc.dup
end
gc.line(px, py, oldpx, oldpy)
gc.draw(images[n])
end
end
m += 1
if m > num_points.to_f / frames.to_f * (mm+1)
mm += 1
end
oldpy = py
oldpx = px
end
il = Magick::ImageList.new
images.each do |f|
il << f
end
il.delay = 50
il.format = 'GIF'
return il.to_blob
end
def get_icon(min_lat, min_lon, max_lat, max_lon)
#puts "getting icon for bbox #{min_lat},#{min_lon} - #{max_lat},#{max_lon}"
width = 50
height = 50
proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height)
gc = Magick::Draw.new
gc.stroke_linejoin('miter')
gc.stroke_width(1)
gc.stroke('#000000')
gc.fill('#000000')
image = Magick::Image.new(width, height) do |image|
image.background_color = 'white'
image.format = 'GIF'
end
oldpx = 0.0
oldpy = 0.0
first = true
points do |p|
px = proj.x(p['longitude'])
py = proj.y(p['latitude'])
gc.dup.line(px, py, oldpx, oldpy).draw(image) unless first
first = false
oldpy = py
oldpx = px
end
return image.to_blob
end
end
class GreatCircle
include Math
# initialise with a base position
def initialize(lat, lon)
@lat = lat * PI / 180
@lon = lon * PI / 180
end
# get the distance from the base position to a given position
def distance(lat, lon)
lat = lat * PI / 180
lon = lon * PI / 180
return 6372.795 * 2 * asin(sqrt(sin((lat - @lat) / 2) ** 2 + cos(@lat) * cos(lat) * sin((lon - @lon)/2) ** 2))
end
# get the worst case bounds for a given radius from the base position
def bounds(radius)
latradius = 2 * asin(sqrt(sin(radius / 6372.795 / 2) ** 2))
lonradius = 2 * asin(sqrt(sin(radius / 6372.795 / 2) ** 2 / cos(@lat) ** 2))
minlat = (@lat - latradius) * 180 / PI
maxlat = (@lat + latradius) * 180 / PI
minlon = (@lon - lonradius) * 180 / PI
maxlon = (@lon + lonradius) * 180 / PI
return { :minlat => minlat, :maxlat => maxlat, :minlon => minlon, :maxlon => maxlon }
end
end
class GeoRSS
def initialize(feed_title='OpenStreetMap GPS Traces', feed_description='OpenStreetMap GPS Traces', feed_url='http://www.openstreetmap.org/traces/')
@doc = XML::Document.new
@doc.encoding = 'UTF-8'
rss = XML::Node.new 'rss'
@doc.root = rss
rss['version'] = "2.0"
rss['xmlns:geo'] = "http://www.w3.org/2003/01/geo/wgs84_pos#"
@channel = XML::Node.new 'channel'
rss << @channel
title = XML::Node.new 'title'
title << feed_title
@channel << title
description_el = XML::Node.new 'description'
@channel << description_el
description_el << feed_description
link = XML::Node.new 'link'
link << feed_url
@channel << link
image = XML::Node.new 'image'
@channel << image
url = XML::Node.new 'url'
url << 'http://www.openstreetmap.org/images/mag_map-rss2.0.png'
image << url
title = XML::Node.new 'title'
title << "OpenStreetMap"
image << title
width = XML::Node.new 'width'
width << '100'
image << width
height = XML::Node.new 'height'
height << '100'
image << height
link = XML::Node.new 'link'
link << feed_url
image << link
end
def add(latitude=0, longitude=0, title_text='dummy title', author_text='anonymous', url='http://www.example.com/', description_text='dummy description', timestamp=DateTime.now)
item = XML::Node.new 'item'
title = XML::Node.new 'title'
item << title
title << title_text
link = XML::Node.new 'link'
link << url
item << link
guid = XML::Node.new 'guid'
guid << url
item << guid
description = XML::Node.new 'description'
description << description_text
item << description
author = XML::Node.new 'author'
author << author_text
item << author
pubDate = XML::Node.new 'pubDate'
pubDate << timestamp.to_s(:rfc822)
item << pubDate
if latitude
lat_el = XML::Node.new 'geo:lat'
lat_el << latitude.to_s
item << lat_el
end
if longitude
lon_el = XML::Node.new 'geo:long'
lon_el << longitude.to_s
item << lon_el
end
@channel << item
end
def to_s
return @doc.to_s
end
end
class API
def get_xml_doc
doc = XML::Document.new
doc.encoding = 'UTF-8'
root = XML::Node.new 'osm'
root['version'] = API_VERSION
root['generator'] = 'OpenStreetMap server'
doc.root = root
return doc
end
end
def self.IPLocation(ip_address)
Timeout::timeout(4) do
Net::HTTP.start('api.hostip.info') do |http|
country = http.get("/country.php?ip=#{ip_address}").body
country = "GB" if country == "UK"
Net::HTTP.start('ws.geonames.org') do |http|
xml = REXML::Document.new(http.get("/countryInfo?country=#{country}").body)
xml.elements.each("geonames/country") do |ele|
minlon = ele.get_text("bBoxWest").to_s
minlat = ele.get_text("bBoxSouth").to_s
maxlon = ele.get_text("bBoxEast").to_s
maxlat = ele.get_text("bBoxNorth").to_s
return { :minlon => minlon, :minlat => minlat, :maxlon => maxlon, :maxlat => maxlat }
end
end
end
end
return nil
rescue Exception
return nil
end
# Construct a random token of a given length
def self.make_token(length = 30)
chars = 'abcdefghijklmnopqrtuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789'
token = ''
length.times do
token += chars[(rand * chars.length).to_i].chr
end
return token
end
# Return an encrypted version of a password
def self.encrypt_password(password, salt)
return Digest::MD5.hexdigest(password) if salt.nil?
return Digest::MD5.hexdigest(salt + password)
end
# Return an SQL fragment to select a given area of the globe
def self.sql_for_area(minlat, minlon, maxlat, maxlon, prefix = nil)
tilesql = QuadTile.sql_for_area(minlat, minlon, maxlat, maxlon, prefix)
minlat = (minlat * 10000000).round
minlon = (minlon * 10000000).round
maxlat = (maxlat * 10000000).round
maxlon = (maxlon * 10000000).round
return "#{tilesql} AND #{prefix}latitude BETWEEN #{minlat} AND #{maxlat} AND #{prefix}longitude BETWEEN #{minlon} AND #{maxlon}"
end
end