Powei Feng
6f2c45c76d
- Push constants is a small set of bytes that can be recorded directly on the command buffer. - Implemented it for the vulkan/gl backend.
88 lines
3.2 KiB
C++
88 lines
3.2 KiB
C++
/*
|
|
* Copyright (C) 2024 The Android Open Source Project
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include "VulkanPipelineLayoutCache.h"
|
|
|
|
#include <vulkan/VulkanResourceAllocator.h>
|
|
|
|
namespace filament::backend {
|
|
|
|
VkPipelineLayout VulkanPipelineLayoutCache::getLayout(
|
|
VulkanDescriptorSetLayoutList const& descriptorSetLayouts, VulkanProgram* program) {
|
|
PipelineLayoutKey key = {};
|
|
uint8_t descSetLayoutCount = 0;
|
|
for (auto layoutHandle: descriptorSetLayouts) {
|
|
if (layoutHandle) {
|
|
auto layout = mAllocator->handle_cast<VulkanDescriptorSetLayout*>(layoutHandle);
|
|
key.descSetLayouts[descSetLayoutCount++] = layout->vklayout;
|
|
}
|
|
}
|
|
|
|
// build the push constant layout key
|
|
uint32_t pushConstantRangeCount = program->getPushConstantRangeCount();
|
|
auto const& pushCostantRanges = program->getPushConstantRanges();
|
|
if (pushConstantRangeCount > 0) {
|
|
assert_invariant(pushConstantRangeCount <= Program::SHADER_TYPE_COUNT);
|
|
for (uint8_t i = 0; i < pushConstantRangeCount; ++i) {
|
|
auto const& range = pushCostantRanges[i];
|
|
auto& pushConstant = key.pushConstant[i];
|
|
if (range.stageFlags & VK_SHADER_STAGE_VERTEX_BIT) {
|
|
pushConstant.stage = static_cast<uint8_t>(ShaderStage::VERTEX);
|
|
}
|
|
if (range.stageFlags & VK_SHADER_STAGE_FRAGMENT_BIT) {
|
|
pushConstant.stage = static_cast<uint8_t>(ShaderStage::FRAGMENT);
|
|
}
|
|
if (range.stageFlags & VK_SHADER_STAGE_COMPUTE_BIT) {
|
|
pushConstant.stage = static_cast<uint8_t>(ShaderStage::COMPUTE);
|
|
}
|
|
pushConstant.size = range.size;
|
|
}
|
|
}
|
|
|
|
auto iter = mPipelineLayouts.find(key);
|
|
if (iter != mPipelineLayouts.end()) {
|
|
PipelineLayoutCacheEntry& entry = mPipelineLayouts[key];
|
|
entry.lastUsed = mTimestamp++;
|
|
return entry.handle;
|
|
}
|
|
|
|
VkPipelineLayoutCreateInfo info{
|
|
.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO,
|
|
.pNext = nullptr,
|
|
.setLayoutCount = (uint32_t) descSetLayoutCount,
|
|
.pSetLayouts = key.descSetLayouts.data(),
|
|
.pushConstantRangeCount = pushConstantRangeCount,
|
|
.pPushConstantRanges = pushCostantRanges,
|
|
};
|
|
|
|
VkPipelineLayout layout;
|
|
vkCreatePipelineLayout(mDevice, &info, VKALLOC, &layout);
|
|
|
|
mPipelineLayouts[key] = {
|
|
.handle = layout,
|
|
.lastUsed = mTimestamp++,
|
|
};
|
|
return layout;
|
|
}
|
|
|
|
void VulkanPipelineLayoutCache::terminate() noexcept {
|
|
for (auto const& [key, entry]: mPipelineLayouts) {
|
|
vkDestroyPipelineLayout(mDevice, entry.handle, VKALLOC);
|
|
}
|
|
}
|
|
|
|
}// namespace filament::backend
|