From dc47eaa046091d72d9b74499e4ade078ff3763d6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kier=C3=A1n=20Meinhardt?= Date: Fri, 28 Jan 2022 16:35:17 +0100 Subject: krebsdance: add flag to generate directed graph --- krebs/5pkgs/simple/krebsdance/default.nix | 167 ++++++++++++++++++------------ 1 file changed, 103 insertions(+), 64 deletions(-) (limited to 'krebs/5pkgs/simple') diff --git a/krebs/5pkgs/simple/krebsdance/default.nix b/krebs/5pkgs/simple/krebsdance/default.nix index bcb859a21..f200625c9 100644 --- a/krebs/5pkgs/simple/krebsdance/default.nix +++ b/krebs/5pkgs/simple/krebsdance/default.nix @@ -2,113 +2,152 @@ writers.writePython3Bin "krebsdance" {} '' import argparse import random + import itertools claws = [ dict( - up='(\\/)', - down='(/\\)', - left='(\\\\)', - right='(//)', + up="(\\/)", + down="(/\\)", + left="(\\\\)", + right="(//)", ), dict( - up='(V)', - down='(A)', - left='>)=', - right='=(<', + up="(V)", + down="(A)", + left=">)=", + right="=(<", ), dict( - up='(U)', - down='(n)', - left=')==', - right='==(', + up="(U)", + down="(n)", + left=")==", + right="==(", ), ] eyes = [ - '°', - '*', - '^', - 'ö', - 'o', - 'O', - 'X', - 'x', - 'U', - 'u', + "°", + "*", + "^", + "ö", + "o", + "O", + "X", + "x", + "U", + "u", ] bodies = [ dict( - left='(', - right=')', + left="(", + right=")", ), dict( - left='{', - right='}', + left="{", + right="}", ), dict( - left='[', - right=']', + left="[", + right="]", ), dict( - left='<', - right='>', + left="<", + right=">", ), dict( - left='|', - right='|', + left="|", + right="|", ), ] mouths = [ - ',,,,', - ',mm,', - '_mm_', - '-mm-', - ';;;;', - ';mm;', - ':mm:', - '::::', - ':ww:', - ':<>:', + ",,,,", + ",mm,", + "_mm_", + "-mm-", + ";;;;", + ";mm;", + ":mm:", + "::::", + ":ww:", + ":<>:", ] + def all_krebses(): + for mouth, body, eye, claw in itertools.product(mouths, bodies, eyes, claws): + yield f'{claw["up"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {claw["up"]}' + + + def krebs_graph() -> str: + return "\n".join( + ["digraph {"] + + [f'"{krebs}"->"{generate(seed=krebs)}"' for krebs in all_krebses()] + + ["}"] + ) + + + def generate(*, seed: str, dancing: bool = False) -> str: + if seed: + random.seed(seed) + clawstyle = random.choice(claws) + body = random.choice(bodies) + eye = random.choice(eyes) + mouth = random.choice(mouths) + if dancing: + return "\n".join( + [ + f'{clawstyle["down"]} {body["left"]}{eye}{mouth}{eye}{body["right"]}{clawstyle["up"]}', + f'{clawstyle["left"]}{body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["right"]}', + f'{clawstyle["right"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["left"]}', + f'{clawstyle["down"]}{body["left"]}{eye}{mouth}{eye}{body["right"]}{clawstyle["down"]}', + ] + ) + else: + return f'{clawstyle["up"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["up"]}' + + + def fixpoints(): + for krebs in all_krebses(): + if generate(seed=krebs) == krebs: + yield krebs + + def main(): parser = argparse.ArgumentParser() parser.add_argument( - 'seed', - nargs='?', - help='random seed to use for generating the krebs variant', + "seed", + nargs="?", + help="random seed to use for generating the krebs variant", ) parser.add_argument( - '--dance', '-d', - dest='dance', - help='if the krebs should dance', + "--dance", + "-d", + dest="dance", + help="if the krebs should dance", default=False, - action='store_true', + action="store_true", ) - args = parser.parse_args() + parser.add_argument( + "--mode", + "-m", + dest="mode", + choices=["graphviz", "plain"], + default="plain", + ) - if args.seed: - random.seed(args.seed) + args = parser.parse_args() - clawstyle = random.choice(claws) - body = random.choice(bodies) - eye = random.choice(eyes) - mouth = random.choice(mouths) - if args.dance: - print(f'{clawstyle["down"]} {body["left"]}{eye}{mouth}{eye}{body["right"]}{clawstyle["up"]}') # noqa - print(f' {clawstyle["left"]}{body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["right"]}') # noqa - print(f'{clawstyle["right"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["left"]}') # noqa - print(f' {clawstyle["down"]}{body["left"]}{eye}{mouth}{eye}{body["right"]}{clawstyle["down"]}') # noqa - else: - print(f'{clawstyle["up"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {clawstyle["up"]}') # noqa + if args.mode == "plain": + print(generate(seed=args.seed, dancing=args.dance)) + elif args.mode == "graphviz": + print(krebs_graph()) - if __name__ == '__main__': + if __name__ == "__main__": main() '' -- cgit v1.2.3