| | ''' |
| | 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 |
| | import math |
| |
|
| | from .framework import * |
| | from .norm_render import NormRender |
| |
|
| |
|
| | class NormalRender(NormRender): |
| |
|
| | def __init__(self, width=1600, height=1200, name='Normal Renderer'): |
| | NormRender.__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): |
| | NormRender.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 euler_to_rot_mat(self, r_x, r_y, r_z): |
| | R_x = np.array([[1, 0, 0], [0, math.cos(r_x), -math.sin(r_x)], |
| | [0, math.sin(r_x), math.cos(r_x)]]) |
| |
|
| | R_y = np.array([[math.cos(r_y), 0, math.sin(r_y)], [0, 1, 0], |
| | [-math.sin(r_y), 0, math.cos(r_y)]]) |
| |
|
| | R_z = np.array([[math.cos(r_z), -math.sin(r_z), 0], |
| | [math.sin(r_z), math.cos(r_z), 0], [0, 0, 1]]) |
| |
|
| | R = np.dot(R_z, np.dot(R_y, R_x)) |
| |
|
| | return R |
| |
|
| | 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() |
| |
|
