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