//
// Copyright (c) 2017-2020 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 "testBase.h"

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

#include <atomic>
#include <string>

namespace {

const char *sample_async_kernel[] = {
    "__kernel void sample_test(__global float *src, __global int *dst)\n"
    "{\n"
    "    size_t tid = get_global_id(0);\n"
    "\n"
    "    dst[tid] = (int)src[tid];\n"
    "\n"
    "}\n"
};

const char *sample_async_kernel_error[] = {
    "__kernel void sample_test(__global float *src, __global int *dst)\n"
    "{\n"
    "    size_t tid = get_global_id(0);\n"
    "\n"
    "    dst[tid] = badcodehere;\n"
    "\n"
    "}\n"
};

// Data passed to a program completion callback
struct TestData
{
    cl_device_id device;
    cl_build_status expectedStatus;
};

std::atomic<int> callbackResult;

}

void CL_CALLBACK test_notify_build_complete(cl_program program, void *userData)
{
    TestData *data = reinterpret_cast<TestData *>(userData);

    // Check user data is valid
    if (data == nullptr)
    {
        log_error("ERROR: User data passed to callback was not valid!\n");
        callbackResult = -1;
        return;
    }

    // Get program build status
    cl_build_status status;
    cl_int err =
        clGetProgramBuildInfo(program, data->device, CL_PROGRAM_BUILD_STATUS,
                              sizeof(cl_build_status), &status, NULL);
    if (err != CL_SUCCESS)
    {
        log_info("ERROR: failed to get build status from callback\n");
        callbackResult = -1;
        return;
    }

    log_info("Program completion callback received build status %d\n", status);

    // Check program build status matches expectation
    if (status != data->expectedStatus)
    {
        log_info("ERROR: build status %d != expected status %d\n", status,
                 data->expectedStatus);
        callbackResult = -1;
    }
    else
    {
        callbackResult = 1;
    }
}

int test_async_build(cl_device_id deviceID, cl_context context,
                     cl_command_queue queue, int num_elements)
{
    cl_int error;

    struct TestDef
    {
        const char **source;
        cl_build_status expectedStatus;
    };

    TestDef testDefs[] = { { sample_async_kernel, CL_BUILD_SUCCESS },
                           { sample_async_kernel_error, CL_BUILD_ERROR } };
    for (TestDef &testDef : testDefs)
    {
        log_info("\nTesting program that should produce status %d\n",
                 testDef.expectedStatus);

        // Create the program
        clProgramWrapper program;
        error = create_single_kernel_helper_create_program(context, &program, 1,
                                                           testDef.source);
        test_error(error, "Unable to create program from source");

        // Start an asynchronous build, registering the completion callback
        TestData testData = { deviceID, testDef.expectedStatus };
        callbackResult = 0;
        error = clBuildProgram(program, 1, &deviceID, NULL,
                               test_notify_build_complete, (void *)&testData);
        // Allow implementations to return synchronous build failures.
        // They still need to call the callback.
        if (!(error == CL_BUILD_PROGRAM_FAILURE
              && testDef.expectedStatus == CL_BUILD_ERROR))
            test_error(error, "Unable to start build");

        // Wait for callback to fire
        int timeout = 20;
        while (callbackResult == 0)
        {
            if (timeout < 0)
            {
                log_error("Timeout while waiting for callback to fire.\n\n");
                return -1;
            }

            log_info(" -- still waiting for callback...\n");
            sleep(1);
            timeout--;
        }

        // Check the callback result
        if (callbackResult == 1)
        {
            log_error("Test passed.\n\n");
        }
        else
        {
            log_error("Async build callback indicated test failure.\n\n");
            return -1;
        }
    }

    return 0;
}