yt.geometry_tests / sphere_grid.py

import time
from yt.mods import *
from yt.geometry.selection_routines import SphereSelector
from yt.data_objects.selection_data_containers import YTSphereBase

class Dummy(object):
    radius = 0.1
    center = [0.5, 0.5, 0.5]

ii = Dummy()


sps = SphereSelector(ii)
pf = load("DD0039/output_0039")

spp = YTSphereBase([0.5, 0.5, 0.5], 0.1, pf=pf)
spp._get_list_of_grids()

gi = sps.select_grids(pf.h.grid_left_edge, pf.h.grid_right_edge)
print gi.sum(),
print gi.sum() / float(len(spp._grids))

pb = get_pbar("Counting", gi.sum())
scount = icount = 0
for i,g in enumerate(pf.h.grids[gi]):
    pb.update(i)
    scount += sps.count_cells(g)
    icount += spp._get_point_indices(g)[0].size
pb.finish()
print scount, icount
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.