helpers/key.py

   1 from .rounded_rect import *
   2 from .action import *
   3 import time
   4 import sys
   5 import pygame
   6
   7 class Key:
   8     row_positions = {
   9         'first':    0,
  10         'second':  50,
  11         'third':  100,
  12         'fourth': 150,
  13         'fifth':  200,
  14         'sixth':  250,
  15     }
  16
  17     default_outer_color = (120, 120, 120)
  18     lighter_outer_color = (200, 200, 200)
  19     default_inner_color = (255, 255, 255)
  20     mapped_inner_color  = (  0, 255,   0)
  21     mapped_unready_inner_color = (255, 165, 0)
  22
  23     def __init__(self, mapping, draw_lock, key_name, key_sym, top, left, width = 48, height = 48, disabled = False):
  24         self.draw_lock = draw_lock
  25         self.mapping = mapping
  26         self.key_name = key_name
  27         self.key_sym  = key_sym
  28
  29         if isinstance(top, str):
  30             self.top = self.row_positions[top]
  31         else:
  32             self.top = top
  33
  34         self.left   = left
  35         self.width  = width
  36         self.height = height
  37
  38         self.bottom = self.top  + self.height
  39         self.right  = self.left + self.width
  40
  41         self.rect     = (self.left, self.top, self.right, self.bottom)
  42         self.position = (self.left, self.top)
  43
  44         if disabled:
  45             self.outer_color = self.lighter_outer_color
  46             self.linewidth = 1
  47         else:
  48             self.outer_color = self.default_outer_color
  49             self.linewidth = 3
  50
  51         self.inner_color = self.default_inner_color
  52         self.actions = []
  53
  54     def square(self, all_actions_ready):
  55         if self.has_actions():
  56             if all_actions_ready:
  57                 self.inner_color = self.mapped_inner_color
  58             else:
  59                 self.inner_color = self.mapped_unready_inner_color
  60
  61         return RoundedRect((0, 0, self.width, self.height),
  62             self.outer_color, self.inner_color, self.linewidth)
  63
  64     def collidepoint(self, position):
  65         return self.surface.get_rect().collidepoint(
  66                 position[0] - self.position[0],
  67                 position[1] - self.position[1]
  68                 )
  69
  70     def draw(self, background_surface):
  71         self.draw_lock.acquire()
  72         all_actions_ready = self.all_actions_ready()
  73
  74         self.surface = self.square(all_actions_ready).surface()
  75
  76         if getattr(sys, 'frozen', False):
  77             police = pygame.font.Font(sys._MEIPASS + "/Ubuntu-Regular.ttf", 14)
  78         else:
  79             police = pygame.font.Font("Ubuntu-Regular.ttf", 14)
  80
  81         text = police.render(self.key_sym, True, (0,0,0))
  82         self.surface.blit(text, (5,5))
  83         background_surface.blit(self.surface, self.position)
  84         self.draw_lock.release()
  85
  86         return not all_actions_ready
  87
  88     def poll_redraw(self, background):
  89         while True:
  90             time.sleep(1)
  91             if self.all_actions_ready():
  92                 self.draw(background)
  93                 self.mapping.blit()
  94                 break
  95
  96     def has_actions(self):
  97         return len(self.actions) > 0
  98
  99     def all_actions_ready(self):
 100         return all(action.ready() for action in self.actions)
 101
 102     def add_action(self, action_name, **arguments):
 103         self.actions.append(Action(action_name, self, **arguments))
 104
 105     def do_actions(self):
 106         print("running actions for {}".format(self.key_sym))
 107         start_time = time.time()
 108         self.mapping.start_running(self, start_time)
 109         for action in self.actions:
 110             if self.mapping.keep_running(self, start_time):
 111                 action.run()
 112
 113         self.mapping.finished_running(self, start_time)
 114
 115     def list_actions(self, screen):
 116         action_descriptions = [action.description() for action in self.actions]
 117         print("actions linked to key {}:".format(self.key_sym))
 118         print("\t" + "\n\t".join(action_descriptions))
 119         self.draw_lock.acquire()
 120         surface = pygame.Surface((800, 250)).convert()
 121         surface.fill((250, 250, 250))
 122         if getattr(sys, 'frozen', False):
 123             police = pygame.font.Font(sys._MEIPASS + "/Ubuntu-Regular.ttf", 14)
 124         else:
 125             police = pygame.font.Font("Ubuntu-Regular.ttf", 14)
 126
 127         offset = 0
 128         police.set_bold(True)
 129         text = police.render("actions linked to key {}:".format(self.key_sym), True, (0,0,0))
 130         surface.blit(text, (0, offset))
 131         offset += police.get_linesize()
 132
 133         police.set_bold(False)
 134         for description in action_descriptions:
 135             text = police.render(description, True, (0,0,0))
 136             surface.blit(text, (0, offset))
 137             offset += police.get_linesize()
 138
 139         screen.blit(surface, (10, 330))
 140         pygame.display.flip()
 141         self.draw_lock.release()
 142
 143