#!/usr/bin/env python3
+# SPDX-FileCopyrightText: Copyright Eric S. Raymond <esr@thyrsus.com>
+# SPDX-License-Identifier: BSD-2-Clause
"""\
-usage: make-graph.py [-a] [-m] [-s]
+usage: make_graph.py [-a] [-d] [-m] [-s] [-v]
-Make a DOT graph of Colossal Cave
+Make a DOT graph of Colossal Cave.
-a = emit graph of entire dungeon
+-d = emit graph of maze all different
+-f = emit graph of forest locations
-m = emit graph of maze all alike
--s = emit graph of surface locations
+-s = emit graph of non-forest surface locations
+-v = include internal symbols in room labels
"""
-# Copyright (c) 2017 by Eric S. Raymond
-# SPDX-License-Identifier: BSD-2-clause
-import sys, yaml, getopt
+# pylint: disable=consider-using-f-string,line-too-long,invalid-name,missing-function-docstring,multiple-imports,redefined-outer-name
+
+import sys, getopt, yaml
+
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)
+ return location_lookup[loc]["conditions"].get("ALLALIKE")
+
+
+def alldifferent(loc):
+ "Select out loci related to the Maze All Alike"
+ return location_lookup[loc]["conditions"].get("ALLDIFFERENT")
-def surface(attrs):
+
+def surface(loc):
"Select out surface locations"
- if ("ABOVE" in attrs["conditions"]) and attrs["conditions"]["ABOVE"]:
- return True
- if ("FOREST" in attrs["conditions"]) and attrs["conditions"]["FOREST"]:
- return True
- return False
+ return location_lookup[loc]["conditions"].get("ABOVE")
+
+
+def forest(loc):
+ return location_lookup[loc]["conditions"].get("FOREST")
+
def abbreviate(d):
- m = {"NORTH":"N", "EAST":"E", "SOUTH":"S", "WEST":"W", "UPWAR":"U", "DOWN":"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"]
+ loc_descriptions = location_lookup[loc]["description"]
+ description = ""
+ if debug:
+ description = loc[4:]
+ longd = loc_descriptions["long"]
+ short = loc_descriptions["maptag"] or loc_descriptions["short"]
+ if short is None and longd is not None and len(longd) < 20:
+ short = loc_descriptions["long"]
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[8:]
+ if (
+ short.startswith("in ")
+ or short.startswith("at ")
+ or short.startswith("on ")
+ ):
short = short[3:]
+ if short.startswith("the "):
+ short = short[4:]
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
+ if debug:
+ description += "\\n"
+ description += short
+ if loc in startlocs:
+ description += "\\n(" + ",".join(startlocs[loc]).lower() + ")"
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 is a location that you can't actually stop in - when you go there
+# it ships some message (which is the point) then shifts you to a next location.
# A forwarder has a zero-length array of notion verbs in its travel section.
#
-# Here is an examoke forwarder kocation:
+# Here is an example forwarder declaration:
#
# - LOC_GRUESOME:
# description:
# {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
+ travel = location_lookup[loc]["travel"]
+ return len(travel) == 1 and len(travel[0]["verbs"]) == 0
+
def forward(loc):
"Chase a location through forwarding links."
loc = location_lookup[loc]["travel"][0]["action"][1]
return loc
+
+def reveal(objname):
+ "Should this object be revealed when mapping?"
+ if "OBJ_" in objname:
+ return False
+ if objname == "VEND":
+ return True
+ obj = object_lookup[objname]
+ return not obj.get("immovable")
+
+
if __name__ == "__main__":
- with open("adventure.yaml", "r") as f:
+ with open("adventure.yaml", "r", encoding="ascii", errors="surrogateescape") as f:
db = yaml.safe_load(f)
location_lookup = dict(db["locations"])
+ object_lookup = dict(db["objects"])
try:
- (options, arguments) = getopt.getopt(sys.argv[1:], "ams")
+ (options, arguments) = getopt.getopt(sys.argv[1:], "adfmsv")
except getopt.GetoptError as e:
print(e)
sys.exit(1)
- subset = "maze"
+ subset = allalike
+ debug = False
for (switch, val) in options:
- if switch == '-a':
- subset = "all"
- elif switch == '-m':
- subset = "maze"
- elif switch == '-s':
- subset = "surface"
+ if switch == "-a":
+ # pylint: disable=unnecessary-lambda-assignment
+ subset = lambda loc: True
+ elif switch == "-d":
+ subset = alldifferent
+ elif switch == "-f":
+ subset = forest
+ elif switch == "-m":
+ subset = allalike
+ elif switch == "-s":
+ subset = surface
+ elif switch == "-v":
+ debug = True
else:
sys.stderr.write(__doc__)
- raise SystemExit(1)
+ raise SystemExit(1)
startlocs = {}
for obj in db["objects"]:
objname = obj[0]
location = obj[1].get("locations")
- if "OBJ" not in objname and location != "LOC_NOWHERE" and ("immovable" not in obj[1] or not obj[1]["immovable"]):
+ if location != "LOC_NOWHERE" and reveal(objname):
if location in startlocs:
startlocs[location].append(objname)
else:
startlocs[location] = [objname]
- startlocs = {}
- for obj in db["objects"]:
- objname = obj[0]
- location = obj[1].get("locations")
- if "OBJ" not in objname and location != "LOC_NOWHERE" and ("immovable" not in obj[1] or not obj[1]["immovable"]):
- if location in startlocs:
- startlocs[location].append(objname)
- else:
- startlocs[location] = [objname]
-
- print("digraph G {")
-
+ # Compute reachability, using forwards.
+ # Dictionary key is (from, to) iff its a valid link,
+ # value is corresponding motion verbs.
+ links = {}
+ nodes = []
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 = roomlabel(loc)
- if loc in startlocs:
- node_label += "\\n" + ",".join(startlocs[loc]).lower()
- print(' %s [shape=box,label="%s"]' % (loc[4:], node_label))
-
- for (loc, attrs) in db["locations"]:
- if subset == "surface" and not surface(attrs):
- continue
+ nodes.append(loc)
travel = attrs["travel"]
if len(travel) > 0:
for dest in travel:
action = dest["action"]
if action[0] == "goto":
dest = forward(action[1])
- if subset == "maze" and not (allalike(loc) or allalike(dest)):
- continue;
- arc = "%s -> %s" % (loc[4:], dest[4:])
- label=",".join(verbs).lower()
- if len(label) > 0:
- arc += ' [label="%s"]' % label
- print(" " + arc)
+ if not (subset(loc) or subset(dest)):
+ continue
+ links[(loc, dest)] = verbs
+
+ neighbors = set()
+ for loc in nodes:
+ for (f, t) in links:
+ if f == "LOC_NOWHERE" or t == "LOC_NOWHERE":
+ continue
+ if (f == loc and subset(t)) or (t == loc and subset(f)):
+ if loc not in neighbors:
+ neighbors.add(loc)
+
+ print("digraph G {")
+
+ for loc in nodes:
+ if not is_forwarder(loc):
+ node_label = roomlabel(loc)
+ if subset(loc):
+ print(' %s [shape=box,label="%s"]' % (loc[4:], node_label))
+ elif loc in neighbors:
+ print(' %s [label="%s"]' % (loc[4:], node_label))
+
+ # Draw arcs
+ for (f, t) in links:
+ arc = "%s -> %s" % (f[4:], t[4:])
+ label = ",".join(links[(f, t)]).lower()
+ if len(label) > 0:
+ arc += ' [label="%s"]' % label
+ print(" " + arc)
print("}")
+
+# end