object_3d.py

import pygame as pg
from matrix_functions import *

class Object3D:
    def __init__(self, render):
       self.render = render
       self.vertexes = np.array([(0, 0, 0, 1), (0, 1, 0, 1), (1, 1, 0, 1), (1, 0, 0, 1),
                                 (0, 0, 1, 1), (0, 1, 1, 1), (1, 1, 1, 1), (1, 0, 1, 1)])

       self.faces = np.array(
           [(0, 1, 2, 3), (4, 5, 6, 7), (0, 4, 5, 1), (2, 3, 7, 6), (1, 2, 6, 5), (0, 3, 7, 4)])
       
       def draw(self): 
           self.screen_projection() 
       
       def screen_projection(self):
           vertexes = self.vertexes @ self.render.camera.camera_matrix()
           vertexes = vertexes @ self.render.projection.projection_matrix
           vertexes /= vertexes[:, -1:].reshape(-1, 1)
           vertexes[(vertexes > 1) | (vertexes < -1)] = 0
           vertexes = vertexes @ self.render.projection.to_screen_matrix
           vertexes = vertexes [:, :2]
           
           for face in self.faces: 
               polygon = vertexes[face]
               if not np.any((polygon == self.render.H_WIDTH) | (polygon == self.render.H_HEIGHT)):
                   pg.draw.polygon(self.render.screen, pg.Color('orange'), polygon, 3)
                   
           for vertex in vertexes:
               if not np.any((vertex == self.render.H_WIDTH) | (vertex == self.render.H_HEIGHT)):
                    pg.draw.circle(self.render.screen, pg.Color('red'), vertex, 5)
                    
                    
       def translate(self, pos):
           self.vertexes = self.vertexes @ translate(pos)

       def scale(self, scale_to):
            self.vertexes = self.vertexes @ scale(scale_to)
        
       def rotate_x(self, angle):
            self.vertexes = self.vertexes @ rotate_x(angle)
            
       def rotaty_y(self, angle):
            self.vertexes = self.vertexes @ rotate_y(angle)
            
       def rotate_z(self, angle):
            self.vertexes = self.vertexes @ rotate_z(angle)