Optimize dijkstra mapping for volume calculation.

[plomrogue2] / plomrogue / commands.py

diff --git a/plomrogue/commands.py b/plomrogue/commands.py
index fd0863284f04c550fb623dd936070bc65f63ba79..6c356658e20ae48c858652f353f96353d879f6b0 100644 (file)
--- a/plomrogue/commands.py
+++ b/plomrogue/commands.py
@@ -1,6 +1,6 @@
 from plomrogue.misc import quote
 from plomrogue.errors import GameError
-from plomrogue.mapping import YX, MapGeometrySquare, MapGeometryHex
+from plomrogue.mapping import YX, MapGeometrySquare, MapGeometryHex, Map
 
 
 
@@ -10,10 +10,60 @@ def cmd_TASKS(game, connection_id):
 cmd_TASKS.argtypes = ''
 
 def cmd_ALL(game, msg, connection_id):
+
+    def lower_msg_by_volume(msg, volume):
+        lowered_msg = ''
+        for c in msg:
+            c = c
+            while random.random() > volume * 8:
+                if c.isupper():
+                    c = c.lower()
+                elif c != '.':
+                    c = '.'
+                else:
+                    c = ' '
+            lowered_msg += c
+        return lowered_msg
+
+    import random
     if not connection_id in game.sessions:
         raise GameError('need to be logged in for this')
-    t = game.get_thing(game.sessions[connection_id], False)
-    game.io.send('CHAT ' + quote(t.nickname + ': ' + msg))
+    speaker = game.get_thing(game.sessions[connection_id], False)
+    n_max = 255
+    map_size = game.map.size_i
+    dijkstra_map = [n_max for i in range(game.map.size_i)]
+    dijkstra_map[game.map.get_position_index(speaker.position)] = 0
+    shrunk = True
+    while shrunk:
+        shrunk = False
+        for i in range(map_size):
+            if game.map.terrain[i] == 'X':
+                continue
+            neighbors = game.map_geometry.get_neighbors_i(i)
+            for direction in [d for d in neighbors if neighbors[d]]:
+                j = neighbors[direction]
+                if dijkstra_map[j] < dijkstra_map[i] - 1:
+                    dijkstra_map[i] = dijkstra_map[j] + 1
+                    shrunk = True
+    #print('DEBUG')
+    #line_to_print = []
+    #x = 0
+    #for n in dijkstra_map:
+    #    line_to_print += ['%3s' % n]
+    #    x += 1
+    #    if x >= game.map.size.x:
+    #        x = 0
+    #        print(' '.join(line_to_print))
+    for c_id in game.sessions:
+        listener = game.get_thing(game.sessions[c_id], create_unfound=False)
+        listener_vol = dijkstra_map[game.map.get_position_index(listener.position)]
+        volume = 1 / max(1, listener_vol)
+        lowered_msg = lower_msg_by_volume(msg, volume)
+        lowered_nick = lower_msg_by_volume(speaker.nickname, volume)
+        game.io.send('CHAT ' +
+                     quote('(volume: %.2f) %s: %s' % (volume, lowered_nick,
+                                                      lowered_msg)),
+                     c_id)
 cmd_ALL.argtypes = 'string'
 
 def cmd_LOGIN(game, nick, connection_id):
@@ -73,6 +123,9 @@ def cmd_TURN(game, n):
 cmd_TURN.argtypes = 'int:nonneg'
 
 def cmd_ANNOTATE(game, yx, msg, pw, connection_id):
+    player = game.get_thing(game.sessions[connection_id], False)
+    if player.fov_stencil[yx] != '.':
+        raise GameError('cannot annotate tile outside field of view')
     if not game.can_do_tile_with_pw(yx, pw):
         raise GameError('wrong password for tile')
     if msg == ' ':
@@ -84,6 +137,9 @@ def cmd_ANNOTATE(game, yx, msg, pw, connection_id):
 cmd_ANNOTATE.argtypes = 'yx_tuple:nonneg string string'
 
 def cmd_PORTAL(game, yx, msg, pw, connection_id):
+    player = game.get_thing(game.sessions[connection_id], False)
+    if player.fov_stencil[yx] != '.':
+        raise GameError('cannot edit portal on tile outside field of view')
     if not game.can_do_tile_with_pw(yx, pw):
         raise GameError('wrong password for tile')
     if msg == ' ':
@@ -97,17 +153,20 @@ cmd_PORTAL.argtypes = 'yx_tuple:nonneg string string'
 def cmd_GOD_ANNOTATE(game, yx, msg):
     game.annotations[yx] = msg
     game.changed = True
-cmd_ANNOTATE.argtypes = 'yx_tuple:nonneg string'
+cmd_GOD_ANNOTATE.argtypes = 'yx_tuple:nonneg string'
 
 def cmd_GOD_PORTAL(game, yx, msg):
     game.portals[yx] = msg
     game.changed = True
-cmd_PORTAL.argtypes = 'yx_tuple:nonneg string'
+cmd_GOD_PORTAL.argtypes = 'yx_tuple:nonneg string'
 
 def cmd_GET_ANNOTATION(game, yx, connection_id):
-    annotation = '(none)';
-    if yx in game.annotations:
-        annotation = game.annotations[yx]
+    player = game.get_thing(game.sessions[connection_id], False)
+    annotation = '(unknown)';
+    if player.fov_stencil[yx] == '.':
+        annotation = '(none)';
+        if yx in game.annotations:
+            annotation = game.annotations[yx]
     game.io.send('ANNOTATION %s %s' % (yx, quote(annotation)))
 cmd_GET_ANNOTATION.argtypes = 'yx_tuple:nonneg'