[Refactor] #1498 Defined set() and reset()

diff --git a/src/timed-effect/player-stun.cpp b/src/timed-effect/player-stun.cpp
index 489020b..2326e20 100644 (file)
--- a/src/timed-effect/player-stun.cpp
+++ b/src/timed-effect/player-stun.cpp
@@ -52,4 +52,10 @@ int PlayerStun::decrease_chance() const
 
 void PlayerStun::set(short value)
 {
+    this->stun = value;
+}
+
+void PlayerStun::reset()
+{
+    this->set(0);
 }

diff --git a/src/timed-effect/player-stun.h b/src/timed-effect/player-stun.h
index 9d58ce8..48b252c 100644 (file)
--- a/src/timed-effect/player-stun.h
+++ b/src/timed-effect/player-stun.h
@@ -16,6 +16,7 @@ public:
     StunRank get_rank() const;
     StunRank get_rank(short value) const;
     int decrease_chance() const;
+    void reset();
     void set(short value);
 
 private: