kass/color_handler.py at master · samuelWilliams99/kass · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import menu
import screen
import math
import settings
import json

_handler_save_path = "/home/pi/python/handler_data.txt"
pedals = [0, 0, 0]

class HandlerSelector:
    name = "COLOR TYPE"
    handlers = []
    force_next_setting = False
    handler_data = {"index": 0}

    @staticmethod
    def get_global_hue():
        return HandlerSelector.get_handler().global_hue

    @staticmethod
    def get_select_text():
        return HandlerSelector.get_handler().name

    @staticmethod
    def add(handler):
        handler.sub_settings = ColorHandlerSettingHandler(handler.name, "")
        handler.add_setting = lambda setting: handler.sub_settings.add(setting)
        handler.get_value = lambda name, **kwargs: handler.sub_settings.get_value(name, **kwargs)
        if not hasattr(handler, "global_hue"):
            handler.global_hue = False
        HandlerSelector.handlers.append(handler)

    @staticmethod
    def on_button_pressed():
        menu.leave_sub_menu()

    @staticmethod
    def on_button_long_pressed():
        handler = HandlerSelector.get_handler()
        if len(handler.sub_settings.setting_names) > 0:
            menu.set_temp_sub_menu(handler.sub_settings)

    @staticmethod
    def on_enter():
        HandlerSelector.force_next_setting = True

    @staticmethod
    def on_value_change(v):
        handler_count = len(HandlerSelector.handlers)
        new_index = min(math.floor(v * handler_count), handler_count - 1)
        if HandlerSelector.handler_index() != new_index or HandlerSelector.force_next_setting:
            HandlerSelector.force_next_setting = False
            HandlerSelector.set_handler_index(new_index)

    @staticmethod
    def handler_index():
        return HandlerSelector.handler_data["index"]

    @staticmethod
    def set_handler_index(i):
        HandlerSelector.handler_data["index"] = i
        HandlerSelector.update_display()
        HandlerSelector.save()

    @staticmethod
    def update_display():
        screen.line1("[" + HandlerSelector.name + "]", True)
        screen.line2(HandlerSelector.get_handler().name, True)

    @staticmethod
    def get_handler():
        return HandlerSelector.handlers[HandlerSelector.handler_index()]

    @staticmethod
    def get_hsv(note, vel):
        return HandlerSelector.get_handler().get_hsv(note, vel)

    @staticmethod
    def loop():
        return HandlerSelector.get_handler().loop()

    @staticmethod
    def save():
        with open(_handler_save_path, 'w') as f:
            json.dump(HandlerSelector.handler_data, f)

    @staticmethod
    def save_settings(name, data):
        HandlerSelector.handler_data[name] = data
        HandlerSelector.save()

try:
    with open(_handler_save_path, 'r') as f:
        HandlerSelector.handler_data = json.load(f)
except:
    pass

class ColorHandlerSettingHandler(settings.SettingHandler):
    def save_json(self, data):
        HandlerSelector.save_settings(self.name, data)

    def load_json(self):
        return HandlerSelector.handler_data.get(self.name, dict())


def add(handler):
    HandlerSelector.add(handler)

def loop():
    HandlerSelector.loop()

def get_global_hue():
    return HandlerSelector.get_global_hue()

def get_hsv(note_index, vel):
    return HandlerSelector.get_hsv(note_index, vel)

def set_pedal(pedal_idx, pedal):
    pedals[pedal_idx] = pedal

import color_handlers