2024-04-03 11:20:42 +00:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
2024-04-04 19:05:23 +00:00
|
|
|
from math import copysign, pi, acos
|
2024-04-04 19:40:37 +00:00
|
|
|
import sys
|
2024-04-04 19:03:41 +00:00
|
|
|
|
2024-04-03 11:20:42 +00:00
|
|
|
import pygame
|
|
|
|
import pygame.freetype
|
2024-04-04 19:03:41 +00:00
|
|
|
|
2024-04-05 12:40:54 +00:00
|
|
|
from gyro import GyroVector, Poincare2Klein
|
2024-04-05 18:06:56 +00:00
|
|
|
from levels import open_level, save_level, make_wall
|
2024-04-05 12:40:54 +00:00
|
|
|
from constants import I, WHITE, BLACK, IROT, ROT, OFFSET
|
2024-04-04 19:03:41 +00:00
|
|
|
from draw import draw
|
2024-04-04 20:55:54 +00:00
|
|
|
from alert import Alert
|
2024-04-03 11:20:42 +00:00
|
|
|
|
2024-04-04 18:03:47 +00:00
|
|
|
gOrigin = GyroVector(0,1)
|
2024-04-04 16:05:20 +00:00
|
|
|
|
2024-04-03 12:47:27 +00:00
|
|
|
SKY = (127,127,255)
|
|
|
|
GROUND = (102, 51, 0)
|
2024-04-03 11:20:42 +00:00
|
|
|
|
2024-04-03 14:07:30 +00:00
|
|
|
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]
|
2024-04-04 18:31:45 +00:00
|
|
|
cRot_surf = font.render("cRot: " + str(copysign(acos(gPlayer.cRot.real),gPlayer.cRot.imag) / pi * 180), font_fg, font_bg)[0]
|
2024-04-03 14:07:30 +00:00
|
|
|
|
|
|
|
return fps_surf, cPos_surf, cRot_surf
|
|
|
|
|
2024-04-04 20:51:26 +00:00
|
|
|
def nonzerocap(n):
|
|
|
|
if n <= 0: return 1
|
|
|
|
return n
|
2024-04-04 18:41:10 +00:00
|
|
|
|
2024-04-06 20:12:28 +00:00
|
|
|
class EngineEvent:
|
|
|
|
def __init__(self, sVariableToModify, fLambda, tsArguments=None):
|
|
|
|
self.sVariableToModify = sVariableToModify
|
|
|
|
self.fLambda = fLambda
|
|
|
|
if tsArguments is not None:
|
|
|
|
self.tsArguments = tsArguments
|
|
|
|
else:
|
|
|
|
self.tsArguments = (sVariableToModify,)
|
|
|
|
|
|
|
|
def defaultcontrols():
|
|
|
|
aoEngineEvents = list()
|
|
|
|
for event in pygame.event.get():
|
|
|
|
if event.type == pygame.QUIT:
|
|
|
|
aoEngineEvents.append(EngineEvent("bCont", lambda _: False))
|
|
|
|
if event.type == pygame.KEYDOWN:
|
|
|
|
if event.key == pygame.K_r:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].rotate(-1), tsArguments=("gPlayer",)))
|
|
|
|
if event.key == pygame.K_F3:
|
|
|
|
aoEngineEvents.append(EngineEvent("debugInfo", lambda args: not args[0]))
|
|
|
|
if event.key == pygame.K_F1:
|
|
|
|
aoEngineEvents.append(EngineEvent("bCap", lambda args: not args[0]))
|
|
|
|
if event.key == pygame.K_F2:
|
|
|
|
def fX(args):
|
|
|
|
#0: wall_buffer
|
|
|
|
#1: gPlayer
|
|
|
|
#2: level
|
|
|
|
if (len(args[0]) == 1):
|
|
|
|
args[0].append(args[1].cPos)
|
|
|
|
make_wall(args[0], args[2])
|
|
|
|
args[0].clear()
|
|
|
|
else:
|
|
|
|
args[0].append(args[1].cPos)
|
|
|
|
aoEngineEvents.append(EngineEvent(None, fX, tsArguments=("wall_buffer", "gPlayer", "level") ))
|
|
|
|
if event.key == pygame.K_F5:
|
|
|
|
#0: level
|
|
|
|
#1: alert_append
|
|
|
|
#2: display
|
|
|
|
def fX(args):
|
|
|
|
filename = save_level(args[0])
|
|
|
|
alert = Alert(f"File saved as {filename}", args[2])
|
|
|
|
|
|
|
|
args[1](alert, 5)
|
|
|
|
aoEngineEvents.append(EngineEvent(None, fX, tsArguments=("level", "alert_append", "display") ))
|
|
|
|
if event.key == pygame.K_F4:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: print(args[0].cPos), tsArguments=("gPlayer",)))
|
|
|
|
keys = pygame.key.get_pressed()
|
|
|
|
if keys[pygame.K_d]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].rotate( ROT), tsArguments=("gPlayer",)))
|
|
|
|
if keys[pygame.K_a]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].rotate(IROT), tsArguments=("gPlayer",)))
|
|
|
|
if keys[pygame.K_q]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].__iadd__(GyroVector(OFFSET * args[0].cRot*I, 1)), tsArguments=("gPlayer",)))
|
|
|
|
if keys[pygame.K_e]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].__isub__(GyroVector(OFFSET * args[0].cRot*I, 1)), tsArguments=("gPlayer",)))
|
|
|
|
if keys[pygame.K_w]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].__isub__(GyroVector(OFFSET * args[0].cRot , 1)), tsArguments=("gPlayer",)))
|
|
|
|
if keys[pygame.K_s]:
|
|
|
|
aoEngineEvents.append(EngineEvent(None, lambda args: args[0].__iadd__(GyroVector(OFFSET * args[0].cRot , 1)), tsArguments=("gPlayer",)))
|
|
|
|
return aoEngineEvents
|
|
|
|
|
2024-04-03 11:20:42 +00:00
|
|
|
def mainLoop():
|
2024-04-05 12:40:54 +00:00
|
|
|
bCap = True
|
|
|
|
bCont = True
|
2024-04-03 11:20:42 +00:00
|
|
|
gPlayer = GyroVector(0,1)
|
|
|
|
display = pygame.display.set_mode((1280,720))
|
2024-04-03 14:07:30 +00:00
|
|
|
clock = pygame.time.Clock()
|
|
|
|
fontSize = 18
|
|
|
|
debugInfo = True
|
2024-04-04 17:01:33 +00:00
|
|
|
wall_buffer = []
|
2024-04-04 20:55:54 +00:00
|
|
|
alerts = []
|
2024-04-06 20:12:28 +00:00
|
|
|
aoEngineEvents = []
|
|
|
|
fvControl_ao = defaultcontrols
|
|
|
|
|
2024-04-04 20:55:54 +00:00
|
|
|
def alert_append(alert, delay):
|
|
|
|
if len(alerts) >= 1:
|
|
|
|
alerts.pop()
|
|
|
|
|
|
|
|
alerts.append(alert)
|
|
|
|
alert.start_hide_thread(alerts, delay)
|
2024-04-03 14:07:30 +00:00
|
|
|
|
2024-04-04 19:40:37 +00:00
|
|
|
try:
|
2024-04-05 18:06:56 +00:00
|
|
|
level = open_level(sys.argv[1])
|
2024-04-04 19:40:37 +00:00
|
|
|
except IndexError:
|
2024-04-05 18:06:56 +00:00
|
|
|
level = open_level("maps/squareroom.json")
|
2024-04-04 19:40:37 +00:00
|
|
|
|
2024-04-05 12:40:54 +00:00
|
|
|
while bCont:
|
2024-04-06 20:12:28 +00:00
|
|
|
aoEngineEvents = fvControl_ao()
|
|
|
|
state = {
|
|
|
|
'bCap': bCap,
|
|
|
|
'bCont': bCont,
|
|
|
|
'gPlayer': gPlayer,
|
|
|
|
'display': display,
|
|
|
|
'fontSize': fontSize,
|
|
|
|
'debugInfo': debugInfo,
|
|
|
|
'wall_buffer': wall_buffer,
|
|
|
|
'alerts': alerts,
|
|
|
|
'alert_append': alert_append,
|
|
|
|
'aoEngineEvents': aoEngineEvents,
|
|
|
|
'level': level
|
|
|
|
}
|
|
|
|
for i in aoEngineEvents:
|
|
|
|
if i.sVariableToModify == "bCont":
|
|
|
|
bCont = i.fLambda(list(map(lambda oX: state[oX], i.tsArguments)))
|
|
|
|
if i.sVariableToModify == "debugInfo":
|
|
|
|
debugInfo = i.fLambda(list(map(lambda oX: state[oX], i.tsArguments)))
|
|
|
|
if i.sVariableToModify == "gPlayer":
|
|
|
|
gPlayer = i.fLambda(list(map(lambda oX: state[oX], i.tsArguments)))
|
|
|
|
if i.sVariableToModify == "bCap":
|
|
|
|
bCap = i.fLambda(list(map(lambda oX: state[oX], i.tsArguments)))
|
|
|
|
if i.sVariableToModify is None:
|
|
|
|
i.fLambda(list(map(lambda oX: state[oX], i.tsArguments)))
|
2024-04-03 14:07:30 +00:00
|
|
|
|
2024-04-03 11:20:42 +00:00
|
|
|
display.fill(WHITE)
|
2024-04-04 19:03:41 +00:00
|
|
|
drawn = draw(level,gPlayer,pi/4,320)
|
2024-04-03 12:47:27 +00:00
|
|
|
pygame.draw.rect(display,SKY, (0,0,1280,360))
|
|
|
|
pygame.draw.rect(display,GROUND, (0,360,1280,360))
|
2024-04-03 11:20:42 +00:00
|
|
|
n = 0
|
|
|
|
for i in drawn:
|
2024-04-04 20:51:26 +00:00
|
|
|
pygame.draw.rect(display,i.color, (n,int((nonzerocap(i.dist)) * 360),8,int((1 -nonzerocap(i.dist)) * 1280)))
|
2024-04-04 16:05:20 +00:00
|
|
|
n += 4
|
2024-04-04 19:08:44 +00:00
|
|
|
gPlayer.normalize()
|
2024-04-03 14:07:30 +00:00
|
|
|
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))
|
2024-04-06 18:43:51 +00:00
|
|
|
pygame.draw.circle(display, BLACK, (100, 200), 100)
|
2024-04-03 14:20:04 +00:00
|
|
|
for i in level:
|
2024-04-06 14:22:34 +00:00
|
|
|
a = Poincare2Klein(i.pointA)
|
|
|
|
b = Poincare2Klein(i.pointB)
|
2024-04-04 17:10:35 +00:00
|
|
|
pygame.draw.line(display, i.color, (a.real * 100 + 100, (a.imag * 100 + 200)), (b.real * 100 + 100, (b.imag * 100 + 200)))
|
2024-04-04 17:02:44 +00:00
|
|
|
c = complex(Poincare2Klein(gPlayer.cPos))
|
|
|
|
pygame.draw.rect(display, WHITE, (c.real * 100 + 95, (c.imag * 100 + 195), 10, 10))
|
2024-04-03 13:06:58 +00:00
|
|
|
|
2024-04-04 20:55:54 +00:00
|
|
|
for alert in alerts:
|
|
|
|
alert.draw(display)
|
|
|
|
|
2024-04-03 11:20:42 +00:00
|
|
|
pygame.display.update()
|
2024-04-04 18:25:24 +00:00
|
|
|
if bCap:
|
|
|
|
clock.tick(60)
|
|
|
|
else:
|
|
|
|
clock.tick()
|
2024-04-03 13:06:58 +00:00
|
|
|
return True
|
2024-04-03 11:20:42 +00:00
|
|
|
|
|
|
|
def main():
|
|
|
|
pygame.init()
|
2024-04-03 13:06:58 +00:00
|
|
|
pygame.display.set_caption('P3HE')
|
2024-04-03 11:20:42 +00:00
|
|
|
if not pygame.get_init():
|
|
|
|
return False
|
|
|
|
retstatus = mainLoop()
|
|
|
|
return retstatus
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if not main():
|
|
|
|
print("An error occured")
|