avico.c File Reference

#include "qcap_private.h"
#include "vfw.h"
#include "aviriff.h"

Include dependency graph for avico.c:

Go to the source code of this file.

Classes
struct	AVICompressor

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (quartz)
static AVICompressor *	impl_from_strmbase_filter (struct strmbase_filter *filter)
static AVICompressor *	impl_from_strmbase_pin (struct strmbase_pin *pin)
static HRESULT	ensure_driver (AVICompressor *This)
static HRESULT	fill_format_info (AVICompressor *This, VIDEOINFOHEADER *src_videoinfo)
static struct strmbase_pin *	avi_compressor_get_pin (struct strmbase_filter *iface, unsigned int index)
static void	avi_compressor_destroy (struct strmbase_filter *iface)
static HRESULT	avi_compressor_query_interface (struct strmbase_filter *iface, REFIID iid, void **out)
static HRESULT	avi_compressor_init_stream (struct strmbase_filter *iface)
static HRESULT	avi_compressor_cleanup_stream (struct strmbase_filter *iface)
static AVICompressor *	impl_from_IPersistPropertyBag (IPersistPropertyBag *iface)
static HRESULT WINAPI	AVICompressorPropertyBag_QueryInterface (IPersistPropertyBag *iface, REFIID riid, void **ppv)
static ULONG WINAPI	AVICompressorPropertyBag_AddRef (IPersistPropertyBag *iface)
static ULONG WINAPI	AVICompressorPropertyBag_Release (IPersistPropertyBag *iface)
static HRESULT WINAPI	AVICompressorPropertyBag_GetClassID (IPersistPropertyBag *iface, CLSID *pClassID)
static HRESULT WINAPI	AVICompressorPropertyBag_InitNew (IPersistPropertyBag *iface)
static HRESULT WINAPI	AVICompressorPropertyBag_Load (IPersistPropertyBag *iface, IPropertyBag *pPropBag, IErrorLog *pErrorLog)
static HRESULT WINAPI	AVICompressorPropertyBag_Save (IPersistPropertyBag *iface, IPropertyBag *pPropBag, BOOL fClearDirty, BOOL fSaveAllProperties)
static HRESULT	sink_query_accept (struct strmbase_pin *base, const AM_MEDIA_TYPE *pmt)
static HRESULT	sink_query_interface (struct strmbase_pin *iface, REFIID iid, void **out)
static HRESULT WINAPI	AVICompressorIn_Receive (struct strmbase_sink *base, IMediaSample *pSample)
static HRESULT	sink_connect (struct strmbase_sink *iface, IPin *peer, const AM_MEDIA_TYPE *mt)
static void	sink_disconnect (struct strmbase_sink *iface)
static HRESULT	source_get_media_type (struct strmbase_pin *base, unsigned int iPosition, AM_MEDIA_TYPE *amt)
static HRESULT WINAPI	AVICompressorOut_DecideBufferSize (struct strmbase_source *base, IMemAllocator *alloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
static HRESULT WINAPI	AVICompressorOut_DecideAllocator (struct strmbase_source *base, IMemInputPin *pPin, IMemAllocator **pAlloc)
HRESULT	avi_compressor_create (IUnknown *outer, IUnknown **out)

Variables
static const struct strmbase_filter_ops	filter_ops
static const IPersistPropertyBagVtbl	PersistPropertyBagVtbl
static const struct strmbase_sink_ops	sink_ops
static const struct strmbase_source_ops	source_ops

Function Documentation

avi_compressor_cleanup_stream()

static HRESULT avi_compressor_cleanup_stream ( struct strmbase_filter *  iface )

static

Definition at line 154 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_filter(iface);
 
    ICCompressEnd(filter->hic);
    return S_OK;
}

avi_compressor_create()

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

Definition at line 472 of file avico.c.

{
    AVICompressor *object;
 
    if (!(object = calloc(1, sizeof(*object))))
        return E_OUTOFMEMORY;
 
    strmbase_filter_init(&object->filter, outer, &CLSID_AVICo, &filter_ops);
    object->IPersistPropertyBag_iface.lpVtbl = &PersistPropertyBagVtbl;
 
    strmbase_sink_init(&object->sink, &object->filter, L"In", &sink_ops, NULL);
    wcscpy(object->sink.pin.name, L"Input");
 
    strmbase_source_init(&object->source, &object->filter, L"Out", &source_ops);
    wcscpy(object->source.pin.name, L"Output");
 
    TRACE("Created AVI compressor %p.\n", object);
    *out = &object->filter.IUnknown_inner;
    return S_OK;
}

avi_compressor_destroy()

static void avi_compressor_destroy ( struct strmbase_filter *  iface )

static

Definition at line 112 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_filter(iface);
 
    if (filter->hic)
        ICClose(filter->hic);
    free(filter->videoinfo);
    strmbase_sink_cleanup(&filter->sink);
    strmbase_source_cleanup(&filter->source);
    strmbase_filter_cleanup(&filter->filter);
    free(filter);
}

avi_compressor_get_pin()

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

static

Definition at line 101 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_filter(iface);
 
    if (index == 0)
        return &filter->sink.pin;
    else if (index == 1)
        return &filter->source.pin;
    return NULL;
}

avi_compressor_init_stream()

static HRESULT avi_compressor_init_stream ( struct strmbase_filter *  iface )

static

Definition at line 138 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_filter(iface);
    HRESULT hr;
 
    if (filter->source.pAllocator && FAILED(hr = IMemAllocator_Commit(filter->source.pAllocator)))
    {
        ERR("Failed to commit allocator, hr %#lx.\n", hr);
        return hr;
    }
 
    filter->frame_cnt = 0;
 
    return S_OK;
}

avi_compressor_query_interface()

static HRESULT avi_compressor_query_interface ( struct strmbase_filter *  iface, REFIID  iid, void **  out  )

static

Definition at line 125 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_filter(iface);
 
    if (IsEqualGUID(iid, &IID_IPersistPropertyBag))
        *out = &filter->IPersistPropertyBag_iface;
    else
        return E_NOINTERFACE;
 
    IUnknown_AddRef((IUnknown *)*out);
    return S_OK;
}

AVICompressorIn_Receive()

static HRESULT WINAPI AVICompressorIn_Receive ( struct strmbase_sink *  base, IMediaSample *  pSample  )

static

Definition at line 299 of file avico.c.

{
    AVICompressor *This = impl_from_strmbase_pin(&base->pin);
    IMemInputPin *meminput = This->source.pMemInputPin;
    VIDEOINFOHEADER *src_videoinfo;
    REFERENCE_TIME start, stop;
    IMediaSample *out_sample;
    AM_MEDIA_TYPE *mt;
    IMediaSample2 *sample2;
    DWORD comp_flags = 0;
    BOOL is_preroll;
    BOOL sync_point;
    BYTE *ptr, *buf;
    LRESULT res;
    HRESULT hres;
 
    TRACE("(%p)->(%p)\n", base, pSample);
 
    if (!meminput)
    {
        WARN("Source is not connected, returning VFW_E_NOT_CONNECTED.\n");
        return VFW_E_NOT_CONNECTED;
    }
 
    if(!This->hic) {
        FIXME("Driver not loaded\n");
        return E_UNEXPECTED;
    }
 
    hres = IMediaSample_QueryInterface(pSample, &IID_IMediaSample2, (void**)&sample2);
    if(SUCCEEDED(hres)) {
        FIXME("Use IMediaSample2\n");
        IMediaSample2_Release(sample2);
    }
 
    is_preroll = IMediaSample_IsPreroll(pSample) == S_OK;
    sync_point = IMediaSample_IsSyncPoint(pSample) == S_OK;
 
    hres = IMediaSample_GetTime(pSample, &start, &stop);
    if(FAILED(hres)) {
        WARN("Failed to get sample time, hr %#lx.\n", hres);
        return hres;
    }
 
    hres = IMediaSample_GetMediaType(pSample, &mt);
    if(FAILED(hres))
        return hres;
 
    hres = IMediaSample_GetPointer(pSample, &ptr);
    if(FAILED(hres)) {
        ERR("Failed to get input buffer pointer, hr %#lx.\n", hres);
        return hres;
    }
 
    if (FAILED(hres = IMemAllocator_GetBuffer(This->source.pAllocator, &out_sample, &start, &stop, 0)))
    {
        ERR("Failed to get sample, hr %#lx.\n", hres);
        return hres;
    }
 
    if (FAILED(hres = IMediaSample_SetTime(out_sample, &start, &stop)))
        ERR("Failed to set time, hr %#lx.\n", hres);
 
    hres = IMediaSample_GetPointer(out_sample, &buf);
    if(FAILED(hres))
        return hres;
 
    if((This->driver_flags & VIDCF_TEMPORAL) && !(This->driver_flags & VIDCF_FASTTEMPORALC))
        FIXME("Unsupported temporal compression\n");
 
    src_videoinfo = (VIDEOINFOHEADER *)This->sink.pin.mt.pbFormat;
    This->videoinfo->bmiHeader.biSizeImage = This->max_frame_size;
    res = ICCompress(This->hic, sync_point ? ICCOMPRESS_KEYFRAME : 0, &This->videoinfo->bmiHeader, buf,
            &src_videoinfo->bmiHeader, ptr, 0, &comp_flags, This->frame_cnt, 0, 0, NULL, NULL);
    if(res != ICERR_OK) {
        ERR("Failed to compress frame, error %Id.\n", res);
        IMediaSample_Release(out_sample);
        return E_FAIL;
    }
 
    IMediaSample_SetActualDataLength(out_sample, This->videoinfo->bmiHeader.biSizeImage);
    IMediaSample_SetPreroll(out_sample, is_preroll);
    IMediaSample_SetSyncPoint(out_sample, (comp_flags&AVIIF_KEYFRAME) != 0);
    IMediaSample_SetDiscontinuity(out_sample, (IMediaSample_IsDiscontinuity(pSample) == S_OK));
 
    if (IMediaSample_GetMediaTime(pSample, &start, &stop) == S_OK)
        IMediaSample_SetMediaTime(out_sample, &start, &stop);
    else
        IMediaSample_SetMediaTime(out_sample, NULL, NULL);
 
    hres = IMemInputPin_Receive(meminput, out_sample);
    if(FAILED(hres))
        WARN("Failed to deliver sample, hr %#lx.\n", hres);
 
    IMediaSample_Release(out_sample);
    This->frame_cnt++;
    return hres;
}

AVICompressorOut_DecideAllocator()

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

static

Definition at line 457 of file avico.c.

{
    TRACE("(%p)->(%p %p)\n", base, pPin, pAlloc);
    return BaseOutputPinImpl_DecideAllocator(base, pPin, pAlloc);
}

AVICompressorOut_DecideBufferSize()

static HRESULT WINAPI AVICompressorOut_DecideBufferSize ( struct strmbase_source *  base, IMemAllocator *  alloc, ALLOCATOR_PROPERTIES *  ppropInputRequest  )

static

Definition at line 439 of file avico.c.

{
    AVICompressor *This = impl_from_strmbase_pin(&base->pin);
    ALLOCATOR_PROPERTIES actual;
 
    TRACE("(%p)\n", This);
 
    if (!ppropInputRequest->cBuffers)
        ppropInputRequest->cBuffers = 1;
    if (ppropInputRequest->cbBuffer < This->max_frame_size)
        ppropInputRequest->cbBuffer = This->max_frame_size;
    if (!ppropInputRequest->cbAlign)
        ppropInputRequest->cbAlign = 1;
 
    return IMemAllocator_SetProperties(alloc, ppropInputRequest, &actual);
}

AVICompressorPropertyBag_AddRef()

static ULONG WINAPI AVICompressorPropertyBag_AddRef ( IPersistPropertyBag *  iface )

static

Definition at line 182 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface);
}

AVICompressorPropertyBag_GetClassID()

static HRESULT WINAPI AVICompressorPropertyBag_GetClassID ( IPersistPropertyBag *  iface, CLSID *  pClassID  )

static

Definition at line 194 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    return IBaseFilter_GetClassID(&This->filter.IBaseFilter_iface, pClassID);
}

AVICompressorPropertyBag_InitNew()

static HRESULT WINAPI AVICompressorPropertyBag_InitNew ( IPersistPropertyBag *  iface )

static

Definition at line 200 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    FIXME("(%p)->()\n", This);
    return E_NOTIMPL;
}

AVICompressorPropertyBag_Load()

static HRESULT WINAPI AVICompressorPropertyBag_Load ( IPersistPropertyBag *  iface, IPropertyBag *  pPropBag, IErrorLog *  pErrorLog  )

static

Definition at line 207 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    BSTR str;
    VARIANT v;
    HRESULT hres;
 
    TRACE("(%p)->(%p %p)\n", This, pPropBag, pErrorLog);
 
    V_VT(&v) = VT_BSTR;
    hres = IPropertyBag_Read(pPropBag, L"FccHandler", &v, NULL);
    if(FAILED(hres)) {
        ERR("Failed to read FccHandler value, hr %#lx.\n", hres);
        return hres;
    }
 
    if(V_VT(&v) != VT_BSTR) {
        FIXME("Got vt %d\n", V_VT(&v));
        VariantClear(&v);
        return E_FAIL;
    }
 
    str = V_BSTR(&v);
    TRACE("FccHandler = %s\n", debugstr_w(str));
    if(SysStringLen(str) != 4) {
        FIXME("Invalid FccHandler len\n");
        SysFreeString(str);
        return E_FAIL;
    }
 
    This->fcc_handler = FCC(str[0], str[1], str[2], str[3]);
    SysFreeString(str);
    return S_OK;
}

AVICompressorPropertyBag_QueryInterface()

static HRESULT WINAPI AVICompressorPropertyBag_QueryInterface ( IPersistPropertyBag *  iface, REFIID  riid, void **  ppv  )

static

Definition at line 176 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv);
}

AVICompressorPropertyBag_Release()

static ULONG WINAPI AVICompressorPropertyBag_Release ( IPersistPropertyBag *  iface )

static

Definition at line 188 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    return IBaseFilter_Release(&This->filter.IBaseFilter_iface);
}

AVICompressorPropertyBag_Save()

static HRESULT WINAPI AVICompressorPropertyBag_Save ( IPersistPropertyBag *  iface, IPropertyBag *  pPropBag, BOOL  fClearDirty, BOOL  fSaveAllProperties  )

static

Definition at line 242 of file avico.c.

{
    AVICompressor *This = impl_from_IPersistPropertyBag(iface);
    FIXME("(%p)->(%p %x %x)\n", This, pPropBag, fClearDirty, fSaveAllProperties);
    return E_NOTIMPL;
}

ensure_driver()

static HRESULT ensure_driver ( AVICompressor *  This )

static

Definition at line 53 of file avico.c.

{
    if(This->hic)
        return S_OK;
 
    This->hic = ICOpen(FCC('v','i','d','c'), This->fcc_handler, ICMODE_COMPRESS);
    if(!This->hic) {
        FIXME("ICOpen failed\n");
        return E_FAIL;
    }
 
    return S_OK;
}

Referenced by fill_format_info(), and sink_query_accept().

fill_format_info()

static HRESULT fill_format_info ( AVICompressor *  This, VIDEOINFOHEADER *  src_videoinfo  )

static

Definition at line 67 of file avico.c.

{
    DWORD size;
    ICINFO icinfo;
    HRESULT hres;
 
    hres = ensure_driver(This);
    if(hres != S_OK)
        return hres;
 
    size = ICGetInfo(This->hic, &icinfo, sizeof(icinfo));
    if(size != sizeof(icinfo))
        return E_FAIL;
 
    size = ICCompressGetFormatSize(This->hic, &src_videoinfo->bmiHeader);
    if(!size) {
        FIXME("ICCompressGetFormatSize failed\n");
        return E_FAIL;
    }
 
    size += FIELD_OFFSET(VIDEOINFOHEADER, bmiHeader);
    if (!(This->videoinfo = calloc(1, size)))
        return E_OUTOFMEMORY;
 
    This->videoinfo_size = size;
    This->driver_flags = icinfo.dwFlags;
    ICCompressGetFormat(This->hic, &src_videoinfo->bmiHeader, &This->videoinfo->bmiHeader);
 
    This->videoinfo->dwBitRate = 10000000/src_videoinfo->AvgTimePerFrame * This->videoinfo->bmiHeader.biSizeImage * 8;
    This->videoinfo->AvgTimePerFrame = src_videoinfo->AvgTimePerFrame;
    This->max_frame_size = This->videoinfo->bmiHeader.biSizeImage;
    return S_OK;
}

Referenced by sink_connect().

impl_from_IPersistPropertyBag()

static AVICompressor * impl_from_IPersistPropertyBag ( IPersistPropertyBag *  iface )

static

Definition at line 171 of file avico.c.

{
    return CONTAINING_RECORD(iface, AVICompressor, IPersistPropertyBag_iface);
}

Referenced by AVICompressorPropertyBag_AddRef(), AVICompressorPropertyBag_GetClassID(), AVICompressorPropertyBag_InitNew(), AVICompressorPropertyBag_Load(), AVICompressorPropertyBag_QueryInterface(), AVICompressorPropertyBag_Release(), and AVICompressorPropertyBag_Save().

impl_from_strmbase_filter()

static AVICompressor * impl_from_strmbase_filter ( struct strmbase_filter *  filter )

inlinestatic

Definition at line 43 of file avico.c.

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

Referenced by avi_compressor_cleanup_stream(), avi_compressor_destroy(), avi_compressor_get_pin(), avi_compressor_init_stream(), avi_compressor_query_interface(), impl_from_strmbase_pin(), and source_get_media_type().

impl_from_strmbase_pin()

static AVICompressor * impl_from_strmbase_pin ( struct strmbase_pin *  pin )

inlinestatic

Definition at line 48 of file avico.c.

{
    return impl_from_strmbase_filter(pin->filter);
}

Referenced by AVICompressorIn_Receive(), AVICompressorOut_DecideBufferSize(), sink_connect(), sink_disconnect(), sink_query_accept(), and sink_query_interface().

sink_connect()

static HRESULT sink_connect ( struct strmbase_sink *  iface, IPin *  peer, const AM_MEDIA_TYPE *  mt  )

static

Definition at line 398 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_pin(&iface->pin);
    return fill_format_info(filter, (VIDEOINFOHEADER *)mt->pbFormat);
}

sink_disconnect()

static void sink_disconnect ( struct strmbase_sink *  iface )

static

Definition at line 404 of file avico.c.

{
    AVICompressor *filter = impl_from_strmbase_pin(&iface->pin);
    free(filter->videoinfo);
    filter->videoinfo = NULL;
}

sink_query_accept()

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

static

Definition at line 260 of file avico.c.

{
    AVICompressor *This = impl_from_strmbase_pin(base);
    VIDEOINFOHEADER *videoinfo;
    HRESULT hres;
    DWORD res;
 
    TRACE("(%p)->(AM_MEDIA_TYPE(%p))\n", base, pmt);
 
    if(!IsEqualIID(&pmt->majortype, &MEDIATYPE_Video))
        return S_FALSE;
 
    if(!IsEqualIID(&pmt->formattype, &FORMAT_VideoInfo)) {
        FIXME("formattype %s unsupported\n", debugstr_guid(&pmt->formattype));
        return S_FALSE;
    }
 
    hres = ensure_driver(This);
    if(hres != S_OK)
        return hres;
 
    videoinfo = (VIDEOINFOHEADER*)pmt->pbFormat;
    res = ICCompressQuery(This->hic, &videoinfo->bmiHeader, NULL);
    return res == ICERR_OK ? S_OK : S_FALSE;
}

sink_query_interface()

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

static

Definition at line 286 of file avico.c.

{
    AVICompressor *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;
}

source_get_media_type()

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

static

Definition at line 420 of file avico.c.

{
    AVICompressor *This = impl_from_strmbase_filter(base->filter);
 
    if(iPosition || !This->videoinfo)
        return S_FALSE;
 
    amt->majortype = MEDIATYPE_Video;
    amt->subtype = MEDIASUBTYPE_PCM;
    amt->bFixedSizeSamples = FALSE;
    amt->bTemporalCompression = (This->driver_flags & VIDCF_TEMPORAL) != 0;
    amt->lSampleSize = This->sink.pin.mt.lSampleSize;
    amt->formattype = FORMAT_VideoInfo;
    amt->pUnk = NULL;
    amt->cbFormat = This->videoinfo_size;
    amt->pbFormat = (BYTE*)This->videoinfo;
    return S_OK;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

quartz

)

Variable Documentation

filter_ops

const struct strmbase_filter_ops filter_ops

static

Initial value:

=
{
    .filter_get_pin = avi_compressor_get_pin,
    .filter_destroy = avi_compressor_destroy,
    .filter_query_interface = avi_compressor_query_interface,
    .filter_init_stream = avi_compressor_init_stream,
    .filter_cleanup_stream = avi_compressor_cleanup_stream,
}

Definition at line 162 of file avico.c.

Referenced by avi_compressor_create().

PersistPropertyBagVtbl

const IPersistPropertyBagVtbl PersistPropertyBagVtbl

static

Initial value:

= {
    AVICompressorPropertyBag_QueryInterface,
    AVICompressorPropertyBag_AddRef,
    AVICompressorPropertyBag_Release,
    AVICompressorPropertyBag_GetClassID,
    AVICompressorPropertyBag_InitNew,
    AVICompressorPropertyBag_Load,
    AVICompressorPropertyBag_Save
}

Definition at line 250 of file avico.c.

Referenced by avi_compressor_create().

sink_ops

const struct strmbase_sink_ops sink_ops

static

Initial value:

=
{
    .base.pin_query_accept = sink_query_accept,
    .base.pin_query_interface = sink_query_interface,
    .pfnReceive = AVICompressorIn_Receive,
    .sink_connect = sink_connect,
    .sink_disconnect = sink_disconnect,
}

Definition at line 411 of file avico.c.

Referenced by avi_compressor_create().

source_ops

const struct strmbase_source_ops source_ops

static

Initial value:

=
{
    .base.pin_get_media_type = source_get_media_type,
    .pfnAttemptConnection = BaseOutputPinImpl_AttemptConnection,
    .pfnDecideBufferSize = AVICompressorOut_DecideBufferSize,
    .pfnDecideAllocator = AVICompressorOut_DecideAllocator,
}

Definition at line 464 of file avico.c.

Referenced by avi_compressor_create().