arc3D/main.py at main · frostdevelop/arc3D · 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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import numpy as np
import math
import pygame as pg
from pygame.locals import QUIT
from arcimport import readobj
from arcdev import *

'''
Material Settings:
0: Color by position
1: Same texture for triangles
2: Mapped Texture
3: Fog
'''

def main():
  pg.init()
  screen = pg.display.set_mode((0, 0), pg.FULLSCREEN)
  info = pg.display.Info()
  width = info.current_w
  height = info.current_h
  clock = pg.time.Clock()
  font = pg.font.Font(pg.font.get_default_font(), 30)
  pg.display.set_caption("Arc3D")
  pg.mouse.set_visible(0)

  background = pg.surface.Surface((width, height))
  background.fill((255, 255, 255))
  screen.blit(background, (0, 0))
  loading = font.render("Loading...", True, (0, 0, 0))
  screen.blit(loading, (width / 2 - loading.get_width() / 2, height / 2 - loading.get_height() / 2))
  pg.display.flip()

  scube = Object(np.asarray([[-2, -2, -2], [2, 2, -2], [-2, 2, -2],[2, -2, -2], [-2, -2, 2], [2, 2, 2], [-2, 2, 2],[2, -2, 2]], dtype=np.float32),np.array([[0, 2, 3], [2, 1, 3], [7, 5, 4], [5, 6, 4], [4, 6, 0], [6, 2, 0], [3, 1, 7], [1, 5, 7], [7, 4, 0],[0, 3, 7], [2, 6, 1], [6, 5, 1]], dtype=np.uint16),2,[10,15,10],[0,0],tex="models/sheepy.png",texcoord=[[0, 0], [0, 1], [1, 0], [1, 1]],texmap=[[0, 1, 2], [1, 3, 2], [0, 1, 2], [1, 3, 2], [0, 1, 2],[1, 3, 2], [0, 1, 2], [1, 3, 2], [2, 0, 1], [1, 3, 2],[0, 1, 2], [1, 3, 2]])

  #stri = Object(np.asarray([[-2,0,-2],[0,0,-4],[2,0,-2]], dtype=np.float32),np.array([[0,2,1]], dtype=np.uint16),2,tex="models/sheepy.png",texcoord=[[0, 0], [0, 1], [1, 0], [1, 1]],texmap=[[0, 1, 2]])

  #iverts, itris, icoord, imap = readobj("models/teapot.obj")
  #teapot = Object(iverts,itris,1,tex="models/sheepy.png")
  iverts, itris, icoord, imap = readobj("models/mountains.obj")
  mountains = Object(iverts,itris,1,[0, -10, -5],[0,0],tex="models/box.jpeg")
  iverts, itris, icoord, imap = readobj("models/Babycrocodile.obj")
  croc = Object(iverts, itris, 2, [0, 10, 0], [0,0], tex="models/BabyCrocodileGreen.png", texcoord=icoord, texmap=imap)

  #iverts, itris, icoord, imap = readobj("models/pacifikytext.obj")
  #pacifikytext = Object(iverts, itris, 0)

  light = Light(np.array([0, 1, 1],dtype=np.float32))
  #light = Light(np.array([0, 1, 0],dtype=np.float32))

  #scene = Scene([[scube,np.asarray([10,15,10], dtype=np.float32)],[croc, np.asarray([0, 10, 0],dtype=np.float32)],[mountains, np.asarray([0, -10, 0], dtype=np.float32)],[pacifikytext,np.asarray([-10,15,-20], dtype=np.float32)]],light,[50, 127, 200])
  scene = Scene([scube,croc,mountains],light,(50, 127, 200))
  #scene = Scene([[scube,np.asarray([10,15,10], dtype=np.float32)],[croc, np.asarray([0, 10, 0],dtype=np.float32)]],light,[50, 127, 200])
  #scene = Scene([[stri,np.asarray([0,0,-0.5], dtype=np.float32)]],light,[50, 127, 200])
  #scene = Scene([[mountains,np.asarray([0,0,0], dtype=np.float32)]],light,[50,127,200])
  #scene = Scene([[scube,np.asarray([0,0,0], dtype=np.float32)]], light, [50,127,200])
  #camera = Camera(np.pi/8,[0,0,-10],0,0, width, height)
  camera = Camera(np.pi / 2, (0.0, 1.5, -5.0), (0,0), 1000, 0.5)
  #camera = Camera(np.pi / 1.1, (0.0, 1.5, -5.0), (0,0), 1000, 0.5)
  renderer = Renderer(width, height, camera, [[1, 0], [0, 1], [1, 1]], False)
  running = True

  while running:
    elapsed_time = clock.tick() * 0.001

    for event in pg.event.get():
      if event.type == pg.QUIT:
        running = False
      if event.type == pg.KEYDOWN:
        if event.key == pg.K_ESCAPE:
          running = False

    renderer.move(elapsed_time)
    #print(camera.velocity)

    #"""
    scene.objects[0].rotation += elapsed_time / 2, elapsed_time
    scene.objects[0].upd()
    scene.objects[1].rotation += elapsed_time/5, 0
    scene.objects[1].upd()

    light.upd((math.sin(pg.time.get_ticks() / 1000), 1, 1))
    #"""

    renderer.render(scene)
    pg.surfarray.blit_array(screen, renderer.surface)

    #"""
    positiontext = font.render(
        f'XYZ: {truncate(camera.position[0])} {truncate(camera.position[1])} {truncate(camera.position[2])}',
        False, (0, 0, 0))
    angletext = font.render(
        f'Angle: {truncate(camera.angle[1])} {truncate(camera.angle[0])}', False,
        (0, 0, 0))
    fps = font.render(f'FPS: {round(1/(elapsed_time + 1e-16))}', False, (0,0,0))
    screen.blit(positiontext, (10, 10))
    screen.blit(angletext, (10, 50))
    screen.blit(fps, (10, 90))
    #"""
    pg.display.flip()

if __name__ == "__main__":
  main()
  pg.quit()