/* * 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 { struct strmbase_sink 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 { struct strmbase_filter filter; IConfigAviMux IConfigAviMux_iface; IConfigInterleaving IConfigInterleaving_iface; IMediaSeeking IMediaSeeking_iface; IPersistMediaPropertyBag IPersistMediaPropertyBag_iface; ISpecifyPropertyPages ISpecifyPropertyPages_iface; InterleavingMode mode; REFERENCE_TIME interleave; REFERENCE_TIME preroll; struct strmbase_source source; IQualityControl IQualityControl_iface; int input_pin_no; AviMuxIn *in[MAX_PIN_NO-1]; REFERENCE_TIME start, stop; AVIMAINHEADER avih; int idx1_entries; int idx1_size; AVIINDEXENTRY *idx1; int cur_stream; LONGLONG cur_time; int buf_pos; BYTE buf[65536]; int movi_off; int out_pos; int size; IStream *stream; } AviMux; static HRESULT create_input_pin(AviMux*); static inline AviMux* impl_from_strmbase_filter(struct strmbase_filter *filter) { return CONTAINING_RECORD(filter, AviMux, filter); } static struct strmbase_pin *avi_mux_get_pin(struct strmbase_filter *iface, unsigned int index) { AviMux *filter = impl_from_strmbase_filter(iface); if (!index) return &filter->source.pin; else if (index <= filter->input_pin_no) return &filter->in[index - 1]->pin.pin; return NULL; } static void avi_mux_destroy(struct strmbase_filter *iface) { AviMux *filter = impl_from_strmbase_filter(iface); int i; strmbase_source_cleanup(&filter->source); for (i = 0; i < filter->input_pin_no; ++i) { IPin_Disconnect(&filter->in[i]->pin.pin.IPin_iface); IMemAllocator_Release(filter->in[i]->samples_allocator); filter->in[i]->samples_allocator = NULL; strmbase_sink_cleanup(&filter->in[i]->pin); heap_free(filter->in[i]); } heap_free(filter->idx1); strmbase_filter_cleanup(&filter->filter); heap_free(filter); ObjectRefCount(FALSE); } static HRESULT avi_mux_query_interface(struct strmbase_filter *iface, REFIID iid, void **out) { AviMux *filter = impl_from_strmbase_filter(iface); if (IsEqualGUID(iid, &IID_IConfigAviMux)) *out = &filter->IConfigAviMux_iface; else if (IsEqualGUID(iid, &IID_IConfigInterleaving)) *out = &filter->IConfigInterleaving_iface; else if (IsEqualGUID(iid, &IID_IMediaSeeking)) *out = &filter->IMediaSeeking_iface; else if (IsEqualGUID(iid, &IID_IPersistMediaPropertyBag)) *out = &filter->IPersistMediaPropertyBag_iface; else if (IsEqualGUID(iid, &IID_ISpecifyPropertyPages)) *out = &filter->ISpecifyPropertyPages_iface; else return E_NOINTERFACE; IUnknown_AddRef((IUnknown *)*out); return S_OK; } static HRESULT out_flush(AviMux *This) { ULONG written; HRESULT hr; if(!This->buf_pos) return S_OK; hr = IStream_Write(This->stream, This->buf, This->buf_pos, &written); if(FAILED(hr)) return hr; if (written != This->buf_pos) return E_FAIL; This->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->stream, li, STREAM_SEEK_SET, NULL); if(FAILED(hr)) return hr; This->out_pos = pos; if(This->out_pos > This->size) This->size = This->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->buf) - This->buf_pos) chunk_size = sizeof(This->buf) - This->buf_pos; else chunk_size = size; memcpy(This->buf + This->buf_pos, data, chunk_size); size -= chunk_size; data = (const BYTE*)data + chunk_size; This->buf_pos += chunk_size; This->out_pos += chunk_size; if (This->out_pos > This->size) This->size = This->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->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->cur_time && !(flags & AM_SAMPLE_TIMEDISCONTINUITY)) { if(closing) break; avimux->cur_stream++; if(avimux->cur_stream >= avimux->input_pin_no-1) { avimux->cur_time += avimux->interleave; avimux->cur_stream = 0; } avimuxin = avimux->in[avimux->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->size; avimux->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->size; avimuxin->ix->aIndex[avimuxin->ix->nEntriesInUse].dwOffset = avimux->size + sizeof(RIFFCHUNK) - avimuxin->ix->qwBaseOffset; hr = out_seek(avimux, avimux->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->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 avi_mux_cleanup_stream(struct strmbase_filter *iface) { AviMux *This = impl_from_strmbase_filter(iface); HRESULT hr; int i; if (This->stream) { AVIEXTHEADER dmlh; RIFFCHUNK rc; RIFFLIST rl; int idx1_off, empty_stream; empty_stream = This->cur_stream; for(i=empty_stream+1; ; i++) { if(i >= This->input_pin_no-1) i = 0; if(i == empty_stream) break; This->cur_stream = i; hr = flush_queue(This, This->in[This->cur_stream], TRUE); if(FAILED(hr)) return hr; } idx1_off = This->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.peer) 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->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->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.peer) 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->movi_off - sizeof(RIFFCHUNK); rl.fccListType = FCC('m','o','v','i'); out_write(This, &rl, sizeof(rl)); out_flush(This); IStream_Release(This->stream); This->stream = NULL; } return S_OK; } static HRESULT avi_mux_init_stream(struct strmbase_filter *iface) { AviMux *This = impl_from_strmbase_filter(iface); HRESULT hr; int i, stream_id; if(This->mode != INTERLEAVE_FULL) { FIXME("mode not supported (%d)\n", This->mode); return E_NOTIMPL; } for(i=0; iinput_pin_no; i++) { IMediaSeeking *ms; LONGLONG cur, stop; if(!This->in[i]->pin.pin.peer) continue; hr = IPin_QueryInterface(This->in[i]->pin.pin.peer, &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->source.pMemInputPin) { hr = IMemInputPin_QueryInterface(This->source.pMemInputPin, &IID_IStream, (void **)&This->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->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.peer) continue; This->avih.dwStreams++; This->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->stream) { IStream_Release(This->stream); This->stream = NULL; } return hr; } } This->movi_off = This->size; This->size += sizeof(RIFFLIST); idx1_add_entry(This, FCC('7','F','x','x'), 0, This->movi_off + sizeof(RIFFLIST), 0); stream_id = 0; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.peer) continue; This->in[i]->ix_off = This->size; This->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->buf_pos = 0; This->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; return S_OK; } static const struct strmbase_filter_ops filter_ops = { .filter_get_pin = avi_mux_get_pin, .filter_destroy = avi_mux_destroy, .filter_query_interface = avi_mux_query_interface, .filter_init_stream = avi_mux_init_stream, .filter_cleanup_stream = avi_mux_cleanup_stream, }; 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->source.pin.peer) { HRESULT hr = IFilterGraph_Reconnect(This->filter.filterInfo.pGraph, &This->source.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 source_query_accept(struct strmbase_pin *base, const AM_MEDIA_TYPE *amt) { FIXME("(%p) stub\n", base); return S_OK; } static HRESULT WINAPI AviMuxOut_AttemptConnection(struct strmbase_source *base, IPin *pReceivePin, const AM_MEDIA_TYPE *pmt) { PIN_DIRECTION dir; HRESULT hr; TRACE("(%p)->(%p AM_MEDIA_TYPE(%p))\n", base, pReceivePin, 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 source_get_media_type(struct strmbase_pin *base, unsigned int iPosition, AM_MEDIA_TYPE *amt) { TRACE("(%p)->(%d %p)\n", base, iPosition, amt); 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(struct strmbase_source *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 struct strmbase_source_ops source_ops = { .base.pin_query_accept = source_query_accept, .base.pin_get_media_type = source_get_media_type, .pfnAttemptConnection = AviMuxOut_AttemptConnection, .pfnDecideAllocator = AviMuxOut_DecideAllocator, }; static inline AviMux *impl_from_out_IPin(IPin *iface) { return CONTAINING_RECORD(iface, AviMux, source.pin.IPin_iface); } 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->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 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); hr = BaseOutputPinImpl_Connect(iface, pReceivePin, pmt); if(FAILED(hr)) return hr; for(i=0; iinput_pin_no; i++) { if(!This->in[i]->pin.pin.peer) 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_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 const IPinVtbl AviMuxOut_PinVtbl = { AviMuxOut_QueryInterface, BasePinImpl_AddRef, BasePinImpl_Release, AviMuxOut_Connect, BaseOutputPinImpl_ReceiveConnection, AviMuxOut_Disconnect, BasePinImpl_ConnectedTo, BasePinImpl_ConnectionMediaType, BasePinImpl_QueryPinInfo, BasePinImpl_QueryDirection, BasePinImpl_QueryId, BasePinImpl_QueryAccept, BasePinImpl_EnumMediaTypes, BasePinImpl_QueryInternalConnections, BaseOutputPinImpl_EndOfStream, BaseOutputPinImpl_BeginFlush, BaseOutputPinImpl_EndFlush, BasePinImpl_NewSegment }; static inline AviMux* impl_from_out_IQualityControl(IQualityControl *iface) { return CONTAINING_RECORD(iface, AviMux, IQualityControl_iface); } static HRESULT WINAPI AviMuxOut_QualityControl_QueryInterface( IQualityControl *iface, REFIID riid, void **ppv) { AviMux *This = impl_from_out_IQualityControl(iface); return IPin_QueryInterface(&This->source.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 sink_query_accept(struct strmbase_pin *base, const 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_Receive(struct strmbase_sink *base, IMediaSample *pSample) { AviMux *avimux = impl_from_strmbase_filter(base->pin.filter); AviMuxIn *avimuxin = CONTAINING_RECORD(base, AviMuxIn, pin); REFERENCE_TIME start, stop; IMediaSample *sample; int frames_no; IMediaSample2 *ms2; BYTE *frame, *buf; DWORD max_size, size; DWORD flags; HRESULT hr; TRACE("pin %p, pSample %p.\n", avimuxin, 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.mt.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 struct strmbase_sink_ops sink_ops = { .base.pin_query_accept = sink_query_accept, .base.pin_get_media_type = strmbase_pin_get_media_type, .pfnReceive = AviMuxIn_Receive, }; static inline AviMux* impl_from_in_IPin(IPin *iface) { struct strmbase_pin *pin = CONTAINING_RECORD(iface, struct strmbase_pin, IPin_iface); return impl_from_strmbase_filter(pin->filter); } static inline AviMuxIn* AviMuxIn_from_IPin(IPin *iface) { return CONTAINING_RECORD(iface, AviMuxIn, pin.pin.IPin_iface); } static HRESULT WINAPI AviMuxIn_QueryInterface(IPin *iface, REFIID riid, void **ppv) { AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); TRACE("pin %p, riid %s, ppv %p.\n", avimuxin, 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 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("pin %p, pConnector %p, pmt %p.\n", avimuxin, pConnector, 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) { AviMuxIn *avimuxin = AviMuxIn_from_IPin(iface); IMediaSample **prev, *cur; HRESULT hr; TRACE("pin %p.\n", avimuxin); 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 const IPinVtbl AviMuxIn_PinVtbl = { AviMuxIn_QueryInterface, BasePinImpl_AddRef, BasePinImpl_Release, BaseInputPinImpl_Connect, AviMuxIn_ReceiveConnection, AviMuxIn_Disconnect, BasePinImpl_ConnectedTo, BasePinImpl_ConnectionMediaType, BasePinImpl_QueryPinInfo, BasePinImpl_QueryDirection, BasePinImpl_QueryId, BasePinImpl_QueryAccept, BasePinImpl_EnumMediaTypes, BasePinImpl_QueryInternalConnections, BaseInputPinImpl_EndOfStream, BaseInputPinImpl_BeginFlush, BaseInputPinImpl_EndFlush, BasePinImpl_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 *start, DWORD cookie) { FIXME("iface %p, start %p, cookie %#x, stub!\n", iface, start, cookie); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_AMStreamControl_StopAt(IAMStreamControl *iface, const REFERENCE_TIME *stop, BOOL send_extra, DWORD cookie) { FIXME("iface %p, stop %p, send_extra %d, cookie %#x, stub!\n", iface, stop, send_extra, cookie); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_AMStreamControl_GetInfo(IAMStreamControl *iface, AM_STREAM_INFO *info) { FIXME("iface %p, info %p, stub!\n", iface, info); 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) { return CONTAINING_RECORD(iface, AviMuxIn, pin.IMemInputPin_iface); } 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); TRACE("pin %p, ppAllocator %p.\n", avimuxin, 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); ALLOCATOR_PROPERTIES props; HRESULT hr; TRACE("pin %p, pAllocator %p, bReadOnly %d.\n", avimuxin, 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); TRACE("pin %p, pProps %p.\n", avimuxin, 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); TRACE("pin %p, pSample %p.\n", avimuxin, 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); HRESULT hr = S_OK; TRACE("pin %p, pSamples %p, nSamples %d, nSamplesProcessed %p.\n", avimuxin, 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("avimuxin %p.\n", avimuxin); if(!This->source.pMemInputPin) return S_FALSE; hr = IMemInputPin_ReceiveCanBlock(This->source.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, const WCHAR *name, VARIANT *value, IErrorLog *error_log) { FIXME("iface %p, name %s, value %p, error_log %p, stub!\n", iface, debugstr_w(name), value, error_log); return E_NOTIMPL; } static HRESULT WINAPI AviMuxIn_PropertyBag_Write(IPropertyBag *iface, const WCHAR *name, VARIANT *value) { FIXME("iface %p, name %s, value %s, stub!\n", iface, debugstr_w(name), debugstr_variant(value)); 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 *filter, Quality q) { FIXME("iface %p, filter %p, type %u, proportion %d, late %s, timestamp %s, stub!\n", iface, filter, 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 *sink) { FIXME("iface %p, sink %p, stub!\n", iface, sink); 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) { WCHAR name[] = {'I','n','p','u','t',' ','0','0',0}; AviMuxIn *object; HRESULT hr; if(avimux->input_pin_no >= MAX_PIN_NO-1) return E_FAIL; name[7] = '0' + (avimux->input_pin_no+1) % 10; name[6] = '0' + (avimux->input_pin_no+1) / 10; if (!(object = heap_alloc_zero(sizeof(*object)))) return E_OUTOFMEMORY; strmbase_sink_init(&object->pin, &AviMuxIn_PinVtbl, &avimux->filter, name, &sink_ops, NULL); object->pin.IMemInputPin_iface.lpVtbl = &AviMuxIn_MemInputPinVtbl; object->IAMStreamControl_iface.lpVtbl = &AviMuxIn_AMStreamControlVtbl; object->IPropertyBag_iface.lpVtbl = &AviMuxIn_PropertyBagVtbl; object->IQualityControl_iface.lpVtbl = &AviMuxIn_QualityControlVtbl; hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (void **)&object->samples_allocator); if (FAILED(hr)) { strmbase_sink_cleanup(&object->pin); heap_free(object); return hr; } hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER, &IID_IMemAllocator, (void **)&object->pin.pAllocator); if (FAILED(hr)) { IMemAllocator_Release(object->samples_allocator); strmbase_sink_cleanup(&object->pin); heap_free(object); return hr; } object->indx = (AVISUPERINDEX *)&object->indx_data; object->ix = (AVISTDINDEX *)object->ix_data; avimux->in[avimux->input_pin_no++] = object; return S_OK; } IUnknown * WINAPI QCAP_createAVIMux(IUnknown *outer, HRESULT *phr) { static const WCHAR output_name[] = {'A','V','I',' ','O','u','t',0}; AviMux *avimux; PIN_INFO info; HRESULT hr; avimux = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(AviMux)); if(!avimux) { *phr = E_OUTOFMEMORY; return NULL; } strmbase_filter_init(&avimux->filter, outer, &CLSID_AviDest, &filter_ops); 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); strmbase_source_init(&avimux->source, &AviMuxOut_PinVtbl, &avimux->filter, output_name, &source_ops); avimux->IQualityControl_iface.lpVtbl = &AviMuxOut_QualityControlVtbl; avimux->cur_stream = 0; avimux->cur_time = 0; avimux->stream = NULL; hr = create_input_pin(avimux); if(FAILED(hr)) { strmbase_source_cleanup(&avimux->source); strmbase_filter_cleanup(&avimux->filter); HeapFree(GetProcessHeap(), 0, avimux); *phr = hr; return NULL; } avimux->interleave = 10000000; ObjectRefCount(TRUE); *phr = S_OK; return &avimux->filter.IUnknown_inner; }