qmk_firmware/keyboards/contra/keymaps/enigma/keymap.c

626 lines
23 KiB
C

/* Copyright 2022 Christopher Swenson
*
* 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 QMK_KEYBOARD_H
enum planck_layers {
_ENIGMA,
_QWERTY,
_FN,
_LOWER,
_RAISE
};
enum planck_normal_keycodes {
QWERTY = SAFE_RANGE,
ENIGMA,
EN_A,
EN_B,
EN_C,
EN_D,
EN_E,
EN_F,
EN_G,
EN_H,
EN_I,
EN_J,
EN_K,
EN_L,
EN_M,
EN_N,
EN_O,
EN_P,
EN_Q,
EN_R,
EN_S,
EN_T,
EN_U,
EN_V,
EN_W,
EN_X,
EN_Y,
EN_Z,
EN_RES,
EN_TEST,
EN_DIAG,
EN_BSPC,
EN_SREF,
EN_SROT,
EN_SPOS,
EN_SRIN,
EN_SPLU
};
const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
/* Enigma
* ,-----------------------------------------------------------------------------------.
* | Esc | Q | W | E | R | T | Y | U | I | O | P | Bksp |
* |------+------+------+------+------+-------------+------+------+------+------+------|
* | Tab | A | S | D | F | G | H | J | K | L | ; | " |
* |------+------+------+------+------+------|------+------+------+------+------+------|
* | Shift| Z | X | C | V | B | N | M | , | . | / |Enter |
* |------+------+------+------+------+------+------+------+------+------+------+------|
* | Ctrl | Fn | Alt | GUI |Lower | Space |Raise | Left | Down | Up |Right |
* `-----------------------------------------------------------------------------------'
*/
[_ENIGMA] = LAYOUT_planck_mit(
QK_GESC, EN_Q, EN_W, EN_E, EN_R, EN_T, EN_Y, EN_U, EN_I, EN_O, EN_P, KC_BSPC,
KC_TAB, EN_A, EN_S, EN_D, EN_F, EN_G, EN_H, EN_J, EN_K, EN_L, KC_SCLN, KC_QUOT,
KC_LSFT, EN_Z, EN_X, EN_C, EN_V, EN_B, EN_N, EN_M, KC_COMM, KC_DOT, KC_SLSH, KC_ENT,
KC_LCTL, MO(_FN), KC_LGUI, KC_LALT, MO(_LOWER), KC_SPC, MO(_RAISE), KC_LEFT, KC_DOWN, KC_UP, KC_RGHT
),
/* Qwerty
* ,-----------------------------------------------------------------------------------.
* | Esc | Q | W | E | R | T | Y | U | I | O | P | Bksp |
* |------+------+------+------+------+-------------+------+------+------+------+------|
* | Tab | A | S | D | F | G | H | J | K | L | ; | " |
* |------+------+------+------+------+------|------+------+------+------+------+------|
* | Shift| Z | X | C | V | B | N | M | , | . | / |Enter |
* |------+------+------+------+------+------+------+------+------+------+------+------|
* | Ctrl | Fn | Alt | GUI |Lower | Space |Raise | Left | Down | Up |Right |
* `-----------------------------------------------------------------------------------'
*/
[_QWERTY] = LAYOUT_planck_mit(
QK_GESC, KC_Q, KC_W, KC_E, KC_R, KC_T, KC_Y, KC_U, KC_I, KC_O, KC_P, KC_BSPC,
KC_TAB, KC_A, KC_S, KC_D, KC_F, KC_G, KC_H, KC_J, KC_K, KC_L, KC_SCLN, KC_QUOT,
KC_LSFT, KC_Z, KC_X, KC_C, KC_V, KC_B, KC_N, KC_M, KC_COMM, KC_DOT, KC_SLSH, KC_ENT,
KC_LCTL, MO(_FN), KC_LGUI, KC_LALT, MO(_LOWER), KC_SPC, MO(_RAISE), KC_LEFT, KC_DOWN, KC_UP, KC_RGHT
),
/* Function
* ,-----------------------------------------------------------------------------------.
* | Boot | Reset| Diag | | | | | | | | Test |Revers|
* |------+------+------+------+------+-------------+------+------+------+------+------|
* | | | | | | | | | | | | |
* |------+------+------+------+------+------|------+------+------+------+------+------|
* | |Reflec|Rotors|Posn.s|Rings |Plugs | | |Enigma|Qwerty| | |
* |------+------+------+------+------+------+------+------+------+------+------+------|
* | | | | | | | | | | | |
* `-----------------------------------------------------------------------------------'
*/
[_FN] = LAYOUT_planck_mit(
QK_BOOT, EN_RES, EN_DIAG, _______, _______, _______, _______, _______, _______, _______, EN_TEST, EN_BSPC,
_______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,
_______, EN_SREF, EN_SROT, EN_SPOS, EN_SRIN, EN_SPLU, _______, _______, QWERTY, ENIGMA, _______, _______,
_______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______
),
/* Lower
* ,-----------------------------------------------------------------------------------.
* | ~ | ! | @ | # | $ | % | ^ | & | * | ( | ) | Bksp |
* |------+------+------+------+------+-------------+------+------+------+------+------|
* | | | | | | | | _ | + | { | } | | |
* |------+------+------+------+------+------|------+------+------+------+------+------|
* | Shift| | | | | | | | | | | |
* |------+------+------+------+------+------+------+------+------+------+------+------|
* | Ctrl | Fn | Alt | GUI |Lower | Space |Raise | Left | Down | Up |Right |
* `-----------------------------------------------------------------------------------'
*/
[_LOWER] = LAYOUT_planck_mit(
KC_TILD, KC_EXLM, KC_AT, KC_HASH, KC_DLR, KC_PERC, KC_CIRC, KC_AMPR, KC_ASTR, KC_LPRN, KC_RPRN, KC_BSPC,
_______, _______, _______, _______, _______, _______, _______, KC_UNDS, KC_PLUS, KC_LCBR, KC_RCBR, KC_PIPE,
KC_LSFT, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,
KC_LCTL, _______, KC_LGUI, KC_LALT, _______, KC_SPC, _______, KC_LEFT, KC_DOWN, KC_UP, KC_RGHT
),
/* RAISE
* ,-----------------------------------------------------------------------------------.
* | ` | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 0 | Bksp |
* |------+------+------+------+------+-------------+------+------+------+------+------|
* | | | | | | | | - | = | [ | ] | \ |
* |------+------+------+------+------+------|------+------+------+------+------+------|
* | Shift| | | | | | | | | | | |
* |------+------+------+------+------+------+------+------+------+------+------+------|
* | Ctrl | Fn | Alt | GUI |Lower | Space |Raise | Left | Down | Up |Right |
* `-----------------------------------------------------------------------------------'
*/
[_RAISE] = LAYOUT_planck_mit(
KC_GRV, KC_1, KC_2, KC_3, KC_4, KC_5, KC_6, KC_7, KC_8, KC_9, KC_0, KC_BSPC,
_______, _______, _______, _______, _______, _______, _______, KC_MINS, KC_EQL, KC_LBRC, KC_RBRC, KC_BSLS,
KC_LSFT, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,
KC_LCTL, _______, KC_LGUI, KC_LALT, _______, KC_SPC, _______, KC_LEFT, KC_DOWN, KC_UP, KC_RGHT
),
};
char rotor_definitions[5][26] = {
"EKMFLGDQVZNTOWYHXUSPAIBRCJ",
"AJDKSIRUXBLHWTMCQGZNPYFVOE",
"BDFHJLCPRTXVZNYEIWGAKMUSQO",
"ESOVPZJAYQUIRHXLNFTGKDCMWB",
"VZBRGITYUPSDNHLXAWMJQOFECK"
};
char reflector_definitions[3][26] = {
"EJMZALYXVBWFCRQUONTSPIKHGD",
"YRUHQSLDPXNGOKMIEBFZCWVJAT",
"FVPJIAOYEDRZXWGCTKUQSBNMHL"
};
char notch[5] = "QEVJZ";
typedef struct Settings {
char rotor_order[3];
char rotor_rings[3];
char rotor_positions[3];
char plugs[25];
int plug_count;
char reflector;
} Settings;
typedef struct KeyboardState {
bool is_setting_reflector;
bool is_setting_rotors;
bool is_setting_rotor_positions;
bool is_setting_rotor_rings;
bool is_setting_plugs;
char setting_progress[26];
int setting_index;
Settings current_settings;
Settings default_settings;
} KeyboardState;
int bound(int letter) {
return ((letter % 26) + 26) % 26;
}
char to_char(int letter) {
return 'A' + letter;
}
int to_int(char letter) {
return letter - 'A';
}
char encipher(char letter, Settings *settings) {
int rotor_2_step = settings->rotor_positions[2] == notch[settings->rotor_order[2] - 1];
int rotor_1_step = settings->rotor_positions[1] == notch[settings->rotor_order[1] - 1];
// Advance the third rotor
settings->rotor_positions[2]++;
if (settings->rotor_positions[2] > 'Z') {
settings->rotor_positions[2] -= 26;
}
// Maybe advance the second rotor, including double steps
if (rotor_2_step || rotor_1_step) {
settings->rotor_positions[1]++;
if (settings->rotor_positions[1] > 'Z') {
settings->rotor_positions[1] -= 26;
}
}
// Maybe advance the first rotor
if (rotor_1_step) {
settings->rotor_positions[0]++;
if (settings->rotor_positions[0] > 'Z') {
settings->rotor_positions[0] -= 26;
}
}
// Swap letters on plugboard
for (int i = 0; i < settings->plug_count * 2; i += 2) {
if (letter == settings->plugs[i]) {
letter = settings->plugs[i + 1];
} else if (letter == settings->plugs[i + 1]) {
letter = settings->plugs[i];
}
}
// Rotors (right to left)
for (int rotor_index = 2; rotor_index >= 0; rotor_index--) {
char *rotor_definition = rotor_definitions[settings->rotor_order[rotor_index] - 1];
int position = to_int(settings->rotor_positions[rotor_index]);
int ring = to_int(settings->rotor_rings[rotor_index]);
int char_index = to_int(letter);
letter = to_char(bound(
to_int(rotor_definition[bound(char_index + position - ring)]) + ring - position
));
}
// Swap via reflector
letter = reflector_definitions[to_int(settings->reflector)][to_int(letter)];
// Rotors in reverse (left to right)
for (int rotor_index = 0; rotor_index < 3; rotor_index++) {
char *rotor_definition = rotor_definitions[settings->rotor_order[rotor_index] - 1];
int position = to_int(settings->rotor_positions[rotor_index]);
int ring = to_int(settings->rotor_rings[rotor_index]);
int search_index;
for (search_index = 0; search_index < 26; search_index++) {
if (
rotor_definition[search_index]
== to_char(bound(to_int(letter) - ring + position))
) {
break;
}
}
letter = to_char(bound(search_index - position + ring));
}
// Plugboard again
for (int i = 0; i < settings->plug_count * 2; i += 2) {
if (letter == settings->plugs[i]) {
letter = settings->plugs[i + 1];
} else if (letter == settings->plugs[i + 1]) {
letter = settings->plugs[i];
}
}
return letter;
}
void init_enigma_default(Settings *settings) {
settings->rotor_order[0] = 1;
settings->rotor_rings[0] = 'A';
settings->rotor_positions[0] = 'A';
settings->rotor_order[1] = 2;
settings->rotor_rings[1] = 'A';
settings->rotor_positions[1] = 'A';
settings->rotor_order[2] = 3;
settings->rotor_rings[2] = 'A';
settings->rotor_positions[2] = 'A';
strcpy(settings->plugs, "");
settings->plug_count = 0;
settings->reflector = 'B';
}
void copy_settings(Settings *from, Settings *to) {
to->rotor_order[0] = from->rotor_order[0];
to->rotor_rings[0] = from->rotor_rings[0];
to->rotor_positions[0] = from->rotor_positions[0];
to->rotor_order[1] = from->rotor_order[1];
to->rotor_rings[1] = from->rotor_rings[1];
to->rotor_positions[1] = from->rotor_positions[1];
to->rotor_order[2] = from->rotor_order[2];
to->rotor_rings[2] = from->rotor_rings[2];
to->rotor_positions[2] = from->rotor_positions[2];
strncpy(to->plugs, from->plugs, from->plug_count * 2);
to->plug_count = from->plug_count;
to->reflector = from->reflector;
}
char *rotor_name(int rotor_number) {
if (rotor_number == 1) {
return "I";
} else if (rotor_number == 2) {
return "II";
} else if (rotor_number == 3) {
return "III";
} else if (rotor_number == 4) {
return "IV";
} else if (rotor_number == 5) {
return "V";
}
return "?";
}
void rotors_reverse(Settings *settings) {
int rotor_2_step = settings->rotor_positions[2]
== to_char(bound(to_int(notch[settings->rotor_order[2] - 1]) + 1));
int rotor_1_step = settings->rotor_positions[1]
== to_char(bound(to_int(notch[settings->rotor_order[1] - 1]) + 1));
// Reverse third rotor
settings->rotor_positions[2]--;
if (settings->rotor_positions[2] < 'A') {
settings->rotor_positions[2] += 26;
}
// Maybe reverse second rotor (including double steps)
if (rotor_2_step || rotor_1_step) {
settings->rotor_positions[1]--;
if (settings->rotor_positions[1] < 'A') {
settings->rotor_positions[1] += 26;
}
}
// Maybe recerse first rotor
if (rotor_1_step) {
settings->rotor_positions[0]--;
if (settings->rotor_positions[0] < 'A') {
settings->rotor_positions[0] += 26;
}
}
}
void reset_settings(KeyboardState *state) {
copy_settings(&state->default_settings, &state->current_settings);
}
void set_layer(uint8_t default_layer) {
default_layer_set((layer_state_t)1 << default_layer);
}
void set_backlight(uint8_t mode, uint8_t hue, uint8_t sat, uint8_t val) {
#ifdef RGBLIGHT_ENABLE
rgblight_enable_noeeprom();
rgblight_mode_noeeprom(mode);
rgblight_sethsv_noeeprom(hue, sat, val);
#endif
}
void clear_working_settings(KeyboardState *state) {
state->is_setting_reflector = false;
state->is_setting_rotors = false;
state->is_setting_rotor_positions = false;
state->is_setting_rotor_rings = false;
state->is_setting_plugs = false;
state->setting_index = 0;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_STATIC_LIGHT, HSV_RED);
}
void send_settings_string(Settings *settings) {
send_char(settings->reflector);
send_string(". ");
send_string(rotor_name(settings->rotor_order[0]));
send_char(settings->rotor_rings[0]);
send_string("-");
send_string(rotor_name(settings->rotor_order[1]));
send_char(settings->rotor_rings[1]);
send_string("-");
send_string(rotor_name(settings->rotor_order[2]));
send_char(settings->rotor_rings[2]);
send_string(" (");
send_char(settings->rotor_positions[0]);
send_string(", ");
send_char(settings->rotor_positions[1]);
send_string(", ");
send_char(settings->rotor_positions[2]);
send_string(") ");
for (int i = 0; i < settings->plug_count; i++) {
send_char(settings->plugs[i * 2]);
send_char(settings->plugs[i * 2 + 1]);
send_string(" ");
}
send_string("\n");
}
void perform_test(Settings *settings) {
for (int i = 0; i < 1000; i++) {
send_char(encipher('A', settings) - ('A' - 'a'));
}
}
void commit_plug_settings(KeyboardState *state) {
state->default_settings.plug_count = state->setting_index / 2;
for (int i = 0; i < state->default_settings.plug_count; i++) {
state->default_settings.plugs[i * 2] = state->setting_progress[i * 2];
state->default_settings.plugs[i * 2 + 1] = state->setting_progress[i * 2 + 1];
}
reset_settings(state);
clear_working_settings(state);
}
void handle_set_reflector(char letter, KeyboardState *state) {
if (letter >= 'A' && letter <= 'C') {
state->default_settings.reflector = letter;
reset_settings(state);
clear_working_settings(state);
}
}
bool handle_set_rotor(char letter, KeyboardState *state) {
bool is_valid = letter >= 'A' && letter <= 'E' && state->setting_index < 3;
if (is_valid) {
state->setting_progress[state->setting_index] = to_int(letter) + 1;
state->setting_index += 1;
if (state->setting_index == 3) {
state->default_settings.rotor_order[0] = state->setting_progress[0];
state->default_settings.rotor_order[1] = state->setting_progress[1];
state->default_settings.rotor_order[2] = state->setting_progress[2];
reset_settings(state);
clear_working_settings(state);
}
}
return is_valid;
}
bool handle_set_rotor_position(char letter, KeyboardState *state) {
bool is_valid = state->setting_index < 3; // Guaranteed to be A-Z already
if (is_valid) {
state->setting_progress[state->setting_index] = letter;
state->setting_index += 1;
if (state->setting_index == 3) {
state->default_settings.rotor_positions[0] = state->setting_progress[0];
state->default_settings.rotor_positions[1] = state->setting_progress[1];
state->default_settings.rotor_positions[2] = state->setting_progress[2];
reset_settings(state);
clear_working_settings(state);
}
}
return is_valid;
}
bool handle_set_rotor_ring(char letter, KeyboardState *state) {
bool is_valid = state->setting_index < 3;
if (is_valid) {
state->setting_progress[state->setting_index] = letter;
state->setting_index += 1;
if (state->setting_index == 3) {
state->default_settings.rotor_rings[0] = state->setting_progress[0];
state->default_settings.rotor_rings[1] = state->setting_progress[1];
state->default_settings.rotor_rings[2] = state->setting_progress[2];
reset_settings(state);
clear_working_settings(state);
}
}
return is_valid;
}
bool handle_set_plug(char letter, KeyboardState *state) {
bool is_valid = state->setting_index < 26;
if (is_valid) {
state->setting_progress[state->setting_index] = letter;
state->setting_index += 1;
}
return is_valid;
}
void handle_enigma_keypress(char letter, bool any_mods, KeyboardState *state) {
bool settings_valid = true;
if (letter) {
if (any_mods) {
tap_code(KC_A + to_int(letter));
} else if (state->is_setting_reflector) {
handle_set_reflector(letter, state);
} else if (state->is_setting_rotors) {
settings_valid = handle_set_rotor(letter, state);
} else if (state->is_setting_rotor_positions) {
settings_valid = handle_set_rotor_position(letter, state);
} else if (state->is_setting_rotor_rings) {
settings_valid = handle_set_rotor_ring(letter, state);
} else if (state->is_setting_plugs) {
settings_valid = handle_set_plug(letter, state);
} else {
char c2 = encipher(letter, &state->current_settings);
send_char(c2 - ('A' - 'a'));
}
}
if (!settings_valid) {
clear_working_settings(state);
}
}
KeyboardState STATE;
void keyboard_pre_init_user(void) {
init_enigma_default(&STATE.default_settings);
init_enigma_default(&STATE.current_settings);
clear_working_settings(&STATE);
}
bool process_record_user(uint16_t keycode, keyrecord_t *record) {
uint8_t letter_index;
bool letter_found = false;
if (record->event.pressed) {
switch (keycode) {
case QWERTY:
set_layer(_QWERTY);
set_backlight(RGBLIGHT_MODE_RAINBOW_SWIRL + 4, HSV_PURPLE);
break;
case ENIGMA:
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_STATIC_LIGHT, HSV_RED);
break;
case EN_SREF:
reset_settings(&STATE);
STATE.is_setting_reflector = true;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_SNAKE, HSV_RED);
break;
case EN_SROT:
reset_settings(&STATE);
STATE.is_setting_rotors = true;
STATE.setting_index = 0;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_SNAKE, 10, 255, 255);
break;
case EN_SPOS:
reset_settings(&STATE);
STATE.is_setting_rotor_positions = true;
STATE.setting_index = 0;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_SNAKE, HSV_ORANGE);
break;
case EN_SRIN:
reset_settings(&STATE);
STATE.is_setting_rotor_rings = true;
STATE.setting_index = 0;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_SNAKE, HSV_GREEN);
break;
case EN_SPLU:
reset_settings(&STATE);
STATE.is_setting_plugs = true;
STATE.setting_index = 0;
set_layer(_ENIGMA);
set_backlight(RGBLIGHT_MODE_SNAKE, HSV_BLUE);
break;
case QK_GESC:
if (
STATE.is_setting_reflector
|| STATE.is_setting_rotors
|| STATE.is_setting_rotor_positions
|| STATE.is_setting_rotor_rings
|| STATE.is_setting_plugs
) {
clear_working_settings(&STATE);
return false;
}
break;
case KC_ENT:
if (STATE.is_setting_plugs) {
commit_plug_settings(&STATE);
return false;
}
break;
case EN_A ... EN_Z:
letter_index = keycode - EN_A;
letter_found = true;
break;
case EN_RES:
reset_settings(&STATE);
break;
case EN_TEST:
perform_test(&STATE.current_settings);
break;
case EN_DIAG:
send_settings_string(&STATE.current_settings);
break;
case EN_BSPC:
rotors_reverse(&STATE.current_settings);
tap_code(KC_BSPC);
break;
}
}
char letter = letter_found ? 'A' + letter_index : 0;
uint8_t mods = get_mods();
bool any_mods = (mods & MOD_MASK_CTRL) || (mods & MOD_MASK_ALT) || (mods & MOD_MASK_GUI);
handle_enigma_keypress(letter, any_mods, &STATE);
return true;
}
void keyboard_post_init_user(void) {
set_layer(_QWERTY);
set_backlight(RGBLIGHT_MODE_RAINBOW_SWIRL + 4, HSV_PURPLE);
}