def read_phi(self):
"""
Read the electrostatic potenential from the NetCDF file.
"""
self.phi = field.get_field_final_timestep(self.cdf_file, 'phi', None)
self.phi = field.field_to_real_space_final_timestep(self.phi)*\
self.rho_star
"""
shape = field.shape
nx = shape[0]
nky = shape[1]
ny = 2*(nky - 1)
nth = shape[2]
field_real_space = np.empty([nx,ny,nth],dtype=float)
field_real_space = np.fft.irfft2(field, axes=[0,1])
field_real_space = np.roll(field_real_space, int(nx/2), axis=0)
return field_real_space*nx*ny
cdf_file = sys.argv[1]
ncfile = Dataset(cdf_file, 'r')
phi = field.get_field_final_timestep(cdf_file, 'phi', None)
ntot = field.get_field_final_timestep(cdf_file, 'ntot', 0)
tperp = field.get_field_final_timestep(cdf_file, 'tperp', 0)
tpar = field.get_field_final_timestep(cdf_file, 'tpar', 0)
ky = np.array(ncfile.variables['ky'][:])
theta = np.array(ncfile.variables['theta'][:])
gradpar = np.array(ncfile.variables['gradpar'][:])
bmag = np.array(ncfile.variables['bmag'][:])
grho = np.array(ncfile.variables['grho'][:])
q_nc = np.array(ncfile.variables['es_heat_flux'][:, 0])
shape = phi.shape
nx = shape[0]
nky = shape[1]
ny = 2*(nky - 1)
nth = shape[2]