diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /gfx/layers/D3D11TextureIMFSampleImage.h | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | gfx/layers/D3D11TextureIMFSampleImage.h | 107 |
1 files changed, 107 insertions, 0 deletions
diff --git a/gfx/layers/D3D11TextureIMFSampleImage.h b/gfx/layers/D3D11TextureIMFSampleImage.h new file mode 100644 index 0000000000..4077d386aa --- /dev/null +++ b/gfx/layers/D3D11TextureIMFSampleImage.h @@ -0,0 +1,107 @@ +/* -*- 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/. */ + +#ifndef GFX_D311_TEXTURE_IMF_SAMPLE_IMAGE_H +#define GFX_D311_TEXTURE_IMF_SAMPLE_IMAGE_H + +#include "ImageContainer.h" +#include "mozilla/RefPtr.h" +#include "mozilla/gfx/Types.h" +#include "mozilla/layers/TextureClient.h" +#include "mozilla/layers/TextureD3D11.h" +#include "mozilla/ThreadSafeWeakPtr.h" + +struct ID3D11Texture2D; +struct IMFSample; + +namespace mozilla { +namespace gl { +class GLBlitHelper; +} +namespace layers { + +class IMFSampleWrapper : public SupportsThreadSafeWeakPtr<IMFSampleWrapper> { + public: + MOZ_DECLARE_REFCOUNTED_TYPENAME(IMFSampleWrapper) + + static RefPtr<IMFSampleWrapper> Create(IMFSample* aVideoSample); + virtual ~IMFSampleWrapper(); + void ClearVideoSample(); + + protected: + explicit IMFSampleWrapper(IMFSample* aVideoSample); + + RefPtr<IMFSample> mVideoSample; +}; + +class IMFSampleUsageInfo final { + public: + NS_INLINE_DECL_THREADSAFE_REFCOUNTING(IMFSampleUsageInfo) + + IMFSampleUsageInfo() = default; + + bool SupportsZeroCopyNV12Texture() { return mSupportsZeroCopyNV12Texture; } + void DisableZeroCopyNV12Texture() { mSupportsZeroCopyNV12Texture = false; } + + protected: + ~IMFSampleUsageInfo() = default; + + Atomic<bool> mSupportsZeroCopyNV12Texture{true}; +}; + +// Image class that wraps ID3D11Texture2D of IMFSample +// Expected to be used in GPU process. +class D3D11TextureIMFSampleImage final : public Image { + public: + D3D11TextureIMFSampleImage(IMFSample* aVideoSample, ID3D11Texture2D* aTexture, + uint32_t aArrayIndex, const gfx::IntSize& aSize, + const gfx::IntRect& aRect, + gfx::ColorSpace2 aColorSpace, + gfx::ColorRange aColorRange); + virtual ~D3D11TextureIMFSampleImage() = default; + + void AllocateTextureClient(KnowsCompositor* aKnowsCompositor, + RefPtr<IMFSampleUsageInfo> aUsageInfo); + + gfx::IntSize GetSize() const override; + already_AddRefed<gfx::SourceSurface> GetAsSourceSurface() override; + TextureClient* GetTextureClient(KnowsCompositor* aKnowsCompositor) override; + gfx::IntRect GetPictureRect() const override { return mPictureRect; } + + ID3D11Texture2D* GetTexture() const; + RefPtr<IMFSampleWrapper> GetIMFSampleWrapper(); + + gfx::ColorRange GetColorRange() const { return mColorRange; } + + private: + friend class gl::GLBlitHelper; + D3D11TextureData* GetData() const { + if (!mTextureClient) { + return nullptr; + } + return mTextureClient->GetInternalData()->AsD3D11TextureData(); + } + + // When ref of IMFSample is held, its ID3D11Texture2D is not reused by + // IMFTransform. + RefPtr<IMFSampleWrapper> mVideoSample; + RefPtr<ID3D11Texture2D> mTexture; + + public: + const uint32_t mArrayIndex; + const gfx::IntSize mSize; + const gfx::IntRect mPictureRect; + const gfx::ColorSpace2 mColorSpace; + const gfx::ColorRange mColorRange; + + private: + RefPtr<TextureClient> mTextureClient; +}; + +} // namespace layers +} // namespace mozilla + +#endif // GFX_D311_TEXTURE_IMF_SAMPLE_IMAGE_H |