156 lines
4.6 KiB
Python
Executable File
156 lines
4.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
try:
|
|
from numpy import complex256
|
|
except ImportError:
|
|
from numpy import complex128 as complex256
|
|
print("Warning! Cannot use full precision!")
|
|
|
|
from time import time_ns, sleep
|
|
from cmath import exp, pi
|
|
import pygame
|
|
import pygame.freetype
|
|
from gyro import *
|
|
from lines import *
|
|
|
|
def deg2rad(rA): return rA/180*PI
|
|
|
|
I = complex256(complex(0,1))
|
|
WHITE = (255,255,255)
|
|
BLACK = (0,0,0)
|
|
OFFSET = 0.015625
|
|
PI = pi
|
|
ROT = cNorm(exp(deg2rad(2)*I))
|
|
IROT = 1/ROT
|
|
F = 5 * 10**7
|
|
F_ = 10**9
|
|
|
|
SKY = (127,127,255)
|
|
GROUND = (102, 51, 0)
|
|
|
|
class Segment:
|
|
def __init__(self,bA,cA,cB,trColor):
|
|
self.isFinite = bA
|
|
self.pointA = cA
|
|
self.pointB = cB
|
|
self.color = trColor
|
|
|
|
class DrawnSegment:
|
|
def __init__(self,height,color):
|
|
self.height = height
|
|
self.color = color
|
|
|
|
level = [
|
|
Segment(True,.6,I*.6,(255,0,0)),
|
|
Segment(True,.6,-I*.6,(0,255,0)),
|
|
Segment(True,-.6,-I*.6,(0,0,255)),
|
|
Segment(True,-.6,I*.6,(255,255,0))
|
|
]
|
|
|
|
def draw(level,gPlayer,fov,res):
|
|
drawn = list()
|
|
irot = exp(fov/res*I)
|
|
iprot = gPlayer.cRot
|
|
for i in range(res):
|
|
m = DrawnSegment(0,BLACK)
|
|
rot = irot**(i-(res/2)) * iprot
|
|
for j in level:
|
|
tA = Poincare2Klein(MobiusAdd(j.pointA,-gPlayer.cPos))
|
|
tB = Poincare2Klein(MobiusAdd(j.pointB,-gPlayer.cPos))
|
|
try: # This function faults from time to time
|
|
cInt = cLineIntersection(tA,tB,complex256(0),rot)
|
|
except ZeroDivisionError:
|
|
continue
|
|
rDist = cDist(0,cInt) * irot.real
|
|
if not (j.isFinite and cBetween(tA,tB,cInt)):
|
|
continue
|
|
if cDot(cInt,rot) > 0:
|
|
continue
|
|
if (1 - m.height) > rDist:
|
|
m = DrawnSegment(1-rDist,j.color)
|
|
drawn.append(m)
|
|
return drawn
|
|
|
|
def cap(rN):
|
|
if rN < 0:
|
|
return 0
|
|
return rN
|
|
|
|
def renderDebugInfo(gPlayer, clock, fontSize = 18):
|
|
font = pygame.freetype.Font(None, fontSize)
|
|
font_fg = (255, 255, 255)
|
|
font_bg = (255, 255, 255, 80)
|
|
|
|
fps_surf = font.render("FPS: " + str(int(clock.get_fps())), font_fg, font_bg)[0]
|
|
cPos_surf = font.render("cPos: " + str(gPlayer.cPos), font_fg, font_bg)[0]
|
|
cRot_surf = font.render("cRot: " + str(gPlayer.cRot), font_fg, font_bg)[0]
|
|
|
|
return fps_surf, cPos_surf, cRot_surf
|
|
|
|
def mainLoop():
|
|
gPlayer = GyroVector(0,1)
|
|
display = pygame.display.set_mode((1280,720))
|
|
clock = pygame.time.Clock()
|
|
fontSize = 18
|
|
debugInfo = True
|
|
|
|
while True:
|
|
#framestart = time_ns()
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
return True
|
|
if event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_q:
|
|
gPlayer.cRot *= -1
|
|
keys = pygame.key.get_pressed()
|
|
if keys[pygame.K_d]:
|
|
gPlayer.rotate(ROT)
|
|
if keys[pygame.K_a]:
|
|
gPlayer.rotate(IROT)
|
|
if keys[pygame.K_w]:
|
|
gPlayer -= GyroVector(OFFSET * gPlayer.cRot, 1)
|
|
if keys[pygame.K_s]:
|
|
gPlayer += GyroVector(OFFSET * gPlayer.cRot, 1)
|
|
if keys[pygame.K_F3]:
|
|
debugInfo = not debugInfo
|
|
|
|
display.fill(WHITE)
|
|
#pygame.draw.rect(display,BLACK, c_tr(Poincare2Klein(gPlayer.cPos) * -100) + (100,100),0)
|
|
drawn = draw(level,gPlayer,PI/2,160)
|
|
pygame.draw.rect(display,SKY, (0,0,1280,360))
|
|
pygame.draw.rect(display,GROUND, (0,360,1280,360))
|
|
n = 0
|
|
for i in drawn:
|
|
pygame.draw.rect(display,i.color, (n,(1 - i.height) * 360,8,cap(i.height) * 1280))
|
|
n += 8
|
|
|
|
if debugInfo:
|
|
fps_surf, cPos_surf, cRot_surf = renderDebugInfo(gPlayer, clock, fontSize)
|
|
|
|
display.blit(fps_surf, (0, 0))
|
|
display.blit(cPos_surf, (0, fps_surf.get_height()))
|
|
display.blit(cRot_surf, (0, fps_surf.get_height() + fontSize))
|
|
for i in level:
|
|
a = complex(Poincare2Klein(MobiusAdd(i.pointA,-gPlayer.cPos)))
|
|
b = complex(Poincare2Klein(MobiusAdd(i.pointB,-gPlayer.cPos)))
|
|
pygame.draw.line(display, WHITE, (a.real * 100 + 100, (a.imag * 100 + 200)), (b.real * 100 + 100, (b.imag * 100 + 200)))
|
|
|
|
pygame.display.update()
|
|
|
|
#frameend = time_ns()
|
|
clock.tick()
|
|
#sleep((frameend-framestart) / F_)
|
|
return True
|
|
|
|
def main():
|
|
pygame.init()
|
|
pygame.display.set_caption('P3HE')
|
|
if not pygame.get_init():
|
|
return False
|
|
retstatus = mainLoop()
|
|
return retstatus
|
|
|
|
if __name__ == "__main__":
|
|
if not main():
|
|
print("An error occured")
|