aboutsummaryrefslogtreecommitdiff
path: root/drivers/led/issi/is31fl3236-mono.c
blob: b80245a1dc0828859a236e7708266bcf208bc3c3 (plain)
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
// Copyright 2024 QMK
// SPDX-License-Identifier: GPL-2.0-or-later

#include "is31fl3236-mono.h"
#include "i2c_master.h"
#include "gpio.h"

#define IS31FL3236_PWM_REGISTER_COUNT 36
#define IS31FL3236_LED_CONTROL_REGISTER_COUNT 36

#ifndef IS31FL3236_I2C_TIMEOUT
#    define IS31FL3236_I2C_TIMEOUT 100
#endif

#ifndef IS31FL3236_I2C_PERSISTENCE
#    define IS31FL3236_I2C_PERSISTENCE 0
#endif

#ifndef IS31FL3236_PWM_FREQUENCY
#    define IS31FL3236_PWM_FREQUENCY IS31FL3236_PWM_FREQUENCY_3K_HZ // OFS - IS31FL3236A only
#endif

const uint8_t i2c_addresses[IS31FL3236_DRIVER_COUNT] = {
    IS31FL3236_I2C_ADDRESS_1,
#ifdef IS31FL3236_I2C_ADDRESS_2
    IS31FL3236_I2C_ADDRESS_2,
#    ifdef IS31FL3236_I2C_ADDRESS_3
    IS31FL3236_I2C_ADDRESS_3,
#        ifdef IS31FL3236_I2C_ADDRESS_4
    IS31FL3236_I2C_ADDRESS_4,
#        endif
#    endif
#endif
};

typedef struct is31fl3236_driver_t {
    uint8_t pwm_buffer[IS31FL3236_PWM_REGISTER_COUNT];
    bool    pwm_buffer_dirty;
    uint8_t led_control_buffer[IS31FL3236_LED_CONTROL_REGISTER_COUNT];
    bool    led_control_buffer_dirty;
} PACKED is31fl3236_driver_t;

is31fl3236_driver_t driver_buffers[IS31FL3236_DRIVER_COUNT] = {{
    .pwm_buffer               = {0},
    .pwm_buffer_dirty         = false,
    .led_control_buffer       = {0},
    .led_control_buffer_dirty = false,
}};

void is31fl3236_write_register(uint8_t index, uint8_t reg, uint8_t data) {
#if IS31FL3236_I2C_PERSISTENCE > 0
    for (uint8_t i = 0; i < IS31FL3236_I2C_PERSISTENCE; i++) {
        if (i2c_write_register(i2c_addresses[index] << 1, reg, &data, 1, IS31FL3236_I2C_TIMEOUT) == I2C_STATUS_SUCCESS) break;
    }
#else
    i2c_write_register(i2c_addresses[index] << 1, reg, &data, 1, IS31FL3236_I2C_TIMEOUT);
#endif
}

void is31fl3236_write_pwm_buffer(uint8_t index) {
#if IS31FL3236_I2C_PERSISTENCE > 0
    for (uint8_t i = 0; i < IS31FL3236_I2C_PERSISTENCE; i++) {
        if (i2c_write_register(i2c_addresses[index] << 1, IS31FL3236_REG_PWM, driver_buffers[index].pwm_buffer, 36, IS31FL3236_I2C_TIMEOUT) == I2C_STATUS_SUCCESS) break;
    }
#else
    i2c_write_register(i2c_addresses[index] << 1, IS31FL3236_REG_PWM, driver_buffers[index].pwm_buffer, 36, IS31FL3236_I2C_TIMEOUT);
#endif
}

void is31fl3236_init_drivers(void) {
    i2c_init();

#if defined(IS31FL3236_SDB_PIN)
    gpio_set_pin_output(IS31FL3236_SDB_PIN);
    gpio_write_pin_high(IS31FL3236_SDB_PIN);
#endif

    for (uint8_t i = 0; i < IS31FL3236_DRIVER_COUNT; i++) {
        is31fl3236_init(i);
    }

    for (uint8_t i = 0; i < IS31FL3236_LED_COUNT; i++) {
        is31fl3236_set_led_control_register(i, true);
    }

    for (uint8_t i = 0; i < IS31FL3236_DRIVER_COUNT; i++) {
        is31fl3236_update_led_control_registers(i);
    }
}

void is31fl3236_init(uint8_t index) {
    // In case we ever want to reinitialize (?)
    is31fl3236_write_register(index, IS31FL3236_REG_RESET, 0x00);

    // Turn off software shutdown
    is31fl3236_write_register(index, IS31FL3236_REG_SHUTDOWN, 0x01);

    // Set all PWM values to zero
    for (uint8_t i = 0; i < IS31FL3236_PWM_REGISTER_COUNT; i++) {
        is31fl3236_write_register(index, IS31FL3236_REG_PWM + i, 0x00);
    }

    // turn off all LEDs in the LED control register
    for (uint8_t i = 0; i < IS31FL3236_LED_CONTROL_REGISTER_COUNT; i++) {
        is31fl3236_write_register(index, IS31FL3236_REG_LED_CONTROL + i, 0x00);
    }

    // Set PWM frequency (IS31FL3236A)
    is31fl3236_write_register(index, IS31FL3236_REG_PWM_FREQUENCY, IS31FL3236_PWM_FREQUENCY);

    // Load PWM registers and LED Control register data
    is31fl3236_write_register(index, IS31FL3236_REG_UPDATE, 0x01);
}

void is31fl3236_set_value(int index, uint8_t value) {
    is31fl3236_led_t led;

    if (index < IS31FL3236_LED_COUNT) {
        memcpy_P(&led, (&g_is31fl3236_leds[index]), sizeof(led));

        if (driver_buffers[led.driver].pwm_buffer[led.v] == value) {
            return;
        }

        driver_buffers[led.driver].pwm_buffer[led.v] = value;
        driver_buffers[led.driver].pwm_buffer_dirty  = true;
    }
}

void is31fl3236_set_value_all(uint8_t value) {
    for (uint8_t i = 0; i < IS31FL3236_LED_COUNT; i++) {
        is31fl3236_set_value(i, value);
    }
}

void is31fl3236_set_led_control_register(uint8_t index, bool value) {
    is31fl3236_led_t led;
    memcpy_P(&led, (&g_is31fl3236_leds[index]), sizeof(led));

    driver_buffers[led.driver].led_control_buffer[led.v] = value ? 0x01 : 0x00;
    driver_buffers[led.driver].led_control_buffer_dirty  = true;
}

void is31fl3236_update_pwm_buffers(uint8_t index) {
    if (driver_buffers[index].pwm_buffer_dirty) {
        is31fl3236_write_pwm_buffer(index);
        // Load PWM registers and LED Control register data
        is31fl3236_write_register(index, IS31FL3236_REG_UPDATE, 0x01);

        driver_buffers[index].pwm_buffer_dirty = false;
    }
}

void is31fl3236_update_led_control_registers(uint8_t index) {
    if (driver_buffers[index].led_control_buffer_dirty) {
        for (uint8_t i = 0; i < IS31FL3236_LED_CONTROL_REGISTER_COUNT; i++) {
            is31fl3236_write_register(index, IS31FL3236_REG_LED_CONTROL + i, driver_buffers[index].led_control_buffer[i]);
        }

        driver_buffers[index].led_control_buffer_dirty = false;
    }
}

void is31fl3236_flush(void) {
    for (uint8_t i = 0; i < IS31FL3236_DRIVER_COUNT; i++) {
        is31fl3236_update_pwm_buffers(i);
    }
}