git.gag.com Git - fw/altos/blob - src/stm/ao_serial_stm.c

   1 /*
   2  * Copyright © 2012 Keith Packard <keithp@keithp.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  16  */
  17
  18 #include <ao.h>
  19
  20 void
  21 ao_debug_out(char c)
  22 {
  23         if (c == '\n')
  24                 ao_debug_out('\r');
  25         while (!(stm_usart1.sr & (1 << STM_USART_SR_TXE)));
  26         stm_usart1.dr = c;
  27 }
  28
  29 static void
  30 _ao_usart_tx_start(struct ao_stm_usart *usart)
  31 {
  32         if (!ao_fifo_empty(usart->tx_fifo) && !usart->tx_started)
  33         {
  34                 usart->tx_started = 1;
  35                 ao_fifo_remove(usart->tx_fifo, usart->reg->dr);
  36         }
  37 }
  38
  39 static void
  40 ao_usart_isr(struct ao_stm_usart *usart, int stdin)
  41 {
  42         uint32_t        sr;
  43
  44         sr = usart->reg->sr;
  45         usart->reg->sr = 0;
  46
  47         if (sr & (1 << STM_USART_SR_RXNE)) {
  48                 char c = usart->reg->dr;
  49                 if (!ao_fifo_full(usart->rx_fifo))
  50                         ao_fifo_insert(usart->rx_fifo, c);
  51                 ao_wakeup(&usart->rx_fifo);
  52                 if (stdin)
  53                         ao_wakeup(&ao_stdin_ready);
  54         }
  55         if (sr & (1 << STM_USART_SR_TC)) {
  56                 usart->tx_started = 0;
  57                 _ao_usart_tx_start(usart);
  58                 ao_wakeup(&usart->tx_fifo);
  59         }
  60 }
  61
  62 int
  63 _ao_usart_pollchar(struct ao_stm_usart *usart)
  64 {
  65         int     c;
  66
  67         if (ao_fifo_empty(usart->rx_fifo))
  68                 c = AO_READ_AGAIN;
  69         else {
  70                 uint8_t u;
  71                 ao_fifo_remove(usart->rx_fifo,u);
  72                 c = u;
  73         }
  74         return c;
  75 }
  76
  77 char
  78 ao_usart_getchar(struct ao_stm_usart *usart)
  79 {
  80         int c;
  81         ao_arch_block_interrupts();
  82         while ((c = _ao_usart_pollchar(usart)) == AO_READ_AGAIN)
  83                 ao_sleep(&usart->rx_fifo);
  84         ao_arch_release_interrupts();
  85         return (char) c;
  86 }
  87
  88 void
  89 ao_usart_putchar(struct ao_stm_usart *usart, char c)
  90 {
  91         ao_arch_block_interrupts();
  92         while (ao_fifo_full(usart->tx_fifo))
  93                 ao_sleep(&usart->tx_fifo);
  94         ao_fifo_insert(usart->tx_fifo, c);
  95         _ao_usart_tx_start(usart);
  96         ao_arch_release_interrupts();
  97 }
  98
  99 void
 100 ao_usart_drain(struct ao_stm_usart *usart)
 101 {
 102         ao_arch_block_interrupts();
 103         while (!ao_fifo_empty(usart->tx_fifo))
 104                 ao_sleep(&usart->tx_fifo);
 105         ao_arch_release_interrupts();
 106 }
 107
 108 static const struct {
 109         uint32_t brr;
 110 } ao_usart_speeds[] = {
 111         [AO_SERIAL_SPEED_4800] = {
 112                 AO_PCLK1 / 4800
 113         },
 114         [AO_SERIAL_SPEED_9600] = {
 115                 AO_PCLK1 / 9600
 116         },
 117         [AO_SERIAL_SPEED_19200] = {
 118                 AO_PCLK1 / 19200
 119         },
 120         [AO_SERIAL_SPEED_57600] = {
 121                 AO_PCLK1 / 57600
 122         },
 123         [AO_SERIAL_SPEED_115200] = {
 124                 AO_PCLK1 / 115200
 125         },
 126 };
 127
 128 void
 129 ao_usart_set_speed(struct ao_stm_usart *usart, uint8_t speed)
 130 {
 131         if (speed > AO_SERIAL_SPEED_115200)
 132                 return;
 133         usart->reg->brr = ao_usart_speeds[speed].brr;
 134 }
 135
 136 void
 137 ao_usart_init(struct ao_stm_usart *usart)
 138 {
 139         usart->reg->cr1 = ((0 << STM_USART_CR1_OVER8) |
 140                           (1 << STM_USART_CR1_UE) |
 141                           (0 << STM_USART_CR1_M) |
 142                           (0 << STM_USART_CR1_WAKE) |
 143                           (0 << STM_USART_CR1_PCE) |
 144                           (0 << STM_USART_CR1_PS) |
 145                           (0 << STM_USART_CR1_PEIE) |
 146                           (0 << STM_USART_CR1_TXEIE) |
 147                           (1 << STM_USART_CR1_TCIE) |
 148                           (1 << STM_USART_CR1_RXNEIE) |
 149                           (0 << STM_USART_CR1_IDLEIE) |
 150                           (1 << STM_USART_CR1_TE) |
 151                           (1 << STM_USART_CR1_RE) |
 152                           (0 << STM_USART_CR1_RWU) |
 153                           (0 << STM_USART_CR1_SBK));
 154
 155         usart->reg->cr2 = ((0 << STM_USART_CR2_LINEN) |
 156                           (STM_USART_CR2_STOP_1 << STM_USART_CR2_STOP) |
 157                           (0 << STM_USART_CR2_CLKEN) |
 158                           (0 << STM_USART_CR2_CPOL) |
 159                           (0 << STM_USART_CR2_CPHA) |
 160                           (0 << STM_USART_CR2_LBCL) |
 161                           (0 << STM_USART_CR2_LBDIE) |
 162                           (0 << STM_USART_CR2_LBDL) |
 163                           (0 << STM_USART_CR2_ADD));
 164
 165         usart->reg->cr3 = ((0 << STM_USART_CR3_ONEBITE) |
 166                           (0 << STM_USART_CR3_CTSIE) |
 167                           (0 << STM_USART_CR3_CTSE) |
 168                           (0 << STM_USART_CR3_RTSE) |
 169                           (0 << STM_USART_CR3_DMAT) |
 170                           (0 << STM_USART_CR3_DMAR) |
 171                           (0 << STM_USART_CR3_SCEN) |
 172                           (0 << STM_USART_CR3_NACK) |
 173                           (0 << STM_USART_CR3_HDSEL) |
 174                           (0 << STM_USART_CR3_IRLP) |
 175                           (0 << STM_USART_CR3_IREN) |
 176                           (0 << STM_USART_CR3_EIE));
 177
 178         /* Pick a 9600 baud rate */
 179         ao_usart_set_speed(usart, AO_SERIAL_SPEED_9600);
 180 }
 181
 182 #if HAS_SERIAL_1
 183
 184 struct ao_stm_usart ao_stm_usart1;
 185
 186 void stm_usart1_isr(void) { ao_usart_isr(&ao_stm_usart1, USE_SERIAL_1_STDIN); }
 187
 188 char
 189 ao_serial1_getchar(void)
 190 {
 191         return ao_usart_getchar(&ao_stm_usart1);
 192 }
 193
 194 void
 195 ao_serial1_putchar(char c)
 196 {
 197         ao_usart_putchar(&ao_stm_usart1, c);
 198 }
 199
 200 int
 201 _ao_serial1_pollchar(void)
 202 {
 203         return _ao_usart_pollchar(&ao_stm_usart1);
 204 }
 205
 206 void
 207 ao_serial1_set_speed(uint8_t speed)
 208 {
 209         ao_usart_set_speed(&ao_stm_usart1, speed);
 210 }
 211 #endif  /* HAS_SERIAL_1 */
 212
 213 #if HAS_SERIAL_2
 214
 215 struct ao_stm_usart ao_stm_usart2;
 216
 217 void stm_usart2_isr(void) { ao_usart_isr(&ao_stm_usart2, USE_SERIAL_2_STDIN); }
 218
 219 char
 220 ao_serial2_getchar(void)
 221 {
 222         return ao_usart_getchar(&ao_stm_usart2);
 223 }
 224
 225 void
 226 ao_serial2_putchar(char c)
 227 {
 228         ao_usart_putchar(&ao_stm_usart2, c);
 229 }
 230
 231 int
 232 _ao_serial2_pollchar(void)
 233 {
 234         return _ao_usart_pollchar(&ao_stm_usart2);
 235 }
 236
 237 void
 238 ao_serial2_set_speed(uint8_t speed)
 239 {
 240         ao_usart_set_speed(&ao_stm_usart2, speed);
 241 }
 242 #endif  /* HAS_SERIAL_2 */
 243
 244 #if HAS_SERIAL_3
 245
 246 struct ao_stm_usart ao_stm_usart3;
 247
 248 void stm_usart3_isr(void) { ao_usart_isr(&ao_stm_usart3, USE_SERIAL_2_STDIN); }
 249
 250 char
 251 ao_serial3_getchar(void)
 252 {
 253         return ao_usart_getchar(&ao_stm_usart3);
 254 }
 255
 256 void
 257 ao_serial3_putchar(char c)
 258 {
 259         ao_usart_putchar(&ao_stm_usart3, c);
 260 }
 261
 262 int
 263 _ao_serial3_pollchar(void)
 264 {
 265         return _ao_usart_pollchar(&ao_stm_usart3);
 266 }
 267
 268 void
 269 ao_serial3_set_speed(uint8_t speed)
 270 {
 271         ao_usart_set_speed(&ao_stm_usart3, speed);
 272 }
 273 #endif  /* HAS_SERIAL_3 */
 274
 275 void
 276 ao_serial_init(void)
 277 {
 278 #if HAS_SERIAL_1
 279         /*
 280          *      TX      RX
 281          *      PA9     PA10
 282          *      PB6     PB7     *
 283          */
 284
 285 #if SERIAL_1_PA9_PA10
 286         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOAEN);
 287
 288         stm_afr_set(&stm_gpioa, 9, STM_AFR_AF7);
 289         stm_afr_set(&stm_gpioa, 10, STM_AFR_AF7);
 290 #else
 291 #if SERIAL_1_PB6_PB7
 292         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOBEN);
 293
 294         stm_afr_set(&stm_gpiob, 6, STM_AFR_AF7);
 295         stm_afr_set(&stm_gpiob, 7, STM_AFR_AF7);
 296 #else
 297 #error "No SERIAL_1 port configuration specified"
 298 #endif
 299 #endif
 300         /* Enable USART */
 301         stm_rcc.apb2enr |= (1 << STM_RCC_APB2ENR_USART1EN);
 302
 303         ao_stm_usart1.reg = &stm_usart1;
 304         ao_usart_init(&ao_stm_usart1);
 305
 306         stm_nvic_set_enable(STM_ISR_USART1_POS);
 307         stm_nvic_set_priority(STM_ISR_USART1_POS, 4);
 308 #if USE_SERIAL_1_STDIN
 309         ao_add_stdio(_ao_serial1_pollchar,
 310                      ao_serial1_putchar,
 311                      NULL);
 312 #endif
 313 #endif
 314
 315 #if HAS_SERIAL_2
 316         /*
 317          *      TX      RX
 318          *      PA2     PA3
 319          *      PD5     PD6
 320          */
 321
 322 #if SERIAL_2_PA2_PA3
 323         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOAEN);
 324
 325         stm_afr_set(&stm_gpioa, 2, STM_AFR_AF7);
 326         stm_afr_set(&stm_gpioa, 3, STM_AFR_AF7);
 327 #else
 328 #if SERIAL_2_PD5_PD6
 329         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIODEN);
 330
 331         stm_afr_set(&stm_gpiod, 5, STM_AFR_AF7);
 332         stm_afr_set(&stm_gpiod, 6, STM_AFR_AF7);
 333 #else
 334 #error "No SERIAL_2 port configuration specified"
 335 #endif
 336 #endif
 337         /* Enable USART */
 338         stm_rcc.apb1enr |= (1 << STM_RCC_APB1ENR_USART2EN);
 339
 340         ao_stm_usart2.reg = &stm_usart2;
 341         ao_usart_init(&ao_stm_usart2);
 342
 343         stm_nvic_set_enable(STM_ISR_USART2_POS);
 344         stm_nvic_set_priority(STM_ISR_USART2_POS, 4);
 345 #if USE_SERIAL_2_STDIN
 346         ao_add_stdio(_ao_serial2_pollchar,
 347                      ao_serial2_putchar,
 348                      NULL);
 349 #endif
 350 #endif
 351
 352 #if HAS_SERIAL_3
 353         /*
 354          *      TX      RX
 355          *      PB10    PB11
 356          *      PC10    PC11
 357          *      PD8     PD9
 358          */
 359 #if SERIAL_3_PB10_PB11
 360         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOBEN);
 361
 362         stm_afr_set(&stm_gpiob, 10, STM_AFR_AF7);
 363         stm_afr_set(&stm_gpiob, 11, STM_AFR_AF7);
 364 #else
 365 #if SERIAL_3_PC10_PC11
 366         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIOCEN);
 367
 368         stm_afr_set(&stm_gpioc, 10, STM_AFR_AF7);
 369         stm_afr_set(&stm_gpioc, 11, STM_AFR_AF7);
 370 #else
 371 #if SERIAL_3_PD8_PD9
 372         stm_rcc.ahbenr |= (1 << STM_RCC_AHBENR_GPIODEN);
 373
 374         stm_afr_set(&stm_gpiod, 8, STM_AFR_AF7);
 375         stm_afr_set(&stm_gpiod, 9, STM_AFR_AF7);
 376 #else
 377 #error "No SERIAL_3 port configuration specified"
 378 #endif
 379 #endif
 380 #endif
 381         /* Enable USART */
 382         stm_rcc.apb1enr |= (1 << STM_RCC_APB1ENR_USART3EN);
 383
 384         ao_stm_usart3.reg = &stm_usart3;
 385         ao_usart_init(&ao_stm_usart3);
 386
 387         stm_nvic_set_enable(STM_ISR_USART3_POS);
 388         stm_nvic_set_priority(STM_ISR_USART3_POS, 4);
 389 #if USE_SERIAL_3_STDIN
 390         ao_add_stdio(_ao_serial3_pollchar,
 391                      ao_serial3_putchar,
 392                      NULL);
 393 #endif
 394 #endif
 395 }
 396
 397