arc3D/sponza.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
import numpy as np
import math
import pygame as pg
from pygame.locals import QUIT
from arcimport import readobj
from arcdev import *

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()
  #try:
  #iverts, itris, icoord, imap = readobj("models/sponzaoneImproved.obj")
  #sponza = Object(iverts,itris,2,[0, -10, -5],[0,0],tex="models/sponza_diff-0.25.png",texcoord=icoord, texmap=imap)
  iverts, itris, icoord, imap = readobj("models/Hall2.obj")
  #print(icoord)
  #sponza = Object(iverts,itris,2,[0, -10, -5],[0,0],tex="models/singlemat.png",texcoord=icoord, texmap=imap)
  sponza = Object(iverts,itris,4,[0, -5, -5],[0,0],color=(153,183,186)) #for some reason dies on texture
  #except e:
      #print(e)

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

  scene = Scene([sponza],light,(50, 127, 200))
  camera = Camera(np.pi / 2, (0.0, 1.5, -5.0), (0,0), 1000, 0.5)
  renderer = Renderer(width, height, camera, [[1, 0], [0, 1], [1, 1]], True)
  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)
    light.upd((math.sin(pg.time.get_ticks() / 1000), 1, 1))

    renderer.render(scene)
    screen.blit(pg.surfarray.make_surface(renderer.surface), (0, 0))

    #"""
    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()
  print("Quit...")
  pg.quit()