Sort AIG PIs by name

sort_pis_by_name.py

# This script requires pyaig and click, they can be installed by running:
# > pip install pyaig click
# run the script as follows:
# > python sort_pis_by_name.py input.aig output.aig
# output.aig would the same as input.aig, except that the inputs would be sorted by name
# the script assumes all PIs have names

import click
from pyaig import *


@click.command()
@click.argument('input', type=click.File("rb"))
@click.argument('output', type=click.File("wb"))
def main(input, output):
    src = read_aiger(input)
    dst = AIG()
    pis = [ (pi, src.get_name_by_id(pi)) for pi in src.get_pis() ]
    M = AIG.fmap()
    for pi, name in sorted(pis, key=lambda x: x[1]):
        M[pi] = dst.create_pi(name)
    dst.compose(src, M)    
    write_aiger(dst, output)

    
if __name__ == "__main__":
    main()