From cb409ab2ce6c2440060051eec895710982644f8e Mon Sep 17 00:00:00 2001
From: Hourier <66951241+Hourier@users.noreply.github.com>
Date: Fri, 17 Sep 2021 17:18:18 +0900
Subject: [PATCH] [Refactor] #1498 Defined set() and reset()

---
 src/timed-effect/player-stun.cpp | 6 ++++++
 src/timed-effect/player-stun.h   | 1 +
 2 files changed, 7 insertions(+)

diff --git a/src/timed-effect/player-stun.cpp b/src/timed-effect/player-stun.cpp
index 489020b27..2326e2026 100644
--- 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 9d58ce857..48b252c00 100644
--- 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:
-- 
2.11.0