From ff893bf17c454f28a8a00f9bf27d4ec4a72727e9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Em=C4=ABls=20Delle?= <emiilsdelle@gmail.com>
Date: Tue, 5 Feb 2019 00:12:23 +0200
Subject: [PATCH] Tmo50 indicators (#5044)

* Create ISO HHKB keymapping for GH60

* Add media controls to Fn layer

* Use M(x) instead of F(x), add Copyright text

* Add README for additional information about TMO50 features/quirks

* Add indicator LED code

* Move indicator code to tmo50.c
---
 keyboards/tmo50/keymaps/default/keymap.c |  1 -
 keyboards/tmo50/tmo50.c                  | 41 ++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/keyboards/tmo50/keymaps/default/keymap.c b/keyboards/tmo50/keymaps/default/keymap.c
index 419893f1a6..7b8b644cc6 100644
--- a/keyboards/tmo50/keymaps/default/keymap.c
+++ b/keyboards/tmo50/keymaps/default/keymap.c
@@ -48,4 +48,3 @@ const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
 		KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS
   ),
 };
-
diff --git a/keyboards/tmo50/tmo50.c b/keyboards/tmo50/tmo50.c
index a42ab4720a..4f6288133d 100644
--- a/keyboards/tmo50/tmo50.c
+++ b/keyboards/tmo50/tmo50.c
@@ -19,6 +19,15 @@ void matrix_init_kb(void) {
 	// put your keyboard start-up code here
 	// runs once when the firmware starts up
 
+  DDRB |= (1 << PB0); //init B0
+  PORTB &= ~(1 << PB0); //turn on B0
+  DDRB |= (1 << PB1);
+  PORTB |= (1<<PB1); //turn off B1
+  DDRB |= (1 << PB2);
+  PORTB |= (1<<PB2);
+  DDRB |= (1 << PB3);
+  PORTB |= (1<<PB3);
+
 	matrix_init_user();
 }
 
@@ -41,3 +50,35 @@ void led_set_kb(uint8_t usb_led) {
 
 	led_set_user(usb_led);
 }
+
+uint32_t layer_state_set_user(uint32_t state)
+{
+  // if on layer 0, turn on B0 LED, otherwise off.
+    if (biton32(state) == 0) {
+        PORTB &= ~(1<<PB0);
+    } else {
+        PORTB |= (1<<PB0);
+    }
+
+  // if on layer 1, turn on B1 LED, otherwise off.
+    if (biton32(state) == 1) {
+        PORTB &= ~(1<<PB1);
+    } else {
+        PORTB |= (1<<PB1);
+    }
+  // if on layer 2, turn on B2 LED, otherwise off.
+    if (biton32(state) == 2) {
+        PORTB &= ~(1<<PB2);
+    } else {
+        PORTB |= (1<<PB2);
+    }
+
+  // if on layer 3, turn on B3 LED, otherwise off.
+    if (biton32(state) == 3) {
+        PORTB &= ~(1<<PB3);
+    } else {
+        PORTB |= (1<<PB3);
+    }
+
+    return state;
+}