tracked_objects_vk.h

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.
 
#ifndef FLUTTER_IMPELLER_RENDERER_BACKEND_VULKAN_TRACKED_OBJECTS_VK_H_
#define FLUTTER_IMPELLER_RENDERER_BACKEND_VULKAN_TRACKED_OBJECTS_VK_H_
 
#include <memory>
 
#include "impeller/renderer/backend/vulkan/command_encoder_vk.h"
#include "impeller/renderer/backend/vulkan/context_vk.h"
 
namespace impeller {
 
/// @brief A per-frame object used to track resource lifetimes and allocate
///        command buffers and descriptor sets.
class TrackedObjectsVK {
 public:
  explicit TrackedObjectsVK(const std::weak_ptr<const ContextVK>& context,
                            const std::shared_ptr<CommandPoolVK>& pool,
                            std::unique_ptr<GPUProbe> probe);
 
  ~TrackedObjectsVK();
 
  bool IsValid() const;
 
  void Track(std::shared_ptr<SharedObjectVK> object);
 
  void Track(std::shared_ptr<const DeviceBuffer> buffer);
 
  bool IsTracking(const std::shared_ptr<const DeviceBuffer>& buffer) const;
 
  void Track(std::shared_ptr<const TextureSourceVK> texture);
 
  bool IsTracking(const std::shared_ptr<const TextureSourceVK>& texture) const;
 
  vk::CommandBuffer GetCommandBuffer() const;
 
  DescriptorPoolVK& GetDescriptorPool();
 
  GPUProbe& GetGPUProbe() const;
 
 private:
  DescriptorPoolVK desc_pool_;
  // `shared_ptr` since command buffers have a link to the command pool.
  std::shared_ptr<CommandPoolVK> pool_;
  vk::UniqueCommandBuffer buffer_;
  std::set<std::shared_ptr<SharedObjectVK>> tracked_objects_;
  std::set<std::shared_ptr<const DeviceBuffer>> tracked_buffers_;
  std::set<std::shared_ptr<const TextureSourceVK>> tracked_textures_;
  std::unique_ptr<GPUProbe> probe_;
  bool is_valid_ = false;
 
  TrackedObjectsVK(const TrackedObjectsVK&) = delete;
 
  TrackedObjectsVK& operator=(const TrackedObjectsVK&) = delete;
};
 
}  // namespace impeller
 
#endif  // FLUTTER_IMPELLER_RENDERER_BACKEND_VULKAN_TRACKED_OBJECTS_VK_H_