vk_scheduler.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "BLI_vector.hh"
 
#include "vk_common.hh"
#include "vk_render_graph_node.hh"
 
namespace blender::gpu::render_graph {
class VKRenderGraph;

class VKScheduler {
 private:
  Vector<NodeHandle> result_;
 
 public:
  [[nodiscard]] Span<NodeHandle> select_nodes_for_image(const VKRenderGraph &render_graph,
                                                        VkImage vk_image);

  [[nodiscard]] Span<NodeHandle> select_nodes_for_buffer(const VKRenderGraph &render_graph,
                                                         VkBuffer vk_buffer);
  [[nodiscard]] Span<NodeHandle> select_nodes(const VKRenderGraph &render_graph);
 
 private:
  void select_all_nodes(const VKRenderGraph &render_graph);
 
  void reorder_nodes(const VKRenderGraph &render_graph);

  void move_transfer_and_dispatch_outside_rendering_scope(const VKRenderGraph &render_graph);

  std::optional<std::pair<int64_t, int64_t>> find_rendering_scope(
      const VKRenderGraph &render_graph, IndexRange search_range) const;
  template<typename FuncT>
  void foreach_rendering_scope(const VKRenderGraph &render_graph, const FuncT &func) const
  {
    for (std::optional<std::pair<int64_t, int64_t>> rendering_scope =
             find_rendering_scope(render_graph, IndexRange(result_.index_range()));
         rendering_scope.has_value();
         rendering_scope =
             find_rendering_scope(render_graph,
                                  IndexRange(rendering_scope.value().second + 1,
                                             result_.size() - rendering_scope.value().second - 1)))
    {
      func(rendering_scope.value().first, rendering_scope.value().second);
    }
  }
 
  void debug_print(const VKRenderGraph &render_graph) const;
};
 
}  // namespace blender::gpu::render_graph