1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import moderngl
import numpy
from headless import HeadlessTestCase
from moderngl_window.opengl.vao import VAO, BufferInfo, VAOError
class VaoTestCase(HeadlessTestCase):
def createProgram(self):
return self.ctx.program(
vertex_shader="""
#version 330
in vec3 position;
in vec3 normal;
out vec3 v_normal;
void main() {
gl_Position = vec4(position, 1.0);
v_normal = normalize(normal);
}
""",
fragment_shader="""
#version 330
out vec4 fragColor;
in vec3 v_normal;
void main() {
float col = length(v_normal);
fragColor = vec4(col);
}
""",
)
def test_create(self):
mesh = VAO("test", mode=moderngl.LINES)
mesh.buffer(numpy.array([0.0, 0.0, 0.0, 1.0, 1.0, 1.0], dtype='f4'), '3f', 'position')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0, 0.0, 1.0], dtype='f4'), '3f', 'normal')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0], dtype='f4'), '2f', 'uv')
# Ensure basic properties are correct
self.assertEqual(mesh.name, "test")
self.assertEqual(mesh.vertex_count, 2)
self.assertEqual(mesh.mode, moderngl.LINES)
# Ensure buffers are present
self.assertIsInstance(mesh.get_buffer_by_name('position'), BufferInfo)
self.assertIsInstance(mesh.get_buffer_by_name('normal'), BufferInfo)
self.assertIsInstance(mesh.get_buffer_by_name('uv'), BufferInfo)
self.assertEqual(mesh.get_buffer_by_name('something'), None)
# Create a progam using a subset of the buffers
prog = self.createProgram()
vao = mesh.instance(prog)
self.assertIsInstance(vao, moderngl.VertexArray)
# Render directly with VAO and VertexArray instance
mesh.render(prog)
vao.render()
def test_illegal_draw_mode(self):
"""Create vao with illegal draw mode"""
with self.assertRaises(VAOError):
VAO(mode=1337)
def test_add_illegal_buffer(self):
"""Attempt to add illegal buffer"""
vao = VAO()
with self.assertRaises(VAOError):
vao.buffer("stuff", '1f', 'in_position')
def test_normalized_types(self):
"""Ensure VAO wrapper can handle normalized types"""
prog = self.createProgram()
mesh = VAO("test", mode=moderngl.LINES)
mesh.buffer(numpy.array([0.0, 0.0, 0.0, 1.0, 1.0, 1.0], dtype='i4'), '3ni', 'position')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0, 0.0, 1.0], dtype='f4'), '3nf', 'normal')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0], dtype='f4'), '2f', 'uv')
# Uncomment in moderngl 5.6+
# mesh.instance(prog)
def test_divisors(self):
"""Test defining buffers with different divisor types"""
mesh = VAO("test", mode=moderngl.LINES)
mesh.buffer(numpy.array([0.0, 0.0, 0.0, 1.0, 1.0, 1.0], dtype='f4'), '3f/v', 'position')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0, 0.0, 1.0], dtype='f4'), '3f/r', 'normal')
mesh.buffer(numpy.array([0.0, 0.0, 1.0, 1.0], dtype='f4'), '2f/i', 'uv')
buffer1 = mesh.get_buffer_by_name('position')
buffer2 = mesh.get_buffer_by_name('normal')
buffer3 = mesh.get_buffer_by_name('uv')
attributes = ['position', 'normal', 'uv']
self.assertEqual(buffer1.content(attributes), (buffer1.buffer, '3f/v', 'position'))
self.assertEqual(buffer2.content(attributes), (buffer2.buffer, '3f/r', 'normal'))
self.assertEqual(buffer3.content(attributes), (buffer3.buffer, '2f/i', 'uv'))
|
