d3d11ComputeEvaluator.h

Go to the documentation of this file.

//
//   Copyright 2015 Pixar
//
//   Licensed under the terms set forth in the LICENSE.txt file available at
//   https://opensubdiv.org/license.
//
 
#ifndef OPENSUBDIV3_OSD_D3D11_COMPUTE_EVALUATOR_H
#define OPENSUBDIV3_OSD_D3D11_COMPUTE_EVALUATOR_H
 
#include "../version.h"
 
struct ID3D11DeviceContext;
struct ID3D11Buffer;
struct ID3D11ComputeShader;
struct ID3D11ClassLinkage;
struct ID3D11ClassInstance;
struct ID3D11ShaderResourceView;
struct ID3D11UnorderedAccessView;
 
#include "../osd/bufferDescriptor.h"
 
namespace OpenSubdiv {
namespace OPENSUBDIV_VERSION {
 
namespace Far {
    class StencilTable;
}
 
namespace Osd {
 
class D3D11StencilTable {
public:
    template <typename DEVICE_CONTEXT>
    static D3D11StencilTable *Create(Far::StencilTable const *stencilTable,
                                      DEVICE_CONTEXT context) {
        return new D3D11StencilTable(stencilTable, context->GetDeviceContext());
    }
 
    static D3D11StencilTable *Create(Far::StencilTable const *stencilTable,
                                      ID3D11DeviceContext *deviceContext) {
        return new D3D11StencilTable(stencilTable, deviceContext);
    }
 
    D3D11StencilTable(Far::StencilTable const *stencilTable,
                       ID3D11DeviceContext *deviceContext);
 
    ~D3D11StencilTable();
 
    // interfaces needed for D3D11ComputeEvaluator
    ID3D11ShaderResourceView *GetSizesSRV() const { return _sizes; }
    ID3D11ShaderResourceView *GetOffsetsSRV() const { return _offsets; }
    ID3D11ShaderResourceView *GetIndicesSRV() const { return _indices; }
    ID3D11ShaderResourceView *GetWeightsSRV() const { return _weights; }
    int GetNumStencils() const { return _numStencils; }
 
private:
    ID3D11ShaderResourceView *_sizes;
    ID3D11ShaderResourceView *_offsets;
    ID3D11ShaderResourceView *_indices;
    ID3D11ShaderResourceView *_weights;
    ID3D11Buffer *_sizesBuffer;
    ID3D11Buffer *_offsetsBuffer;
    ID3D11Buffer *_indicesBuffer;
    ID3D11Buffer *_weightsBuffer;
 
    int _numStencils;
};
 
// ---------------------------------------------------------------------------
 
class D3D11ComputeEvaluator {
public:
    using Instantiatable = bool;    // Enable compiled kernel evaluator cache
 
    static D3D11ComputeEvaluator * Create(BufferDescriptor const &srcDesc,
                                          BufferDescriptor const &dstDesc,
                                          BufferDescriptor const &duDesc,
                                          BufferDescriptor const &dvDesc,
                                          ID3D11DeviceContext *deviceContext);
 
    static D3D11ComputeEvaluator * Create(BufferDescriptor const &srcDesc,
                                          BufferDescriptor const &dstDesc,
                                          BufferDescriptor const &duDesc,
                                          BufferDescriptor const &dvDesc,
                                          BufferDescriptor const &duuDesc,
                                          BufferDescriptor const &duvDesc,
                                          BufferDescriptor const &dvvDesc,
                                          ID3D11DeviceContext *deviceContext);
 
    D3D11ComputeEvaluator();
 
    ~D3D11ComputeEvaluator();
 
    template <typename SRC_BUFFER, typename DST_BUFFER, typename STENCIL_TABLE>
    static bool EvalStencils(
        SRC_BUFFER *srcBuffer, BufferDescriptor const &srcDesc,
        DST_BUFFER *dstBuffer, BufferDescriptor const &dstDesc,
        STENCIL_TABLE const *stencilTable,
        D3D11ComputeEvaluator const *instance,
        ID3D11DeviceContext * deviceContext) {
        if (instance) {
            return instance->EvalStencils(srcBuffer, srcDesc,
                                          dstBuffer, dstDesc,
                                          stencilTable,
                                          deviceContext);
        } else {
            // Create an instance on demand (slow)
            (void)deviceContext;  // unused
            instance = Create(srcDesc, dstDesc,
                              BufferDescriptor(),
                              BufferDescriptor(),
                              deviceContext);
            if (instance) {
                bool r = instance->EvalStencils(srcBuffer, srcDesc,
                                                dstBuffer, dstDesc,
                                                stencilTable,
                                                deviceContext);
                delete instance;
                return r;
            }
            return false;
        }
    }
 
    template <typename SRC_BUFFER, typename DST_BUFFER, typename STENCIL_TABLE>
    bool EvalStencils(
        SRC_BUFFER *srcBuffer, BufferDescriptor const &srcDesc,
        DST_BUFFER *dstBuffer, BufferDescriptor const &dstDesc,
        STENCIL_TABLE const *stencilTable,
        ID3D11DeviceContext *deviceContext) const {
        return EvalStencils(srcBuffer->BindD3D11UAV(deviceContext), srcDesc,
                            dstBuffer->BindD3D11UAV(deviceContext), dstDesc,
                            stencilTable->GetSizesSRV(),
                            stencilTable->GetOffsetsSRV(),
                            stencilTable->GetIndicesSRV(),
                            stencilTable->GetWeightsSRV(),
                            /* start = */ 0,
                            /* end   = */ stencilTable->GetNumStencils(),
                            deviceContext);
    }
 
    bool EvalStencils(ID3D11UnorderedAccessView *srcSRV,
                      BufferDescriptor const &srcDesc,
                      ID3D11UnorderedAccessView *dstUAV,
                      BufferDescriptor const &dstDesc,
                      ID3D11ShaderResourceView *sizesSRV,
                      ID3D11ShaderResourceView *offsetsSRV,
                      ID3D11ShaderResourceView *indicesSRV,
                      ID3D11ShaderResourceView *weightsSRV,
                      int start,
                      int end,
                      ID3D11DeviceContext *deviceContext) const;
 
    bool Compile(BufferDescriptor const &srcDesc,
                 BufferDescriptor const &dstDesc,
                 ID3D11DeviceContext *deviceContext);
 
    static void Synchronize(ID3D11DeviceContext *deviceContext);
 
private:
    ID3D11ComputeShader * _computeShader;
    ID3D11ClassLinkage  * _classLinkage;
    ID3D11ClassInstance * _singleBufferKernel;
    ID3D11ClassInstance * _separateBufferKernel;
    ID3D11Buffer        * _uniformArgs; // uniform parameters for kernels
 
    int _workGroupSize;
};
 
}  // end namespace Osd
 
}  // end namespace OPENSUBDIV_VERSION
using namespace OPENSUBDIV_VERSION;
 
}  // end namespace OpenSubdiv
 
 
#endif  // OPENSUBDIV3_OSD_D3D11_COMPUTE_EVALUATOR_H