//
// 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 <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>


#include "procs.h"

const char *loop_kernel_code =
"__kernel void test_loop(__global int *src, __global int *loopindx, __global int *loopcnt, __global int *dst)\n"
"{\n"
"    int  tid = get_global_id(0);\n"
"    int  n = get_global_size(0);\n"
"    int  i, j;\n"
"\n"
"    dst[tid] = 0;\n"
"    for (i=0,j=loopindx[tid]; i<loopcnt[tid]; i++,j++)\n"
"    {\n"
"        if (j >= n)\n"
"            j = 0;\n"
"        dst[tid] += src[j];\n"
"    }\n"
"\n"
"}\n";


int
verify_loop(int *inptr, int *loopindx, int *loopcnt, int *outptr, int n)
{
    int     r, i, j, k;

    for (i=0; i<n; i++)
    {
        r = 0;
        for (j=0,k=loopindx[i]; j<loopcnt[i]; j++,k++)
        {
            if (k >= n)
                k = 0;
            r += inptr[k];
        }

        if (r != outptr[i])
        {
            log_error("LOOP test failed: %d found, expected %d\n", outptr[i], r);
            return -1;
        }
    }

    log_info("LOOP test passed\n");
    return 0;
}

int test_loop(cl_device_id device, cl_context context, cl_command_queue queue, int num_elements)
{
    cl_mem streams[4];
    cl_int *input_ptr, *loop_indx, *loop_cnt, *output_ptr;
    cl_program program;
    cl_kernel kernel;
    size_t threads[1];
    int err, i;

    size_t length = sizeof(cl_int) * num_elements;
    input_ptr  = (cl_int*)malloc(length);
    loop_indx  = (cl_int*)malloc(length);
    loop_cnt   = (cl_int*)malloc(length);
    output_ptr = (cl_int*)malloc(length);

    streams[0] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL);
    if (!streams[0])
    {
        log_error("clCreateBuffer failed\n");
        return -1;
    }
    streams[1] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL);
    if (!streams[1])
    {
        log_error("clCreateBuffer failed\n");
        return -1;
    }
    streams[2] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL);
    if (!streams[2])
    {
        log_error("clCreateBuffer failed\n");
        return -1;
    }
    streams[3] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL);
    if (!streams[3])
    {
        log_error("clCreateBuffer failed\n");
        return -1;
    }

    MTdata d = init_genrand( gRandomSeed );
    for (i=0; i<num_elements; i++)
    {
        input_ptr[i] = (int)genrand_int32(d);
        loop_indx[i] = (int)get_random_float(0, num_elements-1, d);
        loop_cnt[i] = (int)get_random_float(0, num_elements/32, d);
    }
    free_mtdata(d); d = NULL;

  err = clEnqueueWriteBuffer(queue, streams[0], CL_TRUE, 0, length, input_ptr, 0, NULL, NULL);
  if (err != CL_SUCCESS)
  {
    log_error("clEnqueueWriteBuffer failed\n");
    return -1;
  }
  err = clEnqueueWriteBuffer(queue, streams[1], CL_TRUE, 0, length, loop_indx, 0, NULL, NULL);
  if (err != CL_SUCCESS)
  {
    log_error("clEnqueueWriteBuffer failed\n");
    return -1;
  }
  err = clEnqueueWriteBuffer(queue, streams[2], CL_TRUE, 0, length, loop_cnt, 0, NULL, NULL);
  if (err != CL_SUCCESS)
  {
    log_error("clEnqueueWriteBuffer failed\n");
    return -1;
  }

  err = create_single_kernel_helper(context, &program, &kernel, 1, &loop_kernel_code, "test_loop" );
  if (err)
    return -1;

  err  = clSetKernelArg(kernel, 0, sizeof streams[0], &streams[0]);
  err |= clSetKernelArg(kernel, 1, sizeof streams[1], &streams[1]);
  err |= clSetKernelArg(kernel, 2, sizeof streams[2], &streams[2]);
  err |= clSetKernelArg(kernel, 3, sizeof streams[3], &streams[3]);
    if (err != CL_SUCCESS)
    {
        log_error("clSetKernelArgs failed\n");
        return -1;
    }

    threads[0] = (unsigned int)num_elements;
  err = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, threads, NULL, 0, NULL, NULL);
  if (err != CL_SUCCESS)
  {
    log_error("clEnqueueNDRangeKernel failed\n");
    return -1;
  }

  err = clEnqueueReadBuffer(queue, streams[3], CL_TRUE, 0, length, output_ptr, 0, NULL, NULL);
  if (err != CL_SUCCESS)
  {
    log_error("clReadArray failed\n");
    return -1;
  }

  err = verify_loop(input_ptr, loop_indx, loop_cnt, output_ptr, num_elements);

    // cleanup
    clReleaseMemObject(streams[0]);
    clReleaseMemObject(streams[1]);
    clReleaseMemObject(streams[2]);
    clReleaseMemObject(streams[3]);
    clReleaseKernel(kernel);
    clReleaseProgram(program);
    free(input_ptr);
    free(loop_indx);
    free(loop_cnt);
    free(output_ptr);

    return err;
}