#!/usr/bin/env python3
+"""\
+usage: make-graph.py [-a] [-m] [-s]
-# Make a DOT graph of the dungeon
-#
+Make a DOT graph of Colossal Cave
+
+-a = emit graph of entire dungeon
+-m = emit graph of maze all alike
+-s = emit graph of surface locations
+"""
# Copyright (c) 2017 by Eric S. Raymond
# SPDX-License-Identifier: BSD-2-clause
-import sys, yaml
+import sys, yaml, getopt
def allalike(loc):
"Select out loci related to the Maze All Alike"
- return (loc == "LOC_MISTWEST") or ("ALIKE" in loc) or ("MAZEEND" in loc) or ("STALACTITE" in loc)
+ return ("ALIKE" in loc) or (loc == "LOC_PITBRINK") or ("MAZEEND" in loc) or ("STALACTITE" in loc)
+
+def surface(attrs):
+ "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
def abbreviate(d):
m = {"NORTH":"N", "EAST":"E", "SOUTH":"S", "WEST":"W", "UPWAR":"U", "DOWN":"D"}
with open("adventure.yaml", "r") as f:
db = yaml.safe_load(f)
+ try:
+ (options, arguments) = getopt.getopt(sys.argv[1:], "ams")
+ except getopt.GetoptError as e:
+ print(e)
+ sys.exit(1)
+
+ subset = "maze"
+ for (switch, val) in options:
+ if switch == '-a':
+ subset = "all"
+ elif switch == '-m':
+ subset = "maze"
+ elif switch == '-s':
+ subset = "surface"
+ else:
+ sys.stderr.write(__doc__)
+ 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 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 {")
- for (loc, attrs) in db["locations"]:
- if not allalike(loc):
+
+ for (loc, attrs) in db["locations"]:
+ if subset == "surface" and not surface(attrs):
+ continue
+ if subset == "maze" and not allalike(loc):
+ continue;
+ node_label = loc[4:]
+ 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
travel = attrs["travel"]
if len(travel) > 0:
continue
action = dest["action"]
if action[0] == "goto":
- arc = "%s -> %s" % (loc[4:], action[1][4:])
+ dest = 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)
print("}")
-
-