mirror of
https://github.com/PixarAnimationStudios/OpenSubdiv
synced 2024-12-04 00:41:05 +00:00
44a7cb6a45
* replace void* of all kernel applications with CONTEXT template parameter. It eliminates many static_casts from void* for both far and osd classes. * move the big switch-cases of far default kernel launches out of Refine so that osd controllers can arbitrary mix default kernels and custom kernels. * change FarKernelBatch::kernelType from enum to int, clients can add custom kernel types. * remove a back-pointer to farmesh from subdivision table. * untemplate all subdivision table classes and template their compute methods instead. Those methods take a typed vertex storage. * remove an unused argument FarMesh from the constructor of subdivision table factories.
372 lines
14 KiB
C++
372 lines
14 KiB
C++
//
|
|
// Copyright 2013 Pixar
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "Apache License")
|
|
// with the following modification; you may not use this file except in
|
|
// compliance with the Apache License and the following modification to it:
|
|
// Section 6. Trademarks. is deleted and replaced with:
|
|
//
|
|
// 6. Trademarks. This License does not grant permission to use the trade
|
|
// names, trademarks, service marks, or product names of the Licensor
|
|
// and its affiliates, except as required to comply with Section 4(c) of
|
|
// the License and to reproduce the content of the NOTICE file.
|
|
//
|
|
// You may obtain a copy of the Apache License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the Apache License with the above modification is
|
|
// distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the Apache License for the specific
|
|
// language governing permissions and limitations under the Apache License.
|
|
//
|
|
|
|
#include "../osd/cudaComputeContext.h"
|
|
#include "../osd/cudaComputeController.h"
|
|
|
|
#include <cuda_runtime.h>
|
|
#include <string.h>
|
|
|
|
extern "C" {
|
|
|
|
void OsdCudaComputeFace(float *vertex, float *varying,
|
|
int numUserVertexElements, int numVaryingElements,
|
|
int *F_IT, int *F_ITa, int offset, int tableOffset, int start, int end);
|
|
|
|
void OsdCudaComputeEdge(float *vertex, float *varying,
|
|
int numUserVertexElements, int numVaryingElements,
|
|
int *E_IT, float *E_W, int offset, int tableOffset, int start, int end);
|
|
|
|
void OsdCudaComputeVertexA(float *vertex, float *varying,
|
|
int numUserVertexElements, int numVaryingElements,
|
|
int *V_ITa, float *V_W, int offset, int tableOffset,
|
|
int start, int end, int pass);
|
|
|
|
void OsdCudaComputeVertexB(float *vertex, float *varying,
|
|
int numUserVertexElements, int numVaryingElements,
|
|
int *V_ITa, int *V_IT, float *V_W, int offset, int tableOffset,
|
|
int start, int end);
|
|
|
|
void OsdCudaComputeLoopVertexB(float *vertex, float *varying,
|
|
int numUserVertexElements,
|
|
int numVaryingElements,
|
|
int *V_ITa, int *V_IT, float *V_W, int offset, int tableOffset,
|
|
int start, int end);
|
|
|
|
void OsdCudaComputeBilinearEdge(float *vertex, float *varying,
|
|
int numUserVertexElements,
|
|
int numVaryingElements,
|
|
int *E_IT, int offset, int tableOffset, int start, int end);
|
|
|
|
void OsdCudaComputeBilinearVertex(float *vertex, float *varying,
|
|
int numUserVertexElements,
|
|
int numVaryingElements,
|
|
int *V_ITa, int offset, int tableOffset, int start, int end);
|
|
|
|
void OsdCudaEditVertexAdd(float *vertex, int numUserVertexElements,
|
|
int primVarOffset, int primVarWidth,
|
|
int vertexOffset, int tableOffset,
|
|
int start, int end, int *editIndices, float *editValues);
|
|
|
|
}
|
|
|
|
namespace OpenSubdiv {
|
|
namespace OPENSUBDIV_VERSION {
|
|
|
|
OsdCudaComputeController::OsdCudaComputeController() {
|
|
}
|
|
|
|
OsdCudaComputeController::~OsdCudaComputeController() {
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyBilinearFaceVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * F_IT = context->GetTable(FarSubdivisionTables::F_IT);
|
|
const OsdCudaTable * F_ITa = context->GetTable(FarSubdivisionTables::F_ITa);
|
|
assert(F_IT);
|
|
assert(F_ITa);
|
|
|
|
OsdCudaComputeFace(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(F_IT->GetCudaMemory()),
|
|
static_cast<int*>(F_ITa->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyBilinearEdgeVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * E_IT = context->GetTable(FarSubdivisionTables::E_IT);
|
|
assert(E_IT);
|
|
|
|
OsdCudaComputeBilinearEdge(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(E_IT->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyBilinearVertexVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
assert(V_ITa);
|
|
|
|
OsdCudaComputeBilinearVertex(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyCatmarkFaceVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * F_IT = context->GetTable(FarSubdivisionTables::F_IT);
|
|
const OsdCudaTable * F_ITa = context->GetTable(FarSubdivisionTables::F_ITa);
|
|
assert(F_IT);
|
|
assert(F_ITa);
|
|
|
|
OsdCudaComputeFace(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(F_IT->GetCudaMemory()),
|
|
static_cast<int*>(F_ITa->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyCatmarkEdgeVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * E_IT = context->GetTable(FarSubdivisionTables::E_IT);
|
|
const OsdCudaTable * E_W = context->GetTable(FarSubdivisionTables::E_W);
|
|
assert(E_IT);
|
|
assert(E_W);
|
|
|
|
OsdCudaComputeEdge(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(E_IT->GetCudaMemory()),
|
|
static_cast<float*>(E_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyCatmarkVertexVerticesKernelB(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_IT = context->GetTable(FarSubdivisionTables::V_IT);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_IT);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeVertexB(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<int*>(V_IT->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyCatmarkVertexVerticesKernelA1(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeVertexA(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd(), false);
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyCatmarkVertexVerticesKernelA2(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeVertexA(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd(), true);
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyLoopEdgeVerticesKernel(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * E_IT = context->GetTable(FarSubdivisionTables::E_IT);
|
|
const OsdCudaTable * E_W = context->GetTable(FarSubdivisionTables::E_W);
|
|
assert(E_IT);
|
|
assert(E_W);
|
|
|
|
OsdCudaComputeEdge(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(E_IT->GetCudaMemory()),
|
|
static_cast<float*>(E_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyLoopVertexVerticesKernelB(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_IT = context->GetTable(FarSubdivisionTables::V_IT);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_IT);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeLoopVertexB(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<int*>(V_IT->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd());
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyLoopVertexVerticesKernelA1(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeVertexA(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd(), false);
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyLoopVertexVerticesKernelA2(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaTable * V_ITa = context->GetTable(FarSubdivisionTables::V_ITa);
|
|
const OsdCudaTable * V_W = context->GetTable(FarSubdivisionTables::V_W);
|
|
assert(V_ITa);
|
|
assert(V_W);
|
|
|
|
OsdCudaComputeVertexA(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetCurrentVaryingBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
context->GetVertexDescriptor().numVaryingElements,
|
|
static_cast<int*>(V_ITa->GetCudaMemory()),
|
|
static_cast<float*>(V_W->GetCudaMemory()),
|
|
batch.GetVertexOffset(), batch.GetTableOffset(), batch.GetStart(), batch.GetEnd(), true);
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::ApplyVertexEdits(
|
|
FarKernelBatch const &batch, OsdCudaComputeContext *context) const {
|
|
|
|
assert(context);
|
|
|
|
const OsdCudaHEditTable *edit = context->GetEditTable(batch.GetTableIndex());
|
|
assert(edit);
|
|
|
|
const OsdCudaTable * primvarIndices = edit->GetPrimvarIndices();
|
|
const OsdCudaTable * editValues = edit->GetEditValues();
|
|
|
|
if (edit->GetOperation() == FarVertexEdit::Add) {
|
|
OsdCudaEditVertexAdd(
|
|
context->GetCurrentVertexBuffer(),
|
|
context->GetVertexDescriptor().numVertexElements-3,
|
|
edit->GetPrimvarOffset(),
|
|
edit->GetPrimvarWidth(),
|
|
batch.GetVertexOffset(),
|
|
batch.GetTableOffset(),
|
|
batch.GetStart(),
|
|
batch.GetEnd(),
|
|
static_cast<int*>(primvarIndices->GetCudaMemory()),
|
|
static_cast<float*>(editValues->GetCudaMemory()));
|
|
} else if (edit->GetOperation() == FarVertexEdit::Set) {
|
|
// XXXX TODO
|
|
}
|
|
}
|
|
|
|
void
|
|
OsdCudaComputeController::Synchronize() {
|
|
|
|
cudaThreadSynchronize();
|
|
}
|
|
|
|
} // end namespace OPENSUBDIV_VERSION
|
|
} // end namespace OpenSubdiv
|