OLED Driver Feature

quantum/quantum.c

@@ -275,6 +275,12 @@ bool process_record_quantum(keyrecord_t *record) {
     preprocess_tap_dance(keycode, record);
   #endif
 
+  #if defined(OLED_DRIVER_ENABLE) && !defined(OLED_DISABLE_TIMEOUT)
+    // Wake up oled if user is using those fabulous keys!
+    if (record->event.pressed)
+      oled_on();
+  #endif
+
   if (!(
   #if defined(KEY_LOCK_ENABLE)
     // Must run first to be able to mask key_up events.
@@ -1084,6 +1090,12 @@ void matrix_init_quantum() {
   #ifdef HAPTIC_ENABLE
     haptic_init();
   #endif
+  #ifdef OUTPUT_AUTO_ENABLE
+    set_output(OUTPUT_AUTO);
+  #endif
+  #ifdef OLED_DRIVER_ENABLE
+    oled_init(OLED_ROTATION_0);
+  #endif
   matrix_init_kb();
 }
 
@@ -1120,6 +1132,10 @@ void matrix_scan_quantum() {
     haptic_task();
   #endif
 
+  #ifdef OLED_DRIVER_ENABLE
+    oled_task();
+  #endif
+
   matrix_scan_kb();
 }
 #if defined(BACKLIGHT_ENABLE) && defined(BACKLIGHT_PIN)