X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;f=make_graph.py;h=5a58895a39b0715b504ae620a632576a1151e99b;hb=258b7703f22045be1b6e9fb74bd3391ed94ed86b;hp=16ebcba725c8cf1b503c9da1e2be8332449f87ab;hpb=60d70705063266ed23b1e86d11145770087f660f;p=open-adventure.git diff --git a/make_graph.py b/make_graph.py index 16ebcba..5a58895 100755 --- a/make_graph.py +++ b/make_graph.py @@ -1,45 +1,58 @@ #!/usr/bin/env python3 +# SPDX-FileCopyrightText: (C) Eric S. Raymond +# SPDX-License-Identifier: BSD-2-Clause """\ -usage: make-graph.py [-a] -d] [-m] [-s] +usage: make_graph.py [-a] [-d] [-m] [-s] [-v] 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 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 +# 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 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(loc): "Select out surface locations" 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'] + loc_descriptions = location_lookup[loc]["description"] description = "" if debug: description = loc[4:] @@ -51,8 +64,12 @@ def roomlabel(loc): 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:] @@ -69,8 +86,9 @@ def roomlabel(loc): description += "\\n(" + ",".join(startlocs[loc]).lower() + ")" 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: @@ -85,10 +103,12 @@ def roomlabel(loc): # {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 + travel = location_lookup[loc]["travel"] + return len(travel) == 1 and len(travel[0]["verbs"]) == 0 + def forward(loc): "Chase a location through forwarding links." @@ -96,8 +116,9 @@ def forward(loc): loc = location_lookup[loc]["travel"][0]["action"][1] return loc + def reveal(objname): - "Should this object be revealed when mappinmg?" + "Should this object be revealed when mapping?" if "OBJ_" in objname: return False if objname == "VEND": @@ -105,8 +126,9 @@ def reveal(objname): obj = object_lookup[objname] return not obj.get("immovable") + if __name__ == "__main__": - with open("adventure.yaml", "r", encoding='ascii', errors='surrogateescape') as f: + with open("adventure.yaml", "r", encoding="ascii", errors="surrogateescape") as f: db = yaml.safe_load(f) location_lookup = dict(db["locations"]) @@ -121,17 +143,18 @@ if __name__ == "__main__": subset = allalike debug = False for (switch, val) in options: - if switch == '-a': + if switch == "-a": + # pylint: disable=unnecessary-lambda-assignment subset = lambda loc: True - elif switch == '-d': + elif switch == "-d": subset = alldifferent - elif switch == '-f': + elif switch == "-f": subset = forest - elif switch == '-m': + elif switch == "-m": subset = allalike - elif switch == '-s': + elif switch == "-s": subset = surface - elif switch == '-v': + elif switch == "-v": debug = True else: sys.stderr.write(__doc__) @@ -148,8 +171,8 @@ if __name__ == "__main__": 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 = [] for (loc, attrs) in db["locations"]: @@ -170,7 +193,7 @@ if __name__ == "__main__": neighbors = set() for loc in nodes: for (f, t) in links: - if f == 'LOC_NOWHERE' or t == 'LOC_NOWHERE': + 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: @@ -189,7 +212,7 @@ if __name__ == "__main__": # Draw arcs for (f, t) in links: arc = "%s -> %s" % (f[4:], t[4:]) - label=",".join(links[(f, t)]).lower() + label = ",".join(links[(f, t)]).lower() if len(label) > 0: arc += ' [label="%s"]' % label print(" " + arc)