Colony » Colony Scout » Colony Scout OS

#include "Atmega128rfa1.h"

extern "C"

{

#include <avr/io.h>

#include <avr/interrupt.h>

}

unsigned long millis;

Atmega128rfa1::Atmega128rfa1()

{

}

ISR(TIMER0_COMPA_vect)

{

  millis++;

}

void Atmega128rfa1::init()

{

  // === init serial ===

  // for 16 MHz clock, 76800 baud

  uint16_t baud = 12;

  UBRR0H = baud >> 8;

  UBRR0L = baud;

  // UMSEL0 = 0, asynchronous usart

  // UPM0 = 0, parity check disabled

  // USBS0 = 0, 1 stop bit

  // UCSZ0 = 3, 8-bit

  UCSR0B = _BV(RXEN0) | _BV(TXEN0);

  UCSR0C = _BV(UCSZ01) | _BV(UCSZ00);

  // === init time ===

  // COM0x = 0, pin OC0x not used

  // WGM0 = 2, clear timer on compare match, TOP = OCRA

  // CS0 = 3, 64 prescaler

  TCCR0A = _BV(WGM01);

  TCCR0B = _BV(CS01) | _BV(CS00);

  // enable interrupt on compare match A

  TIMSK0 = _BV(OCIE0A);

  // (1 ms) * 16 MHz / 64 prescaler = 250

  OCR0A = 250;

  millis = 0;

}

int Atmega128rfa1::read()

{

  // TODO make a rx buffer

  if (UCSR0A & _BV(RXC0)) return UDR0;

  else return -1;

}

void Atmega128rfa1::write(uint8_t* data, int length)

{

  // TODO make this non-blocking with a tx buffer

  int i;

  for (i = 0; i < length; i++)

  {

    while (!(UCSR0A & _BV(UDRE0)));

    UDR0 = data[i];

  }

}

unsigned long Atmega128rfa1::time()

{

  unsigned long ret;

  cli();

  ret = millis;

  sei();

  return ret;

}