pgplot 0.1.7

data/demo/pgdemo15.rb

@@ -0,0 +1,78 @@
+require 'narray'
+require 'pgplot'
+include NMath
+include Pgplot
+#-----------------------------------------------------------------------
+# Demonstration program for PGPLOT vector field plot.
+#
+# Program to demonstrate the use of PGVECT along with
+# PGCONB by illustrating the flow around a cylinder with circulation.
+#-----------------------------------------------------------------------
+      twopi = 2*Math::PI
+      blank = -1.0e10
+# number of points in the x and y directions
+      nx = 31
+      ny = 31
+# cylinder radius
+      a = 1.0
+# circulation strength
+      gamma = 2.0
+# freestream velocity
+      vinf = 1.0
+# max and min x and y
+      xmax =  3.0*a
+      xmin = -3.0*a
+      ymax =  3.0*a
+      ymin = -3.0*a
+# point spacing
+      dx = (xmax-xmin)/(nx-1)
+      dy = (ymax-ymin)/(ny-1)
+# compute the stream function, Cp, and u and v velocities
+      a2 = a**2
+      x  = NArray.sfloat(nx,1).indgen!*dx + xmin
+      y  = NArray.sfloat(1,ny).indgen!*dy + ymin
+      r2 = x**2 + y**2
+      r2[(r2.eq 0).where] = 1e-10
+      psi = vinf * y * (1-a2/r2) + gamma/twopi*0.5*log(r2/a)
+      u   = vinf * (1 + a2/r2 - 2*a2*(x/r2)**2) + gamma/twopi * y/r2
+      v   = vinf * x * (-2*a2*y/r2**2) + gamma/twopi * x/r2
+      cp  = 1 - (u**2+v**2)/vinf**2
+      idx = (r2 < a2).where
+      u[idx] = 
+      v[idx] = blank
+#
+# start drawing
+#
+      pgbeg
+      pgenv( x[0], x[-1], y[0], y[-1], 1 )
+      pgiden
+      pglab('X','Y','Flow About a Cylinder with Circulation')
+#
+# contour plot of the stream function (streamlines)
+#
+      tr = [ x[0]-dx, dx, 0.0, y[0]-dy, 0.0, dy ]
+      level = [ 1.0, 0.5, 0.0, -0.5, -1.0 ]
+      pgcont( psi, level, tr )
+#
+# draw cylinder
+#
+      pgbbuf
+      pgsci(0)
+      pgsfs(1)
+      pgcirc(0.0, 0.0, a*1.1)
+      pgsfs(2)
+      pgsci(14)
+      pgcirc(0.0, 0.0, a)
+      pgsci(1)
+      pgebuf
+#
+# vector plot
+#
+      pgsah(2, 45.0, 0.7)
+      pgsch(0.3)
+      tr = [ x[0], dx, 0.0, y[0], 0.0, dy ]
+      pgvect( u[1..-2,1..-2], v[1..-2,1..-2], 0.0, 0, tr, -1.0e10 )
+      pgsch(1.0)
+#
+# finish
+#

data/demo/pgdemo3.rb

@@ -0,0 +1,188 @@
+require 'narray'
+require 'pgplot'
+include Pgplot
+include NMath
+
+def main
+  exit unless pgbeg('?',1,1)
+  print " Routine PGCONT\n"
+  pgex31
+  print " Routine PGCONS\n"
+  pgex32
+  print " Routine PGCONB\n"
+  pgex33
+  print " Routine PGCONT with PGCONL labels\n"
+  pgex36
+  #print " Routine PGCONX with arrow labels\n"
+  #pgex37
+  #print " Routine PGCONX\n"
+  #pgex34
+  print " Routine PGCONF\n"
+  pgexx1
+  pgend
+end
+
+# ====== Utility ======
+
+class PgLinestyle
+  attr_accessor :width
+  attr_accessor :color
+  attr_accessor :style
+
+  def initialize(opt=nil)
+    @width = 1
+    @color = 1
+    @style = 1
+    if opt.is_a?(Hash)
+      @width = opt[:width]  || @width
+      @width = opt['width'] || @width
+      @color = opt[:color]  || @color
+      @color = opt['color'] || @color
+      @style = opt[:style]  || @style
+      @style = opt['style'] || @style
+    end
+  end
+
+  def set
+    pgslw @width
+    pgsci @color
+    pgsls @style
+  end
+end # class PgLinestyle
+
+NC=21
+Pgline_default = PgLinestyle.new
+$sty = (1..NC).collect{|i|
+  if i<10
+    x = PgLinestyle.new(:width =>1, :color =>2, :style =>2)
+  else
+    x = PgLinestyle.new(:width =>1, :color =>3, :style =>1)
+  end
+  if i%5==0
+    x.width = 5
+  end
+  x
+}
+i = NArray.sfloat(40,1).indgen!(1)
+j = NArray.sfloat(1,40).indgen!(1)
+$f = cos( 0.3*sqrt(i*2)-0.4*j/3 ) * cos( 0.4*i/3 ) + (i-j)/40.0
+$lv = NArray.sfloat(NC).indgen!*($f.max-$f.min)/NC+$f.min
+#NArray.span($f.minmax,NC)
+
+# ====== Exsample routine ======
+
+def pgex31
+  pgpage
+  pgsvp(0.05,0.95,0.05,0.95)
+  pgswin(1.0,40.0,1.0,40.0)
+  pgbox('bcts',0.0,0,'bcts',0.0,0)
+  pgmtxt('t',1.0,0.0,0.0,'Contouring using PGCONT')
+  pgbbuf
+  for i in 0...NC
+    $sty[i].set
+    pgcont $f, $lv[i]
+  end
+  Pgline_default.set
+  pgebuf
+end
+
+def pgex32
+  pgpage
+  pgsvp(0.05,0.95,0.05,0.95)
+  pgswin(1.0,40.0,1.0,40.0)
+  pgbox('bcts',0.0,0,'bcts',0.0,0)
+  pgmtxt('t',1.0,0.0,0.0,'Contouring using PGCONS')
+  pgbbuf
+  for i in 0...NC
+    $sty[i].set
+    pgcons $f, $lv[i]
+  end
+  Pgline_default.set
+  pgebuf
+end
+
+def pgex33
+  pgpage
+  pgsvp(0.05,0.95,0.05,0.95)
+  pgswin(1.0,40.0,1.0,40.0)
+  pgbox('bcts',0.0,0,'bcts',0.0,0)
+  pgmtxt('t',1.0,0.0,0.0,'Contouring using PGCONB')
+  pgbbuf
+
+  blank = -65536.0
+
+  f = $f.dup
+  i = NArray.sfloat(40,1).indgen!(1)
+  j = NArray.sfloat(1,40).indgen!(1)
+  r = sqrt((i-20.5)**2 + (j-20.5)**2)
+  idx = ((r>20).or(r<3.0)).where
+  f[idx] = blank
+  pgsci 1
+  pgpt( (i+NArray.int(1,40))[idx], (j+NArray.int(40,1))[idx], 1 )
+
+  for i in 0...NC
+    $sty[i].set
+    pgconb f, $lv[i], blank
+  end
+  Pgline_default.set
+  pgebuf
+end
+
+def pgex36
+  pgpage
+  pgsvp(0.05,0.95,0.05,0.95)
+  pgswin(1.0,40.0,1.0,40.0)
+  pgbox('bcts',0.0,0,'bcts',0.0,0)
+  pgmtxt('t',1.0,0.0,0.0,'Contouring using PGCONT and PGCONL labels')
+  pgbbuf
+
+  for i in 0...NC
+    $sty[i].set
+    pgcons $f, $lv[i]
+  end
+  pgslw 1
+  pgsls 1
+  1.step(20,2) {|i|
+    pgsci $sty[i].color
+    pgconl $f,$lv[i],"%2i"%(i+1),16,8
+  }
+  Pgline_default.set
+  pgebuf
+end
+
+def pgexx1
+  nx=ny=51
+  c = [3.0, 3.2, 3.5, 3.6, 3.766413, 4.0 ,5.0, 10.0, 100.0]
+  xmin =-2.0
+  xmax = 2.0
+  ymin =-2.0
+  ymax = 2.0
+  mu = 0.3
+  dx = (xmax-xmin)/(nx-1)                                      
+  dy = (ymax-ymin)/(ny-1)
+  tr = [xmin-dx, dx, 0.0, ymin-dy, 0.0, dy]
+  x = tr[0] + NArray.sfloat(nx,1).indgen!(1)*tr[1]
+  y = tr[3] + NArray.sfloat(1,ny).indgen!(1)*tr[5]
+  z = (1.0-mu)*(2.0/sqrt((x-mu)**2+y**2)+(x-mu)**2+y**2) +
+       mu*(2.0/sqrt((x+1.0-mu)**2+y**2)+(x+1.0-mu)**2+y**2)      
+
+  pgpage
+  pgvstd
+  pgwnad(xmin, xmax, ymin, ymax)
+  pgsfs(1)
+  for i in 0..c.size-2
+    r = 0.5+0.5*(i-1)/(c.size-1)
+    pgscr(i+10, r, r, r)
+    pgsci(i+10)
+    pgconf(z,c[i]..c[i+1],tr)
+  end
+  pgsci(3)
+  pgcont(z,c,tr)
+  pgsci(1)
+  pgsch(0.6)
+  pgbox('bctsin',1.0,10,'bctsinv',1.0,10)
+  pgsch(1.0)
+  pgmtxt('t',1.0,0.0,0.0,'Contour filling using PGCONF')
+end
+
+main

data/demo/pgdemo4.rb

@@ -0,0 +1,80 @@
+require 'narray'
+require 'pgplot'
+include Pgplot
+include NMath
+
+def setvp
+  pgsvp(0.0, 1.0, 0.0, 1.0)
+  vpx1, vpx2, vpy1, vpy2 = pgqvp(1)
+  d = [vpx2-vpx1, vpy2-vpy1].min/40.0
+  vpx1 = vpx1 + 5.0*d
+  vpx2 = vpx2 - 2.0*d
+  vpy1 = vpy1 + 8.0*d
+  vpy2 = vpy2 - 2.0*d
+  pgvsiz(vpx1, vpx2, vpy1, vpy2)
+end
+
+def palett contra,bright
+  rl =[-0.5, 0.0, 0.17, 0.33, 0.50, 0.67, 0.83, 1.0, 1.7]
+  rr =[	0.0, 0.0,  0.0,  0.0,  0.6,  1.0,  1.0, 1.0, 1.0]
+  rg =[	0.0, 0.0,  0.0,  1.0,  1.0,  1.0,  0.6, 0.0, 1.0]
+  rb =[	0.0, 0.3,  0.8,  1.0,  0.3,  0.0,  0.0, 0.0, 1.0]
+  pgctab(rl, rr, rg, rb)
+end
+
+exit if pgopen('?') < 1
+printf "PGPLOT device type: %s\n", pgqinf('TYPE')
+c1,c2 = pgqcir()
+printf "Number of color indices used for image: %d\n", nc=[0,c2-c1+1].max
+exit if nc<8
+
+nx = 64
+ny = 64
+x = NArray.sfloat(nx,1).indgen!(1)
+y = NArray.sfloat(1,ny).indgen!(1)
+f = cos( sqrt(x*(80.0/nx))*0.6 - y*16.0/(3.0*ny) ) *
+    cos( x*16.0/(3.0*nx) ) +
+    (x/nx - y/ny) + sin(sqrt(x**2+y**2))*0.05
+
+pgpage
+setvp
+pgwnad(0.0, 1.0+nx, 0.0, 1.0+ny)
+bright = 0.5
+contra  = 1.0
+palett(contra, bright)
+pgimag(f)
+pgmtxt('t',1.0,0.0,0.0,'PGIMAG, PGWEDG, and PGCTAB')
+pgsch(0.6)
+pgbox('bcntsi',0.0,0,'bcntsiv',0.0,0)
+pgmtxt('b',3.0,1.0,1.0,'pixel number')
+pgwedg('BI', 4.0, 5.0, f.min,f.max, 'pixel value')
+pgsch(1.0)
+
+angle = 120.0/57.29578
+c = cos(angle)
+s = sin(angle)
+tr = NArray[ -c-s,  2.0*c/nx, 2.0*s/ny,
+	     -c+s, -2.0*s/nx, 2.0*c/ny ]
+
+pgpage
+setvp
+pgwnad(-1.0, 1.0, -1.0, 1.0)
+pgsci(1)
+bright = 0.5
+contra  = 1.0
+palett(contra, bright)
+pgimag(f,nil,tr)
+pgsci(1)
+
+pgcont(f,NArray.sfloat(21).indgen!*(f.max-f.min)/21+f.min, tr)
+pgsls(1)
+pgslw(1)
+pgsci(1)
+#outlin(1,mxi,1,mxj,tr)
+pgmtxt('t',1.0,0.0,0.0,'PGIMAG, PGCONT and PGWEDG')
+pgsch(0.6)
+pgbox('bctsn',0.0,0,'bctsn',0.0,0)
+pgwedg('BI', 4.0, 5.0, f.min, f.max, 'pixel value')
+pgsch(1.0)
+
+pgclos

data/demo/pgdemo9.rb

@@ -0,0 +1,44 @@
+require 'narray'
+require 'pgplot'
+include Pgplot
+include NMath
+
+n=64
+ncol=32
+nlev=9
+
+pgbeg
+
+i = NArray.sfloat(n,1).indgen!(1)
+j = NArray.sfloat(1,n).indgen!(1)
+f = cos(0.6*sqrt(i*2)-(0.4/3)*j) * cos((0.4/3)*i) + (i-j)/n
+fmin = f.min
+fmax = f.max
+
+ia = (f-fmin)/(fmax-fmin)*(ncol-1)+16
+
+ci1,ci2 = pgqcol
+if ci2 < 15+ncol
+  raise 'This program requires a device with at least %d colors'%(15+ncol)
+end
+
+pgpage
+pgscr(0, 0.0, 0.3, 0.2)
+pgsvp(0.05,0.95,0.05,0.95)
+pgwnad(0.0, 1.0, 0.0, 1.0)
+
+for i in 1..ncol
+  r = 0.8*(i-1)/(ncol-1) + 0.2
+  g = 2.0*(i-1-ncol/2)/(ncol-1)
+  g = 0 if g<0
+  b = 0.2 + 0.4*(ncol-i)/ncol
+  pgscr(i+15, r, g, b)
+end
+
+pgpixl(ia,0,1,0,1)
+pgsci(1)
+pgmtxt('t',1.0,0.0,0.0,'Test of PGPIXL')
+pgbox('bcnts',0.0,0,'bcnts',0.0,0)
+
+clev = NArray.sfloat(nlev).indgen!(1) * ((fmax-fmin)/nlev) + fmin
+pgcont(f, clev, NArray[-1.0, 1, 0, -1, 0, 1]/(n-1))

data/ext/cogen.rb

@@ -0,0 +1,158 @@
+# cogen.rb : Code generation script for Ruby/PGPLOT
+#
+#   Copyright (c) 2000,2001 Masahiro TANAKA <masa@ir.isas.ac.jp>
+#
+#   This program is free software.
+#   You can distribute/modify this program
+#   under the same terms as Ruby itself.
+#   NO WARRANTY.
+
+$pgfuncs = %w(
+
+pgend::
+pgbbuf::
+pgebuf::
+pgpage::
+pgpap:1,1:
+pgupdt::
+pgpanl:1,1:
+pgclos::
+
+pgbox:2,1,0,2,1,0:
+pgtbox:2,1,0,2,1,0:
+pgvsiz:1,1,1,1:
+pgvstd::
+pgwnad:1,1,1,1:
+pgsubp:0,0:
+
+pgwedg:2,1,1,1,1,2:
+
+# Draw Funcs
+pgdraw:1,1:
+pgmove:1,1:
+pgrect:1,1,1,1:
+pgarro:1,1,1,1:
+pgcirc:1,1,1:
+pgpt1:1,1,0:
+pgerr1:0,1,1,1,1:
+pglab:2,2,2:
+pgptxt:1,1,1,1,2:
+pgtext:1,1,2:
+pgmtxt:2,1,1,1,2:
+pgetxt::
+
+pgiden::
+pgldev::
+pgsave::
+pgunsa::
+pgeras::
+
+# Set Funcs
+pgsch:1:
+pgscf:0:
+pgsci:0:
+pgsfs:0:
+pgsls:0:
+pgslw:0:
+pgsclp:0:
+pgsitf:0:
+pgslct:0:
+pgstbg:0:
+pgscr:0,1,1,1:
+pgshls:0,1,1,1:
+pgsah:0,1,1:
+pgscrl:1,1:
+pgscir:0,0:
+pgscrn:0,2:0
+pgshs:1,1,1:
+pgsvp:1,1,1,1:
+pgswin:1,1,1,1:
+
+# Query Funcs
+pgqch::1
+pgqcf::0
+pgqci::0
+pgqcs:0:1,1
+pgqfs::0
+pgqls::0
+pgqlw::0
+pgqclp::0
+pgqid::0
+pgqitf::0
+pgqndt::0
+pgqtbg::0
+pgqcr:0:1,1,1
+pgqvp:0:1,1,1,1
+pgqwin::1,1,1,1
+pgqcol::0,0
+pgqcir::0,0
+pgqpos::1,1
+pgqvsz:0:1,1,1,1
+
+).grep(/:.*:/).collect{|i| i.split(":",3)}
+
+
+def pgfuncgen(name, inp, out)
+  inp = inp.split(",").collect{|i| i.to_i}
+  out = out.split(",").collect{|i| i.to_i}
+  ninp = inp.size
+  nout = out.size
+  # int->0, float->1
+  val2 = ["NUM2INT","NUM2DBL","StringValuePtr"]
+  type = ["int","float",nil]
+  conv = ["INT2NUM","rb_float_new",nil]
+  # Initialize Array
+  prot = ["VALUE obj"]
+  pass = []
+  vars = []
+  retn = []
+  inp.each_with_index { |i,x|
+    prot << "VALUE arg#{x}"
+    pass << "#{val2[i]}(arg#{x})"
+  }
+  out.each_with_index { |i,x|
+    vars << "#{type[i]} var#{x};"
+    pass << "&var#{x}"
+    retn << "#{conv[i]}(var#{x})"
+  }
+  if nout==0 then
+    retn = "Qtrue";
+  elsif nout>1 then
+    retn = "rb_ary_new3(#{nout},"+retn.join(",")+")"
+  else
+    retn = retn.join("")
+  end
+
+  vars = vars.join("")
+  prot = prot.join(",")
+  pass = pass.join(",")
+
+  return "
+static VALUE
+  rb_pgplot_#{name}(#{prot})
+{
+  #{vars}
+  c#{name}(#{pass});
+  return #{retn};
+}
+"
+end
+
+
+def cogen_pgplot
+  fin  = open("rb_pgplot.c.in","r")
+  fout = open("rb_pgplot.c","w")
+  while l = fin.gets
+    if    /--- auto-generated funcs will be placed here ---/ =~ l
+      $pgfuncs.each{|x| fout.print pgfuncgen(*x)}
+    elsif /--- auto-generated defs will be placed here ---/  =~ l
+      $pgfuncs.each{|x|
+	n = x[1].split(",").size
+	fout.print "  rb_define_module_function(mPgplot,\"#{x[0]}\",rb_pgplot_#{x[0]},#{n});\n"}
+    else
+      fout.print
+    end
+  end
+end
+
+cogen_pgplot

data/ext/depend

@@ -0,0 +1,4 @@
+rb_pgplot.c: rb_pgplot.c.in cogen.rb
+	$(RUBY) cogen.rb
+
+rb_pgplot.o: rb_pgplot.c $(hdrdir)/ruby.h

data/ext/extconf.rb

@@ -0,0 +1,124 @@
+# extconf.rb : Configure script for Ruby/PGPLOT
+#
+#   Copyright (c) 2000,2001 Masahiro TANAKA <masa@ir.isas.ac.jp>
+#
+#   This program is free software.
+#   You can distribute/modify this program
+#   under the same terms as Ruby itself.
+#   NO WARRANTY.
+#
+# usage: ruby extconf.rb [configure options]
+
+# Narray is now Gem based, so require rubygems
+# so that we can use Gem class to find narray.
+require "mkmf"
+
+#$DEBUG = true
+
+# configure options:
+#  --with-x11-dir=path
+#  --with-x11-include=path
+#  --with-x11-lib=path
+dir_config("x11")
+
+# configure options:
+#  --with-pgplot-dir=path
+#  --with-pgplot-include=path
+#  --with-pgplot-lib=path
+dir_config("pgplot")
+
+# Otherwise you can also specify:
+#  --with-opt-dir=path
+#  --with-opt-include=path
+#  --with-opt-lib=path
+
+# Check PGPLOT Header
+exit unless have_header("cpgplot.h")
+
+def find_dir_w_file(d,h)
+  g = Dir.glob(RbConfig.expand(d+"/"+h))
+  File.dirname(g.last) if g and !g.empty?
+end
+
+def find_dir_in_gemspec
+  begin
+    require 'rubygems'
+    if gemspec=Gem::Specification.find_by_path('narray')
+      return File.join(gemspec.full_gem_path, gemspec.require_path)
+    end
+  rescue
+  end
+  nil
+end
+
+# Check NArray
+gems_dir="$(rubylibprefix)/gems/$(ruby_version)/gems/"
+narray_d="narray-0.[56].*"
+narray_h="narray.h"
+if narray_h_dir =
+    find_dir_in_gemspec ||
+    find_dir_w_file("../"+narray_d,narray_h) ||
+    find_dir_w_file(gems_dir+narray_d,narray_h) ||
+    find_dir_w_file(CONFIG['sitearchdir'],narray_h) ||
+    find_dir_w_file(CONFIG['archdir'],narray_h)
+  $CPPFLAGS = " -I#{narray_h_dir} " + $CPPFLAGS
+end
+exit unless have_header("narray.h")
+
+if RUBY_PLATFORM =~ /cygwin|mingw/
+  $LDFLAGS = " -L#{CONFIG['sitearchdir']} "+$LDFLAGS
+  exit unless have_library("narray","na_make_object")
+end
+
+# Check FORTRAN Libraries
+#
+# SUN WorkShop FORTRAN 77 compiler ver5.0
+# configure options: --with-sunws
+if with_config("sunws")
+  $libs = "-lM77 -lsunmath "+$libs
+  exit unless find_library("F77", "f77_init", "/opt/SUNWspro/lib")
+  $defs.push "-DSPARC_FORTRAN"
+#
+# GNU FORTRAN v4
+elsif have_library("gfortran")
+  $CFLAGS = "-Wall "+$CFLAGS
+  $defs.push "-DGNU_FORTRAN"
+#
+# GNU FORTRAN v3
+elsif have_library("g77")
+  $CFLAGS = "-Wall "+$CFLAGS
+  $defs.push "-DGNU_FORTRAN"
+else
+  puts "failed"
+  exit
+end
+
+# Check GrWin Library (for cygwin (and mingw32?))
+#  configure options: --with-grwin
+if with_config("grwin")
+  #$LDFLAGS = "-Wl,--subsystem,console "+$LDFLAGS
+  if RUBY_PLATFORM =~ /cygwin|mingw/
+    $libs += " -mwindows"
+  end
+  exit unless have_library("GrWin", "GWinit")
+end
+#
+# Check X11 Library
+have_library("X11", "XOpenDisplay")
+
+# Check PNG Library
+libs_save = $libs
+$libs = append_library($libs, "z")
+if !have_library("png","png_create_write_struct")
+  $libs = libs_save
+end
+
+# Check PGPLOT Library
+$libs = append_library($libs, "pgplot")
+exit unless find_library( "cpgplot", "cpgbeg", "/usr/lib",
+			  "/usr/local/lib", "/usr/local/pgplot" )
+
+$objs = %w(rb_pgplot.o kwarg.o)
+
+# Generate Makefile
+create_makefile("pgplot")