//
// Copyright (c) 2017 The Khronos Group Inc.
//
// 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.
//
#include <stdio.h>
#include <stdlib.h>

#include "harness/testHarness.h"
#include "utils.h"
#include "procs.h"


test_definition test_list[] = { ADD_TEST(context_create),
                                ADD_TEST(get_device_ids),
                                ADD_TEST(api),
                                ADD_TEST(kernel),
                                ADD_TEST(other_data_types),
                                ADD_TEST(memory_access),
                                ADD_TEST(interop_user_sync) };

const int test_num = ARRAY_SIZE(test_list);

clGetDeviceIDsFromDX9MediaAdapterKHR_fn clGetDeviceIDsFromDX9MediaAdapterKHR =
    NULL;
clCreateFromDX9MediaSurfaceKHR_fn clCreateFromDX9MediaSurfaceKHR = NULL;
clEnqueueAcquireDX9MediaSurfacesKHR_fn clEnqueueAcquireDX9MediaSurfacesKHR =
    NULL;
clEnqueueReleaseDX9MediaSurfacesKHR_fn clEnqueueReleaseDX9MediaSurfacesKHR =
    NULL;

cl_platform_id gPlatformIDdetected;
cl_device_id gDeviceIDdetected;
cl_device_type gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;

bool MediaSurfaceSharingExtensionInit()
{
    clGetDeviceIDsFromDX9MediaAdapterKHR =
        (clGetDeviceIDsFromDX9MediaAdapterKHR_fn)
            clGetExtensionFunctionAddressForPlatform(
                gPlatformIDdetected, "clGetDeviceIDsFromDX9MediaAdapterKHR");
    if (clGetDeviceIDsFromDX9MediaAdapterKHR == NULL)
    {
        log_error("clGetExtensionFunctionAddressForPlatform("
                  "clGetDeviceIDsFromDX9MediaAdapterKHR) returned NULL.\n");
        return false;
    }

    clCreateFromDX9MediaSurfaceKHR = (clCreateFromDX9MediaSurfaceKHR_fn)
        clGetExtensionFunctionAddressForPlatform(
            gPlatformIDdetected, "clCreateFromDX9MediaSurfaceKHR");
    if (clCreateFromDX9MediaSurfaceKHR == NULL)
    {
        log_error("clGetExtensionFunctionAddressForPlatform("
                  "clCreateFromDX9MediaSurfaceKHR) returned NULL.\n");
        return false;
    }

    clEnqueueAcquireDX9MediaSurfacesKHR =
        (clEnqueueAcquireDX9MediaSurfacesKHR_fn)
            clGetExtensionFunctionAddressForPlatform(
                gPlatformIDdetected, "clEnqueueAcquireDX9MediaSurfacesKHR");
    if (clEnqueueAcquireDX9MediaSurfacesKHR == NULL)
    {
        log_error("clGetExtensionFunctionAddressForPlatform("
                  "clEnqueueAcquireDX9MediaSurfacesKHR) returned NULL.\n");
        return false;
    }

    clEnqueueReleaseDX9MediaSurfacesKHR =
        (clEnqueueReleaseDX9MediaSurfacesKHR_fn)
            clGetExtensionFunctionAddressForPlatform(
                gPlatformIDdetected, "clEnqueueReleaseDX9MediaSurfacesKHR");
    if (clEnqueueReleaseDX9MediaSurfacesKHR == NULL)
    {
        log_error("clGetExtensionFunctionAddressForPlatform("
                  "clEnqueueReleaseDX9MediaSurfacesKHR) returned NULL.\n");
        return false;
    }

    return true;
}

bool DetectPlatformAndDevice()
{
    std::vector<cl_platform_id> platforms;
    cl_uint platformsNum = 0;
    cl_int error = clGetPlatformIDs(0, 0, &platformsNum);
    if (error != CL_SUCCESS)
    {
        print_error(error, "clGetPlatformIDs failed\n");
        return false;
    }

    platforms.resize(platformsNum);
    error = clGetPlatformIDs(platformsNum, &platforms[0], 0);
    if (error != CL_SUCCESS)
    {
        print_error(error, "clGetPlatformIDs failed\n");
        return false;
    }

    bool found = false;
    for (size_t i = 0; i < platformsNum; ++i)
    {
        std::vector<cl_device_id> devices;
        cl_uint devicesNum = 0;
        error = clGetDeviceIDs(platforms[i], gDeviceTypeSelected, 0, 0,
                               &devicesNum);
        if (error != CL_SUCCESS)
        {
            print_error(error, "clGetDeviceIDs failed\n");
            return false;
        }

        devices.resize(devicesNum);
        error = clGetDeviceIDs(platforms[i], gDeviceTypeSelected, devicesNum,
                               &devices[0], 0);
        if (error != CL_SUCCESS)
        {
            print_error(error, "clGetDeviceIDs failed\n");
            return false;
        }

        for (size_t j = 0; j < devicesNum; ++j)
        {
            if (is_extension_available(devices[j], "cl_khr_dx9_media_sharing"))
            {
                gPlatformIDdetected = platforms[i];
                gDeviceIDdetected = devices[j];
                found = true;
                break;
            }
        }
    }

    if (!found)
    {
        log_info("Test was not run, because the media surface sharing "
                 "extension is not supported for any devices.\n");
        return false;
    }

    return true;
}

bool CmdlineParse(int argc, const char *argv[])
{
    char *env_mode = getenv("CL_DEVICE_TYPE");
    if (env_mode != NULL)
    {
        if (strcmp(env_mode, "gpu") == 0
            || strcmp(env_mode, "CL_DEVICE_TYPE_GPU") == 0)
            gDeviceTypeSelected = CL_DEVICE_TYPE_GPU;
        else if (strcmp(env_mode, "cpu") == 0
                 || strcmp(env_mode, "CL_DEVICE_TYPE_CPU") == 0)
            gDeviceTypeSelected = CL_DEVICE_TYPE_CPU;
        else if (strcmp(env_mode, "accelerator") == 0
                 || strcmp(env_mode, "CL_DEVICE_TYPE_ACCELERATOR") == 0)
            gDeviceTypeSelected = CL_DEVICE_TYPE_ACCELERATOR;
        else if (strcmp(env_mode, "default") == 0
                 || strcmp(env_mode, "CL_DEVICE_TYPE_DEFAULT") == 0)
            gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;
        else
        {
            log_error("Unknown CL_DEVICE_TYPE env variable setting: "
                      "%s.\nAborting...\n",
                      env_mode);
            return false;
        }
    }

    for (int i = 0; i < argc; ++i)
    {
        if (strcmp(argv[i], "gpu") == 0
            || strcmp(argv[i], "CL_DEVICE_TYPE_GPU") == 0)
        {
            gDeviceTypeSelected = CL_DEVICE_TYPE_GPU;
            continue;
        }
        else if (strcmp(argv[i], "cpu") == 0
                 || strcmp(argv[i], "CL_DEVICE_TYPE_CPU") == 0)
        {
            gDeviceTypeSelected = CL_DEVICE_TYPE_CPU;
            continue;
        }
        else if (strcmp(argv[i], "accelerator") == 0
                 || strcmp(argv[i], "CL_DEVICE_TYPE_ACCELERATOR") == 0)
        {
            gDeviceTypeSelected = CL_DEVICE_TYPE_ACCELERATOR;
            continue;
        }
        else if (strcmp(argv[i], "CL_DEVICE_TYPE_DEFAULT") == 0)
        {
            gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;
            continue;
        }
        else if (strcmp(argv[i], "sw") == 0 || strcmp(argv[i], "software") == 0)
        {
            CDeviceWrapper::AccelerationType(CDeviceWrapper::ACCELERATION_SW);
        }
    }

    return true;
}

int main(int argc, const char *argv[])
{
    if (!CmdlineParse(argc, argv)) return TEST_FAIL;

    if (!DetectPlatformAndDevice())
    {
        log_info("Test was not run, because the media surface sharing "
                 "extension is not supported\n");
        return TEST_SKIP;
    }

    if (!MediaSurfaceSharingExtensionInit()) return TEST_FAIL;

    return runTestHarness(argc, argv, test_num, test_list, true, 0);
}