ruby-grads 1.0.0

data/lib/grads/lib/colorbar.rb

@@ -0,0 +1,192 @@
+#
+# colorbar :title=>"colorbar"
+#          :color=>[0,10,2]
+#          :clevs=>[]
+#          :ccols=>[],
+#          :kind=>"",
+#          :tics=>[0,2], 
+#          :pos=>[xmin,xmax,ymin,ymax],
+#          :font=>[fw,fh,ft]
+#          :edge=>"triangle"
+#          :dir=>"horizontal"
+#          :line=>true
+
+require "stringio"
+
+class GrADS::Command
+
+  COLORBAR_DEFAULT = {
+    :line=>true,
+    :edge=>"triangle",
+  }
+
+  def colorbar (opt={})
+    import :color, :xcbar
+    opt = COLORBAR_DEFAULT.clone.update(opt)
+    if opt[:color]
+      if opt[:color] == :clevs
+        arg = "-levs " + opt[:clevs].join(" ")
+      else
+        arg = [opt[:color]].flatten.join(" ")
+      end
+      if opt[:kind]
+        arg += " -kind " + opt[:kind]
+      end
+      out = color arg
+      clevs = ccols = nil
+      out.split(/\n/).each do |line|
+        case line 
+        when /clevs/
+          clevs = line.split(/ +/)[1..-1]
+        when /ccols/
+          ccols = line.split(/ +/)[1..-1]      
+        end
+      end
+    elsif opt[:clevs] and opt[:ccols]
+      clevs = opt[:clevs]
+      ccols = opt[:ccols]
+    else
+      if @clevs and @ccols
+        clevs = @clevs.clone
+        ccols = @ccols.clone
+      else
+        out = q :shades
+        io = StringIO.new(out)
+        io.gets
+        clevs = []
+        ccols = []
+        while line = io.gets
+          col, lev = *line.split(/\s+/)[0..1]
+          ccols << col.to_i
+          if lev != "<"
+            clevs << lev.to_f
+          end
+        end
+      end
+    end
+    set_clevs *clevs
+    set_ccols *ccols
+
+    arg = ""
+    if opt[:pos]
+      arg << opt[:pos].join(" ") + " "
+      xmin, xmax, ymin, ymax = *opt[:pos]
+      if not opt[:dir]
+        if ( xmax - xmin ) > ( ymax - ymin )
+          opt[:dir] = "horizontal"
+        else
+          opt[:dir] = "virtical"
+        end
+      end
+    else
+      cnum = ccols.size
+      xmin, xmax, ymin, ymax, opt[:dir] = *_guess_colorbar_position(cnum)
+    end
+
+    levcol = []
+    i = 0
+    clevs.each_index do 
+      levcol << ccols[i]
+      levcol << clevs[i]
+      i += 1
+    end
+    levcol << ccols[i]
+    arg << "-levcol #{levcol.join(' ')} "
+
+    if opt[:tics]
+      if opt[:tics][0]
+        arg << "-fs #{opt[:tics][0]} "
+      end
+      if opt[:tics][1]
+        arg << "-fo #{opt[:tics][1]} "
+      end
+    end
+
+    if opt[:font]
+      if opt[:font][0]
+        arg << "-fw #{opt[:font][0]} "
+      end
+      if opt[:font][1]
+        arg << "-fh #{opt[:font][1]} "
+      end
+      if opt[:font][2]
+        arg << "-ft #{opt[:font][2]} "
+      end
+    end
+
+    if opt[:edge]
+      arg << "-edge #{opt[:edge]} "
+    end
+
+    if opt[:dir]
+      arg << "-dir #{opt[:dir]} "
+    end
+
+    if opt[:line]
+      arg << "-line on"
+    else
+      arg << "-line off"
+    end
+    xcbar arg
+
+    if opt[:title]
+      case opt[:title]
+      when String
+        text = opt[:title]
+        ts = 0.2
+      else
+        text = opt[:title][0]
+        ts   = opt[:title][1] || 0.2
+      end
+      if opt[:dir] == "horizontal" or opt[:dir] == "h"
+        set :strsiz, ts
+        set :string, 1, "bc", 2
+        draw :string, (xmin+xmax)/2.0, ymax+(ymax-ymin)*0.5, text
+      else
+        set :strsiz, ts
+        set :string, 1, "bc", 2, 90
+        draw :string, xmin-(xmax-xmin)*0.5, (ymin+ymax)/2.0, text
+      end
+    end
+
+  end
+
+  def _guess_colorbar_position (cnum)
+
+    out = q :gxinfo
+
+    io = StringIO.new(out)
+    io.gets
+    xsize, ysize = *io.gets.split(/\s+/).values_at(3,5).map{|v| v.to_i}
+    xlmin, xlmax = *io.gets.split(/\s+/).values_at(3,5).map{|v| v.to_i}
+    ylmin, ylmax = *io.gets.split(/\s+/).values_at(3,5).map{|v| v.to_i}
+    xlwid = xlmax - xlmin
+    ylwid = ylmax - ylmin
+
+    if ylmin < 0.6 or xsize-xlmax > 1.5
+      direction = 'vertical'
+      xmin = xlmax + ( xsize - xlmax ) / 2 - 0.4
+      xmax = xmin + 0.2
+      y1wid = 0.5
+      if y1wid * cnum > ysize * 0.8 
+        y1wid = ysize * 0.8 / cnum
+      end
+      ymin = ysize / 2 - y1wid * cnum / 2
+      ymax = ysize / 2 + y1wid * cnum / 2
+    else
+      direction = 'horizontal'
+      ymin = ylmin / 2
+      ymax = ymin + 0.2
+      x1wid = 0.8
+      if x1wid * cnum > xsize * 0.8
+        x1wid = xsize * 0.8 / cnum
+      end
+      xmin = xsize / 2 - x1wid * cnum / 2
+      xmax = xsize / 2 + x1wid * cnum / 2
+    end
+
+    return xmin, xmax, ymin, ymax, direction
+  end
+
+end
+

data/lib/grads/lib/makecpt.rb

@@ -0,0 +1,180 @@
+class GrADS::Command
+
+  # --- makecpt("GMT's makecpt arguments as String")
+  #
+  # Sets clevs and ccols from the output of piped makecpt command.
+  #
+  # The continuous color pallet generated with makecpt option "-Z" 
+  # is not supported.
+  #
+
+  def setcpt (file)
+    io = Kernel::open(file)
+    while line = io.gets
+      if line =~ /COLOR_MODEL\s*=\s*\+?(HSV|RGB)/
+        model = $1
+        break
+      end
+    end
+    bgrb = nil
+    frgb = nil
+    list = []
+    while line = io.gets
+      case line
+      when /\A[\#N]/
+        next
+      when /\AB\s/
+        brgb = line.split(/\s+/)[1..3].map{|v| v.to_f}
+      when /\AF\s/
+        frgb = line.split(/\s+/)[1..3].map{|v| v.to_f}
+      else
+        list << line.lstrip.split(/\s+/)[0..3].map{|v| v.to_f}
+        last = line.lstrip.split(/\s+/)[4..7].map{|v| v.to_f}
+      end
+    end
+    list << last
+    io.close
+    clevs = []
+    ccols = []
+    ic = 16
+    if brgb
+      if model == "HSV"
+        set :rgb, ic, *_hsv2rgb(*brgb)
+      else
+        set :rgb, ic, *brgb
+      end
+      ccols << ic
+    else
+      ccols << 0
+    end
+    ic += 1
+    list.each_with_index do |(v,r,g,b),i|
+      if model == "HSV"
+        set :rgb, i+ic, *_hsv2rgb(r, g, b)
+      else
+        set :rgb, i+ic, r, g, b
+      end
+      clevs << v
+      if ! frgb and i == list.size - 1 
+        ccols << 0
+      else
+        ccols << i+ic
+      end
+    end
+    ic += list.size - 1
+    if frgb
+      if model == "HSV"
+        set :rgb, ic, *_hsv2rgb(*frgb)
+      else
+        set :rgb, ic, *frgb
+      end
+    end
+    set_clevs *clevs
+    set_ccols *ccols
+  end
+
+  def makecpt (*args)
+    io = IO::popen("makecpt #{args.join(' ')}", "r")
+    while line = io.gets
+      if line =~ /COLOR_MODEL\s*=\s*\+?(HSV|RGB)/
+        model = $1
+        break
+      end
+    end
+    bgrb = nil
+    frgb = nil
+    list = []
+    while line = io.gets
+      case line
+      when /\A[\#N]/
+        next
+      when /\AB\s/
+        brgb = line.split(/\s+/)[1..3].map{|v| v.to_f}
+      when /\AF\s/
+        frgb = line.split(/\s+/)[1..3].map{|v| v.to_f}
+      else
+        list << line.split(/\s+/)[0..3].map{|v| v.to_f}
+        last = line.split(/\s+/)[4..7].map{|v| v.to_f}
+      end
+    end
+    list << last
+    io.close
+    clevs = []
+    ccols = []
+    ic = 16
+    if brgb
+      if model == "HSV"
+        set :rgb, ic, *_hsv2rgb(*brgb)
+      else
+        set :rgb, ic, *brgb
+      end
+      ccols << ic
+    else
+      ccols << 0
+    end
+    ic += 1
+    list.each_with_index do |(v,r,g,b),i|
+      if model == "HSV"
+        set :rgb, i+ic, *_hsv2rgb(r, g, b)
+      else
+        set :rgb, i+ic, r, g, b
+      end
+      clevs << v
+      if ! frgb and i == list.size - 1 
+        ccols << 0
+      else
+        ccols << i+ic
+      end
+    end
+    ic += list.size - 1
+    if frgb
+      if model == "HSV"
+        set :rgb, ic, *_hsv2rgb(*frgb)
+      else
+        set :rgb, ic, *frgb
+      end
+    end
+    set_clevs *clevs
+    set_ccols *ccols
+  end
+
+  def _hsv2rgb(h, s, v)
+    h = h.to_f
+    s = s.to_f
+    v = v.to_f
+
+    if s == 0 and h == 0
+      v = (255*v).floor
+      return v, v, v
+    end
+
+    if h == 360
+      h = 0
+    end
+
+    h /= 60.0
+    i = h.floor.to_i
+    f = h - i
+
+    p = (255*v*(1 - s)).floor
+    q = (255*v*(1 - s*f)).floor
+    t = (255*v*(1 - s*(1 - f))).floor
+    v = (255*v).floor
+
+    case i
+    when 0
+      return v, t, p
+    when 1
+      return q, v, p
+    when 2
+      return p, v, t
+    when 3
+      return p, q, v
+    when 4
+      return t, p, v
+    when 5
+      return v, p, q
+    end
+  end
+
+end

data/lib/grads/lib/save_image.rb

@@ -0,0 +1,26 @@
+require "tempfile"
+
+class GrADS::Command
+
+  def save_image (filename, size=640, dpi=300)
+    io = Tempfile.open("CA_GrADS_", ".")
+    basename = io.path
+    enable(:print, basename) {
+      print
+    }
+    case filename
+    when /\.eps/
+      system %{
+        gxeps -R -i #{basename} -o #{filename}
+      }
+    else
+      system %{
+        gxeps -R -i #{basename} -o #{basename}.eps
+        convert -density #{dpi} -resize #{size} +antialias #{basename}.eps #{filename}
+        rm -f #{basename}.eps
+      }
+    end
+    io.close
+  end
+
+end

data/lib/grads/lib/xcbar_with_ccols.rb

@@ -0,0 +1,32 @@
+
+class GrADS::Command
+
+  def xcbar_with_ccols (clevs, ccols, *xcbaropts)
+    import :xcbar
+    nlevs = clevs.length
+    levcol = []
+    nlevs.times do |i|
+      levcol << ccols[i]
+      levcol << clevs[i]
+    end
+    levcol << ccols.last
+    set :clevs, *clevs
+    set :ccols, *ccols
+    xcbar *[xcbaropts,"-levcol #{levcol.join(' ')}"].flatten
+  end
+
+  def xcbar_with_color (colorspec, *xcbaropts)
+    import :xcbar, :color
+    nlevs = clevs.length
+    levcol = []
+    nlevs.times do |i|
+      levcol << ccols[i]
+      levcol << clevs[i]
+    end
+    levcol << ccols.last
+    set :clevs, *clevs
+    set :ccols, *ccols
+    xcbar *[xcbaropts,"-levcol #{levcol.join(' ')}"].flatten
+  end
+
+end

data/lib/grads.rb

@@ -0,0 +1,11 @@
+require "carray"
+
+module GrADS 
+end
+
+require "grads/gridded"
+require "grads/command"
+
+require "grads/lib/makecpt"
+require "grads/lib/colorbar"
+require "grads/lib/save_image"

data/ruby-grads.gemspec

@@ -0,0 +1,23 @@
+
+Gem::Specification::new do |s|
+  version = "1.0.0"
+
+  files = Dir.glob("**/*") - [ 
+                               Dir.glob("carray*.gem"), 
+                             ].flatten
+
+  s.platform    = Gem::Platform::RUBY
+  s.name        = "ruby-grads"
+  s.summary     = "Library for driving GrADS from Ruby"
+  s.description = <<-HERE
+    Library for driving GrADS from Ruby
+  HERE
+  s.version     = version
+  s.author      = "Hiroki Motoyoshi"
+  s.email       = ""
+  s.homepage    = 'https://github.com/himotoyoshi/ruby-grads'
+  s.files       = files
+#  s.extensions  = [ "extconf.rb" ]
+  s.has_rdoc    = false
+  s.required_ruby_version = ">= 1.8.1"
+end

data/test/gradsdraw.rb

@@ -0,0 +1,12 @@
+require "grads"
+
+GrADS.start {
+  draw line, 1, 2, 4, 5
+  draw rec,  1, 2, 4, 5
+  draw recf, 4, 5, 7, 8
+  gets
+  clear
+  draw polyf, 1, 1, 2, 1, 3, 2, 2, 2, 1, 3
+  gets
+}
+

data/test/test.rb

@@ -0,0 +1,41 @@
+require "carray"
+require "grads"
+require "grads/lib/makecpt"
+require "grads/lib/colorbar"
+require "grads/lib/save_image"
+
+x = CArray.float(301).span(-2.5..2.5)[nil,:*]
+y = CArray.float(301).span(-2.5..2.5)[:*,nil]
+
+xx = x.bind_with(y)
+yy = y.bind_with(x)
+
+z = (-(xx**2+yy**2)).exp*2500-300
+th = CAMath.atan2(yy,xx)
+z += (6*th).cos*300
+
+GrADS.start {
+
+  import :cbarn
+
+  set :mproj, :off
+  set :parea, 3, 8, 2, 7
+  set :grid, :off
+
+  zz = template2d(nil, z) { 
+    xdef x
+    ydef y
+  }
+
+  makecpt "-Ccopper -T0/2500/500"
+
+  set :gxout, :shaded
+
+  d zz
+
+  colorbar
+
+  save_image("test.png")
+
+  STDIN.gets
+}

metadata

@@ -0,0 +1,54 @@
+--- !ruby/object:Gem::Specification
+name: ruby-grads
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Hiroki Motoyoshi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-11-27 00:00:00.000000000 Z
+dependencies: []
+description: "    Library for driving GrADS from Ruby\n"
+email: ''
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- extconf.rb
+- lib/grads.rb
+- lib/grads/command.rb
+- lib/grads/gridded.rb
+- lib/grads/lib/colorbar.rb
+- lib/grads/lib/makecpt.rb
+- lib/grads/lib/save_image.rb
+- lib/grads/lib/xcbar_with_ccols.rb
+- ruby-grads.gemspec
+- test/gradsdraw.rb
+- test/test.rb
+homepage: https://github.com/himotoyoshi/ruby-grads
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.8.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.7
+signing_key: 
+specification_version: 4
+summary: Library for driving GrADS from Ruby
+test_files: []