1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
|
/*
* Copyright (C) 2021-2023 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "WebGPUQueueImpl.h"
#if HAVE(WEBGPU_IMPLEMENTATION)
#include "WebGPUBufferImpl.h"
#include "WebGPUCommandBufferImpl.h"
#include "WebGPUConvertToBackingContext.h"
#include "WebGPUTextureImpl.h"
#include <WebGPU/WebGPUExt.h>
#include <wtf/BlockPtr.h>
namespace WebCore::WebGPU {
QueueImpl::QueueImpl(WebGPUPtr<WGPUQueue>&& queue, ConvertToBackingContext& convertToBackingContext)
: m_backing(WTFMove(queue))
, m_convertToBackingContext(convertToBackingContext)
{
}
QueueImpl::~QueueImpl() = default;
void QueueImpl::submit(Vector<std::reference_wrapper<CommandBuffer>>&& commandBuffers)
{
auto backingCommandBuffers = commandBuffers.map([&convertToBackingContext = m_convertToBackingContext.get()](auto commandBuffer) {
return convertToBackingContext.convertToBacking(commandBuffer);
});
wgpuQueueSubmit(m_backing.get(), backingCommandBuffers.size(), backingCommandBuffers.data());
}
static void onSubmittedWorkDoneCallback(WGPUQueueWorkDoneStatus status, void* userdata)
{
auto block = reinterpret_cast<void(^)(WGPUQueueWorkDoneStatus)>(userdata);
block(status);
Block_release(block); // Block_release is matched with Block_copy below in QueueImpl::submit().
}
void QueueImpl::onSubmittedWorkDone(CompletionHandler<void()>&& callback)
{
auto blockPtr = makeBlockPtr([callback = WTFMove(callback)](WGPUQueueWorkDoneStatus) mutable {
callback();
});
wgpuQueueOnSubmittedWorkDone(m_backing.get(), &onSubmittedWorkDoneCallback, Block_copy(blockPtr.get())); // Block_copy is matched with Block_release above in onSubmittedWorkDoneCallback().
}
void QueueImpl::writeBuffer(
const Buffer& buffer,
Size64 bufferOffset,
const void* source,
size_t byteLength,
Size64 dataOffset,
std::optional<Size64> size)
{
// FIXME: Use checked arithmetic and check the cast
wgpuQueueWriteBuffer(m_backing.get(), m_convertToBackingContext->convertToBacking(buffer), bufferOffset, static_cast<const uint8_t*>(source) + dataOffset, static_cast<size_t>(size.value_or(byteLength - dataOffset)));
}
void QueueImpl::writeTexture(
const ImageCopyTexture& destination,
const void* source,
size_t byteLength,
const ImageDataLayout& dataLayout,
const Extent3D& size)
{
WGPUImageCopyTexture backingDestination {
nullptr,
m_convertToBackingContext->convertToBacking(destination.texture),
destination.mipLevel,
destination.origin ? m_convertToBackingContext->convertToBacking(*destination.origin) : WGPUOrigin3D { 0, 0, 0 },
m_convertToBackingContext->convertToBacking(destination.aspect),
};
WGPUTextureDataLayout backingDataLayout {
nullptr,
dataLayout.offset,
dataLayout.bytesPerRow.value_or(0),
dataLayout.rowsPerImage.value_or(1),
};
WGPUExtent3D backingSize = m_convertToBackingContext->convertToBacking(size);
wgpuQueueWriteTexture(m_backing.get(), &backingDestination, source, byteLength, &backingDataLayout, &backingSize);
}
void QueueImpl::copyExternalImageToTexture(
const ImageCopyExternalImage& source,
const ImageCopyTextureTagged& destination,
const Extent3D& copySize)
{
UNUSED_PARAM(source);
UNUSED_PARAM(destination);
UNUSED_PARAM(copySize);
}
void QueueImpl::setLabelInternal(const String& label)
{
wgpuQueueSetLabel(m_backing.get(), label.utf8().data());
}
} // namespace WebCore::WebGPU
#endif // HAVE(WEBGPU_IMPLEMENTATION)
|