from subprocess import run,PIPE
from importlib import import_module
from sys import path,argv
from pathlib import Path

'''
    Gathers the pinmux table and parses it's output to a more human form, with pin designator included
'''

# get the model for our board, generate an index and prep it
dtname = run(['cat','/proc/device-tree/model'],stdout=PIPE,universal_newlines=True).stdout[:-1]
if len(argv) == 2:
    model = argv[1]
else:
    model = dtname

source_path = Path(__file__).resolve()
source_dir = source_path.parent
path.append(str(source_dir) + '/maps')
board = import_module(model.replace(' ','-'))

muxindex = [pin[0] for pin in board.gpio]
for pin in board.gpio:
    pin.append('')

# Get the full pinmux list
rawlist = run(['sudo', 'cat','/sys/kernel/debug/pinctrl/2000000.pinctrl/pinmux-pins'],stdout=PIPE,universal_newlines=True).stdout
pinlist = rawlist.splitlines()[2:]

# interpret list
for line in pinlist:
    pmux = line.split()
    if int(pmux[1]) in muxindex:
        state = 'free' + ' (' + pmux[1] + ')'
        if pmux[3] == 'device':
            state = pmux[6] + ' (' + pmux[4] + ':' + pmux[1] + ')'
        elif pmux[3] == 'GPIO':
            state = 'gpio (' + pmux[4] + ')'
        board.gpio[muxindex.index(int(pmux[1]))][2] = state

# work out column widths
width = []
total = 33  # this includes all the 'fixed width' text
for c in range(0, board.cols):
    wide = 0
    for p in range(c, len(board.gpio), board.cols):
        w = len(board.gpio[p][2])
        wide = w if w > wide else wide
    width.append(wide)
    total += wide

# heading
header = '{} (dtb name: {})\n'.format(board.name, dtname)
print('\n{}{}'.format(int((total-len(header))/2) * ' ', header))

# Output result.

print('Gpio Header:')
pincount = board.cols * board.rows
# heading
for p in range(0, board.cols):
    if p % 2 == 0:
        print('  {}func   des  pin   '.format(' ' * (width[p] -4)), end='')
    else:
        print('    pin  des   func{}  '.format(' ' * (width[p] - 4)), end='')
print()

# gpio pins
gpiopin = 1
for l in range(0, pincount, board.cols):
    for p in range(l, l + board.cols):
        if board.gpio[p][2] is None:
            board.gpio[p][3] = ''
            board.gpio[p][2] = ''
        pad = (width[p - l] - len(board.gpio[p][2])) * ' '
        if p % 2 == 0:
            print('  {}{} {:>5} {:>3} --o'.format(pad, board.gpio[p][2], str(board.gpio[p][1]), str(gpiopin)), end='')
        else:
            print(' o-- {:3} {:5} {}{}  '.format(str(gpiopin), str(board.gpio[p][1]), board.gpio[p][2], pad), end='')
        gpiopin += 1
    print()

# extras
if len(board.gpio) > pincount:
    print('\nOther gpio outputs of interest:')
    for p in board.gpio[pincount:]:
        print('-- {:5} - {}'.format(str(p[1]), p[2]))

print('\nNotes:')
for l in board.notes:
    print('- ' + l)