#include <avr/io.h>
#include <avr/interrupt.h>
#include <stdint.h>
#include <string.h>
#include "circular_buf.h"
#include "uart.h"

struct circular_buffer uartRxBuffer;
struct circular_buffer uartTxBuffer;

// #define HI(x) ((x)>>8)
// #define LO(x) ((x)& 0xFF)

// #define BAUDRATE 9600L  // скорость передачи по UART
// #define BAUDRATE_REG (F_CPU/(16*BAUDRATE)-1) 

void UART_init(void) {
	UCSR0B = (1 << RXEN0) | (1 << TXEN0) | (1 << RXCIE0) | (1<<TXCIE0); // прерывание по приему и передаче
	UCSR0C = (1 << UCSZ01) | (1 << UCSZ00);
	UBRR0H = 0;
	UBRR0L = 103;
}

void UART_send(uint8_t* data, size_t length) { 
  for (size_t i = 0; i < length; i++) {
    if (!buffer_full(&uartTxBuffer)) {   
        write_buffer(&uartTxBuffer, data[i]);
      } else {
        break; // если буфер передачи заполнен, то отправка прерывается  }
      }
  }
  UDR0 = read_buffer(&uartTxBuffer);
  UCSR0B |= (1 << TXCIE0); 
}

// Получение данных из буфера
int UART_receive(uint8_t* data, size_t length) {
    char overflow = 0;  // Флаг переполнения, который устанавливается, если превышен размер массива 
    uint32_t byteCount = 0;  // Счетчик байтов, принятых из буфера приема
    // Пока буфер приема не пуст и не превышен лимит длины массива,
    // функция продолжает читать байты из буфера и сохранять их в массив data.
    while (!buffer_empty(&uartRxBuffer) && byteCount < length) {
        int reader = read_buffer(&uartRxBuffer);  // Прием и запись символа в переменную
        data[byteCount] = reader;  // Запись в массив с индексом byteCount
        byteCount++;
    }
    // Проверка переполнения
    if (byteCount > length) {
        overflow = 1;
    }
	//clear_buffer(&uartRxBuffer);
  return overflow ? -1 : byteCount;  // Возвращает количество успешно принятых байт или -1 в случае переполнения
}

// прерывание по завершению приема
ISR(USART_RX_vect) {
	uint8_t data = UDR0; // читаем из регистра UDR0
	if (!buffer_full(&uartRxBuffer)) {
		write_buffer(&uartRxBuffer, data);// записываем символ в буфер приема
	}
}

ISR(USART_TX_vect) {
	if (!buffer_empty(&uartTxBuffer)) {
		UDR0 = read_buffer(&uartTxBuffer);
	} else {
		UCSR0B &= ~(1 << TXCIE0); // отключаем прерывание, когда все данные отправлены
	}
}