MultiBufferVAO.py demo added

jmacey · jmacey · commit bc49a7c2d243 · 2019-12-04T08:23:24.000Z
diff --git a/MultiBufferVAOFactory/MultiBufferVAOFactory.py b/MultiBufferVAOFactory/MultiBufferVAOFactory.py
@@ -0,0 +1,215 @@
+#!/usr/bin/env python
+from __future__ import print_function
+from PyQt5.QtGui import QOpenGLWindow,QSurfaceFormat
+from PyQt5.QtWidgets import QApplication
+from  PyQt5.QtCore import *
+import sys
+from pyngl import *
+from OpenGL.GL import *
+
+class MainWindow(QOpenGLWindow) :
+  
+  def __init__(self, parent=None):
+    super(QOpenGLWindow, self).__init__(parent)
+    self.mouseGlobalTX=Mat4()
+    self.width=int(1024)
+    self.height=int(720)
+    self.setTitle('Boid')
+    self.spinXFace = int(0)
+    self.spinYFace = int(0)
+    self.rotate = False
+    self.translate = False
+    self.origX = int(0)
+    self.origY = int(0)
+    self.origXPos = int(0)
+    self.origYPos = int(0)
+    self.INCREMENT=0.01
+    self.ZOOM=0.1
+    self.modelPos=Vec3()
+    self.view=Mat4()
+    self.project=Mat4()
+    self.vao=None
+
+  def initializeGL(self) :
+    self.makeCurrent()
+    NGLInit.instance()
+    glClearColor( 0.4, 0.4, 0.4, 1.0 ) 
+    glEnable( GL_DEPTH_TEST )
+    glEnable( GL_MULTISAMPLE )
+    shader=ShaderLib.instance()
+    shader.loadShader('Phong','shaders/PhongVertex.glsl','shaders/PhongFragment.glsl',ErrorExit.OFF)
+    shader.use('Phong')
+    lightPos=Vec4(-2.0,5.0,2.0,0.0)
+    shader.setUniform('light.position',lightPos)
+    shader.setUniform('light.ambient',0.0,0.0,0.0,1.0)
+    shader.setUniform('light.diffuse',1.0,1.0,1.0,1.0);
+    shader.setUniform('light.specular',0.8,0.8,0.8,1.0);
+    shader.setUniform('material.ambient',0.274725,0.1995,0.0745,0.0);
+    shader.setUniform('material.diffuse',0.75164,0.60648,0.22648,0.0);
+    shader.setUniform('material.specular',0.628281,0.555802,0.3666065,0.0);
+    shader.setUniform('material.shininess',51.2);
+    shader.setUniform('viewerPos',0,1,5);
+
+
+    self.view=lookAt(Vec3(0,1,5),Vec3.zero(),Vec3.up())
+    self.project=perspective( 45.0, float(self.width)/float(self.height), 0.05, 350.0 )
+    self.buildVAO()
+
+
+  def buildVAO(self) :
+    verts=VectorVec3([
+    Vec3(0.0,1.0,1.0),
+    Vec3(0.0,0.0,-1.0),
+    Vec3(-0.5,0.0,1.0),
+    Vec3(0.0,1.0,1.0),
+    Vec3(0.0,0.0,-1.0),
+    Vec3(0.5,0.0,1.0),
+    Vec3(0.0,1.0,1.0),
+    Vec3(0.0,0.0,1.5),
+    Vec3(-0.5,0.0,1.0),
+    Vec3(0.0,1.0,1.0),
+    Vec3(0.0,0.0,1.5),
+    Vec3(0.5,0.0,1.0)])
+    
+    normals=VectorVec3()
+    n=calcNormal(verts[2],verts[1],verts[0]);
+    normals.extend([n,n,n])
+    n=calcNormal(verts[3],verts[4],verts[5]);
+    normals.extend([n,n,n])
+    n=calcNormal(verts[6],verts[7],verts[8]);
+    normals.extend([n,n,n])
+    n=calcNormal(verts[11],verts[10],verts[9]);
+    normals.extend([n,n,n])
+
+    self.vao=VAOFactory.createVAO(multiBufferVAO,GL_TRIANGLES)
+    self.vao.bind()
+    self.vao.setData(len(verts)*Vec3.sizeof(),verts)
+    self.vao.setVertexAttributePointer(0,3,GL_FLOAT,0,0)
+    self.vao.setData(len(normals)*Vec3.sizeof(),normals)
+    self.vao.setVertexAttributePointer(1,3,GL_FLOAT,0,0)
+    self.vao.setNumIndices(len(verts)) 
+    self.vao.unbind() 
+
+
+  def loadMatricesToShader(self) :
+    shader = ShaderLib.instance()  
+    shader.use('Phong')
+  
+    MV=  self.view*self.mouseGlobalTX
+    MVP= self.project*MV;
+    normalMatrix=Mat3(MV)
+    normalMatrix.inverse().transpose()
+    shader.setUniform("MV",MV)
+    shader.setUniform("MVP",MVP)
+    shader.setUniform("normalMatrix",normalMatrix)
+    shader.setUniform("M",self.mouseGlobalTX)
+  
+  def paintGL(self):
+    try :
+      self.makeCurrent()
+      glViewport( 0, 0, self.width, self.height )
+      glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT )
+      shader=ShaderLib.instance()
+      rotX=Mat4() 
+      rotY=Mat4() 
+      rotX.rotateX( self.spinXFace ) 
+      rotY.rotateY( self.spinYFace ) 
+      self.mouseGlobalTX = rotY * rotX 
+      self.mouseGlobalTX.m_30  = self.modelPos.m_x 
+      self.mouseGlobalTX.m_31  = self.modelPos.m_y 
+      self.mouseGlobalTX.m_32  = self.modelPos.m_z 
+      self.loadMatricesToShader()
+      self.vao.bind()
+      self.vao.draw()
+      self.vao.unbind()
+    except OpenGL.error.GLError :
+      print ('error')
+
+  def resizeGL(self, w,h) :
+    self.width=int(w* self.devicePixelRatio())
+    self.height=int(h* self.devicePixelRatio())
+    self.project=perspective( 45.0, float( w ) / h, 0.05, 350.0 )
+
+
+
+  def keyPressEvent(self, event) :
+    key=event.key()
+    if key==Qt.Key_Escape :
+      exit()
+    elif key==Qt.Key_W :
+      glPolygonMode(GL_FRONT_AND_BACK,GL_LINE)
+    elif key==Qt.Key_S :
+      glPolygonMode(GL_FRONT_AND_BACK,GL_FILL)
+    elif key==Qt.Key_Space :
+      self.spinXFace=0
+      self.spinYFace=0
+      self.modelPos.set(Vec3.zero())
+    
+    self.update()
+
+  def mouseMoveEvent(self, event) :
+    if self.rotate and event.buttons() == Qt.LeftButton  :
+      diffx = event.x() - self.origX
+      diffy = event.y() - self.origY
+      self.spinXFace += int( 0.5 * diffy )
+      self.spinYFace += int( 0.5 * diffx )
+      self.origX = event.x()
+      self.origY = event.y()
+      self.update()
+
+    elif  self.translate and event.buttons() == Qt.RightButton :
+
+      diffX      = int( event.x() - self.origXPos )
+      diffY      = int( event.y() - self.origYPos )
+      self.origXPos = event.x()
+      self.origYPos = event.y()
+      self.modelPos.m_x += self.INCREMENT * diffX 
+      self.modelPos.m_y -= self.INCREMENT * diffY 
+      self.update() 
+
+  def mousePressEvent(self,event) :
+    if  event.button() == Qt.LeftButton :
+      self.origX  = event.x()
+      self.origY  = event.y()
+      self.rotate = True
+
+    elif  event.button() == Qt.RightButton :
+      self.origXPos  = event.x() 
+      self.origYPos  = event.y() 
+      self.translate = True
+
+  def mouseReleaseEvent(self,event) :
+    if  event.button() == Qt.LeftButton :
+      self.rotate = False
+
+    elif  event.button() == Qt.RightButton :
+      self.translate = False
+
+  def wheelEvent(self,event) :
+    numPixels = event.pixelDelta() 
+
+    if  numPixels.x() > 0  :
+      self.modelPos.m_z += self.ZOOM
+
+    elif  numPixels.x() < 0 :
+      self.modelPos.m_z -= self.ZOOM
+    self.update() 
+
+
+if __name__ == '__main__':
+  app = QApplication(sys.argv)
+  format=QSurfaceFormat()
+  format.setSamples(4) 
+  format.setMajorVersion(4) 
+  format.setMinorVersion(1) 
+  format.setProfile(QSurfaceFormat.CoreProfile) 
+  # now set the depth buffer to 24 bits
+  format.setDepthBufferSize(24) 
+  # set that as the default format for all windows
+  QSurfaceFormat.setDefaultFormat(format) 
+
+  window = MainWindow()
+  window.setFormat(format)
+  window.resize(1024,720)
+  window.show()
+  sys.exit(app.exec_())
