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,49 @@
+import ctypes
+import sys
+import types
+
+import OpenGL.platform as _platform
+from OpenGL.platform.baseplatform import BasePlatform as _BasePlatform
+from OpenGL.platform.osmesa import OSMesaPlatform as _OSMesaPlatform
+
+
+# Fix broken loading of OpenGL extension functions when using OSMesa on Windows OS
+if sys.platform == "win32" and isinstance(_platform.PLATFORM, _OSMesaPlatform):
+    if _platform.PLATFORM.constructFunction.__func__ == _BasePlatform.constructFunction:
+        def constructFunction(
+            self,
+            functionName,
+            dll,
+            resultType = ctypes.c_int,
+            argTypes = (),
+            doc = None,
+            argNames = (),
+            extension = None,
+            deprecated = False,
+            module = None,
+            force_extension = False,
+            error_checker = None,
+        ):
+            """Override construct function to do win32-specific hacks to find entry points"""
+            for force_extension_ in (force_extension, True):
+                try:
+                    return _BasePlatform.constructFunction(
+                        self,
+                        functionName,
+                        dll,
+                        resultType,
+                        argTypes,
+                        doc,
+                        argNames,
+                        extension,
+                        deprecated,
+                        module,
+                        force_extension=force_extension_,
+                        error_checker=error_checker,
+                    )
+                    break
+                except AttributeError:
+                    if force_extension:
+                        raise
+
+        _platform.PLATFORM.constructFunction = types.MethodType(constructFunction, _platform.PLATFORM)

genesis/ext/pyrender/font.py

@@ -2,10 +2,10 @@
 
 Author: Matthew Matl
 """
+import os
 
 import freetype
 import numpy as np
-import os
 
 import OpenGL
 from OpenGL.GL import *
@@ -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,7 +3,6 @@
 
 Author: Matthew Matl
 """
-
 import numpy as np
 from OpenGL.GL import *
 
@@ -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,9 +2,7 @@
 
 Author: Matthew Matl
 """
-
 import sys
-from time import time
 
 import PIL
 import pyglet
@@ -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

@@ -9,7 +9,7 @@
 from OpenGL.GL import *
 from OpenGL.platform import ctypesloader
 from OpenGL.GL import shaders as gl_shader_utils
-from time import time
+
 
 func = None
 
@@ -91,7 +91,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