Add logic for chasing forwarding limks to the graph maker.

[open-adventure.git] / make_graph.py

diff --git a/make_graph.py b/make_graph.py
index 6623aff976f2019694a6b32ad6a57e1ced34e314..90e637d0df2bed9fc19c22937ed97e3e3491e617 100755 (executable)
--- a/make_graph.py
+++ b/make_graph.py
@@ -15,7 +15,7 @@ import sys, yaml, getopt
 
 def allalike(loc):
     "Select out loci related to the Maze All Alike"
-    return ("ALIKE" in loc) or (loc == "LOC_PITBRINK") or ("MAZEEND" in loc) or ("STALACTITE" in loc) or (loc == "LOC_MISTWEST")
+    return ("ALIKE" in loc) or (loc == "LOC_PITBRINK") or ("MAZEEND" in loc) or ("STALACTITE" in loc)
 
 def surface(attrs):
     "Select out surface locations"
@@ -29,10 +29,66 @@ def abbreviate(d):
     m = {"NORTH":"N", "EAST":"E", "SOUTH":"S", "WEST":"W", "UPWAR":"U", "DOWN":"D"}
     return m.get(d, d)
 
+def roomlabel(loc):
+    "Generate a room label from the description, if possible"
+    loc_descriptions = location_lookup[loc]['description']
+    description = loc[4:]
+    short = loc_descriptions["short"]
+    maptag = loc_descriptions["maptag"]
+    if short is not None:
+        if short.startswith("You're "):
+            short = short[7:]
+        if short.startswith("You are "):
+            short = short[8 :]
+        if short.startswith("in ") or short.startswith("at ") or short.startswith("on "):
+            short = short[3:]
+        if short[:3] in {"n/s", "e/w"}:
+            short = short[:3].upper() + short[3:]
+        elif short[:2] in {"ne", "sw", "se", "nw"}:
+            short = short[:2].upper() + short[2:]
+        else:
+            short = short[0].upper() + short[1:]
+    elif loc_descriptions["maptag"] is not None:
+        short = loc_descriptions["maptag"]
+    elif loc_descriptions["long"] is not None and len(loc_descriptions["long"]) < 20:
+        short = loc_descriptions["long"]
+    if short is not None:
+        description += "\\n" + short
+    return description
+
+# A forwarder is a location tat you can't actually stop in - when you go there
+# it ships some message (which is the point) then shifts you to a nexr location.
+# A forwarder has a zero-length array of notion verbs in its travel section.
+#
+# Here is an examoke forwarder kocation:
+#
+# - LOC_GRUESOME:
+#    description:
+#      long: 'There is now one more gruesome aspect to the spectacular vista.'
+#      short: !!null
+#      maptag: !!null
+#    conditions: {DEEP: true}
+#    travel: [
+#      {verbs: [], action: [goto, LOC_NOWHERE]},
+#    ]
+
+def is_forwarder(loc):
+    "Is a location a forwarder?"
+    travel = location_lookup[loc]['travel']
+    return len(travel) ==h 1 and len(travel[0]['verbs']) == 0
+
+def forward(loc):
+    "Chase a location through forwarding links."
+    while is_forwarder(loc):
+        loc = location_lookup[loc]["travel"][0]["action"][1]
+    return loc
+
 if __name__ == "__main__":
     with open("adventure.yaml", "r") as f:
         db = yaml.safe_load(f)
 
+    location_lookup = dict(db["locations"])
+
     try:
         (options, arguments) = getopt.getopt(sys.argv[1:], "ams")
     except getopt.GetoptError as e:
@@ -73,12 +129,14 @@ if __name__ == "__main__":
 
     print("digraph G {")
     
-    for (loc, attrs) in db["locations"]:        
+    for (loc, attrs) in db["locations"]:
+        if is_forwarder(loc):
+            continue
         if subset == "surface" and not surface(attrs):
             continue
         if subset == "maze" and not allalike(loc):
             continue;
-        node_label = loc[4:]
+        node_label = roomlabel(loc)
         if loc in startlocs:
             node_label += "\\n" + ",".join(startlocs[loc]).lower()
         print('    %s [shape=box,label="%s"]' % (loc[4:], node_label))
@@ -94,7 +152,7 @@ if __name__ == "__main__":
                     continue
                 action = dest["action"]
                 if action[0] == "goto":
-                    dest = action[1]
+                    dest = forward(action[1])
                     if subset == "maze" and not (allalike(loc) or allalike(dest)):
                         continue;
                     arc = "%s -> %s" % (loc[4:], dest[4:])