X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=keyboards%2Fviterbi%2Fviterbi.h;h=b19f33154bb889b9eae914fb3fc0f17778cf1cec;hb=cc738e32dd02cbc6e10f3e229332390d3b886e4d;hp=34742391041456a14ded776c5629a8ec41eb3ecb;hpb=800ec55dfca06b4630acf62cbb5f130c4031e4f1;p=qmk_firmware.git

diff --git a/keyboards/viterbi/viterbi.h b/keyboards/viterbi/viterbi.h
index 347423910..b19f33154 100644
--- a/keyboards/viterbi/viterbi.h
+++ b/keyboards/viterbi/viterbi.h
@@ -1,19 +1,22 @@
-#ifndef VITERBI_H
-#define VITERBI_H
+#pragma once
+
+#include "quantum.h"
 
 #ifdef KEYBOARD_viterbi_rev1
     #include "rev1.h"
+#elif KEYBOARD_viterbi_rev2
+    #include "rev2.h"
 #endif
 
 // Used to create a keymap using only KC_ prefixed keys
-#define KC_KEYMAP( \
+#define LAYOUT_kc( \
     L00, L01, L02, L03, L04, L05, L06, R00, R01, R02, R03, R04, R05, R06, \
     L10, L11, L12, L13, L14, L15, L16, R10, R11, R12, R13, R14, R15, R16, \
     L20, L21, L22, L23, L24, L25, L26, R20, R21, R22, R23, R24, R25, R26, \
     L30, L31, L32, L33, L34, L35, L36, R30, R31, R32, R33, R34, R35, R36, \
     L40, L41, L42, L43, L44, L45, L46, R40, R41, R42, R43, R44, R45, R46 \
     ) \
-    KEYMAP( \
+    LAYOUT( \
         KC_##L00, KC_##L01, KC_##L02, KC_##L03, KC_##L04, KC_##L05, KC_##L06, KC_##R00, KC_##R01, KC_##R02, KC_##R03, KC_##R04, KC_##R05, KC_##R06, \
         KC_##L10, KC_##L11, KC_##L12, KC_##L13, KC_##L14, KC_##L15, KC_##L16, KC_##R10, KC_##R11, KC_##R12, KC_##R13, KC_##R14, KC_##R15, KC_##R16, \
         KC_##L20, KC_##L21, KC_##L22, KC_##L23, KC_##L24, KC_##L25, KC_##L26, KC_##R20, KC_##R21, KC_##R22, KC_##R23, KC_##R24, KC_##R25, KC_##R26, \
@@ -21,6 +24,5 @@
         KC_##L40, KC_##L41, KC_##L42, KC_##L43, KC_##L44, KC_##L45, KC_##L46, KC_##R40, KC_##R41, KC_##R42, KC_##R43, KC_##R44, KC_##R45, KC_##R46 \
     )
 
-#include "quantum.h"
+#define LAYOUT_ortho_5x14 LAYOUT
 
-#endif