1 files changed, 39 insertions, 0 deletions
diff --git a/input.c b/input.c
new file mode 100644
index 0000000..03f0a0e
--- /dev/null
+++ b/input.c
@@ -0,0 +1,39 @@
+#include "input.h"
+#include "lzr.h"
+
+static const unsigned int keys[5] = {LZR_BUTTON_LEFT, LZR_BUTTON_RIGHT,
+                                     LZR_BUTTON_UP, LZR_BUTTON_DOWN,
+                                     LZR_BUTTON_O};
+static int states[5] = {0};
+
+void input_update(void)
+{
+	int i = 5;
+	while (i-- > 0)
+		if (LZR_ButtonDown(keys[i]))
+			states[i] =
+			    (states[i] == KS_UP) ? (KS_PRESSED) : (KS_DOWN);
+		else
+			states[i] = KS_UP;
+}
+
+int input_up(unsigned int k)
+{
+	if (k >= 5)
+		return 0;
+	return states[k] == KS_UP;
+}
+
+int input_down(unsigned int k)
+{
+	if (k >= 5)
+		return 0;
+	return states[k] == KS_DOWN || states[k] == KS_PRESSED;
+}
+
+int input_pressed(unsigned int k)
+{
+	if (k >= 5)
+		return 0;
+	return states[k] == KS_PRESSED;
+}