Vulkan: Add several unittests for VulkanImage

Bug: 964030
Change-Id: I26d38fa7bfbeefb3101922681263fb1414346fe0
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2136104
Commit-Queue: Jonathan Backer <backer@chromium.org>
Auto-Submit: Peng Huang <penghuang@chromium.org>
Reviewed-by: Jonathan Backer <backer@chromium.org>
Cr-Commit-Position: refs/heads/master@{#757174}

gpu/vulkan/BUILD.gn

@@ -166,6 +166,7 @@ if (enable_vulkan) {
         "tests/vulkan_test.cc",
         "tests/vulkan_tests_main.cc",
         "vulkan_fence_helper_unittest.cc",
+        "vulkan_image_unittest.cc",
       ]
 
       deps = [
@@ -179,6 +180,7 @@ if (enable_vulkan) {
         "//ui/events/platform",
         "//ui/gfx",
         "//ui/gfx:native_widget_types",
+        "//ui/gfx:test_support",
         "//ui/gfx/geometry",
       ]
 

gpu/vulkan/vulkan_image_unittest.cc

+// Copyright 2020 The Chromium 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 "gpu/vulkan/vulkan_image.h"
+
+#include "build/build_config.h"
+#include "gpu/vulkan/tests/basic_vulkan_test.h"
+#include "gpu/vulkan/vulkan_device_queue.h"
+#include "gpu/vulkan/vulkan_function_pointers.h"
+#include "ui/gfx/geometry/rect.h"
+
+namespace gpu {
+
+namespace {
+
+// TODO(penghuang): add more formats used by chrome.
+const VkFormat kFormats[] = {
+    VK_FORMAT_R8G8B8A8_UNORM,
+    VK_FORMAT_B8G8R8A8_UNORM,
+};
+
+}  // namespace
+
+using VulkanImageTest = BasicVulkanTest;
+
+TEST_F(VulkanImageTest, Create) {
+  constexpr gfx::Size size(100, 100);
+  constexpr VkImageUsageFlags usage =
+      VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT |
+      VK_IMAGE_USAGE_SAMPLED_BIT | VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
+  auto* device_queue = GetDeviceQueue();
+  for (auto format : kFormats) {
+    auto image = VulkanImage::Create(device_queue, size, format, usage);
+    EXPECT_TRUE(image);
+    EXPECT_EQ(image->size(), size);
+    EXPECT_EQ(image->format(), format);
+    EXPECT_GT(image->device_size(), 0u);
+    EXPECT_EQ(image->image_tiling(), VK_IMAGE_TILING_OPTIMAL);
+    EXPECT_EQ(image->image_layout(), VK_IMAGE_LAYOUT_UNDEFINED);
+    EXPECT_NE(image->image(), static_cast<VkImage>(VK_NULL_HANDLE));
+    EXPECT_NE(image->device_memory(),
+              static_cast<VkDeviceMemory>(VK_NULL_HANDLE));
+    EXPECT_EQ(image->handle_types(), 0u);
+    image->Destroy();
+  }
+}
+
+TEST_F(VulkanImageTest, CreateWithExternalMemory) {
+  constexpr gfx::Size size(100, 100);
+  constexpr VkImageUsageFlags usage =
+      VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT |
+      VK_IMAGE_USAGE_SAMPLED_BIT | VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
+  auto* device_queue = GetDeviceQueue();
+  for (auto format : kFormats) {
+    auto image = VulkanImage::CreateWithExternalMemory(device_queue, size,
+                                                       format, usage);
+    EXPECT_TRUE(image);
+    EXPECT_EQ(image->size(), size);
+    EXPECT_EQ(image->format(), format);
+    EXPECT_GT(image->device_size(), 0u);
+    EXPECT_EQ(image->image_tiling(), VK_IMAGE_TILING_OPTIMAL);
+    EXPECT_EQ(image->image_layout(), VK_IMAGE_LAYOUT_UNDEFINED);
+    EXPECT_NE(image->image(), static_cast<VkImage>(VK_NULL_HANDLE));
+    EXPECT_NE(image->device_memory(),
+              static_cast<VkDeviceMemory>(VK_NULL_HANDLE));
+
+#if defined(OS_POSIX)
+    EXPECT_TRUE(image->handle_types() &
+                VK_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD_BIT)
+        << std::hex << "handle_types = 0x" << image->handle_types();
+    const VkExternalMemoryHandleTypeFlagBits kHandleTypes[] = {
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD_BIT,
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_DMA_BUF_BIT_EXT,
+    };
+    // Get fd for all supported types.
+    for (auto handle_type : kHandleTypes) {
+      if ((image->handle_types() & handle_type) == 0)
+        continue;
+      base::ScopedFD scoped_fd = image->GetMemoryFd(handle_type);
+      EXPECT_TRUE(scoped_fd.is_valid())
+          << std::hex << " handle_types = 0x" << image->handle_types()
+          << " handle_type = 0x" << handle_type;
+    }
+#elif defined(OS_WIN)
+    EXPECT_TRUE(image->handle_types() &
+                VK_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_BIT)
+        << std::hex << "handle_types = 0x" << image->handle_types();
+    const VkExternalMemoryHandleTypeFlagBits kHandleTypes[] = {
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_BIT,
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_TEXTURE_BIT,
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP_BIT,
+        VK_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE_BIT,
+    };
+    // Get fd for all supported types.
+    for (auto handle_type : kHandleTypes) {
+      if ((image->handle_types() & handle_type) == 0)
+        continue;
+      base::win::ScopedHandle scoped_handle = image->GetMemoryHandle(
+          static_cast<VkExternalMemoryHandleTypeFlagBits>(handle_type));
+      EXPECT_TRUE(scoped_handle.IsValid())
+          << std::hex << " handle_types = 0x" << image->handle_types()
+          << " handle_type = 0x" << handle_type;
+    }
+#elif defined(OS_FUCHSIA)
+    EXPECT_TRUE(image->handle_types() &
+                VK_EXTERNAL_MEMORY_HANDLE_TYPE_TEMP_ZIRCON_VMO_BIT_FUCHSIA);
+    zx::vmo handle = image->GetMemoryZirconHandle();
+    EXPECT_TRUE(handle);
+#endif
+
+    image->Destroy();
+  }
+}
+
+// TODO(penghuang): Add test for creating from GpuMemoryBufferHandle
+
+}  // namespace gpu