1 files changed, 104 insertions, 0 deletions
diff --git a/src/arch/atmega2560/arch.cc b/src/arch/atmega2560/arch.cc
new file mode 100644
index 0000000..9d7e101
--- /dev/null
+++ b/src/arch/atmega2560/arch.cc
@@ -0,0 +1,104 @@
+/*
+ * Copyright 2021 Daniel Friesel
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+#include "arch.h"
+#include <avr/io.h>
+#include <avr/interrupt.h>
+#include <avr/wdt.h>
+#include <util/delay.h>
+
+void Arch::setup(void)
+{
+	wdt_disable();
+
+#if F_CPU == 16000000UL
+	/* default */
+#else
+#error Unsupported F_CPU
+#endif
+
+#if defined(WITH_LOOP) || defined(TIMER_S)
+	TCCR1A = 0;
+	TCCR1B = _BV(WGM12) | _BV(CS12) | _BV(CS10); // /1024
+	OCR1A = F_CPU / 1024;
+	TIMSK1 = _BV(OCIE1A);
+#endif
+
+	sei();
+}
+
+#ifdef WITH_WAKEUP
+void wakeup();
+#endif
+
+#if defined(WITH_LOOP) || defined(TIMER_S)
+
+#include "driver/uptime.h"
+
+#endif
+
+#if defined(WITH_LOOP)
+extern void loop();
+volatile char run_loop = 0;
+#endif
+
+void Arch::idle_loop(void)
+{
+	while (1) {
+		SMCR = _BV(SE);
+		asm("sleep");
+		SMCR = 0;
+		asm("wdr");
+#ifdef WITH_LOOP
+		if (run_loop) {
+			loop();
+			run_loop = 0;
+		}
+#endif
+#ifdef WITH_WAKEUP
+		wakeup();
+#endif
+	}
+}
+
+void Arch::idle(void)
+{
+	SMCR = _BV(SE);
+	asm("sleep");
+	SMCR = 0;
+	asm("wdr");
+}
+
+void Arch::delay_us(unsigned int const us)
+{
+	for (unsigned int i = 0; i < us; i++) {
+		_delay_us(1);
+	}
+}
+
+void Arch::delay_ms(unsigned int const ms)
+{
+	for (unsigned int i = 0; i < ms; i++) {
+		_delay_ms(1);
+	}
+}
+
+Arch arch;
+
+#if defined(WITH_LOOP) || defined(TIMER_S)
+
+#ifndef __acweaving
+ISR(TIMER1_COMPA_vect)
+{
+#ifdef WITH_LOOP
+	run_loop = 1;
+#endif
+#ifdef TIMER_S
+	uptime.tick_s();
+#endif
+}
+#endif
+
+#endif