X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;f=make_graph.py;h=1cb975c1e95e5ad0e72bd9d8f27b126a27f63571;hb=dee8809e3091a9e06f79fcc47f9a38c4183894e5;hp=e90dc5a98d7c5f97efff6c8a410d0d8f8e609b0e;hpb=2df951ba6041023122137137146427e352902ba1;p=open-adventure.git diff --git a/make_graph.py b/make_graph.py index e90dc5a..1cb975c 100755 --- a/make_graph.py +++ b/make_graph.py @@ -1,25 +1,151 @@ #!/usr/bin/env python3 +"""\ +usage: make-graph.py [-a] -d] [-m] [-s] -# Make a DOT graph of the dungeon -# +Make a DOT graph of Colossal Cave. + +-a = emit graph of entire dungeon +-d = emit graoh of mazw all different +-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, getopt, yaml -def allalike(loc, dest): +def allalike(loc): "Select out loci related to the Maze All Alike" - return ("ALIKE" in loc) or ("MAZEEND" in loc) or ("STALACTITE" in loc) or (loc == "LOC_MISTWEST" and "ALIKE" in dest) + return ("ALIKE" in loc) or (loc == "LOC_PITBRINK") or ("MAZEEND" in loc) or ("STALACTITE" in loc) + +def alldifferent(loc): + "Select out loci related to the Maze All Alike" + return ("DIFFERENT" in loc) or (loc == "LOC_DEADEND13") + +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 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:] + 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[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:] + description += "\\n" + short + 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. +# 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) == 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:], "adms") + except getopt.GetoptError as e: + print(e) + sys.exit(1) + + subset = allalike + for (switch, val) in options: + if switch == '-a': + subset = lambda loc: True + elif switch == '-d': + subset = alldifferent + elif switch == '-m': + subset = allalike + 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 is_forwarder(loc): + continue + if not subset(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"]: travel = attrs["travel"] if len(travel) > 0: @@ -29,9 +155,9 @@ if __name__ == "__main__": continue action = dest["action"] if action[0] == "goto": - dest = action[1] - if not allalike(loc, dest): - continue; + dest = forward(action[1]) + if not (subset(loc) or subset(dest)): + continue arc = "%s -> %s" % (loc[4:], dest[4:]) label=",".join(verbs).lower() if len(label) > 0: @@ -39,4 +165,4 @@ if __name__ == "__main__": print(" " + arc) print("}") - +# end