Browse Source
Add serial checkRx in stepper ISR. Copied HardwareSerial to MarlinSerial (Needed for checkRx).pull/1/head
Erik van der Zalm
13 years ago
16 changed files with 1007 additions and 754 deletions
@ -0,0 +1,213 @@ |
|||||
|
/*
|
||||
|
HardwareSerial.cpp - Hardware serial library for Wiring |
||||
|
Copyright (c) 2006 Nicholas Zambetti. All right reserved. |
||||
|
|
||||
|
This library is free software; you can redistribute it and/or |
||||
|
modify it under the terms of the GNU Lesser General Public |
||||
|
License as published by the Free Software Foundation; either |
||||
|
version 2.1 of the License, or (at your option) any later version. |
||||
|
|
||||
|
This library is distributed in the hope that it will be useful, |
||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
|
Lesser General Public License for more details. |
||||
|
|
||||
|
You should have received a copy of the GNU Lesser General Public |
||||
|
License along with this library; if not, write to the Free Software |
||||
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
||||
|
|
||||
|
Modified 23 November 2006 by David A. Mellis |
||||
|
Modified 28 September 2010 by Mark Sproul |
||||
|
*/ |
||||
|
|
||||
|
#include <stdlib.h> |
||||
|
#include <stdio.h> |
||||
|
#include <string.h> |
||||
|
#include <inttypes.h> |
||||
|
#include "wiring.h" |
||||
|
#include "wiring_private.h" |
||||
|
|
||||
|
// this next line disables the entire HardwareSerial.cpp,
|
||||
|
// this is so I can support Attiny series and any other chip without a uart
|
||||
|
#if defined(UBRRH) || defined(UBRR0H) || defined(UBRR1H) || defined(UBRR2H) || defined(UBRR3H) |
||||
|
|
||||
|
#include "MarlinSerial.h" |
||||
|
|
||||
|
// Define constants and variables for buffering incoming serial data. We're
|
||||
|
// using a ring buffer (I think), in which rx_buffer_head is the index of the
|
||||
|
// location to which to write the next incoming character and rx_buffer_tail
|
||||
|
// is the index of the location from which to read.
|
||||
|
#define RX_BUFFER_SIZE 128 |
||||
|
|
||||
|
struct ring_buffer |
||||
|
{ |
||||
|
unsigned char buffer[RX_BUFFER_SIZE]; |
||||
|
int head; |
||||
|
int tail; |
||||
|
}; |
||||
|
|
||||
|
#if defined(UBRRH) || defined(UBRR0H) |
||||
|
ring_buffer rx_buffer = { { 0 }, 0, 0 }; |
||||
|
#endif |
||||
|
|
||||
|
|
||||
|
inline void store_char(unsigned char c, ring_buffer *rx_buffer) |
||||
|
{ |
||||
|
int i = (unsigned int)(rx_buffer->head + 1) % RX_BUFFER_SIZE; |
||||
|
|
||||
|
// if we should be storing the received character into the location
|
||||
|
// just before the tail (meaning that the head would advance to the
|
||||
|
// current location of the tail), we're about to overflow the buffer
|
||||
|
// and so we don't write the character or advance the head.
|
||||
|
if (i != rx_buffer->tail) { |
||||
|
rx_buffer->buffer[rx_buffer->head] = c; |
||||
|
rx_buffer->head = i; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
|
//#elif defined(SIG_USART_RECV)
|
||||
|
#if defined(USART0_RX_vect) |
||||
|
// fixed by Mark Sproul this is on the 644/644p
|
||||
|
//SIGNAL(SIG_USART_RECV)
|
||||
|
SIGNAL(USART0_RX_vect) |
||||
|
{ |
||||
|
#if defined(UDR0) |
||||
|
unsigned char c = UDR0; |
||||
|
#elif defined(UDR) |
||||
|
unsigned char c = UDR; // atmega8, atmega32
|
||||
|
#else |
||||
|
#error UDR not defined |
||||
|
#endif |
||||
|
store_char(c, &rx_buffer); |
||||
|
} |
||||
|
#endif |
||||
|
|
||||
|
// Constructors ////////////////////////////////////////////////////////////////
|
||||
|
|
||||
|
MarlinSerial::MarlinSerial(ring_buffer *rx_buffer, |
||||
|
volatile uint8_t *ubrrh, volatile uint8_t *ubrrl, |
||||
|
volatile uint8_t *ucsra, volatile uint8_t *ucsrb, |
||||
|
volatile uint8_t *udr, |
||||
|
uint8_t rxen, uint8_t txen, uint8_t rxcie, uint8_t udre, uint8_t u2x) |
||||
|
{ |
||||
|
_rx_buffer = rx_buffer; |
||||
|
_ubrrh = ubrrh; |
||||
|
_ubrrl = ubrrl; |
||||
|
_ucsra = ucsra; |
||||
|
_ucsrb = ucsrb; |
||||
|
_udr = udr; |
||||
|
_rxen = rxen; |
||||
|
_txen = txen; |
||||
|
_rxcie = rxcie; |
||||
|
_udre = udre; |
||||
|
_u2x = u2x; |
||||
|
} |
||||
|
|
||||
|
// Public Methods //////////////////////////////////////////////////////////////
|
||||
|
|
||||
|
void MarlinSerial::begin(long baud) |
||||
|
{ |
||||
|
uint16_t baud_setting; |
||||
|
bool use_u2x = true; |
||||
|
|
||||
|
#if F_CPU == 16000000UL |
||||
|
// hardcoded exception for compatibility with the bootloader shipped
|
||||
|
// with the Duemilanove and previous boards and the firmware on the 8U2
|
||||
|
// on the Uno and Mega 2560.
|
||||
|
if (baud == 57600) { |
||||
|
use_u2x = false; |
||||
|
} |
||||
|
#endif |
||||
|
|
||||
|
if (use_u2x) { |
||||
|
*_ucsra = 1 << _u2x; |
||||
|
baud_setting = (F_CPU / 4 / baud - 1) / 2; |
||||
|
} else { |
||||
|
*_ucsra = 0; |
||||
|
baud_setting = (F_CPU / 8 / baud - 1) / 2; |
||||
|
} |
||||
|
|
||||
|
// assign the baud_setting, a.k.a. ubbr (USART Baud Rate Register)
|
||||
|
*_ubrrh = baud_setting >> 8; |
||||
|
*_ubrrl = baud_setting; |
||||
|
|
||||
|
sbi(*_ucsrb, _rxen); |
||||
|
sbi(*_ucsrb, _txen); |
||||
|
sbi(*_ucsrb, _rxcie); |
||||
|
} |
||||
|
|
||||
|
void MarlinSerial::end() |
||||
|
{ |
||||
|
cbi(*_ucsrb, _rxen); |
||||
|
cbi(*_ucsrb, _txen); |
||||
|
cbi(*_ucsrb, _rxcie); |
||||
|
} |
||||
|
|
||||
|
int MarlinSerial::available(void) |
||||
|
{ |
||||
|
return (unsigned int)(RX_BUFFER_SIZE + _rx_buffer->head - _rx_buffer->tail) % RX_BUFFER_SIZE; |
||||
|
} |
||||
|
|
||||
|
int MarlinSerial::peek(void) |
||||
|
{ |
||||
|
if (_rx_buffer->head == _rx_buffer->tail) { |
||||
|
return -1; |
||||
|
} else { |
||||
|
return _rx_buffer->buffer[_rx_buffer->tail]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
int MarlinSerial::read(void) |
||||
|
{ |
||||
|
// if the head isn't ahead of the tail, we don't have any characters
|
||||
|
if (_rx_buffer->head == _rx_buffer->tail) { |
||||
|
return -1; |
||||
|
} else { |
||||
|
unsigned char c = _rx_buffer->buffer[_rx_buffer->tail]; |
||||
|
_rx_buffer->tail = (unsigned int)(_rx_buffer->tail + 1) % RX_BUFFER_SIZE; |
||||
|
return c; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
void MarlinSerial::flush() |
||||
|
{ |
||||
|
// don't reverse this or there may be problems if the RX interrupt
|
||||
|
// occurs after reading the value of rx_buffer_head but before writing
|
||||
|
// the value to rx_buffer_tail; the previous value of rx_buffer_head
|
||||
|
// may be written to rx_buffer_tail, making it appear as if the buffer
|
||||
|
// don't reverse this or there may be problems if the RX interrupt
|
||||
|
// occurs after reading the value of rx_buffer_head but before writing
|
||||
|
// the value to rx_buffer_tail; the previous value of rx_buffer_head
|
||||
|
// may be written to rx_buffer_tail, making it appear as if the buffer
|
||||
|
// were full, not empty.
|
||||
|
_rx_buffer->head = _rx_buffer->tail; |
||||
|
} |
||||
|
|
||||
|
void MarlinSerial::write(uint8_t c) |
||||
|
{ |
||||
|
while (!((*_ucsra) & (1 << _udre))) |
||||
|
; |
||||
|
|
||||
|
*_udr = c; |
||||
|
} |
||||
|
|
||||
|
void MarlinSerial::checkRx() |
||||
|
{ |
||||
|
if((UCSR0A & (1<<RXC0)) != 0) { |
||||
|
unsigned char c = UDR0; |
||||
|
store_char(c, &rx_buffer); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// Preinstantiate Objects //////////////////////////////////////////////////////
|
||||
|
|
||||
|
#if defined(UBRR0H) && defined(UBRR0L) |
||||
|
MarlinSerial MSerial(&rx_buffer, &UBRR0H, &UBRR0L, &UCSR0A, &UCSR0B, &UDR0, RXEN0, TXEN0, RXCIE0, UDRE0, U2X0); |
||||
|
#else |
||||
|
#error no serial port defined (port 0) |
||||
|
#endif |
||||
|
|
||||
|
|
||||
|
#endif // whole file
|
||||
|
|
||||
@ -0,0 +1,66 @@ |
|||||
|
/*
|
||||
|
HardwareSerial.h - Hardware serial library for Wiring |
||||
|
Copyright (c) 2006 Nicholas Zambetti. All right reserved. |
||||
|
|
||||
|
This library is free software; you can redistribute it and/or |
||||
|
modify it under the terms of the GNU Lesser General Public |
||||
|
License as published by the Free Software Foundation; either |
||||
|
version 2.1 of the License, or (at your option) any later version. |
||||
|
|
||||
|
This library is distributed in the hope that it will be useful, |
||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
|
Lesser General Public License for more details. |
||||
|
|
||||
|
You should have received a copy of the GNU Lesser General Public |
||||
|
License along with this library; if not, write to the Free Software |
||||
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
||||
|
|
||||
|
Modified 28 September 2010 by Mark Sproul |
||||
|
*/ |
||||
|
|
||||
|
#ifndef MarlinSerial_h |
||||
|
#define MarlinSerial_h |
||||
|
|
||||
|
#include <inttypes.h> |
||||
|
|
||||
|
#include "Stream.h" |
||||
|
|
||||
|
struct ring_buffer; |
||||
|
|
||||
|
class MarlinSerial : public Stream |
||||
|
{ |
||||
|
private: |
||||
|
ring_buffer *_rx_buffer; |
||||
|
volatile uint8_t *_ubrrh; |
||||
|
volatile uint8_t *_ubrrl; |
||||
|
volatile uint8_t *_ucsra; |
||||
|
volatile uint8_t *_ucsrb; |
||||
|
volatile uint8_t *_udr; |
||||
|
uint8_t _rxen; |
||||
|
uint8_t _txen; |
||||
|
uint8_t _rxcie; |
||||
|
uint8_t _udre; |
||||
|
uint8_t _u2x; |
||||
|
public: |
||||
|
MarlinSerial(ring_buffer *rx_buffer, |
||||
|
volatile uint8_t *ubrrh, volatile uint8_t *ubrrl, |
||||
|
volatile uint8_t *ucsra, volatile uint8_t *ucsrb, |
||||
|
volatile uint8_t *udr, |
||||
|
uint8_t rxen, uint8_t txen, uint8_t rxcie, uint8_t udre, uint8_t u2x); |
||||
|
void begin(long); |
||||
|
void end(); |
||||
|
virtual int available(void); |
||||
|
virtual int peek(void); |
||||
|
virtual int read(void); |
||||
|
virtual void flush(void); |
||||
|
virtual void write(uint8_t); |
||||
|
virtual void checkRx(void); |
||||
|
using Print::write; // pull in write(str) and write(buf, size) from Print
|
||||
|
}; |
||||
|
|
||||
|
#if defined(UBRRH) || defined(UBRR0H) |
||||
|
extern MarlinSerial MSerial; |
||||
|
#endif |
||||
|
|
||||
|
#endif |
||||
Loading…
Reference in new issue