//
// 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.
//
#ifndef test_conformance_checkers_h
#define test_conformance_checkers_h

#include "harness/compat.h"

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

#include "procs.h"
#include "C_host_memory_block.h"

#define TEST_VALUE 5
typedef cl_char TEST_ELEMENT_TYPE;

enum {SUCCESS, FAILURE=-1000};

extern const char *buffer_write_kernel_code[];

enum BUFFER_TYPE {_BUFFER, _Sub_BUFFER};

template < class T > class cBuffer_checker
{
public:
  cBuffer_checker(cl_device_id deviceID, cl_context context,
                  cl_command_queue queue);
  ~cBuffer_checker();

  cl_device_id m_deviceID;
  cl_context m_context;
  cl_command_queue m_queue;

  clMemWrapper m_buffer, m_buffer_parent;
  enum BUFFER_TYPE m_buffer_type;

  cl_buffer_region m_sub_buffer_region;

  cl_int err;
  cl_bool m_blocking;
  cl_mem_flags buffer_mem_flag;

  C_host_memory_block<T> host_m_0, host_m_1, host_m_2;
  int m_nNumber_elements;

  void *pData, *pData2;

  void * pHost_ptr; // the host ptr at creation

  size_t buffer_origin[3];
  size_t host_origin[3];
  size_t region[3];
  size_t buffer_row_pitch;
  size_t buffer_slice_pitch;
  size_t host_row_pitch;
  size_t host_slice_pitch;

  size_t buffer_origin_bytes[3];
  size_t host_origin_bytes[3];
  size_t region_bytes[3];
  size_t buffer_row_pitch_bytes;
  size_t buffer_slice_pitch_bytes;
  size_t host_row_pitch_bytes;
  size_t host_slice_pitch_bytes;

  cl_int CreateBuffer(cl_mem_flags buffer_mem_flag, void * pdata);
  int get_block_size_bytes() { return (int)(m_nNumber_elements * sizeof(T)); };
  virtual cl_int SetupBuffer() = 0;

  virtual cl_int Setup_Test_Environment();

  virtual cl_int SetupASSubBuffer(cl_mem_flags parent_buffer_flag);

  virtual cl_int verify(cl_int err, cl_event & event);

  virtual cl_int Check_GetMemObjectInfo(cl_mem_flags buffer_mem_flag);

  void Init_rect(int bufforg[3], int host_org[3], int region[3],
                 int buffer_pitch[2], int host_pitch[2]);

  void Init_rect();

  virtual cl_int verify_RW_Buffer() = 0;
  virtual cl_int verify_RW_Buffer_rect() = 0;
  virtual cl_int verify_RW_Buffer_mapping() = 0;
};

template < class T >
cBuffer_checker<T>::cBuffer_checker(cl_device_id deviceID, cl_context context,
                                    cl_command_queue queue)
{
  m_nNumber_elements = 0;

  m_deviceID = deviceID;
  m_context = context;
  m_queue = queue;

  m_blocking = false;

  buffer_mem_flag = CL_MEM_READ_WRITE;
  pData = pData2 = NULL;

  buffer_origin[0] = buffer_origin[1] = buffer_origin[2] = 0;
  host_origin[0] = host_origin[1] = host_origin[2] = 0;
  region[0] = region[1] = region[2] = 0;
  buffer_row_pitch = buffer_slice_pitch = host_row_pitch = host_slice_pitch = 0;

  buffer_origin_bytes[0] = buffer_origin_bytes[1] = buffer_origin_bytes[2] = 0;
  host_origin_bytes[0] = host_origin_bytes[1] = host_origin_bytes[2] = 0;
  region_bytes[0] = region_bytes[1] = region_bytes[2] = 0;
  buffer_row_pitch_bytes = buffer_slice_pitch_bytes = 0;
  host_row_pitch_bytes = host_slice_pitch_bytes = 0;

  pHost_ptr = NULL;
}

template < class T >
cBuffer_checker<T>::~cBuffer_checker()
{
}


template < class T >
cl_int cBuffer_checker<T>::SetupBuffer()
{
  m_buffer_type = _BUFFER;
  return CL_SUCCESS;
}

template < class T >
cl_int cBuffer_checker<T>::Setup_Test_Environment()
{
  return CL_SUCCESS;
}

template < class T >
cl_int cBuffer_checker<T>::SetupASSubBuffer(cl_mem_flags parent_buffer_flag)
{
  m_buffer_type = _Sub_BUFFER;

  int supersize = 8000;
  this-> m_nNumber_elements = 1000;
  T vv1= TEST_VALUE;

  int block_size_in_byte = (int)(supersize * sizeof(T));

  this->host_m_0.Init(supersize);

  m_buffer_parent = clCreateBuffer(this->m_context, parent_buffer_flag,
                                   block_size_in_byte, this->host_m_0.pData, &err);
  test_error(err, "clCreateBuffer error");

  int size = this->m_nNumber_elements; // the size of subbuffer in elements

  cl_uint base_addr_align_bits;
  err = clGetDeviceInfo(m_deviceID, CL_DEVICE_MEM_BASE_ADDR_ALIGN, sizeof base_addr_align_bits, &base_addr_align_bits, NULL);
  test_error(err,"clGetDeviceInfo for CL_DEVICE_MEM_BASE_ADDR_ALIGN");

  int base_addr_align_bytes = base_addr_align_bits/8;

  int buffer_origin[3] = {base_addr_align_bytes, 0, 0};
  int host_origin[3] = {0, 0, 0};
  int region[3] = {size, 1, 1};
  int buffer_pitch[2] = {0, 0};
  int host_pitch[2] = {0, 0};
  this->Init_rect(buffer_origin, host_origin, region, buffer_pitch, host_pitch);

  this->m_nNumber_elements = size; // the size of subbuffer in elements
  this->host_m_1.Init(this->m_nNumber_elements, vv1);

  this->m_sub_buffer_region.origin = this->buffer_origin_bytes[0]; // in bytes
  this->m_sub_buffer_region.size = this->region_bytes[0];

  cl_int err = CL_SUCCESS;
  err = clEnqueueReadBufferRect(
      this->m_queue, m_buffer_parent, CL_TRUE, this->buffer_origin_bytes,
      this->host_origin_bytes, this->region_bytes, this->buffer_row_pitch_bytes,
      this->buffer_slice_pitch_bytes, this->host_row_pitch_bytes,
      this->host_slice_pitch_bytes, this->host_m_1.pData, 0, NULL,
      NULL); // update the mem_1

  if (err == CL_SUCCESS && (parent_buffer_flag & (CL_MEM_HOST_WRITE_ONLY | CL_MEM_HOST_NO_ACCESS))) {
    log_error("Calling clEnqueueReadBufferRect on a memory object created with the CL_MEM_HOST_WRITE_ONLY flag or the CL_MEM_HOST_NO_ACCESS flag should not return CL_SUCCESS\n");
    err = FAILURE;
    return err;
  } else {
    err = CL_SUCCESS;
  }

  cl_mem_flags f;
  if (parent_buffer_flag & CL_MEM_HOST_READ_ONLY)
    f = CL_MEM_HOST_READ_ONLY;
  else if (parent_buffer_flag & CL_MEM_HOST_WRITE_ONLY)
    f = CL_MEM_HOST_WRITE_ONLY;
  else if (parent_buffer_flag & CL_MEM_HOST_NO_ACCESS)
    f = CL_MEM_HOST_NO_ACCESS;

  m_buffer = clCreateSubBuffer(m_buffer_parent, f, CL_BUFFER_CREATE_TYPE_REGION,
                               &(this->m_sub_buffer_region), &err);
  test_error(err, "clCreateSubBuffer error");

  if (parent_buffer_flag | CL_MEM_USE_HOST_PTR)
  {
    this->pHost_ptr = (this->host_m_0.pData  + this->m_sub_buffer_region.origin/sizeof(T));
  }

  T vv2 = 0;
  this->host_m_2.Init(this->m_nNumber_elements, vv2);

  return err;
}

template < class T >
cl_int cBuffer_checker<T>::verify(cl_int err, cl_event & event)
{
  return CL_SUCCESS;
}

template < class T >
cl_int cBuffer_checker<T>::CreateBuffer(cl_mem_flags buffer_mem_flag, void *pdata)
{
  cl_int err = CL_SUCCESS;
  int block_size_in_byte= m_nNumber_elements* sizeof(T);

  m_buffer = clCreateBuffer(m_context, buffer_mem_flag, block_size_in_byte, pdata, &err);

  return err;
};

template < class T >
cl_int cBuffer_checker<T>::Check_GetMemObjectInfo(cl_mem_flags buffer_mem_flag)
{
  cl_int err = CL_SUCCESS;
  cl_mem_flags buffer_mem_flag_Check;
  err = clGetMemObjectInfo(this->m_buffer, CL_MEM_FLAGS, sizeof(cl_mem_flags),
                           &buffer_mem_flag_Check, NULL);

  if (buffer_mem_flag_Check != buffer_mem_flag) {
    log_error("clGetMemObjectInfo result differs from the specified result\n");
    return err;
  }

  cl_uint count = 0;
  err = clGetMemObjectInfo(this->m_buffer, CL_MEM_REFERENCE_COUNT,
                           sizeof(cl_uint), &count, NULL);

  if (count > 1)
    log_info("========= buffer count %d\n", count);

  test_error(err, "clGetMemObjectInfo failed");

  return err;
}

template < class T >
void cBuffer_checker<T>::Init_rect ()
{
  int buffer_origin[3] = {10, 0, 0};
  int host_origin[3] = {10, 0, 0};
  int region[3] = {8, 1, 1};
  int buffer_pitch[2] = {0, 0};
  int host_pitch[2] = {0, 0};

  this->Init_rect(buffer_origin, host_origin, region, buffer_pitch, host_pitch);
}

template < class T >
void cBuffer_checker<T>::Init_rect(int bufforg[3], int host_org[3],
                                   int region_in[3], int buffer_pitch[2], int host_pitch[2])
{
  buffer_origin[0] = bufforg[0];
  buffer_origin[1] = bufforg[1];
  buffer_origin[2] = bufforg[2];

  host_origin[0] = host_org[0];
  host_origin[1] = host_org[1];
  host_origin[2] = host_org[2];

  region[0] = region_in[0];
  region[1] = region_in[1];
  region[2] = region_in[2];

  buffer_row_pitch   = buffer_pitch[0];
  buffer_slice_pitch = buffer_pitch[1];
  host_row_pitch     = host_pitch[0];
  host_slice_pitch   = host_pitch[1];

  int sizeof_element = sizeof(T);
  for (int k=0; k<3; k++)
  {
    buffer_origin_bytes[k] = buffer_origin[k] * sizeof_element;
    host_origin_bytes [k]  = host_origin[k] * sizeof_element;
  }

  region_bytes[0]          = region[0] * sizeof_element;
  region_bytes[1]          = region[1];
  region_bytes[2]          = region[2];
  buffer_row_pitch_bytes   = buffer_row_pitch*  sizeof_element;
  buffer_slice_pitch_bytes = buffer_slice_pitch*  sizeof_element;
  host_row_pitch_bytes     = host_row_pitch*  sizeof_element;
  host_slice_pitch_bytes   = host_slice_pitch*  sizeof_element;
}

#endif