Class: NumRu::GPhys::GribVar

Inherits:
Object
  • Object
show all
Extended by:
GribUtils
Includes:
GribUtils
Defined in:
lib/numru/gphys/grib.rb

Overview

NumRu::GPhys::GribVar

Class Methods

—always_time_seq=(t_or_f) If true is given, a GribVar will always have a time dimension, even when the time is unique (i.e., length of the time coordinate is 1).

—new( file, name, obj, dims )

Methods

—file —name —rank —total —set_var —set_miss —dim_names —dim( index ) —ndims —def_dim(name,index) —put_att( key, value ) —set_att( key, value ) —att( key ) —att_names —shape —typecode —get( indics ) —[]( indics ) —inspect

Constant Summary collapse

@@always_time_seq = 
false

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(file, name) ⇒ GribVar

end definition of class methods GribVar



1878
1879
1880
1881
1882
1883
1884
 
# File 'lib/numru/gphys/grib.rb', line 1878

def initialize(file,name)
  @file = file
  @name = name
  @attr = Hash.new
  @miss = false
  @dims = Array.new
end

Class Method Details

.always_time_seq ⇒ Object 



1709
1710
1711
 
# File 'lib/numru/gphys/grib.rb', line 1709

def always_time_seq
  @@always_time_seq
end

.always_time_seq=(t_or_f) ⇒ Object 



1706
1707
1708
 
# File 'lib/numru/gphys/grib.rb', line 1706

def always_time_seq=(t_or_f)
  @@always_time_seq = t_or_f
end

.parse(file, sgms, name) ⇒ Object 



1758
1759
1760
1761
1762
1763
1764
1765
1766
1767
1768
1769
1770
1771
1772
1773
1774
1775
1776
1777
1778
1779
1780
1781
1782
1783
1784
1785
1786
1787
1788
1789
1790
1791
1792
1793
1794
1795
1796
1797
1798
1799
1800
1801
1802
1803
1804
1805
1806
1807
1808
1809
1810
1811
1812
1813
1814
1815
1816
1817
1818
1819
1820
1821
1822
1823
1824
1825
1826
1827
1828
1829
1830
1831
1832
1833
1834
1835
1836
1837
1838
1839
1840
1841
1842
1843
1844
1845
1846
1847
1848
1849
1850
1851
1852
1853
1854
1855
1856
1857
1858
1859
1860
1861
1862
1863
1864
1865
1866
1867
1868
1869
1870
1871
1872
1873
1874
1875
 
# File 'lib/numru/gphys/grib.rb', line 1758

def parse(file,sgms,name)
  miss = false
  sgms.each{|sgm|
    pds = sgm.pds
    if !vars_same?(sgms[0],sgm)
      raise "coordinate is not same"
    end
    miss = miss | pds.bms?
  }
  pds = sgms[0].pds
  gds = sgms[0].gds
  va = GribVar.new(file,name)
  va.set_sgms(sgms)
  va.put_att("long_name",pds.name)
  std_name = pds.standard_name
  va.put_att("standard_name",std_name) if std_name
  va.put_att("units",pds.unit) if pds.unit
  va.set_miss(miss)
  vdim = Array.new
  x,y = gds.grid
  x_sname = x["short_name"]
  d = va.def_dim(x_sname,x["ij"])
  d.put(x["value"])
  x.each{|k,v|
    if k!="value"&&k!="ij"&&k!="length"
      d.put_att(k,v)
    end
  }
  y_sname = y["short_name"]
  d = va.def_dim(y_sname,y["ij"])
  d.put(y["value"])
  y.each{|k,v|
    if k!="value"&&k!="ij"&&k!="length"
      d.put_att(k,v)
    end
  }
  if sgms.length>=1
    nz = pds.z_value.length
    if nz!=0
      z = Array.new(nz).collect{Array.new}
      n=nil
      m=nil
      sgms.each{|sgm|
        zv = sgm.pds.z_value
        nz.times{|m|
          if zv[m]
            z[m].push( zv[m]["value"] )
          else
            z[m].push(m)
          end
        }
      }
      nz.times{|m|
        z[m].uniq!
        z[m].length==z[0].length || raise("length is not same")
      }
      if z[0].length>1
        d = va.def_dim(pds.z_sname,2)
        d.put_att("long_name",pds.z_type)
        if pds.z_type=="Hybrid level"
          list_pv = gds.list_pv
          zval = pds.z_value[0]
          tmp = Array.new
          z[0].each{|n|
            tmp += [n-1,n]
          }
          tmp.uniq!
          tmp.sort!
          ap = list_pv[0...list_pv.length/2]
          b = list_pv[list_pv.length/2..-1]
          z[0] = {"value"=>NArray.to_na(z[0]),"name"=>zval["name"]}
          z[1] = {"value"=>ap[tmp], "name"=>"ap_half_lev", "long_name"=>"hybrid ap at half levels", "units"=>"Pa"}
          z[2] = {"value"=>b[tmp], "name"=>"b_half_lev", "long_name"=>"hybrid b at half levels", "units"=>"1"}
        elsif nz==1
          zval = pds.z_value[0]
          z = NArray.to_na(z[0])
          d.put_att("units",zval["units"])
          d.put_att("value_type",zval["name"])
        else
          nz.times{|mm|
            m = nz-m-1
            zval = pds.z_value[m]
            z[m+1] = {"value"=>NArray.to_na(z[m]),"name"=>zval["name"],"units"=>zval["units"]}
          }
          z[0] = {"value"=>NArray.sint(z[0].length).indgen,"name"=>"level number"}
        end
        d.put(z)
      else
        va.put_att("level", pds.z_type)
        pds.z_value.each{|zz|
          va.put_att(zz["name"].gsub(/\s/,"_"), "#{zz["value"]} #{zz["units"]}")
        }
      end
    else
      va.put_att("level", pds.z_type)
    end
    len = sgms.length
    time = Array.new(len)
    len.times{|i|
      time[i] = get_time(sgms[i].pds.date)
    }
#          sgms.each{|sgm|
#            time.push( get_time(sgm.pds.date) )
#          }
    time.uniq!
    if @@always_time_seq or time.length>1
      time = NArray.to_na(time)
      d = va.def_dim("time",-1)
      d.put(time)
      d.put_att("long_name","time")
      d.put_att("units","hours since 1900-01-01 00:00:0.0")
    else
      date = pds.date
      va.put_att("time","#{date[0]} #{"%02d"%date[1][0]}:#{"%02d"%date[1][1]}:0.0")
    end
  end
  return va
end

Instance Method Details

#att(key) ⇒ Object 



1931
1932
1933
 
# File 'lib/numru/gphys/grib.rb', line 1931

def att(key)
  @attr[key]
end

#att_names ⇒ Object 



1934
1935
1936
 
# File 'lib/numru/gphys/grib.rb', line 1934

def att_names
  @attr.keys
end

#def_dim(name, index) ⇒ Object 



1918
1919
1920
1921
1922
1923
1924
1925
1926
 
# File 'lib/numru/gphys/grib.rb', line 1918

def def_dim(name,index)
  d = GribDim.new(self,name)
  if index==-1
    @dims.push(d)
  else
    @dims[index] = d
  end
  return d
end

#dim(index) ⇒ Object 



1910
1911
1912
1913
1914
 
# File 'lib/numru/gphys/grib.rb', line 1910

def dim(index)
  index = dim_names.index(index) if String===index
  return nil if index.nil?
  @dims[index]
end

#dim_names ⇒ Object 



1907
1908
1909
 
# File 'lib/numru/gphys/grib.rb', line 1907

def dim_names
  @dims.collect{|d| d.name }
end

#file ⇒ Object 



1885
1886
1887
 
# File 'lib/numru/gphys/grib.rb', line 1885

def file
  @file
end

#get(*indices) ⇒ Object Also known as: [], val 



1949
1950
1951
1952
1953
1954
1955
1956
1957
1958
1959
1960
1961
1962
1963
1964
1965
1966
1967
1968
1969
1970
1971
1972
1973
1974
1975
1976
1977
1978
1979
1980
1981
1982
1983
1984
1985
1986
1987
1988
1989
1990
1991
1992
1993
1994
1995
1996
1997
1998
1999
2000
2001
2002
2003
2004
2005
2006
2007
2008
2009
2010
2011
2012
2013
2014
2015
2016
2017
2018
2019
2020
2021
2022
2023
2024
2025
2026
2027
2028
2029
2030
2031
2032
2033
2034
2035
2036
2037
2038
2039
2040
2041
2042
2043
2044
2045
2046
2047
2048
2049
2050
2051
2052
2053
2054
2055
2056
2057
2058
2059
2060
2061
2062
2063
2064
2065
2066
2067
2068
2069
2070
2071
2072
2073
2074
2075
2076
2077
2078
2079
2080
2081
2082
2083
2084
2085
2086
2087
2088
2089
2090
2091
2092
2093
2094
2095
2096
2097
2098
2099
 
# File 'lib/numru/gphys/grib.rb', line 1949

def get(*indices)
  if @obj.nil?
    return nil
  elsif NArray===@obj||NArrayMiss===@obj
    return @obj[*indices]
  else
    sha = shape
    ndims = sha.length
    after_slice = false
    dimdel = nil
    if indices.length!=0
      after_slicer = Array.new(2,true)
      imap = (3..ndims).collect{Hash.new}  # index map org-->current for z,t
      if indices[0] == false
        indices[0] = [true]*(ndims-indices.length+1)
      elsif indices[-1] == false
        indices[-1] = [true]*(ndims-indices.length+1)
      elsif indices.include?(false)
        raise "invalid indices"
      elsif ndims!=indices.length
        raise "invalid indices"
      end
      dimdel = []
      ndims.times{|n|
        ind = indices[n]
        dlen = sha[n]
        case ind
        when true
          if n<=1
            indices[n] = 0..dlen-1
          else
            (0..dlen-1).each{|i| imap[n-2][i] = i}
          end
          next
        when Fixnum
          sha[n] = 1
          imap[n-2][ind] = 0
          dimdel.push(n)
          next
        when Range
          f = ind.first
          f += dlen if f<0
          e = ind.end
          e += dlen if e<0
          e -= 1 if ind.exclude_end?
          sha[n] = e-f+1
          if n<=1
            indices[n] = f..e
          else
            (f..e).each{|i| imap[n-2][i] = i-f}
          end
          next
        when Hash
          ran, step = ind.to_a[0]
          f = ran.first
          f += dlen if f<0
          e = ran.end
          e += dlen if e<0
          e -= 1 if ran.exclude_end?
          sha[n] = (e-f)/step + 1
          if n<=1
            indices[n] = f..e
            after_slice = true
            after_slicer[n] = (0...sha[n]).collect{|i| i*step}
          else
            ia = NArray.int(sha[n]).indgen! * step + f
            indices[n] = NArray.to_na(ia)
            j=0
            (f..e).step(step){|i| imap[n-2][i] = j; j+=1} if n>=2
          end
          next
        when Array, NArray
          ind = ind.collect{|i|
              raise("Array indices must be integers") unless Integer===i
              i>=0 ? i : i+dlen
          }
          sha[n] = ind.length
          if n<=1
            f = ind.min
            e = ind.max
            indices[n] = f..e
            after_slice = true
            after_slicer[n] = ind.collect{|i| i-f}
          else
            (0...ind.length).each{|i| imap[n-2][ind[i]] = i }
          end
        else
          raise "invalid indices"
        end
      }
      if rank>2
        subsetzt = true
      else
        subsetzt = false
      end
    else
      subsetzt = false
    end
    value = @miss ? NArrayMiss.sfloat(*sha) : NArray.sfloat(*sha)
    if rank==2
      vz = nil
     vt = nil
      index = []
    elsif rank==4
      vz = dim(2).val
      vt = dim(3).val
      index = Array.new(2)
    elsif !dim_names.include?("time")
      vz = dim(2).val
      vt = nil
      index = Array.new(1)
    else
      vt = dim(2).val
      vz = nil
      index = Array.new(1)
    end
    @obj.each{|sgm|
      pds = sgm.pds
      if vz
        index[0] = vz.eq(pds.z_value[0]["value"]).where[0]
      end
      if vt
        index[-1] = vt.eq(get_time(pds.date)).where[0]
      end
      if subsetzt
        to_read = true
        index.each_with_index{|i,d| to_read &&= imap[d][i]}
        next unless to_read
      end
      bds = sgm.bds
      if indices.length==0 || (indices[0]==true&&indices[1]==true)
        val = bds.value(shape[0..1])
      else
        val = bds.value(shape[0..1],indices[0..1])
      end
      d = pds.dfact
      val = val*10.0**(-d)
      val = val[*after_slicer] if after_slice
      if !subsetzt
        idx = index
      else
        idx = []
        index.each_with_index{|i,d| idx.push(imap[d][i])}
      end
      value[true,true,*idx] = val
    }
    dimdel.reverse_each{|d| sha.delete_at(d)} if dimdel
    value.reshape!(*sha)
    return value
  end
end

#inspect ⇒ Object 



2105
2106
2107
 
# File 'lib/numru/gphys/grib.rb', line 2105

def inspect
  "GribVar: #{@file.path}?var=#{@name}"
end

#name ⇒ Object 



1888
1889
1890
 
# File 'lib/numru/gphys/grib.rb', line 1888

def name
  @name
end

#ndims ⇒ Object 



1915
1916
1917
 
# File 'lib/numru/gphys/grib.rb', line 1915

def ndims
  @dims.length
end

#put(val) ⇒ Object 



2102
2103
2104
 
# File 'lib/numru/gphys/grib.rb', line 2102

def put(val)
  @obj = val
end

#put_att(key, val) ⇒ Object Also known as: set_att 



1927
1928
1929
 
# File 'lib/numru/gphys/grib.rb', line 1927

def put_att(key,val)
  @attr[key]=val
end

#rank ⇒ Object 



1897
1898
1899
 
# File 'lib/numru/gphys/grib.rb', line 1897

def rank
  dim_names.length
end

#set_miss(miss) ⇒ Object 



1894
1895
1896
 
# File 'lib/numru/gphys/grib.rb', line 1894

def set_miss(miss)
  @miss = miss
end

#set_sgms(sgms) ⇒ Object 



1891
1892
1893
 
# File 'lib/numru/gphys/grib.rb', line 1891

def set_sgms(sgms)
  @obj = sgms
end

#shape ⇒ Object 



1937
1938
1939
 
# File 'lib/numru/gphys/grib.rb', line 1937

def shape
  @dims.collect{|d| d.length }
end

#total ⇒ Object 



1900
1901
1902
1903
1904
1905
1906
 
# File 'lib/numru/gphys/grib.rb', line 1900

def total
  total = 1
  ndims.times{|i|
    total = total*dim(i).length
  }
  total
end

#typecode ⇒ Object 



1940
1941
1942
1943
1944
1945
1946
1947
1948
 
# File 'lib/numru/gphys/grib.rb', line 1940

def typecode
  if @obj.nil?
    nil
  elsif Array===@obj
    NArray::SFLOAT
  else
    @obj.typecode
  end
end