/*
 %atmel_license%
*/

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "UART.h"

// Constructors ////////////////////////////////////////////////////////////////

UARTClass::UARTClass( Uart* pUart, IRQn_Type dwIrq, uint32_t dwId, RingBuffer* pRx_buffer, RingBuffer* pTx_buffer )
{
  _rx_buffer = pRx_buffer ;
  _tx_buffer = pTx_buffer ;

  _pUart=pUart ;
  _dwIrq=dwIrq ;
  _dwId=dwId ;
}

// Public Methods //////////////////////////////////////////////////////////////

void UARTClass::begin( const uint32_t dwBaudRate )
{
  /* Configure PMC */
  PMC_EnablePeripheral( _dwId ) ;

  /* Reset and disable receiver & transmitter */
  _pUart->UART_CR = UART_CR_RSTRX | UART_CR_RSTTX | UART_CR_RXDIS | UART_CR_TXDIS ;

  /* Configure mode */
  _pUart->UART_MR = UART_MR_PAR_NO ;

  /* Configure baudrate */
  /* Asynchronous, no oversampling */
  _pUart->UART_BRGR = (SystemCoreClock / dwBaudRate) >> 4 ;

  /* Disable PDC channel */
  _pUart->UART_PTCR = UART_PTCR_RXTDIS | UART_PTCR_TXTDIS ;

  /* Configure the ENDRX interrupt */
  _pUart->UART_IER=UART_IER_ENDRX ;

  /* Enable UART interrupt in NVIC */
//  NVIC_EnableIRQ( _dwIrq ) ;

  /* Enable receiver and transmitter */
  _pUart->UART_CR = UART_CR_RXEN | UART_CR_TXEN ;
}

void UARTClass::end( void )
{
  // wait for transmission of outgoing data
  //while ( _tx_buffer->_iHead != _tx_buffer->_iTail )
  //{
  //}

  // clear any received data
  _rx_buffer->_iHead = _rx_buffer->_iTail ;

  PMC_DisablePeripheral( _dwId ) ;
}

int UARTClass::available( void )
{
  return (uint32_t)(SERIAL_BUFFER_SIZE + _rx_buffer->_iHead - _rx_buffer->_iTail) % SERIAL_BUFFER_SIZE ;
}

int UARTClass::peek( void )
{
  if ( _rx_buffer->_iHead == _rx_buffer->_iTail )
  {
    return -1 ;
  }
  else
  {
    return _rx_buffer->_aucBuffer[_rx_buffer->_iTail] ;
  }
}

int UARTClass::read( void )
{
  // if the head isn't ahead of the tail, we don't have any characters
  if ( _rx_buffer->_iHead == _rx_buffer->_iTail )
  {
    return -1 ;
  }
  else
  {
    uint8_t uc = _rx_buffer->_aucBuffer[_rx_buffer->_iTail] ;

    _rx_buffer->_iTail = (unsigned int)(_rx_buffer->_iTail + 1) % SERIAL_BUFFER_SIZE ;

    return uc ;
  }
}

void UARTClass::flush( void )
{
  //while ( _tx_buffer->_iHead != _tx_buffer->_iTail )
  //{
  //}
}

void UARTClass::write( const uint8_t uc_data )
{
  /* Check if the transmitter is ready */
  if ( (_pUart->UART_SR & UART_SR_TXRDY) != UART_SR_TXRDY )
  {
      return ;
  }

  /* Send character */
  _pUart->UART_THR = uc_data ;
}

void UARTClass::IrqHandler( void )
{
  /* Did we receive data ? */
  if ( (_pUart->UART_IER & UART_IER_ENDRX) == UART_IER_ENDRX )
  {
    _rx_buffer->store_char( _pUart->UART_RHR ) ;
  }
}