Make a generic preferred color scheme method

[rails.git] / lib / gpx.rb

diff --git a/lib/gpx.rb b/lib/gpx.rb
index 9b93f07f2b5909929abdee3a49a1bbccd80857a4..45a4dcf5fc0bd62335ea0339409e2c1fbfd49e6d 100644 (file)
--- a/lib/gpx.rb
+++ b/lib/gpx.rb
@@ -4,37 +4,30 @@ module GPX
 
     include LibXML
 
 
     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
       @file = file

+      @maximum_points = options[:maximum_points] || Float::INFINITY

     end
 
     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
       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
           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
           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
           end

-        elsif reader.node_type == XML::Reader::TYPE_END_ELEMENT
-          if reader.name == "trkpt" && point && point.valid?
+        when XML::Reader::TYPE_END_ELEMENT
+          if reader.name == "trkpt" && point&.valid?

             point.altitude ||= 0
             yield point
             @actual_points += 1
             point.altitude ||= 0
             yield point
             @actual_points += 1


@@ -45,71 +38,92 @@ module GPX
       end
     end
 
       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, :options => XML::Parser::Options::NOERROR), &block)
+      end
+    end
+

     def picture(min_lat, min_lon, max_lat, max_lon, num_points)
     def picture(min_lat, min_lon, max_lat, max_lon, num_points)

-      frames = 10
+      nframes = 10

       width = 250
       height = 250
       width = 250
       height = 250

+      delay = 50
+
+      points_per_frame = (num_points.to_f / nframes).ceil
+

       proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height)
 
       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.collect do
-        Magick::Image.new(width, height) do |image|
-          image.background_color = "white"
-          image.format = "GIF"
+      frames = []
+
+      (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[n].draw do |pen|
+          pen.color = white
+          pen.rectangle(0, 0, width, height, true)

         end
         end

-      end

 
 

-      oldpx = 0.0
-      oldpy = 0.0
+        frames[n].draw do |pen|
+          pen.color = black
+          pen.anti_aliasing = true
+          pen.dont_blend = false
+
+          oldpx = 0.0
+          oldpy = 0.0

 
 

-      m = 0
-      mm = 0
-      points do |p|
-        px = proj.x(p.longitude)
-        py = proj.y(p.latitude)
+          first = true

 
 

-        if m > 0
-          frames.times do |n|
-            if n == mm
-              gc = highlightgc.dup
+          points.each_with_index do |p, pt|
+            px = proj.x(p.longitude)
+            py = proj.y(p.latitude)
+
+            if (pt >= (points_per_frame * n)) && (pt <= (points_per_frame * (n + 1)))
+              pen.thickness = 3
+              pen.color = black

             else
             else

-              gc = linegc.dup
+              pen.thickness = 1
+              pen.color = grey

             end
 
             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
           end
         end

-
-        m += 1
-        mm += 1 if m > num_points.to_f / frames.to_f * (mm + 1)
-
-        oldpy = py
-        oldpx = px

       end
 
       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
       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)
     end
 
     def icon(min_lat, min_lon, max_lat, max_lon)


@@ -117,44 +131,53 @@ module GPX
       height = 50
       proj = OSM::Mercator.new(min_lat, min_lon, max_lat, max_lon, width, height)
 
       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 |i|
-        i.background_color = "white"
-        i.format = "GIF"
+      image.draw do |pen|
+        pen.color = white
+        pen.rectangle(0, 0, width, height, true)

       end
 
       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
 
       end
 

-      image.to_blob
+      StringIO.new(image.gif)

     end
   end
 
     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
     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
 end