def triangle(cfg, size=0.5):
'''Rotating triangle with edge coloring specified in a vertex attribute'''
b = size * math.sqrt(3) / 2.0
c = size * 1 / 2.
cfg.duration = 3.
cfg.aspect_ratio = (1, 1)
colors_data = array.array(
'f', [0.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])
colors_buffer = ngl.BufferVec4(data=colors_data)
triangle = ngl.Triangle((-b, -c, 0), (b, -c, 0), (0, size, 0))
p = ngl.Program(fragment=cfg.get_frag('color'),
vertex=cfg.get_vert('triangle'))
p.update_vert_out_vars(color=ngl.IOVec4())
node = ngl.Render(triangle, p)
node.update_attributes(edge_color=colors_buffer)
animkf = [
ngl.AnimKeyFrameFloat(0, 0),
ngl.AnimKeyFrameFloat(cfg.duration / 3., -360 / 3., 'exp_in_out'),
ngl.AnimKeyFrameFloat(2 * cfg.duration / 3., -2 * 360 / 3.,
'exp_in_out'),
ngl.AnimKeyFrameFloat(cfg.duration, -360, 'exp_in_out')
]
node = ngl.Rotate(node, anim=ngl.AnimatedFloat(animkf))
return node
def _get_data_streamed_buffer_vec4_scene(cfg, scale, show_dbg_points):
duration = _N
cfg.duration = duration * scale
cfg.aspect_ratio = (1, 1)
size, data_size, = _N, _N * _N
time_anim = None
if scale != 1:
kfs = [
ngl.AnimKeyFrameFloat(0, 0),
ngl.AnimKeyFrameFloat(cfg.duration, duration),
]
time_anim = ngl.AnimatedTime(kfs)
pts_data = array.array('q')
assert pts_data.itemsize == 8
for i in range(duration):
offset = 10000 if i == 0 else 0
pts_data.extend([i * 1000000 + offset])
vec4_data = array.array('f')
for i in range(duration):
for j in range(data_size):
v = i / float(duration) + j / float(data_size * duration)
vec4_data.extend([v, v, v, v])
pts_buffer = ngl.BufferInt64(data=pts_data)
vec4_buffer = ngl.BufferVec4(data=vec4_data)
streamed_buffer = ngl.StreamedBufferVec4(data_size,
pts_buffer,
vec4_buffer,
time_anim=time_anim,
label='data')
streamed_block = ngl.Block(layout='std140',
label='streamed_block',
fields=(streamed_buffer, ))
shader_params = dict(data_size=data_size, size=size)
quad = ngl.Quad((-1, -1, 0), (2, 0, 0), (0, 2, 0))
program = ngl.Program(
vertex=_RENDER_STREAMEDBUFFER_VERT,
fragment=_RENDER_STREAMEDBUFFER_FRAG % shader_params,
)
program.update_vert_out_vars(var_uvcoord=ngl.IOVec2())
render = ngl.Render(quad, program)
render.update_frag_resources(streamed=streamed_block)
group = ngl.Group(children=(render, ))
if show_dbg_points:
cuepoints = _get_data_streamed_buffer_cuepoints()
group.add_children(get_debug_points(cfg, cuepoints))
return group
def compute_histogram(cfg, show_dbg_points=False):
cfg.duration = 10
cfg.aspect_ratio = (1, 1)
hsize, size, local_size = _N * _N, _N, _N // 2
data = array.array("f")
for i in range(size * size):
data.extend((
cfg.rng.uniform(0.0, 0.5),
cfg.rng.uniform(0.25, 0.75),
cfg.rng.uniform(0.5, 1.0),
1.0,
))
texture_buffer = ngl.BufferVec4(data=data)
texture = ngl.Texture2D(width=size, height=size, data_src=texture_buffer)
texture.set_format("r32g32b32a32_sfloat")
histogram_block = ngl.Block(layout="std140", label="histogram")
histogram_block.add_fields(
ngl.BufferUInt(hsize, label="r"),
ngl.BufferUInt(hsize, label="g"),
ngl.BufferUInt(hsize, label="b"),
ngl.UniformUIVec3(label="max"),
)
shader_params = dict(hsize=hsize, size=size, local_size=local_size)
group_size = hsize // local_size
clear_histogram_shader = _COMPUTE_HISTOGRAM_CLEAR % shader_params
clear_histogram_program = ngl.ComputeProgram(clear_histogram_shader,
workgroup_size=(local_size, 1,
1))
clear_histogram_program.update_properties(hist=ngl.ResourceProps(
writable=True))
clear_histogram = ngl.Compute(
workgroup_count=(group_size, 1, 1),
program=clear_histogram_program,
label="clear_histogram",
)
clear_histogram.update_resources(hist=histogram_block)
group_size = size // local_size
exec_histogram_shader = _COMPUTE_HISTOGRAM_EXEC % shader_params
exec_histogram_program = ngl.ComputeProgram(exec_histogram_shader,
workgroup_size=(local_size,
local_size, 1))
exec_histogram_program.update_properties(hist=ngl.ResourceProps(
writable=True))
exec_histogram = ngl.Compute(workgroup_count=(group_size, group_size, 1),
program=exec_histogram_program,
label="compute_histogram")
exec_histogram.update_resources(hist=histogram_block, source=texture)
exec_histogram_program.update_properties(source=ngl.ResourceProps(
as_image=True))
quad = ngl.Quad((-1, -1, 0), (2, 0, 0), (0, 2, 0))
program = ngl.Program(
vertex=_RENDER_HISTOGRAM_VERT,
fragment=_RENDER_HISTOGRAM_FRAG % shader_params,
)
program.update_vert_out_vars(var_uvcoord=ngl.IOVec2())
render = ngl.Render(quad, program, label="render_histogram")
render.update_frag_resources(hist=histogram_block)
group = ngl.Group(children=(clear_histogram, exec_histogram, render))
if show_dbg_points:
cuepoints = _get_compute_histogram_cuepoints()
group.add_children(get_debug_points(cfg, cuepoints))
return group
keyframes=_get_anim_kf(ngl.AnimKeyFrameBuffer, data)),
animated_buffer_vec4=lambda data: ngl.AnimatedBufferVec4(
keyframes=_get_anim_kf(ngl.AnimKeyFrameBuffer, data)),
animated_quat_mat4=lambda data: ngl.AnimatedQuat(
keyframes=_get_anim_kf(ngl.AnimKeyFrameQuat, data), as_mat4=True),
animated_quat_vec4=lambda data: ngl.AnimatedQuat(
keyframes=_get_anim_kf(ngl.AnimKeyFrameQuat, data), as_mat4=False),
array_float=lambda data: ngl.BufferFloat(data=data),
array_int=lambda data: ngl.BufferInt(data=data),
array_ivec2=lambda data: ngl.BufferIVec2(data=data),
array_ivec3=lambda data: ngl.BufferIVec3(data=data),
array_ivec4=lambda data: ngl.BufferIVec4(data=data),
array_mat4=lambda data: ngl.BufferMat4(data=data),
array_vec2=lambda data: ngl.BufferVec2(data=data),
array_vec3=lambda data: ngl.BufferVec3(data=data),
array_vec4=lambda data: ngl.BufferVec4(data=data),
single_bool=lambda data: ngl.UniformBool(data),
single_float=lambda data: ngl.UniformFloat(data),
single_int=lambda data: ngl.UniformInt(data),
single_ivec2=lambda data: ngl.UniformIVec2(data),
single_ivec3=lambda data: ngl.UniformIVec3(data),
single_ivec4=lambda data: ngl.UniformIVec4(data),
single_uint=lambda data: ngl.UniformUInt(data),
single_uvec2=lambda data: ngl.UniformUIVec2(data),
single_uvec3=lambda data: ngl.UniformUIVec3(data),
single_uvec4=lambda data: ngl.UniformUIVec4(data),
single_mat4=lambda data: ngl.UniformMat4(data),
single_quat_mat4=lambda data: ngl.UniformQuat(data, as_mat4=True),
single_quat_vec4=lambda data: ngl.UniformQuat(data, as_mat4=False),
single_vec2=lambda data: ngl.UniformVec2(data),
single_vec3=lambda data: ngl.UniformVec3(data),
def compute_histogram(cfg, show_dbg_points=False):
random.seed(0)
cfg.duration = 10
cfg.aspect_ratio = (1, 1)
hsize, size, local_size = _N * _N, _N, _N // 2
data = array.array('f')
for i in range(size * size):
data.extend((
random.uniform(0.0, 0.5),
random.uniform(0.25, 0.75),
random.uniform(0.5, 1.0),
1.0,
))
texture_buffer = ngl.BufferVec4(data=data)
texture = ngl.Texture2D(width=size, height=size, data_src=texture_buffer)
histogram_block = ngl.Block(layout='std430', label='histogram')
histogram_block.add_fields(
ngl.BufferUInt(hsize),
ngl.BufferUInt(hsize),
ngl.BufferUInt(hsize),
ngl.UniformUIVec3(),
)
shader_version = '310 es' if cfg.backend == 'gles' else '430'
shader_header = '#version %s\n' % shader_version
if cfg.backend == 'gles' and cfg.system == 'Android':
shader_header += '#extension GL_ANDROID_extension_pack_es31a: require\n'
shader_params = dict(hsize=hsize, size=size, local_size=local_size)
group_size = hsize // local_size
clear_histogram_shader = _COMPUTE_HISTOGRAM_CLEAR % shader_params
clear_histogram_program = ngl.ComputeProgram(shader_header + clear_histogram_shader)
clear_histogram = ngl.Compute(
group_size,
1,
1,
clear_histogram_program,
label='clear_histogram',
)
clear_histogram.update_blocks(histogram=histogram_block)
group_size = size // local_size
exec_histogram_shader = _COMPUTE_HISTOGRAM_EXEC % shader_params
exec_histogram_program = ngl.ComputeProgram(shader_header + exec_histogram_shader)
exec_histogram = ngl.Compute(
group_size,
group_size,
1,
exec_histogram_program,
label='compute_histogram'
)
exec_histogram.update_blocks(histogram=histogram_block)
exec_histogram.update_textures(source=texture)
quad = ngl.Quad((-1, -1, 0), (2, 0, 0), (0, 2, 0))
program = ngl.Program(
vertex=shader_header + _RENDER_HISTOGRAM_VERT,
fragment=shader_header + _RENDER_HISTOGRAM_FRAG % shader_params,
)
render = ngl.Render(quad, program, label='render_histogram')
render.update_blocks(histogram=histogram_block)
group = ngl.Group(children=(clear_histogram, exec_histogram, render,))
if show_dbg_points:
cuepoints = _get_compute_histogram_cuepoints()
group.add_children(get_debug_points(cfg, cuepoints))
return group
