1 files changed, 49 insertions, 0 deletions
diff --git a/src/stm/ao_arch_funcs.h b/src/stm/ao_arch_funcs.h
index 052abb65..309937d5 100644
--- a/src/stm/ao_arch_funcs.h
+++ b/src/stm/ao_arch_funcs.h
@@ -37,6 +37,55 @@ ao_spi_recv(void *block, uint16_t len, uint8_t spi_index);
 void
 ao_spi_init(void);
 
+#define ao_spi_get_mask(reg,mask,bus) do {		\
+		ao_spi_get(bus);			\
+		(reg).bsrr = ((uint32_t) mask) << 16;	\
+	} while (0)
+
+#define ao_spi_put_mask(reg,mask,bus) do {	\
+		(reg).bsrr = mask;		\
+		ao_spi_put(bus);		\
+	} while (0)
+
+#define ao_stm_enable_port(port) do {					\
+		if (&(port) == &stm_gpioa)				\
+			stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOAEN); \
+		else if (&(port) == &stm_gpiob)				\
+			stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOBEN); \
+		else if (&(port) == &stm_gpioc)				\
+			stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOCEN); \
+		else if (&(port) == &stm_gpiod)				\
+			stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIODEN); \
+		else if (&(port) == &stm_gpioe)				\
+			stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOEEN); \
+	} while (0)
+
+
+#define ao_stm_enable_cs(port,bit) do {				\
+		stm_gpio_set(&(port), bit, 1);			\
+		stm_moder_set(&(port), bit, STM_MODER_OUTPUT);	\
+	} while (0)
+
+#define ao_spi_init_cs(port, mask) do {				\
+		ao_stm_enable_port(port);			\
+		if (mask & 0x0001) ao_stm_enable_cs(port, 0);	\
+		if (mask & 0x0002) ao_stm_enable_cs(port, 1);	\
+		if (mask & 0x0004) ao_stm_enable_cs(port, 2);	\
+		if (mask & 0x0008) ao_stm_enable_cs(port, 3);	\
+		if (mask & 0x0010) ao_stm_enable_cs(port, 4);	\
+		if (mask & 0x0020) ao_stm_enable_cs(port, 5);	\
+		if (mask & 0x0040) ao_stm_enable_cs(port, 6);	\
+		if (mask & 0x0080) ao_stm_enable_cs(port, 7);	\
+		if (mask & 0x0100) ao_stm_enable_cs(port, 8);	\
+		if (mask & 0x0200) ao_stm_enable_cs(port, 9);	\
+		if (mask & 0x0400) ao_stm_enable_cs(port, 10);	\
+		if (mask & 0x0800) ao_stm_enable_cs(port, 11);	\
+		if (mask & 0x1000) ao_stm_enable_cs(port, 12);	\
+		if (mask & 0x2000) ao_stm_enable_cs(port, 13);	\
+		if (mask & 0x4000) ao_stm_enable_cs(port, 14);	\
+		if (mask & 0x8000) ao_stm_enable_cs(port, 15);	\
+	} while (0)
+
 /* ao_dma_stm.c
  */