|
''' |
|
MIT License |
|
|
|
Copyright (c) 2019 Shunsuke Saito, Zeng Huang, and Ryota Natsume |
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy |
|
of this software and associated documentation files (the "Software"), to deal |
|
in the Software without restriction, including without limitation the rights |
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|
copies of the Software, and to permit persons to whom the Software is |
|
furnished to do so, subject to the following conditions: |
|
|
|
The above copyright notice and this permission notice shall be included in all |
|
copies or substantial portions of the Software. |
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|
SOFTWARE. |
|
''' |
|
import numpy as np |
|
|
|
from .framework import * |
|
from .cam_render import CamRender |
|
|
|
|
|
class NormalRender(CamRender): |
|
def __init__(self, width=1600, height=1200, name='Normal Renderer'): |
|
CamRender.__init__(self, width, height, name, program_files=['normal.vs', 'normal.fs']) |
|
|
|
self.norm_buffer = glGenBuffers(1) |
|
|
|
self.norm_data = None |
|
|
|
def set_normal_mesh(self, vertices, faces, norms, face_normals): |
|
CamRender.set_mesh(self, vertices, faces) |
|
|
|
self.norm_data = norms[face_normals.reshape([-1])] |
|
|
|
glBindBuffer(GL_ARRAY_BUFFER, self.norm_buffer) |
|
glBufferData(GL_ARRAY_BUFFER, self.norm_data, GL_STATIC_DRAW) |
|
|
|
glBindBuffer(GL_ARRAY_BUFFER, 0) |
|
|
|
def draw(self): |
|
self.draw_init() |
|
|
|
glUseProgram(self.program) |
|
glUniformMatrix4fv(self.model_mat_unif, 1, GL_FALSE, self.model_view_matrix.transpose()) |
|
glUniformMatrix4fv(self.persp_mat_unif, 1, GL_FALSE, self.projection_matrix.transpose()) |
|
|
|
|
|
glBindBuffer(GL_ARRAY_BUFFER, self.vertex_buffer) |
|
|
|
glEnableVertexAttribArray(0) |
|
glVertexAttribPointer(0, self.vertex_dim, GL_DOUBLE, GL_FALSE, 0, None) |
|
|
|
|
|
glBindBuffer(GL_ARRAY_BUFFER, self.norm_buffer) |
|
|
|
glEnableVertexAttribArray(1) |
|
glVertexAttribPointer(1, 3, GL_DOUBLE, GL_FALSE, 0, None) |
|
|
|
glDrawArrays(GL_TRIANGLES, 0, self.n_vertices) |
|
|
|
glDisableVertexAttribArray(1) |
|
glDisableVertexAttribArray(0) |
|
|
|
glBindBuffer(GL_ARRAY_BUFFER, 0) |
|
|
|
glUseProgram(0) |
|
|
|
self.draw_end() |
|
|
