OpenSubdiv/opensubdiv/osd/d3d11DrawContext.h

//
//   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.
//

#pragma once
#ifndef OPENSUBDIV3_OSD_D3D11L_DRAW_CONTEXT_H
#define OPENSUBDIV3_OSD_D3D11L_DRAW_CONTEXT_H

#include "../version.h"

#include "../far/patchTables.h"
#include "../osd/drawContext.h"

#include <map>

struct ID3D11VertexShader;
struct ID3D11HullShader;
struct ID3D11DomainShader;
struct ID3D11GeometryShader;
struct ID3D11PixelShader;

struct ID3D11Buffer;
struct ID3D11ShaderResourceView;
struct ID3D11Device;
struct ID3D11DeviceContext;

namespace OpenSubdiv {
namespace OPENSUBDIV_VERSION {

namespace Osd {

/// \brief D3D11 specialized DrawContext class
///
/// D3D11DrawContext implements the OSD drawing interface with Microsoft(c)
/// DirectX D3D11 API.
/// Some functionality may be disabled depending on compile and run-time driver
/// support.
///
/// Contexts interface the serialized topological data pertaining to the
/// geometric primitives with the capabilities of the selected discrete
/// compute device.
///
class D3D11DrawContext : public DrawContext {
public:
    typedef ID3D11Buffer * VertexBufferBinding;

    virtual ~D3D11DrawContext();

    /// \brief Create an D3D11DrawContext from Far::PatchTables
    ///
    /// @param patchTables          A valid set of Far::PatchTables
    ///
    /// @param pd3d11DeviceContext  A device context
    ///
    static D3D11DrawContext *Create(Far::PatchTables const *patchTables,
                                    ID3D11DeviceContext *pd3d11DeviceContext);

    /// template version for custom context (OpenCL) used by OsdMesh
    template<typename DEVICE_CONTEXT>
    static D3D11DrawContext *Create(Far::PatchTables const *patchtables,
                                    DEVICE_CONTEXT context) {
        return Create(patchtables, context->GetDeviceContext());
    }

    /// Set vbo as a vertex texture (for gregory patch drawing)
    ///
    /// @param vbo                  The vertex buffer object to update
    ///
    /// @param pd3d11DeviceContext  A device context
    ///
    template<class VERTEX_BUFFER>
    void UpdateVertexTexture(VERTEX_BUFFER *vbo, ID3D11DeviceContext *pd3d11DeviceContext) {
        if (vbo)
            updateVertexTexture(vbo->BindD3D11Buffer(pd3d11DeviceContext),
                                pd3d11DeviceContext,
                                vbo->GetNumVertices(),
                                vbo->GetNumElements());
    }

    /// template version for custom context (OpenCL) used by OsdMesh
    template<class VERTEX_BUFFER, class DEVICE_CONTEXT>
    void UpdateVertexTexture(VERTEX_BUFFER *vbo, DEVICE_CONTEXT context) {
        UpdateVertexTexture(vbo, context->GetDeviceContext());
    }

    ID3D11Buffer             *patchIndexBuffer;

    ID3D11Buffer             *patchParamBuffer;
    ID3D11ShaderResourceView *patchParamBufferSRV;

    ID3D11Buffer             *fvarDataBuffer;
    ID3D11ShaderResourceView *fvarDataBufferSRV;

    ID3D11ShaderResourceView *vertexBufferSRV;
    ID3D11Buffer             *vertexValenceBuffer;
    ID3D11ShaderResourceView *vertexValenceBufferSRV;
    ID3D11Buffer             *quadOffsetBuffer;
    ID3D11ShaderResourceView *quadOffsetBufferSRV;

    /// Sets face-varying data buffer
    ///
    /// @param patchTables          A valid set of Far::PatchTables
    ///
    /// @param pd3d11DeviceContext  A device context
    ///
    /// @param fvarWidth            Total face-varying primvar data width in fvarData
    ///
    /// @param fvarData             Vector containing the face-varying data
    ///
    /// @return                     True if the operation was successful
    ///
    bool SetFVarDataTexture(Far::PatchTables const & patchTables,
                            int fvarWidth, FVarData const & fvarData,
                            ID3D11DeviceContext *pd3d11DeviceContext);

    /// template version for custom context (OpenCL) used by OsdMesh
    template<typename DEVICE_CONTEXT>
    bool SetFVarDataTexture(Far::PatchTables const & patchTables,
                            int fvarWidth, FVarData const & fvarData,
                            DEVICE_CONTEXT context) {
        return SetFVarDataTexture(patchTables, fvarWidth, fvarData,
                                  context->GetDeviceContext());
    }

private:
    D3D11DrawContext(int maxValence);


    // allocate buffers from patchTables
    bool create(Far::PatchTables const &patchTables,
                ID3D11DeviceContext *pd3d11DeviceContext);

    void updateVertexTexture(ID3D11Buffer *vbo,
                             ID3D11DeviceContext *pd3d11DeviceContext,
                             int numVertices,
                             int numVertexElements);

    int _numVertices;
};

}  // end namespace Osd

}  // end namespace OPENSUBDIV_VERSION
using namespace OPENSUBDIV_VERSION;

} // end namespace OpenSubdiv

#endif /* OPENSUBDIV3_OSD_D3D11L_DRAW_CONTEXT_H */