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

#-------------------------------------------------------------------------
# drawElements Quality Program utilities
# --------------------------------------
#
# Copyright 2017 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 os
import posixpath
from fnmatch import fnmatch

from ctsbuild.common import DEQP_DIR, writeFile, which, execute

SRC_ROOTS = [
    "execserver",
    "executor",
    "external/vulkancts",
    "framework/common",
    "framework/delibs",
    "framework/egl",
    "framework/opengl",
    "framework/platform/android",
    "framework/qphelper",
    "framework/randomshaders",
    "framework/referencerenderer",
    "framework/xexml",
    "modules",
]

INCLUDE_PATTERNS = [
    "*.cpp",
    "*.c",
]

EXCLUDE_PATTERNS = [
    "execserver/xsWin32TestProcess.cpp",
    "external/vulkancts/modules/vulkan/vktBuildPrograms.cpp",
    "framework/delibs/dethread/standalone_test.c",
    "framework/randomshaders/rsgTest.cpp",
    "executor/tools/*",
    "execserver/tools/*",
    "external/vulkancts/framework/vulkan/vkRenderDocUtil.cpp",
    "external/vulkancts/modules/vulkan/vktTestPackageEntrySC.cpp",
    "external/vulkancts/modules/vulkan/sc/*",
    "external/vulkancts/vkscserver/*",
    "external/vulkancts/vkscpc/*",
    "external/vulkancts/framework/vulkan/generated/vulkansc/*",
    "external/vulkancts/modules/vulkan/video/*",
]

# These are include folders where there are no source c/cpp files
EXTRA_INCLUDE_DIRS = [
    # This only has headers, so is not caught with INCLUDE_PATTERNS
    "external/vulkancts/framework/vulkan/generated/vulkan",
]

TEMPLATE = """
// WARNING: This is auto-generated file. Do not modify, since changes will
// be lost! Modify scripts/gen_android_bp.py instead.

cc_defaults {
    name: "libdeqp_gen",

    srcs: [
{SRC_FILES}    ],
    local_include_dirs: [
{INCLUDES}    ],
}

"""[1:-1]

def matchesAny (filename, patterns):
    for ptrn in patterns:
        if fnmatch(filename, ptrn):
            return True
    return False

def isSourceFile (filename):
    return matchesAny(filename, INCLUDE_PATTERNS) and not matchesAny(filename, EXCLUDE_PATTERNS)

def toPortablePath (nativePath):
    # os.path is so convenient...
    head, tail = os.path.split(nativePath)
    components = [tail]

    while head != None and head != '':
        head, tail = os.path.split(head)
        components.append(tail)

    components.reverse()

    portablePath = ""
    for component in components:
        portablePath = posixpath.join(portablePath, component)

    return portablePath

def getSourceFiles ():
    sources = []

    for srcRoot in SRC_ROOTS:
        baseDir = os.path.join(DEQP_DIR, srcRoot)
        for root, dirs, files in os.walk(baseDir):
            for file in files:
                absPath = os.path.join(root, file)
                nativeRelPath = os.path.relpath(absPath, DEQP_DIR)
                portablePath = toPortablePath(nativeRelPath)

                if isSourceFile(portablePath):
                    sources.append(portablePath)

    sources.sort()

    return sources

def getSourceDirs (sourceFiles):
    seenDirs = set()
    sourceDirs = []

    for sourceFile in sourceFiles:
        sourceDir = posixpath.dirname(sourceFile)

        if not sourceDir in seenDirs:
            sourceDirs.append(sourceDir)
            seenDirs.add(sourceDir)

    sourceDirs.extend(EXTRA_INCLUDE_DIRS)
    sourceDirs.sort()

    return sourceDirs

def genBpStringList (items):
    src = ""

    for item in items:
        src += "        \"%s\",\n" % item

    return src

def genAndroidBp (sourceDirs, sourceFiles):
    src = TEMPLATE
    src = src.replace("{INCLUDES}", genBpStringList(sourceDirs))
    src = src.replace("{SRC_FILES}", genBpStringList(sourceFiles))

    return src

if __name__ == "__main__":
    sourceFiles = getSourceFiles()
    sourceDirs = getSourceDirs(sourceFiles)
    androidBpText = genAndroidBp(sourceDirs, sourceFiles)

    bpFilename = os.path.join(DEQP_DIR, "AndroidGen.bp")
    writeFile(bpFilename, androidBpText)

    # Format the generated file
    if which("bpfmt") != None:
        execute(["bpfmt", "-w", bpFilename])
    else:
        print("Warning: Could not find bpfmt, file won't be formatted.")