#!/usr/bin/env python3

import re


def gen_wires(fin):
    for l in fin:
        lj = {}
        l = l.strip()
        for kvs in l.split():
            name, value = kvs.split(':')
            lj[name] = value

        tile_type, xy, wname = re.match(
            r'(.*)_(X[0-9]*Y[0-9]*)/(.*)', lj['NAME']).groups()
        lj['tile_type'] = tile_type
        lj['xy'] = xy
        lj['wname'] = wname

        lj['l'] = l

        yield lj


def run(node_fin, verbose=0):
    refnodes = {}
    nodei = 0
    for nodei, anode in enumerate(gen_wires(node_fin)):

        def getk(anode):
            return anode['wname']
            #return (anode['tile_type'], anode['wname'])

        if nodei % 100000 == 0:
            print('Check node %d, %u node types' % (nodei, len(refnodes)))
        # Existing node?
        try:
            refnode = refnodes[getk(anode)]
        except KeyError:
            # Set as reference
            refnodes[getk(anode)] = anode
            continue
        k_invariant = (
            'COST_CODE',
            'IS_INPUT_PIN',
            'IS_OUTPUT_PIN',
            'IS_PART_OF_BUS',
            'NUM_INTERSECTS',
            'NUM_TILE_PORTS',
            'SPEED_INDEX',
            'TILE_PATTERN_OFFSET',
        )
        k_varies = (
            'ID_IN_TILE_TYPE',
            'IS_CONNECTED',
            'NUM_DOWNHILL_PIPS',
            'NUM_PIPS',
            'NUM_UPHILL_PIPS',
            'TILE_NAME',
        )
        # Verify equivilence
        for k in k_invariant:
            if k in refnode and k in anode:

                def fail():
                    print('Mismatch on %s' % k)
                    print(refnode[k], anode[k])
                    print(refnode['l'])
                    print(anode['l'])
                    #assert 0

                if refnode[k] != anode[k]:
                    print
                    fail()
            # A key in one but not the other?
            elif k in refnode or k in anode:
                assert 0
            elif k not in k_varies:
                assert 0


if __name__ == '__main__':
    import argparse

    parser = argparse.ArgumentParser(
        description=
        'Determines which info is consistent across wires with the same name')

    parser.add_argument('--verbose', type=int, help='')
    parser.add_argument(
        'node_fn_in',
        default='specimen_001/wire_all.txt',
        nargs='?',
        help='Input file')
    args = parser.parse_args()
    run(open(args.node_fn_in, 'r'), verbose=args.verbose)