实现 IMediaBuffer
[与此页面关联的功能 DirectShow 是旧版功能。 它已被 MediaPlayer、 IMFMediaEngine 和 Media Foundation 中的音频/视频捕获所取代。 这些功能已针对Windows 10和Windows 11进行了优化。 Microsoft 强烈建议新代码尽可能在 Media Foundation 中使用 MediaPlayer、 IMFMediaEngine 和 音频/视频捕获 ,而不是 DirectShow。 如果可能,Microsoft 建议重写使用旧 API 的现有代码以使用新 API。]
在默认 DMO 流式处理模型中,缓冲区通过 IMediaBuffer 接口进行管理。 DMO 的客户端负责实现公开此接口的对象。 IMediaBuffer 接口有三种方法:
- GetBufferAndLength 返回缓冲区 (地址,即保存数据) 的实际内存块,以及缓冲区中任何有效数据的大小。
- GetMaxLength 返回缓冲区的大小。
- SetLength 指定缓冲区中有效数据的长度。
就地处理不需要 IMediaBuffer 接口。 以下代码演示 IMediaBuffer 的最小实现:
// CMediaBuffer class.
#include <dmo.h>
class CMediaBuffer : public IMediaBuffer
{
private:
DWORD m_cbLength;
const DWORD m_cbMaxLength;
LONG m_nRefCount; // Reference count
BYTE *m_pbData;
CMediaBuffer(DWORD cbMaxLength, HRESULT& hr) :
m_nRefCount(1),
m_cbMaxLength(cbMaxLength),
m_cbLength(0),
m_pbData(NULL)
{
m_pbData = new BYTE[cbMaxLength];
if (!m_pbData)
{
hr = E_OUTOFMEMORY;
}
}
~CMediaBuffer()
{
if (m_pbData)
{
delete [] m_pbData;
}
}
public:
// Function to create a new IMediaBuffer object and return
// an AddRef'd interface pointer.
static HRESULT Create(long cbMaxLen, IMediaBuffer **ppBuffer)
{
HRESULT hr = S_OK;
CMediaBuffer *pBuffer = NULL;
if (ppBuffer == NULL)
{
return E_POINTER;
}
pBuffer = new CMediaBuffer(cbMaxLen, hr);
if (pBuffer == NULL)
{
hr = E_OUTOFMEMORY;
}
if (SUCCEEDED(hr))
{
*ppBuffer = pBuffer;
(*ppBuffer)->AddRef();
}
if (pBuffer)
{
pBuffer->Release();
}
return hr;
}
// IUnknown methods.
STDMETHODIMP QueryInterface(REFIID riid, void **ppv)
{
if (ppv == NULL)
{
return E_POINTER;
}
else if (riid == IID_IMediaBuffer || riid == IID_IUnknown)
{
*ppv = static_cast<IMediaBuffer *>(this);
AddRef();
return S_OK;
}
else
{
*ppv = NULL;
return E_NOINTERFACE;
}
}
STDMETHODIMP_(ULONG) AddRef()
{
return InterlockedIncrement(&m_nRefCount);
}
STDMETHODIMP_(ULONG) Release()
{
LONG lRef = InterlockedDecrement(&m_nRefCount);
if (lRef == 0)
{
delete this;
// m_cRef is no longer valid! Return lRef.
}
return lRef;
}
// IMediaBuffer methods.
STDMETHODIMP SetLength(DWORD cbLength)
{
if (cbLength > m_cbMaxLength)
{
return E_INVALIDARG;
}
m_cbLength = cbLength;
return S_OK;
}
STDMETHODIMP GetMaxLength(DWORD *pcbMaxLength)
{
if (pcbMaxLength == NULL)
{
return E_POINTER;
}
*pcbMaxLength = m_cbMaxLength;
return S_OK;
}
STDMETHODIMP GetBufferAndLength(BYTE **ppbBuffer, DWORD *pcbLength)
{
// Either parameter can be NULL, but not both.
if (ppbBuffer == NULL && pcbLength == NULL)
{
return E_POINTER;
}
if (ppbBuffer)
{
*ppbBuffer = m_pbData;
}
if (pcbLength)
{
*pcbLength = m_cbLength;
}
return S_OK;
}
};
相关主题