Merge pull request #135 from yuyu2172/get-texture

Add get method for Texture
glumpy · Feb 19, 2018 · f9d58de · f9d58de
2 parents 9896114 + 8024b8e
commit f9d58de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/examples/gloo-framebuffer.py b/examples/gloo-framebuffer.py
@@ -45,6 +45,8 @@ def on_draw(dt):
     window.clear()
     framebuffer.activate()
     quad_1.draw(gl.GL_TRIANGLE_STRIP)
+    # The texture can be read as np.ndarray.
+    # out_texture = framebuffer.color[0].get()
     framebuffer.deactivate()
     quad_2.draw(gl.GL_TRIANGLE_STRIP)
 

diff --git a/glumpy/gloo/texture.py b/glumpy/gloo/texture.py
@@ -211,7 +211,12 @@ def _delete(self):
         if self.handle > -1:
             gl.glDeleteTextures(np.array([self.handle], dtype=np.uint32))
 
-
+    def get(self):
+        """ Read the texture data back into CPU memory """
+        host = np.zeros(self.shape, self.dtype)
+        gl.glBindTexture(self.target, self._handle)
+        gl.glGetTexImage(self.target, 0, self.cpu_format, self.gtype, host)
+        return host
 
 
 class Texture1D(Texture):