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
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