#include <mega328p.h>

// Standard Input/Output functions
#include <stdio.h>

// Declare your global variables here

#define DATA_REGISTER_EMPTY (1<<UDRE0)
#define RX_COMPLETE (1<<RXC0)
#define FRAMING_ERROR (1<<FE0)
#define PARITY_ERROR (1<<UPE0)
#define DATA_OVERRUN (1<<DOR0)

// USART Receiver buffer
#define MAX_DMA_BUFFERS_COUNT 32     // 4 8 16 32 64 128
#define BUFFER_MASK (MAX_DMA_BUFFERS_COUNT-1)
volatile unsigned char BUFF_RX_BUF[MAX_DMA_BUFFERS_COUNT]={0};
volatile unsigned char DMA_WR_BUF = 0;
volatile unsigned char DMA_RD_BUF = 0;

// USART Receiver interrupt service routine
interrupt [USART_RXC] void usart_rx_isr(void)
{
char status,data;
status=UCSR0A;
data=UDR0;
if ((status & (FRAMING_ERROR | PARITY_ERROR | DATA_OVERRUN))==0)
   { 
      BUFF_RX_BUF[DMA_WR_BUF] = data;
      DMA_WR_BUF++;
     DMA_WR_BUF &= BUFFER_MASK;   
   }
}


#include <string.h>

char b_start=0, b_end=0, len_buff=0, len=0;
bit flag=0;
char buff[24]; // максимальная длинна команды
unsigned long main_p=0;

unsigned long my_atoi(char *str)
{
   unsigned long result = 0;

   if ('0'>*str || *str>'9')
   {
      str++;
   }
   while (*str != '\0')
   {
      if ('0'> *str || *str>'9')
         break;
      else
         result = result * 10 + (*str++ - '0');
   }
   return result;
}

static char respcmp(char *s, const char  *resp){
   while(*resp)
      if(*resp++ != *s++) return 0;
   return 1;
}

void usart_send_str(const char* str)
{
    while(*str) {
    while (!(UCSR0A & (1<<UDRE0))) {}
       UDR0 = *str++;
    }
}

#define SEND(str) usart_send_str(str)

void help(void)      // HELP
{
   SEND("\r\n");
   SEND("\t'0' - turn off LED\r\n");
   SEND("\t'1' - turn on LED\r\n");
   SEND("\t'LED ON' - turn on LED\r\n");
   SEND("\t'LRD OFF' - turn off LED\r\n");
   SEND("\t'TOGGLE' - turn toggle LED\r\n");
   SEND("\t'COUNT=N' - set all value by N (decimal):\r\n");
   SEND("\tSTAT - shows which mode the automatic/off mode is in\r\n");

   SEND("\tHELP - help\r\n");
   SEND("\r\n");
}

// print 32bit unsigned int
void printu(unsigned long val){
    char bufa[11];
    char bufb[10];
    int i;
    int l = 0, bpos = 0;
    if(!val){
        bufa[0] = '0';
        l = 1;
    }else{
        while(val){
            bufb[l++] = val % 10 + '0';
            val /= 10;
        }
        bpos += l;
        for(i = 0; i < l; ++i){
            bufa[--bpos] = bufb[i];
        }
    }
    bufa[l + bpos] = 0;
    SEND(bufa);
}

void main(void)
{ 
// USART initialization
// Communication Parameters: 8 Data, 1 Stop, No Parity
// USART Receiver: On
// USART Transmitter: On
// USART0 Mode: Asynchronous
// USART Baud Rate: 9600
UCSR0A=(0<<RXC0) | (0<<TXC0) | (0<<UDRE0) | (0<<FE0) | (0<<DOR0) | (0<<UPE0) | (0<<U2X0) | (0<<MPCM0);
UCSR0B=(1<<RXCIE0) | (0<<TXCIE0) | (0<<UDRIE0) | (1<<RXEN0) | (1<<TXEN0) | (0<<UCSZ02) | (0<<RXB80) | (0<<TXB80);
UCSR0C=(0<<UMSEL01) | (0<<UMSEL00) | (0<<UPM01) | (0<<UPM00) | (0<<USBS0) | (1<<UCSZ01) | (1<<UCSZ00) | (0<<UCPOL0);
UBRR0H=0x00;
UBRR0L=0x67;

// Global enable interrupts
#asm("sei")

while (1)
      {
      // Place your code here

        while (DMA_RD_BUF != (DMA_WR_BUF)) {
       
            b_start =  DMA_RD_BUF;  // Копируем позиции начала           
            DMA_RD_BUF++;
         DMA_RD_BUF &= BUFFER_MASK;
            b_end = DMA_RD_BUF;    // и конца буфера 
           
            // сборка команды
           len_buff = strlen((void*)buff); // определяем длину
           
            if(b_start < b_end) {
           len = b_end-b_start;
           memcpy(&buff[len_buff],(void*)&BUFF_RX_BUF[b_start],len);
           buff[len_buff+len]=0;
           }

           if(b_start > b_end) {
           len = MAX_DMA_BUFFERS_COUNT-b_start;
           memcpy(&buff[len_buff],(void*)&BUFF_RX_BUF[b_start],len);
           memcpy(&buff[len],BUFF_RX_BUF,b_end);
           buff[len_buff+len+b_end]=0;
           }

           // если BkSp - сдвигаем назад
           len_buff = strlen((void*)buff)-1; // определяем длину
           if((buff[len_buff] == 0x08)|| (buff[len_buff] == 0x7f)){
           // стираем символ
           // но не левее начала буфера
           if(len_buff > 0) buff[len_buff-1]=0; else buff[len_buff]=0;
           }

           // если Enter - ввод закончен
           if(((buff[len_buff] == '\n') || (buff[len_buff] == '\r'))) {
           flag=1; buff[len_buff]='\0'; // стираем символ Enter, заменяем концом строки \0
           }
        }
       
        // поиск обработчика
       if(flag) {
       
        len_buff = strlen((void*)buff); // определяем длину команды
        // если не использовать по символьные команды, то len_buff не нужен

       if(respcmp((void*)buff, "1") && len_buff == strlen("1")) { // это просто пример
            PORTB |= (1<<(0));
       }       

        if(respcmp((void*)buff, "0")&& len_buff == strlen("0")) {
            PORTB &= ~(1<<(0));
       }         
               
       if(respcmp((void*)buff, "LED ON") && len_buff == strlen("LED ON")) {
            PORTB |= (1<<(0));
       }
      
        if(respcmp((void*)buff, "LED OFF") && len_buff == strlen("LED OFF")) {
            PORTB &= ~(1<<(0));
       }
      
        if(respcmp((void*)buff, "TOGGLE") && len_buff == strlen("TOGGLE")) {
            PORTB ^= (1<<(0));
       }
                   
       if(respcmp((void*)buff, "COUNT=")) {
            main_p = my_atoi((buff + sizeof("COUNT=")-1));
       }
       
        if(respcmp((void*)buff, "PRINT") && len_buff == strlen("PRINT")) {
           SEND("\r\nPRINT= "); printu(main_p);SEND("\r\n");
       }
       
       if(respcmp((void*)buff, "HELP") && len_buff == strlen("HELP")) {
            help();
       }       
       
        buff[0]=0;
       flag=0;             
      }
  }

}