Python-3D-Renderer/main.py at main · EthanNoble/Python-3D-Renderer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Developed in Python 3.12.5

import pygame as pg
from environment import Screen
from environment import RenderSpace
from environment import Mesh
from environment import deg_to_rad

WIDTH = 1000
HEIGHT = 1000

screen = Screen(WIDTH, HEIGHT)
time, time_steps = 0, 110

render_space = RenderSpace(ambient_lighting=True)
# Just put in the name of your object file that is in the objects folder
mesh = Mesh(object_file='bunny.obj', origin_translate=True, preload_normals=True)

degree = deg_to_rad(1)

while True:
    for event in pg.event.get():
        if event.type == pg.QUIT:
            pg.quit()
        elif event.type == pg.MOUSEBUTTONDOWN:
            # Mouse scroll up
            if event.button == 4:
                render_space.zoom += render_space.zoom / render_space.zoom_speed
            # Mouse scroll down
            elif event.button == 5:
                render_space.zoom -= render_space.zoom / render_space.zoom_speed
                # Cap maximum zoom out
                if render_space.zoom <= 0:
                    render_space.zoom = render_space.zoom / render_space.zoom_speed

    # Comment out these three lines to stop mesh from rotating without user input
    mesh.rotate(axis='x', theta=degree)
    mesh.rotate(axis='y', theta=degree)
    mesh.rotate(axis='z', theta=degree)

    keys = pg.key.get_pressed()

    if keys[pg.K_w]:
        mesh.rotate(axis='x', theta=degree)
    if keys[pg.K_s]:
        mesh.rotate(axis='x', theta=-degree)
    if keys[pg.K_a]:
        mesh.rotate(axis='y', theta=-degree)
    if keys[pg.K_d]:
        mesh.rotate(axis='y', theta=degree)

    # Reset screen
    screen.set()

    # Update
    time_now = pg.time.get_ticks()
    if time_now - time > time_steps:
        # Obviously using the CPU for 3D graphics rendering
        # is laughably awful but who cares
        render_space.render_mesh(mesh, screen)

    # Flip() display to update screen
    screen.flip()

    # Limit FPS to 60
    screen.time_step(60)

    # Display FPS
    pg.display.set_caption(f'FPS - {int(screen.clock.get_fps())}')