cdp 0.0.1 → 0.0.2

data/gemspec

@@ -3,10 +3,10 @@ $:.unshift File.join(File.dirname(__FILE__),"..","lib")
 
 spec = Gem::Specification.new do |s|
   s.name              = "cdp"
-  s.version           = '0.0.1'
+  s.version           = '0.0.2'
   s.platform          = Gem::Platform::RUBY
   s.files             = ["lib/cdp.rb"] + ["gemspec"]
-#  s.test_file         = "test/test_cdo.rb"
+  s.test_file         = "test/test_cdp.rb"
   s.description       = "Climate Data Processing helpers"
   s.summary           = "---"
   s.author            = "Ralf Mueller"

data/lib/cdp.rb

@@ -1,7 +1,8 @@
 require 'cdo'
-module CDP
+require 'socket'
+module Cdp
 
-  def CDP.setCDO
+  def Cdp.setCDO
     #=============================================================================== 
     # setup of CDO on different machines
     hostname = Socket.gethostname
@@ -17,25 +18,25 @@ module CDP
     end
   end
 
-  def CDP.setDEBUG
+  def Cdp.setDEBUG
     Cdo.checkCdo
     Cdo.debug = true unless ENV['DEBUG'].nil?
   end
 
   #=============================================================================== 
   # compute the experiments from the data directories and link the corresponding files
-  def CDP.splitFilesIntoExperiments(files)
+  def Cdp.splitFilesIntoExperiments(files)
     gridFile = files.pop
     experiments = files.map {|f| File.basename(File.dirname(f))}.uniq.sort_by {|f| f.length}.reverse
     # take the larges part of the filenames as experiment name if the files are in
     # the current directory
     if experiments == ["."] then
-      n = files.map(&:size).min.times.map {|i| 
+      n = files.map(&:size).min.times.map {|i|
         if files.map {|f| f[0,i-1]}.uniq.size == 1
           1
         else
           nil
-        end 
+        end
       }.find_all {|v| not v.nil?}.size-1
       uniqName = files[0][0,n]
       experiments = [uniqName]
@@ -51,4 +52,22 @@ module CDP
 
     [gridFile,experimentFiles,experimentAnalyzedData]
   end
+
+  def Cdp.manualAreaWeights(areaVarname,areaFile,ofile=nil)
+    area    = Cdo.selname(areaVarname,:in => areaFile)
+    areaSum = Cdo.enlarge(area,:in => "-fldsum #{area}")
+    if ofile.nil?
+      weights = Cdo.setname("area_weight",:in => " -div #{area} #{areaSum}")
+    else
+                Cdo.setname("area_weight",:in => " -div #{area} #{areaSum}", :out => ofile)
+    end
+  end
+  def Cdp.manualMaskedAreaWeights(areaVarname,areaFile,maskVarname,maskFile,ofile=nil)
+    maskedArea    = Cdo.div(:in    => " -selname,#{areaVarname} #{areaFile} -selname,#{maskVarname} #{maskFile}")
+    if ofile.nil?
+      maskedAreaSum = Cdo.setname("area_weight", :in => " -div  #{maskedArea}  -enlarge,#{maskedArea} -fldsum #{maskedArea}")
+    else
+                      Cdo.setname("area_weight", :in => " -div  #{maskedArea}  -enlarge,#{maskedArea} -fldsum #{maskedArea}", :out => ofile)
+    end
+  end
 end

data/test/test_cdp.rb

@@ -0,0 +1,22 @@
+$:.unshift File.join(File.dirname(__FILE__),"..","lib")
+require 'test/unit'
+require 'cdp'
+require 'pp'
+
+DATADIR  = ENV['HOME'] + "/data/icon"
+ICONGRID = DATADIR     + "/icon-R2B04-grid-etopo.nc"
+ICONMASK = DATADIR     + "/mask.nc"
+
+class TestCdp < Test::Unit::TestCase
+  def test_area_weight
+    Cdo.debug = true
+    weights = Cdp.manualAreaWeights("cell_area",ICONGRID)
+    assert_equal("1",Cdo.outputkey("value", :in => Cdo.fldsum(:in => weights)).first)
+  end
+  def test_masked_area_weight
+    Cdo.debug = true
+    Cdp.setCDO
+    weights = Cdp.manualMaskedAreaWeights("cell_area",ICONGRID,"wet_c",ICONMASK)
+    assert_equal("1",Cdo.outputkey("value", :in => Cdo.fldsum(:in => weights)).uniq.first)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cdp
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -35,6 +35,7 @@ extra_rdoc_files: []
 files:
 - lib/cdp.rb
 - gemspec
+- test/test_cdp.rb
 homepage: 
 licenses:
 - BSD
@@ -60,4 +61,5 @@ rubygems_version: 1.8.23
 signing_key: 
 specification_version: 3
 summary: ! '---'
-test_files: []
+test_files:
+- test/test_cdp.rb