Browse Source

Actually include symlinked files

master
mid-kid 4 years ago
parent
commit
7b099b5529
  1. 1
      arduino/source/serial.c
  2. 145
      arduino/source/serial.c
  3. 1
      arduino/source/serial.h
  4. 37
      arduino/source/serial.h
  5. 1
      arduino/source/spi.c
  6. 32
      arduino/source/spi.c
  7. 1
      arduino/source/spi.h
  8. 20
      arduino/source/spi.h
  9. 1
      arduino/source/utils.h
  10. 29
      arduino/source/utils.h

1
arduino/source/serial.c

@ -1 +0,0 @@
/home/mid-kid/Stuff/Workspace/bitsandpieces/atmega/serial.c

145
arduino/source/serial.c

@ -0,0 +1,145 @@
#include "serial.h"
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <avr/io.h>
#include <avr/interrupt.h>
#include "utils.h"
#define SERIAL_BUFFER_SIZE 0x40
struct serial_buffer {
volatile unsigned char buffer[SERIAL_BUFFER_SIZE];
volatile unsigned head;
volatile unsigned tail;
};
static volatile struct serial_buffer serial_rx;
static volatile struct serial_buffer serial_tx;
__attribute__((always_inline))
static inline int serial_buffer_isempty(volatile struct serial_buffer *buffer)
{
return buffer->head == buffer->tail;
}
__attribute__((always_inline))
static inline int serial_buffer_isfull(volatile struct serial_buffer *buffer)
{
return ((buffer->head + 1) % SERIAL_BUFFER_SIZE) == buffer->tail;
}
__attribute__((always_inline))
static inline void serial_buffer_put(volatile struct serial_buffer *buffer, char c)
{
volatile unsigned head = buffer->head;
buffer->buffer[head++] = c;
buffer->head = head % SERIAL_BUFFER_SIZE;
}
__attribute__((always_inline))
static inline char serial_buffer_get(volatile struct serial_buffer *buffer)
{
volatile unsigned tail = buffer->tail;
char c = buffer->buffer[tail++];
buffer->tail = tail % SERIAL_BUFFER_SIZE;
return c;
}
static void serial_transmit(void)
{
// Called when UDR0 is ready to receive new data
UDR0 = serial_buffer_get(&serial_tx);
// If we've sent everything, disable the interrupt
if (serial_buffer_isempty(&serial_tx)) cbi(UCSR0B, UDRIE0);
}
static void serial_receive(void)
{
// Called when UDR0 contains new data
// Discard the byte if a parity error has occurred or the buffer is full
if (bit_is_set(UCSR0A, UPE0) || serial_buffer_isfull(&serial_rx)) {
UDR0;
return;
}
// There's not much we can do with a full buffer, unfortunately.
// Delaying the read will only cause it to be replaced with further data.
serial_buffer_put(&serial_rx, UDR0);
}
ISR(USART_UDRE_vect) { serial_transmit(); }
ISR(USART_RX_vect) { serial_receive(); }
__attribute__((always_inline))
inline int serial_putchar_inline(const char c)
{
// If the data register and buffer are empty, just send it straight away
if (bit_is_set(UCSR0A, UDRE0) && serial_buffer_isempty(&serial_tx)) {
UDR0 = c;
return 1;
}
// Make sure there's space in the buffer and put the character
while (serial_buffer_isfull(&serial_tx));
serial_buffer_put(&serial_tx, c);
// Enable the interrupt to transmit as soon as we can
sbi(UCSR0B, UDRIE0);
return 1;
}
int serial_putchar(const char c) { return serial_putchar_inline(c); }
__attribute__((always_inline))
inline int serial_getchar_inline(void)
{
// Wait until there's something in the buffer
while (serial_buffer_isempty(&serial_rx));
// Read the next character
return serial_buffer_get(&serial_rx);
}
int serial_getchar(void) { return serial_getchar_inline(); }
static int stdio_serial_putchar(const char c, __attribute__((unused)) FILE *stream)
{ return serial_putchar(c); }
static int stdio_serial_getchar(__attribute__((unused)) FILE *stream)
{ return serial_getchar(); }
static FILE serial = FDEV_SETUP_STREAM(stdio_serial_putchar, stdio_serial_getchar, _FDEV_SETUP_RW);
unsigned serial_available(void)
{
return ((unsigned)(SERIAL_BUFFER_SIZE + serial_rx.head - serial_rx.tail)) % SERIAL_BUFFER_SIZE;
}
void serial_drain(void)
{
while (bit_is_set(UCSR0B, UDRIE0) || bit_is_set(UCSR0A, RXC0));
}
void serial_init_config(const unsigned long bauds, const uint8_t config)
{
// Calculate UBRRn value as per the datasheet
uint16_t ubrr = F_CPU / 4 / 2 / bauds - 1;
UBRR0H = (ubrr >> 8) & 0xFF;
UBRR0L = ubrr & 0xFF;
// Configure the serial
UCSR0A = _BV(U2X0); // Double speed
UCSR0B = _BV(RXEN0) | _BV(TXEN0) | _BV(RXCIE0); // Enable it and the receive interrupt
UCSR0C = config;
// Setup stdio streams
stdout = stderr = stdin = &serial;
}
void serial_init(const unsigned long bauds) { serial_init_config(bauds, SERIAL_8N1); }

1
arduino/source/serial.h

@ -1 +0,0 @@
/home/mid-kid/Stuff/Workspace/bitsandpieces/atmega/serial.h

37
arduino/source/serial.h

@ -0,0 +1,37 @@
#pragma once
#include <stdint.h>
#define SERIAL_5N1 0
#define SERIAL_6N1 _BV(UCPHA0)
#define SERIAL_7N1 _BV(UDORD0)
#define SERIAL_8N1 _BV(UCPHA0) | _BV(UDORD0)
#define SERIAL_5N2 _BV(USBS0)
#define SERIAL_6N2 _BV(USBS0) | _BV(UCPHA0)
#define SERIAL_7N2 _BV(USBS0) | _BV(UDORD0)
#define SERIAL_8N2 _BV(USBS0) | _BV(UDORD0) | _BV(UCPHA0)
#define SERIAL_5E1 _BV(UPM01)
#define SERIAL_6E1 _BV(UPM01) | _BV(UCPHA0)
#define SERIAL_7E1 _BV(UPM01) | _BV(UDORD0)
#define SERIAL_8E1 _BV(UPM01) | _BV(UCPHA0) | _BV(UDORD0)
#define SERIAL_5E2 _BV(UPM01) | _BV(USBS0)
#define SERIAL_6E2 _BV(UPM01) | _BV(USBS0) | _BV(UCPHA0)
#define SERIAL_7E2 _BV(UPM01) | _BV(USBS0) | _BV(UDORD0)
#define SERIAL_8E2 _BV(UPM01) | _BV(USBS0) | _BV(UDORD0) | _BV(UCPHA0)
#define SERIAL_5O1 _BV(UPM01) | _BV(UPM00)
#define SERIAL_6O1 _BV(UPM01) | _BV(UPM00) | _BV(UCPHA0)
#define SERIAL_7O1 _BV(UPM01) | _BV(UPM00) | _BV(UDORD0)
#define SERIAL_8O1 _BV(UPM01) | _BV(UPM00) | _BV(UCPHA0) | _BV(UDORD0)
#define SERIAL_5O2 _BV(UPM01) | _BV(UPM00) | _BV(USBS0)
#define SERIAL_6O2 _BV(UPM01) | _BV(UPM00) | _BV(USBS0) | _BV(UCPHA0)
#define SERIAL_7O2 _BV(UPM01) | _BV(UPM00) | _BV(USBS0) | _BV(UDORD0)
#define SERIAL_8O2 _BV(UPM01) | _BV(UPM00) | _BV(USBS0) | _BV(UDORD0) | _BV(UCPHA0)
int serial_putchar_inline(const char c);
int serial_putchar(const char c);
int serial_getchar_inline(void);
int serial_getchar(void);
unsigned serial_available(void);
void serial_drain(void);
void serial_init_config(const unsigned long bauds, const uint8_t config);
void serial_init(const unsigned long bauds);

1
arduino/source/spi.c

@ -1 +0,0 @@
/home/mid-kid/Stuff/Workspace/bitsandpieces/atmega/spi.c

32
arduino/source/spi.c

@ -0,0 +1,32 @@
#include "spi.h"
#include <stdint.h>
#include <avr/io.h>
#include "utils.h"
#include "pins.h"
void spi_init(const enum spi_divider divider)
{
DDRB |= pin(PIN_SPI_SS) | pin(PIN_SPI_MOSI) | pin(PIN_SPI_SCK); // Outputs
DDRB &= ~pin(PIN_SPI_MISO); // Inputs
PORTB |= pin(PIN_SPI_MISO); // Pull it up
// Configure SPI as master
SPCR = _BV(SPE) | _BV(MSTR) | (divider & 3);
SPSR = (divider & 4) >> 2;
}
void spi_off(void)
{
SPCR = 0;
SPSR = 1;
}
uint8_t spi_transfer(const uint8_t data)
{
SPDR = data;
__asm__ volatile ("nop");
while (!(SPSR & _BV(SPIF))) __asm__ volatile ("nop");
return SPDR;
}

1
arduino/source/spi.h

@ -1 +0,0 @@
/home/mid-kid/Stuff/Workspace/bitsandpieces/atmega/spi.h

20
arduino/source/spi.h

@ -0,0 +1,20 @@
#pragma once
#include <stdint.h>
enum spi_divider {
SPI_DIV4,
SPI_DIV16,
SPI_DIV64,
SPI_DIV128,
// Double speed
SPI_DIV2,
SPI_DIV8,
SPI_DIV32,
SPI_DIV64_2
};
void spi_init(const enum spi_divider divider);
void spi_off(void);
uint8_t spi_transfer(const uint8_t data);

1
arduino/source/utils.h

@ -1 +0,0 @@
/home/mid-kid/Stuff/Workspace/bitsandpieces/atmega/utils.h

29
arduino/source/utils.h

@ -0,0 +1,29 @@
#pragma once
#include <avr/io.h>
#define cycles(us) (F_CPU * (long long)(us) / 1000000)
#define cbi(sfr, bit) (sfr &= ~_BV(bit))
#define sbi(sfr, bit) (sfr |= _BV(bit))
// Stupid definitions for handling pins.
#define _pin(port, pin) PORT ## port ## pin
#define _port(port, pin) PORT ## port
#define pin(pin) _BV(_pin(pin))
#define port(pin) _port(pin)
#define LOW 0
#define HIGH 1
#define OUTPUT 1
#define INPUT 0
#define _pinmode(port, pin, x) if (x) { sbi(DDR ## port, DD ## port ## pin); } else { cbi(DDR ## port, DD ## port ## pin); }
#define pinmode(pin, x) _pinmode(pin, x)
#define _writepin(port, pin, x) if (x) { sbi(PORT ## port, PORT ## port ## pin); } else { cbi(PORT ## port, PORT ## port ## pin); }
#define writepin(pin, x) _writepin(pin, x)
#define _readpin(port, pin) ((PIN ## port >> PORT ## port ## pin) & 1)
#define readpin(pin) _readpin(pin)
Loading…
Cancel
Save