Object_3d.py

import numpy as np
import pygame as pg
from matrix_functions import *


class Object_3d:
    def __init__(self, render, vertexes, faces):
        self.render = render
        # Cube vertexes and faces
        self.vertexes = np.array([np.array(v) for v in vertexes])
        self.faces = np.array([np.array(face) for face in faces])
        self.font = pg.font.SysFont('Arial', 30, bold=True)
        self.color_faces = [(pg.Color('orange'), face) for face in self.faces]
        self.movement_flag = self.draw_vertexes = True, True
        self.label = ''

    def draw(self):
        self.screen_projection()
        self.movement()

    def movement(self):
        if self.movement_flag:
            self.rotate_y(pg.time.get_ticks() % 0.005)

    def screen_projection(self):
        vertexes = self.vertexes @ self.render.view.view_matrix()
        vertexes = vertexes @ self.render.projection.projection_matrix
        vertexes /= vertexes[:, -1].reshape(-1, 1)
        vertexes[(vertexes > 2) | (vertexes < -2)] = 0
        vertexes = vertexes @ self.render.projection.to_screen_matrix
        vertexes = vertexes[:, :2]

        for index, color_face in enumerate(self.color_faces):
            color, face = color_face
            polygon = vertexes[face]
            if not np.any((polygon == self.render.H_WIDTH) | (polygon == self.render.H_HEIGHT)):
                pg.draw.polygon(self.render.screen, color, polygon, 1)
                if self.label:
                    text = self.font.render(self.label[index], True, pg.Color('white'))
                    self.render.screen.blit(text, polygon[-1])
        if self.draw_vertexes:
            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('white'), vertex, 2)

    def translate(self, point):
        self.vertexes = self.vertexes @ translate_M(point)

    def scale(self, scale_to):
        self.vertexes = self.vertexes @ scale_M(scale_to)

    def rotate_x(self, angle):
        self.vertexes = self.vertexes @ rotate_X(angle)

    def rotate_y(self, angle):
        self.vertexes = self.vertexes @ rotate_Y(angle)

    def rotate_z(self, angle):
        self.vertexes = self.vertexes @ rotate_Z(angle)

# class Axes(Object_3d):
#     def __init__(self, render):
#         super().__init__(render)
#         self.vertexes = np.array([(0, 0, 0, 1), (1, 0, 0, 1), (0, 1, 0, 1), (0, 0, 1, 1)])
#         self.faces = np.array([(0, 1), (0, 2), (0, 3)])
#         self.colors = [pg.Color('red'), pg.Color('green'), pg.Color('blue')]
#         self.color_faces = [(color, face) for color, face in zip(self.colors, self.faces)]
#         self.draw_vertexes = False
#         self.label = 'XYZ'