Add profiler

arctic-masher.py

@@ -1,3 +1,5 @@
+from profilehooks import profile
+
 import pygame
 import sys
 import random
@@ -9,106 +11,112 @@ from player import *
 from blocks import *
 from enemies import *
 
-pygame.init()
-clock = pygame.time.Clock()
+@profile
+def main():
 
-screen = pygame.display.set_mode((screen_width,screen_height))
-pygame.display.set_caption('Arctic Masher')
+    pygame.init()
+    clock = pygame.time.Clock()
 
-players["1"] = (Player(
-    "1",
-    5,
-    5,
-    red,
-    [
-        (pygame.K_w, "n"),
-        (pygame.K_e, "ne"),
-        (pygame.K_d, "e"),
-        (pygame.K_c, "se"),
-        (pygame.K_x, "s"),
-        (pygame.K_z, "sw"),
-        (pygame.K_a, "w"),
-        (pygame.K_q, "nw"),
-        (pygame.K_s, "p"),
-    ],
-))
-players["2"] = (Player(
-    "2",
-    3,
-    3,
-    green,
-    [
-        (pygame.K_KP_8, "n"),
-        (pygame.K_KP_9, "ne"),
-        (pygame.K_KP_6, "e"),
-        (pygame.K_KP_3, "se"),
-        (pygame.K_KP_2, "s"),
-        (pygame.K_KP_1, "sw"),
-        (pygame.K_KP_4, "w"),
-        (pygame.K_KP_7, "nw"),
-        (pygame.K_KP_5, "p"),
-    ],
-))
-players["3"] = (Player(
-    "3",
-    5,
-    5,
-    red,
-    [
-        (pygame.K_y, "n"),
-        (pygame.K_u, "ne"),
-        (pygame.K_j, "e"),
-        (pygame.K_m, "se"),
-        (pygame.K_n, "s"),
-        (pygame.K_b, "sw"),
-        (pygame.K_g, "w"),
-        (pygame.K_t, "nw"),
-        (pygame.K_h, "p"),
-    ],
-))
+    screen = pygame.display.set_mode((screen_width,screen_height))
+    pygame.display.set_caption('Arctic Masher')
 
-mapgen.generateMap()
+    players["1"] = (Player(
+        "1",
+        5,
+        5,
+        red,
+        [
+            (pygame.K_w, "n"),
+            (pygame.K_e, "ne"),
+            (pygame.K_d, "e"),
+            (pygame.K_c, "se"),
+            (pygame.K_x, "s"),
+            (pygame.K_z, "sw"),
+            (pygame.K_a, "w"),
+            (pygame.K_q, "nw"),
+            (pygame.K_s, "p"),
+        ],
+    ))
+    players["2"] = (Player(
+        "2",
+        3,
+        3,
+        green,
+        [
+            (pygame.K_KP_8, "n"),
+            (pygame.K_KP_9, "ne"),
+            (pygame.K_KP_6, "e"),
+            (pygame.K_KP_3, "se"),
+            (pygame.K_KP_2, "s"),
+            (pygame.K_KP_1, "sw"),
+            (pygame.K_KP_4, "w"),
+            (pygame.K_KP_7, "nw"),
+            (pygame.K_KP_5, "p"),
+        ],
+    ))
+    players["3"] = (Player(
+        "3",
+        5,
+        5,
+        red,
+        [
+            (pygame.K_y, "n"),
+            (pygame.K_u, "ne"),
+            (pygame.K_j, "e"),
+            (pygame.K_m, "se"),
+            (pygame.K_n, "s"),
+            (pygame.K_b, "sw"),
+            (pygame.K_g, "w"),
+            (pygame.K_t, "nw"),
+            (pygame.K_h, "p"),
+        ],
+    ))
 
-sprite_sheet_image = pygame.image.load('assets/penguin.png').convert_alpha()
+    mapgen.generateMap()
 
-def get_image(sheet, x, y, width, height, scale, colour):
-    image = pygame.Surface((width, height)).convert_alpha()
-    image.blit(sheet, (0, 0), (x, y, width, height))
-    image = pygame.transform.scale(image, (width * scale, height * scale))
-    image.set_colorkey(colour)
+    sprite_sheet_image = pygame.image.load('assets/penguin.png').convert_alpha()
 
-    return image
-frame_0 = get_image(sprite_sheet_image, 0, 0, 32, 32, (tile-2)/32, black)
-frame_4 = get_image(sprite_sheet_image, 32, 0, 32, 32, (tile-2)/32, black)
+    def get_image(sheet, x, y, width, height, scale, colour):
+        image = pygame.Surface((width, height)).convert_alpha()
+        image.blit(sheet, (0, 0), (x, y, width, height))
+        image = pygame.transform.scale(image, (width * scale, height * scale))
+        image.set_colorkey(colour)
 
-game_font = pygame.font.Font("freesansbold.ttf",32)
+        return image
+    frame_0 = get_image(sprite_sheet_image, 0, 0, 32, 32, (tile-2)/32, black)
+    frame_4 = get_image(sprite_sheet_image, 32, 0, 32, 32, (tile-2)/32, black)
 
-while True:
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT:
-            pygame.quit()
-            sys.exit()
+    game_font = pygame.font.Font("freesansbold.ttf",32)
 
-    keys = pygame.key.get_pressed()
+    while True:
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                pygame.quit()
+                sys.exit()
 
-    for id, player in players.items():
-        player.useKeys(keys)
-    
-    for id, enemy in enemies.items():
-        enemy.runAI()
+        keys = pygame.key.get_pressed()
 
-    # Rendering
-    screen.fill(bg_color)
-    for column in gameMap:
-        for spot in column:
-            if not spot:
-                continue
-            elif isinstance(spot, Block):
-                pygame.draw.rect(screen, blue, spot.sprite)
-            elif isinstance(spot, Enemy):
-                screen.blit(frame_4, ((tile*spot.x)+2, (tile*spot.y)+2))
-            elif isinstance(spot, Player):
-                screen.blit(frame_0, ((tile*spot.x)+2, (tile*spot.y)+2))
+        for id, player in players.items():
+            player.useKeys(keys)
+        
+        for id, enemy in enemies.items():
+            enemy.runAI()
 
-    pygame.display.flip()
-    clock.tick(60)
+        # Rendering
+        screen.fill(bg_color)
+        for column in gameMap:
+            for spot in column:
+                if not spot:
+                    continue
+                elif isinstance(spot, Block):
+                    pygame.draw.rect(screen, blue, spot.sprite)
+                elif isinstance(spot, Enemy):
+                    screen.blit(frame_4, ((tile*spot.x)+2, (tile*spot.y)+2))
+                elif isinstance(spot, Player):
+                    screen.blit(frame_0, ((tile*spot.x)+2, (tile*spot.y)+2))
+
+        pygame.display.flip()
+        clock.tick(60)
+
+if __name__ == '__main__':
+    main()