Module: CodeRunner::Gs2::GSLVectorComplexes

Included in:
CodeRunner::Gs2
Defined in:
lib/gs2crmod/gsl_data.rb

Instance Method Summary collapse

Instance Method Details

#phi_along_field_line_gsl_vector_complex(options) ⇒ Object 



1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
 
# File 'lib/gs2crmod/gsl_data.rb', line 1111

def phi_along_field_line_gsl_vector_complex(options)
Dir.chdir(@directory) do
#     eputs options[:ky]
#     eputs Dir.pwd
    #eputs "Start phi_along_field_line"
    options.convert_to_index(self, :ky)
    if options[:t_index] or options[:t]
      #extra option required is t_index
      raise CRFatal.new("write_phi_over_time is not enabled so this function won't work") unless @write_phi_over_time

      options.convert_to_index(self, :t)
      case @grid_option
      when "single"
        temp = GSL::Vector.alloc(netcdf_file.var('phi_t').get({'start' => [0,0,0,0, options[:t_index] - 1], 'end' => [-1,-1,0,0, options[:t_index] - 1]}).to_a[0][0][0].flatten)
      when "range"
        a = netcdf_file.var('phi_t').get({'start' => [0, 0, options[:kx_index]-1, options[:ky_index] - 1, options[:t_index] - 1], 'end' => [-1, -1, options[:kx_index]-1, options[:ky_index] - 1, options[:t_index]-1]})
        #temp =  GSL::Vector.alloc(a.to_a[0].values_at(*kx_elements).flatten)
        temp =  GSL::Vector.alloc(a.to_a[0][0].flatten)
      when "box"
        options.convert_to_index(self, :ky, :kx)
        kx_elements = gsl_vector('linked_kx_elements', options).to_a
#         pp kx_elements
        a = netcdf_file.var('phi_t').get({
          'start' => [0,0,0,options[:ky_index] - 1, options[:t_index] - 1],
          'end' => [-1,-1,-1, options[:ky_index] - 1, options[:t_index] - 1]
        }).to_a[0][0].values_at(*kx_elements).flatten
#         pp a.index(nil)
#         temp = GSL::Vector.alloc(netcdf_file.var('phi').get.to_a[options[:ky_index] - 1 ].values_at(*kx_elements).flatten)
        #ep a
        temp = GSL::Vector.alloc(a)
      end

      #eputs "End phi_along_field_line"
      return GSL::Vector::Complex.alloc(temp.subvector_with_stride(0, 2), temp.subvector_with_stride(1, 2))
    else
      case @grid_option
      when "single"
        temp = GSL::Vector.alloc(netcdf_file.var('phi').get({'start' => [0,0, 0, 0], 'end' => [-1,-1,0,0]}).to_a.flatten)
      when "range"
        a = netcdf_file.var('phi').get({'start' => [0, 0, 0, options[:ky_index] - 1], 'end' => [-1, -1, -1, options[:ky_index] - 1]})
        #temp =  GSL::Vector.alloc(a.to_a[0].values_at(*kx_elements).flatten)
        temp =  GSL::Vector.alloc(a.to_a[0][0].flatten)
      when "box"
        ep 'kx_elements', kx_elements = gsl_vector('linked_kx_elements', options).to_a
        a = netcdf_file.var('phi').get({'start' => [0, 0, 0, options[:ky_index] - 1], 'end' => [-1, -1, -1, options[:ky_index] - 1]})
        temp =  GSL::Vector.alloc(a.to_a[0].values_at(*kx_elements).flatten)
      else
        raise "invalid grid option"
      end

      vector = GSL::Vector::Complex.alloc(temp.subvector_with_stride(0, 2), temp.subvector_with_stride(1, 2))
      #ep 'vector', vector.real
      return vector
    end
  end
#       eputs data; gets
end

#phi_zonal_gsl_vector_complex(options) ⇒ Object

This function returns a complex GSL vector of the zonal (ky=0) component of phi/phi_t at a given theta index and time index if write_phi_over_time was enabled during the simulation



1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
 
# File 'lib/gs2crmod/gsl_data.rb', line 1171

def phi_zonal_gsl_vector_complex(options)
  Dir.chdir(@directory) do
      raise CRFatal.new("write_eigenfunc is not enabled so this function won't work") unless @write_eigenfunc
      options.convert_to_index(self, :t)
      a = netcdf_file.var('phi0').get({
        'start' => [0,0,0, options[:t_index] - 1],
        'end' => [-1,-1,0, options[:t_index] - 1]
      })
      vector = GSL::Vector::Complex.alloc(GSL::Vector.alloc(a[0,0..-1,0,0]), GSL::Vector.alloc(a[1,0..-1,0,0]))
      return vector
  end
end