compute_pipeline_descriptor.cc

Go to the documentation of this file.

// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#include "impeller/renderer/compute_pipeline_descriptor.h"
 
#include "impeller/core/formats.h"
#include "impeller/renderer/shader_function.h"
#include "impeller/renderer/shader_library.h"
#include "impeller/renderer/vertex_descriptor.h"
 
namespace impeller {
 
ComputePipelineDescriptor::ComputePipelineDescriptor() = default;
 
ComputePipelineDescriptor::~ComputePipelineDescriptor() = default;
 
// Comparable<ComputePipelineDescriptor>
std::size_t ComputePipelineDescriptor::GetHash() const {
  auto seed = fml::HashCombine();
  fml::HashCombineSeed(seed, label_);
  if (entrypoint_) {
    fml::HashCombineSeed(seed, entrypoint_->GetHash());
  }
  fml::HashCombineSeed(seed, workgroup_size_[0], workgroup_size_[1],
                       workgroup_size_[2]);
  return seed;
}
 
// Comparable<ComputePipelineDescriptor>
bool ComputePipelineDescriptor::IsEqual(
    const ComputePipelineDescriptor& other) const {
  return label_ == other.label_ &&
         DeepComparePointer(entrypoint_, other.entrypoint_) &&
         workgroup_size_ == other.workgroup_size_;
}
 
ComputePipelineDescriptor& ComputePipelineDescriptor::SetLabel(
    std::string_view label) {
  label_ = label;
  return *this;
}
 
ComputePipelineDescriptor& ComputePipelineDescriptor::SetStageEntrypoint(
    std::shared_ptr<const ShaderFunction> function) {
  FML_DCHECK(!function || function->GetStage() == ShaderStage::kCompute);
  if (!function || function->GetStage() != ShaderStage::kCompute) {
    return *this;
  }
 
  if (function->GetStage() == ShaderStage::kUnknown) {
    return *this;
  }
 
  entrypoint_ = std::move(function);
 
  return *this;
}
 
std::shared_ptr<const ShaderFunction>
ComputePipelineDescriptor::GetStageEntrypoint() const {
  return entrypoint_;
}
 
ComputePipelineDescriptor& ComputePipelineDescriptor::SetWorkgroupSize(
    std::array<uint32_t, 3> size) {
  workgroup_size_ = size;
  return *this;
}
 
std::array<uint32_t, 3> ComputePipelineDescriptor::GetWorkgroupSize() const {
  return workgroup_size_;
}
 
const std::string& ComputePipelineDescriptor::GetLabel() const {
  return label_;
}
 
bool ComputePipelineDescriptor::RegisterDescriptorSetLayouts(
    const DescriptorSetLayout desc_set_layout[],
    size_t count) {
  descriptor_set_layouts_.reserve(descriptor_set_layouts_.size() + count);
  for (size_t i = 0; i < count; i++) {
    descriptor_set_layouts_.emplace_back(desc_set_layout[i]);
  }
  return true;
}
 
const std::vector<DescriptorSetLayout>&
ComputePipelineDescriptor::GetDescriptorSetLayouts() const {
  return descriptor_set_layouts_;
}
 
}  // namespace impeller