From cfb8df3d49ecb0bbdf55de23d31a2a33a7c20edc Mon Sep 17 00:00:00 2001
From: Christian Heller <c.heller@plomlompom.de>
Date: Thu, 17 Dec 2020 00:58:09 +0100
Subject: [PATCH] Allow negative weariness, but force involuntary dancing to
 get rid of it.

---
 plomrogue/things.py  | 19 ++++++++++++++++---
 rogue_chat_curses.py |  2 +-
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/plomrogue/things.py b/plomrogue/things.py
index 2519798..6637b37 100644
--- a/plomrogue/things.py
+++ b/plomrogue/things.py
@@ -618,6 +618,7 @@ class Thing_Player(ThingAnimate):
     tripping = 0
     need_for_toilet = 0
     standing = True
+    dancing = 0
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -660,12 +661,24 @@ class Thing_Player(ThingAnimate):
         if random.random() > 0.9999:
             if self.standing:
                 self.weariness += 1
-                if self.weariness % 5 == 0:
+            else:
+                self.weariness -= 1
+            if self.weariness > 0 and self.weariness % 5 == 0:
                     self.send_msg('CHAT "All that walking or standing makes '
                                   'you weary, and thereby slower. Find a place '
                                   'to sit or lie down to regain energy."')
-            elif self.weariness > 0:
-                self.weariness -= 1
+            self.game.changed = True
+        if self.dancing and random.random() > 0.99 and not self.next_task[0]:
+            self.dancing -= 1
+            direction = random.choice(self.game.map_geometry.directions)
+            self.set_next_task('MOVE', [direction])
+            if random.random() > 0.9:
+                self.weariness += 1
+                self.game.changed = True
+        if 1000000 * random.random() < -self.weariness:
+            self.send_msg('CHAT "Your body tries to '
+                          'dance off its energy surplus."')
+            self.dancing = 50
             self.game.changed = True
 
     def send_msg(self, msg):
diff --git a/rogue_chat_curses.py b/rogue_chat_curses.py
index bfd5560..e3046cf 100755
--- a/rogue_chat_curses.py
+++ b/rogue_chat_curses.py
@@ -367,7 +367,7 @@ cmd_RANDOM_COLORS.argtypes = ''
 def cmd_STATS(game, bladder_pressure, weariness):
     game.bladder_pressure_new = bladder_pressure
     game.weariness_new = weariness
-cmd_STATS.argtypes = 'int:nonneg int:nonneg'
+cmd_STATS.argtypes = 'int:nonneg int'
 
 class Game(GameBase):
     turn_complete = False
-- 
2.30.2