From eb3fd68203fee7c63245c702914c2acd3332d65a Mon Sep 17 00:00:00 2001
From: Julian Blake Kongslie
Date: Thu, 22 Sep 2022 11:29:07 -0700
Subject: Initial commit.

---
 infra/arbiter.h | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 infra/arbiter.h

(limited to 'infra/arbiter.h')

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;
+        }
+    };
+}
-- 
cgit v1.2.3