35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
import numpy as np
|
|
import pygame as pg
|
|
import os
|
|
from src.vertex import Vertex
|
|
|
|
class Object:
|
|
vertices: list[Vertex]
|
|
lines: list[(Vertex, Vertex)]
|
|
|
|
def __init__(self, vertices: list[Vertex], lines: list[(Vertex, Vertex)]):
|
|
self.vertices = vertices
|
|
self.lines = lines
|
|
|
|
def attach_renderer(self, renderer):
|
|
self.__renderer = renderer
|
|
|
|
def draw(self):
|
|
vertices_matrix = np.array([[v.x, v.y, v.z, 1.0] for v in self.vertices])
|
|
vertices_matrix = vertices_matrix @ self.__renderer.camera.camera_matrix()
|
|
vertices_matrix = vertices_matrix @ self.__renderer.camera.projection_matrix()
|
|
vertices_matrix /= vertices_matrix[:, -1].reshape(-1, 1)
|
|
vertices_matrix = vertices_matrix @ self.__renderer.camera.to_screen_matrix()
|
|
vertices_matrix = vertices_matrix[:, :2]
|
|
vertices = {}
|
|
for i in range(len(self.vertices)):
|
|
vertices[self.vertices[i]] = vertices_matrix[i].tolist()[0]
|
|
|
|
for v in vertices.values():
|
|
pg.draw.circle(self.__renderer.screen, pg.Color("black"), v, 2)
|
|
for l in self.lines:
|
|
v1 = vertices[l[0]]
|
|
v2 = vertices[l[1]]
|
|
pg.draw.line(self.__renderer.screen, pg.Color("black"), v1, v2, 1)
|
|
|
|
|