Add robust fallback for OpenGL on Windows based on OSMesa.

Author: duburcqa

genesis/ext/pyrender/__init__.py

@@ -1,3 +1,4 @@
+from . import _pyopengl
 from .camera import Camera, PerspectiveCamera, OrthographicCamera, IntrinsicsCamera
 from .light import Light, PointLight, DirectionalLight, SpotLight
 from .sampler import Sampler

genesis/ext/pyrender/_pyopengl.py

@@ -0,0 +1,29 @@
+from OpenGL.GL import *
+from OpenGL.GL import __dict__ as _gl_dict
+import OpenGL.platform as _platform
+import OpenGL.arrays as _arrays
+from OpenGL.raw.GL import _errors
+from OpenGL.wrapper import wrapper as _wrapper
+
+
+# Fix broken 'glGenVertexArrays' OpenGL function when using OSMesa on Windows OS
+if not glGenVertexArrays:
+    func = _platform.nullFunction(
+        "glGenVertexArrays",
+        _platform.PLATFORM.GL,
+        resultType=None,
+        argTypes=(GLsizei, _arrays.arraydatatype.GLuintArray),
+        doc=None,
+        argNames=('n', 'arrays'),
+        extension='GL_VERSION_GL_3_0',
+        error_checker=_errors._error_checker,
+        force_extension=True,
+    )
+    glGenVertexArrays = _wrapper(func).setOutput(
+        'arrays', size=lambda x:(x,), pnameArg='n', orPassIn=True
+    )
+
+
+__all__ = [
+    name for name in _gl_dict if not name.startswith('_')
+]

genesis/ext/pyrender/font.py

@@ -2,14 +2,14 @@
 
 Author: Matthew Matl
 """
+import os
 
 import freetype
 import numpy as np
-import os
 
 import OpenGL
-from OpenGL.GL import *
 
+from ._pyopengl import *
 from .constants import TextAlign, FLOAT_SZ
 from .texture import Texture
 from .sampler import Sampler
@@ -122,7 +122,6 @@ def font_pt(self, value):
         self._font_pt = int(value)
 
     def _add_to_context(self):
-
         self._vao = glGenVertexArrays(1)
         glBindVertexArray(self._vao)
         self._vbo = glGenBuffers(1)

genesis/ext/pyrender/numba_gl_wrapper.py

@@ -1,3 +1,5 @@
+from contextlib import ExitStack
+
 import numpy as np
 from numba import *
 from numba import types
@@ -11,7 +13,7 @@
     NativeValue,
 )
 from numba.core import cgutils
-from contextlib import ExitStack
+
 import OpenGL.GL as GL
 from OpenGL._bytes import as_8_bit
 from OpenGL.GL import GLint, GLuint, GLvoidp, GLvoid, GLfloat, GLsizei, GLboolean, GLenum, GLsizeiptr, GLintptr

genesis/ext/pyrender/primitive.py

@@ -3,10 +3,9 @@
 
 Author: Matthew Matl
 """
-
 import numpy as np
-from OpenGL.GL import *
 
+from ._pyopengl import *
 from .constants import FLOAT_SZ, GLTF, UINT_SZ, BufFlags
 from .material import Material, MetallicRoughnessMaterial
 from .utils import format_color_array
@@ -330,8 +329,8 @@ def calc_vertex_normal(self):
 
     def _add_to_context(self):
         if self._vaid is not None:
-            return
             # raise ValueError('Mesh is already bound to a context')
+            return
 
         # Generate and bind VAO
         self._vaid = glGenVertexArrays(1)

genesis/ext/pyrender/renderer.py

@@ -2,16 +2,14 @@
 
 Author: Matthew Matl
 """
-
 import sys
-from time import time
 
 import PIL
 import pyglet
 import numpy as np
-from OpenGL.GL import *
 import matplotlib.pyplot as plt
 
+from ._pyopengl import *
 from .constants import (
     DEFAULT_Z_FAR,
     DEFAULT_Z_NEAR,
@@ -25,7 +23,6 @@
     TextAlign,
 )
 from .font import FontCache
-from .jit_render import JITRenderer
 from .light import DirectionalLight, PointLight, SpotLight
 from .material import MetallicRoughnessMaterial, SpecularGlossinessMaterial
 from .shader_program import ShaderProgramCache

genesis/ext/pyrender/shader_program.py

@@ -6,10 +6,11 @@
 import numpy as np
 
 import OpenGL
-from OpenGL.GL import *
 from OpenGL.platform import ctypesloader
 from OpenGL.GL import shaders as gl_shader_utils
-from time import time
+
+from ._pyopengl import *
+
 
 func = None
 
@@ -91,7 +92,6 @@ class ShaderProgram(object):
     """
 
     def __init__(self, vertex_shader, fragment_shader, geometry_shader=None, defines=None):
-
         self.vertex_shader = vertex_shader
         self.fragment_shader = fragment_shader
         self.geometry_shader = geometry_shader