smartteefilter.c File Reference

#include "qcap_private.h"

Include dependency graph for smartteefilter.c:

Go to the source code of this file.

Classes
struct	SmartTeeFilter

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (quartz)
static SmartTeeFilter *	impl_from_strmbase_filter (struct strmbase_filter *filter)
static SmartTeeFilter *	impl_from_strmbase_pin (struct strmbase_pin *pin)
static struct strmbase_pin *	smart_tee_get_pin (struct strmbase_filter *iface, unsigned int index)
static void	smart_tee_destroy (struct strmbase_filter *iface)
static HRESULT	smart_tee_wait_state (struct strmbase_filter *iface, DWORD timeout)
static HRESULT	sink_query_accept (struct strmbase_pin *base, const AM_MEDIA_TYPE *pmt)
static HRESULT	sink_get_media_type (struct strmbase_pin *base, unsigned int iPosition, AM_MEDIA_TYPE *amt)
static HRESULT	sink_query_interface (struct strmbase_pin *iface, REFIID iid, void **out)
static HRESULT	copy_sample (IMediaSample *inputSample, IMemAllocator *allocator, IMediaSample **pOutputSample)
static HRESULT WINAPI	SmartTeeFilterInput_Receive (struct strmbase_sink *base, IMediaSample *inputSample)
static HRESULT	capture_query_accept (struct strmbase_pin *base, const AM_MEDIA_TYPE *amt)
static HRESULT	source_get_media_type (struct strmbase_pin *iface, unsigned int index, AM_MEDIA_TYPE *mt)
static HRESULT WINAPI	SmartTeeFilterCapture_DecideAllocator (struct strmbase_source *base, IMemInputPin *pPin, IMemAllocator **pAlloc)
static HRESULT	preview_query_accept (struct strmbase_pin *base, const AM_MEDIA_TYPE *amt)
static HRESULT WINAPI	SmartTeeFilterPreview_DecideAllocator (struct strmbase_source *base, IMemInputPin *pPin, IMemAllocator **pAlloc)
HRESULT	smart_tee_create (IUnknown *outer, IUnknown **out)

Variables
static const struct strmbase_filter_ops	filter_ops
static const struct strmbase_sink_ops	sink_ops
static const struct strmbase_source_ops	capture_ops
static const struct strmbase_source_ops	preview_ops

Function Documentation

capture_query_accept()

static HRESULT capture_query_accept ( struct strmbase_pin *  base, const AM_MEDIA_TYPE *  amt  )

static

Definition at line 250 of file smartteefilter.c.

{
    FIXME("(%p) stub\n", base);
    return S_OK;
}

copy_sample()

static HRESULT copy_sample ( IMediaSample *  inputSample, IMemAllocator *  allocator, IMediaSample **  pOutputSample  )

static

Definition at line 121 of file smartteefilter.c.

{
    REFERENCE_TIME startTime, endTime;
    BOOL haveStartTime = TRUE, haveEndTime = TRUE;
    IMediaSample *outputSample = NULL;
    BYTE *ptrIn, *ptrOut;
    AM_MEDIA_TYPE *mediaType = NULL;
    HRESULT hr;
 
    hr = IMediaSample_GetTime(inputSample, &startTime, &endTime);
    if (hr == S_OK)
        ;
    else if (hr == VFW_S_NO_STOP_TIME)
        haveEndTime = FALSE;
    else if (hr == VFW_E_SAMPLE_TIME_NOT_SET)
        haveStartTime = haveEndTime = FALSE;
    else
        goto end;
 
    hr = IMemAllocator_GetBuffer(allocator, &outputSample,
            haveStartTime ? &startTime : NULL, haveEndTime ? &endTime : NULL, 0);
    if (FAILED(hr)) goto end;
    if (IMediaSample_GetSize(outputSample) < IMediaSample_GetActualDataLength(inputSample)) {
        ERR("insufficient space in sample\n");
        hr = VFW_E_BUFFER_OVERFLOW;
        goto end;
    }
 
    hr = IMediaSample_SetTime(outputSample, haveStartTime ? &startTime : NULL, haveEndTime ? &endTime : NULL);
    if (FAILED(hr)) goto end;
 
    hr = IMediaSample_GetPointer(inputSample, &ptrIn);
    if (FAILED(hr)) goto end;
    hr = IMediaSample_GetPointer(outputSample, &ptrOut);
    if (FAILED(hr)) goto end;
    memcpy(ptrOut, ptrIn, IMediaSample_GetActualDataLength(inputSample));
    IMediaSample_SetActualDataLength(outputSample, IMediaSample_GetActualDataLength(inputSample));
 
    hr = IMediaSample_SetDiscontinuity(outputSample, IMediaSample_IsDiscontinuity(inputSample) == S_OK);
    if (FAILED(hr)) goto end;
 
    haveStartTime = haveEndTime = TRUE;
    hr = IMediaSample_GetMediaTime(inputSample, &startTime, &endTime);
    if (hr == S_OK)
        ;
    else if (hr == VFW_S_NO_STOP_TIME)
        haveEndTime = FALSE;
    else if (hr == VFW_E_MEDIA_TIME_NOT_SET)
        haveStartTime = haveEndTime = FALSE;
    else
        goto end;
    hr = IMediaSample_SetMediaTime(outputSample, haveStartTime ? &startTime : NULL, haveEndTime ? &endTime : NULL);
    if (FAILED(hr)) goto end;
 
    hr = IMediaSample_GetMediaType(inputSample, &mediaType);
    if (FAILED(hr)) goto end;
    if (hr == S_OK) {
        hr = IMediaSample_SetMediaType(outputSample, mediaType);
        if (FAILED(hr)) goto end;
    }
 
    hr = IMediaSample_SetPreroll(outputSample, IMediaSample_IsPreroll(inputSample) == S_OK);
    if (FAILED(hr)) goto end;
 
    hr = IMediaSample_SetSyncPoint(outputSample, IMediaSample_IsSyncPoint(inputSample) == S_OK);
    if (FAILED(hr)) goto end;
 
end:
    if (mediaType)
        DeleteMediaType(mediaType);
    if (FAILED(hr) && outputSample) {
        IMediaSample_Release(outputSample);
        outputSample = NULL;
    }
    *pOutputSample = outputSample;
    return hr;
}

Referenced by SmartTeeFilterInput_Receive().

impl_from_strmbase_filter()

static SmartTeeFilter * impl_from_strmbase_filter ( struct strmbase_filter *  filter )

inlinestatic

Definition at line 31 of file smartteefilter.c.

{
    return CONTAINING_RECORD(filter, SmartTeeFilter, filter);
}

Referenced by impl_from_strmbase_pin(), smart_tee_destroy(), and smart_tee_get_pin().

impl_from_strmbase_pin()

static SmartTeeFilter * impl_from_strmbase_pin ( struct strmbase_pin *  pin )

inlinestatic

Definition at line 36 of file smartteefilter.c.

{
    return impl_from_strmbase_filter(pin->filter);
}

Referenced by sink_get_media_type(), sink_query_accept(), sink_query_interface(), SmartTeeFilterCapture_DecideAllocator(), SmartTeeFilterInput_Receive(), SmartTeeFilterPreview_DecideAllocator(), and source_get_media_type().

preview_query_accept()

static HRESULT preview_query_accept ( struct strmbase_pin *  base, const AM_MEDIA_TYPE *  amt  )

static

Definition at line 293 of file smartteefilter.c.

{
    FIXME("(%p) stub\n", base);
    return S_OK;
}

sink_get_media_type()

static HRESULT sink_get_media_type ( struct strmbase_pin *  base, unsigned int  iPosition, AM_MEDIA_TYPE *  amt  )

static

Definition at line 88 of file smartteefilter.c.

{
    SmartTeeFilter *This = impl_from_strmbase_pin(base);
    HRESULT hr;
    TRACE("(%p)->(%d, %p)\n", This, iPosition, amt);
    if (iPosition)
        return S_FALSE;
    EnterCriticalSection(&This->filter.filter_cs);
    if (This->sink.pin.peer)
    {
        CopyMediaType(amt, &This->sink.pin.mt);
        hr = S_OK;
    }
    else
        hr = S_FALSE;
    LeaveCriticalSection(&This->filter.filter_cs);
    return hr;
}

sink_query_accept()

static HRESULT sink_query_accept ( struct strmbase_pin *  base, const AM_MEDIA_TYPE *  pmt  )

static

Definition at line 77 of file smartteefilter.c.

{
    SmartTeeFilter *This = impl_from_strmbase_pin(base);
    TRACE("(%p, AM_MEDIA_TYPE(%p))\n", This, pmt);
    if (!pmt)
        return VFW_E_TYPE_NOT_ACCEPTED;
    /* We'll take any media type, but the output pins will later
     * struggle to connect downstream. */
    return S_OK;
}

sink_query_interface()

static HRESULT sink_query_interface ( struct strmbase_pin *  iface, REFIID  iid, void **  out  )

static

Definition at line 108 of file smartteefilter.c.

{
    SmartTeeFilter *filter = impl_from_strmbase_pin(iface);
 
    if (IsEqualGUID(iid, &IID_IMemInputPin))
        *out = &filter->sink.IMemInputPin_iface;
    else
        return E_NOINTERFACE;
 
    IUnknown_AddRef((IUnknown *)*out);
    return S_OK;
}

smart_tee_create()

HRESULT smart_tee_create	(	IUnknown *	outer,
		IUnknown **	out
	)

Definition at line 317 of file smartteefilter.c.

{
    SmartTeeFilter *object;
    HRESULT hr;
 
    if (!(object = calloc(1, sizeof(*object))))
        return E_OUTOFMEMORY;
 
    strmbase_filter_init(&object->filter, outer, &CLSID_SmartTee, &filter_ops);
    strmbase_sink_init(&object->sink, &object->filter, L"Input", &sink_ops, NULL);
    hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER,
            &IID_IMemAllocator, (void **)&object->sink.pAllocator);
    if (FAILED(hr))
    {
        strmbase_filter_cleanup(&object->filter);
        free(object);
        return hr;
    }
 
    strmbase_source_init(&object->capture, &object->filter, L"Capture", &capture_ops);
    strmbase_source_init(&object->preview, &object->filter, L"Preview", &preview_ops);
 
    TRACE("Created smart tee %p.\n", object);
    *out = &object->filter.IUnknown_inner;
    return S_OK;
}

smart_tee_destroy()

static void smart_tee_destroy ( struct strmbase_filter *  iface )

static

Definition at line 54 of file smartteefilter.c.

{
    SmartTeeFilter *filter = impl_from_strmbase_filter(iface);
 
    strmbase_sink_cleanup(&filter->sink);
    strmbase_source_cleanup(&filter->capture);
    strmbase_source_cleanup(&filter->preview);
    strmbase_filter_cleanup(&filter->filter);
    free(filter);
}

smart_tee_get_pin()

static struct strmbase_pin * smart_tee_get_pin ( struct strmbase_filter *  iface, unsigned int  index  )

static

Definition at line 41 of file smartteefilter.c.

{
    SmartTeeFilter *filter = impl_from_strmbase_filter(iface);
 
    if (index == 0)
        return &filter->sink.pin;
    else if (index == 1)
        return &filter->capture.pin;
    else if (index == 2)
        return &filter->preview.pin;
    return NULL;
}

smart_tee_wait_state()

static HRESULT smart_tee_wait_state ( struct strmbase_filter *  iface, DWORD  timeout  )

static

Definition at line 65 of file smartteefilter.c.

{
    return iface->state == State_Paused ? VFW_S_CANT_CUE : S_OK;
}

SmartTeeFilterCapture_DecideAllocator()

static HRESULT WINAPI SmartTeeFilterCapture_DecideAllocator ( struct strmbase_source *  base, IMemInputPin *  pPin, IMemAllocator **  pAlloc  )

static

Definition at line 275 of file smartteefilter.c.

{
    SmartTeeFilter *This = impl_from_strmbase_pin(&base->pin);
    TRACE("(%p, %p, %p)\n", This, pPin, pAlloc);
    *pAlloc = This->sink.pAllocator;
    IMemAllocator_AddRef(This->sink.pAllocator);
    return IMemInputPin_NotifyAllocator(pPin, This->sink.pAllocator, TRUE);
}

SmartTeeFilterInput_Receive()

static HRESULT WINAPI SmartTeeFilterInput_Receive ( struct strmbase_sink *  base, IMediaSample *  inputSample  )

static

Definition at line 199 of file smartteefilter.c.

{
    SmartTeeFilter *This = impl_from_strmbase_pin(&base->pin);
    IMediaSample *captureSample = NULL;
    IMediaSample *previewSample = NULL;
    HRESULT hrCapture = VFW_E_NOT_CONNECTED, hrPreview = VFW_E_NOT_CONNECTED;
 
    TRACE("(%p)->(%p)\n", This, inputSample);
 
    /* Modifying the image coming out of one pin doesn't modify the image
     * coming out of the other. MSDN claims the filter doesn't copy,
     * but unless it somehow uses copy-on-write, I just don't see how
     * that's possible. */
 
    /* FIXME: we should ideally do each of these in a separate thread */
    EnterCriticalSection(&This->filter.filter_cs);
    if (This->capture.pin.peer)
        hrCapture = copy_sample(inputSample, This->capture.pAllocator, &captureSample);
    LeaveCriticalSection(&This->filter.filter_cs);
    if (SUCCEEDED(hrCapture) && This->capture.pMemInputPin)
        hrCapture = IMemInputPin_Receive(This->capture.pMemInputPin, captureSample);
    if (captureSample)
        IMediaSample_Release(captureSample);
 
    EnterCriticalSection(&This->filter.filter_cs);
    if (This->preview.pin.peer)
        hrPreview = copy_sample(inputSample, This->preview.pAllocator, &previewSample);
    LeaveCriticalSection(&This->filter.filter_cs);
    /* No timestamps on preview stream: */
    if (SUCCEEDED(hrPreview))
        hrPreview = IMediaSample_SetTime(previewSample, NULL, NULL);
    if (SUCCEEDED(hrPreview) && This->preview.pMemInputPin)
        hrPreview = IMemInputPin_Receive(This->preview.pMemInputPin, previewSample);
    if (previewSample)
        IMediaSample_Release(previewSample);
 
    /* FIXME: how to merge the HRESULTs from the 2 pins? */
    if (SUCCEEDED(hrCapture))
        return hrCapture;
    else
        return hrPreview;
}

SmartTeeFilterPreview_DecideAllocator()

static HRESULT WINAPI SmartTeeFilterPreview_DecideAllocator ( struct strmbase_source *  base, IMemInputPin *  pPin, IMemAllocator **  pAlloc  )

static

Definition at line 299 of file smartteefilter.c.

{
    SmartTeeFilter *This = impl_from_strmbase_pin(&base->pin);
    TRACE("(%p, %p, %p)\n", This, pPin, pAlloc);
    *pAlloc = This->sink.pAllocator;
    IMemAllocator_AddRef(This->sink.pAllocator);
    return IMemInputPin_NotifyAllocator(pPin, This->sink.pAllocator, TRUE);
}

source_get_media_type()

static HRESULT source_get_media_type ( struct strmbase_pin *  iface, unsigned int  index, AM_MEDIA_TYPE *  mt  )

static

Definition at line 256 of file smartteefilter.c.

{
    SmartTeeFilter *filter = impl_from_strmbase_pin(iface);
    HRESULT hr = S_OK;
 
    EnterCriticalSection(&filter->filter.filter_cs);
 
    if (!filter->sink.pin.peer)
        hr = VFW_E_NOT_CONNECTED;
    else if (!index)
        CopyMediaType(mt, &filter->sink.pin.mt);
    else
        hr = VFW_S_NO_MORE_ITEMS;
 
    LeaveCriticalSection(&filter->filter.filter_cs);
    return hr;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

quartz

)

Variable Documentation

capture_ops

const struct strmbase_source_ops capture_ops

static

Initial value:

=
{
    .base.pin_query_accept = capture_query_accept,
    .base.pin_get_media_type = source_get_media_type,
    .pfnAttemptConnection = BaseOutputPinImpl_AttemptConnection,
    .pfnDecideAllocator = SmartTeeFilterCapture_DecideAllocator,
}

Definition at line 285 of file smartteefilter.c.

Referenced by smart_tee_create().

filter_ops

const struct strmbase_filter_ops filter_ops

static

Initial value:

=
{
    .filter_get_pin = smart_tee_get_pin,
    .filter_destroy = smart_tee_destroy,
    .filter_wait_state = smart_tee_wait_state,
}

Definition at line 70 of file smartteefilter.c.

Referenced by smart_tee_create().

preview_ops

const struct strmbase_source_ops preview_ops

static

Initial value:

=
{
    .base.pin_query_accept = preview_query_accept,
    .base.pin_get_media_type = source_get_media_type,
    .pfnAttemptConnection = BaseOutputPinImpl_AttemptConnection,
    .pfnDecideAllocator = SmartTeeFilterPreview_DecideAllocator,
}

Definition at line 309 of file smartteefilter.c.

Referenced by smart_tee_create().

sink_ops

const struct strmbase_sink_ops sink_ops

static

Initial value:

=
{
    .base.pin_query_accept = sink_query_accept,
    .base.pin_get_media_type = sink_get_media_type,
    .base.pin_query_interface = sink_query_interface,
    .pfnReceive = SmartTeeFilterInput_Receive,
}

Definition at line 242 of file smartteefilter.c.

Referenced by smart_tee_create().