Make a DOT graph of Colossal Cave.
-a = emit graph of entire dungeon
--d = emit graoh of mazw all different
+-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
--v = include internal sy,no;s in room labels
+-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
+# 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 ("DIFFERENT" in loc) or (loc == "LOC_DEADEND13")
+ return location_lookup[loc]["conditions"].get("ALLDIFFERENT")
def surface(loc):
"Select out surface locations"
- attrs = location_lookup[loc]
- 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"}
return description
# 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 nexr location.
+# 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 example forwarder declaration:
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:], "admsv")
+ (options, arguments) = getopt.getopt(sys.argv[1:], "adfmsv")
except getopt.GetoptError as e:
print(e)
sys.exit(1)
subset = lambda loc: True
elif switch == '-d':
subset = alldifferent
+ elif switch == '-f':
+ subset = forest
elif switch == '-m':
subset = allalike
elif switch == '-s':
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]
# Compute reachability, using forwards.
- # Dictionary ke6y is (from, to) iff its a valid link,
- # value is correspoinding motion verbs.
+ # Dictionary key is (from, to) iff its a valid link,
+ # value is corresponding motion verbs.
links = {}
- nodes = set()
+ nodes = []
for (loc, attrs) in db["locations"]:
- nodes.add(loc)
+ nodes.append(loc)
travel = attrs["travel"]
if len(travel) > 0:
for dest in travel:
print("digraph G {")
for loc in nodes:
- if is_forwarder(loc):
- continue
- 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))
+ 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: