mirror of
https://github.com/PixarAnimationStudios/OpenSubdiv
synced 2024-11-27 05:50:05 +00:00
3ae50d1c50
New text: 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.
197 lines
5.5 KiB
C++
197 lines
5.5 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/tbbEvalStencilsController.h"
|
|
|
|
#include <tbb/parallel_for.h>
|
|
#include <tbb/task_scheduler_init.h>
|
|
|
|
#include <cassert>
|
|
|
|
namespace OpenSubdiv {
|
|
namespace OPENSUBDIV_VERSION {
|
|
|
|
#define grain_size 200
|
|
|
|
OsdTbbEvalStencilsController::OsdTbbEvalStencilsController(int numThreads) {
|
|
|
|
_numThreads = numThreads > 0 ? numThreads : tbb::task_scheduler_init::automatic;
|
|
|
|
tbb::task_scheduler_init init(numThreads);
|
|
}
|
|
|
|
OsdTbbEvalStencilsController::~OsdTbbEvalStencilsController() {
|
|
}
|
|
|
|
|
|
class StencilKernel {
|
|
|
|
public:
|
|
enum Mode { UNDEFINED, POINT, U_DERIV, V_DERIV };
|
|
|
|
StencilKernel( FarStencilTables const * stencils,
|
|
OsdVertexBufferDescriptor ctrlDesc,
|
|
float const * ctrlData ) :
|
|
_stencils(stencils),
|
|
_mode(UNDEFINED),
|
|
_ctrlDesc(ctrlDesc),
|
|
_length(0),
|
|
_outStride(0),
|
|
_outData(0) {
|
|
|
|
_ctrlData = ctrlData + ctrlDesc.offset;
|
|
}
|
|
|
|
bool SetOutput(Mode mode, OsdVertexBufferDescriptor outDesc, float * outData) {
|
|
|
|
if (_ctrlDesc.CanEval(outDesc)) {
|
|
_mode = mode;
|
|
_length = outDesc.length;
|
|
_outStride = outDesc.stride;
|
|
_outData = outData + outDesc.offset;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void operator() (tbb::blocked_range<int> const &r) const {
|
|
|
|
assert(_stencils and _ctrlData and _length and _outStride and _outData);
|
|
|
|
int offset = _stencils->GetOffsets()[r.begin()];
|
|
|
|
int const * sizes = &_stencils->GetSizes()[r.begin()],
|
|
* index = &_stencils->GetControlIndices()[offset];
|
|
|
|
float const * weight;
|
|
|
|
switch (_mode) {
|
|
case POINT : weight = &_stencils->GetWeights()[offset]; break;
|
|
case U_DERIV : weight = &_stencils->GetDuWeights()[offset]; break;
|
|
case V_DERIV : weight = &_stencils->GetDvWeights()[offset]; break;
|
|
default:
|
|
return;
|
|
}
|
|
assert( weight);
|
|
|
|
float * out = _outData + r.begin() * _outStride;
|
|
|
|
for (int i=r.begin(); i<r.end(); ++i, ++sizes) {
|
|
|
|
memset( out, 0, _length * sizeof(float) );
|
|
|
|
for (int j=0; j<(*sizes); ++j, ++index, ++weight) {
|
|
|
|
float const * cv = _ctrlData + (*index)*_ctrlDesc.stride;
|
|
|
|
for (int k=0; k<_length; ++k) {
|
|
out[k] += cv[k] * (*weight);
|
|
}
|
|
}
|
|
out+=_outStride;
|
|
}
|
|
}
|
|
|
|
private:
|
|
FarStencilTables const * _stencils;
|
|
|
|
Mode _mode;
|
|
|
|
OsdVertexBufferDescriptor _ctrlDesc;
|
|
float const * _ctrlData;
|
|
|
|
int _length,
|
|
_outStride;
|
|
|
|
float * _outData;
|
|
};
|
|
|
|
int
|
|
OsdTbbEvalStencilsController::_UpdateValues( OsdCpuEvalStencilsContext * context ) {
|
|
|
|
FarStencilTables const * stencils = context->GetStencilTables();
|
|
if (not stencils)
|
|
return 0;
|
|
|
|
int nstencils = stencils->GetNumStencils();
|
|
if (not nstencils)
|
|
return 0;
|
|
|
|
StencilKernel kernel( stencils, context->GetControlDataDescriptor(),
|
|
context->GetControlData() );
|
|
|
|
|
|
if (not kernel.SetOutput( StencilKernel::POINT,
|
|
context->GetOutputDataDescriptor(),
|
|
context->GetOutputData() ))
|
|
return 0;
|
|
|
|
tbb::blocked_range<int> range(0, nstencils, grain_size);
|
|
|
|
tbb::parallel_for(range, kernel);
|
|
|
|
return nstencils;
|
|
}
|
|
|
|
int
|
|
OsdTbbEvalStencilsController::_UpdateDerivs( OsdCpuEvalStencilsContext * context ) {
|
|
|
|
FarStencilTables const * stencils = context->GetStencilTables();
|
|
if (not stencils)
|
|
return 0;
|
|
|
|
int nstencils = stencils->GetNumStencils();
|
|
if (not nstencils)
|
|
return 0;
|
|
|
|
tbb::blocked_range<int> range(0, nstencils, grain_size);
|
|
|
|
StencilKernel kernel( stencils, context->GetControlDataDescriptor(),
|
|
context->GetControlData() );
|
|
|
|
if (not kernel.SetOutput( StencilKernel::U_DERIV,
|
|
context->GetDuDataDescriptor(),
|
|
context->GetOutputUDerivData() ) )
|
|
return 0;
|
|
|
|
tbb::parallel_for(range, kernel);
|
|
|
|
if (not kernel.SetOutput( StencilKernel::V_DERIV,
|
|
context->GetDvDataDescriptor(),
|
|
context->GetOutputVDerivData() ) )
|
|
return 0;
|
|
|
|
tbb::parallel_for(range, kernel);
|
|
|
|
return nstencils;
|
|
}
|
|
|
|
void
|
|
OsdTbbEvalStencilsController::Synchronize() {
|
|
}
|
|
|
|
|
|
} // end namespace OPENSUBDIV_VERSION
|
|
} // end namespace OpenSubdiv
|