X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;f=make_graph.py;h=82b86af87d34f474ed6ee02b315f2c971b33f384;hb=0aead5634ee9fc5eb085996c5262b25e798b398f;hp=a914204ac8f1702218784c3c3c25ac2ea642035e;hpb=f6fc7f244d5bccf2bbc3a51877e65f5ec6654073;p=open-adventure.git diff --git a/make_graph.py b/make_graph.py index a914204..82b86af 100755 --- a/make_graph.py +++ b/make_graph.py @@ -6,29 +6,30 @@ Make a DOT graph of Colossal Cave. -a = emit graph of entire dungeon -d = emit graoh of mazw 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 +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(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"} @@ -37,9 +38,13 @@ def abbreviate(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"] + 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:] @@ -47,25 +52,26 @@ def roomlabel(loc): 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 +# 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: +# Here is an example forwarder declaration: # # - LOC_GRUESOME: # description: @@ -88,71 +94,64 @@ def forward(loc): loc = location_lookup[loc]["travel"][0]["action"][1] return loc +def reveal(objname): + "Should this object be revealed when mappinmg?" + 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: db = yaml.safe_load(f) location_lookup = dict(db["locations"]) + object_lookup = dict(db["objects"]) try: - (options, arguments) = getopt.getopt(sys.argv[1:], "adms") + (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" + subset = lambda loc: True elif switch == '-d': - subset = "different" + subset = alldifferent + elif switch == '-f': + subset = forest elif switch == '-m': - subset = "maze" + subset = allalike elif switch == '-s': - subset = "surface" + 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 ke6y is (from, to) iff its a valid link, + # value is correspoinding 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; - if subset == "different" and not alldifferent(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: @@ -162,15 +161,36 @@ if __name__ == "__main__": action = dest["action"] if action[0] == "goto": dest = forward(action[1]) - if subset == "maze" and not (allalike(loc) or allalike(dest)): - continue; - if subset == "different" and not (alldifferent(loc) or alldifferent(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