mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-05-14 19:31:25 +00:00
30beb65a7b
Previously didn't tell SPIRV-Cross at which MSL resource indices the different SPIRV binding should be, and were lucky that it's giving indices in increasing order matched the PipelineLayout in all our samples. Fix this by making SPIRV->MSL compilation depend on the PipelineLayout so we can tell SPIRV-Cross which binding goes where. We should do the same for vertex attributes eventually as they are hardcoded to start at kMaxBindingsPerGroup currently. Also a couple unrelated cleanups (unused function, usage of IterateBitSet).
97 lines
3.0 KiB
C++
97 lines
3.0 KiB
C++
// Copyright 2017 The NXT Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef BACKEND_SHADERMODULE_H_
|
|
#define BACKEND_SHADERMODULE_H_
|
|
|
|
#include "backend/Forward.h"
|
|
#include "backend/Builder.h"
|
|
#include "backend/RefCounted.h"
|
|
#include "common/Constants.h"
|
|
|
|
#include "nxt/nxtcpp.h"
|
|
|
|
#include <array>
|
|
#include <bitset>
|
|
#include <vector>
|
|
|
|
namespace spirv_cross {
|
|
class Compiler;
|
|
}
|
|
|
|
namespace backend {
|
|
|
|
class ShaderModuleBase : public RefCounted {
|
|
public:
|
|
ShaderModuleBase(ShaderModuleBuilder* builder);
|
|
|
|
DeviceBase* GetDevice() const;
|
|
|
|
void ExtractSpirvInfo(const spirv_cross::Compiler& compiler);
|
|
|
|
struct PushConstantInfo {
|
|
std::bitset<kMaxPushConstants> mask;
|
|
|
|
std::array<std::string, kMaxPushConstants> names;
|
|
std::array<uint32_t, kMaxPushConstants> sizes;
|
|
std::array<PushConstantType, kMaxPushConstants> types;
|
|
};
|
|
|
|
struct BindingInfo {
|
|
// The SPIRV ID of the resource.
|
|
uint32_t id;
|
|
uint32_t base_type_id;
|
|
nxt::BindingType type;
|
|
bool used = false;
|
|
};
|
|
using ModuleBindingInfo = std::array<std::array<BindingInfo, kMaxBindingsPerGroup>, kMaxBindGroups>;
|
|
|
|
const PushConstantInfo& GetPushConstants() const;
|
|
const ModuleBindingInfo& GetBindingInfo() const;
|
|
const std::bitset<kMaxVertexAttributes>& GetUsedVertexAttributes() const;
|
|
nxt::ShaderStage GetExecutionModel() const;
|
|
|
|
bool IsCompatibleWithPipelineLayout(const PipelineLayoutBase* layout);
|
|
|
|
private:
|
|
bool IsCompatibleWithBindGroupLayout(size_t group, const BindGroupLayoutBase* layout);
|
|
|
|
DeviceBase* device;
|
|
PushConstantInfo pushConstants = {};
|
|
ModuleBindingInfo bindingInfo;
|
|
std::bitset<kMaxVertexAttributes> usedVertexAttributes;
|
|
nxt::ShaderStage executionModel;
|
|
};
|
|
|
|
class ShaderModuleBuilder : public Builder<ShaderModuleBase> {
|
|
public:
|
|
ShaderModuleBuilder(DeviceBase* device);
|
|
|
|
std::vector<uint32_t> AcquireSpirv();
|
|
|
|
// NXT API
|
|
void SetSource(uint32_t codeSize, const uint32_t* code);
|
|
|
|
private:
|
|
friend class ShaderModuleBase;
|
|
|
|
ShaderModuleBase* GetResultImpl() override;
|
|
|
|
std::vector<uint32_t> spirv;
|
|
};
|
|
|
|
}
|
|
|
|
#endif // BACKEND_SHADERMODULE_H_
|