blob: 5c2af9524daea27112b87aa3eeb3e304e890c59f [file] [log] [blame] [edit]
// Copyright 2020 The SwiftShader Authors. All Rights Reserved.
//
// 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 "VkDescriptorSet.hpp"
#include "VkDevice.hpp"
#include "VkImageView.hpp"
#include "VkPipelineLayout.hpp"
namespace vk {
void DescriptorSet::ParseDescriptors(const Array &descriptorSets, const PipelineLayout *layout, Device *device, NotificationType notificationType)
{
if(layout)
{
uint32_t descriptorSetCount = layout->getDescriptorSetCount();
ASSERT(descriptorSetCount <= MAX_BOUND_DESCRIPTOR_SETS);
for(uint32_t i = 0; i < descriptorSetCount; ++i)
{
DescriptorSet *descriptorSet = descriptorSets[i];
if(!descriptorSet)
{
continue;
}
marl::lock lock(descriptorSet->header.mutex);
uint32_t bindingCount = layout->getBindingCount(i);
for(uint32_t j = 0; j < bindingCount; ++j)
{
VkDescriptorType type = layout->getDescriptorType(i, j);
uint32_t descriptorCount = layout->getDescriptorCount(i, j);
uint32_t descriptorSize = layout->getDescriptorSize(i, j);
uint8_t *descriptorMemory = descriptorSet->data + layout->getBindingOffset(i, j);
for(uint32_t k = 0; k < descriptorCount; k++)
{
ImageView *memoryOwner = nullptr;
switch(type)
{
case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
memoryOwner = reinterpret_cast<SampledImageDescriptor *>(descriptorMemory)->memoryOwner;
break;
case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
memoryOwner = reinterpret_cast<StorageImageDescriptor *>(descriptorMemory)->memoryOwner;
break;
default:
break;
}
if(memoryOwner)
{
if(notificationType == PREPARE_FOR_SAMPLING)
{
device->prepareForSampling(memoryOwner);
}
else if((notificationType == CONTENTS_CHANGED) && (type == VK_DESCRIPTOR_TYPE_STORAGE_IMAGE))
{
device->contentsChanged(memoryOwner, Image::USING_STORAGE);
}
}
descriptorMemory += descriptorSize;
}
}
}
}
}
void DescriptorSet::ContentsChanged(const Array &descriptorSets, const PipelineLayout *layout, Device *device)
{
ParseDescriptors(descriptorSets, layout, device, CONTENTS_CHANGED);
}
void DescriptorSet::PrepareForSampling(const Array &descriptorSets, const PipelineLayout *layout, Device *device)
{
ParseDescriptors(descriptorSets, layout, device, PREPARE_FOR_SAMPLING);
}
} // namespace vk