/* * Copyright (C) 2013 Piotr Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include #include #define COBJMACROS #include "windef.h" #include "winbase.h" #include "wtypes.h" #include "dshow.h" #include "vfw.h" #include "aviriff.h" #include "qcap_main.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(qcap); #define MAX_PIN_NO 128 #define AVISUPERINDEX_ENTRIES 2000 #define AVISTDINDEX_ENTRIES 4000 #define ALIGN(x) ((x+1)/2*2) typedef struct { BaseOutputPin pin; IQualityControl IQualityControl_iface; int cur_stream; LONGLONG cur_time; int buf_pos; BYTE buf[65536]; int movi_off; int out_pos; int size; IStream *stream; } AviMuxOut; typedef struct { BaseInputPin pin; IAMStreamControl IAMStreamControl_iface; IPropertyBag IPropertyBag_iface; IQualityControl IQualityControl_iface; REFERENCE_TIME avg_time_per_frame; REFERENCE_TIME stop; int stream_id; LONGLONG stream_time; /* strl chunk */ AVISTREAMHEADER strh; struct { FOURCC fcc; DWORD cb; BYTE data[1]; } *strf; AVISUPERINDEX *indx; BYTE indx_data[FIELD_OFFSET(AVISUPERINDEX, aIndex[AVISUPERINDEX_ENTRIES])]; /* movi chunk */ int ix_off; AVISTDINDEX *ix; BYTE ix_data[FIELD_OFFSET(AVISTDINDEX, aIndex[AVISTDINDEX_ENTRIES])]; IMediaSample *samples_head; IMemAllocator *samples_allocator; } AviMuxIn; typedef struct { BaseFilter filter; IConfigAviMux IConfigAviMux_iface; IConfigInterleaving IConfigInterleaving_iface; IMediaSeeking IMediaSeeking_iface; IPersistMediaPropertyBag IPersistMediaPropertyBag_iface; ISpecifyPropertyPages ISpecifyPropertyPages_iface; InterleavingMode mode; REFERENCE_TIME interleave; REFERENCE_TIME preroll; AviMuxOut *out; int input_pin_no; AviMuxIn *in[MAX_PIN_NO-1]; REFERENCE_TIME start, stop; AVIMAINHEADER avih; int idx1_entries; int idx1_size; AVIINDEXENTRY *idx1; } AviMux; static HRESULT create_input_pin(AviMux*); static inline AviMux* impl_from_BaseFilter(BaseFilter *filter) { return CONTAINING_RECORD(filter, AviMux, filter); } static IPin* WINAPI AviMux_GetPin(BaseFilter *iface, int pos) { AviMux *This = impl_from_BaseFilter(iface); TRACE("(%p)->(%d)\n", This, pos); if(pos == 0) { IPin_AddRef(&This->out->pin.pin.IPin_iface); return &This->out->pin.pin.IPin_iface; }else if(pos>0 && pos<=This->input_pin_no) { IPin_AddRef(&This->in[pos-1]->pin.pin.IPin_iface); return &This->in[pos-1]->pin.pin.IPin_iface; } return NULL; } static const BaseFilterFuncTable filter_func_table = { AviMux_GetPin, }; static inline AviMux* impl_from_IBaseFilter(IBaseFilter *iface) { BaseFilter *filter = CONTAINING_RECORD(iface, BaseFilter, IBaseFilter_iface); return impl_from_BaseFilter(filter); } static HRESULT WINAPI AviMux_QueryInterface(IBaseFilter *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_IBaseFilter(iface); TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv); if(IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IPersist) || IsEqualIID(riid, &IID_IMediaFilter) || IsEqualIID(riid, &IID_IBaseFilter)) *ppv = &This->filter.IBaseFilter_iface; else if(IsEqualIID(riid, &IID_IConfigAviMux)) *ppv = &This->IConfigAviMux_iface; else if(IsEqualIID(riid, &IID_IConfigInterleaving)) *ppv = &This->IConfigInterleaving_iface; else if(IsEqualIID(riid, &IID_IMediaSeeking)) *ppv = &This->IMediaSeeking_iface; else if(IsEqualIID(riid, &IID_IPersistMediaPropertyBag)) *ppv = &This->IPersistMediaPropertyBag_iface; else if(IsEqualIID(riid, &IID_ISpecifyPropertyPages)) *ppv = &This->ISpecifyPropertyPages_iface; else { FIXME("no interface for %s\n", debugstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } static ULONG WINAPI AviMux_Release(IBaseFilter *iface) { AviMux *This = impl_from_IBaseFilter(iface); ULONG ref = InterlockedDecrement(&This->filter.refCount); TRACE("(%p) new refcount: %u\n", This, ref); if(!ref) { int i; BaseOutputPinImpl_Release(&This->out->pin.pin.IPin_iface); for(i=0; iinput_pin_no; i++) { IPin_Disconnect(&This->in[i]->pin.pin.IPin_iface); IMemAllocator_Release(This->in[i]->samples_allocator); This->in[i]->samples_allocator = NULL; BaseInputPinImpl_Release(&This->in[i]->pin.pin.IPin_iface); } HeapFree(GetProcessHeap(), 0, This->idx1); strmbase_filter_cleanup(&This->filter); HeapFree(GetProcessHeap(), 0, This); ObjectRefCount(FALSE); } return ref; } static HRESULT out_flush(AviMux *This) { ULONG written; HRESULT hr; if(!This->out->buf_pos) return S_OK; hr = IStream_Write(This->out->stream, This->out->buf, This->out->buf_pos, &written); if(FAILED(hr)) return hr; if(written != This->out->buf_pos) return E_FAIL; This->out->buf_pos = 0; return S_OK; } static HRESULT out_seek(AviMux *This, int pos) { LARGE_INTEGER li; HRESULT hr; hr = out_flush(This); if(FAILED(hr)) return hr; li.QuadPart = pos; hr = IStream_Seek(This->out->stream, li, STREAM_SEEK_SET, NULL); if(FAILED(hr)) return hr; This->out->out_pos = pos; if(This->out->out_pos > This->out->size) This->out->size = This->out->out_pos; return hr; } static HRESULT out_write(AviMux *This, const void *data, int size) { int chunk_size; HRESULT hr; while(1) { if(size > sizeof(This->out->buf)-This->out->buf_pos) chunk_size = sizeof(This->out->buf)-This->out->buf_pos; else chunk_size = size; memcpy(This->out->buf + This->out->buf_pos, data, chunk_size); size -= chunk_size; data = (const BYTE*)data + chunk_size; This->out->buf_pos += chunk_size; This->out->out_pos += chunk_size; if(This->out->out_pos > This->out->size) This->out->size = This->out->out_pos; if(!size) break; hr = out_flush(This); if(FAILED(hr)) return hr; } return S_OK; } static inline HRESULT idx1_add_entry(AviMux *avimux, DWORD ckid, DWORD flags, DWORD off, DWORD len) { if(avimux->idx1_entries == avimux->idx1_size) { AVIINDEXENTRY *new_idx = HeapReAlloc(GetProcessHeap(), 0, avimux->idx1, sizeof(*avimux->idx1)*2*avimux->idx1_size); if(!new_idx) return E_OUTOFMEMORY; avimux->idx1_size *= 2; avimux->idx1 = new_idx; } avimux->idx1[avimux->idx1_entries].ckid = ckid; avimux->idx1[avimux->idx1_entries].dwFlags = flags; avimux->idx1[avimux->idx1_entries].dwChunkOffset = off; avimux->idx1[avimux->idx1_entries].dwChunkLength = len; avimux->idx1_entries++; return S_OK; } static HRESULT flush_queue(AviMux *avimux, AviMuxIn *avimuxin, BOOL closing) { IMediaSample *sample, **prev, **head_prev; BYTE *data; RIFFCHUNK rf; DWORD size; DWORD flags; HRESULT hr; if(avimux->out->cur_stream != avimuxin->stream_id) return S_OK; while(avimuxin->samples_head) { hr = IMediaSample_GetPointer(avimuxin->samples_head, (BYTE**)&head_prev); if(FAILED(hr)) return hr; head_prev--; hr = IMediaSample_GetPointer(*head_prev, (BYTE**)&prev); if(FAILED(hr)) return hr; prev--; sample = *head_prev; size = IMediaSample_GetActualDataLength(sample); hr = IMediaSample_GetPointer(sample, &data); if(FAILED(hr)) return hr; flags = IMediaSample_IsDiscontinuity(sample)==S_OK ? AM_SAMPLE_TIMEDISCONTINUITY : 0; if(IMediaSample_IsSyncPoint(sample) == S_OK) flags |= AM_SAMPLE_SPLICEPOINT; if(avimuxin->stream_time + (closing ? 0 : avimuxin->strh.dwScale) > avimux->out->cur_time && !(flags & AM_SAMPLE_TIMEDISCONTINUITY)) { if(closing) break; avimux->out->cur_stream++; if(avimux->out->cur_stream >= avimux->input_pin_no-1) { avimux->out->cur_time += avimux->interleave; avimux->out->cur_stream = 0; } avimuxin = avimux->in[avimux->out->cur_stream]; continue; } if(avimuxin->ix->nEntriesInUse == AVISTDINDEX_ENTRIES) { /* TODO: use output pins Deliver/Receive method */ hr = out_seek(avimux, avimuxin->ix_off); if(FAILED(hr)) return hr; hr = out_write(avimux, avimuxin->ix, sizeof(avimuxin->ix_data)); if(FAILED(hr)) return hr; avimuxin->indx->aIndex[avimuxin->indx->nEntriesInUse].qwOffset = avimuxin->ix_off; avimuxin->indx->aIndex[avimuxin->indx->nEntriesInUse].dwSize = sizeof(avimuxin->ix_data); avimuxin->indx->aIndex[avimuxin->indx->nEntriesInUse].dwDuration = AVISTDINDEX_ENTRIES; avimuxin->indx->nEntriesInUse++; memset(avimuxin->ix->aIndex, 0, sizeof(avimuxin->ix->aIndex)*avimuxin->ix->nEntriesInUse); avimuxin->ix->nEntriesInUse = 0; avimuxin->ix->qwBaseOffset = 0; avimuxin->ix_off = avimux->out->size; avimux->out->size += sizeof(avimuxin->ix_data); } if(*head_prev == avimuxin->samples_head) avimuxin->samples_head = NULL; else *head_prev = *prev; avimuxin->stream_time += avimuxin->strh.dwScale; avimuxin->strh.dwLength++; if(!(flags & AM_SAMPLE_TIMEDISCONTINUITY)) { if(!avimuxin->ix->qwBaseOffset) avimuxin->ix->qwBaseOffset = avimux->out->size; avimuxin->ix->aIndex[avimuxin->ix->nEntriesInUse].dwOffset = avimux->out->size + sizeof(RIFFCHUNK) - avimuxin->ix->qwBaseOffset; hr = out_seek(avimux, avimux->out->size); if(FAILED(hr)) { IMediaSample_Release(sample); return hr; } } avimuxin->ix->aIndex[avimuxin->ix->nEntriesInUse].dwSize = size | (flags & AM_SAMPLE_SPLICEPOINT ? 0 : AVISTDINDEX_DELTAFRAME); avimuxin->ix->nEntriesInUse++; rf.fcc = FCC('0'+avimuxin->stream_id/10, '0'+avimuxin->stream_id%10, 'd', flags & AM_SAMPLE_SPLICEPOINT ? 'b' : 'c'); rf.cb = size; hr = idx1_add_entry(avimux, rf.fcc, flags & AM_SAMPLE_SPLICEPOINT ? AVIIF_KEYFRAME : 0, flags & AM_SAMPLE_TIMEDISCONTINUITY ? avimux->idx1[avimux->idx1_entries-1].dwChunkOffset : avimux->out->size, size); if(FAILED(hr)) { IMediaSample_Release(sample); return hr; } if(!(flags & AM_SAMPLE_TIMEDISCONTINUITY)) { hr = out_write(avimux, &rf, sizeof(rf)); if(FAILED(hr)) { IMediaSample_Release(sample); return hr; } hr = out_write(avimux, data, size); if(FAILED(hr)) { IMediaSample_Release(sample); return hr; } flags = 0; hr = out_write(avimux, &flags, ALIGN(rf.cb)-rf.cb); if(FAILED(hr)) { IMediaSample_Release(sample); return hr; } } IMediaSample_Release(sample); } return S_OK; } static HRESULT queue_sample(AviMux *avimux, AviMuxIn *avimuxin, IMediaSample *sample) { IMediaSample **prev, **head_prev; HRESULT hr; hr = IMediaSample_GetPointer(sample, (BYTE**)&prev); if(FAILED(hr)) return hr; prev--; if(avimuxin->samples_head) { hr = IMediaSample_GetPointer(avimuxin->samples_head, (BYTE**)&head_prev); if(FAILED(hr)) return hr; head_prev--; *prev = *head_prev; *head_prev = sample; }else { *prev = sample; } avimuxin->samples_head = sample; IMediaSample_AddRef(sample); return flush_queue(avimux, avimuxin, FALSE); } static HRESULT WINAPI AviMux_Stop(IBaseFilter *iface) { AviMux *This = impl_from_IBaseFilter(iface); HRESULT hr; int i; TRACE("(%p)\n", This); if(This->filter.state == State_Stopped) return S_OK; if(This->out->stream) { AVIEXTHEADER dmlh; RIFFCHUNK rc; RIFFLIST rl; int idx1_off, empty_stream; empty_stream = This->out->cur_stream; for(i=empty_stream+1; ; i++) { if(i >= This->input_pin_no-1) i = 0; if(i == empty_stream) break; This->out->cur_stream = i; hr = flush_queue(This, This->in[This->out->cur_stream], TRUE); if(FAILED(hr)) return hr; } idx1_off = This->out->size; rc.fcc = ckidAVIOLDINDEX; rc.cb = This->idx1_entries * sizeof(*This->idx1); hr = out_write(This, &rc, sizeof(rc)); if(FAILED(hr)) return hr; hr = out_write(This, This->idx1, This->idx1_entries * sizeof(*This->idx1)); if(FAILED(hr)) return hr; /* native writes 8 '\0' characters after the end of RIFF data */ i = 0; hr = out_write(This, &i, sizeof(i)); if(FAILED(hr)) return hr; hr = out_write(This, &i, sizeof(i)); if(FAILED(hr)) return hr; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.pConnectedTo) continue; hr = out_seek(This, This->in[i]->ix_off); if(FAILED(hr)) return hr; This->in[i]->indx->aIndex[This->in[i]->indx->nEntriesInUse].qwOffset = This->in[i]->ix_off; This->in[i]->indx->aIndex[This->in[i]->indx->nEntriesInUse].dwSize = sizeof(This->in[i]->ix_data); This->in[i]->indx->aIndex[This->in[i]->indx->nEntriesInUse].dwDuration = This->in[i]->strh.dwLength; if(This->in[i]->indx->nEntriesInUse) { This->in[i]->indx->aIndex[This->in[i]->indx->nEntriesInUse].dwDuration -= This->in[i]->indx->aIndex[This->in[i]->indx->nEntriesInUse-1].dwDuration; } This->in[i]->indx->nEntriesInUse++; hr = out_write(This, This->in[i]->ix, sizeof(This->in[i]->ix_data)); if(FAILED(hr)) return hr; } hr = out_seek(This, 0); if(FAILED(hr)) return hr; rl.fcc = FCC('R','I','F','F'); rl.cb = This->out->size-sizeof(RIFFCHUNK)-2*sizeof(int); rl.fccListType = FCC('A','V','I',' '); hr = out_write(This, &rl, sizeof(rl)); if(FAILED(hr)) return hr; rl.fcc = FCC('L','I','S','T'); rl.cb = This->out->movi_off - sizeof(RIFFLIST) - sizeof(RIFFCHUNK); rl.fccListType = FCC('h','d','r','l'); hr = out_write(This, &rl, sizeof(rl)); if(FAILED(hr)) return hr; /* FIXME: set This->avih.dwMaxBytesPerSec value */ This->avih.dwTotalFrames = (This->stop-This->start) / 10 / This->avih.dwMicroSecPerFrame; hr = out_write(This, &This->avih, sizeof(This->avih)); if(FAILED(hr)) return hr; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.pConnectedTo) continue; rl.cb = sizeof(FOURCC) + sizeof(AVISTREAMHEADER) + sizeof(RIFFCHUNK) + This->in[i]->strf->cb + sizeof(This->in[i]->indx_data); rl.fccListType = ckidSTREAMLIST; hr = out_write(This, &rl, sizeof(rl)); if(FAILED(hr)) return hr; hr = out_write(This, &This->in[i]->strh, sizeof(AVISTREAMHEADER)); if(FAILED(hr)) return hr; hr = out_write(This, This->in[i]->strf, sizeof(RIFFCHUNK) + This->in[i]->strf->cb); if(FAILED(hr)) return hr; hr = out_write(This, This->in[i]->indx, sizeof(This->in[i]->indx_data)); if(FAILED(hr)) return hr; } rl.cb = sizeof(dmlh) + sizeof(FOURCC); rl.fccListType = ckidODML; hr = out_write(This, &rl, sizeof(rl)); if(FAILED(hr)) return hr; memset(&dmlh, 0, sizeof(dmlh)); dmlh.fcc = ckidAVIEXTHEADER; dmlh.cb = sizeof(dmlh) - sizeof(RIFFCHUNK); dmlh.dwGrandFrames = This->in[0]->strh.dwLength; hr = out_write(This, &dmlh, sizeof(dmlh)); rl.cb = idx1_off - This->out->movi_off - sizeof(RIFFCHUNK); rl.fccListType = FCC('m','o','v','i'); out_write(This, &rl, sizeof(rl)); out_flush(This); IStream_Release(This->out->stream); This->out->stream = NULL; } This->filter.state = State_Stopped; return S_OK; } static HRESULT WINAPI AviMux_Pause(IBaseFilter *iface) { AviMux *This = impl_from_IBaseFilter(iface); FIXME("(%p)\n", This); return E_NOTIMPL; } static HRESULT WINAPI AviMux_Run(IBaseFilter *iface, REFERENCE_TIME tStart) { AviMux *This = impl_from_IBaseFilter(iface); HRESULT hr; int i, stream_id; TRACE("(%p)->(%s)\n", This, wine_dbgstr_longlong(tStart)); if(This->filter.state == State_Running) return S_OK; if(This->mode != INTERLEAVE_FULL) { FIXME("mode not supported (%d)\n", This->mode); return E_NOTIMPL; } if(tStart) FIXME("tStart parameter ignored\n"); for(i=0; iinput_pin_no; i++) { IMediaSeeking *ms; LONGLONG cur, stop; if(!This->in[i]->pin.pin.pConnectedTo) continue; hr = IPin_QueryInterface(This->in[i]->pin.pin.pConnectedTo, &IID_IMediaSeeking, (void**)&ms); if(FAILED(hr)) continue; hr = IMediaSeeking_GetPositions(ms, &cur, &stop); if(FAILED(hr)) { IMediaSeeking_Release(ms); continue; } FIXME("Use IMediaSeeking to fill stream header\n"); IMediaSeeking_Release(ms); } if(This->out->pin.pMemInputPin) { hr = IMemInputPin_QueryInterface(This->out->pin.pMemInputPin, &IID_IStream, (void**)&This->out->stream); if(FAILED(hr)) return hr; } This->idx1_entries = 0; if(!This->idx1_size) { This->idx1_size = 1024; This->idx1 = HeapAlloc(GetProcessHeap(), 0, sizeof(*This->idx1)*This->idx1_size); if(!This->idx1) return E_OUTOFMEMORY; } This->out->size = 3*sizeof(RIFFLIST) + sizeof(AVIMAINHEADER) + sizeof(AVIEXTHEADER); This->start = -1; This->stop = -1; memset(&This->avih, 0, sizeof(This->avih)); for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.pConnectedTo) continue; This->avih.dwStreams++; This->out->size += sizeof(RIFFLIST) + sizeof(AVISTREAMHEADER) + sizeof(RIFFCHUNK) + This->in[i]->strf->cb + sizeof(This->in[i]->indx_data); This->in[i]->strh.dwScale = MulDiv(This->in[i]->avg_time_per_frame, This->interleave, 10000000); This->in[i]->strh.dwRate = This->interleave; hr = IMemAllocator_Commit(This->in[i]->pin.pAllocator); if(FAILED(hr)) { if(This->out->stream) { IStream_Release(This->out->stream); This->out->stream = NULL; } return hr; } } This->out->movi_off = This->out->size; This->out->size += sizeof(RIFFLIST); idx1_add_entry(This, FCC('7','F','x','x'), 0, This->out->movi_off+sizeof(RIFFLIST), 0); stream_id = 0; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.pConnectedTo) continue; This->in[i]->ix_off = This->out->size; This->out->size += sizeof(This->in[i]->ix_data); This->in[i]->ix->fcc = FCC('i','x','0'+stream_id/10,'0'+stream_id%10); This->in[i]->ix->cb = sizeof(This->in[i]->ix_data) - sizeof(RIFFCHUNK); This->in[i]->ix->wLongsPerEntry = 2; This->in[i]->ix->bIndexSubType = 0; This->in[i]->ix->bIndexType = AVI_INDEX_OF_CHUNKS; This->in[i]->ix->dwChunkId = FCC('0'+stream_id/10,'0'+stream_id%10,'d','b'); This->in[i]->ix->qwBaseOffset = 0; This->in[i]->indx->fcc = ckidAVISUPERINDEX; This->in[i]->indx->cb = sizeof(This->in[i]->indx_data) - sizeof(RIFFCHUNK); This->in[i]->indx->wLongsPerEntry = 4; This->in[i]->indx->bIndexSubType = 0; This->in[i]->indx->bIndexType = AVI_INDEX_OF_INDEXES; This->in[i]->indx->dwChunkId = This->in[i]->ix->dwChunkId; This->in[i]->stream_id = stream_id++; } This->out->buf_pos = 0; This->out->out_pos = 0; This->avih.fcc = ckidMAINAVIHEADER; This->avih.cb = sizeof(AVIMAINHEADER) - sizeof(RIFFCHUNK); /* TODO: Use first video stream */ This->avih.dwMicroSecPerFrame = This->in[0]->avg_time_per_frame/10; This->avih.dwPaddingGranularity = 1; This->avih.dwFlags = AVIF_TRUSTCKTYPE | AVIF_HASINDEX; This->avih.dwWidth = ((BITMAPINFOHEADER*)This->in[0]->strf->data)->biWidth; This->avih.dwHeight = ((BITMAPINFOHEADER*)This->in[0]->strf->data)->biHeight; This->filter.state = State_Running; return S_OK; } static HRESULT WINAPI AviMux_EnumPins(IBaseFilter *iface, IEnumPins **ppEnum) { AviMux *This = impl_from_IBaseFilter(iface); TRACE("(%p)->(%p)\n", This, ppEnum); return BaseFilterImpl_EnumPins(iface, ppEnum); } static HRESULT WINAPI AviMux_FindPin(IBaseFilter *iface, LPCWSTR Id, IPin **ppPin) { AviMux *This = impl_from_IBaseFilter(iface); int i; TRACE("(%p)->(%s %p)\n", This, debugstr_w(Id), ppPin); if(!Id || !ppPin) return E_POINTER; if(!lstrcmpiW(Id, This->out->pin.pin.pinInfo.achName)) { IPin_AddRef(&This->out->pin.pin.IPin_iface); *ppPin = &This->out->pin.pin.IPin_iface; return S_OK; } for(i=0; iinput_pin_no; i++) { if(lstrcmpiW(Id, This->in[i]->pin.pin.pinInfo.achName)) continue; IPin_AddRef(&This->in[i]->pin.pin.IPin_iface); *ppPin = &This->in[i]->pin.pin.IPin_iface; return S_OK; } return VFW_E_NOT_FOUND; } static HRESULT WINAPI AviMux_QueryFilterInfo(IBaseFilter *iface, FILTER_INFO *pInfo) { AviMux *This = impl_from_IBaseFilter(iface); FIXME("(%p)->(%p)\n", This, pInfo); return E_NOTIMPL; } static HRESULT WINAPI AviMux_QueryVendorInfo(IBaseFilter *iface, LPWSTR *pVendorInfo) { AviMux *This = impl_from_IBaseFilter(iface); FIXME("(%p)->(%p)\n", This, pVendorInfo); return E_NOTIMPL; } static const IBaseFilterVtbl AviMuxVtbl = { AviMux_QueryInterface, BaseFilterImpl_AddRef, AviMux_Release, BaseFilterImpl_GetClassID, AviMux_Stop, AviMux_Pause, AviMux_Run, BaseFilterImpl_GetState, BaseFilterImpl_SetSyncSource, BaseFilterImpl_GetSyncSource, AviMux_EnumPins, AviMux_FindPin, AviMux_QueryFilterInfo, BaseFilterImpl_JoinFilterGraph, AviMux_QueryVendorInfo }; static inline AviMux* impl_from_IConfigAviMux(IConfigAviMux *iface) { return CONTAINING_RECORD(iface, AviMux, IConfigAviMux_iface); } static HRESULT WINAPI ConfigAviMux_QueryInterface( IConfigAviMux *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_IConfigAviMux(iface); return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv); } static ULONG WINAPI ConfigAviMux_AddRef(IConfigAviMux *iface) { AviMux *This = impl_from_IConfigAviMux(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI ConfigAviMux_Release(IConfigAviMux *iface) { AviMux *This = impl_from_IConfigAviMux(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI ConfigAviMux_SetMasterStream(IConfigAviMux *iface, LONG iStream) { AviMux *This = impl_from_IConfigAviMux(iface); FIXME("(%p)->(%d)\n", This, iStream); return E_NOTIMPL; } static HRESULT WINAPI ConfigAviMux_GetMasterStream(IConfigAviMux *iface, LONG *pStream) { AviMux *This = impl_from_IConfigAviMux(iface); FIXME("(%p)->(%p)\n", This, pStream); return E_NOTIMPL; } static HRESULT WINAPI ConfigAviMux_SetOutputCompatibilityIndex( IConfigAviMux *iface, BOOL fOldIndex) { AviMux *This = impl_from_IConfigAviMux(iface); FIXME("(%p)->(%x)\n", This, fOldIndex); return E_NOTIMPL; } static HRESULT WINAPI ConfigAviMux_GetOutputCompatibilityIndex( IConfigAviMux *iface, BOOL *pfOldIndex) { AviMux *This = impl_from_IConfigAviMux(iface); FIXME("(%p)->(%p)\n", This, pfOldIndex); return E_NOTIMPL; } static const IConfigAviMuxVtbl ConfigAviMuxVtbl = { ConfigAviMux_QueryInterface, ConfigAviMux_AddRef, ConfigAviMux_Release, ConfigAviMux_SetMasterStream, ConfigAviMux_GetMasterStream, ConfigAviMux_SetOutputCompatibilityIndex, ConfigAviMux_GetOutputCompatibilityIndex }; static inline AviMux* impl_from_IConfigInterleaving(IConfigInterleaving *iface) { return CONTAINING_RECORD(iface, AviMux, IConfigInterleaving_iface); } static HRESULT WINAPI ConfigInterleaving_QueryInterface( IConfigInterleaving *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_IConfigInterleaving(iface); return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv); } static ULONG WINAPI ConfigInterleaving_AddRef(IConfigInterleaving *iface) { AviMux *This = impl_from_IConfigInterleaving(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI ConfigInterleaving_Release(IConfigInterleaving *iface) { AviMux *This = impl_from_IConfigInterleaving(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI ConfigInterleaving_put_Mode( IConfigInterleaving *iface, InterleavingMode mode) { AviMux *This = impl_from_IConfigInterleaving(iface); TRACE("(%p)->(%d)\n", This, mode); if(mode>INTERLEAVE_NONE_BUFFERED) return E_INVALIDARG; if(This->mode != mode) { if(This->out->pin.pin.pConnectedTo) { HRESULT hr = IFilterGraph_Reconnect(This->filter.filterInfo.pGraph, &This->out->pin.pin.IPin_iface); if(FAILED(hr)) return hr; } This->mode = mode; } return S_OK; } static HRESULT WINAPI ConfigInterleaving_get_Mode( IConfigInterleaving *iface, InterleavingMode *pMode) { AviMux *This = impl_from_IConfigInterleaving(iface); FIXME("(%p)->(%p)\n", This, pMode); return E_NOTIMPL; } static HRESULT WINAPI ConfigInterleaving_put_Interleaving(IConfigInterleaving *iface, const REFERENCE_TIME *prtInterleave, const REFERENCE_TIME *prtPreroll) { AviMux *This = impl_from_IConfigInterleaving(iface); TRACE("(%p)->(%p %p)\n", This, prtInterleave, prtPreroll); if(prtInterleave) This->interleave = *prtInterleave; if(prtPreroll) This->preroll = *prtPreroll; return S_OK; } static HRESULT WINAPI ConfigInterleaving_get_Interleaving(IConfigInterleaving *iface, REFERENCE_TIME *prtInterleave, REFERENCE_TIME *prtPreroll) { AviMux *This = impl_from_IConfigInterleaving(iface); FIXME("(%p)->(%p %p)\n", This, prtInterleave, prtPreroll); return E_NOTIMPL; } static const IConfigInterleavingVtbl ConfigInterleavingVtbl = { ConfigInterleaving_QueryInterface, ConfigInterleaving_AddRef, ConfigInterleaving_Release, ConfigInterleaving_put_Mode, ConfigInterleaving_get_Mode, ConfigInterleaving_put_Interleaving, ConfigInterleaving_get_Interleaving }; static inline AviMux* impl_from_IMediaSeeking(IMediaSeeking *iface) { return CONTAINING_RECORD(iface, AviMux, IMediaSeeking_iface); } static HRESULT WINAPI MediaSeeking_QueryInterface( IMediaSeeking *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_IMediaSeeking(iface); return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv); } static ULONG WINAPI MediaSeeking_AddRef(IMediaSeeking *iface) { AviMux *This = impl_from_IMediaSeeking(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI MediaSeeking_Release(IMediaSeeking *iface) { AviMux *This = impl_from_IMediaSeeking(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI MediaSeeking_GetCapabilities( IMediaSeeking *iface, DWORD *pCapabilities) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pCapabilities); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_CheckCapabilities( IMediaSeeking *iface, DWORD *pCapabilities) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pCapabilities); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_IsFormatSupported( IMediaSeeking *iface, const GUID *pFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%s)\n", This, debugstr_guid(pFormat)); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_QueryPreferredFormat( IMediaSeeking *iface, GUID *pFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pFormat); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetTimeFormat( IMediaSeeking *iface, GUID *pFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pFormat); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_IsUsingTimeFormat( IMediaSeeking *iface, const GUID *pFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%s)\n", This, debugstr_guid(pFormat)); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_SetTimeFormat( IMediaSeeking *iface, const GUID *pFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%s)\n", This, debugstr_guid(pFormat)); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetDuration( IMediaSeeking *iface, LONGLONG *pDuration) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pDuration); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetStopPosition( IMediaSeeking *iface, LONGLONG *pStop) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pStop); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetCurrentPosition( IMediaSeeking *iface, LONGLONG *pCurrent) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pCurrent); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_ConvertTimeFormat(IMediaSeeking *iface, LONGLONG *pTarget, const GUID *pTargetFormat, LONGLONG Source, const GUID *pSourceFormat) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p %s %s %s)\n", This, pTarget, debugstr_guid(pTargetFormat), wine_dbgstr_longlong(Source), debugstr_guid(pSourceFormat)); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_SetPositions(IMediaSeeking *iface, LONGLONG *pCurrent, DWORD dwCurrentFlags, LONGLONG *pStop, DWORD dwStopFlags) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p %x %p %x)\n", This, pCurrent, dwCurrentFlags, pStop, dwStopFlags); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetPositions(IMediaSeeking *iface, LONGLONG *pCurrent, LONGLONG *pStop) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p %p)\n", This, pCurrent, pStop); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetAvailable(IMediaSeeking *iface, LONGLONG *pEarliest, LONGLONG *pLatest) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p %p)\n", This, pEarliest, pLatest); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_SetRate(IMediaSeeking *iface, double dRate) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%lf)\n", This, dRate); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetRate(IMediaSeeking *iface, double *pdRate) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pdRate); return E_NOTIMPL; } static HRESULT WINAPI MediaSeeking_GetPreroll(IMediaSeeking *iface, LONGLONG *pllPreroll) { AviMux *This = impl_from_IMediaSeeking(iface); FIXME("(%p)->(%p)\n", This, pllPreroll); return E_NOTIMPL; } static const IMediaSeekingVtbl MediaSeekingVtbl = { MediaSeeking_QueryInterface, MediaSeeking_AddRef, MediaSeeking_Release, MediaSeeking_GetCapabilities, MediaSeeking_CheckCapabilities, MediaSeeking_IsFormatSupported, MediaSeeking_QueryPreferredFormat, MediaSeeking_GetTimeFormat, MediaSeeking_IsUsingTimeFormat, MediaSeeking_SetTimeFormat, MediaSeeking_GetDuration, MediaSeeking_GetStopPosition, MediaSeeking_GetCurrentPosition, MediaSeeking_ConvertTimeFormat, MediaSeeking_SetPositions, MediaSeeking_GetPositions, MediaSeeking_GetAvailable, MediaSeeking_SetRate, MediaSeeking_GetRate, MediaSeeking_GetPreroll }; static inline AviMux* impl_from_IPersistMediaPropertyBag(IPersistMediaPropertyBag *iface) { return CONTAINING_RECORD(iface, AviMux, IPersistMediaPropertyBag_iface); } static HRESULT WINAPI PersistMediaPropertyBag_QueryInterface( IPersistMediaPropertyBag *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv); } static ULONG WINAPI PersistMediaPropertyBag_AddRef(IPersistMediaPropertyBag *iface) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI PersistMediaPropertyBag_Release(IPersistMediaPropertyBag *iface) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI PersistMediaPropertyBag_GetClassID( IPersistMediaPropertyBag *iface, CLSID *pClassID) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); return IBaseFilter_GetClassID(&This->filter.IBaseFilter_iface, pClassID); } static HRESULT WINAPI PersistMediaPropertyBag_InitNew(IPersistMediaPropertyBag *iface) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); FIXME("(%p)->()\n", This); return E_NOTIMPL; } static HRESULT WINAPI PersistMediaPropertyBag_Load(IPersistMediaPropertyBag *iface, IMediaPropertyBag *pPropBag, IErrorLog *pErrorLog) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); FIXME("(%p)->()\n", This); return E_NOTIMPL; } static HRESULT WINAPI PersistMediaPropertyBag_Save(IPersistMediaPropertyBag *iface, IMediaPropertyBag *pPropBag, BOOL fClearDirty, BOOL fSaveAllProperties) { AviMux *This = impl_from_IPersistMediaPropertyBag(iface); FIXME("(%p)->()\n", This); return E_NOTIMPL; } static const IPersistMediaPropertyBagVtbl PersistMediaPropertyBagVtbl = { PersistMediaPropertyBag_QueryInterface, PersistMediaPropertyBag_AddRef, PersistMediaPropertyBag_Release, PersistMediaPropertyBag_GetClassID, PersistMediaPropertyBag_InitNew, PersistMediaPropertyBag_Load, PersistMediaPropertyBag_Save }; static inline AviMux* impl_from_ISpecifyPropertyPages(ISpecifyPropertyPages *iface) { return CONTAINING_RECORD(iface, AviMux, ISpecifyPropertyPages_iface); } static HRESULT WINAPI SpecifyPropertyPages_QueryInterface( ISpecifyPropertyPages *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_ISpecifyPropertyPages(iface); return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv); } static ULONG WINAPI SpecifyPropertyPages_AddRef(ISpecifyPropertyPages *iface) { AviMux *This = impl_from_ISpecifyPropertyPages(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI SpecifyPropertyPages_Release(ISpecifyPropertyPages *iface) { AviMux *This = impl_from_ISpecifyPropertyPages(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI SpecifyPropertyPages_GetPages( ISpecifyPropertyPages *iface, CAUUID *pPages) { AviMux *This = impl_from_ISpecifyPropertyPages(iface); FIXME("(%p)->(%p)\n", This, pPages); return E_NOTIMPL; } static const ISpecifyPropertyPagesVtbl SpecifyPropertyPagesVtbl = { SpecifyPropertyPages_QueryInterface, SpecifyPropertyPages_AddRef, SpecifyPropertyPages_Release, SpecifyPropertyPages_GetPages }; static HRESULT WINAPI AviMuxOut_CheckMediaType(BasePin *base, const AM_MEDIA_TYPE *amt) { FIXME("(%p) stub\n", base); return S_OK; } static HRESULT WINAPI AviMuxOut_AttemptConnection(BaseOutputPin *base, IPin *pReceivePin, const AM_MEDIA_TYPE *pmt) { PIN_DIRECTION dir; HRESULT hr; TRACE("(%p)->(%p AM_MEDIA_TYPE(%p))\n", base, pReceivePin, pmt); dump_AM_MEDIA_TYPE(pmt); hr = IPin_QueryDirection(pReceivePin, &dir); if(hr==S_OK && dir!=PINDIR_INPUT) return VFW_E_INVALID_DIRECTION; return BaseOutputPinImpl_AttemptConnection(base, pReceivePin, pmt); } static HRESULT WINAPI AviMuxOut_GetMediaType(BasePin *base, int iPosition, AM_MEDIA_TYPE *amt) { TRACE("(%p)->(%d %p)\n", base, iPosition, amt); if(iPosition < 0) return E_INVALIDARG; if(iPosition > 0) return VFW_S_NO_MORE_ITEMS; amt->majortype = MEDIATYPE_Stream; amt->subtype = MEDIASUBTYPE_Avi; amt->bFixedSizeSamples = TRUE; amt->bTemporalCompression = FALSE; amt->lSampleSize = 1; amt->formattype = GUID_NULL; amt->pUnk = NULL; amt->cbFormat = 0; amt->pbFormat = NULL; return S_OK; } static HRESULT WINAPI AviMuxOut_DecideAllocator(BaseOutputPin *base, IMemInputPin *pPin, IMemAllocator **pAlloc) { ALLOCATOR_PROPERTIES req, actual; HRESULT hr; TRACE("(%p)->(%p %p)\n", base, pPin, pAlloc); hr = BaseOutputPinImpl_InitAllocator(base, pAlloc); if(FAILED(hr)) return hr; hr = IMemInputPin_GetAllocatorRequirements(pPin, &req); if(FAILED(hr)) req.cbAlign = 1; req.cBuffers = 32; req.cbBuffer = 0; req.cbPrefix = 0; hr = IMemAllocator_SetProperties(*pAlloc, &req, &actual); if(FAILED(hr)) return hr; return IMemInputPin_NotifyAllocator(pPin, *pAlloc, TRUE); } static const BaseOutputPinFuncTable AviMuxOut_BaseOutputFuncTable = { { AviMuxOut_CheckMediaType, AviMuxOut_GetMediaType }, AviMuxOut_AttemptConnection, NULL, AviMuxOut_DecideAllocator, }; static inline AviMux* impl_from_out_IPin(IPin *iface) { BasePin *bp = CONTAINING_RECORD(iface, BasePin, IPin_iface); IBaseFilter *bf = bp->pinInfo.pFilter; return impl_from_IBaseFilter(bf); } static HRESULT WINAPI AviMuxOut_QueryInterface(IPin *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv); if(IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IPin)) *ppv = iface; else if(IsEqualIID(riid, &IID_IQualityControl)) *ppv = &This->out->IQualityControl_iface; else { FIXME("no interface for %s\n", debugstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } static ULONG WINAPI AviMuxOut_AddRef(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxOut_Release(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxOut_Connect(IPin *iface, IPin *pReceivePin, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_out_IPin(iface); HRESULT hr; int i; TRACE("(%p)->(%p AM_MEDIA_TYPE(%p))\n", This, pReceivePin, pmt); dump_AM_MEDIA_TYPE(pmt); hr = BaseOutputPinImpl_Connect(iface, pReceivePin, pmt); if(FAILED(hr)) return hr; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.pConnectedTo) continue; hr = IFilterGraph_Reconnect(This->filter.filterInfo.pGraph, &This->in[i]->pin.pin.IPin_iface); if(FAILED(hr)) { BaseOutputPinImpl_Disconnect(iface); break; } } if(hr == S_OK) IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); return hr; } static HRESULT WINAPI AviMuxOut_ReceiveConnection(IPin *iface, IPin *pConnector, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p AM_MEDIA_TYPE(%p)\n", This, pConnector, pmt); dump_AM_MEDIA_TYPE(pmt); return BaseOutputPinImpl_ReceiveConnection(iface, pConnector, pmt); } static HRESULT WINAPI AviMuxOut_Disconnect(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); HRESULT hr; TRACE("(%p)\n", This); hr = BaseOutputPinImpl_Disconnect(iface); if(hr == S_OK) IBaseFilter_Release(&This->filter.IBaseFilter_iface); return hr; } static HRESULT WINAPI AviMuxOut_ConnectedTo(IPin *iface, IPin **pPin) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, pPin); return BasePinImpl_ConnectedTo(iface, pPin); } static HRESULT WINAPI AviMuxOut_ConnectionMediaType(IPin *iface, AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, pmt); return BasePinImpl_ConnectionMediaType(iface, pmt); } static HRESULT WINAPI AviMuxOut_QueryPinInfo(IPin *iface, PIN_INFO *pInfo) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, pInfo); return BasePinImpl_QueryPinInfo(iface, pInfo); } static HRESULT WINAPI AviMuxOut_QueryDirection(IPin *iface, PIN_DIRECTION *pPinDir) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, pPinDir); return BasePinImpl_QueryDirection(iface, pPinDir); } static HRESULT WINAPI AviMuxOut_QueryId(IPin *iface, LPWSTR *Id) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, Id); return BasePinImpl_QueryId(iface, Id); } static HRESULT WINAPI AviMuxOut_QueryAccept(IPin *iface, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(AM_MEDIA_TYPE(%p))\n", This, pmt); dump_AM_MEDIA_TYPE(pmt); return BasePinImpl_QueryAccept(iface, pmt); } static HRESULT WINAPI AviMuxOut_EnumMediaTypes(IPin *iface, IEnumMediaTypes **ppEnum) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%p)\n", This, ppEnum); return BasePinImpl_EnumMediaTypes(iface, ppEnum); } static HRESULT WINAPI AviMuxOut_QueryInternalConnections( IPin *iface, IPin **apPin, ULONG *nPin) { AviMux *This = impl_from_out_IPin(iface); FIXME("(%p)->(%p %p)\n", This, apPin, nPin); return E_NOTIMPL; } static HRESULT WINAPI AviMuxOut_EndOfStream(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)\n", This); return BaseOutputPinImpl_EndOfStream(iface); } static HRESULT WINAPI AviMuxOut_BeginFlush(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)\n", This); return BaseOutputPinImpl_BeginFlush(iface); } static HRESULT WINAPI AviMuxOut_EndFlush(IPin *iface) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)\n", This); return BaseOutputPinImpl_EndFlush(iface); } static HRESULT WINAPI AviMuxOut_NewSegment(IPin *iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate) { AviMux *This = impl_from_out_IPin(iface); TRACE("(%p)->(%s %s %f)\n", This, wine_dbgstr_longlong(tStart), wine_dbgstr_longlong(tStop), dRate); return BasePinImpl_NewSegment(iface, tStart, tStop, dRate); } static const IPinVtbl AviMuxOut_PinVtbl = { AviMuxOut_QueryInterface, AviMuxOut_AddRef, AviMuxOut_Release, AviMuxOut_Connect, AviMuxOut_ReceiveConnection, AviMuxOut_Disconnect, AviMuxOut_ConnectedTo, AviMuxOut_ConnectionMediaType, AviMuxOut_QueryPinInfo, AviMuxOut_QueryDirection, AviMuxOut_QueryId, AviMuxOut_QueryAccept, AviMuxOut_EnumMediaTypes, AviMuxOut_QueryInternalConnections, AviMuxOut_EndOfStream, AviMuxOut_BeginFlush, AviMuxOut_EndFlush, AviMuxOut_NewSegment }; static inline AviMux* impl_from_out_IQualityControl(IQualityControl *iface) { AviMuxOut *amo = CONTAINING_RECORD(iface, AviMuxOut, IQualityControl_iface); return impl_from_IBaseFilter(amo->pin.pin.pinInfo.pFilter); } static HRESULT WINAPI AviMuxOut_QualityControl_QueryInterface( IQualityControl *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_out_IQualityControl(iface); return IPin_QueryInterface(&This->out->pin.pin.IPin_iface, riid, ppv); } static ULONG WINAPI AviMuxOut_QualityControl_AddRef(IQualityControl *iface) { AviMux *This = impl_from_out_IQualityControl(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxOut_QualityControl_Release(IQualityControl *iface) { AviMux *This = impl_from_out_IQualityControl(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxOut_QualityControl_Notify(IQualityControl *iface, IBaseFilter *pSelf, Quality q) { AviMux *This = impl_from_out_IQualityControl(iface); FIXME("(%p)->(%p { 0x%x %u %s %s })\n", This, pSelf, q.Type, q.Proportion, wine_dbgstr_longlong(q.Late), wine_dbgstr_longlong(q.TimeStamp)); return E_NOTIMPL; } static HRESULT WINAPI AviMuxOut_QualityControl_SetSink( IQualityControl *iface, IQualityControl *piqc) { AviMux *This = impl_from_out_IQualityControl(iface); FIXME("(%p)->(%p)\n", This, piqc); return E_NOTIMPL; } static const IQualityControlVtbl AviMuxOut_QualityControlVtbl = { AviMuxOut_QualityControl_QueryInterface, AviMuxOut_QualityControl_AddRef, AviMuxOut_QualityControl_Release, AviMuxOut_QualityControl_Notify, AviMuxOut_QualityControl_SetSink }; static HRESULT WINAPI AviMuxIn_CheckMediaType(BasePin *base, const AM_MEDIA_TYPE *pmt) { TRACE("(%p:%s)->(AM_MEDIA_TYPE(%p))\n", base, debugstr_w(base->pinInfo.achName), pmt); dump_AM_MEDIA_TYPE(pmt); if(IsEqualIID(&pmt->majortype, &MEDIATYPE_Audio) && IsEqualIID(&pmt->formattype, &FORMAT_WaveFormatEx)) return S_OK; if(IsEqualIID(&pmt->majortype, &MEDIATYPE_Interleaved) && IsEqualIID(&pmt->formattype, &FORMAT_DvInfo)) return S_OK; if(IsEqualIID(&pmt->majortype, &MEDIATYPE_Video) && (IsEqualIID(&pmt->formattype, &FORMAT_VideoInfo) || IsEqualIID(&pmt->formattype, &FORMAT_DvInfo))) return S_OK; return S_FALSE; } static HRESULT WINAPI AviMuxIn_GetMediaType(BasePin *base, int iPosition, AM_MEDIA_TYPE *amt) { return S_FALSE; } static HRESULT WINAPI AviMuxIn_Receive(BaseInputPin *base, IMediaSample *pSample) { AviMuxIn *avimuxin = CONTAINING_RECORD(base, AviMuxIn, pin); AviMux *avimux = impl_from_IBaseFilter(base->pin.pinInfo.pFilter); REFERENCE_TIME start, stop; IMediaSample *sample; int frames_no; IMediaSample2 *ms2; BYTE *frame, *buf; DWORD max_size, size; DWORD flags; HRESULT hr; TRACE("(%p:%s)->(%p)\n", base, debugstr_w(base->pin.pinInfo.achName), pSample); hr = IMediaSample_QueryInterface(pSample, &IID_IMediaSample2, (void**)&ms2); if(SUCCEEDED(hr)) { AM_SAMPLE2_PROPERTIES props; memset(&props, 0, sizeof(props)); hr = IMediaSample2_GetProperties(ms2, sizeof(props), (BYTE*)&props); IMediaSample2_Release(ms2); if(FAILED(hr)) return hr; flags = props.dwSampleFlags; frame = props.pbBuffer; size = props.lActual; }else { flags = IMediaSample_IsSyncPoint(pSample) == S_OK ? AM_SAMPLE_SPLICEPOINT : 0; hr = IMediaSample_GetPointer(pSample, &frame); if(FAILED(hr)) return hr; size = IMediaSample_GetActualDataLength(pSample); } if(!avimuxin->pin.pin.mtCurrent.bTemporalCompression) flags |= AM_SAMPLE_SPLICEPOINT; hr = IMediaSample_GetTime(pSample, &start, &stop); if(FAILED(hr)) return hr; if(avimuxin->stop>stop) return VFW_E_START_TIME_AFTER_END; if(avimux->start == -1) avimux->start = start; if(avimux->stop < stop) avimux->stop = stop; if(avimux->avih.dwSuggestedBufferSize < ALIGN(size)+sizeof(RIFFCHUNK)) avimux->avih.dwSuggestedBufferSize = ALIGN(size) + sizeof(RIFFCHUNK); if(avimuxin->strh.dwSuggestedBufferSize < ALIGN(size)+sizeof(RIFFCHUNK)) avimuxin->strh.dwSuggestedBufferSize = ALIGN(size) + sizeof(RIFFCHUNK); frames_no = 1; if(avimuxin->stop!=-1 && start > avimuxin->stop) { frames_no += (double)(start - avimuxin->stop) / 10000000 * avimuxin->strh.dwRate / avimuxin->strh.dwScale + 0.5; } avimuxin->stop = stop; while(--frames_no) { /* TODO: store all control frames in one buffer */ hr = IMemAllocator_GetBuffer(avimuxin->samples_allocator, &sample, NULL, NULL, 0); if(FAILED(hr)) return hr; hr = IMediaSample_SetActualDataLength(sample, 0); if(SUCCEEDED(hr)) hr = IMediaSample_SetDiscontinuity(sample, TRUE); if(SUCCEEDED(hr)) hr = IMediaSample_SetSyncPoint(sample, FALSE); if(SUCCEEDED(hr)) hr = queue_sample(avimux, avimuxin, sample); IMediaSample_Release(sample); if(FAILED(hr)) return hr; } hr = IMemAllocator_GetBuffer(avimuxin->samples_allocator, &sample, NULL, NULL, 0); if(FAILED(hr)) return hr; max_size = IMediaSample_GetSize(sample); if(size > max_size) size = max_size; hr = IMediaSample_SetActualDataLength(sample, size); if(SUCCEEDED(hr)) hr = IMediaSample_SetDiscontinuity(sample, FALSE); if(SUCCEEDED(hr)) hr = IMediaSample_SetSyncPoint(sample, flags & AM_SAMPLE_SPLICEPOINT); /* TODO: avoid unnecessary copying */ if(SUCCEEDED(hr)) hr = IMediaSample_GetPointer(sample, &buf); if(SUCCEEDED(hr)) { memcpy(buf, frame, size); hr = queue_sample(avimux, avimuxin, sample); } IMediaSample_Release(sample); return hr; } static const BaseInputPinFuncTable AviMuxIn_BaseInputFuncTable = { { AviMuxIn_CheckMediaType, AviMuxIn_GetMediaType }, AviMuxIn_Receive }; static inline AviMux* impl_from_in_IPin(IPin *iface) { BasePin *bp = CONTAINING_RECORD(iface, BasePin, IPin_iface); IBaseFilter *bf = bp->pinInfo.pFilter; return impl_from_IBaseFilter(bf); } static inline AviMuxIn* AviMuxIn_from_IPin(IPin *iface) { BasePin *bp = CONTAINING_RECORD(iface, BasePin, IPin_iface); BaseInputPin *bip = CONTAINING_RECORD(bp, BaseInputPin, pin); return CONTAINING_RECORD(bip, AviMuxIn, pin); } static HRESULT WINAPI AviMuxIn_QueryInterface(IPin *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%s %p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), debugstr_guid(riid), ppv); if(IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IPin)) *ppv = &avimuxin->pin.pin.IPin_iface; else if(IsEqualIID(riid, &IID_IAMStreamControl)) *ppv = &avimuxin->IAMStreamControl_iface; else if(IsEqualIID(riid, &IID_IMemInputPin)) *ppv = &avimuxin->pin.IMemInputPin_iface; else if(IsEqualIID(riid, &IID_IPropertyBag)) *ppv = &avimuxin->IPropertyBag_iface; else if(IsEqualIID(riid, &IID_IQualityControl)) *ppv = &avimuxin->IQualityControl_iface; else { FIXME("no interface for %s\n", debugstr_guid(riid)); *ppv = NULL; return E_NOINTERFACE; } IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } static ULONG WINAPI AviMuxIn_AddRef(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxIn_Release(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxIn_Connect(IPin *iface, IPin *pReceivePin, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p AM_MEDIA_TYPE(%p))\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pReceivePin, pmt); dump_AM_MEDIA_TYPE(pmt); return BaseInputPinImpl_Connect(iface, pReceivePin, pmt); } static HRESULT WINAPI AviMuxIn_ReceiveConnection(IPin *iface, IPin *pConnector, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); HRESULT hr; TRACE("(%p:%s)->(%p AM_MEDIA_TYPE(%p))\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pConnector, pmt); dump_AM_MEDIA_TYPE(pmt); if(!pmt) return E_POINTER; hr = BaseInputPinImpl_ReceiveConnection(iface, pConnector, pmt); if(FAILED(hr)) return hr; if(IsEqualIID(&pmt->majortype, &MEDIATYPE_Video) && IsEqualIID(&pmt->formattype, &FORMAT_VideoInfo)) { ALLOCATOR_PROPERTIES req, act; VIDEOINFOHEADER *vih; int size; vih = (VIDEOINFOHEADER*)pmt->pbFormat; avimuxin->strh.fcc = ckidSTREAMHEADER; avimuxin->strh.cb = sizeof(AVISTREAMHEADER) - FIELD_OFFSET(AVISTREAMHEADER, fccType); avimuxin->strh.fccType = streamtypeVIDEO; /* FIXME: fccHandler should be set differently */ avimuxin->strh.fccHandler = vih->bmiHeader.biCompression ? vih->bmiHeader.biCompression : FCC('D','I','B',' '); avimuxin->avg_time_per_frame = vih->AvgTimePerFrame; avimuxin->stop = -1; req.cBuffers = 32; req.cbBuffer = vih->bmiHeader.biSizeImage; req.cbAlign = 1; req.cbPrefix = sizeof(void*); hr = IMemAllocator_SetProperties(avimuxin->samples_allocator, &req, &act); if(SUCCEEDED(hr)) hr = IMemAllocator_Commit(avimuxin->samples_allocator); if(FAILED(hr)) { BasePinImpl_Disconnect(iface); return hr; } size = pmt->cbFormat - FIELD_OFFSET(VIDEOINFOHEADER, bmiHeader); avimuxin->strf = CoTaskMemAlloc(sizeof(RIFFCHUNK) + ALIGN(FIELD_OFFSET(BITMAPINFO, bmiColors[vih->bmiHeader.biClrUsed]))); avimuxin->strf->fcc = ckidSTREAMFORMAT; avimuxin->strf->cb = FIELD_OFFSET(BITMAPINFO, bmiColors[vih->bmiHeader.biClrUsed]); if(size > avimuxin->strf->cb) size = avimuxin->strf->cb; memcpy(avimuxin->strf->data, &vih->bmiHeader, size); }else { FIXME("format not supported: %s %s\n", debugstr_guid(&pmt->majortype), debugstr_guid(&pmt->formattype)); return E_NOTIMPL; } return create_input_pin(This); } static HRESULT WINAPI AviMuxIn_Disconnect(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); IMediaSample **prev, *cur; HRESULT hr; TRACE("(%p:%s)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName)); hr = BasePinImpl_Disconnect(iface); if(FAILED(hr)) return hr; IMemAllocator_Decommit(avimuxin->samples_allocator); while(avimuxin->samples_head) { cur = avimuxin->samples_head; hr = IMediaSample_GetPointer(cur, (BYTE**)&prev); if(FAILED(hr)) break; prev--; cur = avimuxin->samples_head; avimuxin->samples_head = *prev; IMediaSample_Release(cur); if(cur == avimuxin->samples_head) avimuxin->samples_head = NULL; } CoTaskMemFree(avimuxin->strf); avimuxin->strf = NULL; return hr; } static HRESULT WINAPI AviMuxIn_ConnectedTo(IPin *iface, IPin **pPin) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pPin); return BasePinImpl_ConnectedTo(iface, pPin); } static HRESULT WINAPI AviMuxIn_ConnectionMediaType(IPin *iface, AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pmt); return BasePinImpl_ConnectionMediaType(iface, pmt); } static HRESULT WINAPI AviMuxIn_QueryPinInfo(IPin *iface, PIN_INFO *pInfo) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pInfo); return BasePinImpl_QueryPinInfo(iface, pInfo); } static HRESULT WINAPI AviMuxIn_QueryDirection(IPin *iface, PIN_DIRECTION *pPinDir) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pPinDir); return BasePinImpl_QueryDirection(iface, pPinDir); } static HRESULT WINAPI AviMuxIn_QueryId(IPin *iface, LPWSTR *Id) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), Id); return BasePinImpl_QueryId(iface, Id); } static HRESULT WINAPI AviMuxIn_QueryAccept(IPin *iface, const AM_MEDIA_TYPE *pmt) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(AM_MEDIA_TYPE(%p))\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pmt); dump_AM_MEDIA_TYPE(pmt); return BasePinImpl_QueryAccept(iface, pmt); } static HRESULT WINAPI AviMuxIn_EnumMediaTypes(IPin *iface, IEnumMediaTypes **ppEnum) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), ppEnum); return BasePinImpl_EnumMediaTypes(iface, ppEnum); } static HRESULT WINAPI AviMuxIn_QueryInternalConnections( IPin *iface, IPin **apPin, ULONG *nPin) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%p %p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), apPin, nPin); return BasePinImpl_QueryInternalConnections(iface, apPin, nPin); } static HRESULT WINAPI AviMuxIn_EndOfStream(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName)); return BaseInputPinImpl_EndOfStream(iface); } static HRESULT WINAPI AviMuxIn_BeginFlush(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName)); return BaseInputPinImpl_BeginFlush(iface); } static HRESULT WINAPI AviMuxIn_EndFlush(IPin *iface) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName)); return BaseInputPinImpl_EndFlush(iface); } static HRESULT WINAPI AviMuxIn_NewSegment(IPin *iface, REFERENCE_TIME tStart, REFERENCE_TIME tStop, double dRate) { AviMux *This = impl_from_in_IPin(iface); AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("(%p:%s)->(%s %s %f)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), wine_dbgstr_longlong(tStart), wine_dbgstr_longlong(tStop), dRate); return BasePinImpl_NewSegment(iface, tStart, tStop, dRate); } static const IPinVtbl AviMuxIn_PinVtbl = { AviMuxIn_QueryInterface, AviMuxIn_AddRef, AviMuxIn_Release, AviMuxIn_Connect, AviMuxIn_ReceiveConnection, AviMuxIn_Disconnect, AviMuxIn_ConnectedTo, AviMuxIn_ConnectionMediaType, AviMuxIn_QueryPinInfo, AviMuxIn_QueryDirection, AviMuxIn_QueryId, AviMuxIn_QueryAccept, AviMuxIn_EnumMediaTypes, AviMuxIn_QueryInternalConnections, AviMuxIn_EndOfStream, AviMuxIn_BeginFlush, AviMuxIn_EndFlush, AviMuxIn_NewSegment }; static inline AviMuxIn* AviMuxIn_from_IAMStreamControl(IAMStreamControl *iface) { return CONTAINING_RECORD(iface, AviMuxIn, IAMStreamControl_iface); } static HRESULT WINAPI AviMuxIn_AMStreamControl_QueryInterface( IAMStreamControl *iface, REFIID riid, void **ppv) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); return IPin_QueryInterface(&avimuxin->pin.pin.IPin_iface, riid, ppv); } static ULONG WINAPI AviMuxIn_AMStreamControl_AddRef(IAMStreamControl *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxIn_AMStreamControl_Release(IAMStreamControl *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxIn_AMStreamControl_StartAt(IAMStreamControl *iface, const REFERENCE_TIME *ptStart, DWORD dwCookie) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%p %x)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), ptStart, dwCookie); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_AMStreamControl_StopAt(IAMStreamControl *iface, const REFERENCE_TIME *ptStop, BOOL bSendExtra, DWORD dwCookie) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%p %x %x)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), ptStop, bSendExtra, dwCookie); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_AMStreamControl_GetInfo( IAMStreamControl *iface, AM_STREAM_INFO *pInfo) { AviMuxIn *avimuxin = AviMuxIn_from_IAMStreamControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pInfo); return E_NOTIMPL; } static const IAMStreamControlVtbl AviMuxIn_AMStreamControlVtbl = { AviMuxIn_AMStreamControl_QueryInterface, AviMuxIn_AMStreamControl_AddRef, AviMuxIn_AMStreamControl_Release, AviMuxIn_AMStreamControl_StartAt, AviMuxIn_AMStreamControl_StopAt, AviMuxIn_AMStreamControl_GetInfo }; static inline AviMuxIn* AviMuxIn_from_IMemInputPin(IMemInputPin *iface) { BaseInputPin *bip = CONTAINING_RECORD(iface, BaseInputPin, IMemInputPin_iface); return CONTAINING_RECORD(bip, AviMuxIn, pin); } static HRESULT WINAPI AviMuxIn_MemInputPin_QueryInterface( IMemInputPin *iface, REFIID riid, void **ppv) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); return IPin_QueryInterface(&avimuxin->pin.pin.IPin_iface, riid, ppv); } static ULONG WINAPI AviMuxIn_MemInputPin_AddRef(IMemInputPin *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxIn_MemInputPin_Release(IMemInputPin *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxIn_MemInputPin_GetAllocator( IMemInputPin *iface, IMemAllocator **ppAllocator) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), ppAllocator); if(!ppAllocator) return E_POINTER; IMemAllocator_AddRef(avimuxin->pin.pAllocator); *ppAllocator = avimuxin->pin.pAllocator; return S_OK; } static HRESULT WINAPI AviMuxIn_MemInputPin_NotifyAllocator( IMemInputPin *iface, IMemAllocator *pAllocator, BOOL bReadOnly) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); ALLOCATOR_PROPERTIES props; HRESULT hr; TRACE("(%p:%s)->(%p %x)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pAllocator, bReadOnly); if(!pAllocator) return E_POINTER; memset(&props, 0, sizeof(props)); hr = IMemAllocator_GetProperties(pAllocator, &props); if(FAILED(hr)) return hr; props.cbAlign = 1; props.cbPrefix = 8; return IMemAllocator_SetProperties(avimuxin->pin.pAllocator, &props, &props); } static HRESULT WINAPI AviMuxIn_MemInputPin_GetAllocatorRequirements( IMemInputPin *iface, ALLOCATOR_PROPERTIES *pProps) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pProps); if(!pProps) return E_POINTER; pProps->cbAlign = 1; pProps->cbPrefix = 8; return S_OK; } static HRESULT WINAPI AviMuxIn_MemInputPin_Receive( IMemInputPin *iface, IMediaSample *pSample) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); TRACE("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pSample); return avimuxin->pin.pFuncsTable->pfnReceive(&avimuxin->pin, pSample); } static HRESULT WINAPI AviMuxIn_MemInputPin_ReceiveMultiple(IMemInputPin *iface, IMediaSample **pSamples, LONG nSamples, LONG *nSamplesProcessed) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); HRESULT hr = S_OK; TRACE("(%p:%s)->(%p %d %p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pSamples, nSamples, nSamplesProcessed); for(*nSamplesProcessed=0; *nSamplesProcessedpin.pFuncsTable->pfnReceive(&avimuxin->pin, pSamples[*nSamplesProcessed]); if(hr != S_OK) break; } return hr; } static HRESULT WINAPI AviMuxIn_MemInputPin_ReceiveCanBlock(IMemInputPin *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IMemInputPin(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); HRESULT hr; TRACE("(%p:%s)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName)); if(!This->out->pin.pMemInputPin) return S_FALSE; hr = IMemInputPin_ReceiveCanBlock(This->out->pin.pMemInputPin); return hr != S_FALSE ? S_OK : S_FALSE; } static const IMemInputPinVtbl AviMuxIn_MemInputPinVtbl = { AviMuxIn_MemInputPin_QueryInterface, AviMuxIn_MemInputPin_AddRef, AviMuxIn_MemInputPin_Release, AviMuxIn_MemInputPin_GetAllocator, AviMuxIn_MemInputPin_NotifyAllocator, AviMuxIn_MemInputPin_GetAllocatorRequirements, AviMuxIn_MemInputPin_Receive, AviMuxIn_MemInputPin_ReceiveMultiple, AviMuxIn_MemInputPin_ReceiveCanBlock }; static inline AviMuxIn* AviMuxIn_from_IPropertyBag(IPropertyBag *iface) { return CONTAINING_RECORD(iface, AviMuxIn, IPropertyBag_iface); } static HRESULT WINAPI AviMuxIn_PropertyBag_QueryInterface( IPropertyBag *iface, REFIID riid, void **ppv) { AviMuxIn *avimuxin = AviMuxIn_from_IPropertyBag(iface); return IPin_QueryInterface(&avimuxin->pin.pin.IPin_iface, riid, ppv); } static ULONG WINAPI AviMuxIn_PropertyBag_AddRef(IPropertyBag *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IPropertyBag(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxIn_PropertyBag_Release(IPropertyBag *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IPropertyBag(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxIn_PropertyBag_Read(IPropertyBag *iface, LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog *pErrorLog) { AviMuxIn *avimuxin = AviMuxIn_from_IPropertyBag(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%s %p %p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), debugstr_w(pszPropName), pVar, pErrorLog); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_PropertyBag_Write(IPropertyBag *iface, LPCOLESTR pszPropName, VARIANT *pVar) { AviMuxIn *avimuxin = AviMuxIn_from_IPropertyBag(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%s %p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), debugstr_w(pszPropName), pVar); return E_NOTIMPL; } static const IPropertyBagVtbl AviMuxIn_PropertyBagVtbl = { AviMuxIn_PropertyBag_QueryInterface, AviMuxIn_PropertyBag_AddRef, AviMuxIn_PropertyBag_Release, AviMuxIn_PropertyBag_Read, AviMuxIn_PropertyBag_Write }; static inline AviMuxIn* AviMuxIn_from_IQualityControl(IQualityControl *iface) { return CONTAINING_RECORD(iface, AviMuxIn, IQualityControl_iface); } static HRESULT WINAPI AviMuxIn_QualityControl_QueryInterface( IQualityControl *iface, REFIID riid, void **ppv) { AviMuxIn *avimuxin = AviMuxIn_from_IQualityControl(iface); return IPin_QueryInterface(&avimuxin->pin.pin.IPin_iface, riid, ppv); } static ULONG WINAPI AviMuxIn_QualityControl_AddRef(IQualityControl *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IQualityControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface); } static ULONG WINAPI AviMuxIn_QualityControl_Release(IQualityControl *iface) { AviMuxIn *avimuxin = AviMuxIn_from_IQualityControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); return IBaseFilter_Release(&This->filter.IBaseFilter_iface); } static HRESULT WINAPI AviMuxIn_QualityControl_Notify(IQualityControl *iface, IBaseFilter *pSelf, Quality q) { AviMuxIn *avimuxin = AviMuxIn_from_IQualityControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%p { 0x%x %u %s %s })\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), pSelf, q.Type, q.Proportion, wine_dbgstr_longlong(q.Late), wine_dbgstr_longlong(q.TimeStamp)); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_QualityControl_SetSink( IQualityControl *iface, IQualityControl *piqc) { AviMuxIn *avimuxin = AviMuxIn_from_IQualityControl(iface); AviMux *This = impl_from_in_IPin(&avimuxin->pin.pin.IPin_iface); FIXME("(%p:%s)->(%p)\n", This, debugstr_w(avimuxin->pin.pin.pinInfo.achName), piqc); return E_NOTIMPL; } static const IQualityControlVtbl AviMuxIn_QualityControlVtbl = { AviMuxIn_QualityControl_QueryInterface, AviMuxIn_QualityControl_AddRef, AviMuxIn_QualityControl_Release, AviMuxIn_QualityControl_Notify, AviMuxIn_QualityControl_SetSink }; static HRESULT create_input_pin(AviMux *avimux) { static const WCHAR name[] = {'I','n','p','u','t',' ','0','0',0}; PIN_INFO info; HRESULT hr; if(avimux->input_pin_no >= MAX_PIN_NO-1) return E_FAIL; info.dir = PINDIR_INPUT; info.pFilter = &avimux->filter.IBaseFilter_iface; memcpy(info.achName, name, sizeof(name)); info.achName[7] = '0' + (avimux->input_pin_no+1) % 10; info.achName[6] = '0' + (avimux->input_pin_no+1) / 10; hr = BaseInputPin_Construct(&AviMuxIn_PinVtbl, sizeof(AviMuxIn), &info, &AviMuxIn_BaseInputFuncTable, &avimux->filter.csFilter, NULL, (IPin**)&avimux->in[avimux->input_pin_no]); if(FAILED(hr)) return hr; avimux->in[avimux->input_pin_no]->pin.IMemInputPin_iface.lpVtbl = &AviMuxIn_MemInputPinVtbl; avimux->in[avimux->input_pin_no]->IAMStreamControl_iface.lpVtbl = &AviMuxIn_AMStreamControlVtbl; avimux->in[avimux->input_pin_no]->IPropertyBag_iface.lpVtbl = &AviMuxIn_PropertyBagVtbl; avimux->in[avimux->input_pin_no]->IQualityControl_iface.lpVtbl = &AviMuxIn_QualityControlVtbl; avimux->in[avimux->input_pin_no]->samples_head = NULL; hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (void**)&avimux->in[avimux->input_pin_no]->samples_allocator); if(FAILED(hr)) { BaseInputPinImpl_Release(&avimux->in[avimux->input_pin_no]->pin.pin.IPin_iface); return hr; } hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (void**)&avimux->in[avimux->input_pin_no]->pin.pAllocator); if(FAILED(hr)) { IMemAllocator_Release(avimux->in[avimux->input_pin_no]->samples_allocator); BaseInputPinImpl_Release(&avimux->in[avimux->input_pin_no]->pin.pin.IPin_iface); return hr; } avimux->in[avimux->input_pin_no]->stream_time = 0; memset(&avimux->in[avimux->input_pin_no]->strh, 0, sizeof(avimux->in[avimux->input_pin_no]->strh)); avimux->in[avimux->input_pin_no]->strf = NULL; memset(&avimux->in[avimux->input_pin_no]->indx_data, 0, sizeof(avimux->in[avimux->input_pin_no]->indx_data)); memset(&avimux->in[avimux->input_pin_no]->ix_data, 0, sizeof(avimux->in[avimux->input_pin_no]->ix_data)); avimux->in[avimux->input_pin_no]->indx = (AVISUPERINDEX*)&avimux->in[avimux->input_pin_no]->indx_data; avimux->in[avimux->input_pin_no]->ix = (AVISTDINDEX*)avimux->in[avimux->input_pin_no]->ix_data; avimux->input_pin_no++; return S_OK; } IUnknown* WINAPI QCAP_createAVIMux(IUnknown *pUnkOuter, HRESULT *phr) { static const WCHAR output_name[] = {'A','V','I',' ','O','u','t',0}; AviMux *avimux; PIN_INFO info; HRESULT hr; TRACE("(%p)\n", pUnkOuter); if(pUnkOuter) { *phr = CLASS_E_NOAGGREGATION; return NULL; } avimux = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(AviMux)); if(!avimux) { *phr = E_OUTOFMEMORY; return NULL; } BaseFilter_Init(&avimux->filter, &AviMuxVtbl, &CLSID_AviDest, (DWORD_PTR)(__FILE__ ": AviMux.csFilter"), &filter_func_table); avimux->IConfigAviMux_iface.lpVtbl = &ConfigAviMuxVtbl; avimux->IConfigInterleaving_iface.lpVtbl = &ConfigInterleavingVtbl; avimux->IMediaSeeking_iface.lpVtbl = &MediaSeekingVtbl; avimux->IPersistMediaPropertyBag_iface.lpVtbl = &PersistMediaPropertyBagVtbl; avimux->ISpecifyPropertyPages_iface.lpVtbl = &SpecifyPropertyPagesVtbl; info.dir = PINDIR_OUTPUT; info.pFilter = &avimux->filter.IBaseFilter_iface; lstrcpyW(info.achName, output_name); hr = BaseOutputPin_Construct(&AviMuxOut_PinVtbl, sizeof(AviMuxOut), &info, &AviMuxOut_BaseOutputFuncTable, &avimux->filter.csFilter, (IPin**)&avimux->out); if(FAILED(hr)) { strmbase_filter_cleanup(&avimux->filter); HeapFree(GetProcessHeap(), 0, avimux); *phr = hr; return NULL; } avimux->out->IQualityControl_iface.lpVtbl = &AviMuxOut_QualityControlVtbl; avimux->out->cur_stream = 0; avimux->out->cur_time = 0; avimux->out->stream = NULL; hr = create_input_pin(avimux); if(FAILED(hr)) { BaseOutputPinImpl_Release(&avimux->out->pin.pin.IPin_iface); strmbase_filter_cleanup(&avimux->filter); HeapFree(GetProcessHeap(), 0, avimux); *phr = hr; return NULL; } avimux->interleave = 10000000; ObjectRefCount(TRUE); *phr = S_OK; return (IUnknown*)&avimux->filter.IBaseFilter_iface; }