Function main

tools/custom/decomp_switch.py:245–287

()

def main():
    code_str = ""
    if len(argv) < 2:
        code_str = raw_input()
    else:
        if argv[1].casefold() == "-h":
            show_help()
            return

        with open(argv[1], "r") as f:
            code_str = f.readlines()

    flow = build_flow(code_str)
    min_iter = int(input("Min value (inclusive) for switch statement:"))
    max_iter = int(input("Max value (inclusive) for switch statement:")) + 1

    # Dictionary format
    # cases = dict[str, Result]
    cases = dict()

    for i in range(min_iter, max_iter):
        case = traverse_flow(flow, i)
        if not case.dest in cases:
            cases[case.dest] = list()
        cases[case.dest].append(Result(case.branch, i))

    print("====================================================")
    print("Switch analysis:")
    print("====================================================")
    print("(Labels are sorted by their order in the asm,")
    print("so they should be in the original order.)")
    print("====================================================")
    print("Asterisk symbol (*) represents BEQ cases,")
    print("or cases guaranteed to be explicitly specified.")
    print("====================================================")
    print("This can help you pick out explicit cases from the default cases.")
    print("====================================================")

    cases_list = [case for case in cases]
    for case in sorted(cases_list, key=custom_key):
        print(f"Label: {case}")
        values = [str(x) for x in cases[case]]
        print(f"    - Cases: {', '.join(values)}")


main()