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

Hourier · Sep 18, 2021 · cb409ab · cb409ab
1 parent 711ed76
commit cb409ab
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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
@@ -16,6 +16,7 @@ class PlayerStun {
     StunRank get_rank() const;
     StunRank get_rank(short value) const;
     int decrease_chance() const;
+    void reset();
     void set(short value);
 
 private: