612b256821
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
81 lines
2.6 KiB
C++
81 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <mbgl/shaders/shader_program_base.hpp>
|
|
#include <mbgl/mtl/mtl_fwd.hpp>
|
|
#include <mbgl/mtl/uniform_block.hpp>
|
|
#include <mbgl/mtl/vertex_attribute.hpp>
|
|
|
|
#include <Foundation/NSSharedPtr.hpp>
|
|
|
|
#include <optional>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
namespace mbgl {
|
|
namespace shaders {
|
|
struct AttributeInfo {
|
|
AttributeInfo(std::size_t index, gfx::AttributeDataType dataType, std::string_view name);
|
|
std::size_t index;
|
|
gfx::AttributeDataType dataType;
|
|
std::string_view name;
|
|
StringIdentity nameID;
|
|
};
|
|
struct UniformBlockInfo {
|
|
UniformBlockInfo(std::size_t index, bool vertex, bool fragment, std::size_t size, std::size_t id);
|
|
std::size_t index;
|
|
bool vertex;
|
|
bool fragment;
|
|
std::size_t size;
|
|
std::size_t id;
|
|
};
|
|
struct TextureInfo {
|
|
TextureInfo(std::size_t index, std::size_t id);
|
|
std::size_t index;
|
|
std::size_t id;
|
|
};
|
|
} // namespace shaders
|
|
namespace mtl {
|
|
class RenderableResource;
|
|
class RendererBackend;
|
|
class ShaderProgram;
|
|
using UniqueShaderProgram = std::unique_ptr<ShaderProgram>;
|
|
|
|
class ShaderProgram final : public gfx::ShaderProgramBase {
|
|
public:
|
|
ShaderProgram(std::string name,
|
|
RendererBackend& backend,
|
|
MTLFunctionPtr vertexFunction,
|
|
MTLFunctionPtr fragmentFunction);
|
|
~ShaderProgram() noexcept override;
|
|
|
|
static constexpr std::string_view Name{"GenericMTLShader"};
|
|
const std::string_view typeName() const noexcept override { return Name; }
|
|
|
|
MTLRenderPipelineStatePtr getRenderPipelineState(const gfx::Renderable&,
|
|
const MTLVertexDescriptorPtr&,
|
|
const gfx::ColorMode& colorMode) const;
|
|
|
|
std::optional<size_t> getSamplerLocation(const size_t id) const override;
|
|
|
|
const gfx::VertexAttributeArray& getVertexAttributes() const override { return vertexAttributes; }
|
|
|
|
const gfx::UniformBlockArray& getUniformBlocks() const override { return uniformBlocks; }
|
|
gfx::UniformBlockArray& mutableUniformBlocks() override { return uniformBlocks; }
|
|
|
|
void initAttribute(const shaders::AttributeInfo&);
|
|
void initUniformBlock(const shaders::UniformBlockInfo&);
|
|
void initTexture(const shaders::TextureInfo&);
|
|
|
|
protected:
|
|
std::string shaderName;
|
|
RendererBackend& backend;
|
|
MTLFunctionPtr vertexFunction;
|
|
MTLFunctionPtr fragmentFunction;
|
|
UniformBlockArray uniformBlocks;
|
|
VertexAttributeArray vertexAttributes;
|
|
std::array<std::optional<size_t>, shaders::maxTextureCountPerShader> textureBindings;
|
|
};
|
|
|
|
} // namespace mtl
|
|
} // namespace mbgl
|