vial/src/main/python/keymap_editor.py

# SPDX-License-Identifier: GPL-2.0-or-later

from basic_editor import BasicEditor
from keyboard_container import KeyboardContainer
from keycodes import recreate_keyboard_keycodes
from tabbed_keycodes import TabbedKeycodes
from vial_device import VialKeyboard


class KeymapEditor(BasicEditor):

    def __init__(self, layout_editor):
        super().__init__()

        self.keyboard_container = KeyboardContainer(layout_editor)

        self.tabbed_keycodes = TabbedKeycodes()
        self.tabbed_keycodes.keycode_changed.connect(self.on_keycode_changed)

        self.addWidget(self.keyboard_container)
        self.addWidget(self.tabbed_keycodes)

        self.device = None

    def on_keycode_changed(self, code):
        self.keyboard_container.set_key(code)

    def rebuild(self, device):
        super().rebuild(device)
        if self.valid():
            self.keyboard_container.rebuild(device.keyboard)
            recreate_keyboard_keycodes(self.keyboard_container.keyboard)
            self.tabbed_keycodes.recreate_keycode_buttons()
            self.keyboard_container.refresh_layer_display()

    def valid(self):
        return isinstance(self.device, VialKeyboard)

    def save_layout(self):
        return self.keyboard_container.save_layout()

    def restore_layout(self, data):
        self.keyboard_container.restore_layout(data)