blob: 855232badd8707d6647ec1fafe3faca48502d41d [file] [log] [blame]
Greg Daniel164a9f02016-02-22 09:56:40 -05001/*
2 * Copyright 2015 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#include "GrVkGpu.h"
9
Greg Daniela5cb7812017-06-16 09:45:32 -040010#include "GrBackendSemaphore.h"
Greg Daniel7ef28f32017-04-20 16:41:55 +000011#include "GrBackendSurface.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050012#include "GrContextOptions.h"
13#include "GrGeometryProcessor.h"
14#include "GrGpuResourceCacheAccess.h"
egdaniel0e1853c2016-03-17 11:35:45 -070015#include "GrMesh.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050016#include "GrPipeline.h"
17#include "GrRenderTargetPriv.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050018#include "GrTexturePriv.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050019
20#include "GrVkCommandBuffer.h"
egdaniel066df7c2016-06-08 14:02:27 -070021#include "GrVkGpuCommandBuffer.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050022#include "GrVkImage.h"
23#include "GrVkIndexBuffer.h"
24#include "GrVkMemory.h"
25#include "GrVkPipeline.h"
egdaniel22281c12016-03-23 13:49:40 -070026#include "GrVkPipelineState.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050027#include "GrVkRenderPass.h"
28#include "GrVkResourceProvider.h"
Greg Daniel6be35232017-03-01 17:01:09 -050029#include "GrVkSemaphore.h"
Greg Danielc2dd5ed2017-05-05 13:49:11 -040030#include "GrVkTexelBuffer.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050031#include "GrVkTexture.h"
32#include "GrVkTextureRenderTarget.h"
33#include "GrVkTransferBuffer.h"
34#include "GrVkVertexBuffer.h"
35
Matt Sarett485c4992017-02-14 14:18:27 -050036#include "SkConvertPixels.h"
jvanverth900bd4a2016-04-29 13:53:12 -070037#include "SkMipMap.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050038
39#include "vk/GrVkInterface.h"
jvanverthfd359ca2016-03-18 11:57:24 -070040#include "vk/GrVkTypes.h"
Greg Daniel164a9f02016-02-22 09:56:40 -050041
ethannicholasb3058bd2016-07-01 08:22:01 -070042#include "SkSLCompiler.h"
ethannicholasb3058bd2016-07-01 08:22:01 -070043
Forrest Reiling44f85712017-03-27 23:22:20 -070044#if !defined(SK_BUILD_FOR_WIN)
45#include <unistd.h>
46#endif // !defined(SK_BUILD_FOR_WIN)
47
Greg Daniel164a9f02016-02-22 09:56:40 -050048#define VK_CALL(X) GR_VK_CALL(this->vkInterface(), X)
49#define VK_CALL_RET(RET, X) GR_VK_CALL_RET(this->vkInterface(), RET, X)
50#define VK_CALL_ERRCHECK(X) GR_VK_CALL_ERRCHECK(this->vkInterface(), X)
51
egdaniel735109c2016-07-27 08:03:57 -070052#ifdef SK_ENABLE_VK_LAYERS
jvanverthd2497f32016-03-18 12:39:05 -070053VKAPI_ATTR VkBool32 VKAPI_CALL DebugReportCallback(
54 VkDebugReportFlagsEXT flags,
55 VkDebugReportObjectTypeEXT objectType,
56 uint64_t object,
57 size_t location,
58 int32_t messageCode,
59 const char* pLayerPrefix,
60 const char* pMessage,
61 void* pUserData) {
62 if (flags & VK_DEBUG_REPORT_ERROR_BIT_EXT) {
63 SkDebugf("Vulkan error [%s]: code: %d: %s\n", pLayerPrefix, messageCode, pMessage);
Jim Van Verthce3fe232016-11-11 09:41:04 -080064 return VK_TRUE; // skip further layers
jvanverthd2497f32016-03-18 12:39:05 -070065 } else if (flags & VK_DEBUG_REPORT_WARNING_BIT_EXT) {
66 SkDebugf("Vulkan warning [%s]: code: %d: %s\n", pLayerPrefix, messageCode, pMessage);
67 } else if (flags & VK_DEBUG_REPORT_PERFORMANCE_WARNING_BIT_EXT) {
68 SkDebugf("Vulkan perf warning [%s]: code: %d: %s\n", pLayerPrefix, messageCode, pMessage);
69 } else {
70 SkDebugf("Vulkan info/debug [%s]: code: %d: %s\n", pLayerPrefix, messageCode, pMessage);
71 }
72 return VK_FALSE;
73}
jvanverthd2497f32016-03-18 12:39:05 -070074#endif
75
Brian Salomon384fab42017-12-07 12:33:05 -050076sk_sp<GrGpu> GrVkGpu::Make(sk_sp<const GrVkBackendContext> backendContext,
77 const GrContextOptions& options, GrContext* context) {
Greg Daniel02611d92017-07-25 10:05:01 -040078 if (!backendContext) {
Brian Salomonc1889822017-04-24 16:49:05 -040079 return nullptr;
Greg Daniel164a9f02016-02-22 09:56:40 -050080 }
81
Greg Daniel02611d92017-07-25 10:05:01 -040082 if (!backendContext->fInterface->validate(backendContext->fExtensions)) {
Greg Danielfe2965a2016-10-11 16:31:05 -040083 return nullptr;
84 }
85
Brian Salomon384fab42017-12-07 12:33:05 -050086 return sk_sp<GrGpu>(new GrVkGpu(context, options, std::move(backendContext)));
Greg Daniel164a9f02016-02-22 09:56:40 -050087}
88
89////////////////////////////////////////////////////////////////////////////////
90
halcanary9d524f22016-03-29 09:03:52 -070091GrVkGpu::GrVkGpu(GrContext* context, const GrContextOptions& options,
Brian Salomon384fab42017-12-07 12:33:05 -050092 sk_sp<const GrVkBackendContext> backendCtx)
93 : INHERITED(context)
94 , fBackendContext(std::move(backendCtx))
95 , fDevice(fBackendContext->fDevice)
96 , fQueue(fBackendContext->fQueue)
97 , fResourceProvider(this)
98 , fDisconnected(false) {
egdaniel735109c2016-07-27 08:03:57 -070099#ifdef SK_ENABLE_VK_LAYERS
brianosman419ca642016-05-04 08:19:44 -0700100 fCallback = VK_NULL_HANDLE;
Brian Salomon384fab42017-12-07 12:33:05 -0500101 if (fBackendContext->fExtensions & kEXT_debug_report_GrVkExtensionFlag) {
jvanverthfd7bd452016-03-25 06:29:52 -0700102 // Setup callback creation information
jvanverthd2497f32016-03-18 12:39:05 -0700103 VkDebugReportCallbackCreateInfoEXT callbackCreateInfo;
104 callbackCreateInfo.sType = VK_STRUCTURE_TYPE_DEBUG_REPORT_CREATE_INFO_EXT;
105 callbackCreateInfo.pNext = nullptr;
106 callbackCreateInfo.flags = VK_DEBUG_REPORT_ERROR_BIT_EXT |
egdanielef0c10c2016-04-07 07:51:22 -0700107 VK_DEBUG_REPORT_WARNING_BIT_EXT |
jvanverthd2497f32016-03-18 12:39:05 -0700108 //VK_DEBUG_REPORT_INFORMATION_BIT_EXT |
109 //VK_DEBUG_REPORT_DEBUG_BIT_EXT |
egdanielb4aa3622016-04-06 13:47:08 -0700110 VK_DEBUG_REPORT_PERFORMANCE_WARNING_BIT_EXT;
jvanverthd2497f32016-03-18 12:39:05 -0700111 callbackCreateInfo.pfnCallback = &DebugReportCallback;
112 callbackCreateInfo.pUserData = nullptr;
113
jvanverthfd7bd452016-03-25 06:29:52 -0700114 // Register the callback
Brian Salomon384fab42017-12-07 12:33:05 -0500115 GR_VK_CALL_ERRCHECK(this->vkInterface(),
116 CreateDebugReportCallbackEXT(fBackendContext->fInstance,
117 &callbackCreateInfo, nullptr, &fCallback));
jvanverthd2497f32016-03-18 12:39:05 -0700118 }
119#endif
jvanverth633b3562016-03-23 11:01:22 -0700120
ethannicholasb3058bd2016-07-01 08:22:01 -0700121 fCompiler = new SkSL::Compiler();
jvanverth633b3562016-03-23 11:01:22 -0700122
Brian Salomon384fab42017-12-07 12:33:05 -0500123 fVkCaps.reset(new GrVkCaps(options, this->vkInterface(), fBackendContext->fPhysicalDevice,
124 fBackendContext->fFeatures, fBackendContext->fExtensions));
jvanverth633b3562016-03-23 11:01:22 -0700125 fCaps.reset(SkRef(fVkCaps.get()));
126
Greg Daniel8385a8a2018-02-26 13:29:37 -0500127 VK_CALL(GetPhysicalDeviceProperties(fBackendContext->fPhysicalDevice, &fPhysDevProps));
Brian Salomon384fab42017-12-07 12:33:05 -0500128 VK_CALL(GetPhysicalDeviceMemoryProperties(fBackendContext->fPhysicalDevice, &fPhysDevMemProps));
jvanverth633b3562016-03-23 11:01:22 -0700129
130 const VkCommandPoolCreateInfo cmdPoolInfo = {
jvanverth7ec92412016-07-06 09:24:57 -0700131 VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO, // sType
132 nullptr, // pNext
133 VK_COMMAND_POOL_CREATE_TRANSIENT_BIT |
134 VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT, // CmdPoolCreateFlags
Brian Salomon384fab42017-12-07 12:33:05 -0500135 fBackendContext->fGraphicsQueueIndex, // queueFamilyIndex
jvanverth633b3562016-03-23 11:01:22 -0700136 };
halcanary9d524f22016-03-29 09:03:52 -0700137 GR_VK_CALL_ERRCHECK(this->vkInterface(), CreateCommandPool(fDevice, &cmdPoolInfo, nullptr,
jvanverth633b3562016-03-23 11:01:22 -0700138 &fCmdPool));
139
140 // must call this after creating the CommandPool
141 fResourceProvider.init();
jvanverth7ec92412016-07-06 09:24:57 -0700142 fCurrentCmdBuffer = fResourceProvider.findOrCreatePrimaryCommandBuffer();
jvanverth633b3562016-03-23 11:01:22 -0700143 SkASSERT(fCurrentCmdBuffer);
144 fCurrentCmdBuffer->begin(this);
jvanverth6b6ffc42016-06-13 14:28:07 -0700145
146 // set up our heaps
147 fHeaps[kLinearImage_Heap].reset(new GrVkHeap(this, GrVkHeap::kSubAlloc_Strategy, 16*1024*1024));
Greg Daniel4f7d97c2017-04-26 15:10:00 -0400148 fHeaps[kOptimalImage_Heap].reset(new GrVkHeap(this, GrVkHeap::kSubAlloc_Strategy, 64*1024*1024));
jvanverth6b6ffc42016-06-13 14:28:07 -0700149 fHeaps[kSmallOptimalImage_Heap].reset(new GrVkHeap(this, GrVkHeap::kSubAlloc_Strategy, 2*1024*1024));
150 fHeaps[kVertexBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSingleAlloc_Strategy, 0));
151 fHeaps[kIndexBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSingleAlloc_Strategy, 0));
jvanverth4c6e47a2016-07-22 10:34:52 -0700152 fHeaps[kUniformBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSubAlloc_Strategy, 256*1024));
Greg Danielc2dd5ed2017-05-05 13:49:11 -0400153 fHeaps[kTexelBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSingleAlloc_Strategy, 0));
jvanverth6b6ffc42016-06-13 14:28:07 -0700154 fHeaps[kCopyReadBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSingleAlloc_Strategy, 0));
155 fHeaps[kCopyWriteBuffer_Heap].reset(new GrVkHeap(this, GrVkHeap::kSubAlloc_Strategy, 16*1024*1024));
Greg Daniel164a9f02016-02-22 09:56:40 -0500156}
157
Greg Daniel8606cf82017-05-08 16:17:53 -0400158void GrVkGpu::destroyResources() {
159 if (fCurrentCmdBuffer) {
160 fCurrentCmdBuffer->end(this);
161 fCurrentCmdBuffer->unref(this);
162 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500163
164 // wait for all commands to finish
jvanverthddf98352016-03-21 11:46:00 -0700165 fResourceProvider.checkCommandBuffers();
Jim Van Verth09557d72016-11-07 11:10:21 -0500166 VkResult res = VK_CALL(QueueWaitIdle(fQueue));
egdanielf8c2be32016-06-24 13:18:27 -0700167
168 // On windows, sometimes calls to QueueWaitIdle return before actually signalling the fences
169 // on the command buffers even though they have completed. This causes an assert to fire when
170 // destroying the command buffers. Currently this ony seems to happen on windows, so we add a
Jim Van Verth09557d72016-11-07 11:10:21 -0500171 // sleep to make sure the fence signals.
egdanielf8c2be32016-06-24 13:18:27 -0700172#ifdef SK_DEBUG
Greg Daniel80a08dd2017-01-20 10:45:49 -0500173 if (this->vkCaps().mustSleepOnTearDown()) {
egdanielf8c2be32016-06-24 13:18:27 -0700174#if defined(SK_BUILD_FOR_WIN)
Greg Daniel80a08dd2017-01-20 10:45:49 -0500175 Sleep(10); // In milliseconds
egdanielf8c2be32016-06-24 13:18:27 -0700176#else
Greg Daniel80a08dd2017-01-20 10:45:49 -0500177 sleep(1); // In seconds
egdanielf8c2be32016-06-24 13:18:27 -0700178#endif
Greg Daniel80a08dd2017-01-20 10:45:49 -0500179 }
egdanielf8c2be32016-06-24 13:18:27 -0700180#endif
181
egdanielbe9d8212016-09-20 08:54:23 -0700182#ifdef SK_DEBUG
Greg Daniel8a8668b2016-10-31 16:34:42 -0400183 SkASSERT(VK_SUCCESS == res || VK_ERROR_DEVICE_LOST == res);
egdanielbe9d8212016-09-20 08:54:23 -0700184#endif
halcanary9d524f22016-03-29 09:03:52 -0700185
Greg Daniel6be35232017-03-01 17:01:09 -0500186 for (int i = 0; i < fSemaphoresToWaitOn.count(); ++i) {
187 fSemaphoresToWaitOn[i]->unref(this);
188 }
189 fSemaphoresToWaitOn.reset();
190
Greg Daniela5cb7812017-06-16 09:45:32 -0400191 for (int i = 0; i < fSemaphoresToSignal.count(); ++i) {
192 fSemaphoresToSignal[i]->unref(this);
193 }
194 fSemaphoresToSignal.reset();
195
196
egdanielbc9b2962016-09-27 08:00:53 -0700197 fCopyManager.destroyResources(this);
198
Jim Van Verth09557d72016-11-07 11:10:21 -0500199 // must call this just before we destroy the command pool and VkDevice
200 fResourceProvider.destroyResources(VK_ERROR_DEVICE_LOST == res);
Greg Daniel164a9f02016-02-22 09:56:40 -0500201
Greg Daniel8606cf82017-05-08 16:17:53 -0400202 if (fCmdPool != VK_NULL_HANDLE) {
203 VK_CALL(DestroyCommandPool(fDevice, fCmdPool, nullptr));
204 }
jvanverth633b3562016-03-23 11:01:22 -0700205
egdaniel735109c2016-07-27 08:03:57 -0700206#ifdef SK_ENABLE_VK_LAYERS
jvanvertha00980e2016-05-02 13:24:48 -0700207 if (fCallback) {
208 VK_CALL(DestroyDebugReportCallbackEXT(fBackendContext->fInstance, fCallback, nullptr));
jvanvertha00980e2016-05-02 13:24:48 -0700209 }
jvanverthd2497f32016-03-18 12:39:05 -0700210#endif
Greg Daniel8606cf82017-05-08 16:17:53 -0400211
212}
213
214GrVkGpu::~GrVkGpu() {
215 if (!fDisconnected) {
216 this->destroyResources();
217 }
218 delete fCompiler;
219}
220
221
222void GrVkGpu::disconnect(DisconnectType type) {
223 INHERITED::disconnect(type);
224 if (!fDisconnected) {
225 if (DisconnectType::kCleanup == type) {
226 this->destroyResources();
227 } else {
228 fCurrentCmdBuffer->unrefAndAbandon();
229 for (int i = 0; i < fSemaphoresToWaitOn.count(); ++i) {
230 fSemaphoresToWaitOn[i]->unrefAndAbandon();
231 }
Greg Daniela5cb7812017-06-16 09:45:32 -0400232 for (int i = 0; i < fSemaphoresToSignal.count(); ++i) {
233 fSemaphoresToSignal[i]->unrefAndAbandon();
234 }
Greg Daniel8606cf82017-05-08 16:17:53 -0400235 fCopyManager.abandonResources();
236
237 // must call this just before we destroy the command pool and VkDevice
238 fResourceProvider.abandonResources();
239 }
240 fSemaphoresToWaitOn.reset();
Greg Daniela5cb7812017-06-16 09:45:32 -0400241 fSemaphoresToSignal.reset();
Greg Daniel8606cf82017-05-08 16:17:53 -0400242#ifdef SK_ENABLE_VK_LAYERS
243 fCallback = VK_NULL_HANDLE;
244#endif
245 fCurrentCmdBuffer = nullptr;
246 fCmdPool = VK_NULL_HANDLE;
247 fDisconnected = true;
248 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500249}
250
251///////////////////////////////////////////////////////////////////////////////
252
Greg Daniel500d58b2017-08-24 15:59:33 -0400253GrGpuRTCommandBuffer* GrVkGpu::createCommandBuffer(
Robert Phillips95214472017-08-08 18:00:03 -0400254 GrRenderTarget* rt, GrSurfaceOrigin origin,
Greg Daniel500d58b2017-08-24 15:59:33 -0400255 const GrGpuRTCommandBuffer::LoadAndStoreInfo& colorInfo,
256 const GrGpuRTCommandBuffer::StencilLoadAndStoreInfo& stencilInfo) {
257 return new GrVkGpuRTCommandBuffer(this, rt, origin, colorInfo, stencilInfo);
258}
259
260GrGpuTextureCommandBuffer* GrVkGpu::createCommandBuffer(GrTexture* texture,
261 GrSurfaceOrigin origin) {
262 return new GrVkGpuTextureCommandBuffer(this, texture, origin);
egdaniel066df7c2016-06-08 14:02:27 -0700263}
264
Greg Daniela5cb7812017-06-16 09:45:32 -0400265void GrVkGpu::submitCommandBuffer(SyncQueue sync) {
Greg Daniel164a9f02016-02-22 09:56:40 -0500266 SkASSERT(fCurrentCmdBuffer);
267 fCurrentCmdBuffer->end(this);
268
Greg Daniela5cb7812017-06-16 09:45:32 -0400269 fCurrentCmdBuffer->submitToQueue(this, fQueue, sync, fSemaphoresToSignal, fSemaphoresToWaitOn);
Greg Daniel6be35232017-03-01 17:01:09 -0500270
271 for (int i = 0; i < fSemaphoresToWaitOn.count(); ++i) {
272 fSemaphoresToWaitOn[i]->unref(this);
273 }
274 fSemaphoresToWaitOn.reset();
Greg Daniela5cb7812017-06-16 09:45:32 -0400275 for (int i = 0; i < fSemaphoresToSignal.count(); ++i) {
276 fSemaphoresToSignal[i]->unref(this);
277 }
278 fSemaphoresToSignal.reset();
Greg Daniel6be35232017-03-01 17:01:09 -0500279
Greg Daniel164a9f02016-02-22 09:56:40 -0500280 fResourceProvider.checkCommandBuffers();
281
282 // Release old command buffer and create a new one
283 fCurrentCmdBuffer->unref(this);
jvanverth7ec92412016-07-06 09:24:57 -0700284 fCurrentCmdBuffer = fResourceProvider.findOrCreatePrimaryCommandBuffer();
Greg Daniel164a9f02016-02-22 09:56:40 -0500285 SkASSERT(fCurrentCmdBuffer);
286
287 fCurrentCmdBuffer->begin(this);
288}
289
290///////////////////////////////////////////////////////////////////////////////
cdalton1bf3e712016-04-19 10:00:02 -0700291GrBuffer* GrVkGpu::onCreateBuffer(size_t size, GrBufferType type, GrAccessPattern accessPattern,
292 const void* data) {
293 GrBuffer* buff;
cdalton397536c2016-03-25 12:15:03 -0700294 switch (type) {
295 case kVertex_GrBufferType:
296 SkASSERT(kDynamic_GrAccessPattern == accessPattern ||
297 kStatic_GrAccessPattern == accessPattern);
cdalton1bf3e712016-04-19 10:00:02 -0700298 buff = GrVkVertexBuffer::Create(this, size, kDynamic_GrAccessPattern == accessPattern);
egdaniele05bbbb2016-04-19 12:13:41 -0700299 break;
cdalton397536c2016-03-25 12:15:03 -0700300 case kIndex_GrBufferType:
301 SkASSERT(kDynamic_GrAccessPattern == accessPattern ||
302 kStatic_GrAccessPattern == accessPattern);
cdalton1bf3e712016-04-19 10:00:02 -0700303 buff = GrVkIndexBuffer::Create(this, size, kDynamic_GrAccessPattern == accessPattern);
egdaniele05bbbb2016-04-19 12:13:41 -0700304 break;
cdalton397536c2016-03-25 12:15:03 -0700305 case kXferCpuToGpu_GrBufferType:
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400306 SkASSERT(kDynamic_GrAccessPattern == accessPattern ||
307 kStream_GrAccessPattern == accessPattern);
cdalton1bf3e712016-04-19 10:00:02 -0700308 buff = GrVkTransferBuffer::Create(this, size, GrVkBuffer::kCopyRead_Type);
egdaniele05bbbb2016-04-19 12:13:41 -0700309 break;
cdalton397536c2016-03-25 12:15:03 -0700310 case kXferGpuToCpu_GrBufferType:
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400311 SkASSERT(kDynamic_GrAccessPattern == accessPattern ||
312 kStream_GrAccessPattern == accessPattern);
cdalton1bf3e712016-04-19 10:00:02 -0700313 buff = GrVkTransferBuffer::Create(this, size, GrVkBuffer::kCopyWrite_Type);
egdaniele05bbbb2016-04-19 12:13:41 -0700314 break;
Greg Danielc2dd5ed2017-05-05 13:49:11 -0400315 case kTexel_GrBufferType:
Chris Dalton6b65b982017-07-06 11:04:00 -0600316 SkASSERT(kDynamic_GrAccessPattern == accessPattern ||
317 kStatic_GrAccessPattern == accessPattern);
318 buff = GrVkTexelBuffer::Create(this, size, kDynamic_GrAccessPattern == accessPattern);
Greg Danielc2dd5ed2017-05-05 13:49:11 -0400319 break;
320 case kDrawIndirect_GrBufferType:
Ben Wagnerb4aab9a2017-08-16 10:53:04 -0400321 SK_ABORT("DrawIndirect Buffers not supported in vulkan backend.");
Greg Danielc2dd5ed2017-05-05 13:49:11 -0400322 return nullptr;
cdalton397536c2016-03-25 12:15:03 -0700323 default:
Ben Wagnerb4aab9a2017-08-16 10:53:04 -0400324 SK_ABORT("Unknown buffer type.");
cdalton397536c2016-03-25 12:15:03 -0700325 return nullptr;
326 }
cdalton1bf3e712016-04-19 10:00:02 -0700327 if (data && buff) {
328 buff->updateData(data, size);
329 }
330 return buff;
Greg Daniel164a9f02016-02-22 09:56:40 -0500331}
332
333////////////////////////////////////////////////////////////////////////////////
Brian Salomonc320b152018-02-20 14:05:36 -0500334bool GrVkGpu::onGetWritePixelsInfo(GrSurface* dstSurface, GrSurfaceOrigin dstOrigin, int width,
335 int height, GrColorType srcColorType,
336 DrawPreference* drawPreference,
Greg Daniel164a9f02016-02-22 09:56:40 -0500337 WritePixelTempDrawInfo* tempDrawInfo) {
Brian Salomonc320b152018-02-20 14:05:36 -0500338 // We don't want to introduce a sRGB conversion if we trigger a draw.
339 auto srcConfigSRGBEncoded = GrPixelConfigIsSRGBEncoded(dstSurface->config());
Brian Salomon9b009bb2018-02-14 13:53:55 -0500340 if (*drawPreference != kNoDraw_DrawPreference) {
341 // We assume the base class has only inserted a draw for sRGB reasons. So the temp surface
342 // has the config of the original src data. There is no swizzling nor src config spoofing.
Brian Salomonc320b152018-02-20 14:05:36 -0500343 SkASSERT(tempDrawInfo->fWriteColorType == srcColorType);
344 SkASSERT(GrPixelConfigToColorType(tempDrawInfo->fTempSurfaceDesc.fConfig) == srcColorType);
Brian Salomon9b009bb2018-02-14 13:53:55 -0500345 SkASSERT(tempDrawInfo->fSwizzle == GrSwizzle::RGBA());
Brian Salomonc320b152018-02-20 14:05:36 -0500346 // Don't undo a sRGB conversion introduced by our caller via an intermediate draw.
347 srcConfigSRGBEncoded = GrPixelConfigIsSRGBEncoded(tempDrawInfo->fTempSurfaceDesc.fConfig);
Brian Salomon9b009bb2018-02-14 13:53:55 -0500348 }
Brian Salomonc320b152018-02-20 14:05:36 -0500349 if (GrColorTypeIsAlphaOnly(srcColorType)) {
350 srcConfigSRGBEncoded = GrSRGBEncoded::kNo;
351 }
egdaniel4583ec52016-06-27 12:57:00 -0700352 GrRenderTarget* renderTarget = dstSurface->asRenderTarget();
353
Brian Salomonc320b152018-02-20 14:05:36 -0500354 if (GrPixelConfigToColorType(dstSurface->config()) == srcColorType) {
Brian Osman33910292017-04-18 14:38:53 -0400355 // We only support writing pixels to textures. Forcing a draw lets us write to pure RTs.
356 if (!dstSurface->asTexture()) {
357 ElevateDrawPreference(drawPreference, kRequireDraw_DrawPreference);
358 }
359 // If the dst is MSAA, we have to draw, or we'll just be writing to the resolve target.
360 if (renderTarget && renderTarget->numColorSamples() > 1) {
361 ElevateDrawPreference(drawPreference, kRequireDraw_DrawPreference);
egdaniel4583ec52016-06-27 12:57:00 -0700362 }
363 return true;
Greg Daniel164a9f02016-02-22 09:56:40 -0500364 }
365
Brian Salomonc320b152018-02-20 14:05:36 -0500366 // Any color type change requires a draw
Brian Osman33910292017-04-18 14:38:53 -0400367 ElevateDrawPreference(drawPreference, kRequireDraw_DrawPreference);
368
Brian Salomonc320b152018-02-20 14:05:36 -0500369 auto srcAsConfig = GrColorTypeToPixelConfig(srcColorType, srcConfigSRGBEncoded);
370 SkASSERT(srcAsConfig != kUnknown_GrPixelConfig);
371 bool configsAreRBSwaps = GrPixelConfigSwapRAndB(srcAsConfig) == dstSurface->config();
Brian Osman33910292017-04-18 14:38:53 -0400372
Brian Salomonc320b152018-02-20 14:05:36 -0500373 if (!this->vkCaps().isConfigTexturable(srcAsConfig) && configsAreRBSwaps) {
Brian Osman33910292017-04-18 14:38:53 -0400374 tempDrawInfo->fTempSurfaceDesc.fConfig = dstSurface->config();
375 tempDrawInfo->fSwizzle = GrSwizzle::BGRA();
Brian Salomonc320b152018-02-20 14:05:36 -0500376 tempDrawInfo->fWriteColorType = GrPixelConfigToColorType(dstSurface->config());
Brian Osman33910292017-04-18 14:38:53 -0400377 }
378 return true;
Greg Daniel164a9f02016-02-22 09:56:40 -0500379}
380
Brian Salomonc320b152018-02-20 14:05:36 -0500381bool GrVkGpu::onWritePixels(GrSurface* surface, GrSurfaceOrigin origin, int left, int top,
382 int width, int height, GrColorType srcColorType,
Robert Phillips590533f2017-07-11 14:22:35 -0400383 const GrMipLevel texels[], int mipLevelCount) {
Greg Daniel164a9f02016-02-22 09:56:40 -0500384 GrVkTexture* vkTex = static_cast<GrVkTexture*>(surface->asTexture());
385 if (!vkTex) {
386 return false;
387 }
388
jvanverth900bd4a2016-04-29 13:53:12 -0700389 // Make sure we have at least the base level
Robert Phillips590533f2017-07-11 14:22:35 -0400390 if (!mipLevelCount || !texels[0].fPixels) {
jvanverth03509ea2016-03-02 13:19:47 -0800391 return false;
392 }
bsalomona1e6b3b2016-03-02 10:58:23 -0800393
Greg Daniel164a9f02016-02-22 09:56:40 -0500394 bool success = false;
Robert Phillips92de6312017-05-23 07:43:48 -0400395 bool linearTiling = vkTex->isLinearTiled();
396 if (linearTiling) {
Robert Phillips590533f2017-07-11 14:22:35 -0400397 if (mipLevelCount > 1) {
Robert Phillips92de6312017-05-23 07:43:48 -0400398 SkDebugf("Can't upload mipmap data to linear tiled texture");
399 return false;
400 }
401 if (VK_IMAGE_LAYOUT_PREINITIALIZED != vkTex->currentLayout()) {
402 // Need to change the layout to general in order to perform a host write
403 vkTex->setImageLayout(this,
404 VK_IMAGE_LAYOUT_GENERAL,
405 VK_ACCESS_HOST_WRITE_BIT,
406 VK_PIPELINE_STAGE_HOST_BIT,
407 false);
408 this->submitCommandBuffer(kForce_SyncQueue);
409 }
Brian Salomonc320b152018-02-20 14:05:36 -0500410 success = this->uploadTexDataLinear(vkTex, origin, left, top, width, height, srcColorType,
Robert Phillips590533f2017-07-11 14:22:35 -0400411 texels[0].fPixels, texels[0].fRowBytes);
Greg Daniel164a9f02016-02-22 09:56:40 -0500412 } else {
Robert Phillips92de6312017-05-23 07:43:48 -0400413 int currentMipLevels = vkTex->texturePriv().maxMipMapLevel() + 1;
Robert Phillips590533f2017-07-11 14:22:35 -0400414 if (mipLevelCount > currentMipLevels) {
415 if (!vkTex->reallocForMipmap(this, mipLevelCount)) {
jvanverth900bd4a2016-04-29 13:53:12 -0700416 return false;
417 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500418 }
Brian Salomonc320b152018-02-20 14:05:36 -0500419 success = this->uploadTexDataOptimal(vkTex, origin, left, top, width, height, srcColorType,
Robert Phillips590533f2017-07-11 14:22:35 -0400420 texels, mipLevelCount);
Greg Daniel164a9f02016-02-22 09:56:40 -0500421 }
egdaniel4583ec52016-06-27 12:57:00 -0700422
jvanverth900bd4a2016-04-29 13:53:12 -0700423 return success;
Greg Daniel164a9f02016-02-22 09:56:40 -0500424}
425
Brian Salomonc320b152018-02-20 14:05:36 -0500426bool GrVkGpu::onTransferPixels(GrTexture* texture, int left, int top, int width, int height,
427 GrColorType bufferColorType, GrBuffer* transferBuffer,
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400428 size_t bufferOffset, size_t rowBytes) {
429 // Vulkan only supports 4-byte aligned offsets
430 if (SkToBool(bufferOffset & 0x2)) {
431 return false;
432 }
433 GrVkTexture* vkTex = static_cast<GrVkTexture*>(texture);
434 if (!vkTex) {
435 return false;
436 }
437 GrVkTransferBuffer* vkBuffer = static_cast<GrVkTransferBuffer*>(transferBuffer);
438 if (!vkBuffer) {
439 return false;
440 }
441
Greg Daniel660cc992017-06-26 14:55:05 -0400442 SkDEBUGCODE(
443 SkIRect subRect = SkIRect::MakeXYWH(left, top, width, height);
444 SkIRect bounds = SkIRect::MakeWH(texture->width(), texture->height());
445 SkASSERT(bounds.contains(subRect));
446 )
Brian Salomonc320b152018-02-20 14:05:36 -0500447 int bpp = GrColorTypeBytesPerPixel(bufferColorType);
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400448 if (rowBytes == 0) {
Brian Salomonc320b152018-02-20 14:05:36 -0500449 rowBytes = bpp * width;
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400450 }
451
452 // Set up copy region
453 VkBufferImageCopy region;
454 memset(&region, 0, sizeof(VkBufferImageCopy));
455 region.bufferOffset = bufferOffset;
456 region.bufferRowLength = (uint32_t)(rowBytes/bpp);
457 region.bufferImageHeight = 0;
458 region.imageSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
459 region.imageOffset = { left, top, 0 };
460 region.imageExtent = { (uint32_t)width, (uint32_t)height, 1 };
461
462 // Change layout of our target so it can be copied to
463 vkTex->setImageLayout(this,
464 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
465 VK_ACCESS_TRANSFER_WRITE_BIT,
466 VK_PIPELINE_STAGE_TRANSFER_BIT,
467 false);
468
469 // Copy the buffer to the image
470 fCurrentCmdBuffer->copyBufferToImage(this,
471 vkBuffer,
472 vkTex,
473 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
474 1,
475 &region);
476
Greg Daniel0fc4d2d2017-10-12 11:23:36 -0400477 vkTex->texturePriv().markMipMapsDirty();
Jim Van Verth2e5eaf02017-06-21 15:55:46 -0400478 return true;
479}
480
Brian Salomon1fabd512018-02-09 09:54:25 -0500481void GrVkGpu::resolveImage(GrSurface* dst, GrVkRenderTarget* src, const SkIRect& srcRect,
482 const SkIPoint& dstPoint) {
egdaniel4bcd62e2016-08-31 07:37:31 -0700483 SkASSERT(dst);
484 SkASSERT(src && src->numColorSamples() > 1 && src->msaaImage());
485
egdanielfd016d72016-09-27 12:13:05 -0700486 if (this->vkCaps().mustSubmitCommandsBeforeCopyOp()) {
487 this->submitCommandBuffer(GrVkGpu::kSkip_SyncQueue);
488 }
489
egdaniel4bcd62e2016-08-31 07:37:31 -0700490 VkImageResolve resolveInfo;
Brian Salomon1fabd512018-02-09 09:54:25 -0500491 resolveInfo.srcSubresource = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1};
492 resolveInfo.srcOffset = {srcRect.fLeft, srcRect.fTop, 0};
493 resolveInfo.dstSubresource = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1};
494 resolveInfo.dstOffset = {dstPoint.fX, dstPoint.fY, 0};
495 resolveInfo.extent = {(uint32_t)srcRect.width(), (uint32_t)srcRect.height(), 1};
egdaniel4bcd62e2016-08-31 07:37:31 -0700496
Greg Danielbc26c392017-04-18 13:32:10 -0400497 GrVkImage* dstImage;
498 GrRenderTarget* dstRT = dst->asRenderTarget();
499 if (dstRT) {
500 GrVkRenderTarget* vkRT = static_cast<GrVkRenderTarget*>(dstRT);
Greg Danielbc26c392017-04-18 13:32:10 -0400501 dstImage = vkRT;
502 } else {
503 SkASSERT(dst->asTexture());
504 dstImage = static_cast<GrVkTexture*>(dst->asTexture());
505 }
506 dstImage->setImageLayout(this,
507 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
508 VK_ACCESS_TRANSFER_WRITE_BIT,
509 VK_PIPELINE_STAGE_TRANSFER_BIT,
510 false);
egdaniel4bcd62e2016-08-31 07:37:31 -0700511
512 src->msaaImage()->setImageLayout(this,
513 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
514 VK_ACCESS_TRANSFER_READ_BIT,
515 VK_PIPELINE_STAGE_TRANSFER_BIT,
516 false);
517
Greg Danielbc26c392017-04-18 13:32:10 -0400518 fCurrentCmdBuffer->resolveImage(this, *src->msaaImage(), *dstImage, 1, &resolveInfo);
egdaniel4bcd62e2016-08-31 07:37:31 -0700519}
520
Brian Salomon1fabd512018-02-09 09:54:25 -0500521void GrVkGpu::internalResolveRenderTarget(GrRenderTarget* target, bool requiresSubmit) {
egdaniel66933552016-08-24 07:22:19 -0700522 if (target->needsResolve()) {
523 SkASSERT(target->numColorSamples() > 1);
egdaniel52ad2512016-08-04 12:50:01 -0700524 GrVkRenderTarget* rt = static_cast<GrVkRenderTarget*>(target);
525 SkASSERT(rt->msaaImage());
Greg Daniel69d49922017-02-23 09:44:02 -0500526
egdaniel4bcd62e2016-08-31 07:37:31 -0700527 const SkIRect& srcRect = rt->getResolveRect();
egdaniel52ad2512016-08-04 12:50:01 -0700528
Brian Salomon1fabd512018-02-09 09:54:25 -0500529 this->resolveImage(target, rt, srcRect, SkIPoint::Make(srcRect.fLeft, srcRect.fTop));
egdaniel52ad2512016-08-04 12:50:01 -0700530
531 rt->flagAsResolved();
Greg Daniel69d49922017-02-23 09:44:02 -0500532
533 if (requiresSubmit) {
534 this->submitCommandBuffer(kSkip_SyncQueue);
535 }
egdaniel52ad2512016-08-04 12:50:01 -0700536 }
537}
538
Brian Salomonc320b152018-02-20 14:05:36 -0500539bool GrVkGpu::uploadTexDataLinear(GrVkTexture* tex, GrSurfaceOrigin texOrigin, int left, int top,
540 int width, int height, GrColorType dataColorType,
541 const void* data, size_t rowBytes) {
Greg Daniel164a9f02016-02-22 09:56:40 -0500542 SkASSERT(data);
jvanverth900bd4a2016-04-29 13:53:12 -0700543 SkASSERT(tex->isLinearTiled());
Greg Daniel164a9f02016-02-22 09:56:40 -0500544
Greg Daniel660cc992017-06-26 14:55:05 -0400545 SkDEBUGCODE(
546 SkIRect subRect = SkIRect::MakeXYWH(left, top, width, height);
547 SkIRect bounds = SkIRect::MakeWH(tex->width(), tex->height());
548 SkASSERT(bounds.contains(subRect));
549 )
Brian Salomonc320b152018-02-20 14:05:36 -0500550 int bpp = GrColorTypeBytesPerPixel(dataColorType);
Greg Daniel164a9f02016-02-22 09:56:40 -0500551 size_t trimRowBytes = width * bpp;
Greg Daniel660cc992017-06-26 14:55:05 -0400552 if (!rowBytes) {
553 rowBytes = trimRowBytes;
554 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500555
jvanverth900bd4a2016-04-29 13:53:12 -0700556 SkASSERT(VK_IMAGE_LAYOUT_PREINITIALIZED == tex->currentLayout() ||
557 VK_IMAGE_LAYOUT_GENERAL == tex->currentLayout());
558 const VkImageSubresource subres = {
559 VK_IMAGE_ASPECT_COLOR_BIT,
560 0, // mipLevel
561 0, // arraySlice
562 };
563 VkSubresourceLayout layout;
564 VkResult err;
Greg Daniel164a9f02016-02-22 09:56:40 -0500565
jvanverth900bd4a2016-04-29 13:53:12 -0700566 const GrVkInterface* interface = this->vkInterface();
Greg Daniel164a9f02016-02-22 09:56:40 -0500567
jvanverth900bd4a2016-04-29 13:53:12 -0700568 GR_VK_CALL(interface, GetImageSubresourceLayout(fDevice,
egdanielb2df0c22016-05-13 11:30:37 -0700569 tex->image(),
jvanverth900bd4a2016-04-29 13:53:12 -0700570 &subres,
571 &layout));
Greg Daniel164a9f02016-02-22 09:56:40 -0500572
Robert Phillipsb0e93a22017-08-29 08:26:54 -0400573 int texTop = kBottomLeft_GrSurfaceOrigin == texOrigin ? tex->height() - top - height : top;
jvanverth1e305ba2016-06-01 09:39:15 -0700574 const GrVkAlloc& alloc = tex->alloc();
575 VkDeviceSize offset = alloc.fOffset + texTop*layout.rowPitch + left*bpp;
Greg Daniel8385a8a2018-02-26 13:29:37 -0500576 VkDeviceSize offsetDiff = 0;
jvanverth900bd4a2016-04-29 13:53:12 -0700577 VkDeviceSize size = height*layout.rowPitch;
Greg Daniel8385a8a2018-02-26 13:29:37 -0500578 // For Noncoherent buffers we want to make sure the range that we map, both offset and size,
579 // are aligned to the nonCoherentAtomSize limit. We may have to move the initial offset back to
580 // meet the alignment requirements. So we track how far we move back and then adjust the mapped
581 // ptr back up so that this is opaque to the caller.
582 if (SkToBool(alloc.fFlags & GrVkAlloc::kNoncoherent_Flag)) {
583 VkDeviceSize alignment = this->physicalDeviceProperties().limits.nonCoherentAtomSize;
584 offsetDiff = offset & (alignment - 1);
585 offset = offset - offsetDiff;
586 // Make size of the map aligned to nonCoherentAtomSize
587 size = (size + alignment - 1) & ~(alignment - 1);
588 }
589 SkASSERT(offset >= alloc.fOffset);
590 SkASSERT(size <= alloc.fOffset + alloc.fSize);
jvanverth900bd4a2016-04-29 13:53:12 -0700591 void* mapPtr;
jvanverth1e305ba2016-06-01 09:39:15 -0700592 err = GR_VK_CALL(interface, MapMemory(fDevice, alloc.fMemory, offset, size, 0, &mapPtr));
jvanverth900bd4a2016-04-29 13:53:12 -0700593 if (err) {
594 return false;
595 }
Greg Daniel8385a8a2018-02-26 13:29:37 -0500596 mapPtr = reinterpret_cast<char*>(mapPtr) + offsetDiff;
jvanverth900bd4a2016-04-29 13:53:12 -0700597
Robert Phillipsb0e93a22017-08-29 08:26:54 -0400598 if (kBottomLeft_GrSurfaceOrigin == texOrigin) {
jvanverth900bd4a2016-04-29 13:53:12 -0700599 // copy into buffer by rows
600 const char* srcRow = reinterpret_cast<const char*>(data);
601 char* dstRow = reinterpret_cast<char*>(mapPtr)+(height - 1)*layout.rowPitch;
602 for (int y = 0; y < height; y++) {
603 memcpy(dstRow, srcRow, trimRowBytes);
604 srcRow += rowBytes;
605 dstRow -= layout.rowPitch;
606 }
607 } else {
Matt Sarettcf3525a2017-01-24 12:43:41 -0500608 SkRectMemcpy(mapPtr, static_cast<size_t>(layout.rowPitch), data, rowBytes, trimRowBytes,
609 height);
jvanverth900bd4a2016-04-29 13:53:12 -0700610 }
611
Greg Daniele35a99e2018-03-02 11:44:22 -0500612 GrVkMemory::FlushMappedAlloc(this, alloc, offset, size);
jvanverth1e305ba2016-06-01 09:39:15 -0700613 GR_VK_CALL(interface, UnmapMemory(fDevice, alloc.fMemory));
jvanverth900bd4a2016-04-29 13:53:12 -0700614
615 return true;
616}
617
Brian Salomonc320b152018-02-20 14:05:36 -0500618bool GrVkGpu::uploadTexDataOptimal(GrVkTexture* tex, GrSurfaceOrigin texOrigin, int left, int top,
619 int width, int height, GrColorType dataColorType,
Robert Phillips590533f2017-07-11 14:22:35 -0400620 const GrMipLevel texels[], int mipLevelCount) {
jvanverth900bd4a2016-04-29 13:53:12 -0700621 SkASSERT(!tex->isLinearTiled());
622 // The assumption is either that we have no mipmaps, or that our rect is the entire texture
Robert Phillips590533f2017-07-11 14:22:35 -0400623 SkASSERT(1 == mipLevelCount ||
jvanverth900bd4a2016-04-29 13:53:12 -0700624 (0 == left && 0 == top && width == tex->width() && height == tex->height()));
625
Greg Danieldd20e912017-04-07 14:42:23 -0400626 // We assume that if the texture has mip levels, we either upload to all the levels or just the
627 // first.
Robert Phillips590533f2017-07-11 14:22:35 -0400628 SkASSERT(1 == mipLevelCount || mipLevelCount == (tex->texturePriv().maxMipMapLevel() + 1));
Greg Danieldd20e912017-04-07 14:42:23 -0400629
jvanverth900bd4a2016-04-29 13:53:12 -0700630 if (width == 0 || height == 0) {
631 return false;
632 }
633
Brian Salomond1eaf492017-05-18 10:02:08 -0400634 SkASSERT(this->caps()->isConfigTexturable(tex->config()));
Brian Salomonc320b152018-02-20 14:05:36 -0500635 int bpp = GrColorTypeBytesPerPixel(dataColorType);
jvanverth900bd4a2016-04-29 13:53:12 -0700636
637 // texels is const.
jvanverthc578b0632016-05-02 10:58:12 -0700638 // But we may need to adjust the fPixels ptr based on the copyRect, or fRowBytes.
639 // Because of this we need to make a non-const shallow copy of texels.
Robert Phillips0f992772017-07-12 08:24:56 -0400640 SkAutoTMalloc<GrMipLevel> texelsShallowCopy;
641
642 if (mipLevelCount) {
643 texelsShallowCopy.reset(mipLevelCount);
644 memcpy(texelsShallowCopy.get(), texels, mipLevelCount*sizeof(GrMipLevel));
645 }
jvanverth900bd4a2016-04-29 13:53:12 -0700646
jvanverth900bd4a2016-04-29 13:53:12 -0700647 // Determine whether we need to flip when we copy into the buffer
Robert Phillipsb0e93a22017-08-29 08:26:54 -0400648 bool flipY = (kBottomLeft_GrSurfaceOrigin == texOrigin && mipLevelCount);
jvanverth900bd4a2016-04-29 13:53:12 -0700649
Robert Phillips590533f2017-07-11 14:22:35 -0400650 SkTArray<size_t> individualMipOffsets(mipLevelCount);
jvanverthc578b0632016-05-02 10:58:12 -0700651 individualMipOffsets.push_back(0);
652 size_t combinedBufferSize = width * bpp * height;
653 int currentWidth = width;
654 int currentHeight = height;
Greg Daniel55afd6d2017-09-29 09:32:44 -0400655 if (mipLevelCount > 0 && !texelsShallowCopy[0].fPixels) {
656 combinedBufferSize = 0;
657 }
658
Greg Daniel468fd632017-03-22 17:03:45 -0400659 // The alignment must be at least 4 bytes and a multiple of the bytes per pixel of the image
660 // config. This works with the assumption that the bytes in pixel config is always a power of 2.
661 SkASSERT((bpp & (bpp - 1)) == 0);
662 const size_t alignmentMask = 0x3 | (bpp - 1);
Robert Phillips590533f2017-07-11 14:22:35 -0400663 for (int currentMipLevel = 1; currentMipLevel < mipLevelCount; currentMipLevel++) {
jvanverthc578b0632016-05-02 10:58:12 -0700664 currentWidth = SkTMax(1, currentWidth/2);
665 currentHeight = SkTMax(1, currentHeight/2);
Greg Daniel660cc992017-06-26 14:55:05 -0400666
Greg Daniel55afd6d2017-09-29 09:32:44 -0400667 if (texelsShallowCopy[currentMipLevel].fPixels) {
668 const size_t trimmedSize = currentWidth * bpp * currentHeight;
669 const size_t alignmentDiff = combinedBufferSize & alignmentMask;
670 if (alignmentDiff != 0) {
671 combinedBufferSize += alignmentMask - alignmentDiff + 1;
672 }
673 individualMipOffsets.push_back(combinedBufferSize);
674 combinedBufferSize += trimmedSize;
675 } else {
676 individualMipOffsets.push_back(0);
Greg Daniel468fd632017-03-22 17:03:45 -0400677 }
Greg Daniel55afd6d2017-09-29 09:32:44 -0400678 }
679 if (0 == combinedBufferSize) {
680 // We don't actually have any data to upload so just return success
681 return true;
jvanverth900bd4a2016-04-29 13:53:12 -0700682 }
683
684 // allocate buffer to hold our mip data
685 GrVkTransferBuffer* transferBuffer =
686 GrVkTransferBuffer::Create(this, combinedBufferSize, GrVkBuffer::kCopyRead_Type);
Greg Daniel6888c0d2017-08-25 11:55:50 -0400687 if(!transferBuffer) {
Forrest Reilingc04f8452017-04-26 19:26:12 -0700688 return false;
Greg Daniel6888c0d2017-08-25 11:55:50 -0400689 }
jvanverth900bd4a2016-04-29 13:53:12 -0700690
691 char* buffer = (char*) transferBuffer->map();
Robert Phillips590533f2017-07-11 14:22:35 -0400692 SkTArray<VkBufferImageCopy> regions(mipLevelCount);
jvanverth900bd4a2016-04-29 13:53:12 -0700693
jvanverthc578b0632016-05-02 10:58:12 -0700694 currentWidth = width;
695 currentHeight = height;
Greg Daniela1b282b2017-03-28 14:56:46 -0400696 int layerHeight = tex->height();
Robert Phillips590533f2017-07-11 14:22:35 -0400697 for (int currentMipLevel = 0; currentMipLevel < mipLevelCount; currentMipLevel++) {
Greg Daniel55afd6d2017-09-29 09:32:44 -0400698 if (texelsShallowCopy[currentMipLevel].fPixels) {
699 SkASSERT(1 == mipLevelCount || currentHeight == layerHeight);
700 const size_t trimRowBytes = currentWidth * bpp;
701 const size_t rowBytes = texelsShallowCopy[currentMipLevel].fRowBytes
702 ? texelsShallowCopy[currentMipLevel].fRowBytes
703 : trimRowBytes;
jvanverth900bd4a2016-04-29 13:53:12 -0700704
Greg Daniel55afd6d2017-09-29 09:32:44 -0400705 // copy data into the buffer, skipping the trailing bytes
706 char* dst = buffer + individualMipOffsets[currentMipLevel];
707 const char* src = (const char*)texelsShallowCopy[currentMipLevel].fPixels;
708 if (flipY) {
709 src += (currentHeight - 1) * rowBytes;
710 for (int y = 0; y < currentHeight; y++) {
711 memcpy(dst, src, trimRowBytes);
712 src -= rowBytes;
713 dst += trimRowBytes;
714 }
715 } else {
716 SkRectMemcpy(dst, trimRowBytes, src, rowBytes, trimRowBytes, currentHeight);
jvanverth900bd4a2016-04-29 13:53:12 -0700717 }
Greg Daniel55afd6d2017-09-29 09:32:44 -0400718
719 VkBufferImageCopy& region = regions.push_back();
720 memset(&region, 0, sizeof(VkBufferImageCopy));
721 region.bufferOffset = transferBuffer->offset() + individualMipOffsets[currentMipLevel];
722 region.bufferRowLength = currentWidth;
723 region.bufferImageHeight = currentHeight;
724 region.imageSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, SkToU32(currentMipLevel), 0, 1 };
725 region.imageOffset = { left, flipY ? layerHeight - top - currentHeight : top, 0 };
726 region.imageExtent = { (uint32_t)currentWidth, (uint32_t)currentHeight, 1 };
jvanverth900bd4a2016-04-29 13:53:12 -0700727 }
jvanverthc578b0632016-05-02 10:58:12 -0700728 currentWidth = SkTMax(1, currentWidth/2);
729 currentHeight = SkTMax(1, currentHeight/2);
Greg Daniela1b282b2017-03-28 14:56:46 -0400730 layerHeight = currentHeight;
jvanverth900bd4a2016-04-29 13:53:12 -0700731 }
732
jvanverth9d54afc2016-09-20 09:20:03 -0700733 // no need to flush non-coherent memory, unmap will do that for us
jvanverth900bd4a2016-04-29 13:53:12 -0700734 transferBuffer->unmap();
735
jvanverth900bd4a2016-04-29 13:53:12 -0700736 // Change layout of our target so it can be copied to
jvanverth900bd4a2016-04-29 13:53:12 -0700737 tex->setImageLayout(this,
738 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
jvanverth50c46c72016-05-06 12:31:28 -0700739 VK_ACCESS_TRANSFER_WRITE_BIT,
740 VK_PIPELINE_STAGE_TRANSFER_BIT,
jvanverth900bd4a2016-04-29 13:53:12 -0700741 false);
742
743 // Copy the buffer to the image
744 fCurrentCmdBuffer->copyBufferToImage(this,
745 transferBuffer,
746 tex,
747 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
748 regions.count(),
749 regions.begin());
jvanverth900bd4a2016-04-29 13:53:12 -0700750 transferBuffer->unref();
Robert Phillips590533f2017-07-11 14:22:35 -0400751 if (1 == mipLevelCount) {
Greg Daniel0fc4d2d2017-10-12 11:23:36 -0400752 tex->texturePriv().markMipMapsDirty();
Greg Danieldd20e912017-04-07 14:42:23 -0400753 }
jvanverth900bd4a2016-04-29 13:53:12 -0700754
Greg Daniel164a9f02016-02-22 09:56:40 -0500755 return true;
756}
757
758////////////////////////////////////////////////////////////////////////////////
Robert Phillips67d52cf2017-06-05 13:38:13 -0400759sk_sp<GrTexture> GrVkGpu::onCreateTexture(const GrSurfaceDesc& desc, SkBudgeted budgeted,
Brian Salomon58389b92018-03-07 13:01:25 -0500760 const GrMipLevel texels[], int mipLevelCount) {
Greg Daniel164a9f02016-02-22 09:56:40 -0500761 bool renderTarget = SkToBool(desc.fFlags & kRenderTarget_GrSurfaceFlag);
762
763 VkFormat pixelFormat;
Brian Salomonbdecacf2018-02-02 20:32:49 -0500764 SkAssertResult(GrPixelConfigToVkFormat(desc.fConfig, &pixelFormat));
egdaniel0a3a7f72016-06-24 09:22:31 -0700765
Greg Daniel164a9f02016-02-22 09:56:40 -0500766 VkImageUsageFlags usageFlags = VK_IMAGE_USAGE_SAMPLED_BIT;
767 if (renderTarget) {
768 usageFlags |= VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
769 }
770
771 // For now we will set the VK_IMAGE_USAGE_TRANSFER_DESTINATION_BIT and
772 // VK_IMAGE_USAGE_TRANSFER_SOURCE_BIT on every texture since we do not know whether or not we
773 // will be using this texture in some copy or not. Also this assumes, as is the current case,
jvanverth62340062016-04-26 08:01:44 -0700774 // that all render targets in vulkan are also textures. If we change this practice of setting
Greg Daniel164a9f02016-02-22 09:56:40 -0500775 // both bits, we must make sure to set the destination bit if we are uploading srcData to the
776 // texture.
777 usageFlags |= VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_TRANSFER_DST_BIT;
778
Greg Daniel164a9f02016-02-22 09:56:40 -0500779 // This ImageDesc refers to the texture that will be read by the client. Thus even if msaa is
jvanverth62340062016-04-26 08:01:44 -0700780 // requested, this ImageDesc describes the resolved texture. Therefore we always have samples set
Greg Daniel164a9f02016-02-22 09:56:40 -0500781 // to 1.
Robert Phillips590533f2017-07-11 14:22:35 -0400782 int mipLevels = !mipLevelCount ? 1 : mipLevelCount;
Greg Daniel164a9f02016-02-22 09:56:40 -0500783 GrVkImage::ImageDesc imageDesc;
784 imageDesc.fImageType = VK_IMAGE_TYPE_2D;
785 imageDesc.fFormat = pixelFormat;
786 imageDesc.fWidth = desc.fWidth;
787 imageDesc.fHeight = desc.fHeight;
Brian Salomon7128fdd2017-05-22 14:00:07 -0400788 imageDesc.fLevels = mipLevels;
Greg Daniel164a9f02016-02-22 09:56:40 -0500789 imageDesc.fSamples = 1;
Brian Salomon7128fdd2017-05-22 14:00:07 -0400790 imageDesc.fImageTiling = VK_IMAGE_TILING_OPTIMAL;
Greg Daniel164a9f02016-02-22 09:56:40 -0500791 imageDesc.fUsageFlags = usageFlags;
Brian Salomon7128fdd2017-05-22 14:00:07 -0400792 imageDesc.fMemProps = VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT;
Greg Daniel164a9f02016-02-22 09:56:40 -0500793
Greg Daniel0fc4d2d2017-10-12 11:23:36 -0400794 GrMipMapsStatus mipMapsStatus = GrMipMapsStatus::kNotAllocated;
795 if (mipLevels > 1) {
796 mipMapsStatus = GrMipMapsStatus::kValid;
797 for (int i = 0; i < mipLevels; ++i) {
798 if (!texels[i].fPixels) {
799 mipMapsStatus = GrMipMapsStatus::kDirty;
800 break;
801 }
Greg Daniel834f1202017-10-09 15:06:20 -0400802 }
803 }
804
Robert Phillips67d52cf2017-06-05 13:38:13 -0400805 sk_sp<GrVkTexture> tex;
Greg Daniel164a9f02016-02-22 09:56:40 -0500806 if (renderTarget) {
kkinnunen2e6055b2016-04-22 01:48:29 -0700807 tex = GrVkTextureRenderTarget::CreateNewTextureRenderTarget(this, budgeted, desc,
Greg Daniel834f1202017-10-09 15:06:20 -0400808 imageDesc,
Greg Daniel0fc4d2d2017-10-12 11:23:36 -0400809 mipMapsStatus);
Greg Daniel164a9f02016-02-22 09:56:40 -0500810 } else {
Greg Daniel834f1202017-10-09 15:06:20 -0400811 tex = GrVkTexture::CreateNewTexture(this, budgeted, desc, imageDesc,
Greg Daniel0fc4d2d2017-10-12 11:23:36 -0400812 mipMapsStatus);
Greg Daniel164a9f02016-02-22 09:56:40 -0500813 }
814
815 if (!tex) {
816 return nullptr;
817 }
818
Brian Salomonc320b152018-02-20 14:05:36 -0500819 auto colorType = GrPixelConfigToColorType(desc.fConfig);
Robert Phillips590533f2017-07-11 14:22:35 -0400820 if (mipLevelCount) {
Brian Salomon58389b92018-03-07 13:01:25 -0500821 if (!this->uploadTexDataOptimal(tex.get(), kTopLeft_GrSurfaceOrigin, 0, 0, desc.fWidth,
822 desc.fHeight, colorType, texels, mipLevelCount)) {
Greg Daniel164a9f02016-02-22 09:56:40 -0500823 tex->unref();
824 return nullptr;
825 }
826 }
827
Brian Salomond17b4a62017-05-23 16:53:47 -0400828 if (desc.fFlags & kPerformInitialClear_GrSurfaceFlag) {
829 VkClearColorValue zeroClearColor;
830 memset(&zeroClearColor, 0, sizeof(zeroClearColor));
831 VkImageSubresourceRange range;
832 range.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
833 range.baseArrayLayer = 0;
834 range.baseMipLevel = 0;
835 range.layerCount = 1;
836 range.levelCount = 1;
837 tex->setImageLayout(this, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
838 VK_ACCESS_TRANSFER_WRITE_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, false);
Robert Phillips67d52cf2017-06-05 13:38:13 -0400839 this->currentCommandBuffer()->clearColorImage(this, tex.get(), &zeroClearColor, 1, &range);
Brian Salomond17b4a62017-05-23 16:53:47 -0400840 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500841 return tex;
842}
843
844////////////////////////////////////////////////////////////////////////////////
845
Greg Daniel6888c0d2017-08-25 11:55:50 -0400846void GrVkGpu::copyBuffer(GrVkBuffer* srcBuffer, GrVkBuffer* dstBuffer, VkDeviceSize srcOffset,
847 VkDeviceSize dstOffset, VkDeviceSize size) {
848 VkBufferCopy copyRegion;
849 copyRegion.srcOffset = srcOffset;
850 copyRegion.dstOffset = dstOffset;
851 copyRegion.size = size;
852 fCurrentCmdBuffer->copyBuffer(this, srcBuffer, dstBuffer, 1, &copyRegion);
853}
854
jvanverthdb379092016-07-07 11:18:46 -0700855bool GrVkGpu::updateBuffer(GrVkBuffer* buffer, const void* src,
856 VkDeviceSize offset, VkDeviceSize size) {
jvanvertha584de92016-06-30 09:10:52 -0700857 // Update the buffer
jvanverthdb379092016-07-07 11:18:46 -0700858 fCurrentCmdBuffer->updateBuffer(this, buffer, offset, size, src);
jvanvertha584de92016-06-30 09:10:52 -0700859
860 return true;
861}
862
863////////////////////////////////////////////////////////////////////////////////
864
Greg Daniel5254ccc2017-11-13 11:05:52 -0500865static bool check_backend_texture(const GrBackendTexture& backendTex,
866 GrPixelConfig config) {
Greg Daniel52e16d92018-04-10 09:34:07 -0400867 GrVkImageInfo info;
868 if (!backendTex.getVkImageInfo(&info)) {
Brian Salomond17f6582017-07-19 18:28:58 -0400869 return false;
Greg Daniel164a9f02016-02-22 09:56:40 -0500870 }
871
Greg Daniel52e16d92018-04-10 09:34:07 -0400872 if (VK_NULL_HANDLE == info.fImage || VK_NULL_HANDLE == info.fAlloc.fMemory) {
Brian Salomond17f6582017-07-19 18:28:58 -0400873 return false;
jvanverthfd359ca2016-03-18 11:57:24 -0700874 }
Greg Daniel7ef28f32017-04-20 16:41:55 +0000875
Greg Daniel52e16d92018-04-10 09:34:07 -0400876 SkASSERT(GrVkFormatPixelConfigPairIsValid(info.fFormat, config));
Brian Salomond17f6582017-07-19 18:28:58 -0400877 return true;
878}
879
880sk_sp<GrTexture> GrVkGpu::onWrapBackendTexture(const GrBackendTexture& backendTex,
Brian Salomond17f6582017-07-19 18:28:58 -0400881 GrWrapOwnership ownership) {
Greg Daniel5254ccc2017-11-13 11:05:52 -0500882 if (!check_backend_texture(backendTex, backendTex.config())) {
Brian Salomond17f6582017-07-19 18:28:58 -0400883 return nullptr;
884 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500885
Greg Daniel164a9f02016-02-22 09:56:40 -0500886 GrSurfaceDesc surfDesc;
Brian Salomond17f6582017-07-19 18:28:58 -0400887 surfDesc.fFlags = kNone_GrSurfaceFlags;
Greg Daniel7ef28f32017-04-20 16:41:55 +0000888 surfDesc.fWidth = backendTex.width();
889 surfDesc.fHeight = backendTex.height();
890 surfDesc.fConfig = backendTex.config();
Brian Salomonbdecacf2018-02-02 20:32:49 -0500891 surfDesc.fSampleCnt = 1;
Greg Daniel164a9f02016-02-22 09:56:40 -0500892
Greg Daniel52e16d92018-04-10 09:34:07 -0400893 GrVkImageInfo imageInfo;
894 if (!backendTex.getVkImageInfo(&imageInfo)) {
895 return nullptr;
896 }
897 sk_sp<GrVkImageLayout> layout = backendTex.getGrVkImageLayout();
898 SkASSERT(layout);
899 return GrVkTexture::MakeWrappedTexture(this, surfDesc, ownership, imageInfo, std::move(layout));
Brian Salomond17f6582017-07-19 18:28:58 -0400900}
901
902sk_sp<GrTexture> GrVkGpu::onWrapRenderableBackendTexture(const GrBackendTexture& backendTex,
Brian Salomond17f6582017-07-19 18:28:58 -0400903 int sampleCnt,
904 GrWrapOwnership ownership) {
Greg Daniel5254ccc2017-11-13 11:05:52 -0500905 if (!check_backend_texture(backendTex, backendTex.config())) {
Brian Salomond17f6582017-07-19 18:28:58 -0400906 return nullptr;
Greg Daniel164a9f02016-02-22 09:56:40 -0500907 }
Brian Salomond17f6582017-07-19 18:28:58 -0400908
909 GrSurfaceDesc surfDesc;
910 surfDesc.fFlags = kRenderTarget_GrSurfaceFlag;
911 surfDesc.fWidth = backendTex.width();
912 surfDesc.fHeight = backendTex.height();
913 surfDesc.fConfig = backendTex.config();
Brian Salomonbdecacf2018-02-02 20:32:49 -0500914 surfDesc.fSampleCnt = this->caps()->getRenderTargetSampleCount(sampleCnt, backendTex.config());
Brian Salomond17f6582017-07-19 18:28:58 -0400915
Greg Daniel52e16d92018-04-10 09:34:07 -0400916 GrVkImageInfo imageInfo;
917 if (!backendTex.getVkImageInfo(&imageInfo)) {
918 return nullptr;
919 }
920 sk_sp<GrVkImageLayout> layout = backendTex.getGrVkImageLayout();
921 SkASSERT(layout);
922
Brian Salomond17f6582017-07-19 18:28:58 -0400923 return GrVkTextureRenderTarget::MakeWrappedTextureRenderTarget(this, surfDesc, ownership,
Greg Daniel52e16d92018-04-10 09:34:07 -0400924 imageInfo, std::move(layout));
Greg Daniel164a9f02016-02-22 09:56:40 -0500925}
926
Robert Phillipsb0e93a22017-08-29 08:26:54 -0400927sk_sp<GrRenderTarget> GrVkGpu::onWrapBackendRenderTarget(const GrBackendRenderTarget& backendRT){
Greg Daniele79b4732017-04-20 14:07:46 -0400928 // Currently the Vulkan backend does not support wrapping of msaa render targets directly. In
929 // general this is not an issue since swapchain images in vulkan are never multisampled. Thus if
930 // you want a multisampled RT it is best to wrap the swapchain images and then let Skia handle
931 // creating and owning the MSAA images.
Brian Salomonbdecacf2018-02-02 20:32:49 -0500932 if (backendRT.sampleCnt() > 1) {
Greg Daniele79b4732017-04-20 14:07:46 -0400933 return nullptr;
934 }
halcanary9d524f22016-03-29 09:03:52 -0700935
Greg Danielbcf612b2017-05-01 13:50:58 +0000936 const GrVkImageInfo* info = backendRT.getVkImageInfo();
937 if (!info) {
938 return nullptr;
939 }
Brian Osman0b791f52017-03-10 08:30:22 -0500940 if (VK_NULL_HANDLE == info->fImage) {
jvanverthfd359ca2016-03-18 11:57:24 -0700941 return nullptr;
942 }
Greg Daniel164a9f02016-02-22 09:56:40 -0500943
Greg Daniel164a9f02016-02-22 09:56:40 -0500944 GrSurfaceDesc desc;
Brian Salomon0ec981b2017-05-15 13:48:50 -0400945 desc.fFlags = kRenderTarget_GrSurfaceFlag;
Robert Phillips16d8ec62017-07-27 16:16:25 -0400946 desc.fWidth = backendRT.width();
947 desc.fHeight = backendRT.height();
948 desc.fConfig = backendRT.config();
Brian Salomonbdecacf2018-02-02 20:32:49 -0500949 desc.fSampleCnt = 1;
Greg Daniel164a9f02016-02-22 09:56:40 -0500950
Greg Daniel52e16d92018-04-10 09:34:07 -0400951 sk_sp<GrVkImageLayout> layout(new GrVkImageLayout(info->fImageLayout));
952
953 sk_sp<GrVkRenderTarget> tgt = GrVkRenderTarget::MakeWrappedRenderTarget(this, desc, *info,
954 std::move(layout));
Brian Salomonafdc6b12018-03-09 12:02:32 -0500955
956 // We don't allow the client to supply a premade stencil buffer. We always create one if needed.
957 SkASSERT(!backendRT.stencilBits());
958 if (tgt) {
959 SkASSERT(tgt->canAttemptStencilAttachment());
Greg Daniel164a9f02016-02-22 09:56:40 -0500960 }
Brian Salomonafdc6b12018-03-09 12:02:32 -0500961
Greg Daniel164a9f02016-02-22 09:56:40 -0500962 return tgt;
963}
964
Greg Daniel7ef28f32017-04-20 16:41:55 +0000965sk_sp<GrRenderTarget> GrVkGpu::onWrapBackendTextureAsRenderTarget(const GrBackendTexture& tex,
Greg Daniel7ef28f32017-04-20 16:41:55 +0000966 int sampleCnt) {
Brian Osman33910292017-04-18 14:38:53 -0400967
Greg Daniel52e16d92018-04-10 09:34:07 -0400968 GrVkImageInfo imageInfo;
969 if (!tex.getVkImageInfo(&imageInfo)) {
Greg Danielbcf612b2017-05-01 13:50:58 +0000970 return nullptr;
971 }
Greg Daniel52e16d92018-04-10 09:34:07 -0400972 if (VK_NULL_HANDLE == imageInfo.fImage) {
Brian Osman33910292017-04-18 14:38:53 -0400973 return nullptr;
974 }
975
976 GrSurfaceDesc desc;
Greg Daniel7ef28f32017-04-20 16:41:55 +0000977 desc.fFlags = kRenderTarget_GrSurfaceFlag;
Greg Daniel7ef28f32017-04-20 16:41:55 +0000978 desc.fWidth = tex.width();
979 desc.fHeight = tex.height();
Robert Phillips16d8ec62017-07-27 16:16:25 -0400980 desc.fConfig = tex.config();
Brian Salomonbdecacf2018-02-02 20:32:49 -0500981 desc.fSampleCnt = this->caps()->getRenderTargetSampleCount(sampleCnt, tex.config());
982 if (!desc.fSampleCnt) {
983 return nullptr;
984 }
Brian Osman33910292017-04-18 14:38:53 -0400985
Greg Daniel52e16d92018-04-10 09:34:07 -0400986 sk_sp<GrVkImageLayout> layout = tex.getGrVkImageLayout();
987 SkASSERT(layout);
988
989 sk_sp<GrVkRenderTarget> tgt = GrVkRenderTarget::MakeWrappedRenderTarget(this, desc, imageInfo,
990 std::move(layout));
Brian Osman33910292017-04-18 14:38:53 -0400991 return tgt;
992}
993
Robert Phillipsb0e93a22017-08-29 08:26:54 -0400994void GrVkGpu::generateMipmap(GrVkTexture* tex, GrSurfaceOrigin texOrigin) {
jvanverth900bd4a2016-04-29 13:53:12 -0700995 // don't do anything for linearly tiled textures (can't have mipmaps)
jvanverth62340062016-04-26 08:01:44 -0700996 if (tex->isLinearTiled()) {
jvanverth900bd4a2016-04-29 13:53:12 -0700997 SkDebugf("Trying to create mipmap for linear tiled texture");
jvanverth62340062016-04-26 08:01:44 -0700998 return;
999 }
1000
jvanverth62340062016-04-26 08:01:44 -07001001 // determine if we can blit to and from this format
1002 const GrVkCaps& caps = this->vkCaps();
1003 if (!caps.configCanBeDstofBlit(tex->config(), false) ||
egdaniel2f5792a2016-07-06 08:51:23 -07001004 !caps.configCanBeSrcofBlit(tex->config(), false) ||
1005 !caps.mipMapSupport()) {
jvanverth62340062016-04-26 08:01:44 -07001006 return;
1007 }
1008
egdanielfd016d72016-09-27 12:13:05 -07001009 if (this->vkCaps().mustSubmitCommandsBeforeCopyOp()) {
1010 this->submitCommandBuffer(kSkip_SyncQueue);
1011 }
1012
egdaniel66933552016-08-24 07:22:19 -07001013 // We may need to resolve the texture first if it is also a render target
1014 GrVkRenderTarget* texRT = static_cast<GrVkRenderTarget*>(tex->asRenderTarget());
1015 if (texRT) {
Brian Salomon1fabd512018-02-09 09:54:25 -05001016 this->internalResolveRenderTarget(texRT, false);
egdaniel66933552016-08-24 07:22:19 -07001017 }
1018
egdaniel7ac5da82016-07-15 13:41:42 -07001019 int width = tex->width();
1020 int height = tex->height();
1021 VkImageBlit blitRegion;
1022 memset(&blitRegion, 0, sizeof(VkImageBlit));
jvanverth62340062016-04-26 08:01:44 -07001023
jvanverth82c05582016-05-03 11:19:01 -07001024 // SkMipMap doesn't include the base level in the level count so we have to add 1
1025 uint32_t levelCount = SkMipMap::ComputeLevelCount(tex->width(), tex->height()) + 1;
egdaniel7ac5da82016-07-15 13:41:42 -07001026 if (levelCount != tex->mipLevels()) {
1027 const GrVkResource* oldResource = tex->resource();
1028 oldResource->ref();
1029 // grab handle to the original image resource
1030 VkImage oldImage = tex->image();
1031
1032 // change the original image's layout so we can copy from it
1033 tex->setImageLayout(this, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
1034 VK_ACCESS_TRANSFER_READ_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, false);
1035
1036 if (!tex->reallocForMipmap(this, levelCount)) {
1037 oldResource->unref(this);
1038 return;
1039 }
1040 // change the new image's layout so we can blit to it
Greg Daniel31cc7312018-03-05 11:41:06 -05001041 tex->setImageLayout(this, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
egdaniel7ac5da82016-07-15 13:41:42 -07001042 VK_ACCESS_TRANSFER_WRITE_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, false);
1043
1044 // Blit original image to top level of new image
1045 blitRegion.srcSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1046 blitRegion.srcOffsets[0] = { 0, 0, 0 };
1047 blitRegion.srcOffsets[1] = { width, height, 1 };
1048 blitRegion.dstSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1049 blitRegion.dstOffsets[0] = { 0, 0, 0 };
1050 blitRegion.dstOffsets[1] = { width, height, 1 };
1051
1052 fCurrentCmdBuffer->blitImage(this,
1053 oldResource,
1054 oldImage,
1055 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
1056 tex->resource(),
1057 tex->image(),
Greg Daniel31cc7312018-03-05 11:41:06 -05001058 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
egdaniel7ac5da82016-07-15 13:41:42 -07001059 1,
1060 &blitRegion,
1061 VK_FILTER_LINEAR);
1062
jvanverth62340062016-04-26 08:01:44 -07001063 oldResource->unref(this);
egdaniel7ac5da82016-07-15 13:41:42 -07001064 } else {
1065 // change layout of the layers so we can write to them.
Greg Daniel31cc7312018-03-05 11:41:06 -05001066 tex->setImageLayout(this, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
egdaniel7ac5da82016-07-15 13:41:42 -07001067 VK_ACCESS_TRANSFER_WRITE_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, false);
jvanverth62340062016-04-26 08:01:44 -07001068 }
1069
jvanverth50c46c72016-05-06 12:31:28 -07001070 // setup memory barrier
Greg Daniel81b80592017-12-13 10:20:04 -05001071 SkASSERT(GrVkFormatIsSupported(tex->imageFormat()));
jvanverth50c46c72016-05-06 12:31:28 -07001072 VkImageAspectFlags aspectFlags = VK_IMAGE_ASPECT_COLOR_BIT;
1073 VkImageMemoryBarrier imageMemoryBarrier = {
1074 VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER, // sType
Ben Wagnera93a14a2017-08-28 10:34:05 -04001075 nullptr, // pNext
egdaniel7ac5da82016-07-15 13:41:42 -07001076 VK_ACCESS_TRANSFER_WRITE_BIT, // srcAccessMask
1077 VK_ACCESS_TRANSFER_READ_BIT, // dstAccessMask
Greg Daniel31cc7312018-03-05 11:41:06 -05001078 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, // oldLayout
1079 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, // newLayout
jvanverth50c46c72016-05-06 12:31:28 -07001080 VK_QUEUE_FAMILY_IGNORED, // srcQueueFamilyIndex
1081 VK_QUEUE_FAMILY_IGNORED, // dstQueueFamilyIndex
egdanielb2df0c22016-05-13 11:30:37 -07001082 tex->image(), // image
jvanverth50c46c72016-05-06 12:31:28 -07001083 { aspectFlags, 0, 1, 0, 1 } // subresourceRange
1084 };
1085
jvanverth62340062016-04-26 08:01:44 -07001086 // Blit the miplevels
jvanverth82c05582016-05-03 11:19:01 -07001087 uint32_t mipLevel = 1;
1088 while (mipLevel < levelCount) {
1089 int prevWidth = width;
1090 int prevHeight = height;
1091 width = SkTMax(1, width / 2);
1092 height = SkTMax(1, height / 2);
jvanverth62340062016-04-26 08:01:44 -07001093
jvanverth50c46c72016-05-06 12:31:28 -07001094 imageMemoryBarrier.subresourceRange.baseMipLevel = mipLevel - 1;
1095 this->addImageMemoryBarrier(VK_PIPELINE_STAGE_TRANSFER_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT,
1096 false, &imageMemoryBarrier);
1097
1098 blitRegion.srcSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, mipLevel - 1, 0, 1 };
jvanverth82c05582016-05-03 11:19:01 -07001099 blitRegion.srcOffsets[0] = { 0, 0, 0 };
brianosmane9906e72016-06-08 12:44:27 -07001100 blitRegion.srcOffsets[1] = { prevWidth, prevHeight, 1 };
jvanverth82c05582016-05-03 11:19:01 -07001101 blitRegion.dstSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, mipLevel, 0, 1 };
1102 blitRegion.dstOffsets[0] = { 0, 0, 0 };
brianosmane9906e72016-06-08 12:44:27 -07001103 blitRegion.dstOffsets[1] = { width, height, 1 };
jvanverth62340062016-04-26 08:01:44 -07001104 fCurrentCmdBuffer->blitImage(this,
Greg Daniel31cc7312018-03-05 11:41:06 -05001105 tex->resource(),
1106 tex->image(),
1107 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
1108 tex->resource(),
1109 tex->image(),
1110 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
jvanverth62340062016-04-26 08:01:44 -07001111 1,
1112 &blitRegion,
1113 VK_FILTER_LINEAR);
jvanverth82c05582016-05-03 11:19:01 -07001114 ++mipLevel;
jvanverth62340062016-04-26 08:01:44 -07001115 }
Greg Daniel31cc7312018-03-05 11:41:06 -05001116 // This barrier logically is not needed, but it changes the final level to the same layout as
1117 // all the others, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL. This makes tracking of the layouts and
1118 // future layout changes easier.
1119 imageMemoryBarrier.subresourceRange.baseMipLevel = mipLevel - 1;
1120 this->addImageMemoryBarrier(VK_PIPELINE_STAGE_TRANSFER_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT,
1121 false, &imageMemoryBarrier);
1122 tex->updateImageLayout(VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL);
jvanverth62340062016-04-26 08:01:44 -07001123}
1124
Greg Daniel164a9f02016-02-22 09:56:40 -05001125////////////////////////////////////////////////////////////////////////////////
1126
1127GrStencilAttachment* GrVkGpu::createStencilAttachmentForRenderTarget(const GrRenderTarget* rt,
1128 int width,
1129 int height) {
Greg Daniel164a9f02016-02-22 09:56:40 -05001130 SkASSERT(width >= rt->width());
1131 SkASSERT(height >= rt->height());
1132
1133 int samples = rt->numStencilSamples();
1134
egdaniel8f1dcaa2016-04-01 10:10:45 -07001135 const GrVkCaps::StencilFormat& sFmt = this->vkCaps().preferedStencilFormat();
Greg Daniel164a9f02016-02-22 09:56:40 -05001136
1137 GrVkStencilAttachment* stencil(GrVkStencilAttachment::Create(this,
Greg Daniel164a9f02016-02-22 09:56:40 -05001138 width,
1139 height,
1140 samples,
1141 sFmt));
1142 fStats.incStencilAttachmentCreates();
1143 return stencil;
1144}
1145
1146////////////////////////////////////////////////////////////////////////////////
1147
Brian Salomon52e943a2018-03-13 09:32:39 -04001148bool copy_testing_data(GrVkGpu* gpu, const void* srcData, const GrVkAlloc& alloc,
1149 size_t bufferOffset, size_t srcRowBytes, size_t dstRowBytes, int h) {
Greg Daniel8385a8a2018-02-26 13:29:37 -05001150 // For Noncoherent buffers we want to make sure the range that we map, both offset and size,
1151 // are aligned to the nonCoherentAtomSize limit. We may have to move the initial offset back to
1152 // meet the alignment requirements. So we track how far we move back and then adjust the mapped
1153 // ptr back up so that this is opaque to the caller.
1154 VkDeviceSize mapSize = dstRowBytes * h;
1155 VkDeviceSize mapOffset = alloc.fOffset + bufferOffset;
1156 VkDeviceSize offsetDiff = 0;
1157 if (SkToBool(alloc.fFlags & GrVkAlloc::kNoncoherent_Flag)) {
1158 VkDeviceSize alignment = gpu->physicalDeviceProperties().limits.nonCoherentAtomSize;
1159 offsetDiff = mapOffset & (alignment - 1);
1160 mapOffset = mapOffset - offsetDiff;
1161 // Make size of the map aligned to nonCoherentAtomSize
1162 mapSize = (mapSize + alignment - 1) & ~(alignment - 1);
1163 }
1164 SkASSERT(mapOffset >= alloc.fOffset);
1165 SkASSERT(mapSize + mapOffset <= alloc.fOffset + alloc.fSize);
egdaniel3602d4f2016-08-12 11:58:53 -07001166 void* mapPtr;
1167 VkResult err = GR_VK_CALL(gpu->vkInterface(), MapMemory(gpu->device(),
jvanverth9d54afc2016-09-20 09:20:03 -07001168 alloc.fMemory,
Greg Daniel8385a8a2018-02-26 13:29:37 -05001169 mapOffset,
1170 mapSize,
egdaniel3602d4f2016-08-12 11:58:53 -07001171 0,
1172 &mapPtr));
Greg Daniel8385a8a2018-02-26 13:29:37 -05001173 mapPtr = reinterpret_cast<char*>(mapPtr) + offsetDiff;
egdaniel3602d4f2016-08-12 11:58:53 -07001174 if (err) {
1175 return false;
1176 }
1177
Greg Daniel20ece3a2017-03-28 10:24:43 -04001178 if (srcData) {
1179 // If there is no padding on dst we can do a single memcopy.
1180 // This assumes the srcData comes in with no padding.
1181 SkRectMemcpy(mapPtr, static_cast<size_t>(dstRowBytes),
1182 srcData, srcRowBytes, srcRowBytes, h);
1183 } else {
1184 // If there is no srcdata we always copy 0's into the textures so that it is initialized
1185 // with some data.
1186 if (srcRowBytes == static_cast<size_t>(dstRowBytes)) {
1187 memset(mapPtr, 0, srcRowBytes * h);
1188 } else {
1189 for (int i = 0; i < h; ++i) {
1190 memset(mapPtr, 0, srcRowBytes);
1191 mapPtr = SkTAddOffset<void>(mapPtr, static_cast<size_t>(dstRowBytes));
1192 }
1193 }
1194 }
Greg Daniele35a99e2018-03-02 11:44:22 -05001195 GrVkMemory::FlushMappedAlloc(gpu, alloc, mapOffset, mapSize);
jvanverth9d54afc2016-09-20 09:20:03 -07001196 GR_VK_CALL(gpu->vkInterface(), UnmapMemory(gpu->device(), alloc.fMemory));
egdaniel3602d4f2016-08-12 11:58:53 -07001197 return true;
1198}
1199
Brian Salomonf865b052018-03-09 09:01:53 -05001200#if GR_TEST_UTILS
Brian Salomon52e943a2018-03-13 09:32:39 -04001201bool GrVkGpu::createTestingOnlyVkImage(GrPixelConfig config, int w, int h, bool texturable,
1202 bool renderable, GrMipMapped mipMapped, const void* srcData,
1203 GrVkImageInfo* info) {
1204 SkASSERT(texturable || renderable);
1205 if (!texturable) {
1206 SkASSERT(GrMipMapped::kNo == mipMapped);
1207 SkASSERT(!srcData);
1208 }
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001209 VkFormat pixelFormat;
1210 if (!GrPixelConfigToVkFormat(config, &pixelFormat)) {
Brian Salomon52e943a2018-03-13 09:32:39 -04001211 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001212 }
1213
Brian Salomon52e943a2018-03-13 09:32:39 -04001214 if (texturable && !fVkCaps->isConfigTexturable(config)) {
1215 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001216 }
1217
Brian Salomon52e943a2018-03-13 09:32:39 -04001218 if (renderable && !fVkCaps->isConfigRenderable(config)) {
1219 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001220 }
1221
1222 // Currently we don't support uploading pixel data when mipped.
1223 if (srcData && GrMipMapped::kYes == mipMapped) {
Brian Salomon52e943a2018-03-13 09:32:39 -04001224 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001225 }
1226
Brian Salomon52e943a2018-03-13 09:32:39 -04001227 VkImageUsageFlags usageFlags = 0;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001228 usageFlags |= VK_IMAGE_USAGE_TRANSFER_SRC_BIT;
1229 usageFlags |= VK_IMAGE_USAGE_TRANSFER_DST_BIT;
Brian Salomon52e943a2018-03-13 09:32:39 -04001230 if (texturable) {
1231 usageFlags |= VK_IMAGE_USAGE_SAMPLED_BIT;
1232 }
1233 if (renderable) {
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001234 usageFlags |= VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT;
1235 }
1236
1237 VkImage image = VK_NULL_HANDLE;
Greg Daniel8385a8a2018-02-26 13:29:37 -05001238 GrVkAlloc alloc;
Brian Salomonde9f5462018-03-07 14:23:58 -05001239 VkImageLayout initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001240
1241 // Create Image
1242 VkSampleCountFlagBits vkSamples;
1243 if (!GrSampleCountToVkSampleCount(1, &vkSamples)) {
Brian Salomon52e943a2018-03-13 09:32:39 -04001244 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001245 }
1246
1247 // Figure out the number of mip levels.
1248 uint32_t mipLevels = 1;
1249 if (GrMipMapped::kYes == mipMapped) {
1250 mipLevels = SkMipMap::ComputeLevelCount(w, h) + 1;
1251 }
1252
1253 const VkImageCreateInfo imageCreateInfo = {
Brian Salomonde9f5462018-03-07 14:23:58 -05001254 VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO, // sType
1255 nullptr, // pNext
1256 0, // VkImageCreateFlags
1257 VK_IMAGE_TYPE_2D, // VkImageType
1258 pixelFormat, // VkFormat
1259 {(uint32_t)w, (uint32_t)h, 1}, // VkExtent3D
1260 mipLevels, // mipLevels
1261 1, // arrayLayers
1262 vkSamples, // samples
1263 VK_IMAGE_TILING_OPTIMAL, // VkImageTiling
1264 usageFlags, // VkImageUsageFlags
1265 VK_SHARING_MODE_EXCLUSIVE, // VkSharingMode
1266 0, // queueFamilyCount
1267 0, // pQueueFamilyIndices
1268 initialLayout // initialLayout
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001269 };
1270
Brian Salomon52e943a2018-03-13 09:32:39 -04001271 GR_VK_CALL_ERRCHECK(this->vkInterface(),
1272 CreateImage(this->device(), &imageCreateInfo, nullptr, &image));
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001273
Brian Salomonde9f5462018-03-07 14:23:58 -05001274 if (!GrVkMemory::AllocAndBindImageMemory(this, image, false, &alloc)) {
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001275 VK_CALL(DestroyImage(this->device(), image, nullptr));
Brian Salomon52e943a2018-03-13 09:32:39 -04001276 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001277 }
1278
1279 // We need to declare these early so that we can delete them at the end outside of the if block.
Greg Daniel8385a8a2018-02-26 13:29:37 -05001280 GrVkAlloc bufferAlloc;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001281 VkBuffer buffer = VK_NULL_HANDLE;
1282
1283 VkResult err;
1284 const VkCommandBufferAllocateInfo cmdInfo = {
1285 VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO, // sType
1286 nullptr, // pNext
1287 fCmdPool, // commandPool
1288 VK_COMMAND_BUFFER_LEVEL_PRIMARY, // level
1289 1 // bufferCount
1290 };
1291
1292 VkCommandBuffer cmdBuffer;
1293 err = VK_CALL(AllocateCommandBuffers(fDevice, &cmdInfo, &cmdBuffer));
1294 if (err) {
1295 GrVkMemory::FreeImageMemory(this, false, alloc);
1296 VK_CALL(DestroyImage(fDevice, image, nullptr));
Brian Salomon52e943a2018-03-13 09:32:39 -04001297 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001298 }
1299
1300 VkCommandBufferBeginInfo cmdBufferBeginInfo;
1301 memset(&cmdBufferBeginInfo, 0, sizeof(VkCommandBufferBeginInfo));
1302 cmdBufferBeginInfo.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
1303 cmdBufferBeginInfo.pNext = nullptr;
1304 cmdBufferBeginInfo.flags = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
1305 cmdBufferBeginInfo.pInheritanceInfo = nullptr;
1306
1307 err = VK_CALL(BeginCommandBuffer(cmdBuffer, &cmdBufferBeginInfo));
1308 SkASSERT(!err);
1309
1310 size_t bpp = GrBytesPerPixel(config);
Brian Salomonde9f5462018-03-07 14:23:58 -05001311 SkASSERT(w && h);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001312
Brian Salomonde9f5462018-03-07 14:23:58 -05001313 SkTArray<size_t> individualMipOffsets(mipLevels);
1314 individualMipOffsets.push_back(0);
1315 size_t combinedBufferSize = w * bpp * h;
1316 int currentWidth = w;
1317 int currentHeight = h;
1318 // The alignment must be at least 4 bytes and a multiple of the bytes per pixel of the image
1319 // config. This works with the assumption that the bytes in pixel config is always a power
1320 // of 2.
1321 SkASSERT((bpp & (bpp - 1)) == 0);
1322 const size_t alignmentMask = 0x3 | (bpp - 1);
1323 for (uint32_t currentMipLevel = 1; currentMipLevel < mipLevels; currentMipLevel++) {
1324 currentWidth = SkTMax(1, currentWidth / 2);
1325 currentHeight = SkTMax(1, currentHeight / 2);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001326
Brian Salomonde9f5462018-03-07 14:23:58 -05001327 const size_t trimmedSize = currentWidth * bpp * currentHeight;
1328 const size_t alignmentDiff = combinedBufferSize & alignmentMask;
1329 if (alignmentDiff != 0) {
1330 combinedBufferSize += alignmentMask - alignmentDiff + 1;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001331 }
Brian Salomonde9f5462018-03-07 14:23:58 -05001332 individualMipOffsets.push_back(combinedBufferSize);
1333 combinedBufferSize += trimmedSize;
1334 }
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001335
Brian Salomonde9f5462018-03-07 14:23:58 -05001336 VkBufferCreateInfo bufInfo;
1337 memset(&bufInfo, 0, sizeof(VkBufferCreateInfo));
1338 bufInfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
1339 bufInfo.flags = 0;
1340 bufInfo.size = combinedBufferSize;
1341 bufInfo.usage = VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
1342 bufInfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;
1343 bufInfo.queueFamilyIndexCount = 0;
1344 bufInfo.pQueueFamilyIndices = nullptr;
1345 err = VK_CALL(CreateBuffer(fDevice, &bufInfo, nullptr, &buffer));
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001346
Brian Salomonde9f5462018-03-07 14:23:58 -05001347 if (err) {
1348 GrVkMemory::FreeImageMemory(this, false, alloc);
1349 VK_CALL(DestroyImage(fDevice, image, nullptr));
1350 VK_CALL(EndCommandBuffer(cmdBuffer));
1351 VK_CALL(FreeCommandBuffers(fDevice, fCmdPool, 1, &cmdBuffer));
Brian Salomon52e943a2018-03-13 09:32:39 -04001352 return false;
Brian Salomonde9f5462018-03-07 14:23:58 -05001353 }
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001354
Brian Salomonde9f5462018-03-07 14:23:58 -05001355 if (!GrVkMemory::AllocAndBindBufferMemory(this, buffer, GrVkBuffer::kCopyRead_Type, true,
1356 &bufferAlloc)) {
1357 GrVkMemory::FreeImageMemory(this, false, alloc);
1358 VK_CALL(DestroyImage(fDevice, image, nullptr));
1359 VK_CALL(DestroyBuffer(fDevice, buffer, nullptr));
1360 VK_CALL(EndCommandBuffer(cmdBuffer));
1361 VK_CALL(FreeCommandBuffers(fDevice, fCmdPool, 1, &cmdBuffer));
Brian Salomon52e943a2018-03-13 09:32:39 -04001362 return false;
Brian Salomonde9f5462018-03-07 14:23:58 -05001363 }
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001364
Brian Salomonde9f5462018-03-07 14:23:58 -05001365 currentWidth = w;
1366 currentHeight = h;
1367 for (uint32_t currentMipLevel = 0; currentMipLevel < mipLevels; currentMipLevel++) {
1368 SkASSERT(0 == currentMipLevel || !srcData);
1369 size_t currentRowBytes = bpp * currentWidth;
1370 size_t bufferOffset = individualMipOffsets[currentMipLevel];
1371 if (!copy_testing_data(this, srcData, bufferAlloc, bufferOffset, currentRowBytes,
1372 currentRowBytes, currentHeight)) {
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001373 GrVkMemory::FreeImageMemory(this, false, alloc);
1374 VK_CALL(DestroyImage(fDevice, image, nullptr));
Brian Salomonde9f5462018-03-07 14:23:58 -05001375 GrVkMemory::FreeBufferMemory(this, GrVkBuffer::kCopyRead_Type, bufferAlloc);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001376 VK_CALL(DestroyBuffer(fDevice, buffer, nullptr));
1377 VK_CALL(EndCommandBuffer(cmdBuffer));
1378 VK_CALL(FreeCommandBuffers(fDevice, fCmdPool, 1, &cmdBuffer));
Brian Salomon52e943a2018-03-13 09:32:39 -04001379 return false;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001380 }
Brian Salomonde9f5462018-03-07 14:23:58 -05001381 currentWidth = SkTMax(1, currentWidth / 2);
1382 currentHeight = SkTMax(1, currentHeight / 2);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001383 }
Brian Salomonde9f5462018-03-07 14:23:58 -05001384
1385 // Set image layout and add barrier
1386 VkImageMemoryBarrier barrier;
1387 memset(&barrier, 0, sizeof(VkImageMemoryBarrier));
1388 barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
1389 barrier.pNext = nullptr;
1390 barrier.srcAccessMask = GrVkMemory::LayoutToSrcAccessMask(initialLayout);
1391 barrier.dstAccessMask = VK_ACCESS_TRANSFER_WRITE_BIT;
1392 barrier.oldLayout = initialLayout;
1393 barrier.newLayout = VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
1394 barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
1395 barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
1396 barrier.image = image;
1397 barrier.subresourceRange = {VK_IMAGE_ASPECT_COLOR_BIT, 0, mipLevels, 0, 1};
1398
1399 VK_CALL(CmdPipelineBarrier(cmdBuffer, GrVkMemory::LayoutToPipelineStageFlags(initialLayout),
1400 VK_PIPELINE_STAGE_TRANSFER_BIT, 0, 0, nullptr, 0, nullptr, 1,
1401 &barrier));
1402 initialLayout = VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
1403
1404 SkTArray<VkBufferImageCopy> regions(mipLevels);
1405
1406 currentWidth = w;
1407 currentHeight = h;
1408 for (uint32_t currentMipLevel = 0; currentMipLevel < mipLevels; currentMipLevel++) {
1409 // Submit copy command
1410 VkBufferImageCopy& region = regions.push_back();
1411 memset(&region, 0, sizeof(VkBufferImageCopy));
1412 region.bufferOffset = individualMipOffsets[currentMipLevel];
1413 region.bufferRowLength = currentWidth;
1414 region.bufferImageHeight = currentHeight;
1415 region.imageSubresource = {VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1};
1416 region.imageOffset = {0, 0, 0};
1417 region.imageExtent = {(uint32_t)currentWidth, (uint32_t)currentHeight, 1};
1418 currentWidth = SkTMax(1, currentWidth / 2);
1419 currentHeight = SkTMax(1, currentHeight / 2);
1420 }
1421
1422 VK_CALL(CmdCopyBufferToImage(cmdBuffer, buffer, image, initialLayout, regions.count(),
1423 regions.begin()));
1424
Brian Salomon52e943a2018-03-13 09:32:39 -04001425 if (texturable) {
1426 // Change Image layout to shader read since if we use this texture as a borrowed textures
1427 // within Ganesh we require that its layout be set to that
1428 memset(&barrier, 0, sizeof(VkImageMemoryBarrier));
1429 barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
1430 barrier.pNext = nullptr;
1431 barrier.srcAccessMask = GrVkMemory::LayoutToSrcAccessMask(initialLayout);
1432 barrier.dstAccessMask = VK_ACCESS_SHADER_READ_BIT;
1433 barrier.oldLayout = initialLayout;
1434 barrier.newLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
1435 barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
1436 barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
1437 barrier.image = image;
1438 barrier.subresourceRange = {VK_IMAGE_ASPECT_COLOR_BIT, 0, mipLevels, 0, 1};
Brian Salomon52e943a2018-03-13 09:32:39 -04001439 VK_CALL(CmdPipelineBarrier(cmdBuffer,
1440 GrVkMemory::LayoutToPipelineStageFlags(initialLayout),
1441 VK_PIPELINE_STAGE_ALL_GRAPHICS_BIT,
1442 0,
1443 0, nullptr,
1444 0, nullptr,
1445 1, &barrier));
Greg Daniel4f4a53f2018-03-15 10:20:45 -04001446 initialLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
Brian Salomon52e943a2018-03-13 09:32:39 -04001447 }
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001448
1449 // End CommandBuffer
1450 err = VK_CALL(EndCommandBuffer(cmdBuffer));
1451 SkASSERT(!err);
1452
1453 // Create Fence for queue
1454 VkFence fence;
1455 VkFenceCreateInfo fenceInfo;
1456 memset(&fenceInfo, 0, sizeof(VkFenceCreateInfo));
1457 fenceInfo.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO;
1458
1459 err = VK_CALL(CreateFence(fDevice, &fenceInfo, nullptr, &fence));
1460 SkASSERT(!err);
1461
1462 VkSubmitInfo submitInfo;
1463 memset(&submitInfo, 0, sizeof(VkSubmitInfo));
1464 submitInfo.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
1465 submitInfo.pNext = nullptr;
1466 submitInfo.waitSemaphoreCount = 0;
1467 submitInfo.pWaitSemaphores = nullptr;
1468 submitInfo.pWaitDstStageMask = 0;
1469 submitInfo.commandBufferCount = 1;
1470 submitInfo.pCommandBuffers = &cmdBuffer;
1471 submitInfo.signalSemaphoreCount = 0;
1472 submitInfo.pSignalSemaphores = nullptr;
1473 err = VK_CALL(QueueSubmit(this->queue(), 1, &submitInfo, fence));
1474 SkASSERT(!err);
1475
1476 err = VK_CALL(WaitForFences(fDevice, 1, &fence, true, UINT64_MAX));
1477 if (VK_TIMEOUT == err) {
1478 GrVkMemory::FreeImageMemory(this, false, alloc);
1479 VK_CALL(DestroyImage(fDevice, image, nullptr));
1480 GrVkMemory::FreeBufferMemory(this, GrVkBuffer::kCopyRead_Type, bufferAlloc);
1481 VK_CALL(DestroyBuffer(fDevice, buffer, nullptr));
1482 VK_CALL(FreeCommandBuffers(fDevice, fCmdPool, 1, &cmdBuffer));
1483 VK_CALL(DestroyFence(fDevice, fence, nullptr));
1484 SkDebugf("Fence failed to signal: %d\n", err);
1485 SK_ABORT("failing");
1486 }
1487 SkASSERT(!err);
1488
1489 // Clean up transfer resources
1490 if (buffer != VK_NULL_HANDLE) { // workaround for an older NVidia driver crash
1491 GrVkMemory::FreeBufferMemory(this, GrVkBuffer::kCopyRead_Type, bufferAlloc);
1492 VK_CALL(DestroyBuffer(fDevice, buffer, nullptr));
1493 }
1494 VK_CALL(FreeCommandBuffers(fDevice, fCmdPool, 1, &cmdBuffer));
1495 VK_CALL(DestroyFence(fDevice, fence, nullptr));
1496
Brian Salomon52e943a2018-03-13 09:32:39 -04001497 info->fImage = image;
1498 info->fAlloc = alloc;
1499 info->fImageTiling = VK_IMAGE_TILING_OPTIMAL;
1500 info->fImageLayout = initialLayout;
1501 info->fFormat = pixelFormat;
1502 info->fLevelCount = mipLevels;
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001503
Brian Salomon52e943a2018-03-13 09:32:39 -04001504 return true;
1505}
1506
1507GrBackendTexture GrVkGpu::createTestingOnlyBackendTexture(const void* srcData, int w, int h,
1508 GrPixelConfig config, bool isRenderTarget,
1509 GrMipMapped mipMapped) {
Brian Salomon8a375832018-03-14 10:21:40 -04001510 this->handleDirtyContext();
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001511 GrVkImageInfo info;
Brian Salomon52e943a2018-03-13 09:32:39 -04001512 if (!this->createTestingOnlyVkImage(config, w, h, true, isRenderTarget, mipMapped, srcData,
1513 &info)) {
1514 return {};
1515 }
Greg Daniel8a3f55c2018-03-14 17:32:12 +00001516 return GrBackendTexture(w, h, info);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001517}
1518
1519bool GrVkGpu::isTestingOnlyBackendTexture(const GrBackendTexture& tex) const {
1520 SkASSERT(kVulkan_GrBackend == tex.fBackend);
1521
Greg Daniel52e16d92018-04-10 09:34:07 -04001522 GrVkImageInfo backend;
1523 if (!tex.getVkImageInfo(&backend)) {
1524 return false;
1525 }
Greg Daniel164a9f02016-02-22 09:56:40 -05001526
Greg Daniel52e16d92018-04-10 09:34:07 -04001527 if (backend.fImage && backend.fAlloc.fMemory) {
Greg Daniel164a9f02016-02-22 09:56:40 -05001528 VkMemoryRequirements req;
1529 memset(&req, 0, sizeof(req));
1530 GR_VK_CALL(this->vkInterface(), GetImageMemoryRequirements(fDevice,
Greg Daniel52e16d92018-04-10 09:34:07 -04001531 backend.fImage,
Greg Daniel164a9f02016-02-22 09:56:40 -05001532 &req));
1533 // TODO: find a better check
1534 // This will probably fail with a different driver
1535 return (req.size > 0) && (req.size <= 8192 * 8192);
1536 }
1537
1538 return false;
1539}
1540
Brian Salomon26102cb2018-03-09 09:33:19 -05001541void GrVkGpu::deleteTestingOnlyBackendTexture(const GrBackendTexture& tex) {
1542 SkASSERT(kVulkan_GrBackend == tex.fBackend);
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001543
Greg Daniel52e16d92018-04-10 09:34:07 -04001544 GrVkImageInfo info;
1545 if (tex.getVkImageInfo(&info)) {
Robert Phillipsd21b2a52017-12-12 13:01:25 -05001546 // something in the command buffer may still be using this, so force submit
1547 this->submitCommandBuffer(kForce_SyncQueue);
Greg Daniel52e16d92018-04-10 09:34:07 -04001548 GrVkImage::DestroyImageInfo(this, const_cast<GrVkImageInfo*>(&info));
Greg Daniel164a9f02016-02-22 09:56:40 -05001549 }
1550}
1551
Brian Salomon52e943a2018-03-13 09:32:39 -04001552GrBackendRenderTarget GrVkGpu::createTestingOnlyBackendRenderTarget(int w, int h, GrColorType ct,
1553 GrSRGBEncoded srgbEncoded) {
Brian Salomon8a375832018-03-14 10:21:40 -04001554 this->handleDirtyContext();
Brian Salomon52e943a2018-03-13 09:32:39 -04001555 GrVkImageInfo info;
1556 auto config = GrColorTypeToPixelConfig(ct, srgbEncoded);
1557 if (kUnknown_GrPixelConfig == config) {
1558 return {};
1559 }
1560 if (!this->createTestingOnlyVkImage(config, w, h, false, true, GrMipMapped::kNo, nullptr,
1561 &info)) {
1562 return {};
1563 }
Greg Daniel8a3f55c2018-03-14 17:32:12 +00001564 return {w, h, 1, 0, info};
Brian Salomonf865b052018-03-09 09:01:53 -05001565}
1566
Brian Salomon52e943a2018-03-13 09:32:39 -04001567void GrVkGpu::deleteTestingOnlyBackendRenderTarget(const GrBackendRenderTarget& rt) {
1568 SkASSERT(kVulkan_GrBackend == rt.fBackend);
Brian Salomonf865b052018-03-09 09:01:53 -05001569
Brian Salomon52e943a2018-03-13 09:32:39 -04001570 if (const auto* info = rt.getVkImageInfo()) {
1571 // something in the command buffer may still be using this, so force submit
1572 this->submitCommandBuffer(kForce_SyncQueue);
1573 GrVkImage::DestroyImageInfo(this, const_cast<GrVkImageInfo*>(info));
1574 }
1575}
Brian Salomonf865b052018-03-09 09:01:53 -05001576
Greg Daniel26b50a42018-03-08 09:49:58 -05001577void GrVkGpu::testingOnly_flushGpuAndSync() {
1578 this->submitCommandBuffer(kForce_SyncQueue);
1579}
Brian Salomonf865b052018-03-09 09:01:53 -05001580#endif
Greg Daniel26b50a42018-03-08 09:49:58 -05001581
Greg Daniel164a9f02016-02-22 09:56:40 -05001582////////////////////////////////////////////////////////////////////////////////
1583
1584void GrVkGpu::addMemoryBarrier(VkPipelineStageFlags srcStageMask,
1585 VkPipelineStageFlags dstStageMask,
1586 bool byRegion,
1587 VkMemoryBarrier* barrier) const {
1588 SkASSERT(fCurrentCmdBuffer);
1589 fCurrentCmdBuffer->pipelineBarrier(this,
1590 srcStageMask,
1591 dstStageMask,
1592 byRegion,
1593 GrVkCommandBuffer::kMemory_BarrierType,
1594 barrier);
1595}
1596
1597void GrVkGpu::addBufferMemoryBarrier(VkPipelineStageFlags srcStageMask,
1598 VkPipelineStageFlags dstStageMask,
1599 bool byRegion,
1600 VkBufferMemoryBarrier* barrier) const {
1601 SkASSERT(fCurrentCmdBuffer);
1602 fCurrentCmdBuffer->pipelineBarrier(this,
1603 srcStageMask,
1604 dstStageMask,
1605 byRegion,
1606 GrVkCommandBuffer::kBufferMemory_BarrierType,
1607 barrier);
1608}
1609
1610void GrVkGpu::addImageMemoryBarrier(VkPipelineStageFlags srcStageMask,
1611 VkPipelineStageFlags dstStageMask,
1612 bool byRegion,
1613 VkImageMemoryBarrier* barrier) const {
1614 SkASSERT(fCurrentCmdBuffer);
1615 fCurrentCmdBuffer->pipelineBarrier(this,
1616 srcStageMask,
1617 dstStageMask,
1618 byRegion,
1619 GrVkCommandBuffer::kImageMemory_BarrierType,
1620 barrier);
1621}
1622
Greg Daniel51316782017-08-02 15:10:09 +00001623void GrVkGpu::onFinishFlush(bool insertedSemaphore) {
1624 // Submit the current command buffer to the Queue. Whether we inserted semaphores or not does
1625 // not effect what we do here.
Greg Daniel164a9f02016-02-22 09:56:40 -05001626 this->submitCommandBuffer(kSkip_SyncQueue);
1627}
1628
Robert Phillips95214472017-08-08 18:00:03 -04001629void GrVkGpu::clearStencil(GrRenderTarget* target, int clearValue) {
1630 if (!target) {
egdaniel3d5d9ac2016-03-01 12:56:15 -08001631 return;
1632 }
1633 GrStencilAttachment* stencil = target->renderTargetPriv().getStencilAttachment();
1634 GrVkStencilAttachment* vkStencil = (GrVkStencilAttachment*)stencil;
1635
1636
1637 VkClearDepthStencilValue vkStencilColor;
Robert Phillips95214472017-08-08 18:00:03 -04001638 vkStencilColor.depth = 0.0f;
1639 vkStencilColor.stencil = clearValue;
egdaniel3d5d9ac2016-03-01 12:56:15 -08001640
egdaniel3d5d9ac2016-03-01 12:56:15 -08001641 vkStencil->setImageLayout(this,
1642 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
jvanverth50c46c72016-05-06 12:31:28 -07001643 VK_ACCESS_TRANSFER_WRITE_BIT,
1644 VK_PIPELINE_STAGE_TRANSFER_BIT,
egdaniel3d5d9ac2016-03-01 12:56:15 -08001645 false);
1646
egdaniel3d5d9ac2016-03-01 12:56:15 -08001647 VkImageSubresourceRange subRange;
1648 memset(&subRange, 0, sizeof(VkImageSubresourceRange));
1649 subRange.aspectMask = VK_IMAGE_ASPECT_STENCIL_BIT;
1650 subRange.baseMipLevel = 0;
1651 subRange.levelCount = 1;
1652 subRange.baseArrayLayer = 0;
1653 subRange.layerCount = 1;
1654
1655 // TODO: I imagine that most times we want to clear a stencil it will be at the beginning of a
1656 // draw. Thus we should look into using the load op functions on the render pass to clear out
1657 // the stencil there.
1658 fCurrentCmdBuffer->clearDepthStencilImage(this, vkStencil, &vkStencilColor, 1, &subRange);
1659}
1660
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001661inline bool can_copy_image(const GrSurface* dst, GrSurfaceOrigin dstOrigin,
1662 const GrSurface* src, GrSurfaceOrigin srcOrigin,
Greg Daniel164a9f02016-02-22 09:56:40 -05001663 const GrVkGpu* gpu) {
egdaniel4bcd62e2016-08-31 07:37:31 -07001664 const GrRenderTarget* dstRT = dst->asRenderTarget();
1665 const GrRenderTarget* srcRT = src->asRenderTarget();
1666 if (dstRT && srcRT) {
1667 if (srcRT->numColorSamples() != dstRT->numColorSamples()) {
1668 return false;
1669 }
1670 } else if (dstRT) {
1671 if (dstRT->numColorSamples() > 1) {
1672 return false;
1673 }
1674 } else if (srcRT) {
1675 if (srcRT->numColorSamples() > 1) {
1676 return false;
1677 }
egdaniel17b89252016-04-05 07:23:38 -07001678 }
1679
Michael Jurka3251ed82017-04-05 09:52:55 -07001680 // We require that all vulkan GrSurfaces have been created with transfer_dst and transfer_src
egdaniel17b89252016-04-05 07:23:38 -07001681 // as image usage flags.
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001682 if (srcOrigin == dstOrigin &&
egdaniel17b89252016-04-05 07:23:38 -07001683 GrBytesPerPixel(src->config()) == GrBytesPerPixel(dst->config())) {
Greg Daniel164a9f02016-02-22 09:56:40 -05001684 return true;
1685 }
1686
Greg Daniel164a9f02016-02-22 09:56:40 -05001687 return false;
1688}
1689
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001690void GrVkGpu::copySurfaceAsCopyImage(GrSurface* dst, GrSurfaceOrigin dstOrigin,
1691 GrSurface* src, GrSurfaceOrigin srcOrigin,
egdaniel17b89252016-04-05 07:23:38 -07001692 GrVkImage* dstImage,
1693 GrVkImage* srcImage,
Greg Daniel164a9f02016-02-22 09:56:40 -05001694 const SkIRect& srcRect,
1695 const SkIPoint& dstPoint) {
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001696 SkASSERT(can_copy_image(dst, dstOrigin, src, srcOrigin, this));
Greg Daniel164a9f02016-02-22 09:56:40 -05001697
Greg Daniel164a9f02016-02-22 09:56:40 -05001698 // These flags are for flushing/invalidating caches and for the dst image it doesn't matter if
1699 // the cache is flushed since it is only being written to.
egdaniel17b89252016-04-05 07:23:38 -07001700 dstImage->setImageLayout(this,
jvanverth50c46c72016-05-06 12:31:28 -07001701 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
1702 VK_ACCESS_TRANSFER_WRITE_BIT,
1703 VK_PIPELINE_STAGE_TRANSFER_BIT,
1704 false);
Greg Daniel164a9f02016-02-22 09:56:40 -05001705
egdaniel17b89252016-04-05 07:23:38 -07001706 srcImage->setImageLayout(this,
1707 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
jvanverth50c46c72016-05-06 12:31:28 -07001708 VK_ACCESS_TRANSFER_READ_BIT,
1709 VK_PIPELINE_STAGE_TRANSFER_BIT,
egdaniel17b89252016-04-05 07:23:38 -07001710 false);
Greg Daniel164a9f02016-02-22 09:56:40 -05001711
1712 // Flip rect if necessary
1713 SkIRect srcVkRect = srcRect;
1714 int32_t dstY = dstPoint.fY;
1715
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001716 if (kBottomLeft_GrSurfaceOrigin == srcOrigin) {
1717 SkASSERT(kBottomLeft_GrSurfaceOrigin == dstOrigin);
Greg Daniel164a9f02016-02-22 09:56:40 -05001718 srcVkRect.fTop = src->height() - srcRect.fBottom;
1719 srcVkRect.fBottom = src->height() - srcRect.fTop;
1720 dstY = dst->height() - dstPoint.fY - srcVkRect.height();
1721 }
1722
1723 VkImageCopy copyRegion;
1724 memset(&copyRegion, 0, sizeof(VkImageCopy));
1725 copyRegion.srcSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1726 copyRegion.srcOffset = { srcVkRect.fLeft, srcVkRect.fTop, 0 };
1727 copyRegion.dstSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1728 copyRegion.dstOffset = { dstPoint.fX, dstY, 0 };
egdanielc355bc82016-04-27 11:31:59 -07001729 copyRegion.extent = { (uint32_t)srcVkRect.width(), (uint32_t)srcVkRect.height(), 1 };
Greg Daniel164a9f02016-02-22 09:56:40 -05001730
1731 fCurrentCmdBuffer->copyImage(this,
egdaniel17b89252016-04-05 07:23:38 -07001732 srcImage,
Greg Daniel164a9f02016-02-22 09:56:40 -05001733 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
egdaniel17b89252016-04-05 07:23:38 -07001734 dstImage,
Greg Daniel164a9f02016-02-22 09:56:40 -05001735 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
1736 1,
1737 &copyRegion);
jvanverth900bd4a2016-04-29 13:53:12 -07001738
1739 SkIRect dstRect = SkIRect::MakeXYWH(dstPoint.fX, dstPoint.fY,
1740 srcRect.width(), srcRect.height());
Brian Salomon1fabd512018-02-09 09:54:25 -05001741 this->didWriteToSurface(dst, dstOrigin, &dstRect);
Greg Daniel164a9f02016-02-22 09:56:40 -05001742}
1743
egdaniel17b89252016-04-05 07:23:38 -07001744inline bool can_copy_as_blit(const GrSurface* dst,
1745 const GrSurface* src,
1746 const GrVkImage* dstImage,
1747 const GrVkImage* srcImage,
1748 const GrVkGpu* gpu) {
egdaniel66933552016-08-24 07:22:19 -07001749 // We require that all vulkan GrSurfaces have been created with transfer_dst and transfer_src
egdaniel17b89252016-04-05 07:23:38 -07001750 // as image usage flags.
1751 const GrVkCaps& caps = gpu->vkCaps();
1752 if (!caps.configCanBeDstofBlit(dst->config(), dstImage->isLinearTiled()) ||
1753 !caps.configCanBeSrcofBlit(src->config(), srcImage->isLinearTiled())) {
1754 return false;
1755 }
1756
1757 // We cannot blit images that are multisampled. Will need to figure out if we can blit the
1758 // resolved msaa though.
1759 if ((dst->asRenderTarget() && dst->asRenderTarget()->numColorSamples() > 1) ||
1760 (src->asRenderTarget() && src->asRenderTarget()->numColorSamples() > 1)) {
1761 return false;
1762 }
1763
1764 return true;
1765}
1766
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001767void GrVkGpu::copySurfaceAsBlit(GrSurface* dst, GrSurfaceOrigin dstOrigin,
1768 GrSurface* src, GrSurfaceOrigin srcOrigin,
egdaniel17b89252016-04-05 07:23:38 -07001769 GrVkImage* dstImage,
1770 GrVkImage* srcImage,
1771 const SkIRect& srcRect,
1772 const SkIPoint& dstPoint) {
1773 SkASSERT(can_copy_as_blit(dst, src, dstImage, srcImage, this));
1774
egdaniel17b89252016-04-05 07:23:38 -07001775 dstImage->setImageLayout(this,
1776 VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
jvanverth50c46c72016-05-06 12:31:28 -07001777 VK_ACCESS_TRANSFER_WRITE_BIT,
1778 VK_PIPELINE_STAGE_TRANSFER_BIT,
egdaniel17b89252016-04-05 07:23:38 -07001779 false);
1780
egdaniel17b89252016-04-05 07:23:38 -07001781 srcImage->setImageLayout(this,
1782 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
jvanverth50c46c72016-05-06 12:31:28 -07001783 VK_ACCESS_TRANSFER_READ_BIT,
1784 VK_PIPELINE_STAGE_TRANSFER_BIT,
egdaniel17b89252016-04-05 07:23:38 -07001785 false);
1786
1787 // Flip rect if necessary
1788 SkIRect srcVkRect;
egdaniel8af936d2016-04-07 10:17:47 -07001789 srcVkRect.fLeft = srcRect.fLeft;
1790 srcVkRect.fRight = srcRect.fRight;
egdaniel17b89252016-04-05 07:23:38 -07001791 SkIRect dstRect;
1792 dstRect.fLeft = dstPoint.fX;
egdaniel8af936d2016-04-07 10:17:47 -07001793 dstRect.fRight = dstPoint.fX + srcRect.width();
egdaniel17b89252016-04-05 07:23:38 -07001794
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001795 if (kBottomLeft_GrSurfaceOrigin == srcOrigin) {
egdaniel17b89252016-04-05 07:23:38 -07001796 srcVkRect.fTop = src->height() - srcRect.fBottom;
1797 srcVkRect.fBottom = src->height() - srcRect.fTop;
1798 } else {
egdaniel8af936d2016-04-07 10:17:47 -07001799 srcVkRect.fTop = srcRect.fTop;
1800 srcVkRect.fBottom = srcRect.fBottom;
egdaniel17b89252016-04-05 07:23:38 -07001801 }
1802
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001803 if (kBottomLeft_GrSurfaceOrigin == dstOrigin) {
egdaniel17b89252016-04-05 07:23:38 -07001804 dstRect.fTop = dst->height() - dstPoint.fY - srcVkRect.height();
1805 } else {
1806 dstRect.fTop = dstPoint.fY;
1807 }
1808 dstRect.fBottom = dstRect.fTop + srcVkRect.height();
1809
1810 // If we have different origins, we need to flip the top and bottom of the dst rect so that we
1811 // get the correct origintation of the copied data.
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001812 if (srcOrigin != dstOrigin) {
egdaniel17b89252016-04-05 07:23:38 -07001813 SkTSwap(dstRect.fTop, dstRect.fBottom);
1814 }
1815
1816 VkImageBlit blitRegion;
1817 memset(&blitRegion, 0, sizeof(VkImageBlit));
1818 blitRegion.srcSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1819 blitRegion.srcOffsets[0] = { srcVkRect.fLeft, srcVkRect.fTop, 0 };
Greg Daniele76071c2016-11-02 11:57:06 -04001820 blitRegion.srcOffsets[1] = { srcVkRect.fRight, srcVkRect.fBottom, 1 };
egdaniel17b89252016-04-05 07:23:38 -07001821 blitRegion.dstSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
1822 blitRegion.dstOffsets[0] = { dstRect.fLeft, dstRect.fTop, 0 };
Greg Daniele76071c2016-11-02 11:57:06 -04001823 blitRegion.dstOffsets[1] = { dstRect.fRight, dstRect.fBottom, 1 };
egdaniel17b89252016-04-05 07:23:38 -07001824
1825 fCurrentCmdBuffer->blitImage(this,
egdanielb2df0c22016-05-13 11:30:37 -07001826 *srcImage,
1827 *dstImage,
egdaniel17b89252016-04-05 07:23:38 -07001828 1,
1829 &blitRegion,
1830 VK_FILTER_NEAREST); // We never scale so any filter works here
jvanverth900bd4a2016-04-29 13:53:12 -07001831
Brian Salomon1fabd512018-02-09 09:54:25 -05001832 this->didWriteToSurface(dst, dstOrigin, &dstRect);
egdaniel17b89252016-04-05 07:23:38 -07001833}
1834
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001835inline bool can_copy_as_resolve(const GrSurface* dst, GrSurfaceOrigin dstOrigin,
1836 const GrSurface* src, GrSurfaceOrigin srcOrigin,
egdaniel4bcd62e2016-08-31 07:37:31 -07001837 const GrVkGpu* gpu) {
1838 // Our src must be a multisampled render target
Brian Salomonbdecacf2018-02-02 20:32:49 -05001839 if (!src->asRenderTarget() || 1 == src->asRenderTarget()->numColorSamples()) {
egdaniel4bcd62e2016-08-31 07:37:31 -07001840 return false;
1841 }
1842
Greg Daniel7d6f6662017-04-18 16:40:35 -04001843 // The dst must not be a multisampled render target, expect in the case where the dst is the
1844 // resolve texture connected to the msaa src. We check for this in case we are copying a part of
1845 // a surface to a different region in the same surface.
1846 if (dst->asRenderTarget() && dst->asRenderTarget()->numColorSamples() > 1 && dst != src) {
egdaniel4bcd62e2016-08-31 07:37:31 -07001847 return false;
1848 }
1849
1850 // Surfaces must have the same origin.
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001851 if (srcOrigin != dstOrigin) {
egdaniel4bcd62e2016-08-31 07:37:31 -07001852 return false;
1853 }
1854
1855 return true;
1856}
1857
Brian Salomon1fabd512018-02-09 09:54:25 -05001858void GrVkGpu::copySurfaceAsResolve(GrSurface* dst, GrSurfaceOrigin dstOrigin, GrSurface* src,
1859 GrSurfaceOrigin srcOrigin, const SkIRect& origSrcRect,
1860 const SkIPoint& origDstPoint) {
egdaniel4bcd62e2016-08-31 07:37:31 -07001861 GrVkRenderTarget* srcRT = static_cast<GrVkRenderTarget*>(src->asRenderTarget());
Brian Salomon1fabd512018-02-09 09:54:25 -05001862 SkIRect srcRect = origSrcRect;
1863 SkIPoint dstPoint = origDstPoint;
1864 if (kBottomLeft_GrSurfaceOrigin == srcOrigin) {
1865 SkASSERT(kBottomLeft_GrSurfaceOrigin == dstOrigin);
1866 srcRect = {origSrcRect.fLeft, src->height() - origSrcRect.fBottom,
1867 origSrcRect.fRight, src->height() - origSrcRect.fTop};
1868 dstPoint.fY = dst->height() - dstPoint.fY - srcRect.height();
1869 }
1870 this->resolveImage(dst, srcRT, srcRect, dstPoint);
egdaniel4bcd62e2016-08-31 07:37:31 -07001871}
1872
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001873bool GrVkGpu::onCopySurface(GrSurface* dst, GrSurfaceOrigin dstOrigin,
1874 GrSurface* src, GrSurfaceOrigin srcOrigin,
Greg Daniel55fa6472018-03-16 16:13:10 -04001875 const SkIRect& srcRect, const SkIPoint& dstPoint,
1876 bool canDiscardOutsideDstRect) {
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001877 if (can_copy_as_resolve(dst, dstOrigin, src, srcOrigin, this)) {
1878 this->copySurfaceAsResolve(dst, dstOrigin, src, srcOrigin, srcRect, dstPoint);
egdanielec440992016-09-13 09:54:11 -07001879 return true;
egdaniel4bcd62e2016-08-31 07:37:31 -07001880 }
1881
egdanielfd016d72016-09-27 12:13:05 -07001882 if (this->vkCaps().mustSubmitCommandsBeforeCopyOp()) {
1883 this->submitCommandBuffer(GrVkGpu::kSkip_SyncQueue);
1884 }
1885
Greg Daniel55fa6472018-03-16 16:13:10 -04001886 if (fCopyManager.copySurfaceAsDraw(this, dst, dstOrigin, src, srcOrigin, srcRect, dstPoint,
1887 canDiscardOutsideDstRect)) {
Brian Salomon3d86a192018-02-27 16:46:11 -05001888 auto dstRect = srcRect.makeOffset(dstPoint.fX, dstPoint.fY);
1889 this->didWriteToSurface(dst, dstOrigin, &dstRect);
egdanielbc9b2962016-09-27 08:00:53 -07001890 return true;
1891 }
1892
egdaniel17b89252016-04-05 07:23:38 -07001893 GrVkImage* dstImage;
1894 GrVkImage* srcImage;
egdaniel4bcd62e2016-08-31 07:37:31 -07001895 GrRenderTarget* dstRT = dst->asRenderTarget();
1896 if (dstRT) {
1897 GrVkRenderTarget* vkRT = static_cast<GrVkRenderTarget*>(dstRT);
1898 dstImage = vkRT->numColorSamples() > 1 ? vkRT->msaaImage() : vkRT;
1899 } else {
1900 SkASSERT(dst->asTexture());
egdaniel17b89252016-04-05 07:23:38 -07001901 dstImage = static_cast<GrVkTexture*>(dst->asTexture());
egdaniel17b89252016-04-05 07:23:38 -07001902 }
egdaniel4bcd62e2016-08-31 07:37:31 -07001903 GrRenderTarget* srcRT = src->asRenderTarget();
1904 if (srcRT) {
1905 GrVkRenderTarget* vkRT = static_cast<GrVkRenderTarget*>(srcRT);
1906 srcImage = vkRT->numColorSamples() > 1 ? vkRT->msaaImage() : vkRT;
egdaniel17b89252016-04-05 07:23:38 -07001907 } else {
egdaniel4bcd62e2016-08-31 07:37:31 -07001908 SkASSERT(src->asTexture());
1909 srcImage = static_cast<GrVkTexture*>(src->asTexture());
egdaniel17b89252016-04-05 07:23:38 -07001910 }
1911
Brian Osman13dddce2017-05-09 13:19:50 -04001912 // For borrowed textures, we *only* want to copy using draws (to avoid layout changes)
1913 if (srcImage->isBorrowed()) {
1914 return false;
1915 }
1916
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001917 if (can_copy_image(dst, dstOrigin, src, srcOrigin, this)) {
1918 this->copySurfaceAsCopyImage(dst, dstOrigin, src, srcOrigin, dstImage, srcImage,
1919 srcRect, dstPoint);
egdaniel17b89252016-04-05 07:23:38 -07001920 return true;
1921 }
1922
1923 if (can_copy_as_blit(dst, src, dstImage, srcImage, this)) {
Robert Phillipsb0e93a22017-08-29 08:26:54 -04001924 this->copySurfaceAsBlit(dst, dstOrigin, src, srcOrigin, dstImage, srcImage,
1925 srcRect, dstPoint);
Greg Daniel164a9f02016-02-22 09:56:40 -05001926 return true;
1927 }
1928
Greg Daniel164a9f02016-02-22 09:56:40 -05001929 return false;
1930}
1931
Brian Salomon9b009bb2018-02-14 13:53:55 -05001932bool GrVkGpu::onGetReadPixelsInfo(GrSurface* srcSurface, GrSurfaceOrigin srcOrigin, int width,
Brian Salomonc320b152018-02-20 14:05:36 -05001933 int height, size_t rowBytes, GrColorType dstColorType,
Brian Salomon9b009bb2018-02-14 13:53:55 -05001934 DrawPreference* drawPreference,
Greg Daniel164a9f02016-02-22 09:56:40 -05001935 ReadPixelTempDrawInfo* tempDrawInfo) {
Brian Salomonc320b152018-02-20 14:05:36 -05001936 // We don't want to introduce a sRGB conversion if we trigger a draw.
1937 auto dstConfigSRGBEncoded = GrPixelConfigIsSRGBEncoded(srcSurface->config());
Brian Salomon9b009bb2018-02-14 13:53:55 -05001938 if (*drawPreference != kNoDraw_DrawPreference) {
1939 // We assume the base class has only inserted a draw for sRGB reasons. So the
1940 // the temp surface has the config of the dst data. There is no swizzling nor dst config.
1941 // spoofing.
Brian Salomonc320b152018-02-20 14:05:36 -05001942 SkASSERT(tempDrawInfo->fReadColorType == dstColorType);
1943 SkASSERT(GrPixelConfigToColorType(tempDrawInfo->fTempSurfaceDesc.fConfig) == dstColorType);
Brian Salomon9b009bb2018-02-14 13:53:55 -05001944 SkASSERT(tempDrawInfo->fSwizzle == GrSwizzle::RGBA());
Brian Salomonc320b152018-02-20 14:05:36 -05001945 // Don't undo a sRGB conversion introduced by our caller via an intermediate draw.
1946 dstConfigSRGBEncoded = GrPixelConfigIsSRGBEncoded(tempDrawInfo->fTempSurfaceDesc.fConfig);
1947 }
1948 if (GrColorTypeIsAlphaOnly(dstColorType)) {
1949 dstConfigSRGBEncoded = GrSRGBEncoded::kNo;
Brian Salomon9b009bb2018-02-14 13:53:55 -05001950 }
1951
Brian Salomonc320b152018-02-20 14:05:36 -05001952 if (GrPixelConfigToColorType(srcSurface->config()) == dstColorType) {
egdaniel4583ec52016-06-27 12:57:00 -07001953 return true;
Greg Daniel164a9f02016-02-22 09:56:40 -05001954 }
1955
Brian Osman33910292017-04-18 14:38:53 -04001956 // Any config change requires a draw
1957 ElevateDrawPreference(drawPreference, kRequireDraw_DrawPreference);
Brian Salomonc320b152018-02-20 14:05:36 -05001958 tempDrawInfo->fTempSurfaceDesc.fConfig =
1959 GrColorTypeToPixelConfig(dstColorType, dstConfigSRGBEncoded);
1960 tempDrawInfo->fReadColorType = dstColorType;
Greg Daniel164a9f02016-02-22 09:56:40 -05001961
Brian Salomonc320b152018-02-20 14:05:36 -05001962 return kUnknown_GrPixelConfig != tempDrawInfo->fTempSurfaceDesc.fConfig;
Greg Daniel164a9f02016-02-22 09:56:40 -05001963}
1964
Brian Salomon9b009bb2018-02-14 13:53:55 -05001965bool GrVkGpu::onReadPixels(GrSurface* surface, GrSurfaceOrigin origin, int left, int top, int width,
Brian Salomonc320b152018-02-20 14:05:36 -05001966 int height, GrColorType dstColorType, void* buffer, size_t rowBytes) {
1967 if (GrPixelConfigToColorType(surface->config()) != dstColorType) {
Greg Daniel164a9f02016-02-22 09:56:40 -05001968 return false;
1969 }
1970
egdaniel66933552016-08-24 07:22:19 -07001971 GrVkImage* image = nullptr;
1972 GrVkRenderTarget* rt = static_cast<GrVkRenderTarget*>(surface->asRenderTarget());
1973 if (rt) {
1974 // resolve the render target if necessary
1975 switch (rt->getResolveType()) {
1976 case GrVkRenderTarget::kCantResolve_ResolveType:
1977 return false;
1978 case GrVkRenderTarget::kAutoResolves_ResolveType:
1979 break;
1980 case GrVkRenderTarget::kCanResolve_ResolveType:
Brian Salomon1fabd512018-02-09 09:54:25 -05001981 this->internalResolveRenderTarget(rt, false);
egdaniel66933552016-08-24 07:22:19 -07001982 break;
1983 default:
Ben Wagnerb4aab9a2017-08-16 10:53:04 -04001984 SK_ABORT("Unknown resolve type");
egdaniel66933552016-08-24 07:22:19 -07001985 }
1986 image = rt;
1987 } else {
1988 image = static_cast<GrVkTexture*>(surface->asTexture());
1989 }
1990
1991 if (!image) {
Greg Daniel164a9f02016-02-22 09:56:40 -05001992 return false;
1993 }
1994
1995 // Change layout of our target so it can be used as copy
egdaniel66933552016-08-24 07:22:19 -07001996 image->setImageLayout(this,
1997 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
1998 VK_ACCESS_TRANSFER_READ_BIT,
1999 VK_PIPELINE_STAGE_TRANSFER_BIT,
2000 false);
Greg Daniel164a9f02016-02-22 09:56:40 -05002001
Brian Salomonc320b152018-02-20 14:05:36 -05002002 int bpp = GrColorTypeBytesPerPixel(dstColorType);
egdaniel6fa0a912016-09-12 11:51:29 -07002003 size_t tightRowBytes = bpp * width;
Robert Phillipsb0e93a22017-08-29 08:26:54 -04002004 bool flipY = kBottomLeft_GrSurfaceOrigin == origin;
Greg Daniel164a9f02016-02-22 09:56:40 -05002005
Greg Daniel164a9f02016-02-22 09:56:40 -05002006 VkBufferImageCopy region;
2007 memset(&region, 0, sizeof(VkBufferImageCopy));
egdaniel6fa0a912016-09-12 11:51:29 -07002008
2009 bool copyFromOrigin = this->vkCaps().mustDoCopiesFromOrigin();
2010 if (copyFromOrigin) {
2011 region.imageOffset = { 0, 0, 0 };
2012 region.imageExtent = { (uint32_t)(left + width),
2013 (uint32_t)(flipY ? surface->height() - top : top + height),
2014 1
2015 };
2016 } else {
2017 VkOffset3D offset = {
2018 left,
2019 flipY ? surface->height() - top - height : top,
2020 0
2021 };
2022 region.imageOffset = offset;
2023 region.imageExtent = { (uint32_t)width, (uint32_t)height, 1 };
2024 }
2025
2026 size_t transBufferRowBytes = bpp * region.imageExtent.width;
Greg Daniel3cdfa092018-02-26 16:14:10 -05002027 size_t imageRows = bpp * region.imageExtent.height;
egdaniel6fa0a912016-09-12 11:51:29 -07002028 GrVkTransferBuffer* transferBuffer =
Greg Daniel3cdfa092018-02-26 16:14:10 -05002029 static_cast<GrVkTransferBuffer*>(this->createBuffer(transBufferRowBytes * imageRows,
egdaniel6fa0a912016-09-12 11:51:29 -07002030 kXferGpuToCpu_GrBufferType,
2031 kStream_GrAccessPattern));
2032
2033 // Copy the image to a buffer so we can map it to cpu memory
jvanverthdb379092016-07-07 11:18:46 -07002034 region.bufferOffset = transferBuffer->offset();
egdaniel88e8aef2016-06-27 14:34:55 -07002035 region.bufferRowLength = 0; // Forces RowLength to be width. We handle the rowBytes below.
Greg Daniel164a9f02016-02-22 09:56:40 -05002036 region.bufferImageHeight = 0; // Forces height to be tightly packed. Only useful for 3d images.
2037 region.imageSubresource = { VK_IMAGE_ASPECT_COLOR_BIT, 0, 0, 1 };
Greg Daniel164a9f02016-02-22 09:56:40 -05002038
2039 fCurrentCmdBuffer->copyImageToBuffer(this,
egdaniel66933552016-08-24 07:22:19 -07002040 image,
Greg Daniel164a9f02016-02-22 09:56:40 -05002041 VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
2042 transferBuffer,
2043 1,
2044 &region);
2045
2046 // make sure the copy to buffer has finished
2047 transferBuffer->addMemoryBarrier(this,
2048 VK_ACCESS_TRANSFER_WRITE_BIT,
2049 VK_ACCESS_HOST_READ_BIT,
2050 VK_PIPELINE_STAGE_TRANSFER_BIT,
2051 VK_PIPELINE_STAGE_HOST_BIT,
2052 false);
2053
2054 // We need to submit the current command buffer to the Queue and make sure it finishes before
2055 // we can copy the data out of the buffer.
2056 this->submitCommandBuffer(kForce_SyncQueue);
Greg Daniel88fdee92018-02-24 22:41:50 +00002057 void* mappedMemory = transferBuffer->map();
Greg Daniele35a99e2018-03-02 11:44:22 -05002058 const GrVkAlloc& transAlloc = transferBuffer->alloc();
2059 GrVkMemory::InvalidateMappedAlloc(this, transAlloc, transAlloc.fOffset, VK_WHOLE_SIZE);
Greg Daniel164a9f02016-02-22 09:56:40 -05002060
egdaniel6fa0a912016-09-12 11:51:29 -07002061 if (copyFromOrigin) {
2062 uint32_t skipRows = region.imageExtent.height - height;
2063 mappedMemory = (char*)mappedMemory + transBufferRowBytes * skipRows + bpp * left;
2064 }
2065
egdaniel88e8aef2016-06-27 14:34:55 -07002066 if (flipY) {
2067 const char* srcRow = reinterpret_cast<const char*>(mappedMemory);
2068 char* dstRow = reinterpret_cast<char*>(buffer)+(height - 1) * rowBytes;
2069 for (int y = 0; y < height; y++) {
2070 memcpy(dstRow, srcRow, tightRowBytes);
egdaniel6fa0a912016-09-12 11:51:29 -07002071 srcRow += transBufferRowBytes;
egdaniel88e8aef2016-06-27 14:34:55 -07002072 dstRow -= rowBytes;
2073 }
2074 } else {
Matt Sarettcf3525a2017-01-24 12:43:41 -05002075 SkRectMemcpy(buffer, rowBytes, mappedMemory, transBufferRowBytes, tightRowBytes, height);
egdaniel88e8aef2016-06-27 14:34:55 -07002076 }
Greg Daniel164a9f02016-02-22 09:56:40 -05002077
2078 transferBuffer->unmap();
2079 transferBuffer->unref();
Greg Daniel164a9f02016-02-22 09:56:40 -05002080 return true;
2081}
egdaniel066df7c2016-06-08 14:02:27 -07002082
egdaniel27bb2842016-07-07 11:58:35 -07002083// The RenderArea bounds we pass into BeginRenderPass must have a start x value that is a multiple
2084// of the granularity. The width must also be a multiple of the granularity or eaqual to the width
2085// the the entire attachment. Similar requirements for the y and height components.
2086void adjust_bounds_to_granularity(SkIRect* dstBounds, const SkIRect& srcBounds,
2087 const VkExtent2D& granularity, int maxWidth, int maxHeight) {
2088 // Adjust Width
egdanield5797b32016-09-20 12:57:45 -07002089 if ((0 != granularity.width && 1 != granularity.width)) {
2090 // Start with the right side of rect so we know if we end up going pass the maxWidth.
2091 int rightAdj = srcBounds.fRight % granularity.width;
2092 if (rightAdj != 0) {
2093 rightAdj = granularity.width - rightAdj;
2094 }
2095 dstBounds->fRight = srcBounds.fRight + rightAdj;
2096 if (dstBounds->fRight > maxWidth) {
2097 dstBounds->fRight = maxWidth;
2098 dstBounds->fLeft = 0;
2099 } else {
2100 dstBounds->fLeft = srcBounds.fLeft - srcBounds.fLeft % granularity.width;
2101 }
egdaniel27bb2842016-07-07 11:58:35 -07002102 } else {
egdanield5797b32016-09-20 12:57:45 -07002103 dstBounds->fLeft = srcBounds.fLeft;
2104 dstBounds->fRight = srcBounds.fRight;
egdaniel27bb2842016-07-07 11:58:35 -07002105 }
2106
2107 // Adjust height
egdanield5797b32016-09-20 12:57:45 -07002108 if ((0 != granularity.height && 1 != granularity.height)) {
2109 // Start with the bottom side of rect so we know if we end up going pass the maxHeight.
2110 int bottomAdj = srcBounds.fBottom % granularity.height;
2111 if (bottomAdj != 0) {
2112 bottomAdj = granularity.height - bottomAdj;
2113 }
2114 dstBounds->fBottom = srcBounds.fBottom + bottomAdj;
2115 if (dstBounds->fBottom > maxHeight) {
2116 dstBounds->fBottom = maxHeight;
2117 dstBounds->fTop = 0;
2118 } else {
2119 dstBounds->fTop = srcBounds.fTop - srcBounds.fTop % granularity.height;
2120 }
egdaniel27bb2842016-07-07 11:58:35 -07002121 } else {
egdanield5797b32016-09-20 12:57:45 -07002122 dstBounds->fTop = srcBounds.fTop;
2123 dstBounds->fBottom = srcBounds.fBottom;
egdaniel27bb2842016-07-07 11:58:35 -07002124 }
2125}
2126
Greg Daniel22bc8652017-03-22 15:45:43 -04002127void GrVkGpu::submitSecondaryCommandBuffer(const SkTArray<GrVkSecondaryCommandBuffer*>& buffers,
egdaniel9cb63402016-06-23 08:37:05 -07002128 const GrVkRenderPass* renderPass,
2129 const VkClearValue* colorClear,
Robert Phillipsb0e93a22017-08-29 08:26:54 -04002130 GrVkRenderTarget* target, GrSurfaceOrigin origin,
egdaniel9cb63402016-06-23 08:37:05 -07002131 const SkIRect& bounds) {
egdaniele7d1b242016-07-01 08:06:45 -07002132 const SkIRect* pBounds = &bounds;
2133 SkIRect flippedBounds;
Robert Phillipsb0e93a22017-08-29 08:26:54 -04002134 if (kBottomLeft_GrSurfaceOrigin == origin) {
egdaniele7d1b242016-07-01 08:06:45 -07002135 flippedBounds = bounds;
2136 flippedBounds.fTop = target->height() - bounds.fBottom;
2137 flippedBounds.fBottom = target->height() - bounds.fTop;
2138 pBounds = &flippedBounds;
2139 }
2140
egdaniel27bb2842016-07-07 11:58:35 -07002141 // The bounds we use for the render pass should be of the granularity supported
2142 // by the device.
2143 const VkExtent2D& granularity = renderPass->granularity();
2144 SkIRect adjustedBounds;
2145 if ((0 != granularity.width && 1 != granularity.width) ||
2146 (0 != granularity.height && 1 != granularity.height)) {
2147 adjust_bounds_to_granularity(&adjustedBounds, *pBounds, granularity,
2148 target->width(), target->height());
2149 pBounds = &adjustedBounds;
2150 }
2151
Robert Phillips95214472017-08-08 18:00:03 -04002152#ifdef SK_DEBUG
2153 uint32_t index;
2154 bool result = renderPass->colorAttachmentIndex(&index);
2155 SkASSERT(result && 0 == index);
2156 result = renderPass->stencilAttachmentIndex(&index);
2157 if (result) {
2158 SkASSERT(1 == index);
2159 }
2160#endif
2161 VkClearValue clears[2];
2162 clears[0].color = colorClear->color;
Robert Phillips8c326e92017-08-10 13:50:17 -04002163 clears[1].depthStencil.depth = 0.0f;
2164 clears[1].depthStencil.stencil = 0;
Robert Phillips95214472017-08-08 18:00:03 -04002165
2166 fCurrentCmdBuffer->beginRenderPass(this, renderPass, clears, *target, *pBounds, true);
Greg Daniel22bc8652017-03-22 15:45:43 -04002167 for (int i = 0; i < buffers.count(); ++i) {
2168 fCurrentCmdBuffer->executeCommands(this, buffers[i]);
2169 }
Greg Daniel164a9f02016-02-22 09:56:40 -05002170 fCurrentCmdBuffer->endRenderPass(this);
egdaniel66933552016-08-24 07:22:19 -07002171
Brian Salomon1fabd512018-02-09 09:54:25 -05002172 this->didWriteToSurface(target, origin, &bounds);
Greg Daniel164a9f02016-02-22 09:56:40 -05002173}
egdaniel9cb63402016-06-23 08:37:05 -07002174
Greg Daniel6be35232017-03-01 17:01:09 -05002175GrFence SK_WARN_UNUSED_RESULT GrVkGpu::insertFence() {
jvanverth84741b32016-09-30 08:39:02 -07002176 VkFenceCreateInfo createInfo;
2177 memset(&createInfo, 0, sizeof(VkFenceCreateInfo));
2178 createInfo.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO;
2179 createInfo.pNext = nullptr;
2180 createInfo.flags = 0;
2181 VkFence fence = VK_NULL_HANDLE;
Greg Daniel6be35232017-03-01 17:01:09 -05002182
2183 VK_CALL_ERRCHECK(CreateFence(this->device(), &createInfo, nullptr, &fence));
2184 VK_CALL(QueueSubmit(this->queue(), 0, nullptr, fence));
2185
2186 GR_STATIC_ASSERT(sizeof(GrFence) >= sizeof(VkFence));
jvanverth84741b32016-09-30 08:39:02 -07002187 return (GrFence)fence;
2188}
2189
Greg Daniel6be35232017-03-01 17:01:09 -05002190bool GrVkGpu::waitFence(GrFence fence, uint64_t timeout) {
2191 SkASSERT(VK_NULL_HANDLE != (VkFence)fence);
2192
2193 VkResult result = VK_CALL(WaitForFences(this->device(), 1, (VkFence*)&fence, VK_TRUE, timeout));
jvanverth84741b32016-09-30 08:39:02 -07002194 return (VK_SUCCESS == result);
2195}
2196
2197void GrVkGpu::deleteFence(GrFence fence) const {
Greg Daniel6be35232017-03-01 17:01:09 -05002198 VK_CALL(DestroyFence(this->device(), (VkFence)fence, nullptr));
2199}
2200
Greg Daniela5cb7812017-06-16 09:45:32 -04002201sk_sp<GrSemaphore> SK_WARN_UNUSED_RESULT GrVkGpu::makeSemaphore(bool isOwned) {
2202 return GrVkSemaphore::Make(this, isOwned);
Greg Daniel6be35232017-03-01 17:01:09 -05002203}
2204
Greg Daniel48661b82018-01-22 16:11:35 -05002205sk_sp<GrSemaphore> GrVkGpu::wrapBackendSemaphore(const GrBackendSemaphore& semaphore,
2206 GrResourceProvider::SemaphoreWrapType wrapType,
2207 GrWrapOwnership ownership) {
2208 return GrVkSemaphore::MakeWrapped(this, semaphore.vkSemaphore(), wrapType, ownership);
Greg Daniela5cb7812017-06-16 09:45:32 -04002209}
2210
Greg Daniel48661b82018-01-22 16:11:35 -05002211void GrVkGpu::insertSemaphore(sk_sp<GrSemaphore> semaphore, bool flush) {
Greg Daniel6be35232017-03-01 17:01:09 -05002212 GrVkSemaphore* vkSem = static_cast<GrVkSemaphore*>(semaphore.get());
2213
Greg Daniel48661b82018-01-22 16:11:35 -05002214 GrVkSemaphore::Resource* resource = vkSem->getResource();
2215 if (resource->shouldSignal()) {
Greg Daniel17b7c052018-01-09 13:55:33 -05002216 resource->ref();
2217 fSemaphoresToSignal.push_back(resource);
2218 }
Greg Daniela5cb7812017-06-16 09:45:32 -04002219
2220 if (flush) {
2221 this->submitCommandBuffer(kSkip_SyncQueue);
2222 }
Greg Daniel6be35232017-03-01 17:01:09 -05002223}
2224
Greg Daniel48661b82018-01-22 16:11:35 -05002225void GrVkGpu::waitSemaphore(sk_sp<GrSemaphore> semaphore) {
Greg Daniel6be35232017-03-01 17:01:09 -05002226 GrVkSemaphore* vkSem = static_cast<GrVkSemaphore*>(semaphore.get());
2227
Greg Daniel48661b82018-01-22 16:11:35 -05002228 GrVkSemaphore::Resource* resource = vkSem->getResource();
2229 if (resource->shouldWait()) {
2230 resource->ref();
2231 fSemaphoresToWaitOn.push_back(resource);
2232 }
jvanverth84741b32016-09-30 08:39:02 -07002233}
Brian Osman13dddce2017-05-09 13:19:50 -04002234
2235sk_sp<GrSemaphore> GrVkGpu::prepareTextureForCrossContextUsage(GrTexture* texture) {
2236 SkASSERT(texture);
2237 GrVkTexture* vkTexture = static_cast<GrVkTexture*>(texture);
2238 vkTexture->setImageLayout(this,
2239 VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL,
2240 VK_ACCESS_SHADER_READ_BIT,
2241 VK_PIPELINE_STAGE_ALL_GRAPHICS_BIT,
2242 false);
2243 this->submitCommandBuffer(kSkip_SyncQueue);
2244
2245 // The image layout change serves as a barrier, so no semaphore is needed
2246 return nullptr;
2247}
Greg Danielf5d87582017-12-18 14:48:15 -05002248