mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-19 14:56:36 +00:00
15e11ff303
Original commit message from CVS: * sys/dshowvideosink/dshowvideofakesrc.cpp: * sys/dshowvideosink/dshowvideofakesrc.h: * sys/dshowvideosink/dshowvideosink.cpp: * sys/dshowvideosink/dshowvideosink.h: Fix up copyright notice on new plugin.
281 lines
9.2 KiB
C++
281 lines
9.2 KiB
C++
/* GStreamer
|
|
* Copyright (C) 2008 Pioneers of the Inevitable <songbird@songbirdnest.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include "dshowvideofakesrc.h"
|
|
|
|
// {A0A5CF33-BD0C-4158-9A56-3011DEE3AF6B}
|
|
const GUID CLSID_VideoFakeSrc =
|
|
{ 0xa0a5cf33, 0xbd0c, 0x4158, { 0x9a, 0x56, 0x30, 0x11, 0xde, 0xe3, 0xaf, 0x6b } };
|
|
|
|
/* output pin*/
|
|
VideoFakeSrcPin::VideoFakeSrcPin (CBaseFilter *pFilter, CCritSec *sec, HRESULT *hres):
|
|
CBaseOutputPin("VideoFakeSrcPin", pFilter, sec, hres, L"output")
|
|
{
|
|
}
|
|
|
|
VideoFakeSrcPin::~VideoFakeSrcPin()
|
|
{
|
|
}
|
|
|
|
HRESULT VideoFakeSrcPin::GetMediaType(int iPosition, CMediaType *pMediaType)
|
|
{
|
|
GST_DEBUG ("GetMediaType(%d) called", iPosition);
|
|
if(iPosition == 0) {
|
|
*pMediaType = m_MediaType;
|
|
return S_OK;
|
|
}
|
|
|
|
return VFW_S_NO_MORE_ITEMS;
|
|
}
|
|
|
|
/* This seems to be called to notify us of the actual media type being used,
|
|
* even though SetMediaType isn't called. How bizarre! */
|
|
HRESULT VideoFakeSrcPin::CheckMediaType(const CMediaType *pmt)
|
|
{
|
|
GST_DEBUG ("CheckMediaType called: %p", pmt);
|
|
|
|
/* The video renderer will request a different stride, which we must accept.
|
|
* So, we accept arbitrary strides (and do memcpy() to convert if needed),
|
|
* and require the rest of the media type to match
|
|
*/
|
|
if (IsEqualGUID(pmt->majortype,m_MediaType.majortype) &&
|
|
IsEqualGUID(pmt->subtype,m_MediaType.subtype) &&
|
|
IsEqualGUID(pmt->formattype,m_MediaType.formattype) &&
|
|
pmt->cbFormat >= m_MediaType.cbFormat)
|
|
{
|
|
if (IsEqualGUID(pmt->formattype, FORMAT_VideoInfo)) {
|
|
VIDEOINFOHEADER *newvh = (VIDEOINFOHEADER *)pmt->pbFormat;
|
|
VIDEOINFOHEADER *curvh = (VIDEOINFOHEADER *)m_MediaType.pbFormat;
|
|
|
|
if ((memcmp ((void *)&newvh->rcSource, (void *)&curvh->rcSource, sizeof (RECT)) == 0) &&
|
|
(memcmp ((void *)&newvh->rcTarget, (void *)&curvh->rcTarget, sizeof (RECT)) == 0) &&
|
|
(newvh->bmiHeader.biCompression == curvh->bmiHeader.biCompression) &&
|
|
(newvh->bmiHeader.biHeight == curvh->bmiHeader.biHeight) &&
|
|
(newvh->bmiHeader.biWidth >= curvh->bmiHeader.biWidth))
|
|
{
|
|
GST_DEBUG ("CheckMediaType has same media type, width %d (%d image)", newvh->bmiHeader.biWidth, curvh->bmiHeader.biWidth);
|
|
|
|
/* OK, compatible! */
|
|
return S_OK;
|
|
}
|
|
else {
|
|
GST_WARNING ("Looked similar, but aren't...");
|
|
}
|
|
}
|
|
|
|
}
|
|
GST_WARNING ("Different media types, FAILING!");
|
|
return S_FALSE;
|
|
}
|
|
|
|
HRESULT VideoFakeSrcPin::DecideBufferSize (IMemAllocator *pAlloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
|
|
{
|
|
ALLOCATOR_PROPERTIES properties;
|
|
GST_DEBUG ("Required allocator properties: %d, %d, %d, %d",
|
|
ppropInputRequest->cbAlign, ppropInputRequest->cbBuffer,
|
|
ppropInputRequest->cbPrefix, ppropInputRequest->cBuffers);
|
|
|
|
ppropInputRequest->cbBuffer = m_SampleSize;
|
|
ppropInputRequest->cBuffers = 1;
|
|
|
|
/* First set the buffer descriptions we're interested in */
|
|
HRESULT hres = pAlloc->SetProperties(ppropInputRequest, &properties);
|
|
GST_DEBUG ("Actual Allocator properties: %d, %d, %d, %d",
|
|
properties.cbAlign, properties.cbBuffer,
|
|
properties.cbPrefix, properties.cBuffers);
|
|
|
|
/* Then actually allocate the buffers */
|
|
pAlloc->Commit();
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP
|
|
VideoFakeSrcPin::Notify(IBaseFilter * pSender, Quality q)
|
|
{
|
|
/* Implementing this usefully is not required, but the base class
|
|
* has an assertion here... */
|
|
/* TODO: Map this to GStreamer QOS events? */
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP VideoFakeSrcPin::SetMediaType (AM_MEDIA_TYPE *pmt)
|
|
{
|
|
m_MediaType.Set (*pmt);
|
|
m_SampleSize = m_MediaType.GetSampleSize();
|
|
|
|
GST_DEBUG ("SetMediaType called. SampleSize is %d", m_SampleSize);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/* If the destination buffer is a different shape (strides, etc.) from the source
|
|
* buffer, we have to copy. Do that here, for supported video formats.
|
|
*
|
|
* TODO: When possible (when these things DON'T differ), we should buffer-alloc the
|
|
* final output buffer, and not do this copy */
|
|
STDMETHODIMP VideoFakeSrcPin::CopyToDestinationBuffer (byte *srcbuf, byte *dstbuf)
|
|
{
|
|
VIDEOINFOHEADER *vh = (VIDEOINFOHEADER *)m_MediaType.pbFormat;
|
|
GST_DEBUG ("Rendering a frame");
|
|
|
|
byte *src, *dst;
|
|
int dststride, srcstride, rows;
|
|
guint32 fourcc = vh->bmiHeader.biCompression;
|
|
|
|
/* biHeight is always negative; we don't want that. */
|
|
int height = ABS (vh->bmiHeader.biHeight);
|
|
int width = vh->bmiHeader.biWidth;
|
|
|
|
/* YUY2 is the preferred layout for DirectShow, so we will probably get this
|
|
* most of the time */
|
|
if ((fourcc == GST_MAKE_FOURCC ('Y', 'U', 'Y', '2')) ||
|
|
(fourcc == GST_MAKE_FOURCC ('Y', 'U', 'Y', 'V')) ||
|
|
(fourcc == GST_MAKE_FOURCC ('U', 'Y', 'V', 'Y')))
|
|
{
|
|
/* Nice and simple */
|
|
int srcstride = GST_ROUND_UP_4 (vh->rcSource.right * 2);
|
|
int dststride = width * 2;
|
|
|
|
for (int i = 0; i < height; i++) {
|
|
memcpy (dstbuf + dststride * i, srcbuf + srcstride * i, srcstride);
|
|
}
|
|
}
|
|
else if (fourcc == GST_MAKE_FOURCC ('Y', 'V', '1', '2')) {
|
|
for (int component = 0; component < 3; component++) {
|
|
// TODO: Get format properly rather than hard-coding it. Use gst_video_* APIs *?
|
|
if (component == 0) {
|
|
srcstride = GST_ROUND_UP_4 (vh->rcSource.right);
|
|
src = srcbuf;
|
|
}
|
|
else {
|
|
srcstride = GST_ROUND_UP_4 ( GST_ROUND_UP_2 (vh->rcSource.right) / 2);
|
|
if (component == 1)
|
|
src = srcbuf + GST_ROUND_UP_4 (vh->rcSource.right) * GST_ROUND_UP_2 (vh->rcSource.bottom);
|
|
else
|
|
src = srcbuf + GST_ROUND_UP_4 (vh->rcSource.right) * GST_ROUND_UP_2 (vh->rcSource.bottom) +
|
|
srcstride * (GST_ROUND_UP_2 (vh->rcSource.bottom) / 2);
|
|
}
|
|
|
|
/* Is there a better way to do this? This is ICK! */
|
|
if (component == 0) {
|
|
dststride = width;
|
|
dst = dstbuf;
|
|
rows = height;
|
|
} else if (component == 1) {
|
|
dststride = width / 2;
|
|
dst = dstbuf + width * height;
|
|
rows = height/2;
|
|
}
|
|
else {
|
|
dststride = width / 2;
|
|
dst = dstbuf + width * height +
|
|
width/2 * height/2;
|
|
rows = height/2;
|
|
}
|
|
|
|
for (int i = 0; i < rows; i++) {
|
|
memcpy (dst + i * dststride, src + i * srcstride, srcstride);
|
|
}
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
GstFlowReturn VideoFakeSrcPin::PushBuffer(GstBuffer *buffer)
|
|
{
|
|
IMediaSample *pSample = NULL;
|
|
|
|
byte *data = GST_BUFFER_DATA (buffer);
|
|
|
|
/* TODO: Use more of the arguments here? */
|
|
HRESULT hres = GetDeliveryBuffer(&pSample, NULL, NULL, 0);
|
|
if (SUCCEEDED (hres))
|
|
{
|
|
BYTE *sample_buffer;
|
|
AM_MEDIA_TYPE *mediatype;
|
|
|
|
pSample->GetPointer(&sample_buffer);
|
|
pSample->GetMediaType(&mediatype);
|
|
if (mediatype)
|
|
SetMediaType (mediatype);
|
|
|
|
if(sample_buffer)
|
|
{
|
|
/* Copy to the destination stride.
|
|
* This is not just a simple memcpy because of the different strides.
|
|
* TODO: optimise for the same-stride case and avoid the copy entirely.
|
|
*/
|
|
CopyToDestinationBuffer (data, sample_buffer);
|
|
}
|
|
|
|
pSample->SetDiscontinuity(FALSE); /* Decoded frame; unimportant */
|
|
pSample->SetSyncPoint(TRUE); /* Decoded frame; always a valid syncpoint */
|
|
pSample->SetPreroll(FALSE); /* For non-displayed frames.
|
|
Not used in GStreamer */
|
|
|
|
/* Disable synchronising on this sample. We instead let GStreamer handle
|
|
* this at a higher level, inside BaseSink. */
|
|
pSample->SetTime(NULL, NULL);
|
|
|
|
hres = Deliver(pSample);
|
|
pSample->Release();
|
|
|
|
if (SUCCEEDED (hres))
|
|
return GST_FLOW_OK;
|
|
else if (hres == VFW_E_NOT_CONNECTED)
|
|
return GST_FLOW_NOT_LINKED;
|
|
else
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
else {
|
|
GST_WARNING ("Could not get sample for delivery to sink: %x", hres);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP VideoFakeSrcPin::Flush ()
|
|
{
|
|
DeliverBeginFlush();
|
|
DeliverEndFlush();
|
|
return S_OK;
|
|
}
|
|
|
|
VideoFakeSrc::VideoFakeSrc() : CBaseFilter("VideoFakeSrc", NULL, &m_critsec, CLSID_VideoFakeSrc)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
m_pOutputPin = new VideoFakeSrcPin ((CSource *)this, &m_critsec, &hr);
|
|
}
|
|
|
|
int VideoFakeSrc::GetPinCount()
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
CBasePin *VideoFakeSrc::GetPin(int n)
|
|
{
|
|
return (CBasePin *)m_pOutputPin;
|
|
}
|
|
|
|
VideoFakeSrcPin *VideoFakeSrc::GetOutputPin()
|
|
{
|
|
return m_pOutputPin;
|
|
}
|