| /* |
| * Copyright (C) 2021 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 "RemoteImageDecoderAVFManager.h" |
| |
| #if ENABLE(GPU_PROCESS) && HAVE(AVASSETREADER) |
| |
| #include "GPUProcessConnection.h" |
| #include "RemoteImageDecoderAVF.h" |
| #include "RemoteImageDecoderAVFManagerMessages.h" |
| #include "RemoteImageDecoderAVFProxyMessages.h" |
| #include "SharedBufferCopy.h" |
| #include "WebProcess.h" |
| |
| namespace WebKit { |
| |
| using namespace WebCore; |
| |
| RefPtr<RemoteImageDecoderAVF> RemoteImageDecoderAVFManager::createImageDecoder(FragmentedSharedBuffer& data, const String& mimeType, AlphaOption alphaOption, GammaAndColorProfileOption gammaAndColorProfileOption) |
| { |
| std::optional<ImageDecoderIdentifier> imageDecoderIdentifier; |
| if (!ensureGPUProcessConnection().connection().sendSync(Messages::RemoteImageDecoderAVFProxy::CreateDecoder(IPC::SharedBufferCopy(data), mimeType), Messages::RemoteImageDecoderAVFProxy::CreateDecoder::Reply(imageDecoderIdentifier), 0)) |
| return nullptr; |
| |
| if (!imageDecoderIdentifier) |
| return nullptr; |
| |
| auto remoteImageDecoder = RemoteImageDecoderAVF::create(*this, *imageDecoderIdentifier, data, mimeType); |
| m_remoteImageDecoders.add(*imageDecoderIdentifier, remoteImageDecoder); |
| |
| return remoteImageDecoder; |
| } |
| |
| void RemoteImageDecoderAVFManager::deleteRemoteImageDecoder(const ImageDecoderIdentifier& identifier) |
| { |
| m_remoteImageDecoders.take(identifier); |
| if (m_gpuProcessConnection) |
| m_gpuProcessConnection->connection().send(Messages::RemoteImageDecoderAVFProxy::DeleteDecoder(identifier), 0); |
| } |
| |
| RemoteImageDecoderAVFManager::RemoteImageDecoderAVFManager(WebProcess& process) |
| : m_process(process) |
| { |
| } |
| |
| RemoteImageDecoderAVFManager::~RemoteImageDecoderAVFManager() |
| { |
| if (m_gpuProcessConnection) |
| m_gpuProcessConnection->messageReceiverMap().removeMessageReceiver(Messages::RemoteImageDecoderAVFManager::messageReceiverName()); |
| } |
| |
| void RemoteImageDecoderAVFManager::gpuProcessConnectionDidClose(GPUProcessConnection& connection) |
| { |
| ASSERT(m_gpuProcessConnection == &connection); |
| connection.removeClient(*this); |
| m_gpuProcessConnection->messageReceiverMap().removeMessageReceiver(Messages::RemoteImageDecoderAVFManager::messageReceiverName()); |
| m_gpuProcessConnection = nullptr; |
| // FIXME: Do we need to do more when m_remoteImageDecoders is not empty to re-create them? |
| } |
| |
| const char* RemoteImageDecoderAVFManager::supplementName() |
| { |
| return "RemoteImageDecoderAVFManager"; |
| } |
| |
| GPUProcessConnection& RemoteImageDecoderAVFManager::ensureGPUProcessConnection() |
| { |
| if (!m_gpuProcessConnection) { |
| m_gpuProcessConnection = m_process.ensureGPUProcessConnection(); |
| m_gpuProcessConnection->addClient(*this); |
| m_gpuProcessConnection->messageReceiverMap().addMessageReceiver(Messages::RemoteImageDecoderAVFManager::messageReceiverName(), *this); |
| } |
| return *m_gpuProcessConnection; |
| } |
| |
| void RemoteImageDecoderAVFManager::setUseGPUProcess(bool useGPUProcess) |
| { |
| if (!useGPUProcess) { |
| ImageDecoder::resetFactories(); |
| return; |
| } |
| |
| ImageDecoder::clearFactories(); |
| ImageDecoder::installFactory({ |
| RemoteImageDecoderAVF::supportsMediaType, |
| RemoteImageDecoderAVF::canDecodeType, |
| [this](FragmentedSharedBuffer& data, const String& mimeType, AlphaOption alphaOption, GammaAndColorProfileOption gammaAndColorProfileOption) { |
| return createImageDecoder(data, mimeType, alphaOption, gammaAndColorProfileOption); |
| } |
| }); |
| } |
| |
| void RemoteImageDecoderAVFManager::encodedDataStatusChanged(const ImageDecoderIdentifier& identifier, size_t frameCount, const WebCore::IntSize& size, bool hasTrack) |
| { |
| if (!m_remoteImageDecoders.contains(identifier)) |
| return; |
| |
| auto remoteImageDecoder = m_remoteImageDecoders.get(identifier); |
| if (!remoteImageDecoder) |
| return; |
| |
| remoteImageDecoder->encodedDataStatusChanged(frameCount, size, hasTrack); |
| } |
| |
| } |
| |
| #endif |