1 files changed, 40 insertions, 0 deletions
diff --git a/src/arch/atmega2560/driver/stdout.cc b/src/arch/atmega2560/driver/stdout.cc
new file mode 100644
index 0000000..58a3e7e
--- /dev/null
+++ b/src/arch/atmega2560/driver/stdout.cc
@@ -0,0 +1,40 @@
+/*
+ * Copyright 2021 Daniel Friesel
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+#include "driver/stdout.h"
+#include <avr/io.h>
+#include <avr/interrupt.h>
+
+#ifndef BAUD
+#define BAUD 115200UL
+#endif
+
+#include <util/setbaud.h>
+
+void StandardOutput::setup()
+{
+	UBRR0H = UBRRH_VALUE;
+	UBRR0L = UBRRL_VALUE;
+
+#if USE_2X
+	UCSR0A |= _BV(U2X0);
+#else
+	UCSR0A &= ~_BV(U2X0);
+#endif
+
+	UCSR0B |= _BV(RXEN0) | _BV(TXEN0);
+	UCSR0C = _BV(UCSZ01) | _BV(UCSZ00);
+}
+
+void StandardOutput::put(char c)
+{
+	while (!(UCSR0A & _BV(UDRE0)));
+	UDR0 = c;
+	if (c == '\n') {
+		put('\r');
+	}
+}
+
+StandardOutput kout;