//
// 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 "testBase.h"
#include "action_classes.h"


extern const char *IGetStatusString(cl_int status);

#define PRINT_OPS 0

int test_waitlist(cl_device_id device, cl_context context,
                  cl_command_queue queue, Action *actionToTest, bool multiple)
{
    NDRangeKernelAction actions[2];
    clEventWrapper events[3];
    cl_int status[3];
    cl_int error;

    if (multiple)
        log_info("\tExecuting reference event 0, then reference event 1 with "
                 "reference event 0 in its waitlist, then test event 2 with "
                 "reference events 0 and 1 in its waitlist.\n");
    else
        log_info("\tExecuting reference event 0, then test event 2 with "
                 "reference event 0 in its waitlist.\n");

    // Set up the first base action to wait against
    error = actions[0].Setup(device, context, queue);
    test_error(error, "Unable to setup base event to wait against");

    if (multiple)
    {
        // Set up a second event to wait against
        error = actions[1].Setup(device, context, queue);
        test_error(error, "Unable to setup second base event to wait against");
    }

    // Now set up the actual action to test
    error = actionToTest->Setup(device, context, queue);
    test_error(error, "Unable to set up test event");

    // Execute all events now
    if (PRINT_OPS) log_info("\tExecuting action 0...\n");
    error = actions[0].Execute(queue, 0, NULL, &events[0]);
    test_error(error, "Unable to execute first event");

    if (multiple)
    {
        if (PRINT_OPS) log_info("\tExecuting action 1...\n");
        error = actions[1].Execute(queue, 1, &events[0], &events[1]);
        test_error(error, "Unable to execute second event");
    }

    // Sanity check
    if (multiple)
    {
        if (PRINT_OPS) log_info("\tChecking status of action 1...\n");
        error = clGetEventInfo(events[1], CL_EVENT_COMMAND_EXECUTION_STATUS,
                               sizeof(status[1]), &status[1], NULL);
        test_error(error, "Unable to get event status");
    }
    if (PRINT_OPS) log_info("\tChecking status of action 0...\n");
    error = clGetEventInfo(events[0], CL_EVENT_COMMAND_EXECUTION_STATUS,
                           sizeof(status[0]), &status[0], NULL);
    test_error(error, "Unable to get event status");

    log_info("\t\tEvent status after starting reference events: reference "
             "event 0: %s, reference event 1: %s, test event 2: %s.\n",
             IGetStatusString(status[0]),
             (multiple ? IGetStatusString(status[1]) : "N/A"), "N/A");

    if ((status[0] == CL_COMPLETE) || (multiple && status[1] == CL_COMPLETE))
    {
        log_info("WARNING: Reference event(s) already completed before we "
                 "could execute test event! Possible that the reference event "
                 "blocked (implicitly passing)\n");
        return 0;
    }

    if (PRINT_OPS) log_info("\tExecuting action to test...\n");
    error = actionToTest->Execute(queue, (multiple) ? 2 : 1, &events[0],
                                  &events[2]);
    test_error(error, "Unable to execute test event");

    // Hopefully, the first event is still running
    if (PRINT_OPS) log_info("\tChecking status of action to test 2...\n");
    error = clGetEventInfo(events[2], CL_EVENT_COMMAND_EXECUTION_STATUS,
                           sizeof(status[2]), &status[2], NULL);
    test_error(error, "Unable to get event status");
    if (multiple)
    {
        if (PRINT_OPS) log_info("\tChecking status of action 1...\n");
        error = clGetEventInfo(events[1], CL_EVENT_COMMAND_EXECUTION_STATUS,
                               sizeof(status[1]), &status[1], NULL);
        test_error(error, "Unable to get event status");
    }
    if (PRINT_OPS) log_info("\tChecking status of action 0...\n");
    error = clGetEventInfo(events[0], CL_EVENT_COMMAND_EXECUTION_STATUS,
                           sizeof(status[0]), &status[0], NULL);
    test_error(error, "Unable to get event status");

    log_info("\t\tEvent status after starting test event: reference event 0: "
             "%s, reference event 1: %s, test event 2: %s.\n",
             IGetStatusString(status[0]),
             (multiple ? IGetStatusString(status[1]) : "N/A"),
             IGetStatusString(status[2]));

    if (multiple)
    {
        if (status[0] == CL_COMPLETE && status[1] == CL_COMPLETE)
        {
            log_info("WARNING: Both events completed, so unable to test "
                     "further (implicitly passing).\n");
            clFinish(queue);
            return 0;
        }

        if (status[1] == CL_COMPLETE && status[0] != CL_COMPLETE)
        {
            log_error(
                "ERROR: Test failed because the second wait event is complete "
                "and the first is not.(status: 0: %s and 1: %s)\n",
                IGetStatusString(status[0]), IGetStatusString(status[1]));
            clFinish(queue);
            return -1;
        }
    }
    else
    {
        if (status[0] == CL_COMPLETE)
        {
            log_info("WARNING: Reference event completed, so unable to test "
                     "further (implicitly passing).\n");
            clFinish(queue);
            return 0;
        }
        if (status[0] != CL_RUNNING && status[0] != CL_QUEUED
            && status[0] != CL_SUBMITTED)
        {
            log_error(
                "ERROR: Test failed because first wait event is not currently "
                "running, queued, or submitted! (status: 0: %s)\n",
                IGetStatusString(status[0]));
            clFinish(queue);
            return -1;
        }
    }

    if (status[2] != CL_QUEUED && status[2] != CL_SUBMITTED)
    {
        log_error("ERROR: Test event is not waiting to run! (status: 2: %s)\n",
                  IGetStatusString(status[2]));
        clFinish(queue);
        return -1;
    }

    // Now wait for the first reference event
    if (PRINT_OPS) log_info("\tWaiting for action 1 to finish...\n");
    error = clWaitForEvents(1, &events[0]);
    test_error(error, "Unable to wait for reference event");

    // Grab statuses again
    if (PRINT_OPS) log_info("\tChecking status of action to test 2...\n");
    error = clGetEventInfo(events[2], CL_EVENT_COMMAND_EXECUTION_STATUS,
                           sizeof(status[2]), &status[2], NULL);
    test_error(error, "Unable to get event status");
    if (multiple)
    {
        if (PRINT_OPS) log_info("\tChecking status of action 1...\n");
        error = clGetEventInfo(events[1], CL_EVENT_COMMAND_EXECUTION_STATUS,
                               sizeof(status[1]), &status[1], NULL);
        test_error(error, "Unable to get event status");
    }
    if (PRINT_OPS) log_info("\tChecking status of action 0...\n");
    error = clGetEventInfo(events[0], CL_EVENT_COMMAND_EXECUTION_STATUS,
                           sizeof(status[0]), &status[0], NULL);
    test_error(error, "Unable to get event status");

    log_info("\t\tEvent status after waiting for reference event 0: reference "
             "event 0: %s, reference event 1: %s, test event 2: %s.\n",
             IGetStatusString(status[0]),
             (multiple ? IGetStatusString(status[1]) : "N/A"),
             IGetStatusString(status[2]));

    // Sanity
    if (status[0] != CL_COMPLETE)
    {
        log_error("ERROR: Waited for first event but it's not complete "
                  "(status: 0: %s)\n",
                  IGetStatusString(status[0]));
        clFinish(queue);
        return -1;
    }

    // If we're multiple, and the second event isn't complete, then our test
    // event should still be queued
    if (multiple && status[1] != CL_COMPLETE)
    {
        if (status[1] == CL_RUNNING && status[2] == CL_RUNNING)
        {
            log_error("ERROR: Test event and second event are both running.\n");
            clFinish(queue);
            return -1;
        }
        if (status[2] != CL_QUEUED && status[2] != CL_SUBMITTED)
        {
            log_error("ERROR: Test event did not wait for second event before "
                      "starting! (status of ref: 1: %s, of test: 2: %s)\n",
                      IGetStatusString(status[1]), IGetStatusString(status[2]));
            clFinish(queue);
            return -1;
        }

        // Now wait for second event to complete, too
        if (PRINT_OPS) log_info("\tWaiting for action 1 to finish...\n");
        error = clWaitForEvents(1, &events[1]);
        test_error(error, "Unable to wait for second reference event");

        // Grab statuses again
        if (PRINT_OPS) log_info("\tChecking status of action to test 2...\n");
        error = clGetEventInfo(events[2], CL_EVENT_COMMAND_EXECUTION_STATUS,
                               sizeof(status[2]), &status[2], NULL);
        test_error(error, "Unable to get event status");
        if (multiple)
        {
            if (PRINT_OPS) log_info("\tChecking status of action 1...\n");
            error = clGetEventInfo(events[1], CL_EVENT_COMMAND_EXECUTION_STATUS,
                                   sizeof(status[1]), &status[1], NULL);
            test_error(error, "Unable to get event status");
        }
        if (PRINT_OPS) log_info("\tChecking status of action 0...\n");
        error = clGetEventInfo(events[0], CL_EVENT_COMMAND_EXECUTION_STATUS,
                               sizeof(status[0]), &status[0], NULL);
        test_error(error, "Unable to get event status");

        log_info(
            "\t\tEvent status after waiting for reference event 1: reference "
            "event 0: %s, reference event 1: %s, test event 2: %s.\n",
            IGetStatusString(status[0]),
            (multiple ? IGetStatusString(status[1]) : "N/A"),
            IGetStatusString(status[2]));

        // Sanity
        if (status[1] != CL_COMPLETE)
        {
            log_error("ERROR: Waited for second reference event but it didn't "
                      "complete (status: 1: %s)\n",
                      IGetStatusString(status[1]));
            clFinish(queue);
            return -1;
        }
    }

    // At this point, the test event SHOULD be running, but if it completed, we
    // consider it a pass
    if (status[2] == CL_COMPLETE)
    {
        log_info("WARNING: Test event already completed. Assumed valid.\n");
        clFinish(queue);
        return 0;
    }
    if (status[2] != CL_RUNNING && status[2] != CL_SUBMITTED
        && status[2] != CL_QUEUED)
    {
        log_error("ERROR: Second event did not start running after reference "
                  "event(s) completed! (status: 2: %s)\n",
                  IGetStatusString(status[2]));
        clFinish(queue);
        return -1;
    }

    // Wait for the test event, then return
    if (PRINT_OPS) log_info("\tWaiting for action 2 to test to finish...\n");
    error = clWaitForEvents(1, &events[2]);
    test_error(error, "Unable to wait for test event");

    error |= clGetEventInfo(events[2], CL_EVENT_COMMAND_EXECUTION_STATUS,
                            sizeof(status[2]), &status[2], NULL);
    test_error(error, "Unable to get event status");

    log_info("\t\tEvent status after waiting for test event: reference event "
             "0: %s, reference event 1: %s, test event 2: %s.\n",
             IGetStatusString(status[0]),
             (multiple ? IGetStatusString(status[1]) : "N/A"),
             IGetStatusString(status[2]));

    // Sanity
    if (status[2] != CL_COMPLETE)
    {
        log_error("ERROR: Test event didn't complete (status: 2: %s)\n",
                  IGetStatusString(status[2]));
        clFinish(queue);
        return -1;
    }

    clFinish(queue);
    return 0;
}

#define TEST_ACTION(name)                                                      \
    {                                                                          \
        name##Action action;                                                   \
        log_info("-- Testing " #name " (waiting on 1 event)...\n");            \
        if ((error = test_waitlist(deviceID, context, queue, &action, false))  \
            != CL_SUCCESS)                                                     \
            retVal++;                                                          \
        clFinish(queue);                                                       \
    }                                                                          \
    if (error                                                                  \
        == CL_SUCCESS) /* Only run multiples test if single test passed */     \
    {                                                                          \
        name##Action action;                                                   \
        log_info("-- Testing " #name " (waiting on 2 events)...\n");           \
        if ((error = test_waitlist(deviceID, context, queue, &action, true))   \
            != CL_SUCCESS)                                                     \
            retVal++;                                                          \
        clFinish(queue);                                                       \
    }

int test_waitlists(cl_device_id deviceID, cl_context context,
                   cl_command_queue oldQueue, int num_elements)
{
    cl_int error;
    int retVal = 0;
    cl_command_queue_properties props = CL_QUEUE_OUT_OF_ORDER_EXEC_MODE_ENABLE;

    if (!checkDeviceForQueueSupport(deviceID, props))
    {
        log_info("WARNING: Device does not support out-of-order exec mode; "
                 "skipping test.\n");
        return 0;
    }

    clCommandQueueWrapper queue =
        clCreateCommandQueue(context, deviceID, props, &error);
    test_error(error, "Unable to create out-of-order queue");

    log_info("\n");

    TEST_ACTION(NDRangeKernel)

    TEST_ACTION(ReadBuffer)
    TEST_ACTION(WriteBuffer)
    TEST_ACTION(MapBuffer)
    TEST_ACTION(UnmapBuffer)

    if (checkForImageSupport(deviceID) == CL_IMAGE_FORMAT_NOT_SUPPORTED)
    {
        log_info("\nNote: device does not support images. Skipping remainder "
                 "of waitlist tests...\n");
    }
    else
    {
        TEST_ACTION(ReadImage2D)
        TEST_ACTION(WriteImage2D)
        TEST_ACTION(CopyImage2Dto2D)
        TEST_ACTION(Copy2DImageToBuffer)
        TEST_ACTION(CopyBufferTo2DImage)
        TEST_ACTION(MapImage)

        if (checkFor3DImageSupport(deviceID) == CL_IMAGE_FORMAT_NOT_SUPPORTED)
            log_info("Device does not support 3D images. Skipping remainder of "
                     "waitlist tests...\n");
        else
        {
            TEST_ACTION(ReadImage3D)
            TEST_ACTION(WriteImage3D)
            TEST_ACTION(CopyImage2Dto3D)
            TEST_ACTION(CopyImage3Dto2D)
            TEST_ACTION(CopyImage3Dto3D)
            TEST_ACTION(Copy3DImageToBuffer)
            TEST_ACTION(CopyBufferTo3DImage)
        }
    }

    return retVal;
}