示例#1
0
def new_mesh_sampler(camera, render_source, engine):
    params = ensure_code_unit_params(camera._get_sampler_params(render_source))
    args = (
        np.atleast_3d(params['vp_pos']),
        np.atleast_3d(params['vp_dir']),
        params['center'],
        params['bounds'],
        np.atleast_3d(params['image']).astype('float64'),
        params['x_vec'],
        params['y_vec'],
        params['width'],
    )
    kwargs = {'lens_type': params['lens_type']}
    if engine == 'embree':
        sampler = mesh_traversal.EmbreeMeshSampler(*args, **kwargs)
    elif engine == 'yt':
        sampler = bounding_volume_hierarchy.BVHMeshSampler(*args, **kwargs)
    return sampler
示例#2
0
文件: utils.py 项目: pshriwise/yt
def new_mesh_sampler(camera, render_source, engine):
    params = ensure_code_unit_params(camera._get_sampler_params(render_source))
    args = (
        np.atleast_3d(params["vp_pos"]),
        np.atleast_3d(params["vp_dir"]),
        params["center"],
        params["bounds"],
        np.atleast_3d(params["image"]).astype("float64"),
        params["x_vec"],
        params["y_vec"],
        params["width"],
    )
    kwargs = {"lens_type": params["lens_type"]}
    if engine == "embree":
        sampler = mesh_traversal.EmbreeMeshSampler(*args, **kwargs)
    elif engine == "yt":
        sampler = bounding_volume_hierarchy.BVHMeshSampler(*args, **kwargs)
    return sampler