Merge pull request #46 from bdunahu/bdunahu

 Ensure all stages only do work if they are not 'OK' 

1 files changed, 19 insertions, 1 deletions

diff --git a/inc/wb.h b/inc/wb.h
index 62cef17..c4755ea 100644
--- a/inc/wb.h
+++ b/inc/wb.h
@@ -14,9 +14,27 @@ class WB : public Stage
 	 */
 	WB(Stage *next);
 	using Stage::advance;
-	
+
   private:
 	void advance_helper() override;
+	/**
+	 * Performs the actual work of storing into a register.
+	 */
+	void write_handler();
+	/**
+	 * Performs the actual work of processing a jump instruction.
+	 */
+	void jump_handler();
+	/**
+	 * @return true if the current instruction is an R or I type and is not a
+	 * STORE.
+	 */
+	bool should_write();
+	/**
+	 * @return true if the current instruction is a J type and is not a push.
+	 * STORE.
+	 */
+	bool should_jump();
 };
 
 #endif /* WB_H_INCLUDED */