start of Menu implementation

game/util/control.py

@@ -0,0 +1,83 @@
+import pygame, sys
+from util import write_save
+
+class Controls_Handler():
+    def __init__(self, save):
+        self.save_file = save
+        self.curr_block = save["current_profile"]
+        self.controls = self.save_file["controls"][str(self.curr_block)]
+        self.setup()
+
+    def update(self, actions):
+        if self.selected: self.set_new_control()
+        else: self.navigate_menu(actions)
+
+    def render(self, surface):
+        self.draw_text(surface, "Control Profile " + str(self.curr_block+1) , 20, pygame.Color((0,0,0)), 480 / 2, 270/8)
+        self.display_controls(surface, self.save_file["controls"][str(self.curr_block)])
+        if self.curr_block == self.save_file["current_profile"]: self.draw_text(surface, "*"  , 20, pygame.Color((0,0,0)), 20, 20)
+
+    def navigate_menu(self, actions):
+        # Move the cursor up and down
+        if actions["Down"]: self.curr_index = (self.curr_index + 1) % (len(self.save_file["controls"][str(self.curr_block)]) + 1)
+        if actions["Up"]: self.curr_index = (self.curr_index - 1) % (len(self.save_file["controls"][str(self.curr_block)]) + 1)
+        # Switch between profiles
+        if actions["Left"]: self.curr_block = (self.curr_block -1) % len(self.save_file["controls"])
+        if actions["Right"]: self.curr_block = (self.curr_block +1) % len(self.save_file["controls"])
+        # Handle Selection
+        if actions["Action1"] or actions["Start"]:
+            # Set the current profile to be the main one
+            if self.cursor_dict[self.curr_index] == "Set":
+                self.controls = self.save_file["controls"][str(self.curr_block)]
+                self.save_file["current_profile"] = self.curr_block
+                write_save(self.save_file)
+            else:
+                self.selected = True
+
+    def set_new_control(self):
+        selected_control = self.cursor_dict[self.curr_index]
+        done = False
+        while not done:
+            for event in pygame.event.get():
+                    if event.type == pygame.QUIT:
+                        done = True
+                        pygame.quit()
+                        sys.exit()
+                    elif event.type == pygame.KEYDOWN:
+                        if event.key == pygame.K_ESCAPE:
+                            done = True
+                            pygame.quit()
+                            sys.exit()
+                        elif event.key not in self.save_file["controls"][str(self.curr_block)].values():
+                            self.save_file["controls"][str(self.curr_block)][selected_control] = event.key
+                            write_save(self.save_file)
+                            self.selected = False
+                            done = True
+
+    def display_controls(self,surface, controls):
+        color = (255,13,5) if self.selected else (255,250,239)
+        pygame.draw.rect(surface, color, (80 , 270/4 - 10 + (self.curr_index*30), 320, 20) )
+        i = 0
+        for control in controls:
+            self.draw_text(surface, control + ' - ' + pygame.key.name(controls[control]),20,
+                            pygame.Color((0,0,0)), 480 / 2, 270/4 + i)
+            i += 30
+        self.draw_text(surface, "Set Current Profile",20, pygame.Color((0,0,0)), 480 / 2, 270/4 + i)
+
+    def setup(self):
+        self.selected = False
+        self.font = pygame.font.Font("RetroFont.ttf", 20)
+        self.cursor_dict = {}
+        self.curr_index = 0
+        i = 0
+        for control in self.controls:
+            self.cursor_dict[i] = control
+            i += 1
+        self.cursor_dict[i] = "Set"
+
+    def draw_text(self,surface, text, size, color, x, y):
+        text_surface = self.font.render(text, True, color, size)
+        text_surface.set_colorkey((0,0,0))
+        text_rect = text_surface.get_rect()
+        text_rect.center = (x, y)
+        surface.blit(text_surface, text_rect)