def generate_container_data(test_dir):
file_to_write = os.path.join(test_dir, "containers.h5")
ds = yt.load("enzo_tiny_cosmology/DD0046/DD0046")
for key, value in cont_dict.items():
cont = getattr(ds, camelcase_to_underscore(value[0]), None)
if cont is not None:
c = cont(*value[1], **value[2])
c["density"].write_hdf5(file_to_write, dataset_name=key)
# Special handling
sp1 = ds.sphere(*cont_dict["sp1"][1])
prj = ds.proj("density", 1, data_source=sp1)
prj["density"].write_hdf5(file_to_write, dataset_name="prj3")
sp2 = ds.sphere(*cont_dict["sp2"][1])
conditions = ["obj['kT'] > 0.5"]
cr = sp2.cut_region(conditions)
cr["density"].write_hdf5(file_to_write, dataset_name="cr")
for i, grid in enumerate(ds.index.grids):
grid["density"].write_hdf5(file_to_write, dataset_name="grid_%04d" % (i+1))
def generate_container_data(test_dir):
file_to_write = os.path.join(test_dir, "containers.h5")
ds = yt.load("enzo_tiny_cosmology/DD0046/DD0046")
for key, value in cont_dict.items():
cont = getattr(ds, camelcase_to_underscore(value[0]), None)
if cont is not None:
c = cont(*value[1], **value[2])
c["density"].write_hdf5(file_to_write, dataset_name=key)
# Special handling
sp1 = ds.sphere(*cont_dict["sp1"][1])
prj = ds.proj("density", 1, data_source=sp1)
prj["density"].write_hdf5(file_to_write, dataset_name="prj3")
sp2 = ds.sphere(*cont_dict["sp2"][1])
conditions = ["obj['kT'] > 0.5"]
cr = sp2.cut_region(conditions)
cr["density"].write_hdf5(file_to_write, dataset_name="cr")
for i, grid in enumerate(ds.index.grids):
grid["density"].write_hdf5(file_to_write,
dataset_name="grid_%04d" % (i + 1))
def __new__(cls, data_source, *args, **kwargs):
inst = object.__new__(cls)
inst.data_source = data_source
for f in inst.keys():
setattr(inst, camelcase_to_underscore(f), inst[f])
return inst
