Colony » Colony Scout » Colony Scout OS

#include <stdint.h>

#include <avr/io.h>

#include <avr/interrupt.h>

#include "tiny-twi.h"

#include "bomi2c.h"

char last_bit;

char count;

uint16_t data;

volatile char currently_sending;

char sending_counter;

uint16_t sending_data;

#define READ_BIT(pin, bit) (((pin) >> bit) & 1)

#define SET_BIT(pin, bit) ((pin) |= (1 << (bit)))

#define CLEAR_BIT(pin, bit) ((pin) &= ~(1 << (bit)))

#define PRESCALAR 8

#define TIME_US(us) (F_CPU * (us) / 1000000 / PRESCALAR)

#define ADDRESS 0x3 

#define DETECT PB2

#define EMIT PB4

static void bom_init(void) {

  // setup INT0

  //EICRA |= _BV(ISC01) | _BV(ISC00);

  //EIMSK |= _BV(INT0);

  init_timer0();

  start_timer0();

  init_timer1();

  // setup output pin

  SET_BIT(DDRB, 4);

  sei();

  bom_start();

}

////////////////////////////////// Receiving ///////////////////////////////////

static void init_timer0(void) {

  TCCROA |= _BV(WGM01);    // enable reset on compare A

  TIMSK  |= _BV(OCIE0A);   // enable timer0 compare match A

}

static void start_timer0(void) {

  TCNT0 = 0; // Reset timer 0 count to 0

  TCCR0B |= _BV(CS01);     // setup clkio and start the timer

}

static void start_detect(void) {

  data = 0;

  count = 0;

  last_bit = 1;

  start_timer0();

}

static void stop_detect(void) {

  TCCR0B &= ~0x07; // Stop timer0

}

ISR(TIMER0_COMPA_vect) {

  char this_bit = READ_BIT(PINB, DETECT);

  if (this_bit == 0) {

    if (last_bit == 0) {

      stop_detect();

    }

  } else {

    data = data << 1 | (~last_bit & 1);

    count++;

  }

  last_bit = this_bit;

  if (count == 15) {

    stop_detect();

    smb_send_data(&data, sizeof(data));

  }

}

////////////////////////////////////////////////////////////////////////////////

//////////////////////////////////// Sending ///////////////////////////////////

static void start_timer1() {

  TCNT1 = 0; // Reset timer 1 count to 0

  TCCR1 |= _BV(CS12) // Set prescalar to 8 and start timer1

}

static void stop_timer1() {

  TCCR1 &= ~0x0F; // Stop the timer without clearing CTC1

}

static void init_timer1() {

  OCR1B = TIME_US(320); // Set match b to fire every 320 us

  TIMSK |= _BV(OCIE1A) | _BV(OCIE1B); // Enable interrupt for match a and b

  TCCR1 |= _BV(CTC1);   // Enables resetting timer1 after matches match c

}

static void send_data(uint16_t data) {

  sending_data = data;

  sending_counter = 0;

  currently_sending = 1;

  SET_BIT(PORTB, EMIT);

  start_timer1();

  send_next_bit();

}

static void send_next_bit() {

  char next_bit = sending_data >> (14 - sending_counter) & 1;

  char ocr_temp = next_bit ? TIME_US(2000-40) : TIME_US(1000-20);

  OCR1A = ocr_temp; // Compare A - Turns on transmitter

  OCR1C = ocr_temp; // Reset the timer when match a fires

}

// Compare A - Turns on transmitter

ISR(TIMER1_COMPA_vect) {

  SET_BIT(PORTB, EMIT); 

  sending_counter++;

  send_next_bit();

}

// Compare B - Turns off transmitter after 320us

ISR(TIMER1_COMPB_vect) {

  CLEAR_BIT(PORTB, EMIT);

  if (sending_counter >= 15) {

    currently_sending = 0;

    stop_timer1();

  }

}

////////////////////////////////////////////////////////////////////////////////

ISR(INT0_vect) {

  bom_start();

}

static void slave_rx(uint8_t *buf, int len) {

  if (len >= 3) {

    switch (buf[0]) {

      case BOM_I2C_SEND:

        if (!currently_sending) send_data((uint16_t) buf[1] << 8 | buf[2]);  

        break;

    }

  }

}

int main() {

  bom_init();

  smb_init(slave_rx);

  smb_set_address(ADDRESS);

  for (;;);;;;;;;;;;;;;;;;; 

}