72f2f9cb75
With CI's NVIDIA GPU, SharedHandle of ID3D11Texture2D of hardware decoded video during no video copy caused rendering problem. When SharedHandle is not used, the rendering problem did not happen. But when video is rendered to WebGL texture, SharedHandle need to be used. In this case, D3D11TextureIMFSampleImage copies original ID3D11Texture2D to a new ID3D11Texture2D and use a shared handled of the copied texture. And no video copy of future video frames are disabled. NoCopyNV12Texture is renamed to ZeroCopyNV12Texture to clarify its meaning. Differential Revision: https://phabricator.services.mozilla.com/D144598
105 lines
3.3 KiB
C++
105 lines
3.3 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/. */
|
|
|
|
#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::YUVColorSpace 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::YUVColorSpace GetYUVColorSpace() const { return mYUVColorSpace; }
|
|
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;
|
|
const uint32_t mArrayIndex;
|
|
const gfx::IntSize mSize;
|
|
const gfx::IntRect mPictureRect;
|
|
const gfx::YUVColorSpace mYUVColorSpace;
|
|
const gfx::ColorRange mColorRange;
|
|
RefPtr<TextureClient> mTextureClient;
|
|
};
|
|
|
|
} // namespace layers
|
|
} // namespace mozilla
|
|
|
|
#endif // GFX_D311_TEXTURE_IMF_SAMPLE_IMAGE_H
|