150 lines
5.8 KiB
C
150 lines
5.8 KiB
C
/* Copyright 2021 Yang Hu
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "unicorne.h"
|
|
#include "i2c_master.h"
|
|
|
|
// Custom i2c init to enable internal pull up resistor for i2c.
|
|
void i2c_init(void) {
|
|
static bool is_initialised = false;
|
|
if (!is_initialised) {
|
|
is_initialised = true;
|
|
|
|
// Try releasing special pins for a short time
|
|
palSetLineMode(I2C1_SCL_PIN, PAL_MODE_INPUT);
|
|
palSetLineMode(I2C1_SDA_PIN, PAL_MODE_INPUT);
|
|
|
|
chThdSleepMilliseconds(10);
|
|
// Use internal pull up since we do not have pull up on i2c pins in v1 design.
|
|
palSetLineMode(I2C1_SCL_PIN, PAL_MODE_ALTERNATE(I2C1_SCL_PAL_MODE) | PAL_OUTPUT_TYPE_OPENDRAIN | PAL_STM32_PUPDR_PULLUP);
|
|
palSetLineMode(I2C1_SDA_PIN, PAL_MODE_ALTERNATE(I2C1_SDA_PAL_MODE) | PAL_OUTPUT_TYPE_OPENDRAIN | PAL_STM32_PUPDR_PULLUP);
|
|
}
|
|
}
|
|
|
|
// LED matrix
|
|
// physical location
|
|
// 2 3 4 5
|
|
//
|
|
// 1 6
|
|
// 0 7
|
|
#ifdef RGB_MATRIX_ENABLE
|
|
// clang-format off
|
|
led_config_t g_led_config = {{
|
|
// Key Matrix to LED Index
|
|
// Since we only have 8 LEDs, map the keys near them to the same LED.
|
|
{2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5},
|
|
{2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5},
|
|
{1, 1, 1, 0, 0, 0, 7, 7, 7, 6, 6, 6},
|
|
{NO_LED, NO_LED, NO_LED, 0, 0, 0, 7, 7, 7, NO_LED, NO_LED, NO_LED},
|
|
}, {// LED Index to Physical Position
|
|
{94, 60},
|
|
{18, 44},
|
|
{8, 10},
|
|
{94, 10},
|
|
{130,10},
|
|
{216, 10},
|
|
{208, 44},
|
|
{130, 60}
|
|
}, {// LED Index to Flag
|
|
LED_FLAG_ALL, LED_FLAG_ALL, LED_FLAG_ALL, LED_FLAG_ALL, LED_FLAG_ALL,
|
|
LED_FLAG_ALL, LED_FLAG_ALL, LED_FLAG_ALL
|
|
}};
|
|
// clang-format on
|
|
#endif
|
|
|
|
#ifdef OLED_ENABLE
|
|
// OLED shared code
|
|
// The oled is vertical. Need to rotate 270 degrees.
|
|
__attribute__((weak)) oled_rotation_t oled_init_user(oled_rotation_t rotation) { return OLED_ROTATION_270; }
|
|
|
|
// Render layer status on OLED. To be re-implemented by keymaps since layers are
|
|
// defined there.
|
|
__attribute__((weak)) void oled_render_layer(void) { return; }
|
|
|
|
// Oneshot mods status
|
|
uint8_t osmods;
|
|
|
|
__attribute__((weak)) void oled_render_mods(void) {
|
|
static const char PROGMEM ctrl[] = {0xb6, 0xb7, 10, 0xd6, 0xd7, 10, 0x20, 0x20, 10, 0};
|
|
static const char PROGMEM shift[] = {0x87, 0x88, 0x89, 10, 0xa7, 0xa8, 0xa9, 10, 0xc7, 0xc8, 0xc9, 10, 0};
|
|
static const char PROGMEM alt[] = {0x84, 0x85, 0x86, 10, 0xa4, 0xa5, 0xa6, 10, 0xc4, 0xc5, 0xc6, 10, 0};
|
|
static const char PROGMEM ctrl_alt[] = {0xb6, 0xb7, 0x84, 0x85, 0x86, 0xd6, 0xd7, 0xa4, 0xa5, 0xa6, 0x20, 0x20, 0xc4, 0xc5, 0xc6, 0};
|
|
|
|
static const char PROGMEM ctrl_shift[] = {0xb6, 0xb7, 0x87, 0x88, 0x89, 0xd6, 0xd7, 0xa7, 0xa8, 0xa9, 0x20, 0x20, 0xc7, 0xc8, 0xc9, 0};
|
|
static const char PROGMEM c_a_shift[] = {0xb6, 0xb7, 0x84, 0x85, 0x86, 0xd6, 0xd7, 0xa4, 0xa5, 0xa6, 0x20, 0x20, 0xc4, 0xc5, 0xc6, 0x87, 0x88, 0x89, 10, 0xa7, 0xa8, 0xa9, 10, 0xc7, 0xc8, 0xc9, 10, 0};
|
|
// Now check mod status and render.
|
|
static uint8_t mods;
|
|
mods = get_mods() | osmods;
|
|
if ((mods & MOD_MASK_CTRL) && (mods & MOD_MASK_ALT) && (mods & MOD_MASK_SHIFT)) {
|
|
oled_write_P(c_a_shift, false);
|
|
} else if ((mods & MOD_MASK_CTRL) && (mods & MOD_MASK_ALT)) {
|
|
oled_write_P(ctrl_alt, false);
|
|
} else if ((mods & MOD_MASK_CTRL) && (mods & MOD_MASK_SHIFT)) {
|
|
oled_write_P(ctrl_shift, false);
|
|
} else if ((mods & MOD_MASK_SHIFT) && (mods & MOD_MASK_ALT)) {
|
|
oled_write_P(alt, false);
|
|
oled_write_P(shift, false);
|
|
} else if (mods & MOD_MASK_CTRL) {
|
|
oled_write_P(ctrl, false);
|
|
} else if (mods & MOD_MASK_ALT) {
|
|
oled_write_P(alt, false);
|
|
} else if (mods & MOD_MASK_SHIFT) {
|
|
oled_write_P(shift, false);
|
|
} else {
|
|
for (int i = 0; i < 6; ++i) {
|
|
oled_write_ln(" ", false);
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
void oneshot_mods_changed_user(uint8_t mods) { osmods = mods; }
|
|
|
|
// Call this from "led_update_user" and use `led_state.caps_lock` to check
|
|
// the status of capslock.
|
|
__attribute__((weak)) void oled_render_capslock(bool caps_on) {
|
|
static const char PROGMEM capslock_logo[] = {0x9c, 0x9d, 0x9e, 0x9f, 10, 0xbc, 0xbd, 0xbe, 0xbf, 10, 0xdc, 0xdd, 0xde, 0xdf, 10, 0};
|
|
if (caps_on) {
|
|
oled_write_P(capslock_logo, false);
|
|
} else {
|
|
for (int i = 0; i < 3; ++i) {
|
|
oled_write_ln(" ", false);
|
|
}
|
|
}
|
|
}
|
|
|
|
char keylog_str[24] = {};
|
|
|
|
const char code_to_name[60] = {' ', ' ', ' ', ' ', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', 'R', 'E', 'B', 'T', '_', '-', '=', '[', ']', '\\', '#', ';', '\'', '`', ',', '.', '/', ' ', ' ', ' '};
|
|
|
|
void set_keylog(uint16_t keycode, keyrecord_t *record) {
|
|
if ((keycode >= QK_MOD_TAP && keycode <= QK_MOD_TAP_MAX) || (keycode >= QK_LAYER_TAP && keycode <= QK_LAYER_TAP_MAX)) {
|
|
keycode = keycode & 0xFF;
|
|
}
|
|
}
|
|
|
|
__attribute__((weak)) void oled_render_keylog(void) { oled_write(keylog_str, false); }
|
|
|
|
// Keymaps can override this function
|
|
__attribute__((weak)) bool oled_task_kb(void) {
|
|
if (!oled_task_user()) { return false; }
|
|
/* oled_render_keylog(); */
|
|
oled_render_layer();
|
|
oled_render_mods();
|
|
led_t led_state = host_keyboard_led_state();
|
|
oled_render_capslock(led_state.caps_lock);
|
|
return true;
|
|
}
|
|
#endif
|