parent
312ab6d897
commit
2b38adb74d
@ -0,0 +1,137 @@ |
||||
## === CPU settings ===
|
||||
# CPU type
|
||||
MCU = atmega328p
|
||||
# CPU frequency
|
||||
F_CPU = 16000000
|
||||
# Fuses
|
||||
LFUSE = 0xFF
|
||||
HFUSE = 0xDE
|
||||
EFUSE = 0x05
|
||||
|
||||
|
||||
## === Source files ===
|
||||
# Main C file
|
||||
MAIN = main.c
|
||||
# Extra C files in this folder
|
||||
LOCAL_SOURCE =
|
||||
|
||||
# Library directory (with C files)
|
||||
EXTRA_SOURCE_DIR = lib/
|
||||
# C files in the library directory
|
||||
EXTRA_SOURCE_FILES = uart.c
|
||||
|
||||
|
||||
## === Programmer ===
|
||||
PROGRAMMER_TYPE = arduino
|
||||
PROGRAMMER_ARGS = -b 57600 -P /dev/ttyUSB0
|
||||
|
||||
|
||||
## === C flags ===
|
||||
|
||||
CFLAGS = -std=gnu99 -mmcu=$(MCU) -DF_CPU=$(F_CPU)UL -I. -I$(EXTRA_SOURCE_DIR)
|
||||
CFLAGS += -funsigned-char -funsigned-bitfields -fpack-struct -fshort-enums
|
||||
CFLAGS += -Wall -Wno-main -Wno-strict-prototypes -Wno-comment
|
||||
CFLAGS += -g2 -Wextra -Wfatal-errors -Wno-unused-but-set-variable
|
||||
CFLAGS += -ffunction-sections -fdata-sections -Wl,--gc-sections -Wl,--relax
|
||||
# CFLAGS += -lm ## Math
|
||||
# CFLAGS += -Wl,-u,vfprintf -lprintf_flt -lm ## for floating-point printf
|
||||
# CFLAGS += -Wl,-u,vfprintf -lprintf_min ## for smaller printf
|
||||
CFLAGS_BUILD = $(CFLAGS) -Os
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
## Defined programs / locations
|
||||
CC = avr-gcc
|
||||
OBJCOPY = avr-objcopy
|
||||
OBJDUMP = avr-objdump
|
||||
AVRSIZE = avr-size
|
||||
AVRDUDE = avrdude
|
||||
|
||||
## === File lists ===
|
||||
TARGET = $(strip $(basename $(MAIN)))
|
||||
SRC1 = $(TARGET).c
|
||||
SRC = $(SRC1)
|
||||
EXTRA_SOURCE = $(addprefix $(EXTRA_SOURCE_DIR), $(EXTRA_SOURCE_FILES))
|
||||
SRC += $(EXTRA_SOURCE)
|
||||
SRC += $(LOCAL_SOURCE)
|
||||
|
||||
HEADERS = $(SRC:.c=.h)
|
||||
OBJ = $(SRC:.c=.o)
|
||||
|
||||
|
||||
## === File generation ===
|
||||
all: $(TARGET).hex size |
||||
pre: $(TARGET).pre |
||||
|
||||
%.hex: %.elf |
||||
$(OBJCOPY) -R .eeprom -O ihex $< $@
|
||||
|
||||
%.elf: $(SRC) |
||||
$(CC) $(CFLAGS_BUILD) $(SRC) --output $@
|
||||
|
||||
%.pre: $(SRC1) |
||||
$(CC) $(CFLAGS) -E $(SRC1) --output $@
|
||||
|
||||
%.eeprom: %.elf |
||||
$(OBJCOPY) -j .eeprom --change-section-lma .eeprom=0 -O ihex $< $@
|
||||
|
||||
%.lst: %.elf |
||||
$(OBJDUMP) -S $< > $@
|
||||
|
||||
# Show debug info
|
||||
debug: |
||||
@echo
|
||||
@echo "Source files:" $(SRC)
|
||||
@echo "MCU, F_CPU, BAUD:" $(MCU), $(F_CPU), $(BAUD)
|
||||
@echo
|
||||
|
||||
|
||||
# Disassemble the ELF
|
||||
disassemble: $(TARGET).lst |
||||
dis: disassemble |
||||
lst: disassemble |
||||
|
||||
# Make eeprom file
|
||||
eeprom: $(TARGET).eeprom |
||||
|
||||
# Show how big the resulting program is
|
||||
size: $(TARGET).elf |
||||
$(AVRSIZE) -C --mcu=$(MCU) $(TARGET).elf
|
||||
|
||||
# Clean all produced trash
|
||||
clean: |
||||
rm -f $(TARGET).elf $(TARGET).hex $(TARGET).obj \
|
||||
$(TARGET).o $(TARGET).d $(TARGET).eep $(TARGET).lst \
|
||||
$(TARGET).lss $(TARGET).sym $(TARGET).map $(TARGET)~ \
|
||||
$(TARGET).eeprom
|
||||
|
||||
# Clean all trash
|
||||
purge: |
||||
rm -f *.elf *.hex *.obj *.o *.d *.eep *.lst *.lss *.sym *.map *~
|
||||
|
||||
|
||||
## === avrdude ===
|
||||
|
||||
flash: $(TARGET).hex |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -U flash:w:$<
|
||||
|
||||
flashe: $(TARGET).eeprom |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -U eeprom:w:$<
|
||||
|
||||
shell: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -nt
|
||||
|
||||
|
||||
# === fuses ===
|
||||
|
||||
FUSE_STRING = -U lfuse:w:$(LFUSE):m -U hfuse:w:$(HFUSE):m -U efuse:w:$(EFUSE):m
|
||||
|
||||
fuses: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) \
|
||||
$(PROGRAMMER_ARGS) $(FUSE_STRING)
|
||||
show_fuses: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -nv
|
||||
|
||||
set_default_fuses: FUSE_STRING = -U lfuse:w:$(LFUSE):m -U hfuse:w:$(HFUSE):m -U efuse:w:$(EFUSE):m |
||||
set_default_fuses: fuses |
||||
@ -0,0 +1,228 @@ |
||||
#include <avr/io.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <util/delay.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
#include <stdlib.h> |
||||
|
||||
#include "uart.h" |
||||
|
||||
|
||||
void _uart_init_do(uint16_t ubrr) { |
||||
/*Set baud rate */ |
||||
UBRR0H = (uint8_t) (ubrr >> 8); |
||||
UBRR0L = (uint8_t) ubrr; |
||||
|
||||
// Enable Rx and Tx
|
||||
UCSR0B = (1 << RXEN0) | (1 << TXEN0); |
||||
|
||||
// 8-bit data, 1 stop bit
|
||||
UCSR0C = (0b11 << UCSZ00); |
||||
} |
||||
|
||||
|
||||
/** Enable or disable RX ISR */ |
||||
void uart_isr_rx(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << RXCIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << RXCIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Enable or disable TX ISR (1 byte is sent) */ |
||||
void uart_isr_tx(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << TXCIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << TXCIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Enable or disable DRE ISR (all is sent) */ |
||||
void uart_isr_dre(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << UDRIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << UDRIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send byte over UART */ |
||||
void uart_tx(uint8_t data) |
||||
{ |
||||
// Wait for transmit buffer
|
||||
while (!uart_tx_ready()); |
||||
// send it
|
||||
UDR0 = data; |
||||
} |
||||
|
||||
|
||||
/** Receive one byte over UART */ |
||||
uint8_t uart_rx() |
||||
{ |
||||
// Wait for data to be received
|
||||
while (!uart_rx_ready()); |
||||
// Get and return received data from buffer
|
||||
return UDR0; |
||||
} |
||||
|
||||
|
||||
/** Send string over UART */ |
||||
void uart_puts(const char* str) |
||||
{ |
||||
while (*str) { |
||||
uart_tx(*str++); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send progmem string over UART */ |
||||
void uart_puts_pgm(const char* str) |
||||
{ |
||||
char c; |
||||
while ((c = pgm_read_byte(str++))) { |
||||
uart_tx(c); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Clear receive buffer */ |
||||
void uart_flush() |
||||
{ |
||||
uint8_t dummy; |
||||
while (UCSR0A & (1 << RXC0)) |
||||
dummy = UDR0; |
||||
} |
||||
|
||||
|
||||
/** Send CRLF */ |
||||
void uart_nl() |
||||
{ |
||||
uart_tx(13); |
||||
uart_tx(10); |
||||
} |
||||
|
||||
|
||||
char tmpstr[12]; // buffer for number rendering
|
||||
|
||||
void _uart_putnf(const uint8_t places); |
||||
|
||||
|
||||
/** Send signed int8 */ |
||||
void uart_putu(const uint8_t num) |
||||
{ |
||||
utoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} |
||||
|
||||
|
||||
/** Send unsigned int8 */ |
||||
void uart_putn(const int8_t num) |
||||
{ |
||||
itoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} |
||||
|
||||
|
||||
/** Send unsigned int as float */ |
||||
void uart_putiu(const uint16_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
utoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
utoa(num, tmpstr, 10); |
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send signed int as float */ |
||||
void uart_puti(const int16_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
itoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
if (num < 0) { |
||||
uart_tx('-'); |
||||
itoa(-num, tmpstr, 10); |
||||
} else { |
||||
itoa(num, tmpstr, 10); |
||||
} |
||||
|
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send unsigned long as float */ |
||||
void uart_putlu(const uint32_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
ultoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
ultoa(num, tmpstr, 10); |
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send signed long as float */ |
||||
void uart_putl(const int32_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
ltoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
if (num < 0) { |
||||
uart_tx('-'); |
||||
ltoa(-num, tmpstr, 10); |
||||
} else { |
||||
ltoa(num, tmpstr, 10); |
||||
} |
||||
|
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Print number in tmp string as float with given decimal point position */ |
||||
void _uart_putnf(const uint8_t places) |
||||
{ |
||||
// measure text length
|
||||
uint8_t len = 0; |
||||
while(tmpstr[len] != 0) len++; |
||||
|
||||
int8_t at = len - places; |
||||
|
||||
// print virtual zeros
|
||||
if (at <= 0) { |
||||
uart_tx('0'); |
||||
uart_tx('.'); |
||||
while(at <= -1) { |
||||
uart_tx('0'); |
||||
at++; |
||||
} |
||||
at = -1; |
||||
} |
||||
|
||||
// print the number
|
||||
uint8_t i = 0; |
||||
while(i < len) { |
||||
if (at-- == 0) { |
||||
uart_tx('.'); |
||||
} |
||||
|
||||
uart_tx(tmpstr[i++]); |
||||
} |
||||
} |
||||
|
||||
@ -0,0 +1,88 @@ |
||||
#pragma once |
||||
|
||||
//
|
||||
// Utilities for UART communication.
|
||||
//
|
||||
// First, init uart with desired baud rate using uart_init(baud).
|
||||
// Then enable interrupts you want with uart_isr_XXX().
|
||||
//
|
||||
|
||||
#include <avr/io.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <util/delay.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
|
||||
|
||||
/** Init UART for given baudrate */ |
||||
void _uart_init_do(uint16_t ubrr); // internal, needed for the macro.
|
||||
#define uart_init(baud) _uart_init_do(F_CPU / 16 / (baud) - 1) |
||||
|
||||
/** Check if there's a byte in the RX register */ |
||||
#define uart_rx_ready() (0 != (UCSR0A & (1 << RXC0))) |
||||
|
||||
/** Check if transmission of everything is done */ |
||||
#define uart_tx_ready() (0 != (UCSR0A & (1 << UDRE0))) |
||||
|
||||
|
||||
|
||||
// Enable UART interrupts
|
||||
|
||||
/** Enable or disable RX ISR */ |
||||
void uart_isr_rx(bool enable); |
||||
|
||||
/** Enable or disable TX ISR (1 byte is sent) */ |
||||
void uart_isr_tx(bool enable); |
||||
|
||||
/** Enable or disable DRE ISR (all is sent) */ |
||||
void uart_isr_dre(bool enable); |
||||
|
||||
|
||||
|
||||
// Basic IO
|
||||
|
||||
/** Receive one byte over UART */ |
||||
uint8_t uart_rx(); |
||||
|
||||
/** Send byte over UART */ |
||||
#define uart_putc(data) uart_tx((data)) |
||||
void uart_tx(uint8_t data); |
||||
|
||||
/** Clear receive buffer */ |
||||
void uart_flush(); |
||||
|
||||
|
||||
|
||||
// Strings
|
||||
|
||||
/** Send string over UART */ |
||||
void uart_puts(const char* str); |
||||
|
||||
/** Send progmem string over UART */ |
||||
void uart_puts_pgm(const char* str); |
||||
|
||||
|
||||
// Numbers
|
||||
|
||||
/** Send unsigned int */ |
||||
void uart_putn(const int8_t num); |
||||
|
||||
/** Send signed int */ |
||||
void uart_putu(const uint8_t num); |
||||
|
||||
/** Send unsigned int */ |
||||
void uart_puti(const int16_t num, const uint8_t places); |
||||
|
||||
/** Send signed int */ |
||||
void uart_putiu(const uint16_t num, const uint8_t places); |
||||
|
||||
/** Send unsigned long */ |
||||
void uart_putlu(const uint32_t num, const uint8_t places); |
||||
|
||||
/** Send signed long */ |
||||
void uart_putl(const int32_t num, const uint8_t places); |
||||
|
||||
// Extras
|
||||
|
||||
/** Send CRLF */ |
||||
void uart_nl(); |
||||
@ -0,0 +1,32 @@ |
||||
#include <avr/io.h> |
||||
#include <util/delay.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <avr/interrupt.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
#include <stdlib.h> |
||||
|
||||
#include "lib/uart.h" |
||||
|
||||
|
||||
ISR(USART_RX_vect) |
||||
{ |
||||
uint8_t c = uart_rx(); |
||||
|
||||
uart_putc(c); |
||||
uart_putc(' '); |
||||
uart_putn(c); |
||||
uart_nl(); |
||||
} |
||||
|
||||
|
||||
void main() |
||||
{ |
||||
uart_init(9600); |
||||
uart_isr_rx(1); |
||||
sei(); |
||||
|
||||
uart_puts_pgm(PSTR("UART key analyzer\r\n")); |
||||
|
||||
while(1); |
||||
} |
||||
@ -0,0 +1,137 @@ |
||||
## === CPU settings ===
|
||||
# CPU type
|
||||
MCU = atmega328p
|
||||
# CPU frequency
|
||||
F_CPU = 16000000
|
||||
# Fuses
|
||||
LFUSE = 0xFF
|
||||
HFUSE = 0xDE
|
||||
EFUSE = 0x05
|
||||
|
||||
|
||||
## === Source files ===
|
||||
# Main C file
|
||||
MAIN = main.c
|
||||
# Extra C files in this folder
|
||||
LOCAL_SOURCE =
|
||||
|
||||
# Library directory (with C files)
|
||||
EXTRA_SOURCE_DIR = lib/
|
||||
# C files in the library directory
|
||||
EXTRA_SOURCE_FILES = uart.c uart_ansi.c
|
||||
|
||||
|
||||
## === Programmer ===
|
||||
PROGRAMMER_TYPE = arduino
|
||||
PROGRAMMER_ARGS = -b 57600 -P /dev/ttyUSB0
|
||||
|
||||
|
||||
## === C flags ===
|
||||
|
||||
CFLAGS = -std=gnu99 -mmcu=$(MCU) -DF_CPU=$(F_CPU)UL -I. -I$(EXTRA_SOURCE_DIR)
|
||||
CFLAGS += -funsigned-char -funsigned-bitfields -fpack-struct -fshort-enums
|
||||
CFLAGS += -Wall -Wno-main -Wno-strict-prototypes -Wno-comment
|
||||
CFLAGS += -g2 -Wextra -Wfatal-errors -Wno-unused-but-set-variable
|
||||
CFLAGS += -ffunction-sections -fdata-sections -Wl,--gc-sections -Wl,--relax
|
||||
# CFLAGS += -lm ## Math
|
||||
# CFLAGS += -Wl,-u,vfprintf -lprintf_flt -lm ## for floating-point printf
|
||||
# CFLAGS += -Wl,-u,vfprintf -lprintf_min ## for smaller printf
|
||||
CFLAGS_BUILD = $(CFLAGS) -Os
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
## Defined programs / locations
|
||||
CC = avr-gcc
|
||||
OBJCOPY = avr-objcopy
|
||||
OBJDUMP = avr-objdump
|
||||
AVRSIZE = avr-size
|
||||
AVRDUDE = avrdude
|
||||
|
||||
## === File lists ===
|
||||
TARGET = $(strip $(basename $(MAIN)))
|
||||
SRC1 = $(TARGET).c
|
||||
SRC = $(SRC1)
|
||||
EXTRA_SOURCE = $(addprefix $(EXTRA_SOURCE_DIR), $(EXTRA_SOURCE_FILES))
|
||||
SRC += $(EXTRA_SOURCE)
|
||||
SRC += $(LOCAL_SOURCE)
|
||||
|
||||
HEADERS = $(SRC:.c=.h)
|
||||
OBJ = $(SRC:.c=.o)
|
||||
|
||||
|
||||
## === File generation ===
|
||||
all: $(TARGET).hex size |
||||
pre: $(TARGET).pre |
||||
|
||||
%.hex: %.elf |
||||
$(OBJCOPY) -R .eeprom -O ihex $< $@
|
||||
|
||||
%.elf: $(SRC) |
||||
$(CC) $(CFLAGS_BUILD) $(SRC) --output $@
|
||||
|
||||
%.pre: $(SRC1) |
||||
$(CC) $(CFLAGS) -E $(SRC1) --output $@
|
||||
|
||||
%.eeprom: %.elf |
||||
$(OBJCOPY) -j .eeprom --change-section-lma .eeprom=0 -O ihex $< $@
|
||||
|
||||
%.lst: %.elf |
||||
$(OBJDUMP) -S $< > $@
|
||||
|
||||
# Show debug info
|
||||
debug: |
||||
@echo
|
||||
@echo "Source files:" $(SRC)
|
||||
@echo "MCU, F_CPU, BAUD:" $(MCU), $(F_CPU), $(BAUD)
|
||||
@echo
|
||||
|
||||
|
||||
# Disassemble the ELF
|
||||
disassemble: $(TARGET).lst |
||||
dis: disassemble |
||||
lst: disassemble |
||||
|
||||
# Make eeprom file
|
||||
eeprom: $(TARGET).eeprom |
||||
|
||||
# Show how big the resulting program is
|
||||
size: $(TARGET).elf |
||||
$(AVRSIZE) -C --mcu=$(MCU) $(TARGET).elf
|
||||
|
||||
# Clean all produced trash
|
||||
clean: |
||||
rm -f $(TARGET).elf $(TARGET).hex $(TARGET).obj \
|
||||
$(TARGET).o $(TARGET).d $(TARGET).eep $(TARGET).lst \
|
||||
$(TARGET).lss $(TARGET).sym $(TARGET).map $(TARGET)~ \
|
||||
$(TARGET).eeprom
|
||||
|
||||
# Clean all trash
|
||||
purge: |
||||
rm -f *.elf *.hex *.obj *.o *.d *.eep *.lst *.lss *.sym *.map *~
|
||||
|
||||
|
||||
## === avrdude ===
|
||||
|
||||
flash: $(TARGET).hex |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -U flash:w:$<
|
||||
|
||||
flashe: $(TARGET).eeprom |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -U eeprom:w:$<
|
||||
|
||||
shell: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -nt
|
||||
|
||||
|
||||
# === fuses ===
|
||||
|
||||
FUSE_STRING = -U lfuse:w:$(LFUSE):m -U hfuse:w:$(HFUSE):m -U efuse:w:$(EFUSE):m
|
||||
|
||||
fuses: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) \
|
||||
$(PROGRAMMER_ARGS) $(FUSE_STRING)
|
||||
show_fuses: |
||||
$(AVRDUDE) -c $(PROGRAMMER_TYPE) -p $(MCU) $(PROGRAMMER_ARGS) -nv
|
||||
|
||||
set_default_fuses: FUSE_STRING = -U lfuse:w:$(LFUSE):m -U hfuse:w:$(HFUSE):m -U efuse:w:$(EFUSE):m |
||||
set_default_fuses: fuses |
||||
@ -0,0 +1,228 @@ |
||||
#include <avr/io.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <util/delay.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
#include <stdlib.h> |
||||
|
||||
#include "uart.h" |
||||
|
||||
|
||||
void _uart_init_do(uint16_t ubrr) { |
||||
/*Set baud rate */ |
||||
UBRR0H = (uint8_t) (ubrr >> 8); |
||||
UBRR0L = (uint8_t) ubrr; |
||||
|
||||
// Enable Rx and Tx
|
||||
UCSR0B = (1 << RXEN0) | (1 << TXEN0); |
||||
|
||||
// 8-bit data, 1 stop bit
|
||||
UCSR0C = (0b11 << UCSZ00); |
||||
} |
||||
|
||||
|
||||
/** Enable or disable RX ISR */ |
||||
void uart_isr_rx(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << RXCIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << RXCIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Enable or disable TX ISR (1 byte is sent) */ |
||||
void uart_isr_tx(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << TXCIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << TXCIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Enable or disable DRE ISR (all is sent) */ |
||||
void uart_isr_dre(bool yes) |
||||
{ |
||||
if(yes) { |
||||
UCSR0B |= (1 << UDRIE0); |
||||
} else { |
||||
UCSR0B &= ~(1 << UDRIE0); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send byte over UART */ |
||||
void uart_tx(uint8_t data) |
||||
{ |
||||
// Wait for transmit buffer
|
||||
while (!uart_tx_ready()); |
||||
// send it
|
||||
UDR0 = data; |
||||
} |
||||
|
||||
|
||||
/** Receive one byte over UART */ |
||||
uint8_t uart_rx() |
||||
{ |
||||
// Wait for data to be received
|
||||
while (!uart_rx_ready()); |
||||
// Get and return received data from buffer
|
||||
return UDR0; |
||||
} |
||||
|
||||
|
||||
/** Send string over UART */ |
||||
void uart_puts(const char* str) |
||||
{ |
||||
while (*str) { |
||||
uart_tx(*str++); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send progmem string over UART */ |
||||
void uart_puts_pgm(const char* str) |
||||
{ |
||||
char c; |
||||
while ((c = pgm_read_byte(str++))) { |
||||
uart_tx(c); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Clear receive buffer */ |
||||
void uart_flush() |
||||
{ |
||||
uint8_t dummy; |
||||
while (UCSR0A & (1 << RXC0)) |
||||
dummy = UDR0; |
||||
} |
||||
|
||||
|
||||
/** Send CRLF */ |
||||
void uart_nl() |
||||
{ |
||||
uart_tx(13); |
||||
uart_tx(10); |
||||
} |
||||
|
||||
|
||||
char tmpstr[12]; // buffer for number rendering
|
||||
|
||||
void _uart_putnf(const uint8_t places); |
||||
|
||||
|
||||
/** Send signed int8 */ |
||||
void uart_putu(const uint8_t num) |
||||
{ |
||||
utoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} |
||||
|
||||
|
||||
/** Send unsigned int8 */ |
||||
void uart_putn(const int8_t num) |
||||
{ |
||||
itoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} |
||||
|
||||
|
||||
/** Send unsigned int as float */ |
||||
void uart_putiu(const uint16_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
utoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
utoa(num, tmpstr, 10); |
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send signed int as float */ |
||||
void uart_puti(const int16_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
itoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
if (num < 0) { |
||||
uart_tx('-'); |
||||
itoa(-num, tmpstr, 10); |
||||
} else { |
||||
itoa(num, tmpstr, 10); |
||||
} |
||||
|
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send unsigned long as float */ |
||||
void uart_putlu(const uint32_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
ultoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
ultoa(num, tmpstr, 10); |
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send signed long as float */ |
||||
void uart_putl(const int32_t num, const uint8_t places) |
||||
{ |
||||
if (!places) { |
||||
ltoa(num, tmpstr, 10); |
||||
uart_puts(tmpstr); |
||||
} else { |
||||
if (num < 0) { |
||||
uart_tx('-'); |
||||
ltoa(-num, tmpstr, 10); |
||||
} else { |
||||
ltoa(num, tmpstr, 10); |
||||
} |
||||
|
||||
_uart_putnf(places); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Print number in tmp string as float with given decimal point position */ |
||||
void _uart_putnf(const uint8_t places) |
||||
{ |
||||
// measure text length
|
||||
uint8_t len = 0; |
||||
while(tmpstr[len] != 0) len++; |
||||
|
||||
int8_t at = len - places; |
||||
|
||||
// print virtual zeros
|
||||
if (at <= 0) { |
||||
uart_tx('0'); |
||||
uart_tx('.'); |
||||
while(at <= -1) { |
||||
uart_tx('0'); |
||||
at++; |
||||
} |
||||
at = -1; |
||||
} |
||||
|
||||
// print the number
|
||||
uint8_t i = 0; |
||||
while(i < len) { |
||||
if (at-- == 0) { |
||||
uart_tx('.'); |
||||
} |
||||
|
||||
uart_tx(tmpstr[i++]); |
||||
} |
||||
} |
||||
|
||||
@ -0,0 +1,88 @@ |
||||
#pragma once |
||||
|
||||
//
|
||||
// Utilities for UART communication.
|
||||
//
|
||||
// First, init uart with desired baud rate using uart_init(baud).
|
||||
// Then enable interrupts you want with uart_isr_XXX().
|
||||
//
|
||||
|
||||
#include <avr/io.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <util/delay.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
|
||||
|
||||
/** Init UART for given baudrate */ |
||||
void _uart_init_do(uint16_t ubrr); // internal, needed for the macro.
|
||||
#define uart_init(baud) _uart_init_do(F_CPU / 16 / (baud) - 1) |
||||
|
||||
/** Check if there's a byte in the RX register */ |
||||
#define uart_rx_ready() (0 != (UCSR0A & (1 << RXC0))) |
||||
|
||||
/** Check if transmission of everything is done */ |
||||
#define uart_tx_ready() (0 != (UCSR0A & (1 << UDRE0))) |
||||
|
||||
|
||||
|
||||
// Enable UART interrupts
|
||||
|
||||
/** Enable or disable RX ISR */ |
||||
void uart_isr_rx(bool enable); |
||||
|
||||
/** Enable or disable TX ISR (1 byte is sent) */ |
||||
void uart_isr_tx(bool enable); |
||||
|
||||
/** Enable or disable DRE ISR (all is sent) */ |
||||
void uart_isr_dre(bool enable); |
||||
|
||||
|
||||
|
||||
// Basic IO
|
||||
|
||||
/** Receive one byte over UART */ |
||||
uint8_t uart_rx(); |
||||
|
||||
/** Send byte over UART */ |
||||
#define uart_putc(data) uart_tx((data)) |
||||
void uart_tx(uint8_t data); |
||||
|
||||
/** Clear receive buffer */ |
||||
void uart_flush(); |
||||
|
||||
|
||||
|
||||
// Strings
|
||||
|
||||
/** Send string over UART */ |
||||
void uart_puts(const char* str); |
||||
|
||||
/** Send progmem string over UART */ |
||||
void uart_puts_pgm(const char* str); |
||||
|
||||
|
||||
// Numbers
|
||||
|
||||
/** Send unsigned int */ |
||||
void uart_putn(const int8_t num); |
||||
|
||||
/** Send signed int */ |
||||
void uart_putu(const uint8_t num); |
||||
|
||||
/** Send unsigned int */ |
||||
void uart_puti(const int16_t num, const uint8_t places); |
||||
|
||||
/** Send signed int */ |
||||
void uart_putiu(const uint16_t num, const uint8_t places); |
||||
|
||||
/** Send unsigned long */ |
||||
void uart_putlu(const uint32_t num, const uint8_t places); |
||||
|
||||
/** Send signed long */ |
||||
void uart_putl(const int32_t num, const uint8_t places); |
||||
|
||||
// Extras
|
||||
|
||||
/** Send CRLF */ |
||||
void uart_nl(); |
||||
@ -0,0 +1,580 @@ |
||||
#include <avr/io.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <avr/interrupt.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
|
||||
#include "uart.h" |
||||
#include "uart_ansi.h" |
||||
|
||||
void _vt_apply_style(); |
||||
void _vt_reset_attribs_do(); |
||||
void _vt_style_do(); |
||||
void _vt_color_do(); |
||||
|
||||
|
||||
void vt_goto(uint8_t x, uint8_t y) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(x); |
||||
uart_putc(';'); |
||||
uart_putu(y); |
||||
uart_putc('H'); |
||||
} |
||||
|
||||
|
||||
void vt_goto_x(uint8_t x) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(x); |
||||
uart_putc('`'); |
||||
} |
||||
|
||||
|
||||
void vt_goto_y(uint8_t y) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(y); |
||||
uart_putc('d'); |
||||
} |
||||
|
||||
|
||||
void vt_move(int8_t x, int8_t y) |
||||
{ |
||||
vt_move_x(x); |
||||
vt_move_y(y); |
||||
} |
||||
|
||||
|
||||
void vt_move_x(int8_t x) |
||||
{ |
||||
if (x < 0) { |
||||
vt_left(-x); |
||||
} else { |
||||
vt_right(x); |
||||
} |
||||
} |
||||
|
||||
|
||||
void vt_move_y(int8_t y) |
||||
{ |
||||
if (y < 0) { |
||||
vt_up(-y); |
||||
} else { |
||||
vt_down(y); |
||||
} |
||||
} |
||||
|
||||
|
||||
void vt_up(uint8_t y) |
||||
{ |
||||
if (y == 0) return; |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(y); |
||||
uart_putc('A'); |
||||
} |
||||
|
||||
|
||||
void vt_down(uint8_t y) |
||||
{ |
||||
if (y == 0) return; |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(y); |
||||
uart_putc('B'); |
||||
} |
||||
|
||||
|
||||
void vt_left(uint8_t x) |
||||
{ |
||||
if (x == 0) return; |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(x); |
||||
uart_putc('D'); |
||||
} |
||||
|
||||
|
||||
void vt_right(uint8_t x) |
||||
{ |
||||
if (x == 0) return; |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(x); |
||||
uart_putc('C'); |
||||
} |
||||
|
||||
|
||||
void vt_scroll(int8_t y) |
||||
{ |
||||
while (y < 0) { |
||||
uart_putc(27); |
||||
uart_putc('D'); // up
|
||||
y++; |
||||
} |
||||
|
||||
while (y > 0) { |
||||
uart_putc(27); |
||||
uart_putc('M'); // down
|
||||
y--; |
||||
} |
||||
} |
||||
|
||||
|
||||
void vt_scroll_set(uint8_t from, uint8_t to) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(from); |
||||
uart_putc(';'); |
||||
uart_putu(to); |
||||
uart_putc('r'); |
||||
} |
||||
|
||||
|
||||
void vt_scroll_reset() |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putc('r'); |
||||
} |
||||
|
||||
|
||||
|
||||
typedef struct { |
||||
uint8_t flags; |
||||
uint8_t fg; |
||||
uint8_t bg; |
||||
} vt_style_t; |
||||
|
||||
vt_style_t saved_style; |
||||
vt_style_t current_style; |
||||
|
||||
void vt_save() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[s")); |
||||
|
||||
saved_style = current_style; |
||||
} |
||||
|
||||
|
||||
void vt_restore() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[u")); |
||||
|
||||
current_style = saved_style; |
||||
} |
||||
|
||||
|
||||
/** Disable all text attributes (excluding color) */ |
||||
void vt_attr_reset() |
||||
{ |
||||
current_style.flags = 0; |
||||
|
||||
_vt_reset_attribs_do(); |
||||
_vt_apply_style(); |
||||
} |
||||
|
||||
|
||||
/** Set color to white on black */ |
||||
void vt_color_reset() |
||||
{ |
||||
current_style.fg = VT_WHITE; |
||||
current_style.bg = VT_BLACK; |
||||
|
||||
_vt_color_do(); |
||||
} |
||||
|
||||
|
||||
/** Enable or disable a text attribute */ |
||||
void vt_attr(uint8_t attribute, bool on) |
||||
{ |
||||
// flags are powers of two
|
||||
// so this can handle multiple OR'd flags
|
||||
for(uint8_t c = 1; c <= VT_FAINT; c *= 2) { |
||||
if (attribute & c) { |
||||
if (on) { |
||||
current_style.flags |= c; |
||||
} else { |
||||
current_style.flags &= ~c; |
||||
} |
||||
} |
||||
} |
||||
|
||||
_vt_apply_style(); |
||||
} |
||||
|
||||
|
||||
/** Send style and color commands */ |
||||
void _vt_apply_style() |
||||
{ |
||||
_vt_reset_attribs_do(); |
||||
_vt_style_do(); |
||||
_vt_color_do(); |
||||
} |
||||
|
||||
|
||||
/** Set color 0..7 */ |
||||
void vt_color(uint8_t fg, uint8_t bg) |
||||
{ |
||||
current_style.fg = fg; |
||||
current_style.bg = bg; |
||||
_vt_color_do(); |
||||
} |
||||
|
||||
|
||||
/** Set FG color 0..7 */ |
||||
void vt_color_fg(uint8_t fg) |
||||
{ |
||||
current_style.fg = fg; |
||||
_vt_color_do(); |
||||
} |
||||
|
||||
|
||||
/** Set BG color 0..7 */ |
||||
void vt_color_bg(uint8_t bg) |
||||
{ |
||||
current_style.bg = bg; |
||||
_vt_color_do(); |
||||
} |
||||
|
||||
|
||||
/** Send reset command */ |
||||
inline void _vt_reset_attribs_do() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[m")); // reset
|
||||
} |
||||
|
||||
|
||||
/** Send commands for text attribs */ |
||||
void _vt_style_do() |
||||
{ |
||||
if (current_style.flags & VT_BOLD) { |
||||
uart_puts_pgm(PSTR("\x1B[1m")); |
||||
} |
||||
|
||||
if (current_style.flags & VT_FAINT) { |
||||
uart_puts_pgm(PSTR("\x1B[2m")); |
||||
} |
||||
|
||||
if (current_style.flags & VT_ITALIC) { |
||||
uart_puts_pgm(PSTR("\x1B[3m")); |
||||
} |
||||
|
||||
if (current_style.flags & VT_UNDERLINE) { |
||||
uart_puts_pgm(PSTR("\x1B[4m")); |
||||
} |
||||
|
||||
if (current_style.flags & VT_BLINK) { |
||||
uart_puts_pgm(PSTR("\x1B[5m")); |
||||
} |
||||
|
||||
if (current_style.flags & VT_REVERSE) { |
||||
uart_puts_pgm(PSTR("\x1B[7m")); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** Send commands for xolor */ |
||||
void _vt_color_do() |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(30 + current_style.fg); |
||||
uart_putc(';'); |
||||
uart_putu(40 + current_style.bg); |
||||
uart_putc('m'); |
||||
} |
||||
|
||||
|
||||
/** Insert blank lines febore the current line */ |
||||
void vt_insert_lines(uint8_t count) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(count); |
||||
uart_putc('L'); |
||||
} |
||||
|
||||
|
||||
/** Delete lines from the current line down */ |
||||
void vt_delete_lines(uint8_t count) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(count); |
||||
uart_putc('M'); |
||||
} |
||||
|
||||
|
||||
/** Insert empty characters at cursor */ |
||||
void vt_insert_chars(uint8_t count) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(count); |
||||
uart_putc('@'); |
||||
} |
||||
|
||||
|
||||
/** Delete characters at cursor */ |
||||
void vt_delete_chars(uint8_t count) |
||||
{ |
||||
uart_putc(27); |
||||
uart_putc('['); |
||||
uart_putu(count); |
||||
uart_putc('P'); |
||||
} |
||||
|
||||
|
||||
void vt_clear() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[2J")); |
||||
} |
||||
|
||||
|
||||
void vt_erase_forth() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[K")); |
||||
} |
||||
|
||||
|
||||
void vt_erase_back() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[1K")); |
||||
} |
||||
|
||||
|
||||
void vt_erase_line() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[2K")); |
||||
} |
||||
|
||||
|
||||
void vt_erase_above() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[1J")); |
||||
} |
||||
|
||||
void vt_erase_below() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[J")); |
||||
} |
||||
|
||||
|
||||
void vt_home() |
||||
{ |
||||
uart_puts_pgm(PSTR("\x1B[H")); |
||||
} |
||||
|
||||
|
||||
/** Initialize helper variables */ |
||||
void vt_init() |
||||
{ |
||||
vt_reset(); |
||||
} |
||||
|
||||
|
||||
/** Reset state and clear screen */ |
||||
void vt_reset() |
||||
{ |
||||
// reset color and attributes
|
||||
vt_color_reset(); |
||||
vt_attr_reset(); |
||||
vt_scroll_reset(); |
||||
|
||||
// clear screen
|
||||
vt_clear(); |
||||
|
||||
// go to top left
|
||||
vt_home(); |
||||
|
||||
// overwrite saved state
|
||||
vt_save(); |
||||
} |
||||
|
||||
|
||||
|
||||
// Assigned keyhandler
|
||||
void (*_vt_kh)(uint8_t, bool) = NULL; |
||||
|
||||
/** Assign a key handler (later used with vt_handle_key) */ |
||||
void vt_set_key_handler(void (*handler)(uint8_t, bool)) |
||||
{ |
||||
_vt_kh = handler; |
||||
} |
||||
|
||||
|
||||
// state machine states
|
||||
typedef enum { |
||||
GROUND = 0, |
||||
ESC = 1, |
||||
BR = 2, |
||||
O = 3, |
||||
WAITING_TILDE = 4 |
||||
} KSTATE; |
||||
|
||||
// code received before started to wait for a tilde
|
||||
uint8_t _before_wtilde; |
||||
// current state
|
||||
KSTATE _kstate = GROUND; |
||||
|
||||
|
||||
|
||||
void _vt_kh_abort() |
||||
{ |
||||
switch (_kstate) { |
||||
case ESC: |
||||
_vt_kh(VK_ESC, true); |
||||
break; |
||||
|
||||
case BR: |
||||
_vt_kh(VK_ESC, true); |
||||
_vt_kh('[', false); |
||||
break; |
||||
|
||||
case O: |
||||
_vt_kh(VK_ESC, true); |
||||
_vt_kh('O', false); |
||||
break; |
||||
|
||||
case WAITING_TILDE: |
||||
_vt_kh(VK_ESC, true); |
||||
_vt_kh('[', false); |
||||
vt_handle_key(_before_wtilde); |
||||
break; |
||||
|
||||
case GROUND: |
||||
// nop
|
||||
break; |
||||
} |
||||
|
||||
_kstate = GROUND; |
||||
} |
||||
|
||||
|
||||
/**
|
||||
* Handle a key received over UART |
||||
* Takes care of multi-byte keys and translates them to special |
||||
* constants. |
||||
*/ |
||||
void vt_handle_key(uint8_t c) |
||||
{ |
||||
if (_vt_kh == NULL) return; |
||||
|
||||
switch (_kstate) { |
||||
case GROUND: |
||||
switch (c) { |
||||
case 27: |
||||
_kstate = ESC; |
||||
break; |
||||
|
||||
case VK_ENTER: |
||||
case VK_TAB: |
||||
case VK_BACKSPACE: |
||||
_vt_kh(c, true); |
||||
return; |
||||
|
||||
default: |
||||
_vt_kh(c, false); |
||||
return; |
||||
} |
||||
|
||||
break; // continue to next char
|
||||
|
||||
case ESC: |
||||
switch (c) { |
||||
case '[': |
||||
_kstate = BR; |
||||
break; // continue to next char
|
||||
|
||||
case 'O': |
||||
_kstate = O; |
||||
break; // continue to next char
|
||||
|
||||
default: |
||||
// bad code
|
||||
_vt_kh_abort(); |
||||
vt_handle_key(c); |
||||
return; |
||||
} |
||||
break; |
||||
|
||||
case BR: |
||||
switch (c) { |
||||
// arrows
|
||||
case 65: |
||||
case 66: |
||||
case 67: |
||||
case 68: |
||||
_vt_kh(c, true); |
||||
_kstate = GROUND; |
||||
return; |
||||
|
||||
// ins del pgup pgdn
|
||||
case 50: |
||||
case 51: |
||||
case 53: |
||||
case 54: |
||||
// wait for terminating tilde
|
||||
_before_wtilde = c; |
||||
_kstate = WAITING_TILDE; |
||||
break; // continue to next char
|
||||
|
||||
// bad key
|
||||
default: |
||||
_vt_kh_abort(); |
||||
vt_handle_key(c); |
||||
return; |
||||
} |
||||
break; |
||||
|
||||
case O: |
||||
switch (c) { |
||||
// F keys
|
||||
case 80: |
||||
case 81: |
||||
case 82: |
||||
case 83: |
||||
// home, end
|
||||
case 72: |
||||
case 70: |
||||
_vt_kh(c, true); |
||||
_kstate = GROUND; |
||||
return; |
||||
|
||||
// bad key
|
||||
default: |
||||
_vt_kh_abort(); |
||||
vt_handle_key(c); |
||||
return; |
||||
} |
||||
|
||||
case WAITING_TILDE: |
||||
if (c != '~') { |
||||
_vt_kh_abort(); |
||||
vt_handle_key(c); |
||||
return; |
||||
} else { |
||||
_vt_kh(_before_wtilde, true); |
||||
_kstate = GROUND; |
||||
return; |
||||
} |
||||
} |
||||
|
||||
// wait for next key
|
||||
if (_kstate != GROUND) { |
||||
_delay_ms(2); |
||||
if (!uart_rx_ready()) { |
||||
// abort receiving
|
||||
_vt_kh_abort(); |
||||
|
||||
} else { |
||||
vt_handle_key(uart_rx()); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,192 @@ |
||||
#pragma once |
||||
|
||||
//
|
||||
// ANSI / VT100 utilities for UART
|
||||
//
|
||||
// To use this, first call uart_init(baud) and vt_init()
|
||||
// To print stuff on the screen, use uart_puts() etc from uart.h
|
||||
//
|
||||
|
||||
#include <avr/io.h> |
||||
#include <stdlib.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
#include "uart.h" |
||||
|
||||
|
||||
|
||||
// INIT
|
||||
|
||||
/** Initialize helper variables */ |
||||
void vt_init(); |
||||
|
||||
/** Reset state and clear screen */ |
||||
void vt_reset(); |
||||
|
||||
|
||||
|
||||
// CURSOR MOVE
|
||||
|
||||
/** Move cursor to top left corner */ |
||||
void vt_home(); |
||||
|
||||
/** Jump to a location on the screen */ |
||||
void vt_goto(uint8_t x, uint8_t y); |
||||
|
||||
/** Jump to given X, keep Y */ |
||||
void vt_goto_x(uint8_t x); |
||||
|
||||
/** Jump to given Y, keep X */ |
||||
void vt_goto_y(uint8_t y); |
||||
|
||||
/** Move cursor relative to current location */ |
||||
void vt_move(int8_t x, int8_t y); |
||||
|
||||
/** Move cursor horizontally */ |
||||
void vt_move_x(int8_t x); |
||||
|
||||
/** Move cursor vertically */ |
||||
void vt_move_y(int8_t y); |
||||
|
||||
/** Move cursor up y cells */ |
||||
void vt_up(uint8_t y); |
||||
|
||||
/** Move cursor down y cells */ |
||||
void vt_down(uint8_t y); |
||||
|
||||
/** Move cursor left x cells */ |
||||
void vt_left(uint8_t x); |
||||
|
||||
/** Move cursor right x cells */ |
||||
void vt_right(uint8_t x); |
||||
|
||||
|
||||
|
||||
// SCROLLING
|
||||
|
||||
/** Scroll y lines down (like up/down, but moves window if needed) */ |
||||
void vt_scroll(int8_t down); |
||||
|
||||
/** Set scrolling region (lines) */ |
||||
void vt_scroll_set(uint8_t from, uint8_t to); |
||||
|
||||
|
||||
/** Sets scrolling region to the entire screen. */ |
||||
void vt_scroll_reset(); |
||||
|
||||
|
||||
// COLOR
|
||||
|
||||
#define VT_BLACK 0 |
||||
#define VT_RED 1 |
||||
#define VT_GREEN 2 |
||||
#define VT_YELLOW 3 |
||||
#define VT_BLUE 4 |
||||
#define VT_MAGENTA 5 |
||||
#define VT_CYAN 6 |
||||
#define VT_WHITE 7 |
||||
|
||||
/** Set color 0..7 */ |
||||
void vt_color(uint8_t fg, uint8_t bg); |
||||
|
||||
/** Set FG color 0..7 */ |
||||
void vt_color_fg(uint8_t fg); |
||||
|
||||
/** Set BG color 0..7 */ |
||||
void vt_color_bg(uint8_t bg); |
||||
|
||||
/** Set color to white on black */ |
||||
void vt_color_reset(); |
||||
|
||||
|
||||
|
||||
// STYLES
|
||||
|
||||
#define VT_BOLD 1 |
||||
#define VT_UNDERLINE 2 |
||||
#define VT_BLINK 4 |
||||
#define VT_REVERSE 8 |
||||
#define VT_ITALIC 16 |
||||
#define VT_FAINT 32 |
||||
|
||||
/** Enable or disable a text attribute */ |
||||
void vt_attr(uint8_t attribute, bool on); |
||||
|
||||
/** Disable all text attributes (excluding color) */ |
||||
void vt_attr_reset(); |
||||
|
||||
|
||||
|
||||
// SAVE & RESTORE
|
||||
|
||||
/** Save cursor position & text attributes */ |
||||
void vt_save(); |
||||
|
||||
/** Restore cursor to saved values */ |
||||
void vt_restore(); |
||||
|
||||
|
||||
|
||||
// MODIFY
|
||||
|
||||
|
||||
/** Insert blank lines febore the current line */ |
||||
void vt_insert_lines(uint8_t count); |
||||
|
||||
/** Delete lines from the current line down */ |
||||
void vt_delete_lines(uint8_t count); |
||||
|
||||
/** Insert empty characters at cursor */ |
||||
void vt_insert_chars(uint8_t count); |
||||
|
||||
/** Delete characters at cursor */ |
||||
void vt_delete_chars(uint8_t count); |
||||
|
||||
|
||||
|
||||
// ERASING
|
||||
|
||||
/** Clear the screen */ |
||||
void vt_clear(); |
||||
|
||||
/** Erase to the end of line */ |
||||
void vt_erase_forth(); |
||||
|
||||
/** Erase line to cursor */ |
||||
void vt_erase_back(); |
||||
|
||||
/** Erase entire line */ |
||||
void vt_erase_line(); |
||||
|
||||
/** Erase screen below the line */ |
||||
void vt_erase_above(); |
||||
|
||||
/** Erase screen above the line */ |
||||
void vt_erase_below(); |
||||
|
||||
|
||||
|
||||
// KEY HANDLER
|
||||
|
||||
// Special keys from key handler
|
||||
#define VK_LEFT 68 |
||||
#define VK_RIGHT 67 |
||||
#define VK_UP 65 |
||||
#define VK_DOWN 66 |
||||
#define VK_DELETE 51 |
||||
#define VK_INSERT 50 |
||||
#define VK_PGUP 53 |
||||
#define VK_PGDN 54 |
||||
#define VK_HOME 72 |
||||
#define VK_END 70 |
||||
#define VK_F1 80 |
||||
#define VK_F2 81 |
||||
#define VK_F3 82 |
||||
#define VK_F4 83 |
||||
#define VK_BACKSPACE 8 |
||||
#define VK_TAB 9 |
||||
#define VK_ENTER 13 |
||||
#define VK_ESC 27 |
||||
|
||||
void vt_handle_key(uint8_t c); |
||||
void vt_set_key_handler(void (*handler)(uint8_t, bool)); |
||||
@ -0,0 +1,46 @@ |
||||
#include <avr/io.h> |
||||
#include <util/delay.h> |
||||
#include <avr/pgmspace.h> |
||||
#include <avr/interrupt.h> |
||||
#include <stdbool.h> |
||||
#include <stdint.h> |
||||
#include <stdlib.h> |
||||
|
||||
#include "lib/uart.h" |
||||
#include "lib/uart_ansi.h" |
||||
|
||||
ISR(USART_RX_vect) |
||||
{ |
||||
vt_handle_key(uart_rx()); |
||||
} |
||||
|
||||
|
||||
void key_handler(uint8_t code, bool special) |
||||
{ |
||||
if (special) { |
||||
uart_puts_pgm(PSTR("Special key: ")); |
||||
} else { |
||||
uart_puts_pgm(PSTR("Character: ")); |
||||
} |
||||
|
||||
uart_putc(code); |
||||
uart_putc(32); |
||||
uart_putu(code); |
||||
uart_nl(); |
||||
} |
||||
|
||||
|
||||
void main() |
||||
{ |
||||
uart_init(9600); |
||||
uart_isr_rx(1); |
||||
vt_init(); |
||||
|
||||
vt_set_key_handler(&key_handler); |
||||
|
||||
sei(); |
||||
|
||||
uart_puts_pgm(PSTR("UART key handler test!\r\n")); |
||||
|
||||
while(1); |
||||
} |
||||
Loading…
Reference in new issue