1 files changed, 50 insertions, 0 deletions
diff --git a/inc/controller.h b/inc/controller.h
new file mode 100644
index 0000000..0cafe10
--- /dev/null
+++ b/inc/controller.h
@@ -0,0 +1,50 @@
+#ifndef CONTROLLER_H
+#define CONTROLLER_H
+#include "stage.h"
+
+/**
+ * Houses the clock, and acts as the main API to the GUI.
+ */
+class Controller : public Stage
+{
+  public:
+	/**
+	 * Constructor.
+	 * @param The storage object to use.
+	 * @param Whether or not efficient pipelining will be used.
+	 * @return A newly allocated controller object.
+	 */
+	Controller(Storage *storage, bool is_pipelined);
+	~Controller();
+
+	/**
+	 * Direct the simulator to run for `number` clock cycles.
+	 * @param the number of clock cycles to run for.
+	 */
+	void run_for(int number);
+	/**
+	 * @return the current clock cycle.
+	 */
+	int get_clock_cycle();
+	/**
+	 * @return a copy of gprs.
+	 */
+	std::array<int, GPR_NUM> get_gprs();
+	/**
+	 * @return the pc.
+	 */
+	int get_pc();
+
+  private:
+	/**
+	 * Helper for run_for.
+	 * Advances the simulation by a single cycle.
+	 */
+	void advance();
+	/**
+	 * The current clock cycle.
+	 */
+	int clock_cycle;
+};
+
+#endif /* CONTROLLER_H_INCLUDED */