gecko-dev/gfx/layers/wr/WebRenderTextureHostWrapper...

76 строки
2.8 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "WebRenderTextureHostWrapper.h"
#include "mozilla/layers/AsyncImagePipelineManager.h"
#include "mozilla/layers/WebRenderTextureHost.h"
#include "mozilla/webrender/RenderTextureHostWrapper.h"
#include "mozilla/webrender/RenderThread.h"
#include "mozilla/webrender/WebRenderAPI.h"
#include "mozilla/webrender/RenderThread.h"
namespace mozilla {
namespace layers {
class ScheduleUpdateRenderTextureHost : public wr::NotificationHandler {
public:
ScheduleUpdateRenderTextureHost(uint64_t aSrcExternalImageId,
uint64_t aWrappedExternalImageId)
: mSrcExternalImageId(aSrcExternalImageId),
mWrappedExternalImageId(aWrappedExternalImageId) {}
virtual void Notify(wr::Checkpoint aCheckpoint) override {
if (aCheckpoint == wr::Checkpoint::FrameTexturesUpdated) {
MOZ_ASSERT(wr::RenderThread::IsInRenderThread());
wr::RenderThread::Get()->UpdateRenderTextureHost(mSrcExternalImageId,
mWrappedExternalImageId);
} else {
MOZ_ASSERT(aCheckpoint == wr::Checkpoint::TransactionDropped);
}
}
protected:
uint64_t mSrcExternalImageId;
uint64_t mWrappedExternalImageId;
};
WebRenderTextureHostWrapper::WebRenderTextureHostWrapper(
AsyncImagePipelineManager* aManager)
: mExternalImageId(aManager->GetNextExternalImageId()) {
MOZ_ASSERT(aManager);
MOZ_COUNT_CTOR(WebRenderTextureHostWrapper);
RefPtr<wr::RenderTextureHost> texture = new wr::RenderTextureHostWrapper();
wr::RenderThread::Get()->RegisterExternalImage(wr::AsUint64(mExternalImageId),
texture.forget());
}
WebRenderTextureHostWrapper::~WebRenderTextureHostWrapper() {
MOZ_COUNT_DTOR(WebRenderTextureHostWrapper);
wr::RenderThread::Get()->UnregisterExternalImage(
wr::AsUint64(mExternalImageId));
}
void WebRenderTextureHostWrapper::UpdateWebRenderTextureHost(
wr::TransactionBuilder& aTxn, WebRenderTextureHost* aTextureHost) {
MOZ_ASSERT(aTextureHost);
// AsyncImagePipelineManager is responsible of holding compositable ref of
// wrapped WebRenderTextureHost by using ForwardingTextureHost.
// ScheduleUpdateRenderTextureHost does not need to handle it.
aTxn.Notify(wr::Checkpoint::FrameTexturesUpdated,
MakeUnique<ScheduleUpdateRenderTextureHost>(
wr::AsUint64(mExternalImageId),
wr::AsUint64(aTextureHost->GetExternalImageKey())));
mWrTextureHost = aTextureHost;
}
} // namespace layers
} // namespace mozilla