athenakit.athena_data

Core data container for a single AthenaK simulation snapshot.

Top-level loader

athenakit.athena_data.load(filename, **kwargs)

AthenaData

class athenakit.athena_data.AthenaData(num=0, version='1.0')

Bases: object

__init__(num=0, version='1.0')

add_data(name, data)

add_data_func(name, func)

add_gr_data()

average(var, weights='ones', where=None, **kwargs)

config()

config_coord()

d(var, **kwargs)

data(var, **kwargs)

property data_list

divergence(fx, fy, fz, **kwargs)

get_avg(varl, *args, **kwargs)

get_hist(varl, bins=128, scales='log', weights='vol', **kwargs)

get_hist2d(varl, bins=128, scales='log', weights='vol', **kwargs)

get_profile(bin_var, varl, bins=256, weights='vol', **kwargs)

get_profile2d(bin_varl, varl, bins=256, weights='vol', **kwargs)

get_slice(varl, **kwargs)

get_slice_centers(zoom=0, level=0, xyz=[], axis='z')

get_slice_coord(zoom=0, level=0, xyz=[], axis='z')

get_slice_faces(zoom=0, level=0, xyz=[], axis='z')

get_slice_for_plot(var='dens', key=None, vec=None, stream=None, vecx='velx', vecy='vely', zoom=0, level=0, xyz=[], unit=1.0, xyunit=1.0, axis='z')

get_sum(varl, *args, **kwargs)

gradient(f, axis=None, edge_order=1, **kwargs)

header(blockname, keyname, astype=<class 'str'>, default=None)

histogram(*args, **kwargs)

histogram2d(*args, **kwargs)

include_ghost()

interpolate(varl='dens', points=[[0.0, 0.0, 0.0]])

load(filename, config=True, add_gr=False, **kwargs)

load_athdf(filename)

load_binary(filename)

load_hdf5(filename, **kwargs)

load_pickle(filename, **kwargs)

property mb_dx

property n

plot(*args, **kwargs)

plot_hist2d(*args, **kwargs)

plot_image(x, y, img, title='', label='', xlabel='X', ylabel='Y', xscale='linear', yscale='linear', cmap='viridis', norm='log', save=False, figfolder=None, figlabel='', figname='', dpi=200, fig=None, ax=None, colorbar=True, returnall=False, aspect='auto', xticks=None, yticks=None, xticklabels=None, yticklabels=None, **kwargs)

plot_phase(var='dens,temp', key='vol', bins=128, range=None, weights='vol', where=None, title='', label='', xlabel=None, ylabel=None, xscale='log', yscale='log', unit=1.0, cmap='viridis', norm='log', extent=None, density=False, save=False, savepath='', figdir='../figure/Simu_', figpath='', x=None, y=None, xshift=0.0, xunit=1.0, yshift=0.0, yunit=1.0, fig=None, ax=None, dpi=128, **kwargs)

plot_profile(var='r,dens', unit=1.0, xunit=1.0, bins=256, weights='vol', range=None, where=None, fig=None, ax=None, dpi=200, xscale='log', yscale='log', xlabel=None, ylabel=None, returnall=False, **kwargs)

plot_slice(var='dens', key=None, vec=None, stream=None, vecx='velx', vecy='vely', zoom=0, level=0, xyz=[], unit=1.0, xyunit=1.0, axis='z', fig=None, ax=None, dpi=200, norm='log', cmap='viridis', aspect='equal', xlabel=None, ylabel=None, title='', label='', colorbar=True, quiver_para={}, stream_para={}, returnall=False, **kwargs)

plot_slice_by_prof(var='dens', key=None, data=None, zoom=0, level=0, xyz=[], unit=1.0, bins=None, title='', label='', xlabel='X', ylabel='Y', cmap='viridis', norm='log', save=False, figdir='../figure/Simu_', figpath=None, savepath='', savelabel='', figlabel='', dpi=200, vec=None, stream=None, circle=True, fig=None, ax=None, xyunit=1.0, colorbar=True, returnall=False, stream_color='k', stream_linewidth=1.0, stream_arrowsize=1.0, vecx='velx', vecy='vely', vel_method='ave', aspect='equal', **kwargs)

plot_stream(dpi=200, fig=None, ax=None, x=None, y=None, u=None, v=None, xyunit=1.0, color='k', linewidth=None, arrowsize=None)

save(filename, except_keys=[], default_except_keys=['binary', 'h5file', 'h5dic', 'coord', 'data_raw', 'data_func', 'mb_list'], **kwargs)

save_hdf5(dic, filename)

save_pickle(dic, filename)

set_avg(*args, **kwargs)

set_hist(varl, key=None, bins=128, scales='log', weights='vol', **kwargs)

set_hist2d(varl, key=None, bins=128, scales='log', weights='vol', **kwargs)

set_profile(bin_var, varl, key=None, bins=256, weights='vol', **kwargs)

set_profile2d(bin_varl, varl, key=None, bins=256, weights='vol', **kwargs)

set_slice(varl, key='z', **kwargs)

set_slice_by_profile(bin_varl, varl, key=None, **kwargs)

set_sum(*args, **kwargs)

slice(var='dens', zoom=0, level=0, xyz=[], axis='z')

sum(var, weights=1.0, where=None, **kwargs)

xyz(zoom=0, level=None, axis=None)

AthenaDataSet

A time-series container that holds multiple AthenaData snapshots keyed by output number.

class athenakit.athena_data.AthenaDataSet(version='1.0')

Bases: object

__call__(n=None)

Call self as a function.

__getitem__(n=None)

__init__(version='1.0')

property ad

clear()

items()

keys()

load(ns, path=None, dtype=None, verbose=False, **kwargs)

property ns

pop(n)

popitem()

values()