1 files changed, 42 insertions, 0 deletions

diff --git a/infra/arbiter.h b/infra/arbiter.h
new file mode 100644
index 0000000..5dd1647
--- /dev/null
+++ b/infra/arbiter.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include <cassert>
+#include <optional>
+#include <utility>
+
+#include "infra/sim.h"
+
+namespace infra {
+    template<typename T, unsigned int peers> struct priority_arbiter : public sim {
+        std::array<port<T>, peers> peerp;
+        port<T> *outp = nullptr;
+
+        void clock() {
+            for (unsigned int i = 0; i < peers; ++i) {
+                if (outp->can_write() && peerp[i].can_read())
+                    outp->write(peerp[i].read());
+            }
+        }
+    };
+
+    template<typename T, unsigned int peers> struct round_robin_arbiter : public sim {
+        std::array<port<T>, peers> peerp;
+        port<T> *outp = nullptr;
+        unsigned int initial = 0;
+
+        void clock() {
+            bool initially_empty = outp->can_write();
+            for (unsigned int i = initial; i < peers; ++i) {
+                if (outp->can_write() && peerp[i].can_read())
+                    outp->write(peerp[i].read());
+            }
+            for (unsigned int i = 0; i < initial; ++i) {
+                if (outp->can_write() && peerp[i].can_read())
+                    outp->write(peerp[i].read());
+            }
+            if (initially_empty && !outp->can_write())
+                if (++initial == peers)
+                    initial = 0;
+        }
+    };
+}