parent
db26607b2d
commit
618de224e0
@ -1,6 +1,8 @@ |
||||
[submodule "libopencm3"] |
||||
path = libopencm3 |
||||
url = https://github.com/libopencm3/libopencm3 |
||||
ignore = all |
||||
[submodule "STM32duino-bootloader"] |
||||
path = STM32duino-bootloader |
||||
url = https://github.com/rogerclarkmelbourne/STM32duino-bootloader.git |
||||
ignore = all |
||||
|
@ -0,0 +1,235 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles the communication with a HC-05 bluetooth module */ |
||||
/* peripherals used: USART (check source for details) */ |
||||
|
||||
/* standard libraries */ |
||||
#include <stdint.h> // standard integer types |
||||
#include <stdio.h> // standard I/O facilities |
||||
#include <stdlib.h> // general utilities |
||||
#include <string.h> // string utilities |
||||
|
||||
/* STM32 (including CM3) libraries */ |
||||
#include <libopencm3/stm32/rcc.h> // real-time control clock library |
||||
#include <libopencm3/stm32/gpio.h> // general purpose input output library |
||||
#include <libopencm3/stm32/usart.h> // universal synchronous asynchronous receiver transmitter library |
||||
#include <libopencm3/cm3/nvic.h> // interrupt handler |
||||
#include <libopencmsis/core_cm3.h> // Cortex M3 utilities |
||||
|
||||
#include "global.h" // common utilities |
||||
#include "bluetooth_hc-05.h" // USART header and definitions |
||||
|
||||
/* which USART to use to talk to the bluetooth module */ |
||||
#define USART USART2 |
||||
#define USART_RCC RCC_USART2 |
||||
#define USART_IRQ NVIC_USART2_IRQ |
||||
#define USART_PORT GPIOA |
||||
#define USART_PIN_TX GPIO_USART2_TX |
||||
#define USART_PIN_RX GPIO_USART3_RX |
||||
#define USART_BAUDRATE 9600 |
||||
/* AT mode pin
|
||||
* to configure the HC-05 bluetooth module you need to put it in AT mode and use AT commands |
||||
* there are several ways to get into AT mode, and there are several AT modes |
||||
* - in full AT mode you can send all [defined] AT commands |
||||
* - in mini AT mode you can't send some AT commands, such as AT+NAME |
||||
* the AT mode is defined by the state of pin 34 of the module (top right) |
||||
* - high: full AT mode |
||||
* - low: mini AT mode |
||||
* this pin is often connect to a switch on the adapter board (connected to VCC) |
||||
* to enter AT mode: |
||||
* - set pin 34 high when powering up, the LED will blink slowly, the baudrate will be 38400, bluetooth will be off |
||||
* - set pin 34 high after powering up, the LED will keep blinking fast, the baudrate is user defined (AT+UART), bluetooth communication will keep working |
||||
* setting pin 34 low after setting it high will put it into mini AT mode, except when bluetooth is not connected |
||||
* when booting normally (fast blinking LED), the module will not respond to AT commands before setting pin 34 high |
||||
* connect pin 34 to a STM32 GPIO to be able to go into AT mode */ |
||||
#define AT_PORT GPIOB |
||||
#define AT_RCC RCC_GPIOB |
||||
#define AT_PIN GPIO5 |
||||
|
||||
/* input and output ring buffer, indexes, and available memory */ |
||||
static uint8_t rx_buffer[BT_BUFFER] = {0}; |
||||
static volatile uint8_t rx_i = 0; |
||||
static volatile uint8_t rx_used = 0; |
||||
static uint8_t tx_buffer[BT_BUFFER] = {0}; |
||||
static volatile uint8_t tx_i = 0; |
||||
static volatile uint8_t tx_used = 0; |
||||
/* show the user how much data received over bluetooth is ready */ |
||||
volatile uint8_t bt_received = 0; // same as rx_used, but since the user can write this variable we don't rely on it
|
||||
|
||||
/* display configuration of bluetooth module */ |
||||
void bt_info(void) |
||||
{ |
||||
char* at_commands[] = {"AT+VERSION?","AT+ADDR?","AT+ROLE?","AT+UART?","AT+CMODE?","AT+STATE?","AT+NAME?"}; |
||||
char* ok = "OK\r\n"; |
||||
gpio_set(AT_PORT, AT_PIN); // enable AT mode
|
||||
for (uint8_t i=0; i<LENGTH(at_commands); i++) { // go through commands
|
||||
for (uint8_t j=0; j<strlen(at_commands[i]); j++) { // send command
|
||||
bt_putchar_nonblocking(at_commands[i][j]); // send character
|
||||
} |
||||
bt_putchar_nonblocking('\r'); // send end of AT command
|
||||
bt_putchar_nonblocking('\n'); // send end of AT command
|
||||
uint8_t ok_i = 0; // how much of the ok sequence has been detected
|
||||
while (ok_i<strlen(ok)) { // print until OK has been received
|
||||
char c = bt_getchar(); // received character
|
||||
printf("%c",c); // print received character
|
||||
if (ok_i<strlen(ok) && c==ok[ok_i]) { |
||||
ok_i++; |
||||
} else { |
||||
ok_i = 0; |
||||
} |
||||
} |
||||
|
||||
} |
||||
gpio_clear(AT_PORT, AT_PIN); // disable AT mode
|
||||
} |
||||
|
||||
/* setup communication to bluetooth module */ |
||||
void bt_setup(void) |
||||
{ |
||||
rcc_periph_clock_enable(USART_RCC); // enable USART clock
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, USART_PIN_TX); // setup GPIO pin USART transmit
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_INPUT, GPIO_CNF_INPUT_PULL_UPDOWN, USART_PIN_RX); // setup GPIO pin USART receive
|
||||
gpio_set(USART_PORT, USART_PIN_RX); // pull up to avoid noise when not connected
|
||||
|
||||
/* setup UART part */ |
||||
rcc_periph_clock_enable(USART_RCC); // enable clock for USART block
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, USART_PIN_TX); // setup GPIO pin USART transmit
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_INPUT, GPIO_CNF_INPUT_PULL_UPDOWN, USART_PIN_RX); // setup GPIO pin USART receive
|
||||
gpio_set(USART_PORT, USART_PIN_RX); // pull up to avoid noise when not connected
|
||||
usart_set_baudrate(USART, 9600); // set baudrate
|
||||
usart_set_databits(USART, 8); // set data width (most common: 8 bits)
|
||||
usart_set_stopbits(USART, USART_STOPBITS_1); // set stop bit (most common: 1 bit)
|
||||
usart_set_parity(USART, USART_PARITY_NONE); // set parity (most common: none)
|
||||
usart_set_mode(USART, USART_MODE_TX_RX); // enable USART to receive and transmit
|
||||
usart_set_flow_control(USART, USART_FLOWCONTROL_NONE); // no hardware flow control
|
||||
nvic_enable_irq(USART_IRQ); // enable the USART interrupt
|
||||
usart_enable_rx_interrupt(USART); // enable receive interrupt
|
||||
usart_enable(USART); // enable USART
|
||||
|
||||
/* reset buffer states */ |
||||
tx_i = 0; |
||||
tx_used = 0; |
||||
rx_i = 0; |
||||
rx_used = 0; |
||||
bt_received = 0; |
||||
|
||||
/* configure AT mode pin */ |
||||
rcc_periph_clock_enable(AT_RCC); // enable clock for GPIO block
|
||||
gpio_set_mode(AT_PORT, GPIO_MODE_OUTPUT_2_MHZ, GPIO_CNF_OUTPUT_PUSHPULL, AT_PIN); // set pin to control AT mode
|
||||
|
||||
/* configure HC-05 bluetooth module */ |
||||
gpio_set(AT_PORT, AT_PIN); // enable AT mode
|
||||
bool responded = false; // did the mode respond to AT command
|
||||
uint32_t baudrates[] = {115200, 38400, 9600, 57600, 19200}; // possible baudrate (order by preference)
|
||||
uint8_t baudrate_i = 0; // the detected baudrate
|
||||
for (baudrate_i = 0; baudrate_i<LENGTH(baudrates) && !responded; baudrate_i++) { // find module baudrate
|
||||
usart_set_baudrate(USART, baudrates[baudrate_i]); // set baudrate to test
|
||||
usart_send_blocking(USART, 'A'); // send AT command to test if it responded
|
||||
usart_send_blocking(USART, 'T'); // send AT command to test if it responded
|
||||
usart_send_blocking(USART, '\r'); // send AT command to test if it responded
|
||||
usart_send_blocking(USART, '\n'); // send AT command to test if it responded
|
||||
for (uint32_t i = 0; i<0xc0000; i++) { // wait 30ms for response (empiric value)
|
||||
__asm__("nop"); |
||||
} |
||||
if ((rx_used==4 && bt_getchar()=='O' && bt_getchar()=='K' && bt_getchar()=='\r' && bt_getchar()=='\n') || (rx_used==11 && bt_getchar()=='E' && bt_getchar()=='R' && bt_getchar()=='R' && bt_getchar()=='O' && bt_getchar()=='R' && bt_getchar()==':' && bt_getchar()=='(' && bt_getchar()=='0' && bt_getchar()==')' && bt_getchar()=='\r' && bt_getchar()=='\n')) { |
||||
responded = true; // seems we found the right baudrate since it responded
|
||||
} |
||||
// reset buffer states
|
||||
rx_i = 0; |
||||
rx_used = 0; |
||||
} |
||||
gpio_clear(AT_PORT, AT_PIN); // disable AT mode
|
||||
if (responded) { |
||||
printf("baudrate: %lubps\n",baudrates[--baudrate_i]); |
||||
gpio_clear(AT_PORT, AT_PIN); |
||||
} else { |
||||
printf("couldn't find baudrate\n"); |
||||
} |
||||
//usart_set_baudrate(USART, USART_BAUDRATE); // set final baudrate
|
||||
|
||||
bt_info(); |
||||
|
||||
} |
||||
|
||||
/* send character over bluetooth (blocking) */ |
||||
void bt_putchar_blocking(char c) |
||||
{ |
||||
bt_flush(); // empty buffer first
|
||||
usart_send_blocking(USART, c); // send character
|
||||
} |
||||
|
||||
/* ensure all data has been transmitted (blocking) */ |
||||
void bt_flush(void) |
||||
{ |
||||
while (tx_used) { // idle until buffer is empty
|
||||
__WFI(); // sleep until interrupt
|
||||
} |
||||
usart_wait_send_ready(USART); // wait until transmit register is empty (transmission might not be complete)
|
||||
} |
||||
|
||||
/* get character received over bluetooth (blocking) */ |
||||
char bt_getchar(void) |
||||
{ |
||||
while (!rx_used) { // idle until data is available
|
||||
__WFI(); // sleep until interrupt;
|
||||
} |
||||
char to_return = rx_buffer[rx_i]; // get the next available character
|
||||
rx_i = (rx_i+1)%sizeof(rx_buffer); // update used buffer
|
||||
rx_used--; // update used buffer
|
||||
bt_received = rx_used; // update available data
|
||||
return to_return; |
||||
} |
||||
|
||||
/* send character over bluetooth (non-blocking until buffer is full) */ |
||||
void bt_putchar_nonblocking(char c) |
||||
{ |
||||
while (tx_used>=sizeof(tx_buffer)) { // idle until buffer has some space
|
||||
usart_enable_tx_interrupt(USART); // enable transmit interrupt
|
||||
__WFI(); // sleep until something happened
|
||||
} |
||||
tx_buffer[(tx_i+tx_used)%sizeof(tx_buffer)] = c; // put character in buffer
|
||||
tx_used++; // update used buffer
|
||||
usart_enable_tx_interrupt(USART); // enable transmit interrupt
|
||||
} |
||||
|
||||
#if (USART==USART1) |
||||
void usart1_isr(void) |
||||
#elif (USART==USART2) |
||||
void usart2_isr(void) |
||||
#elif (USART==USART3) |
||||
void usart3_isr(void) |
||||
#endif |
||||
{ // USART interrupt
|
||||
if (usart_get_interrupt_source(USART, USART_SR_TXE)) { // data has been transmitted
|
||||
if (!tx_used) { // no data in the buffer to transmit
|
||||
usart_disable_tx_interrupt(USART); // disable transmit interrupt
|
||||
} else { |
||||
usart_send(USART,tx_buffer[tx_i]); // put data in transmit register
|
||||
tx_i = (tx_i+1)%sizeof(rx_buffer); // update location on buffer
|
||||
tx_used--; // update used size
|
||||
} |
||||
} |
||||
if (usart_get_interrupt_source(USART, USART_SR_RXNE)) { // data has been received
|
||||
// only save data if there is space in the buffer
|
||||
if (rx_used>=sizeof(rx_buffer)) { |
||||
usart_recv(USART); // read to clear interrupt
|
||||
} else { |
||||
rx_buffer[(rx_i+rx_used)%sizeof(rx_buffer)] = usart_recv(USART); // put character in buffer
|
||||
rx_used++; // update used buffer
|
||||
bt_received = rx_used; // update available data
|
||||
} |
||||
} |
||||
} |
@ -0,0 +1,36 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles the communication with a HC-05 bluetooth module */ |
||||
/* peripherals used: USART (check source for details) */ |
||||
#pragma once |
||||
|
||||
/* RX and TX buffer sizes */ |
||||
#define BT_BUFFER 128 |
||||
/* show the user how much received is available */ |
||||
extern volatile uint8_t bt_received; |
||||
|
||||
/* display configuration of bluetooth module */ |
||||
void bt_info(void); |
||||
/* setup communication to bluetooth module */ |
||||
void bt_setup(void); |
||||
/* send character over bluetooth (blocking) */ |
||||
void bt_putchar_blocking(char c); |
||||
/* ensure all data has been transmitted (blocking) */ |
||||
void bt_flush(void); |
||||
/* get character received over bluetooth (blocking) */ |
||||
char bt_getchar(void); |
||||
/* send character over bluetooth (non-blocking until buffer is full) */ |
||||
void bt_putchar_nonblocking(char c); |
@ -0,0 +1,113 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library is used to store (read and write) data in flash */ |
||||
/* peripherals used: none */ |
||||
|
||||
/* standard libraries */ |
||||
#include <stdint.h> // standard integer types |
||||
#include <stdlib.h> // general utilities |
||||
|
||||
/* STM32 (including CM3) libraries */ |
||||
#include <libopencm3/stm32/desig.h> // device signature utilities |
||||
#include <libopencm3/stm32/flash.h> // flash utilities |
||||
|
||||
#include "flash_storage.h" // flash storage library API |
||||
#include "global.h" // global definitions |
||||
|
||||
// the flash page size (medium-density devices have 1KiB page size)
|
||||
#define PAGE_SIZE 1024 |
||||
|
||||
/* read <size> data from address <address> into <buffer>
|
||||
* returns if read succeeded */ |
||||
bool storage_read(uint32_t address, uint8_t *buffer, size_t size) |
||||
{ |
||||
// verify it's in the storage area
|
||||
if (address<STORAGE_START || (address+size)>STORAGE_END) { |
||||
return false; |
||||
} |
||||
if (buffer==NULL || size==0) { |
||||
return false; |
||||
} |
||||
|
||||
// copy data byte per byte
|
||||
// a more efficient way would be to copy words, than the remaining bytes
|
||||
for (size_t i=0; i<size; i++) { |
||||
buffer[i] = *((uint8_t*)address+i); |
||||
} |
||||
|
||||
return true; |
||||
} |
||||
|
||||
/* write <size> data from <buffer> to address <address>
|
||||
* returns if write succeeded */ |
||||
bool storage_write(uint32_t address, uint8_t *buffer, size_t size) |
||||
{ |
||||
// verify it's in the storage area
|
||||
if (address<STORAGE_START || (address+size)>STORAGE_END) { |
||||
return false; |
||||
} |
||||
if (buffer==NULL || size==0) { |
||||
return false; |
||||
} |
||||
|
||||
uint8_t page[PAGE_SIZE]; // the complete page to write
|
||||
|
||||
flash_unlock(); // unlock flash to be able to write it
|
||||
// go through memory
|
||||
while (size) { |
||||
uint32_t page_pre = address%PAGE_SIZE; // the beginning data size in the page
|
||||
address -= page_pre; // go to beginning of the page
|
||||
storage_read(address, &page[0], page_pre); // copy existing data
|
||||
if (size>=PAGE_SIZE-page_pre) { // no need to read tailing page data
|
||||
for (uint16_t i=0; i<PAGE_SIZE-page_pre; i++) { // put buffer in page
|
||||
page[page_pre+i] = buffer[i]; |
||||
} |
||||
buffer += PAGE_SIZE-page_pre; // adjust remaining buffer
|
||||
size -= PAGE_SIZE-page_pre; // adjust remaining size
|
||||
} else { // need read tailing page data
|
||||
for (uint16_t i=0; i<size; i++) { // put buffer in page
|
||||
page[page_pre+i] = buffer[i]; |
||||
} |
||||
buffer += size; // adjust remaining buffer
|
||||
storage_read(address+page_pre+size, &page[page_pre+size], PAGE_SIZE-page_pre-size); // read tailing page data
|
||||
size = 0; // adjust remaining size
|
||||
} |
||||
|
||||
// write page
|
||||
flash_erase_page(address); // erase current page
|
||||
if (flash_get_status_flags()!=FLASH_SR_EOP) { // operation went wrong
|
||||
flash_lock(); // lock back flash to protect it
|
||||
return false; |
||||
} |
||||
for (uint16_t i=0; i<PAGE_SIZE/2; i++) { // write whole page
|
||||
flash_program_half_word(address+i*2, *((uint16_t*)page+i)); |
||||
if (flash_get_status_flags()!=FLASH_SR_EOP) { // operation went wrong
|
||||
flash_lock(); // lock back flash to protect it
|
||||
return false; |
||||
} |
||||
if (*((uint16_t*)address+i)!=*((uint16_t*)page+i)) { // verify the programmed data is right
|
||||
flash_lock(); // lock back flash to protect it
|
||||
return false; |
||||
} |
||||
} |
||||
|
||||
address += PAGE_SIZE; // go to next page
|
||||
|
||||
} |
||||
flash_lock(); // lock back flash to protect it
|
||||
|
||||
return true; |
||||
}
|
@ -0,0 +1,33 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library is used to store (read and write) data in flash */ |
||||
/* peripherals used: none */ |
||||
#include <libopencm3/stm32/desig.h> // device signature utilities |
||||
|
||||
// how much data (in bytes) should we be able to store (be sure it's available and does not overlap the firmware)
|
||||
#define STORAGE_SIZE 2048 |
||||
// the end of the flash area where to store data
|
||||
#define STORAGE_END FLASH_BASE+DESIG_FLASH_SIZE |
||||
// the start of the flash area where to store data (be sure it's after the firmware data)
|
||||
// we will only use the last kilobytes
|
||||
#define STORAGE_START STORAGE_END-STORAGE_SIZE |
||||
|
||||
/* read <size> data from address <address> into <buffer>
|
||||
* returns if read succeeded */ |
||||
bool storage_read(uint32_t address, uint8_t *buffer, size_t size); |
||||
/* write <size> data from <buffer> to address <address>
|
||||
* returns if write succeeded */ |
||||
bool storage_write(uint32_t address, uint8_t *buffer, size_t size); |
@ -0,0 +1,175 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library is used to drive a WS2812b LED chain */ |
||||
/* peripherals used: SPI , timer, DMA (check source for details) */ |
||||
|
||||
/* standard libraries */ |
||||
#include <stdint.h> // standard integer types |
||||
#include <stdlib.h> // general utilities |
||||
|
||||
/* STM32 (including CM3) libraries */ |
||||
#include <libopencm3/stm32/rcc.h> // real-time control clock library |
||||
#include <libopencm3/stm32/gpio.h> // general purpose input output library |
||||
#include <libopencm3/stm32/spi.h> // SPI library |
||||
#include <libopencm3/stm32/timer.h> // timer library |
||||
#include <libopencm3/stm32/dma.h> // DMA library |
||||
#include <libopencm3/cm3/nvic.h> // interrupt handler |
||||
#include <libopencmsis/core_cm3.h> // Cortex M3 utilities |
||||
|
||||
#include "led_ws2812b.h" // LED WS2812b library API |
||||
#include "global.h" // global definitions |
||||
|
||||
/* WS2812b peripheral configuration */ |
||||
#define WS2812B_SPI SPI1 |
||||
#define WS2812B_SPI_DR SPI1_DR |
||||
#define WS2812B_SPI_RCC RCC_SPI1 |
||||
#define WS2812B_SPI_PORT GPIOA |
||||
#define WS2812B_SPI_CLK GPIO_SPI1_SCK |
||||
#define WS2812B_SPI_DOUT GPIO_SPI1_MISO |
||||
#define WS2812B_TIMER TIM3 |
||||
#define WS2812B_TIMER_RCC RCC_TIM3 |
||||
#define WS2812B_TIMER_OC TIM_OC3 |
||||
#define WS2812B_CLK_RCC RCC_GPIOB |
||||
#define WS2812B_CLK_PORT GPIOB |
||||
#define WS2812B_CLK_PIN GPIO_TIM3_CH3 |
||||
#define WS2812B_DMA DMA1 // DMA1 supports SPI1_TX interrupt
|
||||
#define WS2812B_DMA_RCC RCC_DMA1 // follows previous definition
|
||||
#define WS2812B_DMA_CH DMA_CHANNEL3 // only DMA1 channel 3 supports SPI1_TX interrupt
|
||||
#define WS2812B_DMA_IRQ NVIC_DMA1_CHANNEL3_IRQ // follows previous definition
|
||||
#define WS2812B_DMA_ISR dma1_channel3_isr // follows previous definition
|
||||
|
||||
/* template to encode one byte
|
||||
* for each WS2812b bit which needs to be transfered we require to transfer 3 SPI bits |
||||
* the first SPI bit is the high start of the WS2812b bit frame |
||||
* the second SPI bit determines if the WS2812b bit is a 0 or 1 |
||||
* the third SPI bit is the last part of the WS2812b bit frame, which is always low |
||||
* only the first 24 bits (3*8) are used */ |
||||
#define WS2812B_SPI_TEMPLATE 0b10010010010010010010010000000000 |
||||
|
||||
uint8_t ws2812b_data[WS2812B_LEDS*3*3+40*3/8] = {0}; // SPI encode data to be shifted out for WS2812b + the 50us reset
|
||||
static volatile bool transmit_flag = false; // is transmission ongoing
|
||||
|
||||
/* set color of a single LED
|
||||
* transmission needs to be done separately */ |
||||
void ws2812b_set_rgb(uint16_t led, uint8_t red, uint8_t green, uint8_t blue) |
||||
{ |
||||
// verify the led exists
|
||||
if (led>=WS2812B_LEDS) { |
||||
return; |
||||
} |
||||
// wait for transmission to complete before changing the color
|
||||
while (transmit_flag) { |
||||
__WFI(); |
||||
} |
||||
|
||||
uint8_t colors[] = {green, red, blue}; |
||||
|
||||
// generate the pattern
|
||||
for (uint8_t color=0; color<LENGTH(colors); color++) { |
||||
uint32_t bits_color = WS2812B_SPI_TEMPLATE; // template to encode the bits in
|
||||
for (uint8_t bit=0; bit<8; bit++) { |
||||
bits_color |= ((colors[color]>>bit)&0b1)<<(bit*3+9); // encode the data bits in the pattern
|
||||
} |
||||
// store pattern
|
||||
ws2812b_data[led*3*3+color*3+0] = (bits_color>>24); |
||||
ws2812b_data[led*3*3+color*3+1] = (bits_color>>16); |
||||
ws2812b_data[led*3*3+color*3+2] = (bits_color>>8); |
||||
} |
||||
} |
||||
|
||||
/* transmit colors to LEDs */ |
||||
void ws2812b_transmit(void) |
||||
{ |
||||
while (transmit_flag) { // wait for previous transmission to complete
|
||||
__WFI(); |
||||
} |
||||
transmit_flag = true; // remember transmission started
|
||||
dma_set_memory_address(WS2812B_DMA, WS2812B_DMA_CH, (uint32_t)ws2812b_data); |
||||
dma_set_number_of_data(WS2812B_DMA, WS2812B_DMA_CH, LENGTH(ws2812b_data)); // set the size of the data to transmit
|
||||
dma_enable_transfer_complete_interrupt(WS2812B_DMA, WS2812B_DMA_CH); // warm when transfer is complete to stop transmission
|
||||
dma_enable_channel(WS2812B_DMA, WS2812B_DMA_CH); // enable DMA channel
|
||||
|
||||
spi_enable_tx_dma(WS2812B_SPI); // use DMA to provide data stream to be transfered
|
||||
|
||||
timer_set_counter(WS2812B_TIMER, 0); // reset timer counter fro clean clock
|
||||
timer_enable_counter(WS2812B_TIMER); // start timer to generate clock
|
||||
} |
||||
|
||||
/* setup WS2812b LED controller */ |
||||
void ws2812b_setup(void) |
||||
{ |
||||
/* setup timer to generate clock of (using PWM): 800kHz*3 */ |
||||
rcc_periph_clock_enable(WS2812B_CLK_RCC); // enable clock for GPIO peripheral
|
||||
gpio_set_mode(WS2812B_CLK_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, WS2812B_CLK_PIN); // set pin a output
|
||||
rcc_periph_clock_enable(RCC_AFIO); // enable clock for alternate function (PWM)
|
||||
rcc_periph_clock_enable(WS2812B_TIMER_RCC); // enable clock for timer peripheral
|
||||
timer_reset(WS2812B_TIMER); // reset timer state
|
||||
timer_set_mode(WS2812B_TIMER, TIM_CR1_CKD_CK_INT, TIM_CR1_CMS_EDGE, TIM_CR1_DIR_UP); // set timer mode, use undivided timer clock, edge alignment (simple count), and count up
|
||||
timer_set_prescaler(WS2812B_TIMER, 0); // no prescaler to keep most precise timer (72MHz/2^16=1099<800kHz)
|
||||
timer_set_period(WS2812B_TIMER, rcc_ahb_frequency/800000/3-1); // set the clock frequency to 800kHz*3bit since we need to send 3 bits to output a 800kbps stream
|
||||
timer_set_oc_value(WS2812B_TIMER, WS2812B_TIMER_OC, rcc_ahb_frequency/800000/3/2); // duty cycle to 50%
|
||||
timer_set_oc_mode(WS2812B_TIMER, WS2812B_TIMER_OC, TIM_OCM_PWM1); // set timer to generate PWM (used as clock)
|
||||
timer_enable_oc_output(WS2812B_TIMER, WS2812B_TIMER_OC); // enable output to generate the clock
|
||||
|
||||
/* setup SPI to transmit data (we are slave and the clock comes from the above PWM): 3 SPI bits for 1 WS2812b bit */ |
||||
rcc_periph_clock_enable(WS2812B_SPI_RCC); // enable clock for SPI peripheral
|
||||
gpio_set_mode(WS2812B_SPI_PORT, GPIO_MODE_INPUT, GPIO_CNF_INPUT_FLOAT, WS2812B_SPI_CLK); // set clock as input
|
||||
gpio_set_mode(WS2812B_SPI_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, WS2812B_SPI_DOUT); // set MISO as output
|
||||
spi_reset(WS2812B_SPI); // clear SPI values to default
|
||||
spi_set_slave_mode(WS2812B_SPI); // set SPI as slave (since we use the clock as input)
|
||||
spi_set_bidirectional_transmit_only_mode(WS2812B_SPI); // we won't receive data
|
||||
spi_set_unidirectional_mode(WS2812B_SPI); // we only need to transmit data
|
||||
spi_set_dff_8bit(WS2812B_SPI); // use 8 bits for simpler encoding (but there will be more interrupts)
|
||||
spi_set_clock_polarity_1(WS2812B_SPI); // clock is high when idle
|
||||
spi_set_clock_phase_1(WS2812B_SPI); // output data on second edge (rising)
|
||||
spi_send_msb_first(WS2812B_SPI); // send least significant bit first
|
||||
spi_enable_software_slave_management(WS2812B_SPI); // control the slave select in software (since there is no master)
|
||||
spi_set_nss_low(WS2812B_SPI); // set NSS low so we can output
|
||||
spi_enable(WS2812B_SPI); // enable SPI
|
||||
// do not disable SPI or set NSS high since it will put MISO high, breaking the beginning of the next transmission
|
||||
|
||||
/* configure DMA to provide the pattern to be shifted out from SPI to the WS2812b LEDs */ |
||||
rcc_periph_clock_enable(WS2812B_DMA_RCC); // enable clock for DMA peripheral
|
||||
dma_channel_reset(WS2812B_DMA, WS2812B_DMA_CH); // start with fresh channel configuration
|
||||
dma_set_memory_address(WS2812B_DMA, WS2812B_DMA_CH, (uint32_t)ws2812b_data); // set bit pattern as source address
|
||||
dma_set_peripheral_address(WS2812B_DMA, WS2812B_DMA_CH, (uint32_t)&WS2812B_SPI_DR); // set SPI as peripheral destination address
|
||||
dma_set_read_from_memory(WS2812B_DMA, WS2812B_DMA_CH); // set direction from memory to peripheral
|
||||
dma_enable_memory_increment_mode(WS2812B_DMA, WS2812B_DMA_CH); // go through bit pattern
|
||||
dma_set_memory_size(WS2812B_DMA, WS2812B_DMA_CH, DMA_CCR_MSIZE_8BIT); // read 8 bits from memory
|
||||
dma_set_peripheral_size(WS2812B_DMA, WS2812B_DMA_CH, DMA_CCR_PSIZE_8BIT); // write 8 bits to peripheral
|
||||
dma_set_priority(WS2812B_DMA, WS2812B_DMA_CH, DMA_CCR_PL_HIGH); // set priority to high since time is crucial for the peripheral
|
||||
nvic_enable_irq(WS2812B_DMA_IRQ); // enable interrupts for this DMA channel
|
||||
|
||||
// reset color
|
||||
for (uint16_t led=0; led<WS2812B_LEDS; led++) { |
||||
ws2812b_set_rgb(led, 0x00, 0x00, 0x00); // switch off (set to black)
|
||||
} |
||||
ws2812b_transmit(); // set LEDs
|
||||
} |
||||
|
||||
/* data transmission finished */ |
||||
void WS2812B_DMA_ISR(void) |
||||
{ |
||||
if (dma_get_interrupt_flag(WS2812B_DMA, WS2812B_DMA_CH, DMA_TCIF)) { // transfer completed
|
||||
dma_clear_interrupt_flags(WS2812B_DMA, WS2812B_DMA_CH, DMA_TCIF); // clear flag
|
||||
dma_disable_transfer_complete_interrupt(WS2812B_DMA, WS2812B_DMA_CH); // stop warning transfer completed
|
||||
spi_disable_tx_dma(WS2812B_SPI); // stop SPI asking for more data
|
||||
while (SPI_SR(WS2812B_SPI) & SPI_SR_BSY); // wait for data to be shifted out
|
||||
timer_disable_counter(WS2812B_TIMER); // stop clock
|
||||
dma_disable_channel(WS2812B_DMA, WS2812B_DMA_CH); // stop using DMA
|
||||
transmit_flag = false; // transmission completed
|
||||
}
|
||||
} |
@ -0,0 +1,29 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library is used to drive a WS2812b LED chain */ |
||||
/* peripherals used: SPI , timer, DMA (check source for details) */ |
||||
#pragma once |
||||
|
||||
/* number of LEDs */ |
||||
#define WS2812B_LEDS 60 |
||||
|
||||
/* set color of a single LED
|
||||
* transmission needs to be done separately */ |
||||
void ws2812b_set_rgb(uint16_t led, uint8_t red, uint8_t green, uint8_t blue); |
||||
/* transmit colors to LEDs */ |
||||
void ws2812b_transmit(void); |
||||
/* setup WS2812b LED controller */ |
||||
void ws2812b_setup(void); |
@ -0,0 +1,175 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles USART communication over pulse coded InfraRed transmission */ |
||||
/* peripherals used: USART, timer (check source for details) */ |
||||
|
||||
/* standard libraries */ |
||||
#include <stdint.h> // standard integer types |
||||
#include <stdio.h> // standard I/O facilities |
||||
#include <stdlib.h> // general utilities |
||||
|
||||
/* STM32 (including CM3) libraries */ |
||||
#include <libopencm3/stm32/rcc.h> // real-time control clock library |
||||
#include <libopencm3/stm32/gpio.h> // general purpose input output library |
||||
#include <libopencm3/stm32/usart.h> // universal synchronous asynchronous receiver transmitter library |
||||
#include <libopencm3/stm32/timer.h> // timer library |
||||
#include <libopencm3/cm3/nvic.h> // interrupt handler |
||||
#include <libopencmsis/core_cm3.h> // Cortex M3 utilities |
||||
|
||||
#include "usart_ir.h" // USART InfraRed header and definitions |
||||
|
||||
/* which USART to use */ |
||||
#define USART USART3 |
||||
#define USART_RCC RCC_USART3 |
||||
#define USART_IRQ NVIC_USART3_IRQ |
||||
#define USART_PORT GPIOB |
||||
#define USART_PIN_TX GPIO_USART3_TX |
||||
#define USART_PIN_RX GPIO_USART3_RX |
||||
|
||||
/* which timer to use to create PWM for IR modulation */ |
||||
#define TIMER TIM2 |
||||
#define TIMER_RCC RCC_TIM2 |
||||
#define TIMER_OC TIM_OC2 |
||||
#define TIMER_PORT GPIOA |
||||
#define TIMER_PORT_RCC RCC_GPIOA |
||||
#define TIMER_PIN GPIO_TIM2_CH2 |
||||
|
||||
#define USART_BAUDRATE 2400 // serial baudrate (in bits per second, with 8N1 configuration)
|
||||
#define USART_BUFFER 128 // RX and TX buffer sizes
|
||||
#define IR_MODULATION 38000 // infra-red modulation frequency
|
||||
|
||||
/* input and output ring buffer, indexes, and available memory */ |
||||
static uint8_t rx_buffer[USART_BUFFER] = {0}; |
||||
static volatile uint8_t rx_i = 0; |
||||
static volatile uint8_t rx_used = 0; |
||||
static uint8_t tx_buffer[USART_BUFFER] = {0}; |
||||
static volatile uint8_t tx_i = 0; |
||||
static volatile uint8_t tx_used = 0; |
||||
/* show the user how much data received over USART is ready */ |
||||
volatile uint8_t usart_ir_received = 0; // same as rx_used, but since the user can write this variable we don't rely on it
|
||||
|
||||
/* setup USART/IR peripheral */ |
||||
void usart_ir_setup(void) |
||||
{ |
||||
/* setup timer to generate infra-red pulse modulation (using PWM) */ |
||||
rcc_periph_clock_enable(TIMER_PORT_RCC); // enable clock for GPIO peripheral
|
||||
gpio_set_mode(TIMER_PORT, GPIO_MODE_OUTPUT_2_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, TIMER_PIN); // set pin a output
|
||||
rcc_periph_clock_enable(RCC_AFIO); // enable clock for alternate function (PWM)
|
||||
rcc_periph_clock_enable(TIMER_RCC); // enable clock for timer peripheral
|
||||
timer_reset(TIMER); // reset timer state
|
||||
timer_set_mode(TIMER, TIM_CR1_CKD_CK_INT, TIM_CR1_CMS_EDGE, TIM_CR1_DIR_UP); // set timer mode, use undivided timer clock, edge alignment (simple count), and count up
|
||||
timer_set_prescaler(TIMER, 0); // no pre-scaler to keep most precise timer (72MHz/2^16=1099Hz)
|
||||
timer_set_period(TIMER, rcc_ahb_frequency/IR_MODULATION-1+15); // set the infra-red modulation frequency (plus hand tuning)
|
||||
timer_set_oc_value(TIMER, TIMER_OC, rcc_ahb_frequency/IR_MODULATION/2-1); // duty cycle to 50%
|
||||
timer_set_oc_mode(TIMER, TIMER_OC, TIM_OCM_PWM1); // set timer to generate PWM
|
||||
timer_enable_oc_output(TIMER, TIMER_OC); // enable output to provide the modulation
|
||||
timer_enable_counter(TIMER); // start timer to generate modulation
|
||||
|
||||
/* enable USART I/O peripheral */ |
||||
rcc_periph_clock_enable(USART_RCC); // enable clock for USART peripheral
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, USART_PIN_TX); // setup GPIO pin USART transmit
|
||||
gpio_set_mode(USART_PORT, GPIO_MODE_INPUT, GPIO_CNF_INPUT_PULL_UPDOWN, USART_PIN_RX); // setup GPIO pin USART receive
|
||||
gpio_set(USART_PORT, USART_PIN_RX); // pull up to avoid noise when not connected
|
||||
|
||||
/* setup UART parameters */ |
||||
usart_set_baudrate(USART, USART_BAUDRATE); |
||||
usart_set_databits(USART, 8); |
||||
usart_set_stopbits(USART, USART_STOPBITS_1); |
||||
usart_set_mode(USART, USART_MODE_TX_RX); |
||||
usart_set_parity(USART, USART_PARITY_NONE); |
||||
usart_set_flow_control(USART, USART_FLOWCONTROL_NONE); |
||||
|
||||
nvic_enable_irq(USART_IRQ); // enable the USART interrupt
|
||||
usart_enable_rx_interrupt(USART); // enable receive interrupt
|
||||
usart_enable(USART); // enable USART
|
||||
|
||||
/* reset buffer states */ |
||||
tx_i = 0; |
||||
tx_used = 0; |
||||
rx_i = 0; |
||||
rx_used = 0; |
||||
usart_ir_received = 0; |
||||
} |
||||
|
||||
/* put character on USART/IR (blocking) */ |
||||
void usart_ir_putchar_blocking(char c) |
||||
{ |
||||
usart_ir_flush(); // empty buffer first
|
||||
usart_send_blocking(USART, c); // send character
|
||||
} |
||||
|
||||
/* ensure all data has been transmitted (blocking) */ |
||||
void usart_ir_flush(void) |
||||
{ |
||||
while (tx_used) { // idle until buffer is empty
|
||||
__WFI(); // sleep until interrupt
|
||||
} |
||||
usart_wait_send_ready(USART); // wait until transmit register is empty (transmission might not be complete)
|
||||
} |
||||
|
||||
/* get character from USART/IR (blocking) */ |
||||
char usart_ir_getchar(void) |
||||
{ |
||||
while (!rx_used) { // idle until data is available
|
||||
__WFI(); // sleep until interrupt;
|
||||
} |
||||
char to_return = rx_buffer[rx_i]; // get the next available character
|
||||
rx_i = (rx_i+1)%sizeof(rx_buffer); // update used buffer
|
||||
rx_used--; // update used buffer
|
||||
usart_ir_received = rx_used; // update available data
|
||||
return to_return; |
||||
} |
||||
|
||||
/* put character on USART/IR (non-blocking until buffer is full) */ |
||||
void usart_ir_putchar_nonblocking(char c) |
||||
{ |
||||
while (tx_used>=sizeof(tx_buffer)) { // idle until buffer has some space
|
||||
usart_enable_tx_interrupt(USART); // enable transmit interrupt
|
||||
__WFI(); // sleep until something happened
|
||||
} |
||||
tx_buffer[(tx_i+tx_used)%sizeof(tx_buffer)] = c; // put character in buffer
|
||||
tx_used++; // update used buffer
|
||||
usart_enable_tx_interrupt(USART); // enable transmit interrupt
|
||||
} |
||||
|
||||
#if (USART==USART1) |
||||
void usart1_isr(void) |
||||
#elif (USART==USART2) |
||||
void usart2_isr(void) |
||||
#elif (USART==USART3) |
||||
void usart3_isr(void) |
||||
#endif |
||||
{ // USART interrupt
|
||||
if (usart_get_interrupt_source(USART, USART_SR_TXE)) { // data has been transmitted
|
||||
if (!tx_used) { // no data in the buffer to transmit
|
||||
usart_disable_tx_interrupt(USART); // disable transmit interrupt
|
||||
} else { |
||||
usart_send(USART,tx_buffer[tx_i]); // put data in transmit register
|
||||
tx_i = (tx_i+1)%sizeof(rx_buffer); // update location on buffer
|
||||
tx_used--; // update used size
|
||||
} |
||||
} |
||||
if (usart_get_interrupt_source(USART, USART_SR_RXNE)) { // data has been received
|
||||
// only save data if there is space in the buffer
|
||||
if (rx_used>=sizeof(rx_buffer)) { |
||||
usart_recv(USART); // read to clear interrupt
|
||||
} else { |
||||
rx_buffer[(rx_i+rx_used)%sizeof(rx_buffer)] = usart_recv(USART); // put character in buffer
|
||||
rx_used++; // update used buffer
|
||||
usart_ir_received = rx_used; // update available data
|
||||
} |
||||
} |
||||
} |
@ -0,0 +1,32 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles USART communication over pulse coded InfraRed transmission */ |
||||
/* peripherals used: USART, timer (check source for details) */ |
||||
#pragma once |
||||
|
||||
/* show the user how much data has been received and is available */ |
||||
extern volatile uint8_t usart_ir_received; |
||||
|
||||
/* setup USART/IR peripheral */ |
||||
void usart_ir_setup(void); |
||||
/* put character on USART/IR (blocking) */ |
||||
void usart_ir_putchar_blocking(char c); |
||||
/* ensure all data has been transmitted (blocking) */ |
||||
void usart_ir_flush(void); |
||||
/* get character from USART/IR (blocking) */ |
||||
char usart_ir_getchar(void); |
||||
/* put character on USART/IR (non-blocking until buffer is full) */ |
||||
void usart_ir_putchar_nonblocking(char c); |
@ -0,0 +1,152 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles IrDA SIR (USART based) communication */ |
||||
/* this library handles IrDA communication */ |
||||
|
||||
/* standard libraries */ |
||||
#include <stdint.h> // standard integer types |
||||
#include <stdio.h> // standard I/O facilities |
||||
#include <stdlib.h> // general utilities |
||||
|
||||
/* STM32 (including CM3) libraries */ |
||||
#include <libopencm3/stm32/rcc.h> // real-time control clock library |
||||
#include <libopencm3/stm32/gpio.h> // general purpose input output library |
||||
#include <libopencm3/stm32/usart.h> // universal synchronous asynchronous receiver transmitter library |
||||
#include <libopencm3/cm3/nvic.h> // interrupt handler |
||||
#include <libopencmsis/core_cm3.h> // Cortex M3 utilities |
||||
|
||||
#include "usart_irda.h" // IrDA header and definitions |
||||
|
||||
/* which USART to use for IrDA */ |
||||
#define IRDA USART3 |
||||
#define IRDA_RCC RCC_USART3 |
||||
#define IRDA_IRQ NVIC_USART3_IRQ |
||||
#define IRDA_PORT GPIOB |
||||
#define IRDA_PIN_TX GPIO_USART3_TX |
||||
#define IRDA_PIN_RX GPIO_USART3_RX |
||||
/* serial baudrate, in bits per second (with 8N1 8 bits, no parity bit, 1 stop bit settings) */ |
||||
#define IRDA_BAUDRATE 9600 |
||||
/* RX and TX buffer sizes */ |
||||
#define IRDA_BUFFER 128 |
||||
|
||||
/* input and output ring buffer, indexes, and available memory */ |
||||
static uint8_t rx_buffer[IRDA_BUFFER] = {0}; |
||||
static volatile uint8_t rx_i = 0; |
||||
static volatile uint8_t rx_used = 0; |
||||
static uint8_t tx_buffer[IRDA_BUFFER] = {0}; |
||||
static volatile uint8_t tx_i = 0; |
||||
static volatile uint8_t tx_used = 0; |
||||
/* show the user how much data received over IrDA is ready */ |
||||
volatile uint8_t irda_received = 0; // same as rx_used, but since the user can write this variable we don't rely on it
|
||||
|
||||
/* setup IrDA peripheral */ |
||||
void irda_setup(void) |
||||
{ |
||||
rcc_periph_clock_enable(IRDA_RCC); // enable clock for USART/IrDA block
|
||||
gpio_set_mode(IRDA_PORT, GPIO_MODE_OUTPUT_10_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, IRDA_PIN_TX); // setup GPIO pin USART/IrDA transmit
|
||||
gpio_set_mode(IRDA_PORT, GPIO_MODE_INPUT, GPIO_CNF_INPUT_PULL_UPDOWN, IRDA_PIN_RX); // setup GPIO pin USART/IrDA receive
|
||||
gpio_set(IRDA_PORT, IRDA_PIN_RX); // pull up to avoid noise when not connected
|
||||
|
||||
/* setup UART/IrDA parameters */ |
||||
usart_set_baudrate(IRDA, IRDA_BAUDRATE); |
||||
usart_set_databits(IRDA, 8); |
||||
usart_set_stopbits(IRDA, USART_STOPBITS_1); |
||||
usart_set_mode(IRDA, USART_MODE_TX_RX); |
||||
usart_set_parity(IRDA, USART_PARITY_NONE); |
||||
usart_set_flow_control(IRDA, USART_FLOWCONTROL_NONE); |
||||
USART_CR3(IRDA) |= USART_CR3_IREN; // enable IrDA SIR ENDEC block (using IREN)
|
||||
|
||||
nvic_enable_irq(IRDA_IRQ); // enable the USART/IrDA interrupt
|
||||
usart_enable_rx_interrupt(IRDA); // enable receive interrupt
|
||||
usart_enable(IRDA); // enable USART/IrDA
|
||||
|
||||
/* reset buffer states */ |
||||
tx_i = 0; |
||||
tx_used = 0; |
||||
rx_i = 0; |
||||
rx_used = 0; |
||||
irda_received = 0; |
||||
} |
||||
|
||||
/* put character on IrDA (blocking) */ |
||||
void irda_putchar_blocking(char c) |
||||
{ |
||||
irda_flush(); // empty buffer first
|
||||
usart_send_blocking(IRDA, c); // send character
|
||||
} |
||||
|
||||
/* ensure all data has been transmitted (blocking) */ |
||||
void irda_flush(void) |
||||
{ |
||||
while (tx_used) { // idle until buffer is empty
|
||||
__WFI(); // sleep until interrupt
|
||||
} |
||||
usart_wait_send_ready(IRDA); // wait until transmit register is empty (transmission might not be complete)
|
||||
} |
||||
|
||||
/* get character from IrDA (blocking) */ |
||||
char irda_getchar(void) |
||||
{ |
||||
while (!rx_used) { // idle until data is available
|
||||
__WFI(); // sleep until interrupt;
|
||||
} |
||||
char to_return = rx_buffer[rx_i]; // get the next available character
|
||||
rx_i = (rx_i+1)%sizeof(rx_buffer); // update used buffer
|
||||
rx_used--; // update used buffer
|
||||
irda_received = rx_used; // update available data
|
||||
return to_return; |
||||
} |
||||
|
||||
/* put character on IrDA (non-blocking until buffer is full) */ |
||||
void irda_putchar_nonblocking(char c) |
||||
{ |
||||
while (tx_used>=sizeof(tx_buffer)) { // idle until buffer has some space
|
||||
usart_enable_tx_interrupt(IRDA); // enable transmit interrupt
|
||||
__WFI(); // sleep until something happened
|
||||
} |
||||
tx_buffer[(tx_i+tx_used)%sizeof(tx_buffer)] = c; // put character in buffer
|
||||
tx_used++; // update used buffer
|
||||
usart_enable_tx_interrupt(IRDA); // enable transmit interrupt
|
||||
} |
||||
|
||||
#if (IRDA==USART1) |
||||
void usart1_isr(void) |
||||
#elif (IRDA==USART2) |
||||
void usart2_isr(void) |
||||
#elif (IRDA==USART3) |
||||
void usart3_isr(void) |
||||
#endif |
||||
{ // USART interrupt
|
||||
if (usart_get_interrupt_source(IRDA, USART_SR_TXE)) { // data has been transmitted
|
||||
if (!tx_used) { // no data in the buffer to transmit
|
||||
usart_disable_tx_interrupt(IRDA); // disable transmit interrupt
|
||||
} else { |
||||
usart_send(IRDA,tx_buffer[tx_i]); // put data in transmit register
|
||||
tx_i = (tx_i+1)%sizeof(rx_buffer); // update location on buffer
|
||||
tx_used--; // update used size
|
||||
} |
||||
} |
||||
if (usart_get_interrupt_source(IRDA, USART_SR_RXNE)) { // data has been received
|
||||
// only save data if there is space in the buffer
|
||||
if (rx_used>=sizeof(rx_buffer)) { |
||||
usart_recv(IRDA); // read to clear interrupt
|
||||
} else { |
||||
rx_buffer[(rx_i+rx_used)%sizeof(rx_buffer)] = usart_recv(IRDA); // put character in buffer
|
||||
rx_used++; // update used buffer
|
||||
irda_received = rx_used; // update available data
|
||||
} |
||||
} |
||||
} |
@ -0,0 +1,31 @@ |
||||
/* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
/* Copyright (c) 2016 King Kévin <kingkevin@cuvoodoo.info> */ |
||||
/* this library handles IrDA SIR (USART based) communication */ |
||||
/* peripherals used: USART (check source for details) */ |
||||
|
||||
/* show the user how much received is available */ |
||||
|