fix command design pattern

1 files changed, 10 insertions, 6 deletions

diff --git a/ToggleMuseumPauseCommand.cpp b/ToggleMuseumPauseCommand.cpp
index 57176af..bdd2ef6 100644
--- a/ToggleMuseumPauseCommand.cpp
+++ b/ToggleMuseumPauseCommand.cpp
@@ -1,13 +1,17 @@
 #include "ToggleMuseumPauseCommand.h"
 #include "Museum.h"
 
-void ToggleMuseumPauseCommand::set(bool paused) {
-	Museum & museum = this->get_museum();
-	museum.set_pause(paused);
+ToggleMuseumPauseCommand::ToggleMuseumPauseCommand(Museum & m) : museum(m) {
+	this->toggle = true;
 }
 
-void ToggleMuseumPauseCommand::toggle() {
-	Museum & museum = this->get_museum();
-	museum.set_pause(!museum.get_pause());
+ToggleMuseumPauseCommand::ToggleMuseumPauseCommand(Museum & m, bool set) : museum(m) {
+	this->toggle = false;
+	this->value = set;
+}
+
+void ToggleMuseumPauseCommand::execute() {
+	if (this->toggle) this->value = !this->museum.get_pause();
+	this->museum.set_pause(this->value);
 }