""" Original code from https://github.com/skullydazed/kle2xy """ import hjson from decimal import Decimal class KLE2xy(list): """Abstract interface for interacting with a KLE layout. """ def __init__(self, layout=None, name='', invert_y=True): super(KLE2xy, self).__init__() self.name = name self.invert_y = invert_y self.key_width = Decimal('19.05') self.key_skel = {'decal': False, 'border_color': 'none', 'keycap_profile': '', 'keycap_color': 'grey', 'label_color': 'black', 'label_size': 3, 'label_style': 4, 'width': Decimal('1'), 'height': Decimal('1')} self.rows = Decimal(0) self.columns = Decimal(0) if layout: self.parse_layout(layout) @property def width(self): """Returns the width of the keyboard plate. """ return (Decimal(self.columns) * self.key_width) + self.key_width / 2 @property def height(self): """Returns the height of the keyboard plate. """ return (self.rows * self.key_width) + self.key_width / 2 @property def size(self): """Returns the size of the keyboard plate. """ return (self.width, self.height) def attrs(self, properties): """Parse the keyboard properties dictionary. """ # FIXME: Store more than just the keyboard name. if 'name' in properties: self.name = properties['name'] def parse_layout(self, layout): # noqa FIXME(skullydazed): flake8 says this has a complexity of 25, it should be refactored. # Wrap this in a dictionary so hjson will parse KLE raw data layout = '{"layout": [' + layout + ']}' layout = hjson.loads(layout)['layout'] # Initialize our state machine current_key = self.key_skel.copy() current_row = Decimal(0) current_col = Decimal(0) if isinstance(layout[0], dict): self.attrs(layout[0]) layout = layout[1:] for row_num, row in enumerate(layout): self.append([]) # Process the current row for key in row: if isinstance(key, dict): if 'w' in key and key['w'] != Decimal(1): current_key['width'] = Decimal(key['w']) if 'w2' in key and 'h2' in key and key['w2'] == 1.5 and key['h2'] == 1: # FIXME: ISO Key uses these params: {x:0.25,w:1.25,h:2,w2:1.5,h2:1,x2:-0.25} current_key['isoenter'] = True if 'h' in key and key['h'] != Decimal(1): current_key['height'] = Decimal(key['h']) if 'a' in key: current_key['label_style'] = self.key_skel['label_style'] = max(min(int(key['a']), 9), 0) if 'f' in key: current_key['label_size'] = self.key_skel['label_size'] = max(min(int(key['f']), 9), 1) if 'p' in key: current_key['keycap_profile'] = self.key_skel['keycap_profile'] = key['p'] if 'c' in key: current_key['keycap_color'] = self.key_skel['keycap_color'] = key['c'] if 't' in key: # FIXME: Need to do better validation, plus figure out how to support multiple colors if '\n' in key['t']: key['t'] = key['t'].split('\n')[0] if key['t'] == "0": key['t'] = "#000000" current_key['label_color'] = self.key_skel['label_color'] = key['t'] if 'x' in key: current_col += Decimal(key['x']) if 'y' in key: current_row += Decimal(key['y']) if 'd' in key: current_key['decal'] = True else: current_key['name'] = key current_key['row'] = round(current_row, 2) current_key['column'] = round(current_col, 2) # x,y (units mm) is the center of the key x_center = current_col + current_key['width'] / 2 y_center = current_row + current_key['height'] / 2 current_key['x'] = x_center * self.key_width current_key['y'] = y_center * self.key_width # Tend to our row/col count current_col += current_key['width'] if current_col > self.columns: self.columns = current_col # Invert the y-axis if neccesary if self.invert_y: current_key['y'] = -current_key['y'] # Store this key self[-1].append(current_key) current_key = self.key_skel.copy() # Move to the next row current_col = Decimal(0) current_row += Decimal(1) if current_row > self.rows: self.rows = Decimal(current_row)