| ''' |
| 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() |
|
|
