impeller/renderer/backend/vulkan/swapchain/khr/khr_surface_vk.cc

// 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/backend/vulkan/swapchain/khr/khr_surface_vk.h"

#include "impeller/core/formats.h"
#include "impeller/renderer/backend/vulkan/swapchain/khr/khr_swapchain_image_vk.h"
#include "impeller/renderer/backend/vulkan/texture_vk.h"
#include "impeller/renderer/surface.h"

namespace impeller {

std::unique_ptr<KHRSurfaceVK> KHRSurfaceVK::WrapSwapchainImage(
    const std::shared_ptr<Context>& context,
    std::shared_ptr<KHRSwapchainImageVK>& swapchain_image,
    SwapCallback swap_callback,
    bool enable_msaa) {
  if (!context || !swapchain_image || !swap_callback) {
    return nullptr;
  }

  std::shared_ptr<Texture> msaa_tex;
  if (enable_msaa) {
    TextureDescriptor msaa_tex_desc;
    msaa_tex_desc.storage_mode = StorageMode::kDeviceTransient;
    msaa_tex_desc.type = TextureType::kTexture2DMultisample;
    msaa_tex_desc.sample_count = SampleCount::kCount4;
    msaa_tex_desc.format = swapchain_image->GetPixelFormat();
    msaa_tex_desc.size = swapchain_image->GetSize();
    msaa_tex_desc.usage = TextureUsage::kRenderTarget;

    if (!swapchain_image->GetMSAATexture()) {
      msaa_tex = context->GetResourceAllocator()->CreateTexture(msaa_tex_desc);
      msaa_tex->SetLabel("ImpellerOnscreenColorMSAA");
      if (!msaa_tex) {
        VALIDATION_LOG << "Could not allocate MSAA color texture.";
        return nullptr;
      }
    } else {
      msaa_tex = swapchain_image->GetMSAATexture();
    }
  }

  TextureDescriptor resolve_tex_desc;
  resolve_tex_desc.type = TextureType::kTexture2D;
  resolve_tex_desc.format = swapchain_image->GetPixelFormat();
  resolve_tex_desc.size = swapchain_image->GetSize();
  resolve_tex_desc.usage = TextureUsage::kRenderTarget;
  resolve_tex_desc.sample_count = SampleCount::kCount1;
  resolve_tex_desc.storage_mode = StorageMode::kDevicePrivate;

  std::shared_ptr<Texture> resolve_tex =
      std::make_shared<TextureVK>(context,         //
                                  swapchain_image  //
      );

  if (!resolve_tex) {
    VALIDATION_LOG << "Could not wrap resolve texture.";
    return nullptr;
  }
  resolve_tex->SetLabel("ImpellerOnscreenResolve");

  ColorAttachment color0;
  color0.clear_color = Color::DarkSlateGray();
  color0.load_action = LoadAction::kClear;
  if (enable_msaa) {
    color0.texture = msaa_tex;
    color0.store_action = StoreAction::kMultisampleResolve;
    color0.resolve_texture = resolve_tex;
  } else {
    color0.texture = resolve_tex;
    color0.store_action = StoreAction::kStore;
  }

  RenderTarget render_target_desc;
  render_target_desc.SetColorAttachment(color0, 0u);
  render_target_desc.SetupDepthStencilAttachments(
      /*context=*/*context,                            //
      /*allocator=*/*context->GetResourceAllocator(),  //
      /*size=*/swapchain_image->GetSize(),             //
      /*msaa=*/enable_msaa,                            //
      /*label=*/"Onscreen",                            //
      /*stencil_attachment_config=*/
      RenderTarget::kDefaultStencilAttachmentConfig,                       //
      /*depth_stencil_texture=*/swapchain_image->GetDepthStencilTexture()  //
  );

  // The constructor is private. So make_unique may not be used.
  return std::unique_ptr<KHRSurfaceVK>(
      new KHRSurfaceVK(render_target_desc, std::move(swap_callback)));
}

KHRSurfaceVK::KHRSurfaceVK(const RenderTarget& target,
                           SwapCallback swap_callback)
    : Surface(target), swap_callback_(std::move(swap_callback)) {}

KHRSurfaceVK::~KHRSurfaceVK() = default;

bool KHRSurfaceVK::Present() const {
  return swap_callback_ ? swap_callback_() : false;
}

}  // namespace impeller