ruby-grads 1.0.0 → 1.0.7

data/lib/grads/command.rb

@@ -75,8 +75,15 @@ class GrADS::Command
       return @expr
     end
     alias to_str to_s
-    def [] (hash)
-      spec = ["(",hash.map{ |k,v| [k,"=",v].join("") }.join(","),")"].join("")
+    def [] (expr)
+      case expr
+      when Hash
+        spec = ["(",hash.map{ |k,v| [k,"=",v].join("") }.join(","),")"].join("")
+      when String
+        spec = "("+expr+")"
+      else
+        raise "unknown expression"
+      end
       return Expression.new([@expr,spec].join(""))
     end
     def method_missing (id, *argv)
@@ -176,7 +183,11 @@ class GrADS::Command
   attr_reader :log
 
   def method_missing (id, *argv)
-    return id, *argv
+    if argv.empty?
+      return id
+    else
+      super
+    end
   end
 
   def quit
@@ -402,7 +413,6 @@ class GrADS::Command
     "reinit",
     "reset",
     "run",
-    "sdfwrite",
     "swap",
     "undefine",
   ].each do |name|
@@ -413,9 +423,21 @@ class GrADS::Command
      }  
   end
 
+  [
+    "sdfwrite",
+  ].each do |name|
+     class_eval %{
+       def #{name} (*argv)
+         if argv.empty?
+           return :#{name}
+         else
+           put "#{name} " + argv.join(" ")
+         end
+       end
+     }  
+  end
+
   def clear (*argv)
-    @ccols = nil
-    @clevs = nil
     put "clear " + argv.join(" ")
   end
 
@@ -518,13 +540,20 @@ class GrADS::Command
   end
 
   def set_clevs (*argv)
-    @clevs = argv
     set :clevs, *argv
   end
 
   def set_ccols (*argv)
-    @ccols = argv
-    set :ccols, *argv
+    list = []
+    argv.each do |arg|
+      case arg
+      when RGB
+        list << set_rgb(arg)
+      else
+        list << arg
+      end
+    end
+    set :ccols, *list
   end
   
   def import (*args)
@@ -646,7 +675,7 @@ class GrADS::Command
   end
 
   begin
-    require "netcdflib"
+    require "simple-netcdf"
     def get_var (expr)
       @var_id ||= 0
       @var_id += 1
@@ -730,6 +759,395 @@ class GrADS::Command
     end
   end
 
+  class Area
+
+    def initialize (x1, x2, y1, y2)
+      @x1 = x1
+      @x2 = x2
+      @y1 = y1
+      @y2 = y2
+    end
+    
+    attr_reader :x1, :x2, :y1, :y2
+    
+    def w
+      return @x2 - @x1
+    end
+    
+    def h
+      return @y2 - @y1
+    end
+    
+    def center
+      return @x1 + w/2
+    end
+    
+    def mid
+      return @y1 + h/2
+    end
+
+  end
+
+  def len (str)
+    ans = query("string #{str}")
+    return subwrd(ans, 4).to_f
+  end
+
+  def parea (x1, x2, y1, y2)
+    set :parea, x1, x2, y1, y2
+    set :clip, x1, x2, y1, y2
+    yield Area.new(x1, x2, y1, y2)
+  end
+
+  def vpage (x1, x2, y1, y2)
+    set :vpage, x1, x2, y1, y2
+    yield Area.new(x1, x2, y1, y2)
+  end
+  
+  def display_nothing (dummy)
+    ans = query("dims")
+    case ans.scan("varying").size
+    when 2
+      set :gxout, :contour
+      set :clevs, -9.99e99
+      set :ccolor, -1
+      set :clab, :off
+      set :cmark, 0
+    when 1
+      set :gxout, :line
+      set :line, -1
+      set :ylab, :off
+      set :vrange, -9.99e99, -9.99e98
+    end
+    if block_given?
+      yield
+    end
+    case dummy
+    when String
+      put "display #{dummy}"
+    else
+      display dummy
+    end
+  end
+
+  def axis_off
+    set :frame, :off
+    set :xlab, :off
+    set :ylab, :off
+    set :grid, :off
+    set :grads, :off
+    set :mpdraw, :off
+  end
+
+  def axis_on
+    set :frame, :on
+    set :xlab, :on
+    set :ylab, :on
+    set :grid, :on
+    set :grads, :on
+    set :mpdraw, :on
+  end
+  
+  def plot (var, *args)
+    begin
+      axis_off
+      if block_given?
+        yield
+      end
+      set :gxout, *args
+      case var
+      when String
+        put "display #{var}"
+      else
+        display var
+      end
+    ensure
+        axis_on
+    end
+  end
+  
+  def axis (dummy, &block)
+    begin
+      axis_on
+      display_nothing(dummy, &block)
+    ensure
+      axis_off
+    end
+  end
+
+  def draw_ylabel (text, area:, color: 1, just: "bc", thickness: 3, rot: 90, offset: "M", side: "l")
+    set :string, color, just, thickness, rot
+    case side
+    when "l"
+      draw :string, area.x1-len(offset), area.mid, text
+    when "r"
+      draw :string, area.x2+len(offset), area.mid, text
+    else
+      raise "invalid side sepecification"
+    end
+  end
+
+  def draw_xlabel (text, area:, color: 1, just: "bc", thickness: 3, rot: 0, offset: "M", side: "b")
+    set :string, color, just, thickness, rot
+    case side
+    when "b"
+      draw :string, area.center, area.y1-len(offset), text
+    when "t"
+      draw :string, area.center, area.y2+len(offset), text
+    else
+      raise "invalid side sepecification"
+    end
+  end
+  
+  COLOR_TABLE = {
+    "background" => 0,
+    "foregraound" => 1,
+    "red" => 2,
+    "green" => 3,
+    "dark blue" => 4,
+    "light blue" => 5,
+    "magenta" => 6,
+    "yellow" => 7,
+    "orange" => 8,
+    "purple" => 9,
+    "yellow green" => 10,
+    "medium blue" => 11,
+    "dark yellow" => 12,
+    "aqua" => 13,
+    "dark purple" => 14,
+    "gray" => 15,
+  }
+  
+  def COLOR (name)
+    if COLOR_TABLE.has_key?(name)
+      COLOR_TABLE[name]
+    else
+      -1
+    end
+  end
+
+  class LINE
+
+    def initialize (type = 1, color: 1, thickness: 3)
+      case type
+      when "","solid"
+        @type    = 1
+      when "--", "long dash"
+        @type    = 2
+      when "-", "short dash"
+        @type    = 3
+      when "-- -", "long dash short dash"
+        @type    = 4
+      when ".", "dotted"
+        @type    = 5
+      when ". -", "dot dash"
+        @type    = 6
+      when ". . -", "dot dot dash"
+        @type    = 7
+      else
+        @type    = type
+      end
+      @color = color
+      @thickness    = thickness
+    end
+    
+    attr_reader :type, :color
+    
+    def to_a
+      return [@color, @type, @thickness]
+    end
+   
+  end
+  
+  class TILE
+
+    def self.reset
+      @@number = 0
+    end
+    
+    def self.next_number
+      @@number += 1
+      if @@number > 2047
+        reset
+      end
+      return @@number
+    end
+
+    reset
+    
+    def initialize (type, width: 6, height: nil, thickness: 3, color: 1, bgcolor: -1)
+      case type
+      when "."
+        @type    = 2
+      when "\\"
+        @type    = 3
+      when "/"
+        @type    = 4
+      when "x"
+        @type    = 5
+      when "|"
+        @type    = 6
+      when "-"
+        @type    = 7
+      when "#", "+"
+        @type    = 8
+      else
+        @type    = type
+      end
+      @width   = width
+      @height  = height || width
+      @thickness    = thickness
+      @color   = color
+      @bgcolor = bgcolor
+    end
+    
+    attr_reader :type, :color
+    
+    def to_a
+      return [@type, @width, @height, @thickness, @color, @bgcolor]
+    end
+   
+  end
+  
+  class RGB
+    
+    @@number = nil
+    @@table = {}
+    
+    def self.reset
+      @@number = 100
+    end
+    
+    def self.next_number
+      @@number += 1
+      if @@number > 2047
+        reset
+      end
+      return @@number
+    end
+    
+    reset
+    
+    def initialize (r, g, b, a=255)
+      @r = r
+      @g = g
+      @b = b
+      @a = a
+    end
+    
+    attr_reader :r, :g, :b, :a
+    
+    def to_s
+      return format("%i %i %i %i", @r, @g, @b, @a)
+    end
+
+    def to_a
+      return [@r, @g, @b, @a]
+    end
+    
+    def % (alpha)
+      if alpha >= 0 and alpha <= 1
+        a = (@a*alpha).to_i
+      else
+        raise "invalid alpha value"
+      end
+      return RGB.new(@r, @g, @b, a)
+    end
+
+    def * (scale)
+      if scale >= 0 and scale <= 1
+        return RGB.new((@r*scale).to_i, (@g*scale).to_i, (@b*scale).to_i, @a)
+      else
+        raise "invalid scale value"
+      end
+    end
+
+    def / (scale)
+      if scale >= 1
+        return RGB.new((@r/scale).to_i, (@g/scale).to_i, (@b/scale).to_i, @a)
+      else
+        raise "invalid scale value"
+      end
+    end
+    
+    def + (other)
+      r = [self.r, other.r].max
+      g = [self.g, other.g].max
+      b = [self.b, other.b].max
+      a = [self.a, other.a].max
+      return RGB.new(r,g,b,a)
+    end
+    
+  end
+
+  def LINE (*argv)
+    return LINE.new(*argv)
+  end
+
+  def TILE (*argv)
+    return TILE.new(*argv)
+  end
+
+  def RGB (*argv)
+    case argv.size
+    when 1
+      return GrADS::Command::RGB_TABLE[argv[0]]
+    else
+      return GrADS::Command::RGB.new(*argv)
+    end
+  end
+
+  def self.RGB (*argv)
+    case argv.size
+    when 1
+      return GrADS::Command::RGB_TABLE[argv[0]]
+    else
+      return GrADS::Command::RGB.new(*argv)
+    end
+  end
+
+  RGB_TABLE = {
+    "red"   => RGB(255,0,0),
+    "green" => RGB(0,255,0),
+    "blue"  => RGB(0,0,255),
+  }
+  RGB_TABLE.update({
+    "cyan"  => RGB("green") + RGB("blue"),
+  })
+
+  def set_line (obj)
+    args = obj.to_a
+    if obj.color.is_a?(RGB)
+      args[0] = set_rgb(obj.color)
+    end    
+    set :line, *args
+    set :ccolor, args[0]
+    set :cstyle, args[1]
+    set :cthick, args[2]
+  end
+
+  def set_rgb (*argv)
+    list = []
+    argv.each do |obj|
+      num = RGB.next_number
+      case obj
+      when TILE
+        nt = TILE.next_number
+        args = obj.to_a
+        if obj.color.is_a?(RGB)
+          c = set_rgb(obj.color)
+          args[4] = c
+        end
+        set :tile, nt, *args.to_a
+        set :rgb, num, :tile, nt
+      when RGB
+        set :rgb, num, *obj.to_a        
+      end
+      list << num
+    end
+    return list
+  end
+  
 end
 
 def GrADS.start (*argv, &block)
@@ -748,4 +1166,3 @@ def GrADS.script (name, definition)
 end
 
 
-

data/lib/grads/gridded.rb

@@ -34,6 +34,16 @@
 require "carray"
 require "strscan"
 
+begin
+  require "simple-proj"
+  require "simple-proj-carray"
+rescue LoadError
+  begin
+    require "proj4r"
+  rescue LoadError
+  end
+end
+
 module GrADS
 end
 
@@ -128,9 +138,10 @@ end
 
 class GrADS::Gridded
 
-  def initialize (ctl_file)
+  def initialize (ctl_file, radius: nil)
     @ctl_file = ctl_file
     @entries = scan_ctl_file(ctl_file)
+    @radius = radius
     parse_entries(@entries)
   end
 
@@ -319,7 +330,13 @@ class GrADS::Gridded
   end
 
   def parse_pdef_lcc (args)
-    require "proj4r"
+    if defined? PROJ4 and defined? PROJ4::Proj
+      projlib = "proj4r"
+    elsif defined? PROJ 
+      projlib = "simple-proj"
+    else
+      raise "can't find proj related library"
+    end
     rlat  = args[1].to_f
     rlon  = args[2].to_f
     xi    = args[3].to_f
@@ -330,21 +347,61 @@ class GrADS::Gridded
     dx    = args[8].to_f
     dy    = args[9].to_f
     slat0 = slat1 > 0 ? 90 : -90
-    proj = PROJ4::Proj.new %{
-      +ellps=sphere +proj=lcc \
-      +lat_1=#{slat1} +lat_2=#{slat2} +lat_0=90 +lon_0=#{slon}
-    }
+    case projlib
+    when "proj4r"
+      if @radius
+        proj = PROJ4::Proj.new %{
+          +ellps=sphere +a=#{@radius} +b=#{@radius} +proj=lcc \
+          +lat_1=#{slat1} +lat_2=#{slat2} +lon_0=#{slon}
+        }
+      else
+        proj = PROJ4::Proj.new %{
+          +ellps=sphere +proj=lcc \
+          +lat_1=#{slat1} +lat_2=#{slat2} +lon_0=#{slon}
+        }
+      end
+    when "simple-proj"
+      if @radius
+        proj = PROJ.new %{
+          +ellps=sphere +a=#{@radius} +b=#{@radius} +proj=lcc \
+          +lat_1=#{slat1} +lat_2=#{slat2} +lon_0=#{slon}
+        }
+      else
+        proj = PROJ.new %{
+          +ellps=sphere +proj=lcc \
+          +lat_1=#{slat1} +lat_2=#{slat2} +lon_0=#{slon}
+        }
+      end
+    end
     rx, ry = proj.forward(rlon, rlat)
     rx0 = rx - (xi-1)*dx
     ry0 = ry - (yi-1)*dy
-    @forward = lambda { |lon, lat|
-      mx, my = proj.forward(lon, lat)
-      [(mx - rx0)/dx, (my - ry0)/dy]
-    }
-    @inverse = lambda { |fi, fj|
-      mx = fi*dx + rx0
-      my = fj*dy + ry0
-      proj.inverse(mx, my)
+    case projlib
+    when "proj4r"
+      @forward = lambda { |lon, lat|
+        mx, my = proj.forward(lon, lat)
+        [(mx - rx0)/dx, (my - ry0)/dy]
+      }
+      @inverse = lambda { |fi, fj|
+        mx = fi*dx + rx0
+        my = fj*dy + ry0
+        proj.inverse(mx, my)
+      }      
+    when "simple-proj"
+      @forward = lambda { |lon, lat|
+        mx, my = proj.forward_carray(lon, lat)
+        [(mx - rx0)/dx, (my - ry0)/dy]
+      }
+      @inverse = lambda { |fi, fj|
+        mx = fi*dx + rx0
+        my = fj*dy + ry0
+        proj.inverse_carray(mx, my)
+      }
+    end
+    @xy = lambda {
+      x = CArray.float(xsize).seq(rx0, dx)
+      y = CArray.float(ysize).seq(ry0, dy)
+      [x, y]
     }
   end
 
@@ -495,6 +552,18 @@ class GrADS::Gridded
   def forward (rlon, rlat)
     return @forward[rlon, rlat]
   end
+  
+  def lonlat
+    fi = CArray.float(xsize).seq
+    fj = CArray.float(ysize).seq
+    ii = fi[ysize,:%]
+    jj = fj[:%,xsize]
+    return @inverse[ii, jj]
+  end
+
+  def xy
+    return @xy[]
+  end
 
   def var (name)
     unless @varnames.include?(name)