# -*- coding: utf-8 -*-

#-------------------------------------------------------------------------
# drawElements Quality Program utilities
# --------------------------------------
#
# Copyright 2015 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#-------------------------------------------------------------------------

import re
import sys
from fnmatch import fnmatch

def fail (msg):
    print("ERROR: " + msg)
    sys.exit(-1)

# filename -> [case name]
def readCaseList (filename):
    f = open(filename, 'rb')
    cases = []
    for line in f:
        if line[0:6] == "TEST: ":
            cases.append(line[6:].strip())
    f.close()
    return cases

# filename -> [(filter, min, recommended)]
def readResolutionList (filename):
    f = open(filename, 'rb')
    resList = []
    for line in f:
        line = line.strip()
        params = line.split('\t')
        if len(params) == 3:
            resList.append((params[0], params[1], params[2]))
        elif len(params) != 0:
            fail("Invalid line in resolution list: '%s'" % line)
    f.close()
    return resList

def getMatchingCases (cases, pattern):
    matching = []
    for case in cases:
        if fnmatch(case, pattern):
            matching.append(case)
    return matching

def isResolutionOk (res):
    return re.match('^[1-9][0-9]*x[1-9][0-9]*$', res) != None

if __name__ == "__main__":
    if len(sys.argv) != 3:
        print("%s: [caselist] [resolution list]" % sys.argv[0])
        sys.exit(-1)

    caseList = readCaseList(sys.argv[1])
    resList = readResolutionList(sys.argv[2])

    # Pass 1: quick check for resolution values
    for pattern, minRes, recRes in resList:
        if not isResolutionOk(minRes) or not isResolutionOk(recRes):
            fail("Invalid resolution: '%s %s %s'" % (pattern, minRes, recRes))

    # Pass 2: check that each case is specified by one rule
    markedCases = set()
    for pattern, minRes, recRes in resList:
        matchingCases = getMatchingCases(caseList, pattern)

        if len(matchingCases) == 0:
            fail("Pattern '%s' does not match any cases" % pattern)

        for case in matchingCases:
            if case in markedCases:
                fail("Case '%s' specified multiple times (when processing '%s')" % (case, pattern))
            markedCases.add(case)

    for case in caseList:
        if not case in markedCases:
            fail("Case '%s' not specified by any rule" % case)