From 58dfb817362225e21c5c4c594f8da764376cb0bf Mon Sep 17 00:00:00 2001
From: Azalea Gui <me@hydev.org>
Date: Tue, 7 Mar 2023 23:48:55 -0500
Subject: [PATCH] [O] Reset position on rollback

---
 tngame/telnet.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tngame/telnet.py b/tngame/telnet.py
index c3b28c3..39379ef 100644
--- a/tngame/telnet.py
+++ b/tngame/telnet.py
@@ -24,6 +24,7 @@ ASCII_HEIGHT = ASCII_CAT.count('\n')
 ASCII_WIDTH = max(len(line.strip('\n')) for line in ASCII_CAT.splitlines())
 
 SNOW_DENSITY = 0.05  # Snow particles per pixel on screen
+SNOW_SPEED = 8  # Snow fall speed in pixels per second
 
 
 COLORS = [RGB.from_hex(v) for v in {
@@ -54,6 +55,9 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
     snow: list[SnowParticle]
     last_update_ns: int
 
+    def rand_velocity() -> tuple[float, float]:
+        return random.randrange(-1, 1) * SNOW_SPEED, random.randrange(1, 2) * SNOW_SPEED
+
     # Create snow particles
     def create_snow(count: int | None) -> list[SnowParticle]:
         # Calculate snow particle count based on screen size and density
@@ -68,8 +72,7 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
             y = random.randint(0, height)
 
             # Generate random x and y velocity
-            xv = random.randint(-1, 1)
-            yv = random.randint(1, 2)
+            xv, yv = rand_velocity()
 
             # Generate random color
             color = random.choice(COLORS)
@@ -100,9 +103,10 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
 
             if particle.y >= height:
                 particle.y = 0
+                particle.xv, particle.yv = rand_velocity()
 
             # Draw snow particle
-            writer.write(f'\x1b[{int(particle.y)};{int(particle.x)}H'
+            writer.write(f'\x1b[{round(particle.y)};{round(particle.x)}H'
                          f'{particle.color.to_ansi_rgb()}*'
                          f'\x1b[0m')