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
90 91 92 |
# File 'lib/gs2crmod/read_netcdf.rb', line 90 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
115 116 117 |
# File 'lib/gs2crmod/read_netcdf.rb', line 115 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
77 78 79 80 81 82 83 84 85 86 87 88 |
# File 'lib/gs2crmod/read_netcdf.rb', line 77 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
62 63 64 65 66 67 68 69 70 71 72 73 |
# File 'lib/gs2crmod/read_netcdf.rb', line 62 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
93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
# File 'lib/gs2crmod/read_netcdf.rb', line 93 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' 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
74 75 76 |
# File 'lib/gs2crmod/read_netcdf.rb', line 74 def ends(dims, ) dims.map{|d| dim_end(d.name, )} end |
#graphkit(variable, options) ⇒ Object
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 |
# File 'lib/gs2crmod/read_netcdf.rb', line 118 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(: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 |
# File 'lib/gs2crmod/read_netcdf.rb', line 48 def read_variable(varname, ) #start = get_start(dims, options) dims = dimensions(varname) narray = @file.var(varname).get('start'=>starts(dims, ), 'end'=>ends(dims, )) shape = narray.shape shape.delete_if{|i| i==1} #p 'shape', shape; STDIN.gets narray.reshape!(*shape) narray end |
#starts(dims, options) ⇒ Object
59 60 61 |
# File 'lib/gs2crmod/read_netcdf.rb', line 59 def starts(dims, ) dims.map{|d| dim_start(d.name, )} end |