IM23a-bachmannj2
82 lines
3.7 KiB
Python
82 lines
3.7 KiB
Python
import pygame, sys
|
|
from game.util.util import write_save, get_font
|
|
|
|
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((255, 255, 255)), 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((255, 255, 255)), 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 pygame.key.get_pressed()[pygame.K_RETURN]:
|
|
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 = (173, 2, 2) if self.selected else (207, 190, 41)
|
|
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((255,255,255)), 480 / 2, 270/4 + i)
|
|
i += 30
|
|
self.draw_text(surface, "Set Current Profile",20, pygame.Color((255, 255, 255)), 480 / 2, 270/4 + i)
|
|
|
|
def setup(self):
|
|
self.selected = False
|
|
self.font = get_font(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) |