q3dviewer 1.0.0__py3-none-any.whl

q3dviewer/custom_items/gaussian_item.py

@@ -0,0 +1,267 @@
+#!/usr/bin/env python3
+
+import numpy as np
+import pyqtgraph.opengl as gl
+from OpenGL.GL import *
+from OpenGL.GLU import *
+from OpenGL.GLUT import *
+import numpy as np
+import os
+from PyQt5.QtWidgets import QComboBox, QLabel
+from OpenGL.GL import shaders
+
+
+def div_round_up(x, y):
+    return int((x + y - 1) / y)
+
+
+def set_uniform_mat4(shader, content, name):
+    content = content.T
+    glUniformMatrix4fv(
+        glGetUniformLocation(shader, name),
+        1,
+        GL_FALSE,
+        content.astype(np.float32)
+    )
+
+
+def set_uniform_1int(shader, content, name):
+    glUniform1i(
+        glGetUniformLocation(shader, name),
+        content
+    )
+
+
+def set_uniform_v2(shader, contents, name):
+    glUniform2f(
+        glGetUniformLocation(shader, name),
+        *contents
+    )
+
+
+def set_uniform_v3(shader, contents, name):
+    glUniform3f(
+        glGetUniformLocation(shader, name),
+        *contents
+    )
+
+
+class GaussianItem(gl.GLGraphicsItem.GLGraphicsItem):
+    def __init__(self, **kwds):
+        super().__init__()
+        self.need_update_gs = False
+        self.sh_dim = 0
+        self.gs_data = np.empty([0])
+        self.prev_Rz = np.array([np.inf, np.inf, np.inf])
+        self.path = os.path.dirname(__file__)
+        try:
+            import torch
+            if not torch.cuda.is_available():
+                raise ImportError
+            self.cuda_pw = None
+            self.sort = self.torch_sort
+        except ImportError:
+                self.sort = self.opengl_sort
+
+    def addSetting(self, layout):
+        label1 = QLabel("set render mode:")
+        layout.addWidget(label1)
+        combo = QComboBox()
+        combo.addItem("render normal guassian")
+        combo.addItem("render ball")
+        combo.addItem("render inverse guassian")
+        combo.currentIndexChanged.connect(self.on_combobox_selection)
+        layout.addWidget(combo)
+
+    def on_combobox_selection(self, index):
+        glUseProgram(self.program)
+        set_uniform_1int(self.program, index, "render_mod")
+        glUseProgram(0)
+
+    def initializeGL(self):
+        fragment_shader = open(self.path + '/../shaders/gau_frag.glsl', 'r').read()
+        vertex_shader = open(self.path + '/../shaders/gau_vert.glsl', 'r').read()
+        sort_shader = open(self.path + '/../shaders/sort_by_key.glsl', 'r').read()
+        prep_shader = open(self.path + '/../shaders/gau_prep.glsl', 'r').read()
+
+        self.sort_program = shaders.compileProgram(
+            shaders.compileShader(sort_shader, GL_COMPUTE_SHADER))
+
+        self.prep_program = shaders.compileProgram(
+            shaders.compileShader(prep_shader, GL_COMPUTE_SHADER))
+
+        self.program = shaders.compileProgram(
+            shaders.compileShader(vertex_shader, GL_VERTEX_SHADER),
+            shaders.compileShader(fragment_shader, GL_FRAGMENT_SHADER),
+        )
+        positions = np.array([[0.0, 0.0, 0.0], [1.0, 0.0, 0.0], [0.0, 1.0, 0.0]], dtype=np.float32)
+
+        self.vao = glGenVertexArrays(1)
+
+        # trade a gaussian as a square (4 2d points)
+        square_vert = np.array([-1, 1, 1, 1, 1, -1, -1, -1], dtype=np.float32)
+        indices = np.array([0, 1, 2, 0, 2, 3], dtype=np.uint32)
+
+        # set the vertices for square
+        vbo = glGenBuffers(1)
+        glBindVertexArray(self.vao)
+        glBindBuffer(GL_ARRAY_BUFFER, vbo)
+        glBufferData(GL_ARRAY_BUFFER, square_vert.nbytes, square_vert, GL_STATIC_DRAW)
+        pos = glGetAttribLocation(self.program, 'vert')
+        glVertexAttribPointer(pos, 2, GL_FLOAT, False, 0, None)
+        glEnableVertexAttribArray(pos)
+        glBindBuffer(GL_ARRAY_BUFFER, 0)
+
+        # the vert's indices for drawing square
+        self.ebo = glGenBuffers(1)
+        glBindVertexArray(self.vao)
+        glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, self.ebo)
+        glBufferData(GL_ELEMENT_ARRAY_BUFFER,
+                     indices.nbytes, indices, GL_STATIC_DRAW)
+        glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0)
+        glBindVertexArray(0)
+
+        # add SSBO for gaussian data
+        self.ssbo_gs = glGenBuffers(1)
+        self.ssbo_gi = glGenBuffers(1)
+        self.ssbo_dp = glGenBuffers(1)
+        self.ssbo_pp = glGenBuffers(1)
+
+        width = self._GLGraphicsItem__view.deviceWidth()
+        height = self._GLGraphicsItem__view.deviceHeight()
+
+        # set constant parameter for gaussian shader
+        project_matrix = np.array(self._GLGraphicsItem__view.projectionMatrix().data(), np.float32).reshape([4, 4]).T
+        focal_x = project_matrix[0, 0] * width / 2
+        focal_y = project_matrix[1, 1] * height / 2
+        glUseProgram(self.prep_program)
+        set_uniform_mat4(self.prep_program, project_matrix, 'projection_matrix')
+        set_uniform_v2(self.prep_program, [focal_x, focal_y], 'focal')
+        glUseProgram(0)
+
+        glUseProgram(self.program)
+        set_uniform_v2(self.program, [width, height], 'win_size')
+        set_uniform_1int(self.program, 0, "render_mod")
+        glUseProgram(0)
+
+        # opengl settings
+        glDisable(GL_CULL_FACE)
+        glEnable(GL_BLEND)
+        glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
+
+    def update_gs(self):
+        if (self.need_update_gs):
+            # compute sorting size
+            self.num_sort = int(2**np.ceil(np.log2(self.gs_data.shape[0])))
+
+            # set input gaussian data
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, self.ssbo_gs)
+            glBufferData(GL_SHADER_STORAGE_BUFFER, self.gs_data.nbytes, self.gs_data.reshape(-1), GL_STATIC_DRAW)
+            glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, self.ssbo_gs)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0)
+
+            # set depth for sorting
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, self.ssbo_dp)
+            glBufferData(GL_SHADER_STORAGE_BUFFER, self.num_sort * 4, None, GL_STATIC_DRAW)
+            glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, self.ssbo_dp)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0)
+
+            # set index for sorting (the index need be initialized)
+            gi = np.arange(self.num_sort, dtype=np.uint32)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, self.ssbo_gi)
+            glBufferData(GL_SHADER_STORAGE_BUFFER, self.num_sort * 4, gi, GL_STATIC_DRAW)
+            glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, self.ssbo_gi)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0)
+
+            # set preprocess buffer
+            # the dim of preprocess data is 12 u(3), covinv(3), color(3), area(2), alpha(1)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, self.ssbo_pp)
+            glBufferData(GL_SHADER_STORAGE_BUFFER, self.gs_data.shape[0] * 4 * 12,
+                         None, GL_STATIC_DRAW)
+            glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 3, self.ssbo_pp)
+            glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0)
+
+            glUseProgram(self.prep_program)
+            set_uniform_1int(self.prep_program, self.sh_dim, "sh_dim")
+            set_uniform_1int(self.prep_program, self.gs_data.shape[0], "gs_num")
+            glUseProgram(0)
+            self.need_update_gs = False
+
+    def paint(self):
+        # get current view matrix
+        self.view_matrix = np.array(self._GLGraphicsItem__view.viewMatrix().data(), np.float32).reshape([4, 4]).T
+
+        # if gaussian data is update, renew vao, ssbo, etc...
+        self.update_gs()
+
+        if (self.gs_data.shape[0] == 0):
+            return
+
+        # preprocess and sort gaussian by compute shader.
+        self.preprocess_gs()
+        self.try_sort()
+
+        # draw by vert shader
+        glUseProgram(self.program)
+        # bind vao and ebo
+        glBindVertexArray(self.vao)
+        glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, self.ebo)
+        # draw instances
+        glDrawElementsInstanced(GL_TRIANGLES, 6, GL_UNSIGNED_INT, None, self.gs_data.shape[0])
+        # upbind vao and ebo
+        glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0)
+        glBindVertexArray(0)
+        glUseProgram(0)
+
+    def try_sort(self):
+        # don't sort if the depths are not change.
+        Rz = self.view_matrix[2, :3]
+        if (np.linalg.norm(self.prev_Rz - Rz) > 0.1):
+            # import torch
+            # torch.cuda.synchronize()
+            # start = time.time()
+            self.sort()
+            self.prev_Rz = Rz
+            # torch.cuda.synchronize()
+            # end = time.time()
+            # time_diff = end - start
+            # print(time_diff)
+
+    def opengl_sort(self):
+        glUseProgram(self.sort_program)
+        # can we move this loop to gpu?
+        for level in 2**np.arange(1, int(np.ceil(np.log2(self.num_sort))+1)):  # level = level*2
+            for stage in level/2**np.arange(1, np.log2(level)+1):   # stage =stage / 2
+                set_uniform_1int(self.sort_program, int(level), "level")
+                set_uniform_1int(self.sort_program, int(stage), "stage")
+                glDispatchCompute(div_round_up(self.num_sort//2, 256), 1, 1)
+                glMemoryBarrier(GL_SHADER_STORAGE_BARRIER_BIT)
+        # glFinish()
+        glUseProgram(0)
+
+    def torch_sort(self):
+        import torch
+        if self.cuda_pw is None:
+            self.cuda_pw = torch.tensor(self.gs_data[:, :3]).cuda()
+        Rz = torch.tensor(self.view_matrix[2, :3]).cuda()
+        depth = Rz @ self.cuda_pw.T
+        index = torch.argsort(depth).type(torch.int32).cpu().numpy()
+        glBindBuffer(GL_SHADER_STORAGE_BUFFER, self.ssbo_gi)
+        glBufferData(GL_SHADER_STORAGE_BUFFER, index.nbytes, index, GL_STATIC_DRAW)
+        glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, self.ssbo_gi)
+        glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0)
+        return index
+
+    def preprocess_gs(self):
+        glUseProgram(self.prep_program)
+        set_uniform_mat4(self.prep_program, self.view_matrix, 'view_matrix')
+        glDispatchCompute(div_round_up(self.gs_data.shape[0], 256), 1, 1)
+        glMemoryBarrier(GL_SHADER_STORAGE_BARRIER_BIT)
+        glUseProgram(0)
+
+    def setData(self, **kwds):
+        if 'gs_data' in kwds:
+            gs_data = kwds.pop('gs_data')
+            self.gs_data = np.ascontiguousarray(gs_data, dtype=np.float32)
+            self.sh_dim = self.gs_data.shape[-1] - (3 + 4 + 3 + 1)
+        self.need_update_gs = True

q3dviewer/custom_items/grid_item.py

@@ -0,0 +1,37 @@
+import pyqtgraph.opengl as gl
+from OpenGL.GL import *
+from PyQt5.QtWidgets import QLabel, QDoubleSpinBox
+
+
+class GridItem(gl.GLGridItem):
+    def __init__(self, size, spacing, color=(255, 255, 255, 76.5)):
+        super(GridItem, self).__init__(color=color)
+        self.setSize0(size)
+        self.setSpacing0(spacing)
+
+    def addSetting(self, layout):
+        label1 = QLabel("Set size:")
+        layout.addWidget(label1)
+        box1 = QDoubleSpinBox()
+        box1.setSingleStep(1.0)
+        layout.addWidget(box1)
+        box1.setValue(self.size0)
+        box1.valueChanged.connect(self.setSize0)
+        box1.setRange(0, 100000)
+
+        label2 = QLabel("Set spacing:")
+        layout.addWidget(label2)
+        box2 = QDoubleSpinBox()
+        layout.addWidget(box2)
+        box2.setSingleStep(0.1)
+        box2.setValue(self.spacing0)
+        box2.valueChanged.connect(self.setSpacing0)
+        box2.setRange(0, 1000)
+
+    def setSize0(self, size):
+        self.size0 = size
+        self.setSize(self.size0, self.size0)
+
+    def setSpacing0(self, spacing):
+        self.spacing0 = spacing
+        self.setSpacing(self.spacing0, self.spacing0)

q3dviewer/custom_items/image_item.py

@@ -0,0 +1,169 @@
+import pyqtgraph.opengl as gl
+from OpenGL.GL import *
+import numpy as np
+from OpenGL.GL import shaders
+from PIL import Image as PIL_Image
+from PyQt5.QtWidgets import QLabel, QSpinBox
+
+
+# Vertex and Fragment shader source code
+vertex_shader_source = """
+#version 330 core
+layout(location = 0) in vec3 position;
+layout(location = 1) in vec2 texCoord;
+
+out vec2 TexCoord;
+
+uniform mat4 view_matrix;
+uniform mat4 project_matrix;
+
+void main()
+{
+    gl_Position = vec4(position, 1.0);
+    TexCoord = texCoord;
+}
+"""
+
+fragment_shader_source = """
+#version 330 core
+in vec2 TexCoord;
+out vec4 color;
+uniform sampler2D ourTexture;
+void main()
+{
+    color = texture(ourTexture, TexCoord);
+}
+"""
+
+
+def set_uniform_mat4(shader, content, name):
+    content = content.T
+    glUniformMatrix4fv(
+        glGetUniformLocation(shader, name),
+        1,
+        GL_FALSE,
+        content.astype(np.float32)
+    )
+
+
+class ImageItem(gl.GLGraphicsItem.GLGraphicsItem):
+    def __init__(self, pos=np.array([0, 0]), size=np.array([1280/2, 720/2])):
+        gl.GLGraphicsItem.GLGraphicsItem.__init__(self)
+        self.pos = pos  # bottom-left
+        self.size = size
+        self.image = np.zeros((self.size[0], self.size[1], 4), dtype=np.uint8)
+        self.alpha = 255
+
+    def initializeGL(self):
+        # Rectangle vertices and texture coordinates
+        width = self._GLGraphicsItem__view.deviceWidth()
+        height = self._GLGraphicsItem__view.deviceHeight()
+        x0, y0 = self.pos
+        x1, y1 = self.pos + self.size
+        x0 = x0 / width * 2 - 1
+        y0 = y0 / height * 2 - 1
+        x1 = x1 / width * 2 - 1
+        y1 = y1 / height * 2 - 1
+
+        self.vertices = np.array([
+            # positions          # texture coords
+            [x0, y0,  0.0,  0.0, 0.0],  # bottom-left
+            [x1, y0,  0.0,  1.0, 0.0],  # bottom-right
+            [x1,  y1,  0.0,  1.0, 1.0],  # top-right
+            [x0,  y1,  0.0,  0.0, 1.0],   # top-left
+        ], dtype=np.float32)
+
+        indices = np.array([
+            0, 1, 2,  # first triangle
+            2, 3, 0   # second triangle
+        ], dtype=np.uint32)
+
+        self.vao = glGenVertexArrays(1)
+        vbo = glGenBuffers(1)
+        ebo = glGenBuffers(1)
+
+        glBindVertexArray(self.vao)
+
+        glBindBuffer(GL_ARRAY_BUFFER, vbo)
+        glBufferData(GL_ARRAY_BUFFER, self.vertices.itemsize *
+                     5 * 4, self.vertices, GL_STATIC_DRAW)
+
+        glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, ebo)
+        glBufferData(GL_ELEMENT_ARRAY_BUFFER,
+                     indices.nbytes, indices, GL_STATIC_DRAW)
+
+        # Vertex positions
+        glVertexAttribPointer(0, 3, GL_FLOAT, GL_FALSE,
+                              20, ctypes.c_void_p(0))
+        glEnableVertexAttribArray(0)
+
+        # Texture coordinates
+        glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE,
+                              20, ctypes.c_void_p(12))
+        glEnableVertexAttribArray(1)
+
+        # Compile shaders and create shader program
+        self.program = shaders.compileProgram(
+            shaders.compileShader(vertex_shader_source, GL_VERTEX_SHADER),
+            shaders.compileShader(fragment_shader_source, GL_FRAGMENT_SHADER),
+        )
+
+        self.texture = glGenTextures(1)
+        glBindTexture(GL_TEXTURE_2D, self.texture)
+        glBindVertexArray(0)
+
+    def setData(self, data):
+        if isinstance(data, np.ndarray):
+            pass
+        elif isinstance(data, PIL_Image.Image):
+            data = np.array(data)
+        else:
+            print("not support image type")
+            raise NotImplementedError
+        
+        if data.ndim == 2:  # Grayscale image
+            data = np.stack((data,) * 3 + (np.ones_like(data) * 255,), axis=-1)
+        elif data.shape[-1] == 3:  # RGB image
+            alpha_channel = np.ones((data.shape[0], data.shape[1], 1), dtype=data.dtype) * self.alpha
+            data = np.concatenate((data, alpha_channel), axis=-1)
+        self.image = data
+
+    def paint(self):
+        if self.image is not None:
+            img_data = self.image
+            img_data = np.flipud(img_data)  # Flip the image vertically
+            img_data = img_data.tobytes()
+            glBindTexture(GL_TEXTURE_2D, self.texture)
+            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, self.image.shape[1],
+                         self.image.shape[0], 0, GL_RGBA, GL_UNSIGNED_BYTE, img_data)
+            glGenerateMipmap(GL_TEXTURE_2D)
+            glBindTexture(GL_TEXTURE_2D, 0)
+            self.image = None
+
+        glEnable(GL_DEPTH_TEST)
+        glEnable(GL_BLEND)
+        glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
+
+        glUseProgram(self.program)
+        glBindVertexArray(self.vao)
+        glBindTexture(GL_TEXTURE_2D, self.texture)
+        glDrawElements(GL_TRIANGLES, 6, GL_UNSIGNED_INT, None)
+        glBindTexture(GL_TEXTURE_2D, 0)
+        glBindVertexArray(0)
+        glUseProgram(0)
+
+        glDisable(GL_DEPTH_TEST)
+        glDisable(GL_BLEND)
+
+    def addSetting(self, layout):
+        label1 = QLabel("Set Alpha:")
+        layout.addWidget(label1)
+        box1 = QSpinBox()
+        box1.setSingleStep(1)
+        box1.setRange(0, 255)
+        box1.setValue(self.alpha)
+        box1.valueChanged.connect(self.setAlpha)
+        layout.addWidget(box1)
+
+    def setAlpha(self, alpha):
+        self.alpha = alpha

q3dviewer/custom_items/text_item.py

@@ -0,0 +1,53 @@
+from pyqtgraph.Qt import QtCore
+import pyqtgraph.opengl as gl
+from OpenGL.GL import *
+from PyQt5 import QtGui, QtCore
+
+
+class GL2DTextItem(gl.GLGraphicsItem.GLGraphicsItem):
+    """Draws text over opengl 3D."""
+
+    def __init__(self, **kwds):
+        """All keyword arguments are passed to setData()"""
+        gl.GLGraphicsItem.GLGraphicsItem.__init__(self)
+        glopts = kwds.pop('glOptions', 'additive')
+        self.setGLOptions(glopts)
+        self.pos = (100, 100)
+        self.color = QtCore.Qt.GlobalColor.white
+        self.text = ''
+        self.font = QtGui.QFont('Helvetica', 16)
+        self.setData(**kwds)
+
+    def setData(self, **kwds):
+        args = ['pos', 'color', 'text', 'size', 'font']
+        for k in kwds.keys():
+            if k not in args:
+                raise ValueError('Invalid keyword argument: %s (allowed arguments are %s)' % (k, str(args)))
+        for arg in args:
+            if arg in kwds:
+                value = kwds[arg]
+                if arg == 'pos':
+                    self.pos = value
+                elif arg == 'color':
+                    value = value
+                elif arg == 'font':
+                    if isinstance(value, QtGui.QFont) is False:
+                        raise TypeError('"font" must be QFont.')
+                elif arg == 'size':
+                    self.font.setPointSize(value)
+                setattr(self, arg, value)
+        self.update()
+
+    def paint(self):
+        if len(self.text) < 1:
+            return
+        self.setupGLState()
+
+        text_pos = QtCore.QPointF(*self.pos)
+        painter = QtGui.QPainter(self.view())
+        painter.setPen(self.color)
+        painter.setFont(self.font)
+        painter.setRenderHints(QtGui.QPainter.RenderHint.Antialiasing | QtGui.QPainter.RenderHint.TextAntialiasing)
+        painter.drawText(text_pos, self.text)
+        painter.end()
+

q3dviewer/custom_items/trajectory_item.py

@@ -0,0 +1,78 @@
+import pyqtgraph.opengl as gl
+from OpenGL.GL import *
+import numpy as np
+import threading
+
+
+class TrajectoryItem(gl.GLGridItem):
+    def __init__(self, width=1, color=(0, 1, 0, 1)):
+        super(TrajectoryItem, self).__init__()
+        self.width = width
+        self.buff = np.empty((0, 3), np.float32)
+        self.wait_add_data = None
+        self.mutex = threading.Lock()
+        self.CAPACITY = 100000
+        self.valid_buff_top = 0
+        self.color = color
+
+    def addSetting(self, layout):
+        pass
+
+    def setData(self, data, append=True):
+        self.mutex.acquire()
+        data = data.astype(np.float32).reshape(-1, 3)
+        if (append is False):
+            self.wait_add_data = data
+            self.add_buff_loc = 0
+        else:
+            if (self.wait_add_data is None):
+                self.wait_add_data = data
+            else:
+                self.wait_add_data = np.concatenate([self.wait_add_data, data])
+            self.add_buff_loc = self.valid_buff_top
+        self.mutex.release()
+
+    def updateRenderBuffer(self):
+        if (self.wait_add_data is None):
+            return
+        self.mutex.acquire()
+
+        new_buff_top = self.add_buff_loc + self.wait_add_data.shape[0]
+        if new_buff_top > self.buff.shape[0]:
+            buff_capacity = self.buff.shape[0]
+            while (new_buff_top > buff_capacity):
+                buff_capacity += self.CAPACITY
+            self.buff = np.empty((buff_capacity, 3), np.float32)
+            self.buff[self.add_buff_loc:new_buff_top] = self.wait_add_data
+            glBindBuffer(GL_ARRAY_BUFFER, self.vbo)
+            glBufferData(GL_ARRAY_BUFFER, self.buff.nbytes,
+                         self.buff, GL_DYNAMIC_DRAW)
+            glBindBuffer(GL_ARRAY_BUFFER, 0)
+        else:
+            self.buff[self.add_buff_loc:new_buff_top] = self.wait_add_data
+            glBindBuffer(GL_ARRAY_BUFFER, self.vbo)
+            glBufferSubData(GL_ARRAY_BUFFER, self.add_buff_loc * 12,
+                            self.wait_add_data.shape[0] * 12, self.wait_add_data)
+        self.valid_buff_top = new_buff_top
+        self.wait_add_data = None
+        self.mutex.release()
+
+    def initializeGL(self):
+        self.vbo = glGenBuffers(1)
+
+    def paint(self):
+        self.setupGLState()
+        self.updateRenderBuffer()
+        glEnable(GL_BLEND)
+        glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
+        glBindBuffer(GL_ARRAY_BUFFER, self.vbo)
+        glEnableClientState(GL_VERTEX_ARRAY)
+        glVertexPointer(3, GL_FLOAT, 0, None)
+        glLineWidth(self.width)
+        glColor4f(*self.color)  # z is blue
+
+        glDrawArrays(GL_LINE_STRIP, 0, self.valid_buff_top)
+        glDisableClientState(GL_VERTEX_ARRAY)
+
+        glBindBuffer(GL_ARRAY_BUFFER, 0)
+        glUseProgram(0)