main.py

import os
import json
import sys
if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
    os.chdir(sys._MEIPASS)
else:
    os.chdir(os.path.dirname(os.path.realpath(__file__)))

import pygame
import pygame.locals
pygame.init()

import lib.constants
import lib.globals
import lib.font
import lib.sound
import lib.utils
import lib.debug
import lib.scene.title

if __name__ == '__main__':
    # Hide cursor
    pygame.mouse.set_cursor((8, 8), (0, 0), (0, 0, 0, 0, 0, 0, 0, 0), (0, 0, 0, 0, 0, 0, 0, 0))
    lib.sound.playBgm(lib.sound.BGM.TITLE)

    lib.globals.nextScene = lib.scene.title
    loop = True
    freeze = False
    while loop:
        keyPressed = pygame.key.get_pressed()
        lib.globals.keysLastFrame = lib.globals.keys
        lib.globals.keys = {k: keyPressed[k] for k in (
            pygame.K_UP,
            pygame.K_DOWN,
            pygame.K_LEFT,
            pygame.K_RIGHT,
            pygame.K_LSHIFT,
            pygame.K_LCTRL,
            pygame.K_LALT,
            pygame.K_z,
            pygame.K_x,
            pygame.K_c,
            pygame.K_p,
            pygame.K_ESCAPE,
        )}

        for e in pygame.event.get():
            if e.type == pygame.QUIT:
                loop = False

        if lib.globals.keys[pygame.K_p] and not lib.globals.keysLastFrame[pygame.K_p]:
            freeze = not freeze
            lib.sound.SFX.PAUSE.play()
            if freeze:
                pygame.mixer.music.pause()
                pygame.display.set_caption(f'{lib.constants.TITLE} (Paused)')
            else:
                pygame.mixer.music.unpause()
                pygame.display.set_caption(lib.constants.TITLE)

        if not freeze:
            lib.globals.currentScene = lib.globals.nextScene
            lib.globals.currentScene.update()
            lib.globals.currentScene.draw(lib.globals.screen)

            if lib.globals.config['inputDisplay']:
                for key, pos in (
                    (pygame.K_LSHIFT, (10, 10, 45, 20)),
                    (pygame.K_z, (10, 35, 20, 20)),
                    (pygame.K_x, (35, 35, 20, 20)),
                    (pygame.K_LEFT, (60, 35, 20, 20)),
                    (pygame.K_DOWN, (85, 35, 20, 20)),
                    (pygame.K_UP, (85, 10, 20, 20)),
                    (pygame.K_RIGHT, (110, 35, 20, 20)),
                ):
                    pygame.draw.rect(
                        lib.globals.screen,
                        lib.constants.DEBUG_INPUT_DISPLAY_PRESSED if lib.globals.keys[key] else lib.constants.DEBUG_INPUT_DISPLAY_NOTPRESSED,
                        (1140 + pos[0], 895 + pos[1], pos[2], pos[3])
                    )

            fps = lib.globals.clock.get_fps()
            lib.globals.screen.blit(
                lib.font.FONT_SMALL.render(
                    f'FPS: {fps:.02f}',
                   pygame.Color(255, 255, 255) if abs(1 - fps / 60) < .05 else pygame.Color(255, 255, 0),
                ),
                (1190, 930),
            )
            pygame.display.flip()

        lib.globals.clock.tick(
            240 if (
                lib.globals.keys[pygame.K_LCTRL] and (
                    not lib.globals.keys[pygame.K_LALT] or not lib.globals.groupBoss.sprite
                )
            ) else 60
        )

    # Write config and savedata before quit
    with open(lib.constants.PATH_CONFIG, 'w', encoding='utf-8') as f:
        json.dump(lib.globals.config, f, separators=(',', ':'))
    with open(lib.constants.PATH_SAVEDATA, 'w', encoding='utf-8') as f:
        json.dump(lib.globals.savedata, f, separators=(',', ':'))

    pygame.quit()