//
// 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 "harness/compat.h"

#include <stdio.h>
#include <string.h>

#if !defined(__APPLE__)
#include <CL/cl.h>
#endif

#include "procs.h"
#include "gl/setup.h"
#include "harness/testHarness.h"
#include "harness/parseParameters.h"

#if !defined(_WIN32)
#include <unistd.h>
#endif

static cl_context sCurrentContext = NULL;


#define TEST_FN_REDIRECT(fn) ADD_TEST(redirect_##fn)
#define TEST_FN_REDIRECTOR(fn)                                                 \
    int test_redirect_##fn(cl_device_id device, cl_context context,            \
                           cl_command_queue queue, int numElements)            \
    {                                                                          \
        int error;                                                             \
        clCommandQueueWrapper realQueue = clCreateCommandQueueWithProperties(  \
            sCurrentContext, device, 0, &error);                               \
        test_error(error, "Unable to create command queue");                   \
        return test_##fn(device, sCurrentContext, realQueue, numElements);     \
    }

// buffers:
TEST_FN_REDIRECTOR(buffers)
TEST_FN_REDIRECTOR(buffers_getinfo)

// 1D images:
TEST_FN_REDIRECTOR(images_read_1D)
TEST_FN_REDIRECTOR(images_write_1D)
TEST_FN_REDIRECTOR(images_1D_getinfo)

// 1D image arrays:
TEST_FN_REDIRECTOR(images_read_1Darray)
TEST_FN_REDIRECTOR(images_write_1Darray)
TEST_FN_REDIRECTOR(images_1Darray_getinfo)

// 2D images:
TEST_FN_REDIRECTOR(images_read_2D)
TEST_FN_REDIRECTOR(images_read_cube)
TEST_FN_REDIRECTOR(images_write)
TEST_FN_REDIRECTOR(images_write_cube)
TEST_FN_REDIRECTOR(images_2D_getinfo)
TEST_FN_REDIRECTOR(images_cube_getinfo)

// 2D image arrays:
TEST_FN_REDIRECTOR(images_read_2Darray)
TEST_FN_REDIRECTOR(images_write_2Darray)
TEST_FN_REDIRECTOR(images_2Darray_getinfo)

// 3D images:
TEST_FN_REDIRECTOR(images_read_3D)
TEST_FN_REDIRECTOR(images_write_3D)
TEST_FN_REDIRECTOR(images_3D_getinfo)

#ifdef GL_VERSION_3_2

TEST_FN_REDIRECTOR(images_read_texturebuffer)
TEST_FN_REDIRECTOR(images_write_texturebuffer)
TEST_FN_REDIRECTOR(images_texturebuffer_getinfo)

// depth textures
TEST_FN_REDIRECTOR(images_read_2D_depth)
TEST_FN_REDIRECTOR(images_write_2D_depth)
TEST_FN_REDIRECTOR(images_read_2Darray_depth)
TEST_FN_REDIRECTOR(images_write_2Darray_depth)

TEST_FN_REDIRECTOR(images_read_2D_multisample)
TEST_FN_REDIRECTOR(images_read_2Darray_multisample)
TEST_FN_REDIRECTOR(image_methods_depth)
TEST_FN_REDIRECTOR(image_methods_multisample)
#endif

// Renderbuffer-backed images:
TEST_FN_REDIRECTOR(renderbuffer_read)
TEST_FN_REDIRECTOR(renderbuffer_write)
TEST_FN_REDIRECTOR(renderbuffer_getinfo)

TEST_FN_REDIRECTOR(fence_sync)

test_definition test_list[] = { TEST_FN_REDIRECT(buffers),
                                TEST_FN_REDIRECT(buffers_getinfo),

                                TEST_FN_REDIRECT(images_read_1D),
                                TEST_FN_REDIRECT(images_write_1D),
                                TEST_FN_REDIRECT(images_1D_getinfo),

                                TEST_FN_REDIRECT(images_read_1Darray),
                                TEST_FN_REDIRECT(images_write_1Darray),
                                TEST_FN_REDIRECT(images_1Darray_getinfo),

                                TEST_FN_REDIRECT(images_read_2D),
                                TEST_FN_REDIRECT(images_write),
                                TEST_FN_REDIRECT(images_2D_getinfo),

                                TEST_FN_REDIRECT(images_read_cube),
                                TEST_FN_REDIRECT(images_write_cube),
                                TEST_FN_REDIRECT(images_cube_getinfo),

                                TEST_FN_REDIRECT(images_read_2Darray),
                                TEST_FN_REDIRECT(images_write_2Darray),
                                TEST_FN_REDIRECT(images_2Darray_getinfo),

                                TEST_FN_REDIRECT(images_read_3D),
                                TEST_FN_REDIRECT(images_write_3D),
                                TEST_FN_REDIRECT(images_3D_getinfo),

                                TEST_FN_REDIRECT(renderbuffer_read),
                                TEST_FN_REDIRECT(renderbuffer_write),
                                TEST_FN_REDIRECT(renderbuffer_getinfo) };

test_definition test_list32[] = {
    TEST_FN_REDIRECT(images_read_texturebuffer),
    TEST_FN_REDIRECT(images_write_texturebuffer),
    TEST_FN_REDIRECT(images_texturebuffer_getinfo),

    TEST_FN_REDIRECT(fence_sync),
    TEST_FN_REDIRECT(images_read_2D_depth),
    TEST_FN_REDIRECT(images_write_2D_depth),
    TEST_FN_REDIRECT(images_read_2Darray_depth),
    TEST_FN_REDIRECT(images_write_2Darray_depth),
    TEST_FN_REDIRECT(images_read_2D_multisample),
    TEST_FN_REDIRECT(images_read_2Darray_multisample),
    TEST_FN_REDIRECT(image_methods_depth),
    TEST_FN_REDIRECT(image_methods_multisample)
};

const int test_num = ARRAY_SIZE(test_list);
const int test_num32 = ARRAY_SIZE(test_list32);

int main(int argc, const char *argv[])
{
    gTestRounding = true;
    int error = 0;
    int numErrors = 0;

    test_start();
    argc = parseCustomParam(argc, argv);
    if (argc == -1)
    {
        return -1;
    }

    cl_device_type requestedDeviceType = CL_DEVICE_TYPE_DEFAULT;

    /* Do we have a CPU/GPU specification? */
    if (argc > 1)
    {
        if (strcmp(argv[argc - 1], "gpu") == 0
            || strcmp(argv[argc - 1], "CL_DEVICE_TYPE_GPU") == 0)
        {
            requestedDeviceType = CL_DEVICE_TYPE_GPU;
            argc--;
        }
        else if (strcmp(argv[argc - 1], "cpu") == 0
                 || strcmp(argv[argc - 1], "CL_DEVICE_TYPE_CPU") == 0)
        {
            requestedDeviceType = CL_DEVICE_TYPE_CPU;
            argc--;
        }
        else if (strcmp(argv[argc - 1], "accelerator") == 0
                 || strcmp(argv[argc - 1], "CL_DEVICE_TYPE_ACCELERATOR") == 0)
        {
            requestedDeviceType = CL_DEVICE_TYPE_ACCELERATOR;
            argc--;
        }
        else if (strcmp(argv[argc - 1], "CL_DEVICE_TYPE_DEFAULT") == 0)
        {
            requestedDeviceType = CL_DEVICE_TYPE_DEFAULT;
            argc--;
        }
    }

    if (argc > 1 && strcmp(argv[1], "-list") == 0)
    {
        log_info("Available 2.x tests:\n");
        for (int i = 0; i < test_num; i++)
            log_info("\t%s\n", test_list[i].name);

        log_info("Available 3.2 tests:\n");
        for (int i = 0; i < test_num32; i++)
            log_info("\t%s\n", test_list32[i].name);

        log_info("Note: Any 3.2 test names must follow 2.1 test names on the "
                 "command line.\n");
        log_info("Use environment variables to specify desired device.\n");

        return 0;
    }

    // Check to see if any 2.x or 3.2 test names were specified on the command
    // line.
    unsigned first_32_testname = 0;

    for (int j = 1; (j < argc) && (!first_32_testname); ++j)
        for (int i = 0; i < test_num32; ++i)
            if (strcmp(test_list32[i].name, argv[j]) == 0)
            {
                first_32_testname = j;
                break;
            }

    // Create the environment for the test.
    GLEnvironment *glEnv = GLEnvironment::Instance();

    // Check if any devices of the requested type support CL/GL interop.
    int supported = glEnv->SupportsCLGLInterop(requestedDeviceType);
    if (supported == 0)
    {
        log_info("Test not run because GL-CL interop is not supported for any "
                 "devices of the requested type.\n");
        return 0;
    }
    else if (supported == -1)
    {
        log_error("Unable to setup the test or failed to determine if CL-GL "
                  "interop is supported.\n");
        return -1;
    }

    // Initialize function pointers.
    error = init_clgl_ext();
    if (error < 0)
    {
        return error;
    }

    // OpenGL tests for non-3.2
    // ////////////////////////////////////////////////////////
    if ((argc == 1) || (first_32_testname != 1))
    {

        // At least one device supports CL-GL interop, so init the test.
        if (glEnv->Init(&argc, (char **)argv, CL_FALSE))
        {
            log_error(
                "Failed to initialize the GL environment for this test.\n");
            return -1;
        }

        // Create a context to use and then grab a device (or devices) from it
        sCurrentContext = glEnv->CreateCLContext();
        if (sCurrentContext == NULL)
        {
            log_error("ERROR: Unable to obtain CL context from GL\n");
            return -1;
        }

        size_t numDevices = 0;
        cl_device_id *deviceIDs;

        error = clGetContextInfo(sCurrentContext, CL_CONTEXT_DEVICES, 0, NULL,
                                 &numDevices);
        if (error != CL_SUCCESS)
        {
            print_error(error, "Unable to get device count from context");
            return -1;
        }
        deviceIDs = (cl_device_id *)malloc(numDevices);
        if (deviceIDs == NULL)
        {
            print_error(error, "malloc failed");
            return -1;
        }
        error = clGetContextInfo(sCurrentContext, CL_CONTEXT_DEVICES,
                                 numDevices, deviceIDs, NULL);
        if (error != CL_SUCCESS)
        {
            print_error(error, "Unable to get device list from context");
            return -1;
        }

        numDevices /= sizeof(cl_device_id);

        if (numDevices < 1)
        {
            log_error("No devices found.\n");
            return -1;
        }

        // Execute tests.
        int argc_ = (first_32_testname) ? first_32_testname : argc;

        for (size_t i = 0; i < numDevices; i++)
        {
            log_info("\nTesting OpenGL 2.x\n");
            if (printDeviceHeader(deviceIDs[i]) != CL_SUCCESS)
            {
                return -1;
            }

            // Note: don't use the entire harness, because we have a different
            // way of obtaining the device (via the context)
            test_harness_config config{};
            config.forceNoContextCreation = true;
            config.numElementsToUse = 1024;
            config.queueProps = 0;
            error = parseAndCallCommandLineTests(argc_, argv, deviceIDs[i],
                                                 test_num, test_list, config);
            if (error != 0) break;
        }

        numErrors += error;

        // Clean-up.
        free(deviceIDs);
        clReleaseContext(sCurrentContext);
        // delete glEnv;
    }

    // OpenGL 3.2 tests.
    // ////////////////////////////////////////////////////////
    if ((argc == 1) || first_32_testname)
    {

        // At least one device supports CL-GL interop, so init the test.
        if (glEnv->Init(&argc, (char **)argv, CL_TRUE))
        {
            log_error(
                "Failed to initialize the GL environment for this test.\n");
            return -1;
        }

        // Create a context to use and then grab a device (or devices) from it
        sCurrentContext = glEnv->CreateCLContext();
        if (sCurrentContext == NULL)
        {
            log_error("ERROR: Unable to obtain CL context from GL\n");
            return -1;
        }

        size_t numDevices = 0;
        cl_device_id *deviceIDs;

        error = clGetContextInfo(sCurrentContext, CL_CONTEXT_DEVICES, 0, NULL,
                                 &numDevices);
        if (error != CL_SUCCESS)
        {
            print_error(error, "Unable to get device count from context");
            return -1;
        }
        deviceIDs = (cl_device_id *)malloc(numDevices);
        if (deviceIDs == NULL)
        {
            print_error(error, "malloc failed");
            return -1;
        }
        error = clGetContextInfo(sCurrentContext, CL_CONTEXT_DEVICES,
                                 numDevices, deviceIDs, NULL);
        if (error != CL_SUCCESS)
        {
            print_error(error, "Unable to get device list from context");
            return -1;
        }

        numDevices /= sizeof(cl_device_id);

        if (numDevices < 1)
        {
            log_error("No devices found.\n");
            return -1;
        }

        int argc_ = (first_32_testname) ? 1 + (argc - first_32_testname) : argc;
        const char **argv_ =
            (first_32_testname) ? &argv[first_32_testname - 1] : argv;

        // Execute the tests.
        for (size_t i = 0; i < numDevices; i++)
        {
            log_info("\nTesting OpenGL 3.2\n");
            if (printDeviceHeader(deviceIDs[i]) != CL_SUCCESS)
            {
                return -1;
            }

            // Note: don't use the entire harness, because we have a different
            // way of obtaining the device (via the context)
            test_harness_config config{};
            config.forceNoContextCreation = true;
            config.numElementsToUse = 1024;
            config.queueProps = 0;
            error = parseAndCallCommandLineTests(
                argc_, argv_, deviceIDs[i], test_num32, test_list32, config);
            if (error != 0) break;
        }

        numErrors += error;

        // Clean-up.
        free(deviceIDs);
        clReleaseContext(sCurrentContext);
        delete glEnv;
    }

    // All done.
    return numErrors;
}