more fixes

1 files changed, 15 insertions, 0 deletions

diff --git a/backend/Player.cpp b/backend/Player.cpp
index 8a44a3a..46d924f 100644
--- a/backend/Player.cpp
+++ b/backend/Player.cpp
@@ -1,6 +1,7 @@
 #include <assert.h>
 
 #include "Player.h"
+#include "Exception.h"
 #include "print.h"
 #include "Dungeon.h"
 #include "util.h"
@@ -46,6 +47,20 @@ void Player::set_location(Location & location) {
 	lprtf("Je staat nu bij de locatie %s.\n", location.get_name().c_str());
 }
 
+void Player::equip(Object * object) {
+	if (object == nullptr) return;
+
+	auto weapon = dynamic_cast<WeaponObject *>(object);
+	if (weapon != nullptr)
+		return this->equip(weapon);
+
+	auto armor = dynamic_cast<ArmorObject *>(object);
+	if (armor != nullptr)
+		return this->equip(armor);
+
+	throw Exception("object \"%s\" is niet draagbaar", object->get_name().c_str());
+}
+
 void Player::equip(WeaponObject * weapon) {
 	if (weapon == nullptr) return;