def test_amr_kdtree_coverage():
return #TESTDISABLED
domain_dims = (32, 32, 32)
data = np.zeros(domain_dims) + 0.25
fo = [
ic.CoredSphere(0.05, 0.3, [0.7, 0.4, 0.75], {"density": (0.25, 100.0)})
]
rc = [fm.flagging_method_registry["overdensity"](8.0)]
ug = load_uniform_grid({"density": data}, domain_dims, 1.0)
ds = refine_amr(ug, rc, fo, 5)
kd = AMRKDTree(ds)
volume = kd.count_volume()
yield assert_equal, volume, \
np.prod(ds.domain_right_edge - ds.domain_left_edge)
cells = kd.count_cells()
true_cells = ds.all_data().quantities['TotalQuantity']('Ones')[0]
yield assert_equal, cells, true_cells
# This largely reproduces the AMRKDTree.tree.check_tree() functionality
tree_ok = True
for node in kd.tree.trunk.depth_traverse():
if node.grid is None:
continue
grid = ds.index.grids[node.grid - kd._id_offset]
dds = grid.dds
gle = grid.LeftEdge
nle = node.get_left_edge()
nre = node.get_right_edge()
li = np.rint((nle - gle) / dds).astype('int32')
ri = np.rint((nre - gle) / dds).astype('int32')
dims = (ri - li).astype('int32')
tree_ok *= np.all(grid.LeftEdge <= nle)
tree_ok *= np.all(grid.RightEdge >= nre)
tree_ok *= np.all(dims > 0)
yield assert_equal, True, tree_ok
def test_amr_kdtree_coverage():
return #TESTDISABLED
domain_dims = (32, 32, 32)
data = np.zeros(domain_dims) + 0.25
fo = [ic.CoredSphere(0.05, 0.3, [0.7, 0.4, 0.75],
{"density": (0.25, 100.0)})]
rc = [fm.flagging_method_registry["overdensity"](8.0)]
ug = load_uniform_grid({"density": data}, domain_dims, 1.0)
ds = refine_amr(ug, rc, fo, 5)
kd = AMRKDTree(ds)
volume = kd.count_volume()
yield assert_equal, volume, \
np.prod(ds.domain_right_edge - ds.domain_left_edge)
cells = kd.count_cells()
true_cells = ds.all_data().quantities['TotalQuantity']('Ones')[0]
yield assert_equal, cells, true_cells
# This largely reproduces the AMRKDTree.tree.check_tree() functionality
tree_ok = True
for node in depth_traverse(kd.tree.trunk):
if node.grid is None:
continue
grid = ds.index.grids[node.grid - kd._id_offset]
dds = grid.dds
gle = grid.LeftEdge
nle = get_left_edge(node)
nre = get_right_edge(node)
li = np.rint((nle-gle)/dds).astype('int32')
ri = np.rint((nre-gle)/dds).astype('int32')
dims = (ri - li).astype('int32')
tree_ok *= np.all(grid.LeftEdge <= nle)
tree_ok *= np.all(grid.RightEdge >= nre)
tree_ok *= np.all(dims > 0)
yield assert_equal, True, tree_ok
# In this example we will show how to use the AMRKDTree to take a simulation
# with 8 levels of refinement and only use levels 0-3 to render the dataset.
# We begin by loading up yt, and importing the AMRKDTree
import numpy as np
import yt
from yt.utilities.amr_kdtree.api import AMRKDTree
# Load up a dataset and define the kdtree
ds = yt.load('IsolatedGalaxy/galaxy0030/galaxy0030')
kd = AMRKDTree(ds)
# Print out specifics of KD Tree
print("Total volume of all bricks = %i" % kd.count_volume())
print("Total number of cells = %i" % kd.count_cells())
# Define a camera and take an volume rendering.
tf = yt.ColorTransferFunction((-30, -22))
cam = ds.camera([0.5, 0.5, 0.5], [0.2, 0.3, 0.4], 0.10, 256,
tf, volume=kd)
tf.add_layers(4, 0.01, col_bounds=[-27.5, -25.5], colormap='RdBu_r')
cam.snapshot("v1.png", clip_ratio=6.0)
# This rendering is okay, but lets say I'd like to improve it, and I don't want
# to spend the time rendering the high resolution data. What we can do is
# generate a low resolution version of the AMRKDTree and pass that in to the
# camera. We do this by specifying a maximum refinement level of 6.
kd_low_res = AMRKDTree(ds, max_level=6)
print(kd_low_res.count_volume())
# Load up a dataset and define the kdtree
ds = yt.load('IsolatedGalaxy/galaxy0030/galaxy0030')
im, sc = yt.volume_render(ds, 'density', fname='v0.png')
sc.camera.set_width(ds.arr(100, 'kpc'))
render_source = sc.get_source(0)
kd = render_source.volume
# Print out specifics of KD Tree
print("Total volume of all bricks = %i" % kd.count_volume())
print("Total number of cells = %i" % kd.count_cells())
new_source = ds.all_data()
new_source.max_level = 3
kd_low_res = AMRKDTree(ds, data_source=new_source)
print(kd_low_res.count_volume())
print(kd_low_res.count_cells())
# Now we pass this in as the volume to our camera, and render the snapshot
# again.
render_source.set_volume(kd_low_res)
render_source.set_field('density')
sc.render()
sc.save("v1.png", sigma_clip=6.0)
# This operation was substantiall faster. Now lets modify the low resolution
# rendering until we find something we like.
tf = render_source.transfer_function
tf.clear()
tf.add_layers(4,