| // |
| // 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 <stdio.h> |
| #include <stdlib.h> |
| |
| #include "../../test_common/harness/errorHelpers.h" |
| #include "../../test_common/harness/kernelHelpers.h" |
| |
| #include "utils.h" |
| |
| int kernel_functions(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements, |
| unsigned int iterationNum, unsigned int width, unsigned int height, cl_dx9_media_adapter_type_khr adapterType, |
| TSurfaceFormat surfaceFormat, TSharedHandleType sharedHandle) |
| { |
| const unsigned int FRAME_NUM = 2; |
| const cl_uchar MAX_VALUE = 255 / 2; |
| const std::string PROGRAM_STR = |
| "__kernel void TestFunction( read_only image2d_t planeIn, write_only image2d_t planeOut, " |
| NL " sampler_t sampler, __global int *planeRes)" |
| NL "{" |
| NL " int w = get_global_id(0);" |
| NL " int h = get_global_id(1);" |
| NL " int width = get_image_width(planeIn);" |
| NL " int height = get_image_height(planeOut);" |
| NL " float4 color0 = read_imagef(planeIn, sampler, (int2)(w,h)) + 0.2f;" |
| NL " float4 color1 = read_imagef(planeIn, sampler, (float2)(w,h)) + 0.2f;" |
| NL " color0 = (color0 == color1) ? color0: (float4)(0.5, 0.5, 0.5, 0.5);" |
| NL " write_imagef(planeOut, (int2)(w,h), color0);" |
| NL " if(w == 0 && h == 0)" |
| NL " {" |
| NL " planeRes[0] = width;" |
| NL " planeRes[1] = height;" |
| NL " }" |
| NL "}"; |
| |
| CResult result; |
| |
| std::auto_ptr<CDeviceWrapper> deviceWrapper; |
| if (!DeviceCreate(adapterType, deviceWrapper)) |
| { |
| result.ResultSub(CResult::TEST_ERROR); |
| return result.Result(); |
| } |
| |
| std::vector<std::vector<cl_uchar> > bufferIn(FRAME_NUM); |
| std::vector<std::vector<cl_uchar> > bufferExp(FRAME_NUM); |
| size_t frameSize = width * height * 3 / 2; |
| cl_uchar step = MAX_VALUE / FRAME_NUM; |
| for (size_t i = 0; i < FRAME_NUM; ++i) |
| { |
| if (!YUVGenerate(surfaceFormat, bufferIn[i], width, height, static_cast<cl_uchar>(step * i), static_cast<cl_uchar>(step * (i + 1))) || |
| !YUVGenerate(surfaceFormat, bufferExp[i], width, height, static_cast<cl_uchar>(step * i), static_cast<cl_uchar>(step * (i + 1)), 0.2)) |
| { |
| result.ResultSub(CResult::TEST_ERROR); |
| return result.Result(); |
| } |
| } |
| |
| while (deviceWrapper->AdapterNext()) |
| { |
| if (surfaceFormat != SURFACE_FORMAT_NV12 && !SurfaceFormatCheck(adapterType, *deviceWrapper, surfaceFormat)) |
| { |
| std::string sharedHandleStr = (sharedHandle == SHARED_HANDLE_ENABLED)? "yes": "no"; |
| std::string formatStr; |
| std::string adapterStr; |
| SurfaceFormatToString(surfaceFormat, formatStr); |
| AdapterToString(adapterType, adapterStr); |
| log_info("Skipping test case, image format is not supported by a device (adapter type: %s, format: %s, shared handle: %s)\n", |
| adapterStr.c_str(), formatStr.c_str(), sharedHandleStr.c_str()); |
| return result.Result(); |
| } |
| |
| void *objectSrcHandle = 0; |
| std::auto_ptr<CSurfaceWrapper> surfaceSrc; |
| if (!MediaSurfaceCreate(adapterType, width, height, surfaceFormat, *deviceWrapper, surfaceSrc, |
| (sharedHandle == SHARED_HANDLE_ENABLED) ? true: false, &objectSrcHandle)) |
| { |
| log_error("Media surface creation failed for %i adapter\n", deviceWrapper->AdapterIdx()); |
| result.ResultSub(CResult::TEST_ERROR); |
| return result.Result(); |
| } |
| |
| void *objectDstHandle = 0; |
| std::auto_ptr<CSurfaceWrapper> surfaceDst; |
| if (!MediaSurfaceCreate(adapterType, width, height, surfaceFormat, *deviceWrapper, surfaceDst, |
| (sharedHandle == SHARED_HANDLE_ENABLED) ? true: false, &objectDstHandle)) |
| { |
| log_error("Media surface creation failed for %i adapter\n", deviceWrapper->AdapterIdx()); |
| result.ResultSub(CResult::TEST_ERROR); |
| return result.Result(); |
| } |
| |
| cl_context_properties contextProperties[] = { |
| CL_CONTEXT_PLATFORM, (cl_context_properties)gPlatformIDdetected, |
| AdapterTypeToContextInfo(adapterType), (cl_context_properties)deviceWrapper->Device(), |
| 0, |
| }; |
| |
| cl_int error; |
| clContextWrapper ctx = clCreateContext(&contextProperties[0], 1, &gDeviceIDdetected, NULL, NULL, &error); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clCreateContext failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| #if defined(_WIN32) |
| cl_dx9_surface_info_khr surfaceInfoSrc; |
| surfaceInfoSrc.resource = *(static_cast<CD3D9SurfaceWrapper *>(surfaceSrc.get())); |
| surfaceInfoSrc.shared_handle = objectSrcHandle; |
| |
| cl_dx9_surface_info_khr surfaceInfoDst; |
| surfaceInfoDst.resource = *(static_cast<CD3D9SurfaceWrapper *>(surfaceDst.get())); |
| surfaceInfoDst.shared_handle = objectDstHandle; |
| #else |
| void *surfaceInfoSrc = 0; |
| void *surfaceInfoDst = 0; |
| return TEST_NOT_IMPLEMENTED; |
| #endif |
| |
| std::vector<cl_mem> memObjSrcList; |
| std::vector<cl_mem> memObjDstList; |
| unsigned int planesNum = PlanesNum(surfaceFormat); |
| std::vector<clMemWrapper> planeSrcList(planesNum); |
| std::vector<clMemWrapper> planeDstList(planesNum); |
| for (unsigned int planeIdx = 0; planeIdx < planesNum; ++planeIdx) |
| { |
| planeSrcList[planeIdx] = clCreateFromDX9MediaSurfaceKHR(ctx, CL_MEM_READ_WRITE, adapterType, &surfaceInfoSrc, planeIdx, &error); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clCreateFromDX9MediaSurfaceKHR failed for plane %i: %s\n", planeIdx, IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| memObjSrcList.push_back(planeSrcList[planeIdx]); |
| |
| planeDstList[planeIdx] = clCreateFromDX9MediaSurfaceKHR(ctx, CL_MEM_READ_WRITE, adapterType, &surfaceInfoDst, planeIdx, &error); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clCreateFromDX9MediaSurfaceKHR failed for plane %i: %s\n", planeIdx, IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| memObjDstList.push_back(planeDstList[planeIdx]); |
| } |
| |
| clCommandQueueWrapper cmdQueue = clCreateCommandQueueWithProperties(ctx, gDeviceIDdetected, 0, &error ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to create command queue: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| if (!ImageInfoVerify(adapterType, memObjSrcList, width, height, surfaceSrc, objectSrcHandle)) |
| { |
| log_error("Image info verification failed\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| for (size_t frameIdx = 0; frameIdx < iterationNum; ++frameIdx) |
| { |
| if (!YUVSurfaceSet(surfaceFormat, surfaceSrc, bufferIn[frameIdx % FRAME_NUM], width, height)) |
| { |
| result.ResultSub(CResult::TEST_ERROR); |
| return result.Result(); |
| } |
| |
| error = clEnqueueAcquireDX9MediaSurfacesKHR(cmdQueue, static_cast<cl_uint>(memObjSrcList.size()), &memObjSrcList[0], 0, 0, 0); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clEnqueueAcquireDX9MediaSurfacesKHR failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| error = clEnqueueAcquireDX9MediaSurfacesKHR(cmdQueue, static_cast<cl_uint>(memObjDstList.size()), &memObjDstList[0], 0, 0, 0); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clEnqueueAcquireDX9MediaSurfacesKHR failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| clSamplerWrapper sampler = clCreateSampler( ctx, CL_FALSE, CL_ADDRESS_NONE, CL_FILTER_NEAREST, &error ); |
| if(error != CL_SUCCESS) |
| { |
| log_error("Unable to create sampler\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| clProgramWrapper program; |
| clKernelWrapper kernel; |
| const char *progPtr = PROGRAM_STR.c_str(); |
| if(create_single_kernel_helper(ctx, &program, &kernel, 1, (const char **)&progPtr, "TestFunction")) |
| result.ResultSub(CResult::TEST_FAIL); |
| |
| size_t bufferSize = sizeof(cl_int) * 2; |
| clMemWrapper imageRes = clCreateBuffer( ctx, CL_MEM_READ_WRITE, bufferSize, NULL, &error); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clCreateBuffer failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| size_t offset = 0; |
| size_t origin[3] = {0,0,0}; |
| std::vector<cl_uchar> out( frameSize, 0 ); |
| for (size_t i = 0; i < memObjSrcList.size(); ++i) |
| { |
| size_t planeWidth = (i == 0) ? width: width / 2; |
| size_t planeHeight = (i == 0) ? height: height / 2; |
| size_t regionPlane[3] = {planeWidth, planeHeight, 1}; |
| size_t threads[ 2 ] = { planeWidth, planeHeight }; |
| |
| error = clSetKernelArg( kernel, 0, sizeof( memObjSrcList[i] ), &memObjSrcList[i] ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to set kernel arguments" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clSetKernelArg( kernel, 1, sizeof( memObjDstList[i] ), &memObjDstList[i] ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to set kernel arguments" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clSetKernelArg( kernel, 2, sizeof( sampler ), &sampler ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to set kernel arguments" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clSetKernelArg( kernel, 3, sizeof( imageRes ), &imageRes ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to set kernel arguments" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| size_t localThreads[ 2 ]; |
| error = get_max_common_2D_work_group_size( ctx, kernel, threads, localThreads ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to get work group size to use" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clEnqueueNDRangeKernel( cmdQueue, kernel, 2, NULL, threads, localThreads, 0, NULL, NULL ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to execute test kernel" ); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| std::vector<cl_uint> imageResOut(2, 0); |
| error = clEnqueueReadBuffer( cmdQueue, imageRes, CL_TRUE, 0, bufferSize, &imageResOut[0], 0, NULL, NULL ); |
| if (error != CL_SUCCESS) |
| { |
| log_error("Unable to read buffer"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(imageResOut[0] != planeWidth) |
| { |
| log_error("Invalid width value, test = %i, expected = %i\n", imageResOut[0], planeWidth); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(imageResOut[1] != planeHeight) |
| { |
| log_error("Invalid height value, test = %i, expected = %i\n", imageResOut[1], planeHeight); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clEnqueueReadImage(cmdQueue, memObjDstList[i], CL_TRUE, origin, regionPlane, 0, 0, &out[offset], 0, 0, 0); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clEnqueueReadImage failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| offset += planeWidth * planeHeight; |
| } |
| |
| if (!YUVCompare(surfaceFormat, out, bufferExp[frameIdx % FRAME_NUM], width, height)) |
| { |
| log_error("Frame idx: %i, OCL objects are different than expected\n", frameIdx); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clEnqueueReleaseDX9MediaSurfacesKHR(cmdQueue, static_cast<cl_uint>(memObjSrcList.size()), &memObjSrcList[0], 0, 0, 0); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clEnqueueReleaseDX9MediaSurfacesKHR failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| error = clEnqueueReleaseDX9MediaSurfacesKHR(cmdQueue, static_cast<cl_uint>(memObjDstList.size()), &memObjDstList[0], 0, 0, 0); |
| if (error != CL_SUCCESS) |
| { |
| log_error("clEnqueueReleaseDX9MediaSurfacesKHR failed: %s\n", IGetErrorString(error)); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| std::vector<cl_uchar> bufferOut(frameSize, 0); |
| if (!YUVSurfaceGet(surfaceFormat, surfaceDst, bufferOut, width, height)) |
| { |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| if (!YUVCompare(surfaceFormat, bufferOut, bufferExp[frameIdx % FRAME_NUM], width, height)) |
| { |
| log_error("Frame idx: %i, media surface is different than expected\n", frameIdx); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| } |
| } |
| |
| if (!deviceWrapper->Status()) |
| { |
| std::string adapter; |
| AdapterToString(adapterType, adapter); |
| log_error("%s init failed\n", adapter.c_str()); |
| result.ResultSub(CResult::TEST_FAIL); |
| return result.Result(); |
| } |
| |
| return result.Result(); |
| } |
| |
| int test_kernel(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements) |
| { |
| CResult result; |
| |
| #if defined(_WIN32) |
| //D3D9 |
| if(kernel_functions(deviceID, context, queue, num_elements, 10, 256, 256, CL_ADAPTER_D3D9_KHR, |
| SURFACE_FORMAT_NV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (D3D9, NV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 3, 256, 256, CL_ADAPTER_D3D9_KHR, |
| SURFACE_FORMAT_YV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (D3D9, YV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| //D3D9EX |
| if(kernel_functions(deviceID, context, queue, num_elements, 5, 256, 512, CL_ADAPTER_D3D9EX_KHR, |
| SURFACE_FORMAT_NV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (D3D9EX, NV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 7, 512, 256, CL_ADAPTER_D3D9EX_KHR, |
| SURFACE_FORMAT_NV12, SHARED_HANDLE_ENABLED) != 0) |
| { |
| log_error("\nTest case (D3D9EX, NV12, shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 10, 256, 256, CL_ADAPTER_D3D9EX_KHR, |
| SURFACE_FORMAT_YV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (D3D9EX, YV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 15, 128, 128, CL_ADAPTER_D3D9EX_KHR, |
| SURFACE_FORMAT_YV12, SHARED_HANDLE_ENABLED) != 0) |
| { |
| log_error("\nTest case (D3D9EX, YV12, shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| //DXVA |
| if(kernel_functions(deviceID, context, queue, num_elements, 20, 128, 128, CL_ADAPTER_DXVA_KHR, |
| SURFACE_FORMAT_NV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (DXVA, NV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 40, 64, 64, CL_ADAPTER_DXVA_KHR, |
| SURFACE_FORMAT_NV12, SHARED_HANDLE_ENABLED) != 0) |
| { |
| log_error("\nTest case (DXVA, NV12, shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 5, 512, 512, CL_ADAPTER_DXVA_KHR, |
| SURFACE_FORMAT_YV12, SHARED_HANDLE_DISABLED) != 0) |
| { |
| log_error("\nTest case (DXVA, YV12, no shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| if(kernel_functions(deviceID, context, queue, num_elements, 2, 1024, 1024, CL_ADAPTER_DXVA_KHR, |
| SURFACE_FORMAT_YV12, SHARED_HANDLE_ENABLED) != 0) |
| { |
| log_error("\nTest case (DXVA, YV12, shared handle) failed\n\n"); |
| result.ResultSub(CResult::TEST_FAIL); |
| } |
| |
| #else |
| return TEST_NOT_IMPLEMENTED; |
| #endif |
| |
| return result.Result(); |
| } |
