//
// 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"
#include "harness/conversions.h"
#include "harness/typeWrappers.h"

struct work_item_data
{
    cl_uint workDim;
    cl_uint globalSize[ 3 ];
    cl_uint globalID[ 3 ];
    cl_uint localSize[ 3 ];
    cl_uint localID[ 3 ];
    cl_uint numGroups[ 3 ];
    cl_uint groupID[ 3 ];
};

static const char *workItemKernelCode =
"typedef struct {\n"
"    uint workDim;\n"
"    uint globalSize[ 3 ];\n"
"    uint globalID[ 3 ];\n"
"    uint localSize[ 3 ];\n"
"    uint localID[ 3 ];\n"
"    uint numGroups[ 3 ];\n"
"    uint groupID[ 3 ];\n"
" } work_item_data;\n"
"\n"
"__kernel void sample_kernel( __global work_item_data *outData )\n"
"{\n"
"    int id = get_global_id(0);\n"
"   outData[ id ].workDim = (uint)get_work_dim();\n"
"    for( uint i = 0; i < get_work_dim(); i++ )\n"
"   {\n"
"       outData[ id ].globalSize[ i ] = (uint)get_global_size( i );\n"
"       outData[ id ].globalID[ i ] = (uint)get_global_id( i );\n"
"       outData[ id ].localSize[ i ] = (uint)get_local_size( i );\n"
"       outData[ id ].localID[ i ] = (uint)get_local_id( i );\n"
"       outData[ id ].numGroups[ i ] = (uint)get_num_groups( i );\n"
"       outData[ id ].groupID[ i ] = (uint)get_group_id( i );\n"
"   }\n"
"}";

#define NUM_TESTS 1

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

    clProgramWrapper program;
    clKernelWrapper kernel;
    clMemWrapper outData;
    work_item_data    testData[ 10240 ];
    size_t threads[3], localThreads[3];
    MTdata d;


    error = create_single_kernel_helper( context, &program, &kernel, 1, &workItemKernelCode, "sample_kernel" );
    test_error( error, "Unable to create testing kernel" );

    outData = clCreateBuffer( context, CL_MEM_READ_WRITE, sizeof( testData ), NULL, &error );
    test_error( error, "Unable to create output buffer" );

    error = clSetKernelArg( kernel, 0, sizeof( outData ), &outData );
    test_error( error, "Unable to set kernel arg" );

    d = init_genrand( gRandomSeed );
    for( size_t dim = 1; dim <= 3; dim++ )
    {
        for( int i = 0; i < NUM_TESTS; i++  )
        {
            size_t numItems = 1;
            for( size_t j = 0; j < dim; j++ )
            {
                // All of our thread sizes should be within the max local sizes, since they're all <= 20
                threads[ j ] = (size_t)random_in_range( 1, 20, d );
                localThreads[ j ] = threads[ j ] / (size_t)random_in_range( 1, (int)threads[ j ], d );
                while( localThreads[ j ] > 1 && ( threads[ j ] % localThreads[ j ] != 0 ) )
                    localThreads[ j ]--;

                numItems *= threads[ j ];

                // Hack for now: localThreads > 1 are iffy
                localThreads[ j ] = 1;
            }
            error = clEnqueueNDRangeKernel( queue, kernel, (cl_uint)dim, NULL, threads, localThreads, 0, NULL, NULL );
            test_error( error, "Unable to run kernel" );

            error = clEnqueueReadBuffer( queue, outData, CL_TRUE, 0, sizeof( testData ), testData, 0, NULL, NULL );
            test_error( error, "Unable to read results" );

            // Validate
            for( size_t q = 0; q < threads[0]; q++ )
            {
                // We can't really validate the actual value of each one, but we can validate that they're within a sane range
                if( testData[ q ].workDim != (cl_uint)dim )
                {
                    log_error( "ERROR: get_work_dim() did not return proper value for %d dimensions (expected %d, got %d)\n", (int)dim, (int)dim, (int)testData[ q ].workDim );
                    free_mtdata(d);
                    return -1;
                }
                for( size_t j = 0; j < dim; j++ )
                {
                    if( testData[ q ].globalSize[ j ] != (cl_uint)threads[ j ] )
                    {
                        log_error( "ERROR: get_global_size(%d) did not return proper value for %d dimensions (expected %d, got %d)\n",
                                    (int)j, (int)dim, (int)threads[ j ], (int)testData[ q ].globalSize[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                    if (testData[q].globalID[j] >= (cl_uint)threads[j])
                    {
                        log_error( "ERROR: get_global_id(%d) did not return proper value for %d dimensions (max %d, got %d)\n",
                                  (int)j, (int)dim, (int)threads[ j ], (int)testData[ q ].globalID[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                    if( testData[ q ].localSize[ j ] != (cl_uint)localThreads[ j ] )
                    {
                        log_error( "ERROR: get_local_size(%d) did not return proper value for %d dimensions (expected %d, got %d)\n",
                                  (int)j, (int)dim, (int)localThreads[ j ], (int)testData[ q ].localSize[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                    if (testData[q].localID[j] >= (cl_uint)localThreads[j])
                    {
                        log_error( "ERROR: get_local_id(%d) did not return proper value for %d dimensions (max %d, got %d)\n",
                                  (int)j, (int)dim, (int)localThreads[ j ], (int)testData[ q ].localID[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                    size_t groupCount = ( threads[ j ] + localThreads[ j ] - 1 ) / localThreads[ j ];
                    if( testData[ q ].numGroups[ j ] != (cl_uint)groupCount )
                    {
                        log_error( "ERROR: get_num_groups(%d) did not return proper value for %d dimensions (expected %d with global dim %d and local dim %d, got %d)\n",
                                  (int)j, (int)dim, (int)groupCount, (int)threads[ j ], (int)localThreads[ j ], (int)testData[ q ].numGroups[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                    if (testData[q].groupID[j] >= (cl_uint)groupCount)
                    {
                        log_error( "ERROR: get_group_id(%d) did not return proper value for %d dimensions (max %d, got %d)\n",
                                  (int)j, (int)dim, (int)groupCount, (int)testData[ q ].groupID[ j ] );
                        free_mtdata(d);
                        return -1;
                    }
                }
            }
        }
    }

    free_mtdata(d);
    return 0;
}