From 14519aa7ee2ada669bea50632a60d1ca6a147502 Mon Sep 17 00:00:00 2001
From: Robert Pelnar <robert.pelnar@seznam.cz>
Date: Wed, 13 Sep 2017 13:35:22 +0200
Subject: [PATCH] Support the second serial (original commit
 7d2d6c6fdcb296e776c7143cc72f6ca2271aa11e on MK2 by AndreeeCZ)

---
 Firmware/MarlinSerial.cpp | 45 +++++++++++++++++++++++++++--
 Firmware/MarlinSerial.h   | 59 ++++++++++++++++++++++++++++-----------
 Firmware/Marlin_main.cpp  | 50 ++++++++++++++++++++++++++++++++-
 3 files changed, 134 insertions(+), 20 deletions(-)

diff --git a/Firmware/MarlinSerial.cpp b/Firmware/MarlinSerial.cpp
index fba87cc8..8606b9f2 100644
--- a/Firmware/MarlinSerial.cpp
+++ b/Firmware/MarlinSerial.cpp
@@ -23,6 +23,8 @@
 #include "Marlin.h"
 #include "MarlinSerial.h"
 
+int selectedSerialPort; 
+
 #ifndef AT90USB
 // this next line disables the entire HardwareSerial.cpp, 
 // this is so I can support Attiny series and any other chip without a UART
@@ -64,6 +66,23 @@ FORCE_INLINE void store_char(unsigned char c)
           store_char(c);
       }
   }
+  
+  SIGNAL(USART2_RX_vect)
+  {
+      if (selectedSerialPort == 1) {
+        // Test for a framing error.
+        if (UCSR2A & (1<<FE2)) {
+            // Characters received with the framing errors will be ignored.
+            // The temporary variable "c" was made volatile, so the compiler does not optimize this out.
+            volatile unsigned char c = UDR2;
+        } else {
+            // Read the input register.
+            unsigned char c = UDR2;
+            store_char(c);
+            
+        }
+      }
+  }
 #endif
 
 // Constructors ////////////////////////////////////////////////////////////////
@@ -88,7 +107,7 @@ void MarlinSerial::begin(long baud)
     useU2X = false;
   }
 #endif
-  
+// set up the first (original serial port)
   if (useU2X) {
     M_UCSRxA = 1 << M_U2Xx;
     baud_setting = (F_CPU / 4 / baud - 1) / 2;
@@ -104,13 +123,35 @@ void MarlinSerial::begin(long baud)
   sbi(M_UCSRxB, M_RXENx);
   sbi(M_UCSRxB, M_TXENx);
   sbi(M_UCSRxB, M_RXCIEx);
+  
+  
+// set up the second serial port
+  if (useU2X) {
+        UCSR2A = 1 << U2X2;
+        baud_setting = (F_CPU / 4 / baud - 1) / 2;
+    } else {
+        UCSR2A = 0;
+        baud_setting = (F_CPU / 8 / baud - 1) / 2;
+    }
+
+    // assign the baud_setting, a.k.a. ubbr (USART Baud Rate Register)
+    UBRR2H = baud_setting >> 8;
+    UBRR2L = baud_setting;
+
+    sbi(UCSR2B, RXEN2);
+    sbi(UCSR2B, TXEN2);
+    sbi(UCSR2B, RXCIE2);
 }
 
 void MarlinSerial::end()
 {
   cbi(M_UCSRxB, M_RXENx);
   cbi(M_UCSRxB, M_TXENx);
-  cbi(M_UCSRxB, M_RXCIEx);  
+  cbi(M_UCSRxB, M_RXCIEx);
+  
+  cbi(UCSR2B, RXEN2);
+  cbi(UCSR2B, TXEN2);
+  cbi(UCSR2B, RXCIE2); 
 }
 
 
diff --git a/Firmware/MarlinSerial.h b/Firmware/MarlinSerial.h
index 6776b568..30f5b83d 100644
--- a/Firmware/MarlinSerial.h
+++ b/Firmware/MarlinSerial.h
@@ -73,6 +73,7 @@
 // is the index of the location from which to read.
 #define RX_BUFFER_SIZE 128
 
+extern int selectedSerialPort;
 
 struct ring_buffer
 {
@@ -110,24 +111,48 @@ class MarlinSerial //: public Stream
     }
     
     
-    FORCE_INLINE void checkRx(void)
+    void checkRx(void)
     {
-        if((M_UCSRxA & (1<<M_RXCx)) != 0) {
-            // Test for a framing error.
-            if (M_UCSRxA & (1<<M_FEx)) {
-                // Characters received with the framing errors will be ignored.
-                // The temporary variable "c" was made volatile, so the compiler does not optimize this out.
-                volatile unsigned char c = M_UDRx;
-            } else {
-                unsigned char c  =  M_UDRx;
-                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;
+        if (selectedSerialPort == 0) {
+            if((M_UCSRxA & (1<<M_RXCx)) != 0) {
+                // Test for a framing error.
+                if (M_UCSRxA & (1<<M_FEx)) {
+                    // Characters received with the framing errors will be ignored.
+                    // The temporary variable "c" was made volatile, so the compiler does not optimize this out.
+                    volatile unsigned char c = M_UDRx;
+                } else {
+                    unsigned char c  =  M_UDRx;
+                    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;
+                    }
+                    selectedSerialPort = 0;
+                }
+            }
+        } else if(selectedSerialPort == 1) {
+            if((UCSR2A & (1<<RXC2)) != 0) {
+                // Test for a framing error.
+                if (UCSR2A & (1<<FE2)) {
+                    // Characters received with the framing errors will be ignored.
+                    // The temporary variable "c" was made volatile, so the compiler does not optimize this out.
+                    volatile unsigned char c = UDR2;
+                } else {
+                    unsigned char c  =  UDR2;
+                    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;
+                    }
+                    selectedSerialPort = 1;
                 }
             }
         }
diff --git a/Firmware/Marlin_main.cpp b/Firmware/Marlin_main.cpp
index dde6cf36..90694b20 100644
--- a/Firmware/Marlin_main.cpp
+++ b/Firmware/Marlin_main.cpp
@@ -1111,6 +1111,17 @@ void setup()
     lcd_print_at_PGM(0, 2, PSTR("    3D  Printers    "));
 	setup_killpin();
 	setup_powerhold();
+	farm_mode = eeprom_read_byte((uint8_t*)EEPROM_FARM_MODE); 
+	EEPROM_read_B(EEPROM_FARM_NUMBER, &farm_no); 
+	//if ((farm_mode == 0xFF && farm_no == 0) || (farm_no == 0xFFFF)) farm_mode = false; //if farm_mode has not been stored to eeprom yet and farm number is set to zero or EEPROM is fresh, deactivate farm mode  
+	if (farm_no == 0xFFFF) farm_no = 0; 
+	if (farm_mode) 
+	{ 
+		prusa_statistics(8); 
+		selectedSerialPort = 1; 
+	}
+	else
+		selectedSerialPort = 0; 
 	MYSERIAL.begin(BAUDRATE);
 	SERIAL_PROTOCOLLNPGM("start");
 	SERIAL_ECHO_START;
@@ -1523,6 +1534,12 @@ void get_command()
 		  rx_buffer_full = true;				//sets flag that buffer was full	
 	  }
     char serial_char = MYSERIAL.read();
+	if (selectedSerialPort == 1)
+	{
+		selectedSerialPort = 0; 
+		MYSERIAL.write(serial_char); 
+		selectedSerialPort = 1; 
+	} 
       TimeSent = millis();
       TimeNow = millis();
 
@@ -2082,6 +2099,9 @@ void homeaxis(int axis)
         axis_known_position[axis] = true;
 #ifdef TMC2130
 		tmc2130_home_exit();
+//        destination[axis] += 2;
+//        plan_buffer_line(destination[X_AXIS], destination[Y_AXIS], destination[Z_AXIS], destination[E_AXIS], homing_feedrate[axis]/60, active_extruder);
+//        st_synchronize();
 #endif
     }
     else if ((axis==Z_AXIS)?HOMEAXIS_DO(Z):0)
@@ -2331,7 +2351,35 @@ void process_commands()
         trace();
         prusa_sd_card_upload = true;
         card.openFile(strchr_pointer+4,false);
-    } else if(code_seen("Fir")){
+	} else if (code_seen("SN")) { 
+        if (farm_mode) { 
+            selectedSerialPort = 0; 
+            MSerial.write(";S"); 
+            // S/N is:CZPX0917X003XC13518 
+            int numbersRead = 0; 
+ 
+            while (numbersRead < 19) { 
+                while (MSerial.available() > 0) { 
+                    uint8_t serial_char = MSerial.read(); 
+                    selectedSerialPort = 1; 
+                    MSerial.write(serial_char); 
+                    numbersRead++; 
+                    selectedSerialPort = 0; 
+                } 
+            } 
+            selectedSerialPort = 1; 
+            MSerial.write('\n'); 
+            /*for (int b = 0; b < 3; b++) { 
+                tone(BEEPER, 110); 
+                delay(50); 
+                noTone(BEEPER); 
+                delay(50); 
+            }*/ 
+        } else { 
+            MYSERIAL.println("Not in farm mode."); 
+        } 
+		
+	} else if(code_seen("Fir")){
 
       SERIAL_PROTOCOLLN(FW_version);