1 files changed, 59 insertions, 0 deletions
diff --git a/src/controller.rs b/src/controller.rs
new file mode 100644
index 0000000..cc01d0f
--- /dev/null
+++ b/src/controller.rs
@@ -0,0 +1,59 @@
+
+pub trait Controller {
+    fn read(&self) -> u8;
+    fn write(&self, data: u8);
+}
+
+pub mod stdctl {
+    use core::cell::Cell;
+    use controller::Controller;
+    #[derive(Copy, Clone)]
+    pub enum Button {
+        A = 0,
+        B = 1,
+        Select = 2,
+        Start = 3,
+        Up = 4,
+        Down = 5,
+        Left = 6,
+        Right = 7,
+        Null = 8,
+    }
+    pub struct Joystick {
+        strobe: Cell<bool>,
+        reg: Cell<u8>,
+        back_reg: Cell<u8>
+    }
+
+    impl Joystick {
+        pub fn new() -> Self {
+            Joystick{reg: Cell::new(0), strobe: Cell::new(false), back_reg: Cell::new(0)}
+        }
+
+        pub fn set(&self, buttons: &[bool]) {
+            let mut reg = 0;
+            for (i, v) in buttons.iter().enumerate() {
+                if *v {
+                    reg |= 1 << i;
+                }
+            }
+            self.reg.set(reg);
+            self.back_reg.set(reg);
+        }
+    }
+
+    impl Controller for Joystick {
+        fn read(&self) -> u8 {
+            let res = self.reg.get() & 1;
+            if !self.strobe.get() {
+                self.reg.set(self.reg.get() >> 1);
+            }
+            res
+        }
+        
+        fn write(&self, data: u8) {
+            self.strobe.set(data & 1 == 1);
+            self.reg.set(self.back_reg.get());
+        }
+    }
+}