X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/ef7f3d800cbdd49b692df10d312e5fd880e2e938..43db18ae911fef3ad7b36e9ed27e5eff469ef863:/lib/gpx.rb diff --git a/lib/gpx.rb b/lib/gpx.rb index 0a7a88c10..3e1cb9afa 100644 --- a/lib/gpx.rb +++ b/lib/gpx.rb @@ -4,36 +4,29 @@ module GPX include LibXML - attr_reader :possible_points - attr_reader :actual_points - attr_reader :tracksegs + attr_reader :possible_points, :actual_points, :tracksegs - def initialize(file) + def initialize(file, options = {}) @file = file + @maximum_points = options[:maximum_points] || Float::INFINITY end - def points - @possible_points = 0 - @actual_points = 0 - @tracksegs = 0 - - @file.rewind - - reader = XML::Reader.io(@file) - + def parse_file(reader) point = nil while reader.read - if reader.node_type == XML::Reader::TYPE_ELEMENT + case reader.node_type + when XML::Reader::TYPE_ELEMENT if reader.name == "trkpt" point = TrkPt.new(@tracksegs, reader["lat"].to_f, reader["lon"].to_f) @possible_points += 1 + raise FileTooBigError if @possible_points > @maximum_points elsif reader.name == "ele" && point point.altitude = reader.read_string.to_f elsif reader.name == "time" && point - point.timestamp = DateTime.parse(reader.read_string) + point.timestamp = Time.parse(reader.read_string).utc end - elsif reader.node_type == XML::Reader::TYPE_END_ELEMENT + when XML::Reader::TYPE_END_ELEMENT if reader.name == "trkpt" && point && point.valid? point.altitude ||= 0 yield point @@ -45,79 +38,92 @@ module GPX end end + def points(&block) + return enum_for(:points) unless block + + @possible_points = 0 + @actual_points = 0 + @tracksegs = 0 + + begin + Archive::Reader.open_filename(@file).each_entry_with_data do |entry, data| + parse_file(XML::Reader.string(data), &block) if entry.regular? + end + rescue Archive::Error + io = ::File.open(@file) + + case Marcel::MimeType.for(io) + when "application/gzip" then io = Zlib::GzipReader.open(@file) + when "application/x-bzip" then io = Bzip2::FFI::Reader.open(@file) + end + + parse_file(XML::Reader.io(io), &block) + end + end + def picture(min_lat, min_lon, max_lat, max_lon, num_points) - frames = 10 + nframes = 10 width = 250 height = 250 - proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height) + delay = 50 + + points_per_frame = (num_points.to_f / nframes).ceil - linegc = Magick::Draw.new - linegc.stroke_linejoin('miter') - linegc.stroke_width(1) - linegc.stroke('#BBBBBB') - linegc.fill('#BBBBBB') + proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height) - highlightgc = Magick::Draw.new - highlightgc.stroke_linejoin('miter') - highlightgc.stroke_width(3) - highlightgc.stroke('#000000') - highlightgc.fill('#000000') + frames = [] - images = [] + (0...nframes).each do |n| + frames[n] = GD2::Image::IndexedColor.new(width, height) + black = frames[n].palette.allocate(GD2::Color[0, 0, 0]) + white = frames[n].palette.allocate(GD2::Color[255, 255, 255]) + grey = frames[n].palette.allocate(GD2::Color[187, 187, 187]) - frames.times do - image = Magick::Image.new(width, height) do |image| - image.background_color = 'white' - image.format = 'GIF' + frames[n].draw do |pen| + pen.color = white + pen.rectangle(0, 0, width, height, true) end - images << image - end + frames[n].draw do |pen| + pen.color = black + pen.anti_aliasing = true + pen.dont_blend = false - oldpx = 0.0 - oldpy = 0.0 + oldpx = 0.0 + oldpy = 0.0 - first = true + first = true - m = 0 - mm = 0 - points do |p| - px = proj.x(p.longitude) - py = proj.y(p.latitude) + points.each_with_index do |p, pt| + px = proj.x(p.longitude) + py = proj.y(p.latitude) - if m > 0 - frames.times do |n| - if n == mm - gc = highlightgc.dup + if (pt >= (points_per_frame * n)) && (pt <= (points_per_frame * (n + 1))) + pen.thickness = 3 + pen.color = black else - gc = linegc.dup + pen.thickness = 1 + pen.color = grey end - gc.line(px, py, oldpx, oldpy) - - gc.draw(images[n]) + pen.line(px, py, oldpx, oldpy) unless first + first = false + oldpy = py + oldpx = px 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 + image = GD2::AnimatedGif.new + image.add(frames.first) + frames.each do |frame| + image.add(frame, :delay => delay) end + image.end - il.delay = 50 - il.format = 'GIF' - - il.to_blob + output = StringIO.new + image.export(output) + output end def icon(min_lat, min_lon, max_lat, max_lon) @@ -125,44 +131,53 @@ module GPX 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 = GD2::Image::IndexedColor.new(width, height) + + black = image.palette.allocate(GD2::Color[0, 0, 0]) + white = image.palette.allocate(GD2::Color[255, 255, 255]) - image = Magick::Image.new(width, height) do |image| - image.background_color = 'white' - image.format = 'GIF' + image.draw do |pen| + pen.color = white + pen.rectangle(0, 0, width, height, true) end - oldpx = 0.0 - oldpy = 0.0 + image.draw do |pen| + pen.color = black + pen.anti_aliasing = true + pen.dont_blend = false - first = true + oldpx = 0.0 + oldpy = 0.0 - points do |p| - px = proj.x(p.longitude) - py = proj.y(p.latitude) + first = true - gc.dup.line(px, py, oldpx, oldpy).draw(image) unless first + points do |p| + px = proj.x(p.longitude) + py = proj.y(p.latitude) - first = false - oldpy = py - oldpx = px + pen.line(px, py, oldpx, oldpy) unless first + + first = false + oldpy = py + oldpx = px + end end - image.to_blob + StringIO.new(image.gif) end end - private - - class TrkPt < Struct.new(:segment, :latitude, :longitude, :altitude, :timestamp) + TrkPt = Struct.new(:segment, :latitude, :longitude, :altitude, :timestamp) do def valid? latitude && longitude && timestamp && latitude >= -90 && latitude <= 90 && longitude >= -180 && longitude <= 180 end end + + class FileTooBigError < RuntimeError + def initialise + super("GPX File contains too many points") + end + end end