5 files changed, 30 insertions, 17 deletions
diff --git a/mcu/mcu.h b/mcu/mcu.h
index 2caf432..2a950c8 100644
--- a/mcu/mcu.h
+++ b/mcu/mcu.h
@@ -1,19 +1,15 @@
-#if (defined __AVR_ATmega328P__ || defined __AVR_ATmega328__ || \
+#define __MCU_ATmega328p__ \
+    ((defined __AVR_ATmega328P__ || defined __AVR_ATmega328__ || \
         defined __AVR_ATmega1688PA__ || defined __AVR_ATmega168A__ || \
         defined __AVR_ATmega88PA__ || defined __AVR_ATmega88A__ || \
-        defined __AVR_ATmega48PA__ || defined AVR_ATmega48A__)
-#include "ATmega328P.h"
-#endif
+        defined __AVR_ATmega48PA__ || defined AVR_ATmega48A__))
 
-#if (defined __AVR_ATmega328U4__ || defined __AVR_ATmega16U4)
-#include "ATmega32U4.h"
-#endif
+#define __MCU_ATmega32U4__ \
+    ((defined __AVR_ATmega32U4__ || defined __AVR_ATmega16U4))
 
-#if (defined __AVR_ATtiny4313__ || defined __AVR_ATtiny2313A__)
-#include "ATtiny4313.h"
-#endif
+#define __MCU_ATtiny4313__ \
+    ((defined __AVR_ATtiny4313__ || defined __AVR_ATtiny2313A__))
 
-#if (defined __AVR_ATtiny85__ || defined __AVR_ATtiny45__ || \
-        defined __AVR_ATtiny25__)
-#include "ATtiny85.h"
-#endif
+#define __MCU_ATtiny85__ \
+    ((defined __AVR_ATtiny85__ || defined __AVR_ATtiny45__ || \
+        defined __AVR_ATtiny25__))
diff --git a/mcu/mcu_def.h b/mcu/mcu_def.h
new file mode 100644
index 0000000..b423a77
--- /dev/null
+++ b/mcu/mcu_def.h
@@ -0,0 +1,17 @@
+#include "mcu.h"
+
+#if __MCU_ATmega328p__
+#include "ATmega328P.h"
+#endif
+
+#if __MCU_ATmega32U4__
+#include "ATmega32U4.h"
+#endif
+
+#if __MCU_ATtiny4313__
+#include "ATtiny4313.h"
+#endif
+
+#if __MCU_ATtiny85__
+#include "ATtiny85.h"
+#endif
diff --git a/src/spi.c b/src/spi.c
index 8462905..77ed8a3 100644
--- a/src/spi.c
+++ b/src/spi.c
@@ -1,5 +1,5 @@
 #include "../spi.h"
-#include "mcu/mcu.h"
+#include "mcu/mcu_def.h"
 
 inline void ioe_spi_join(void) {
     // TODO
diff --git a/src/spi_usart.c b/src/spi_usart.c
index 40eb06d..2e5d612 100644
--- a/src/spi_usart.c
+++ b/src/spi_usart.c
@@ -1,5 +1,5 @@
 #include "../spi_usart.h"
-#include "mcu/mcu.h"
+#include "../mcu/mcu_def.h"
 
 inline void ioe_spi_usart_init(void) {
 }
diff --git a/src/spi_usi.c b/src/spi_usi.c
index 7bdb5c2..8baa8e2 100644
--- a/src/spi_usi.c
+++ b/src/spi_usi.c
@@ -1,5 +1,5 @@
 #include "../spi_usi.h"
-#include "../mcu/mcu.h"
+#include "../mcu/mcu_def.h"
 
 #ifdef IOE_SPI_USI_MASTER