mirror of
https://github.com/PixarAnimationStudios/OpenSubdiv
synced 2024-11-26 21:40:07 +00:00
Add OpenCL kernel and dispatcher. not to be compiled yet.
This commit is contained in:
Takahito Tejima
parent
91bb75f0ec
commit
deea155b06
542
opensubdiv/osd/clDispatcher.cpp
Normal file
542
opensubdiv/osd/clDispatcher.cpp
Normal file
@ -0,0 +1,542 @@
|
||||
//
|
||||
// Copyright (C) Pixar. All rights reserved.
|
||||
//
|
||||
// This license governs use of the accompanying software. If you
|
||||
// use the software, you accept this license. If you do not accept
|
||||
// the license, do not use the software.
|
||||
//
|
||||
// 1. Definitions
|
||||
// The terms "reproduce," "reproduction," "derivative works," and
|
||||
// "distribution" have the same meaning here as under U.S.
|
||||
// copyright law. A "contribution" is the original software, or
|
||||
// any additions or changes to the software.
|
||||
// A "contributor" is any person or entity that distributes its
|
||||
// contribution under this license.
|
||||
// "Licensed patents" are a contributor's patent claims that read
|
||||
// directly on its contribution.
|
||||
//
|
||||
// 2. Grant of Rights
|
||||
// (A) Copyright Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free copyright license to reproduce its contribution,
|
||||
// prepare derivative works of its contribution, and distribute
|
||||
// its contribution or any derivative works that you create.
|
||||
// (B) Patent Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free license under its licensed patents to make, have
|
||||
// made, use, sell, offer for sale, import, and/or otherwise
|
||||
// dispose of its contribution in the software or derivative works
|
||||
// of the contribution in the software.
|
||||
//
|
||||
// 3. Conditions and Limitations
|
||||
// (A) No Trademark License- This license does not grant you
|
||||
// rights to use any contributor's name, logo, or trademarks.
|
||||
// (B) If you bring a patent claim against any contributor over
|
||||
// patents that you claim are infringed by the software, your
|
||||
// patent license from such contributor to the software ends
|
||||
// automatically.
|
||||
// (C) If you distribute any portion of the software, you must
|
||||
// retain all copyright, patent, trademark, and attribution
|
||||
// notices that are present in the software.
|
||||
// (D) If you distribute any portion of the software in source
|
||||
// code form, you may do so only under this license by including a
|
||||
// complete copy of this license with your distribution. If you
|
||||
// distribute any portion of the software in compiled or object
|
||||
// code form, you may only do so under a license that complies
|
||||
// with this license.
|
||||
// (E) The software is licensed "as-is." You bear the risk of
|
||||
// using it. The contributors give no express warranties,
|
||||
// guarantees or conditions. You may have additional consumer
|
||||
// rights under your local laws which this license cannot change.
|
||||
// To the extent permitted under your local laws, the contributors
|
||||
// exclude the implied warranties of merchantability, fitness for
|
||||
// a particular purpose and non-infringement.
|
||||
//
|
||||
#include "../version.h"
|
||||
#include "../osd/clDispatcher.h"
|
||||
#include "../osd/local.h"
|
||||
|
||||
#if defined(_WIN32)
|
||||
#include <windows.h>
|
||||
#elif defined(__APPLE__)
|
||||
#include <OpenCL/opencl.h>
|
||||
#else
|
||||
#include <GL/glx.h>
|
||||
#include <CL/opencl.h>
|
||||
#endif
|
||||
|
||||
#ifdef _MSC_VER
|
||||
#define snprintf _snprintf
|
||||
#endif
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#define CL_CHECK_ERROR(x, ...) { if(x != CL_SUCCESS) { printf("ERROR %d : ", x); printf(__VA_ARGS__);} }
|
||||
|
||||
namespace OpenSubdiv {
|
||||
namespace OPENSUBDIV_VERSION {
|
||||
|
||||
static const char *clSource =
|
||||
#include "clKernel.inc"
|
||||
;
|
||||
|
||||
std::vector<OsdClKernelDispatcher::ClKernel> OsdClKernelDispatcher::kernelRegistry;
|
||||
|
||||
// XXX: context and queue should be moved to client code
|
||||
cl_context OsdClKernelDispatcher::_clContext = NULL;
|
||||
cl_command_queue OsdClKernelDispatcher::_clQueue = NULL;
|
||||
|
||||
OsdClVertexBuffer::OsdClVertexBuffer(int numElements, int numVertices,
|
||||
cl_context clContext, cl_command_queue clQueue) :
|
||||
OsdGpuVertexBuffer(numElements, numVertices),
|
||||
_clVbo(NULL),
|
||||
_clQueue(clQueue) {
|
||||
|
||||
// register vbo as cl resource
|
||||
cl_int ciErrNum;
|
||||
_clVbo = clCreateFromGLBuffer(clContext, CL_MEM_READ_WRITE, _vbo, &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateFromGLBuffer\n");
|
||||
}
|
||||
|
||||
OsdClVertexBuffer::~OsdClVertexBuffer() {
|
||||
|
||||
if (_clVbo)
|
||||
clReleaseMemObject(_clVbo);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClVertexBuffer::UpdateData(const float *src, int numVertices) {
|
||||
|
||||
size_t size = numVertices * _numElements * sizeof(float);
|
||||
Map();
|
||||
clEnqueueWriteBuffer(_clQueue, _clVbo, true, 0, size, src, 0, NULL, NULL);
|
||||
Unmap();
|
||||
}
|
||||
|
||||
void *
|
||||
OsdClVertexBuffer::Map() {
|
||||
|
||||
clEnqueueAcquireGLObjects(_clQueue, 1, &_clVbo, 0, 0, 0);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClVertexBuffer::Unmap() {
|
||||
|
||||
clEnqueueReleaseGLObjects(_clQueue, 1, &_clVbo, 0, 0, 0);
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------------------
|
||||
OsdClKernelDispatcher::DeviceTable::~DeviceTable() {
|
||||
|
||||
if (devicePtr) clReleaseMemObject(devicePtr);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::DeviceTable::Copy(cl_context context, int size, const void *table) {
|
||||
|
||||
if (size > 0) {
|
||||
cl_int ciErrNum;
|
||||
if (devicePtr)
|
||||
clReleaseMemObject(devicePtr);
|
||||
devicePtr = clCreateBuffer(context, CL_MEM_READ_WRITE|CL_MEM_COPY_HOST_PTR, size,
|
||||
const_cast<void*>(table), &ciErrNum);
|
||||
|
||||
CL_CHECK_ERROR(ciErrNum, "Table copy %p\n", table);
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------------------------------
|
||||
|
||||
OsdClKernelDispatcher::OsdClKernelDispatcher(int levels) :
|
||||
OsdKernelDispatcher(levels) {
|
||||
|
||||
_tables.resize(TABLE_MAX);
|
||||
|
||||
if (_clContext == NULL) initCL();
|
||||
}
|
||||
|
||||
OsdClKernelDispatcher::~OsdClKernelDispatcher() {
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::CopyTable(int tableIndex, size_t size, const void *ptr) {
|
||||
|
||||
_tables[tableIndex].Copy(_clContext, size, ptr);
|
||||
}
|
||||
|
||||
OsdVertexBuffer *
|
||||
OsdClKernelDispatcher::InitializeVertexBuffer(int numElements, int numVertices) {
|
||||
|
||||
return new OsdClVertexBuffer(numElements, numVertices, _clContext, _clQueue);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::BindVertexBuffer(OsdVertexBuffer *vertex, OsdVertexBuffer *varying) {
|
||||
|
||||
if (vertex)
|
||||
_currentVertexBuffer = dynamic_cast<OsdClVertexBuffer *>(vertex);
|
||||
else
|
||||
_currentVertexBuffer = NULL;
|
||||
|
||||
if (varying)
|
||||
_currentVaryingBuffer = dynamic_cast<OsdClVertexBuffer *>(varying);
|
||||
else
|
||||
_currentVaryingBuffer = NULL;
|
||||
|
||||
int numVertexElements = vertex ? vertex->GetNumElements() : 0;
|
||||
int numVaryingElements = varying ? varying->GetNumElements() : 0;
|
||||
|
||||
if (_currentVertexBuffer) {
|
||||
_currentVertexBuffer->Map();
|
||||
}
|
||||
if (_currentVaryingBuffer) {
|
||||
_currentVaryingBuffer->Map();
|
||||
}
|
||||
|
||||
// find cl kernel from registry (create it if needed)
|
||||
std::vector<ClKernel>::iterator it =
|
||||
std::find_if(kernelRegistry.begin(), kernelRegistry.end(),
|
||||
ClKernel::Match(numVertexElements, numVaryingElements));
|
||||
|
||||
if (it != kernelRegistry.end()) {
|
||||
_clKernel = &(*it);
|
||||
} else {
|
||||
kernelRegistry.push_back(ClKernel());
|
||||
_clKernel = &kernelRegistry.back();
|
||||
_clKernel->Compile(_clContext, numVertexElements, numVaryingElements);
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::UnbindVertexBuffer() {
|
||||
|
||||
if (_currentVertexBuffer) {
|
||||
_currentVertexBuffer->Unmap();
|
||||
}
|
||||
if (_currentVaryingBuffer) {
|
||||
_currentVaryingBuffer->Unmap();
|
||||
}
|
||||
|
||||
_currentVertexBuffer = NULL;
|
||||
_currentVaryingBuffer = NULL;
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::Synchronize() {
|
||||
clFinish(_clQueue);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyCatmarkFaceVerticesKernel(FarMesh<OsdVertex> * mesh, int offset,
|
||||
int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
cl_kernel kernel = _clKernel->GetCatmarkFaceKernel();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[F_IT].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[F_ITa].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(int), &_tableOffsets[F_IT][level-1]);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[F_ITa][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), &offset);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), &start);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), &end);
|
||||
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "face kernel lv[%d] %d\n", level, ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyCatmarkEdgeVerticesKernel(FarMesh<OsdVertex> * mesh, int offset,
|
||||
int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
cl_kernel kernel = _clKernel->GetCatmarkEdgeKernel();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[E_IT].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[E_W].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(int), &_tableOffsets[E_IT][level-1]);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[E_W][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), &offset);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), &start);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), &end);
|
||||
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "edge kernel %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyCatmarkVertexVerticesKernelB(FarMesh<OsdVertex> * mesh, int offset,
|
||||
int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
cl_kernel kernel = _clKernel->GetCatmarkVertexKernelB();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[V_ITa].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[V_IT].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(cl_mem), &_tables[V_W].devicePtr);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[V_ITa][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), &_tableOffsets[V_IT][level-1]);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), &_tableOffsets[V_W][level-1]);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), (void*)&offset);
|
||||
clSetKernelArg(kernel, 9, sizeof(int), (void*)&start);
|
||||
clSetKernelArg(kernel, 10, sizeof(int), (void*)&end);
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "vertex kernel 1 %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyCatmarkVertexVerticesKernelA(FarMesh<OsdVertex> * mesh, int offset,
|
||||
bool pass, int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
int ipass = pass;
|
||||
cl_kernel kernel = _clKernel->GetCatmarkVertexKernelA();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[V_ITa].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[V_W].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(int), &_tableOffsets[V_ITa][level-1]);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[V_W][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), (void*)&offset);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), (void*)&start);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), (void*)&end);
|
||||
clSetKernelArg(kernel, 9, sizeof(int), (void*)&ipass);
|
||||
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "vertex kernel 2 %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyLoopEdgeVerticesKernel(FarMesh<OsdVertex> * mesh, int offset,
|
||||
int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
cl_kernel kernel = _clKernel->GetLoopEdgeKernel();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[E_IT].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[E_W].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(int), &_tableOffsets[E_IT][level-1]);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[E_W][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), &offset);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), &start);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), &end);
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "edge kernel %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyLoopVertexVerticesKernelB(FarMesh<OsdVertex> * mesh, int offset,
|
||||
int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
cl_kernel kernel = _clKernel->GetLoopVertexKernelB();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[V_ITa].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[V_IT].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(cl_mem), &_tables[V_W].devicePtr);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[V_ITa][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), &_tableOffsets[V_IT][level-1]);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), &_tableOffsets[V_W][level-1]);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), &offset);
|
||||
clSetKernelArg(kernel, 9, sizeof(int), &start);
|
||||
clSetKernelArg(kernel, 10, sizeof(int), &end);
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "vertex kernel 1 %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::ApplyLoopVertexVerticesKernelA(FarMesh<OsdVertex> * mesh, int offset,
|
||||
bool pass, int level, int start, int end, void * data) const {
|
||||
|
||||
cl_int ciErrNum;
|
||||
size_t globalWorkSize[1] = { end-start };
|
||||
int ipass = pass;
|
||||
cl_kernel kernel = _clKernel->GetLoopVertexKernelA();
|
||||
|
||||
clSetKernelArg(kernel, 0, sizeof(cl_mem), GetVertexBuffer());
|
||||
clSetKernelArg(kernel, 1, sizeof(cl_mem), GetVaryingBuffer());
|
||||
clSetKernelArg(kernel, 2, sizeof(cl_mem), &_tables[V_ITa].devicePtr);
|
||||
clSetKernelArg(kernel, 3, sizeof(cl_mem), &_tables[V_W].devicePtr);
|
||||
clSetKernelArg(kernel, 4, sizeof(int), &_tableOffsets[V_ITa][level-1]);
|
||||
clSetKernelArg(kernel, 5, sizeof(int), &_tableOffsets[V_W][level-1]);
|
||||
clSetKernelArg(kernel, 6, sizeof(int), (void*)&offset);
|
||||
clSetKernelArg(kernel, 7, sizeof(int), (void*)&start);
|
||||
clSetKernelArg(kernel, 8, sizeof(int), (void*)&end);
|
||||
clSetKernelArg(kernel, 9, sizeof(int), (void*)&ipass);
|
||||
ciErrNum = clEnqueueNDRangeKernel(_clQueue, kernel, 1, NULL, globalWorkSize, NULL, 0, NULL, NULL);
|
||||
CL_CHECK_ERROR(ciErrNum, "vertex kernel 2 %d\n", ciErrNum);
|
||||
}
|
||||
|
||||
// XXX: initCL should be removed from libosd
|
||||
void
|
||||
OsdClKernelDispatcher::initCL() {
|
||||
|
||||
cl_int ciErrNum;
|
||||
|
||||
cl_platform_id cpPlatform = 0;
|
||||
cl_uint num_platforms;
|
||||
ciErrNum = clGetPlatformIDs(0, NULL, &num_platforms);
|
||||
if (ciErrNum != CL_SUCCESS) {
|
||||
OSD_ERROR("Error %i in clGetPlatformIDs call.\n", ciErrNum);
|
||||
exit(1);
|
||||
}
|
||||
if (num_platforms == 0) {
|
||||
OSD_ERROR("No OpenCL platform found.\n");
|
||||
exit(1);
|
||||
}
|
||||
cl_platform_id *clPlatformIDs;
|
||||
clPlatformIDs = new cl_platform_id[num_platforms];
|
||||
ciErrNum = clGetPlatformIDs(num_platforms, clPlatformIDs, NULL);
|
||||
char chBuffer[1024];
|
||||
for (cl_uint i = 0; i < num_platforms; ++i) {
|
||||
ciErrNum = clGetPlatformInfo(clPlatformIDs[i], CL_PLATFORM_NAME, 1024, chBuffer,NULL);
|
||||
if (ciErrNum == CL_SUCCESS) {
|
||||
cpPlatform = clPlatformIDs[i];
|
||||
}
|
||||
}
|
||||
// -------------
|
||||
cl_device_id cdDevice;
|
||||
clGetDeviceIDs(cpPlatform, CL_DEVICE_TYPE_GPU, 1, &cdDevice, NULL);
|
||||
|
||||
#if defined(_WIN32)
|
||||
cl_context_properties props[] = {
|
||||
CL_GL_CONTEXT_KHR, (cl_context_properties)wglGetCurrentContext(),
|
||||
CL_WGL_HDC_KHR, (cl_context_properties)wglGetCurrentDC(),
|
||||
CL_CONTEXT_PLATFORM, (cl_context_properties)cpPlatform,
|
||||
0
|
||||
};
|
||||
#elif defined(__APPLE__)
|
||||
CGLContextObj kCGLContext = CGLGetCurrentContext();
|
||||
CGLShareGroupObj kCGLShareGroup = CGLGetShareGroup(kCGLContext);
|
||||
cl_context_properties props[] = {
|
||||
CL_CONTEXT_PROPERTY_USE_CGL_SHAREGROUP_APPLE, (cl_context_properties)kCGLShareGroup,
|
||||
0
|
||||
};
|
||||
#else
|
||||
cl_context_properties props[] = {
|
||||
CL_GL_CONTEXT_KHR, (cl_context_properties)glXGetCurrentContext(),
|
||||
CL_GLX_DISPLAY_KHR, (cl_context_properties)glXGetCurrentDisplay(),
|
||||
CL_CONTEXT_PLATFORM, (cl_context_properties)cpPlatform,
|
||||
0
|
||||
};
|
||||
#endif
|
||||
|
||||
// XXX context creation should be moved to client code
|
||||
_clContext = clCreateContext(props, 1, &cdDevice, NULL, NULL, &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateContext\n");
|
||||
|
||||
_clQueue = clCreateCommandQueue(_clContext, cdDevice, 0, &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateCommandQueue\n");
|
||||
}
|
||||
|
||||
void
|
||||
OsdClKernelDispatcher::uninitCL() {
|
||||
|
||||
// XXX: who calls this function...
|
||||
clReleaseCommandQueue(_clQueue);
|
||||
clReleaseContext(_clContext);
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------------
|
||||
|
||||
OsdClKernelDispatcher::ClKernel::ClKernel() :
|
||||
_clCatmarkFace(NULL),
|
||||
_clCatmarkEdge(NULL),
|
||||
_clCatmarkVertexA(NULL),
|
||||
_clCatmarkVertexB(NULL),
|
||||
_clLoopEdge(NULL),
|
||||
_clLoopVertexA(NULL),
|
||||
_clLoopVertexB(NULL),
|
||||
_clProgram(NULL) {
|
||||
}
|
||||
|
||||
OsdClKernelDispatcher::ClKernel::~ClKernel() {
|
||||
|
||||
if (_clCatmarkFace)
|
||||
clReleaseKernel(_clCatmarkFace);
|
||||
if (_clCatmarkEdge)
|
||||
clReleaseKernel(_clCatmarkEdge);
|
||||
if (_clCatmarkVertexA)
|
||||
clReleaseKernel(_clCatmarkVertexA);
|
||||
if (_clCatmarkVertexB)
|
||||
clReleaseKernel(_clCatmarkVertexB);
|
||||
|
||||
if (_clLoopEdge)
|
||||
clReleaseKernel(_clLoopEdge);
|
||||
if (_clLoopVertexA)
|
||||
clReleaseKernel(_clLoopVertexA);
|
||||
if (_clLoopVertexB)
|
||||
clReleaseKernel(_clLoopVertexB);
|
||||
|
||||
if (_clProgram) clReleaseProgram(_clProgram);
|
||||
}
|
||||
|
||||
bool
|
||||
OsdClKernelDispatcher::ClKernel::Compile(cl_context clContext, int numVertexElements, int numVaryingElements) {
|
||||
|
||||
cl_int ciErrNum;
|
||||
|
||||
_numVertexElements = numVertexElements;
|
||||
_numVaryingElements = numVaryingElements;
|
||||
|
||||
char constantDefine[256];
|
||||
snprintf(constantDefine, 256, "#define NUM_VERTEX_ELEMENTS %d\n"
|
||||
"#define NUM_VARYING_ELEMENTS %d\n", numVertexElements, numVaryingElements);
|
||||
|
||||
const char *sources[] = { constantDefine, clSource };
|
||||
|
||||
_clProgram = clCreateProgramWithSource(clContext, 2, sources, 0, &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateProgramWithSource\n");
|
||||
|
||||
ciErrNum = clBuildProgram(_clProgram, 0, NULL, NULL, NULL, NULL);
|
||||
if (ciErrNum != CL_SUCCESS) {
|
||||
OSD_ERROR("ERROR in clBuildProgram %d\n", ciErrNum);
|
||||
//char cBuildLog[10240];
|
||||
//clGetProgramBuildInfo(_clProgram, cdDevice, CL_PROGRAM_BUILD_LOG,
|
||||
// sizeof(cBuildLog), cBuildLog, NULL);
|
||||
//OSD_ERROR(cBuildLog);
|
||||
return false;
|
||||
}
|
||||
|
||||
// -------
|
||||
|
||||
_clCatmarkFace = clCreateKernel(_clProgram, "computeFace", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel face\n");
|
||||
_clCatmarkEdge = clCreateKernel(_clProgram, "computeEdge", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel edge\n");
|
||||
_clCatmarkVertexA = clCreateKernel(_clProgram, "computeVertexA", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel vertex a\n");
|
||||
_clCatmarkVertexB = clCreateKernel(_clProgram, "computeVertexB", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel vertex b\n");
|
||||
_clLoopEdge = clCreateKernel(_clProgram, "computeEdge", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel edge\n");
|
||||
_clLoopVertexA = clCreateKernel(_clProgram, "computeVertexA", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel vertex a\n");
|
||||
_clLoopVertexB = clCreateKernel(_clProgram, "computeLoopVertexB", &ciErrNum);
|
||||
CL_CHECK_ERROR(ciErrNum, "clCreateKernel vertex b\n");
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
} // end namespace OPENSUBDIV_VERSION
|
||||
} // end namespace OpenSubdiv
|
||||
215
opensubdiv/osd/clDispatcher.h
Normal file
215
opensubdiv/osd/clDispatcher.h
Normal file
@ -0,0 +1,215 @@
|
||||
//
|
||||
// Copyright (C) Pixar. All rights reserved.
|
||||
//
|
||||
// This license governs use of the accompanying software. If you
|
||||
// use the software, you accept this license. If you do not accept
|
||||
// the license, do not use the software.
|
||||
//
|
||||
// 1. Definitions
|
||||
// The terms "reproduce," "reproduction," "derivative works," and
|
||||
// "distribution" have the same meaning here as under U.S.
|
||||
// copyright law. A "contribution" is the original software, or
|
||||
// any additions or changes to the software.
|
||||
// A "contributor" is any person or entity that distributes its
|
||||
// contribution under this license.
|
||||
// "Licensed patents" are a contributor's patent claims that read
|
||||
// directly on its contribution.
|
||||
//
|
||||
// 2. Grant of Rights
|
||||
// (A) Copyright Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free copyright license to reproduce its contribution,
|
||||
// prepare derivative works of its contribution, and distribute
|
||||
// its contribution or any derivative works that you create.
|
||||
// (B) Patent Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free license under its licensed patents to make, have
|
||||
// made, use, sell, offer for sale, import, and/or otherwise
|
||||
// dispose of its contribution in the software or derivative works
|
||||
// of the contribution in the software.
|
||||
//
|
||||
// 3. Conditions and Limitations
|
||||
// (A) No Trademark License- This license does not grant you
|
||||
// rights to use any contributor's name, logo, or trademarks.
|
||||
// (B) If you bring a patent claim against any contributor over
|
||||
// patents that you claim are infringed by the software, your
|
||||
// patent license from such contributor to the software ends
|
||||
// automatically.
|
||||
// (C) If you distribute any portion of the software, you must
|
||||
// retain all copyright, patent, trademark, and attribution
|
||||
// notices that are present in the software.
|
||||
// (D) If you distribute any portion of the software in source
|
||||
// code form, you may do so only under this license by including a
|
||||
// complete copy of this license with your distribution. If you
|
||||
// distribute any portion of the software in compiled or object
|
||||
// code form, you may only do so under a license that complies
|
||||
// with this license.
|
||||
// (E) The software is licensed "as-is." You bear the risk of
|
||||
// using it. The contributors give no express warranties,
|
||||
// guarantees or conditions. You may have additional consumer
|
||||
// rights under your local laws which this license cannot change.
|
||||
// To the extent permitted under your local laws, the contributors
|
||||
// exclude the implied warranties of merchantability, fitness for
|
||||
// a particular purpose and non-infringement.
|
||||
//
|
||||
#ifndef OSD_CL_DISPATCHER_H
|
||||
#define OSD_CL_DISPATCHER_H
|
||||
|
||||
#include "../version.h"
|
||||
#include "../osd/kernelDispatcher.h"
|
||||
#include "../osd/vertexBuffer.h"
|
||||
|
||||
#if defined(__APPLE__)
|
||||
#include <OpenCL/opencl.h>
|
||||
#else
|
||||
#include <CL/opencl.h>
|
||||
#endif
|
||||
|
||||
namespace OpenSubdiv {
|
||||
namespace OPENSUBDIV_VERSION {
|
||||
|
||||
class OsdClVertexBuffer : public OsdGpuVertexBuffer {
|
||||
public:
|
||||
OsdClVertexBuffer(int numElements, int numVertices, cl_context context, cl_command_queue queue);
|
||||
virtual ~OsdClVertexBuffer();
|
||||
|
||||
virtual void UpdateData(const float *src, int numVertices);
|
||||
void * Map();
|
||||
void Unmap();
|
||||
|
||||
cl_mem const * GetClBuffer() const { return &_clVbo; }
|
||||
|
||||
protected:
|
||||
cl_mem _clVbo;
|
||||
cl_command_queue _clQueue;
|
||||
};
|
||||
|
||||
class OsdClKernelDispatcher : public OsdKernelDispatcher
|
||||
{
|
||||
public:
|
||||
OsdClKernelDispatcher(int levels);
|
||||
virtual ~OsdClKernelDispatcher();
|
||||
|
||||
|
||||
virtual void ApplyCatmarkFaceVerticesKernel(FarMesh<OsdVertex> * mesh, int offset, int level, int start, int end, void * data) const;
|
||||
|
||||
virtual void ApplyCatmarkEdgeVerticesKernel(FarMesh<OsdVertex> * mesh, int offset, int level, int start, int end, void * data) const;
|
||||
|
||||
virtual void ApplyCatmarkVertexVerticesKernelB(FarMesh<OsdVertex> * mesh, int offset, int level, int start, int end, void * data) const;
|
||||
|
||||
virtual void ApplyCatmarkVertexVerticesKernelA(FarMesh<OsdVertex> * mesh, int offset, bool pass, int level, int start, int end, void * data) const;
|
||||
|
||||
|
||||
virtual void ApplyLoopEdgeVerticesKernel(FarMesh<OsdVertex> * mesh, int offset, int level, int start, int end, void * data) const;
|
||||
|
||||
virtual void ApplyLoopVertexVerticesKernelB(FarMesh<OsdVertex> * mesh, int offset, int level, int start, int end, void * data) const;
|
||||
|
||||
virtual void ApplyLoopVertexVerticesKernelA(FarMesh<OsdVertex> * mesh, int offset, bool pass, int level, int start, int end, void * data) const;
|
||||
|
||||
|
||||
virtual void CopyTable(int tableIndex, size_t size, const void *ptr);
|
||||
|
||||
virtual void OnKernelLaunch() {}
|
||||
|
||||
virtual void OnKernelFinish() {}
|
||||
|
||||
virtual OsdVertexBuffer *InitializeVertexBuffer(int numElements, int count);
|
||||
|
||||
virtual void BindVertexBuffer(OsdVertexBuffer *vertex, OsdVertexBuffer *varying);
|
||||
|
||||
virtual void UnbindVertexBuffer();
|
||||
|
||||
virtual void Synchronize();
|
||||
|
||||
static OsdKernelDispatcher * Create(int levels) {
|
||||
return new OsdClKernelDispatcher(levels);
|
||||
}
|
||||
|
||||
static void Register() {
|
||||
Factory::GetInstance().Register(Create, kCL);
|
||||
}
|
||||
|
||||
protected:
|
||||
class ClKernel {
|
||||
public:
|
||||
ClKernel();
|
||||
~ClKernel();
|
||||
|
||||
bool Compile(cl_context clContext, int numVertexElements, int numVaryingElements);
|
||||
|
||||
cl_kernel GetCatmarkFaceKernel() const { return _clCatmarkFace; }
|
||||
cl_kernel GetCatmarkEdgeKernel() const { return _clCatmarkEdge; }
|
||||
cl_kernel GetCatmarkVertexKernelA() const { return _clCatmarkVertexA; }
|
||||
cl_kernel GetCatmarkVertexKernelB() const { return _clCatmarkVertexB; }
|
||||
cl_kernel GetLoopEdgeKernel() const { return _clLoopEdge; }
|
||||
cl_kernel GetLoopVertexKernelA() const { return _clLoopVertexA; }
|
||||
cl_kernel GetLoopVertexKernelB() const { return _clLoopVertexB; }
|
||||
|
||||
struct Match {
|
||||
Match(int numVertexElements, int numVaryingElements) :
|
||||
_numVertexElements(numVertexElements), _numVaryingElements(numVaryingElements) {}
|
||||
bool operator() (const ClKernel &kernel) {
|
||||
return (kernel._numVertexElements == _numVertexElements
|
||||
&& kernel._numVaryingElements == _numVaryingElements);
|
||||
}
|
||||
int _numVertexElements, _numVaryingElements;
|
||||
};
|
||||
|
||||
friend class Match;
|
||||
|
||||
protected:
|
||||
cl_program _clProgram;
|
||||
cl_kernel _clCatmarkFace, _clCatmarkEdge, _clCatmarkVertexA, _clCatmarkVertexB;
|
||||
cl_kernel _clLoopEdge, _clLoopVertexA, _clLoopVertexB;
|
||||
int _numVertexElements, _numVaryingElements;
|
||||
};
|
||||
|
||||
struct DeviceTable
|
||||
{
|
||||
DeviceTable() : devicePtr(NULL) {}
|
||||
~DeviceTable();
|
||||
|
||||
void Copy(cl_context context, int size, const void *ptr);
|
||||
|
||||
cl_mem devicePtr;
|
||||
};
|
||||
|
||||
cl_mem const * GetVertexBuffer() const {
|
||||
|
||||
static cl_mem nul = NULL;
|
||||
return _currentVertexBuffer ? _currentVertexBuffer->GetClBuffer() : &nul;
|
||||
}
|
||||
|
||||
cl_mem const * GetVaryingBuffer() const {
|
||||
|
||||
static cl_mem nul = NULL;
|
||||
return _currentVaryingBuffer ? _currentVaryingBuffer->GetClBuffer() : &nul;
|
||||
}
|
||||
|
||||
void initCL();
|
||||
|
||||
void uninitCL();
|
||||
|
||||
OsdClVertexBuffer *_currentVertexBuffer,
|
||||
*_currentVaryingBuffer;
|
||||
|
||||
std::vector<DeviceTable> _tables;
|
||||
|
||||
ClKernel * _clKernel;
|
||||
|
||||
// XXX: context and queue should be moved to client code
|
||||
static cl_context _clContext;
|
||||
static cl_command_queue _clQueue;
|
||||
|
||||
// static shader registry (XXX tentative..)
|
||||
static std::vector<ClKernel> kernelRegistry;
|
||||
};
|
||||
|
||||
} // end namespace OPENSUBDIV_VERSION
|
||||
using namespace OPENSUBDIV_VERSION;
|
||||
|
||||
} // end namespace OpenSubdiv
|
||||
|
||||
#endif // OSD_CL_DISPATCHER_H
|
||||
292
opensubdiv/osd/clKernel.cl
Normal file
292
opensubdiv/osd/clKernel.cl
Normal file
@ -0,0 +1,292 @@
|
||||
//
|
||||
// Copyright (C) Pixar. All rights reserved.
|
||||
//
|
||||
// This license governs use of the accompanying software. If you
|
||||
// use the software, you accept this license. If you do not accept
|
||||
// the license, do not use the software.
|
||||
//
|
||||
// 1. Definitions
|
||||
// The terms "reproduce," "reproduction," "derivative works," and
|
||||
// "distribution" have the same meaning here as under U.S.
|
||||
// copyright law. A "contribution" is the original software, or
|
||||
// any additions or changes to the software.
|
||||
// A "contributor" is any person or entity that distributes its
|
||||
// contribution under this license.
|
||||
// "Licensed patents" are a contributor's patent claims that read
|
||||
// directly on its contribution.
|
||||
//
|
||||
// 2. Grant of Rights
|
||||
// (A) Copyright Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free copyright license to reproduce its contribution,
|
||||
// prepare derivative works of its contribution, and distribute
|
||||
// its contribution or any derivative works that you create.
|
||||
// (B) Patent Grant- Subject to the terms of this license,
|
||||
// including the license conditions and limitations in section 3,
|
||||
// each contributor grants you a non-exclusive, worldwide,
|
||||
// royalty-free license under its licensed patents to make, have
|
||||
// made, use, sell, offer for sale, import, and/or otherwise
|
||||
// dispose of its contribution in the software or derivative works
|
||||
// of the contribution in the software.
|
||||
//
|
||||
// 3. Conditions and Limitations
|
||||
// (A) No Trademark License- This license does not grant you
|
||||
// rights to use any contributor's name, logo, or trademarks.
|
||||
// (B) If you bring a patent claim against any contributor over
|
||||
// patents that you claim are infringed by the software, your
|
||||
// patent license from such contributor to the software ends
|
||||
// automatically.
|
||||
// (C) If you distribute any portion of the software, you must
|
||||
// retain all copyright, patent, trademark, and attribution
|
||||
// notices that are present in the software.
|
||||
// (D) If you distribute any portion of the software in source
|
||||
// code form, you may do so only under this license by including a
|
||||
// complete copy of this license with your distribution. If you
|
||||
// distribute any portion of the software in compiled or object
|
||||
// code form, you may only do so under a license that complies
|
||||
// with this license.
|
||||
// (E) The software is licensed "as-is." You bear the risk of
|
||||
// using it. The contributors give no express warranties,
|
||||
// guarantees or conditions. You may have additional consumer
|
||||
// rights under your local laws which this license cannot change.
|
||||
// To the extent permitted under your local laws, the contributors
|
||||
// exclude the implied warranties of merchantability, fitness for
|
||||
// a particular purpose and non-infringement.
|
||||
//
|
||||
|
||||
struct Vertex
|
||||
{
|
||||
float v[NUM_VERTEX_ELEMENTS];
|
||||
};
|
||||
|
||||
struct Varying
|
||||
{
|
||||
float v[NUM_VARYING_ELEMENTS];
|
||||
};
|
||||
|
||||
__global void clearVertex(struct Vertex *vertex) {
|
||||
|
||||
for (int i = 0; i < NUM_VERTEX_ELEMENTS; i++) {
|
||||
vertex->v[i] = 0;
|
||||
}
|
||||
}
|
||||
__global void clearVarying(struct Varying *varying) {
|
||||
|
||||
for (int i = 0; i < NUM_VARYING_ELEMENTS; i++) {
|
||||
varying->v[i] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
__global void addWithWeight(struct Vertex *dst, __global struct Vertex *src, float weight) {
|
||||
|
||||
for (int i = 0; i < NUM_VERTEX_ELEMENTS; i++) {
|
||||
dst->v[i] += src->v[i] * weight;
|
||||
}
|
||||
}
|
||||
|
||||
__global void addVaryingWithWeight(struct Varying *dst, __global struct Varying *src, float weight) {
|
||||
|
||||
for (int i = 0; i < NUM_VARYING_ELEMENTS; i++) {
|
||||
dst->v[i] += src->v[i] * weight;
|
||||
}
|
||||
}
|
||||
|
||||
// ----------------------------------------------------------------------------------------
|
||||
|
||||
__kernel void computeFace(__global struct Vertex *vertex,
|
||||
__global struct Varying *varying,
|
||||
__global int *F_IT,
|
||||
__global int *F_ITa,
|
||||
int ofs_F_IT, int ofs_F_ITa,
|
||||
int offset, int start, int end) {
|
||||
|
||||
F_IT += ofs_F_IT;
|
||||
F_ITa += ofs_F_ITa;
|
||||
|
||||
int i = start + get_global_id(0);
|
||||
int h = F_ITa[2*i];
|
||||
int n = F_ITa[2*i+1];
|
||||
|
||||
float weight = 1.0f/n;
|
||||
|
||||
struct Vertex dst;
|
||||
struct Varying dstVarying;
|
||||
clearVertex(&dst);
|
||||
clearVarying(&dstVarying);
|
||||
for (int j=0; j<n; ++j) {
|
||||
int index = F_IT[h+j];
|
||||
addWithWeight(&dst, &vertex[index], weight);
|
||||
if(varying) addVaryingWithWeight(&dstVarying, &varying[index], weight);
|
||||
}
|
||||
vertex[i+offset] = dst;
|
||||
if(varying) varying[i+offset] = dstVarying;
|
||||
}
|
||||
|
||||
__kernel void computeEdge(__global struct Vertex *vertex,
|
||||
__global struct Varying *varying,
|
||||
__global int *E_IT,
|
||||
__global float *E_W,
|
||||
int ofs_E_IT, int ofs_E_W,
|
||||
int offset, int start, int end) {
|
||||
|
||||
E_IT += ofs_E_IT;
|
||||
E_W += ofs_E_W;
|
||||
|
||||
int i = start + get_global_id(0);
|
||||
int eidx0 = E_IT[4*i+0];
|
||||
int eidx1 = E_IT[4*i+1];
|
||||
int eidx2 = E_IT[4*i+2];
|
||||
int eidx3 = E_IT[4*i+3];
|
||||
|
||||
float vertWeight = E_W[i*2+0];
|
||||
|
||||
// Fully sharp edge : vertWeight = 0.5f;
|
||||
struct Vertex dst;
|
||||
struct Varying dstVarying;
|
||||
clearVertex(&dst);
|
||||
clearVarying(&dstVarying);
|
||||
|
||||
addWithWeight(&dst, &vertex[eidx0], vertWeight);
|
||||
addWithWeight(&dst, &vertex[eidx1], vertWeight);
|
||||
|
||||
if (eidx2 > -1) {
|
||||
float faceWeight = E_W[i*2+1];
|
||||
|
||||
addWithWeight(&dst, &vertex[eidx2], faceWeight);
|
||||
addWithWeight(&dst, &vertex[eidx3], faceWeight);
|
||||
}
|
||||
|
||||
vertex[i+offset] = dst;
|
||||
|
||||
if (varying) {
|
||||
addVaryingWithWeight(&dstVarying, &varying[eidx0], 0.5f);
|
||||
addVaryingWithWeight(&dstVarying, &varying[eidx1], 0.5f);
|
||||
varying[i+offset] = dstVarying;
|
||||
}
|
||||
}
|
||||
|
||||
__kernel void computeVertexA(__global struct Vertex *vertex,
|
||||
__global struct Varying *varying,
|
||||
__global int *V_ITa,
|
||||
__global float *V_W,
|
||||
int ofs_V_ITa, int ofs_V_W,
|
||||
int offset, int start, int end, int pass) {
|
||||
V_ITa += ofs_V_ITa;
|
||||
V_W += ofs_V_W;
|
||||
|
||||
int i = start + get_global_id(0);
|
||||
int n = V_ITa[5*i+1];
|
||||
int p = V_ITa[5*i+2];
|
||||
int eidx0 = V_ITa[5*i+3];
|
||||
int eidx1 = V_ITa[5*i+4];
|
||||
|
||||
float weight = (pass==1) ? V_W[i] : 1.0f - V_W[i];
|
||||
|
||||
// In the case of fractional weight, the weight must be inverted since
|
||||
// the value is shared with the k_Smooth kernel (statistically the
|
||||
// k_Smooth kernel runs much more often than this one)
|
||||
if (weight>0.0f && weight<1.0f && n > 0)
|
||||
weight=1.0f-weight;
|
||||
|
||||
struct Vertex dst;
|
||||
if (not pass)
|
||||
clearVertex(&dst);
|
||||
else
|
||||
dst = vertex[i+offset];
|
||||
|
||||
if (eidx0==-1 || (pass==0 && (n==-1)) ) {
|
||||
addWithWeight(&dst, &vertex[p], weight);
|
||||
} else {
|
||||
addWithWeight(&dst, &vertex[p], weight * 0.75f);
|
||||
addWithWeight(&dst, &vertex[eidx0], weight * 0.125f);
|
||||
addWithWeight(&dst, &vertex[eidx1], weight * 0.125f);
|
||||
}
|
||||
vertex[i+offset] = dst;
|
||||
|
||||
if (not pass && varying) {
|
||||
struct Varying dstVarying;
|
||||
clearVarying(&dstVarying);
|
||||
addVaryingWithWeight(&dstVarying, &varying[p], 1.0f);
|
||||
varying[i+offset] = dstVarying;
|
||||
}
|
||||
}
|
||||
|
||||
__kernel void computeVertexB(__global struct Vertex *vertex,
|
||||
__global struct Varying *varying,
|
||||
__global int *V_ITa,
|
||||
__global int *V_IT,
|
||||
__global float *V_W,
|
||||
int ofs_V_ITa, int ofs_V_IT, int ofs_V_W,
|
||||
int offset, int start, int end) {
|
||||
V_ITa += ofs_V_ITa;
|
||||
V_IT += ofs_V_IT;
|
||||
V_W += ofs_V_W;
|
||||
|
||||
int i = start + get_global_id(0);
|
||||
int h = V_ITa[5*i];
|
||||
int n = V_ITa[5*i+1];
|
||||
int p = V_ITa[5*i+2];
|
||||
|
||||
float weight = V_W[i];
|
||||
float wp = 1.0f/(float)(n*n);
|
||||
float wv = (n-2.0f) * n * wp;
|
||||
|
||||
struct Vertex dst;
|
||||
clearVertex(&dst);
|
||||
|
||||
addWithWeight(&dst, &vertex[p], weight * wv);
|
||||
|
||||
for (int j = 0; j < n; ++j) {
|
||||
addWithWeight(&dst, &vertex[V_IT[h+j*2]], weight * wp);
|
||||
addWithWeight(&dst, &vertex[V_IT[h+j*2+1]], weight * wp);
|
||||
}
|
||||
vertex[i+offset] = dst;
|
||||
|
||||
if (varying) {
|
||||
struct Varying dstVarying;
|
||||
clearVarying(&dstVarying);
|
||||
addVaryingWithWeight(&dstVarying, &varying[p], 1.0f);
|
||||
varying[i+offset] = dstVarying;
|
||||
}
|
||||
}
|
||||
|
||||
__kernel void computeLoopVertexB(__global struct Vertex *vertex,
|
||||
__global struct Varying *varying,
|
||||
__global int *V_ITa,
|
||||
__global int *V_IT,
|
||||
__global float *V_W,
|
||||
int ofs_V_ITa, int ofs_V_IT, int ofs_V_W,
|
||||
int offset, int start, int end) {
|
||||
|
||||
V_ITa += ofs_V_ITa;
|
||||
V_IT += ofs_V_IT;
|
||||
V_W += ofs_V_W;
|
||||
|
||||
int i = start + get_global_id(0);
|
||||
int h = V_ITa[5*i];
|
||||
int n = V_ITa[5*i+1];
|
||||
int p = V_ITa[5*i+2];
|
||||
|
||||
float weight = V_W[i];
|
||||
float wp = 1.0f/(float)(n);
|
||||
float beta = 0.25f * cos((float)(M_PI) * 2.0f * wp) + 0.375f;
|
||||
beta = beta * beta;
|
||||
beta = (0.625f - beta) * wp;
|
||||
|
||||
struct Vertex dst;
|
||||
clearVertex(&dst);
|
||||
addWithWeight(&dst, &vertex[p], weight * (1.0f - (beta * n)));
|
||||
|
||||
for (int j = 0; j < n; ++j) {
|
||||
addWithWeight(&dst, &vertex[V_IT[h+j]], weight * beta);
|
||||
}
|
||||
vertex[i+offset] = dst;
|
||||
|
||||
if (varying) {
|
||||
struct Varying dstVarying;
|
||||
clearVarying(&dstVarying);
|
||||
addVaryingWithWeight(&dstVarying, &varying[p], 1.0f);
|
||||
varying[i+offset] = dstVarying;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user