MarlinSerial.cpp 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394
  1. /*
  2. HardwareSerial.cpp - Hardware serial library for Wiring
  3. Copyright (c) 2006 Nicholas Zambetti. All right reserved.
  4. This library is free software; you can redistribute it and/or
  5. modify it under the terms of the GNU Lesser General Public
  6. License as published by the Free Software Foundation; either
  7. version 2.1 of the License, or (at your option) any later version.
  8. This library is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11. Lesser General Public License for more details.
  12. You should have received a copy of the GNU Lesser General Public
  13. License along with this library; if not, write to the Free Software
  14. Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  15. Modified 23 November 2006 by David A. Mellis
  16. Modified 28 September 2010 by Mark Sproul
  17. */
  18. #include "Marlin.h"
  19. #include "MarlinSerial.h"
  20. uint8_t selectedSerialPort = 0;
  21. #ifndef AT90USB
  22. // this next line disables the entire HardwareSerial.cpp,
  23. // this is so I can support Attiny series and any other chip without a UART
  24. #if defined(UBRRH) || defined(UBRR0H) || defined(UBRR1H) || defined(UBRR2H) || defined(UBRR3H)
  25. #if UART_PRESENT(SERIAL_PORT)
  26. ring_buffer rx_buffer = { { 0 }, 0, 0 };
  27. #endif
  28. FORCE_INLINE void store_char(unsigned char c)
  29. {
  30. int i = (unsigned int)(rx_buffer.head + 1) % RX_BUFFER_SIZE;
  31. // if we should be storing the received character into the location
  32. // just before the tail (meaning that the head would advance to the
  33. // current location of the tail), we're about to overflow the buffer
  34. // and so we don't write the character or advance the head.
  35. if (i != rx_buffer.tail) {
  36. rx_buffer.buffer[rx_buffer.head] = c;
  37. rx_buffer.head = i;
  38. }
  39. }
  40. //#elif defined(SIG_USART_RECV)
  41. #if defined(M_USARTx_RX_vect)
  42. // fixed by Mark Sproul this is on the 644/644p
  43. //SIGNAL(SIG_USART_RECV)
  44. SIGNAL(M_USARTx_RX_vect)
  45. {
  46. // Test for a framing error.
  47. if (M_UCSRxA & (1<<M_FEx))
  48. {
  49. // Characters received with the framing errors will be ignored.
  50. // Dummy register read (discard)
  51. (void)(*(char *)M_UDRx);
  52. }
  53. else
  54. {
  55. // Read the input register.
  56. unsigned char c = M_UDRx;
  57. if (selectedSerialPort == 0)
  58. store_char(c);
  59. #ifdef DEBUG_DUMP_TO_2ND_SERIAL
  60. UDR1 = c;
  61. #endif //DEBUG_DUMP_TO_2ND_SERIAL
  62. }
  63. }
  64. #ifndef SNMM
  65. SIGNAL(USART1_RX_vect)
  66. {
  67. // Test for a framing error.
  68. if (UCSR1A & (1<<FE1))
  69. {
  70. // Characters received with the framing errors will be ignored.
  71. // Dummy register read (discard)
  72. (void)(*(char *)UDR1);
  73. }
  74. else
  75. {
  76. // Read the input register.
  77. unsigned char c = UDR1;
  78. if (selectedSerialPort == 1)
  79. store_char(c);
  80. #ifdef DEBUG_DUMP_TO_2ND_SERIAL
  81. M_UDRx = c;
  82. #endif //DEBUG_DUMP_TO_2ND_SERIAL
  83. }
  84. }
  85. #endif
  86. #endif
  87. // Constructors ////////////////////////////////////////////////////////////////
  88. MarlinSerial::MarlinSerial()
  89. {
  90. }
  91. // Public Methods //////////////////////////////////////////////////////////////
  92. void MarlinSerial::begin(long baud)
  93. {
  94. uint16_t baud_setting;
  95. bool useU2X = true;
  96. #if F_CPU == 16000000UL && SERIAL_PORT == 0
  97. // hard-coded exception for compatibility with the bootloader shipped
  98. // with the Duemilanove and previous boards and the firmware on the 8U2
  99. // on the Uno and Mega 2560.
  100. if (baud == 57600) {
  101. useU2X = false;
  102. }
  103. #endif
  104. // set up the first (original serial port)
  105. if (useU2X) {
  106. M_UCSRxA = 1 << M_U2Xx;
  107. baud_setting = (F_CPU / 4 / baud - 1) / 2;
  108. } else {
  109. M_UCSRxA = 0;
  110. baud_setting = (F_CPU / 8 / baud - 1) / 2;
  111. }
  112. // assign the baud_setting, a.k.a. ubbr (USART Baud Rate Register)
  113. M_UBRRxH = baud_setting >> 8;
  114. M_UBRRxL = baud_setting;
  115. sbi(M_UCSRxB, M_RXENx);
  116. sbi(M_UCSRxB, M_TXENx);
  117. sbi(M_UCSRxB, M_RXCIEx);
  118. #ifndef SNMM
  119. if (selectedSerialPort == 1) { //set up also the second serial port
  120. if (useU2X) {
  121. UCSR1A = 1 << U2X1;
  122. baud_setting = (F_CPU / 4 / baud - 1) / 2;
  123. } else {
  124. UCSR1A = 0;
  125. baud_setting = (F_CPU / 8 / baud - 1) / 2;
  126. }
  127. // assign the baud_setting, a.k.a. ubbr (USART Baud Rate Register)
  128. UBRR1H = baud_setting >> 8;
  129. UBRR1L = baud_setting;
  130. sbi(UCSR1B, RXEN1);
  131. sbi(UCSR1B, TXEN1);
  132. sbi(UCSR1B, RXCIE1);
  133. }
  134. #endif
  135. }
  136. void MarlinSerial::end()
  137. {
  138. cbi(M_UCSRxB, M_RXENx);
  139. cbi(M_UCSRxB, M_TXENx);
  140. cbi(M_UCSRxB, M_RXCIEx);
  141. #ifndef SNMM
  142. cbi(UCSR1B, RXEN1);
  143. cbi(UCSR1B, TXEN1);
  144. cbi(UCSR1B, RXCIE1);
  145. #endif
  146. }
  147. int MarlinSerial::peek(void)
  148. {
  149. if (rx_buffer.head == rx_buffer.tail) {
  150. return -1;
  151. } else {
  152. return rx_buffer.buffer[rx_buffer.tail];
  153. }
  154. }
  155. int MarlinSerial::read(void)
  156. {
  157. // if the head isn't ahead of the tail, we don't have any characters
  158. if (rx_buffer.head == rx_buffer.tail) {
  159. return -1;
  160. } else {
  161. unsigned char c = rx_buffer.buffer[rx_buffer.tail];
  162. rx_buffer.tail = (unsigned int)(rx_buffer.tail + 1) % RX_BUFFER_SIZE;
  163. return c;
  164. }
  165. }
  166. void MarlinSerial::flush()
  167. {
  168. // don't reverse this or there may be problems if the RX interrupt
  169. // occurs after reading the value of rx_buffer_head but before writing
  170. // the value to rx_buffer_tail; the previous value of rx_buffer_head
  171. // may be written to rx_buffer_tail, making it appear as if the buffer
  172. // don't reverse this or there may be problems if the RX interrupt
  173. // occurs after reading the value of rx_buffer_head but before writing
  174. // the value to rx_buffer_tail; the previous value of rx_buffer_head
  175. // may be written to rx_buffer_tail, making it appear as if the buffer
  176. // were full, not empty.
  177. rx_buffer.head = rx_buffer.tail;
  178. }
  179. /// imports from print.h
  180. void MarlinSerial::print(char c, int base)
  181. {
  182. print((long) c, base);
  183. }
  184. void MarlinSerial::print(unsigned char b, int base)
  185. {
  186. print((unsigned long) b, base);
  187. }
  188. void MarlinSerial::print(int n, int base)
  189. {
  190. print((long) n, base);
  191. }
  192. void MarlinSerial::print(unsigned int n, int base)
  193. {
  194. print((unsigned long) n, base);
  195. }
  196. void MarlinSerial::print(long n, int base)
  197. {
  198. if (base == 0) {
  199. write(n);
  200. } else if (base == 10) {
  201. if (n < 0) {
  202. print('-');
  203. n = -n;
  204. }
  205. printNumber(n, 10);
  206. } else {
  207. printNumber(n, base);
  208. }
  209. }
  210. void MarlinSerial::print(unsigned long n, int base)
  211. {
  212. if (base == 0) write(n);
  213. else printNumber(n, base);
  214. }
  215. void MarlinSerial::print(double n, int digits)
  216. {
  217. printFloat(n, digits);
  218. }
  219. void MarlinSerial::println(void)
  220. {
  221. print('\r');
  222. print('\n');
  223. }
  224. void MarlinSerial::println(const String &s)
  225. {
  226. print(s);
  227. println();
  228. }
  229. void MarlinSerial::println(const char c[])
  230. {
  231. print(c);
  232. println();
  233. }
  234. void MarlinSerial::println(char c, int base)
  235. {
  236. print(c, base);
  237. println();
  238. }
  239. void MarlinSerial::println(unsigned char b, int base)
  240. {
  241. print(b, base);
  242. println();
  243. }
  244. void MarlinSerial::println(int n, int base)
  245. {
  246. print(n, base);
  247. println();
  248. }
  249. void MarlinSerial::println(unsigned int n, int base)
  250. {
  251. print(n, base);
  252. println();
  253. }
  254. void MarlinSerial::println(long n, int base)
  255. {
  256. print(n, base);
  257. println();
  258. }
  259. void MarlinSerial::println(unsigned long n, int base)
  260. {
  261. print(n, base);
  262. println();
  263. }
  264. void MarlinSerial::println(double n, int digits)
  265. {
  266. print(n, digits);
  267. println();
  268. }
  269. // Private Methods /////////////////////////////////////////////////////////////
  270. void MarlinSerial::printNumber(unsigned long n, uint8_t base)
  271. {
  272. unsigned char buf[8 * sizeof(long)]; // Assumes 8-bit chars.
  273. unsigned long i = 0;
  274. if (n == 0) {
  275. print('0');
  276. return;
  277. }
  278. while (n > 0) {
  279. buf[i++] = n % base;
  280. n /= base;
  281. }
  282. for (; i > 0; i--)
  283. print((char) (buf[i - 1] < 10 ?
  284. '0' + buf[i - 1] :
  285. 'A' + buf[i - 1] - 10));
  286. }
  287. void MarlinSerial::printFloat(double number, uint8_t digits)
  288. {
  289. // Handle negative numbers
  290. if (number < 0.0)
  291. {
  292. print('-');
  293. number = -number;
  294. }
  295. // Round correctly so that print(1.999, 2) prints as "2.00"
  296. double rounding = 0.5;
  297. for (uint8_t i=0; i<digits; ++i)
  298. rounding /= 10.0;
  299. number += rounding;
  300. // Extract the integer part of the number and print it
  301. unsigned long int_part = (unsigned long)number;
  302. double remainder = number - (double)int_part;
  303. print(int_part);
  304. // Print the decimal point, but only if there are digits beyond
  305. if (digits > 0)
  306. print(".");
  307. // Extract digits from the remainder one at a time
  308. while (digits-- > 0)
  309. {
  310. remainder *= 10.0;
  311. int toPrint = int(remainder);
  312. print(toPrint);
  313. remainder -= toPrint;
  314. }
  315. }
  316. // Preinstantiate Objects //////////////////////////////////////////////////////
  317. MarlinSerial MSerial;
  318. #endif // whole file
  319. #endif // !AT90USB
  320. // For AT90USB targets use the UART for BT interfacing
  321. #if defined(AT90USB) && defined (BTENABLED)
  322. HardwareSerial bt;
  323. #endif