Class: CodeRunner::Gs2::NetcdfSmartReader
- Inherits:
-
Object
- Object
- CodeRunner::Gs2::NetcdfSmartReader
- Defined in:
- lib/gs2crmod/read_netcdf.rb
Direct Known Subclasses
Instance Method Summary collapse
- #axiskit(variable, options) ⇒ Object
- #check_no_r(non_flat_dims) ⇒ Object
- #dim_end(name, options) ⇒ Object
- #dim_start(name, options) ⇒ Object
- #dimension_variable_name(n) ⇒ Object
- #dimensions(varname) ⇒ Object
- #ends(dims, options) ⇒ Object
- #graphkit(variable, options) ⇒ Object
-
#initialize(file) ⇒ NetcdfSmartReader
constructor
A new instance of NetcdfSmartReader.
- #read_variable(varname, options) ⇒ Object
- #starts(dims, options) ⇒ Object
Constructor Details
#initialize(file) ⇒ NetcdfSmartReader
Returns a new instance of NetcdfSmartReader.
41 42 43 |
# File 'lib/gs2crmod/read_netcdf.rb', line 41 def initialize(file) @file = file end |
Instance Method Details
#axiskit(variable, options) ⇒ Object
103 104 105 |
# File 'lib/gs2crmod/read_netcdf.rb', line 103 def axiskit(variable, ) GraphKit::AxisKit.autocreate(data: read_variable(variable, ), units: @file.var(variable).att('Units').get, title: @file.var(variable).att('Description').get.sub(/(,|summed|average).*$/, '').sub(/[vV]alues of (the )?/, '').sub(/ coordinate/, '')) end |
#check_no_r(non_flat_dims) ⇒ Object
130 131 132 |
# File 'lib/gs2crmod/read_netcdf.rb', line 130 def check_no_r(non_flat_dims) raise "Please specify the r index for real or imaginary" if non_flat_dims.include? @file.dim('r') end |
#dim_end(name, options) ⇒ Object
90 91 92 93 94 95 96 97 98 99 100 101 |
# File 'lib/gs2crmod/read_netcdf.rb', line 90 def dim_end(name, ) sym = name.to_sym if i=[sym + :_index] return i-1 elsif i=[sym + :_element] return i elsif i=[sym + :max] return i else return -1 end end |
#dim_start(name, options) ⇒ Object
75 76 77 78 79 80 81 82 83 84 85 86 |
# File 'lib/gs2crmod/read_netcdf.rb', line 75 def dim_start(name, ) sym = name.to_sym if i=[sym + :_index] return i-1 elsif i=[sym + :_element] return i elsif i=[sym + :min] return i else return 0 end end |
#dimension_variable_name(n) ⇒ Object
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# File 'lib/gs2crmod/read_netcdf.rb', line 106 def dimension_variable_name(n) case n when 'X' 'kx' when 'Y' 'ky' when 'z' 'kz' when 'e' 'energy' when 'l' 'lambda' when 't' n when 'm' 'hermite' when 'p' 'hankel' when 's' 's' else raise "Unknown dimension #{n}" end end |
#dimensions(varname) ⇒ Object
44 45 46 47 |
# File 'lib/gs2crmod/read_netcdf.rb', line 44 def dimensions(varname) #p 'varname', varname @file.var(varname).dims end |
#ends(dims, options) ⇒ Object
87 88 89 |
# File 'lib/gs2crmod/read_netcdf.rb', line 87 def ends(dims, ) dims.map{|d| dim_end(d.name, )} end |
#graphkit(variable, options) ⇒ Object
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 |
# File 'lib/gs2crmod/read_netcdf.rb', line 133 def graphkit(variable, ) non_flat_dims=dimensions(variable).find_all{|dim| dim_start(dim.name, ) != dim_end(dim.name, ) and dim.length != 1} check_no_r(non_flat_dims) axiskits = non_flat_dims.map{|dim| dimvar = dimension_variable_name(dim.name); axiskit(dimvar, )} + [axiskit(variable, )] hash = {} axes = [:x, :y, :z, :f] axiskits.each_with_index{|ax, i| hash[axes[i]] = ax} kit = GraphKit.autocreate(hash) opts = .dup opts.delete(:modify_variable) opts.delete(:graphkit_name) #kit.data[0].title += " with options: " + opts.to_s kit.data[0].title += " " + opts.to_s.gsub(/_(index|element)/, '') if kit.zlabel kit.zlabel = "'#{kit.zlabel}' rotate by 90" #kit.zlabel = nil end kit end |
#read_variable(varname, options) ⇒ Object
48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
# File 'lib/gs2crmod/read_netcdf.rb', line 48 def read_variable(varname, ) #start = get_start(dims, options) if varname == 's' # dummy response for species index return NArray.float(@nspec||1) end dims = dimensions(varname) narray = @file.var(varname).get('start'=>starts(dims, ), 'end'=>ends(dims, )) if [:modify_variable] hsh = dims.inject({}){|hsh, dim| opts = .dup opts[:modify_variable] = nil dimval = read_variable(dimension_variable_name(dim.name), opts) hsh[dim.name] = dimval hsh } narray = [:modify_variable].call(varname, narray, hsh) end shape = narray.shape shape.delete_if{|i| i==1} #p 'shape', shape; STDIN.gets narray.reshape!(*shape) narray end |
#starts(dims, options) ⇒ Object
72 73 74 |
# File 'lib/gs2crmod/read_netcdf.rb', line 72 def starts(dims, ) dims.map{|d| dim_start(d.name, )} end |