#!/usr/bin/python
# -*- coding: UTF-8 -*-
# vim: ts=4 sw=4 et:

"""
The script sorts SND_PCI_QUIRK groups by vendor and product id pair.
"""

__author__ = "Andy Shevchenko <andy@smile.org.ua>"
__revision__ = "$Id$"

import sys
import os
import re

def read_file(fname):
    """ Read given file """
    fdes = open(fname, 'rt')
    result = fdes.readlines()
    fdes.close()
    return result

def save_file(fname, lines):
    """ Write lines to the given file """
    fdes = open(fname, 'wt')
    fdes.writelines(lines)
    fdes.flush()
    fdes.close()

def get_vpid(vid, pid):
    """ Construct value of the vendor and product id pair """
    vconv = {
        'PCI_VENDOR_ID_INTEL':  '0x8086',
        'PCI_VENDOR_ID_DELL':   '0x1028',
    }
    if vid in vconv.keys():
        vid = vconv[vid]
    return (int(vid, 16) << 16) + int(pid, 16)

def parse_quirks(quirks):
    """ Parse buffer as SND_PCI_QUIRK group """
    rexp_begin = re.compile("^\s*[/\*]*\s*SND_PCI_QUIRK\((?P<vid>[^,]+),\s*(?P<pid>[^,]+)")
    rexp_end = re.compile("\),\s*(/\*.+\*/)?\s*$")

    xbuffer = {} # vendor, product: prefix lines + line + postfix lines

    flag = False
    vpid = 0
    prefix = []

    for line in quirks:
        if line.strip() == '':
            continue
        rmatch = rexp_begin.match(line)
        if rmatch:
            vpid = get_vpid(rmatch.group('vid'), rmatch.group('pid'))

            if xbuffer.has_key(vpid):
                print "Warning: the line '%s' is duplicate!" % line.strip()
                xbuffer[vpid].append(line)
            else:
                xbuffer[vpid] = prefix
                xbuffer[vpid].append(line)
                prefix = []

            if rexp_end.search(line):
                flag = False
            else:
                flag = True
        else:
            if flag == False:
                prefix.append(line)
            else:
                xbuffer[vpid].append(line)

            if rexp_end.search(line):
                flag = False
    return xbuffer

def sort_quirks(quirks):
    """ Sort SND_PCI_QUIRK group by vendor and product id pair """
    xbuffer = []

    keys = quirks.keys()
    keys.sort()

    for key in keys:
        xbuffer.extend(quirks[key])

    return xbuffer

def catch_quirk_lines(lines):
    """ Catch SND_PCI_QUIRK group from lines and sort it """
    rexp_begin = re.compile("\s*struct\s+snd_pci_quirk\s+.+\s*=\s*{\s*$")

    flag = False
    xbuffer = []
    quirks = []
    for line in lines:
        if flag == False and rexp_begin.search(line):
            xbuffer.append(line)
            quirks.extend(xbuffer)
            flag = True
            xbuffer = []
            continue
        if flag == True and line.strip() in ('{}', '{} /* terminator */'):
            quirks.extend(sort_quirks(parse_quirks(xbuffer)))
            flag = False
            xbuffer = []
        xbuffer.append(line)
    quirks.extend(xbuffer)
    return quirks

def main():
    """ MAIN """
    try:
        fname = sys.argv[1]
    except IndexError:
        print "Usage: %s <input file>" % os.path.basename(sys.argv[0])
        sys.exit(1)
    lines = read_file(fname)
    quirks = catch_quirk_lines(lines)
    save_file('%s.sorted' % fname, quirks)

if __name__ == '__main__':
    main()