src/graphics/examples/vkproto/common/image_view.cc

// Copyright 2018 The Fuchsia 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 "image_view.h"

#include "src/graphics/examples/vkproto/common/utils.h"

namespace vkp {

ImageView::ImageView(std::shared_ptr<vk::Device> device, const vk::PhysicalDevice &physical_device,
                     const vk::Extent2D &extent)
    : initialized_(false), device_(device), physical_device_(physical_device), extent_(extent) {}

bool ImageView::Init() {
  RTN_IF_MSG(false, initialized_, "ImageView is already initialized.\n");
  RTN_IF_MSG(false, !device_, "Device must be initialized.\n");

  format_ = vk::Format::eB8G8R8A8Unorm;

  // Create vk::Image.
  vk::ImageCreateInfo image_info;
  image_info.extent = vk::Extent3D(extent_.width, extent_.height, 1);
  image_info.arrayLayers = 1;
  image_info.format = format_;
  image_info.imageType = vk::ImageType::e2D;
  image_info.initialLayout = vk::ImageLayout::eUndefined;
  image_info.tiling = vk::ImageTiling::eOptimal;
  image_info.mipLevels = 1;
  image_info.samples = vk::SampleCountFlagBits::e1;
  image_info.sharingMode = vk::SharingMode::eExclusive;
  image_info.usage =
      vk::ImageUsageFlagBits::eColorAttachment | vk::ImageUsageFlagBits::eTransferSrc;
  auto [r_image, image] = device_->createImageUnique(image_info);
  RTN_IF_VKH_ERR(false, r_image, "Failed to create image.\n");
  image_ = std::move(image);

  // Allocate memory for |image_| and bind it.
  auto image_memory_requirements = device_->getImageMemoryRequirements(*image_);
  vk::MemoryAllocateInfo alloc_info;
  alloc_info.allocationSize = image_memory_requirements.size;
  RTN_IF_VKH_ERR(
      false,
      FindMemoryIndex(physical_device_, image_memory_requirements.memoryTypeBits,
                      vk::MemoryPropertyFlagBits::eDeviceLocal, &alloc_info.memoryTypeIndex),
      "Failed to find matching memory index.");
  auto [r_image_memory, image_memory] = device_->allocateMemoryUnique(alloc_info);
  RTN_IF_VKH_ERR(false, r_image_memory, "Failed to allocate device memory for image.\n");
  image_memory_ = std::move(image_memory);

  RTN_IF_VKH_ERR(false, device_->bindImageMemory(*image_, *image_memory_, 0),
                 "Failed to bind device memory to image.\n");

  // Create vk::ImageView on |image_|.
  vk::ImageSubresourceRange range;
  range.aspectMask = vk::ImageAspectFlagBits::eColor;
  range.layerCount = 1;
  range.levelCount = 1;

  vk::ImageViewCreateInfo view_info;
  view_info.format = format_;
  view_info.subresourceRange = range;
  view_info.viewType = vk::ImageViewType::e2D;
  view_info.image = *image_;
  auto [r_image_view, image_view] = device_->createImageViewUnique(view_info);
  RTN_IF_VKH_ERR(false, r_image_view, "Failed to create image view.\n");
  image_view_ = std::move(image_view);

  initialized_ = true;

  return initialized_;
}

bool ImageView::Init(vk::UniqueImage image, vk::UniqueDeviceMemory image_memory,
                     vk::Format format) {
  RTN_IF_MSG(false, initialized_, "ImageView is already initialized.\n");
  RTN_IF_MSG(false, !device_, "Device must be initialized.\n");
  format_ = format;
  image_ = std::move(image);
  image_memory_ = std::move(image_memory);
  // Create vk::ImageView on |image_|.
  vk::ImageSubresourceRange range;
  range.aspectMask = vk::ImageAspectFlagBits::eColor;
  range.layerCount = 1;
  range.levelCount = 1;

  vk::ImageViewCreateInfo view_info;
  view_info.format = format_;
  view_info.subresourceRange = range;
  view_info.viewType = vk::ImageViewType::e2D;
  view_info.image = *image_;
  auto [r_image_view, image_view] = device_->createImageViewUnique(view_info);
  RTN_IF_VKH_ERR(false, r_image_view, "Failed to create image view.\n");
  image_view_ = std::move(image_view);

  initialized_ = true;

  return initialized_;
}

}  // namespace vkp