Function test_eglctx_auto_fbo_rbo

tests/headless_opengl_tests.py:244–288

()

def test_eglctx_auto_fbo_rbo():
    render_w = 1024
    render_h = 1024
    gl.glClear(gl.GL_COLOR_BUFFER_BIT | gl.GL_DEPTH_BUFFER_BIT)

    mesh_path = 'assets/meshes/bunny.ply'
    img_path = 'test_eglctx_auto_fbo_rbo.png'
    dpt_path = 'test_eglctx_auto_fbo_rbo_dpt.png'
    camera = Camera(H=render_h, W=render_w,
                    K=torch.tensor([[render_w * 592 / 512, 0., render_w / 2],
                                    [0., render_h * 592 / 512, render_h / 2],
                                    [0., 0., 1.]]),
                    R=torch.tensor([[0.9908, -0.1353, 0.0000],
                                    [-0.1341, -0.9815, -0.1365],
                                    [0.0185, 0.1353, -0.9906]]),
                    T=torch.tensor([[0.0178],
                                    [0.0953],
                                    [0.3137]]),
                    n=0.02,
                    f=1,
                    )
    mesh = Mesh(filename=mesh_path, shade_flat=False)
    mesh.offscreen_render(eglctx, camera)

    # Read result
    gl.glReadBuffer(gl.GL_COLOR_ATTACHMENT0)
    img_buf = gl.glReadPixels(0, 0, render_w, render_h, gl.GL_RGBA, gl.GL_UNSIGNED_BYTE)
    img = np.frombuffer(img_buf, np.uint8).reshape(render_h, render_w, 4)[::-1]
    save_image(img_path, img)
    log(f'Rendered image saved at: {blue(img_path)}')

    # Read result
    # gl.glReadBuffer(gl.GL_COLOR_ATTACHMENT1)
    # dpt_buf = gl.glReadPixels(0, 0, render_w, render_h, gl.GL_RGBA, gl.GL_UNSIGNED_BYTE)  # lost precision...
    # dpt = np.frombuffer(dpt_buf, np.uint8).reshape(render_h, render_w, 4)[::-1]
    # dpt = dpt[..., 0] / 1 + dpt[..., 0] / 256 + dpt[..., 0] / 65536 + dpt[..., 0] / 16777216
    dpt_buf = gl.glReadPixels(0, 0, render_w, render_h, gl.GL_DEPTH_COMPONENT, gl.GL_FLOAT)  # lost precision... # MARK: SYNC
    # gl.glBindTexture(gl.GL_TEXTURE_2D, eglctx.tex_dpt)
    # dpt_buf = gl.glGetTexImage(gl.GL_TEXTURE_2D, 0, gl.GL_DEPTH_COMPONENT, gl.GL_FLOAT)
    dpt = np.frombuffer(dpt_buf, np.float32).reshape(render_h, render_w, 1)[::-1]
    dpt = linearize_depth(dpt, camera.n, camera.f)  # this is the z component in camera coordinates
    dpt = (dpt.min() - dpt) / (dpt.min() - dpt.max())

    save_image(dpt_path, dpt)
    log(f'Rendered depth saved at: {blue(dpt_path)}')


if __name__ == '__main__':