iconPlot 0.0.2 → 0.0.3

data/gemspec

@@ -3,7 +3,7 @@ $:.unshift File.join(File.dirname(__FILE__),"..","lib")
 
 spec = Gem::Specification.new do |s|
   s.name              = "iconPlot"
-  s.version           = '0.0.2'
+  s.version           = '0.0.3'
   s.platform          = Gem::Platform::RUBY
   s.files             = ["lib/iconPlot.rb"] + ["gemspec"] + ["contrib/nclsh"] + Dir.glob("lib/*ncl")
   s.test_file         = "test/test_iconPlot.rb"
@@ -14,6 +14,9 @@ spec = Gem::Specification.new do |s|
   s.homepage          = "https://github.com/Try2Code/iconPlot"
   s.license           = "GPLv2"
   s.required_ruby_version = ">= 1.8"
+  s.add_dependency('cdo')
+  s.add_dependency('extcsv')
+  s.add_dependency('gnuplot')
 end
 
 # vim:ft=ruby

data/lib/iconPlot.rb

@@ -1,4 +1,7 @@
 require 'fileutils'
+require 'cdo'
+require 'extcsv'
+require 'shellwords'
 
 class IconPlot < Struct.new(:caller,:plotter,:libdir,:otype,:display,:cdo,:debug)
   def IconPlot.gemPath
@@ -52,6 +55,36 @@ class IconPlot < Struct.new(:caller,:plotter,:libdir,:otype,:display,:cdo,:debug
   def vectorPlot(ifile,ofile,varname,opts={})
     plot(ifile,ofile,varname,'vector',opts)
   end
+  def levelPlot(ifile,ofile,varname,opts={})
+    Cdo.setCdo('/home/ram/src/cdo/trunk/cdo/build/bin/cdo')
+    operation = opts[:operation].nil? ? 'fldmin' : opts[:operation]
+
+    data = createData(ifile,varname,operation)
+    icon = ExtCsv.new("array","plain",data.transpose)
+    setDatetime(icon)
+
+    # Plot data with automatic splitting by depth
+    unless icon.datacolumns.include?(varname)
+      warn "Variable cannot be found!"
+      exit -1
+    end
+    Cdo.debug = true
+    unit = Cdo.showunit(:in => "-selname,#{varname} #{ifile}").first
+    ExtCsvDiagram.plot_xy(icon,"datetime",varname,
+                          "ICON: #{operation} on #{varname} (file:#{ifile})", # Change title here
+    :label_position => 'below',:skipColumnCheck => true,
+      :type => 'lines',:groupBy => ["depth"], :onlyGroupTitle => true,
+      #                     :addSettings => ["logscale y"],     # Commend theses out for large scale values like Vert_Mixing_V
+      #                     :yrange => '[0.0001:10]',           # Otherwise you'll see nothing reasonable
+      :terminal => true ? 'png' : 'x11',
+      :ylabel => "#{varname} [#{Shellwords.escape(unit)}]",     # Correct the label if necessary
+      :input_time_format => "'%Y%m%d %H:%M:%S'",
+        :filename => ofile,
+        :output_time_format => '"%d.%m.%y \n %H:%M"',:size => "1600,600")
+      return "#{ofile}.png"
+  end
+  def scatterPlot(ifile,ofile,varname,opts={})
+  end
 
   def del(file)
     FileUtils.rm(file) if File.exists?(file)
@@ -65,4 +98,51 @@ class IconPlot < Struct.new(:caller,:plotter,:libdir,:otype,:display,:cdo,:debug
   def showVector(ifile,ofile,vars,opts={})
     show(vectorPlot(ifile,ofile,vars,opts))
   end
+
+  def createData(ifile,varname,operation)
+    # Temporal file for text output
+    dataFile = MyTempfile.path
+
+    # read the date
+    IO.popen("echo 'date|time|depth|#{varname}' > #{dataFile}")
+    Cdo.debug = true
+    Cdo.outputkey('date,time,level,value', 
+                  :in => "-#{operation} -selname,#{varname} #{ifile} >>#{dataFile} 2>&1")
+
+    # postprocessing for correct time values
+    data = []
+    File.open(dataFile).each_with_index {|line,lineIndex|
+      next if line.chomp.empty?
+      _t = line.chomp.gsub(/ +/,'|').split('|')
+      if 0 == lineIndex then
+        data << _t
+        next
+      end
+      if "0" == _t[1] then
+        _t[1] = '00:00:00'
+      else
+        time = _t[1].reverse
+        timeStr = ''
+        while time.size > 2 do
+          timeStr << time[0,2] << ':'
+          time = time[2..-1]
+        end
+        timeStr << time.ljust(2,'0') unless time.size == 0
+        _t[1] = timeStr.reverse
+      end
+      data << _t
+    }
+    data
+  end
+
+  def setDatetime(extcsvObj)
+    unless (extcsvObj.respond_to?(:date) and extcsvObj.respond_to?(:time))
+      warn "Cannot set datetime due to missing date and time attributes"
+      raise ArgumentError
+    end
+    # Create datetime column for timeseries plot
+    extcsvObj.datetime = []
+    extcsvObj.date.each_with_index{|date,i| extcsvObj.datetime << [date,extcsvObj.time[i]].join(' ') }
+    extcsvObj.datacolumns << "datetime"
+  end
 end

data/test/test_iconPlot.rb

@@ -9,6 +9,7 @@ class TestIconPlot < Test::Unit::TestCase
   PLOTLIB = "/home/ram/src/git/icon/scripts/postprocessing/tools"
   LS                    = 'ls -crtlh'
   OCE_PLOT_TEST_FILE    = ENV['HOME']+'/data/icon/oce.nc'
+  OCELONG_PLOT_TEST_FILE    = ENV['HOME']+'/data/icon/oceLong.nc'
   ATM_PLOT_TEST_FILE    = ENV['HOME']+'/data/icon/atm.nc'
   OCE_REGPLOT_TEST_FILE = ENV['HOME']+'/data/icon/regular_oce.nc' #remapnn,r180x90
   ATM_REGPLOT_TEST_FILE = ENV['HOME']+'/data/icon/regular_atm.nc' #remapnn,n63 (no sections), r180x90 (with sections)
@@ -30,4 +31,8 @@ class TestIconPlot < Test::Unit::TestCase
     assert_includes(p.plotter.split(File::SEPARATOR),'gems')
     assert_includes(p.libdir.split(File::SEPARATOR),'gems')
   end
+  def test_levelPlot
+    p = IconPlot.new
+    p.show( p.levelPlot(OCELONG_PLOT_TEST_FILE,'test_levelPlot_00','T',:operation => :fldmax))
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: iconPlot
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,8 +9,41 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-05-23 00:00:00.000000000 Z
-dependencies: []
+date: 2012-08-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: cdo
+  requirement: &16945020 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *16945020
+- !ruby/object:Gem::Dependency
+  name: extcsv
+  requirement: &16944200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *16944200
+- !ruby/object:Gem::Dependency
+  name: gnuplot
+  requirement: &16943700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *16943700
 description: ! 'Plot with ncl via Ruby: requires NCL 6.* and CDO 1.5.*'
 email: stark.dreamdetective@gmail.com
 executables: []