Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1540 lines
37 KiB

/****************************************************************************
*
* EDITSTRM.C
*
* routines for reading Standard AVI files
*
* Copyright (c) 1992 Microsoft Corporation. All Rights Reserved.
*
* You have a royalty-free right to use, modify, reproduce and
* distribute the Sample Files (and/or any modified version) in
* any way you find useful, provided that you agree that
* Microsoft has no warranty obligations or liability for any
* Sample Application Files which are modified.
*
***************************************************************************/
#include <win32.h>
#include <compman.h>
#include "avifile.h"
#include <storage.h>
#include "editstrm.h"
#ifdef DEBUG
static void CDECL dprintf(LPSTR, ...);
#define DPF dprintf
#else
#define DPF ; / ## /
#endif
/*
* memcopy.asm
*/
#ifdef WIN32
#define MemCopy(dst, src, cnt) memmove(dst,src,cnt)
#else
EXTERN_C LONG FAR PASCAL MemCopy(HPSTR, HPSTR, DWORD);
#endif
STDAPI EditStreamCut(PAVISTREAM pavi, LONG FAR *plStart, LONG FAR *plLength, PAVISTREAM FAR * ppResult)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
hr = pedit->Cut(plStart, plLength, ppResult);
pedit->Release();
return hr;
}
STDAPI EditStreamCopy(PAVISTREAM pavi, LONG FAR *plStart, LONG FAR *plLength, PAVISTREAM FAR * ppResult)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
hr = pedit->Copy(plStart, plLength, ppResult);
pedit->Release();
return hr;
}
STDAPI EditStreamPaste(PAVISTREAM pavi, LONG FAR *plPos, LONG FAR *plLength, PAVISTREAM pstream, LONG lStart, LONG lLength)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
hr = pedit->Paste(plPos, plLength, pstream, lStart, lLength);
pedit->Release();
return hr;
}
STDAPI EditStreamClone(PAVISTREAM pavi, PAVISTREAM FAR *ppResult)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
hr = pedit->Clone(ppResult);
pedit->Release();
return hr;
}
STDAPI EditStreamSetInfo(PAVISTREAM pavi, AVISTREAMINFO FAR *lpInfo, LONG cbInfo)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
hr = pedit->SetInfo(lpInfo, cbInfo);
pedit->Release();
return hr;
}
STDAPI EditStreamSetName(PAVISTREAM pavi, LPCSTR lpszName)
{
PAVIEDITSTREAM pedit = NULL;
HRESULT hr;
AVISTREAMINFO info;
pavi->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &pedit);
if (!pedit)
return ResultFromScode(E_NOINTERFACE);
pavi->Info(&info, sizeof(info));
_fstrncpy(info.szName, lpszName, sizeof(info.szName));
hr = pedit->SetInfo(&info, sizeof(info));
pedit->Release();
return hr;
}
// #define EDITCHECK
///////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////
#define USUAL_ALLOC 16
#define EXTRA_ALLOC 8
HRESULT CEditStream::AllocEditSpace(LONG l, LONG cNew)
{
LPEDIT p;
LONG size;
if (cedits + cNew > maxedits) {
size = maxedits + max(cNew + EXTRA_ALLOC, USUAL_ALLOC);
p = (LPEDIT) GlobalReAllocPtr(edits, size * sizeof(EDIT), GHND | GMEM_SHARE);
if (!p)
return ResultFromScode(AVIERR_MEMORY);
edits = p;
maxedits = size;
}
if (l < cedits)
MemCopy((HPSTR) &edits[l + cNew],
(HPSTR) &edits[l],
(cedits - l) * sizeof(EDIT));
cedits += cNew;
return AVIERR_OK;
}
HRESULT CEditStream::PossiblyRemoveEdit(LONG l)
{
if (edits[l].lLength > 0)
return AVIERR_OK;
if (edits[l].pavi)
AVIStreamRelease(edits[l].pavi);
--cedits;
if (l < cedits)
MemCopy((HPSTR) &edits[l],
(HPSTR) &edits[l + 1],
(cedits - l) * sizeof(EDIT));
return AVIERR_OK;
}
CEditStream FAR * CEditStream::NewEditStream(PAVISTREAM psSource)
{
CEditStream FAR * pedit;
pedit = new CEditStream;
if (pedit)
(pedit->Create)((LONG) psSource, 0);
// !!! error check
return pedit;
}
STDAPI CreateEditableStream(
PAVISTREAM FAR * ppsEditable,
PAVISTREAM psSource)
{
// First, check if the stream is already editable....
if (psSource) {
PAVIEDITSTREAM paviedit = NULL;
psSource->QueryInterface(IID_IAVIEditStream, (LPVOID FAR *) &paviedit);
if (paviedit) {
paviedit->Clone(ppsEditable);
paviedit->Release();
return AVIERR_OK;
}
}
*ppsEditable = (PAVISTREAM) CEditStream::NewEditStream(psSource);
if (!*ppsEditable)
return ResultFromScode(AVIERR_MEMORY);
return AVIERR_OK;
}
///////////////////////////////////////////////////////////////////////////
//
// EditStreamOpen()
//
// open a single stream of a particular type from a AVI file.
//
// params:
// szFile - AVI file name
// fccType - stream type 0 for any type
// iStream - zero based stream number
//
// returns:
// a PAVISTREAM for the specifed stream or NULL.
//
// example:
//
// EditStreamOpen(pavi, "Foo.avi", 0, 0)
//
// will open stream 0 (the first stream)
//
// EditStreamOpen(pavi, "Foo.avi", 1)
//
// will open stream 1
//
// EditStreamOpenStream(pavi, "Foo.avi", 'vids', 0)
//
// will open the first video stream
//
// AVIOpenStream(pavi, "Foo.avi", 'auds', 0)
//
// will open the first audio stream
//
///////////////////////////////////////////////////////////////////////////
HRESULT STDMETHODCALLTYPE CEditStream::Create(LONG lParam1, LONG lParam2)
{
this->edits = (LPEDIT) GlobalAllocPtr(GHND | GMEM_SHARE, USUAL_ALLOC * sizeof(EDIT));
if (this->edits == NULL)
return ResultFromScode(AVIERR_MEMORY);
this->maxedits = USUAL_ALLOC;
this->ulRefCount = 1;
this->pgf = NULL;
this->psgf = NULL;
this->lpbiLast = NULL;
this->fFullFrames = FALSE;
this->edits[0].pavi = (PAVISTREAM)lParam1;
_fmemset(&this->sinfo, 0, sizeof(this->sinfo));
this->cedits = 1;
if (this->edits[0].pavi) {
AVIStreamAddRef(this->edits[0].pavi);
this->edits[0].lStart = AVIStreamStart(this->edits[0].pavi);
this->edits[0].lLength = AVIStreamLength(this->edits[0].pavi);
AVIStreamInfo(this->edits[0].pavi, &this->sinfo, sizeof(this->sinfo));
CheckEditList();
} else {
this->edits[0].lStart = 0;
this->edits[0].lLength = 0;
}
DPF("Edit %08lx: Usage++=%lx\n", (DWORD) (LPVOID) this, 1L);
//
// all done return success.
//
return 0; // success
}
///////////////////////////////////////////////////////////////////////////
//
// EditStreamAddRef()
//
// increase the reference count of the stream
//
///////////////////////////////////////////////////////////////////////////
ULONG STDMETHODCALLTYPE CEditStream::AddRef()
{
DPF("Edit %08lx: Usage++=%lx\n", (DWORD) (LPVOID) this, this->ulRefCount + 1);
return ++this->ulRefCount;
}
///////////////////////////////////////////////////////////////////////////
//
// EditStreamRelease()
//
// close a EditStream stream
//
///////////////////////////////////////////////////////////////////////////
ULONG STDMETHODCALLTYPE CEditStream::Release()
{
LONG l;
DPF("Edit %08lx: Usage--=%lx\n", (DWORD) (LPVOID) this, this->ulRefCount - 1);
if (--this->ulRefCount)
return this->ulRefCount;
// free edits....
for (l = 0; l < this->cedits; l++) {
if (this->edits[l].pavi)
AVIStreamRelease(this->edits[l].pavi);
}
GlobalFreePtr(this->edits);
this->edits = 0;
if (this->pgf)
AVIStreamGetFrameClose(this->pgf);
delete this;
return 0;
}
LPBITMAPINFOHEADER NEAR PASCAL CEditStream::CallGetFrame(
PAVISTREAM p,
LONG l)
{
if (psgf != p) {
PGETFRAME pgfNew;
pgfNew = AVIStreamGetFrameOpen(p, NULL);
if (!pgfNew)
return NULL;
if (pgf) {
#ifdef DEBUG
DPF("Trying to SetFormat %dx%dx%d '%4.4s'\n",
(int)lpbiLast->biWidth,
(int)lpbiLast->biHeight,
(int)lpbiLast->biBitCount,
(lpbiLast->biCompression == BI_RGB ? (LPSTR)"None" :
lpbiLast->biCompression == BI_RLE8 ? (LPSTR)"Rle8" :
lpbiLast->biCompression == BI_RLE4 ? (LPSTR)"Rle4" :
(LPSTR)&lpbiLast->biCompression));
#endif
if (pgfNew->SetFormat(lpbiLast, NULL, 0, 0, -1, -1) != AVIERR_OK) {
DPF("Couldn't AVIStreamGetFrameSetFormat!\n");
AVIStreamGetFrameClose(pgfNew);
return NULL;
}
AVIStreamGetFrameClose(pgf);
}
pgf = pgfNew;
psgf = p;
}
lpbiLast = (LPBITMAPINFOHEADER) AVIStreamGetFrame(pgf, l);
if (lpbiLast)
sinfo.dwSuggestedBufferSize = lpbiLast->biSizeImage;
return lpbiLast;
}
///////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////
HRESULT STDMETHODCALLTYPE CEditStream::ReadFormat(LONG lPos, LPVOID lpFormat, LONG FAR *lpcbFormat)
{
PAVISTREAM p;
LONG l;
HRESULT hr;
if ((lPos < (LONG) sinfo.dwStart) ||
(lPos >= (LONG) (sinfo.dwStart + sinfo.dwLength))) {
return ResultFromScode(AVIERR_BADPARAM);
}
hr = ResolveEdits(lPos, &p, &l, NULL, FALSE);
if (hr != 0) {
DPF("ReadFormat: ResolveEdits failed!\n");
return hr;
}
if (fFullFrames) {
LPBITMAPINFOHEADER lpbi;
LONG lSize;
// This isn't really right: we really need to make the formats
// agree. Should we just get the format from the first frame?
lpbi = CallGetFrame(p, l);
if (!lpbi) {
DPF("ReadFormat: GetFrame failed!\n");
return ResultFromScode(E_FAIL);
}
lSize = lpbi->biSize + lpbi->biClrUsed * sizeof(RGBQUAD);
if (lpFormat)
hmemcpy(lpFormat, lpbi, min(*lpcbFormat, lSize));
*lpcbFormat = lSize;
return 0;
} else {
return AVIStreamReadFormat(p, l, lpFormat, lpcbFormat);
}
}
///////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////
HRESULT STDMETHODCALLTYPE CEditStream::Info(AVISTREAMINFO FAR * psi, LONG lSize)
{
if (psi)
hmemcpy(psi, &sinfo, min(lSize, sizeof(sinfo)));
return 0; // !!! sizeof(pavi->sinfo);
}
///////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////
LONG STDMETHODCALLTYPE CEditStream::FindSample(LONG lPos, LONG lFlags)
{
PAVISTREAM p;
LONG l;
LONG edit;
LONG lRet;
HRESULT hr;
if ((lPos < (LONG) sinfo.dwStart) ||
(lPos >= (LONG) (sinfo.dwStart + sinfo.dwLength))) {
return -1;
}
hr = ResolveEdits(lPos, &p, &l, &edit, TRUE);
if (hr != 0) {
DPF("FindSample: error from ResolveEdits!\n");
return -1;
}
if (lFlags & FIND_FORMAT) {
// !!!! This isn't right, obviously.
if (lFlags & FIND_PREV)
return 0;
else
return -1;
}
if (this->fFullFrames) {
return lPos;
}
// !!! This won't really work, especially for searching forward.
lRet = AVIStreamFindSample(p, l, lFlags);
#ifdef DEBUG
if (lRet < edits[edit].lStart) {
DPF("We were about to return a key frame before a segment: returning %ld instead of %ld.\n", edits[edit].lStart, lRet);
}
#endif
// DPF("FindSample: lPos = %ld, Key = %ld\n", lPos, lPos - (l - lRet));
return lPos - (l - lRet);
}
///////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////
HRESULT STDMETHODCALLTYPE CEditStream::Read(
LONG lStart,
LONG lSamples,
LPVOID lpBuffer,
LONG cbBuffer,
LONG FAR * plBytes,
LONG FAR * plSamples)
{
PAVISTREAM p;
LONG l;
LONG edit;
LONG lSamplesRead;
LONG lBytesRead;
LONG lSamplesThisTime;
HRESULT hr;
if (plBytes)
*plBytes = 0;
if (plSamples)
*plSamples = 0;
#ifdef TOOMUCHDEBUG
if (lpBuffer) {
DPF("Read %08lx: Start = %ld Length = %ld\n", (DWORD) (LPVOID) this, lStart, lSamples);
}
#endif
if ((lStart < (LONG) sinfo.dwStart) ||
(lStart >= (LONG) (sinfo.dwStart + sinfo.dwLength))) {
DPF("Read at position %ld, start = %lu, len = %lu\n", lStart, sinfo.dwStart, sinfo.dwStart + sinfo.dwLength);
return ResultFromScode(AVIERR_BADPARAM);
}
while (lSamples) {
hr = ResolveEdits(lStart, &p, &l, &edit, FALSE);
if (hr != 0) {
DPF("Read: ResolveEdits failed!\n");
return ResultFromScode(E_FAIL);
}
// Don't read past the end of this edit.
if ((l - this->edits[edit].lStart) + lSamples > this->edits[edit].lLength)
lSamplesThisTime = this->edits[edit].lLength - (l - this->edits[edit].lStart);
else
lSamplesThisTime = lSamples;
if (this->fFullFrames) {
LPBITMAPINFOHEADER lpbi;
LPVOID lp;
lpbi = CallGetFrame(p, l);
if (!lpbi) {
DPF("Read: GetFrame failed!\n");
return ResultFromScode(E_FAIL);
}
//
// a NULL buffer means return the size buffer needed to read
// the given sample.
//
if (lpBuffer == NULL)
goto exit;
lp = (LPBYTE) lpbi + lpbi->biSize + lpbi->biClrUsed * sizeof(RGBQUAD);
if (cbBuffer >= (LONG) lpbi->biSizeImage) {
hmemcpy(lpBuffer, lp, lpbi->biSizeImage);
exit:
if (plBytes)
*plBytes = lpbi->biSizeImage;
if (plSamples)
*plSamples = 1;
return 0;
}
if (plBytes)
*plBytes = 0;
if (plSamples)
*plSamples = 0;
return ResultFromScode(AVIERR_BUFFERTOOSMALL);
} else {
hr = AVIStreamRead(p, l, lSamplesThisTime, lpBuffer, cbBuffer,
&lBytesRead, &lSamplesRead);
if (hr != NOERROR)
return hr;
if (plBytes)
*plBytes += lBytesRead;
if (plSamples)
*plSamples += lSamplesRead;
lpBuffer = (BYTE _huge *) lpBuffer + lBytesRead;
lStart += lSamplesThisTime;
lSamples -= lSamplesThisTime;
cbBuffer -= lBytesRead;
// If we've read up to the end of the file,
// stop now, rather than return an error....
if (lStart >= (LONG) (this->sinfo.dwLength + this->sinfo.dwStart))
break;
}
}
#ifdef TOOMUCHDEBUG
if (lpBuffer && plBytes) {
DPF("Read %08lx: Bytes Read = %ld\n", (DWORD) (LPVOID) this, *plBytes);
}
#endif
return 0;
}
void CEditStream::CheckEditList()
{
#ifdef EDITCHECK
LONG lTotal = 0;
LONG l;
DPF("Edit list %08lx: %s\n", (DWORD) this, fFullFrames ? (LPSTR) " (Using full frames)" : (LPSTR) "");
for (l = 0; l < cedits; l++) {
DPF("\t\t%ld:\t%08lx\t%ld\t%ld\n", l, (DWORD) (LPVOID) edits[l].pavi, edits[l].lStart, edits[l].lLength);
lTotal += edits[l].lLength;
}
if (lTotal != (LONG) sinfo.dwLength) {
* (LPSTR) 0 = 0;
}
#endif
}
HRESULT CEditStream::ResolveEdits(LONG lPos, PAVISTREAM FAR *ppavi,
LONG FAR *plPos, LONG FAR *pl, BOOL fAllowEnd)
{
LONG edit;
//
// Search edit list, get position...
//
if (lPos < (LONG) this->sinfo.dwStart) {
DPF("ResolveEdits: Read at %ld, before start at %ld\n", lPos, this->sinfo.dwStart);
return ResultFromScode(AVIERR_BADPARAM);
}
lPos -= (LONG) this->sinfo.dwStart;
for (edit = 0; edit < this->cedits; edit++) {
if (lPos < this->edits[edit].lLength) {
*ppavi = this->edits[edit].pavi;
*plPos = lPos + this->edits[edit].lStart;
if (pl)
*pl = edit;
return 0;
}
lPos -= this->edits[edit].lLength;
}
// Normally, we don't return a position at the end of an edit--we instead
// go to the next thing.
if (lPos == 0 && fAllowEnd) {
edit--;
*ppavi = this->edits[edit].pavi;
*plPos = this->edits[edit].lStart + this->edits[edit].lLength;
if (pl)
*pl = edit;
return 0;
}
*ppavi = 0;
*plPos = 0;
if (pl)
*pl = 0;
return ResultFromScode(AVIERR_BADPARAM);
}
//
// Cut:
//
// Takes start, length to cut out
//
// returns actual start, length cut, along with new stream
//
STDMETHODIMP CEditStream::Cut(LONG FAR *plStart, LONG FAR *plLength, PAVISTREAM FAR * ppResult)
{
HRESULT hr = AVIERR_OK;
PAVISTREAM p;
LONG l;
LONG edit;
LONG lStart, lLength;
l = AVIStreamLength(this);
if (ppResult)
*ppResult = 0;
if (!plStart || !plLength) {
return ResultFromScode(AVIERR_BADPARAM);
}
if (*plStart < 0) {
return ResultFromScode(AVIERR_BADPARAM);
}
if (*plLength < 0 || *plStart + *plLength > l) {
if (*plStart >= l)
return ResultFromScode(AVIERR_BADPARAM);
*plLength = l - *plStart;
}
#ifdef KEYALWAYS
// Make cut end at key frame
for (l = *plStart + *plLength; l < AVIStreamLength(this); l++) {
if (AVIStreamFindSample(this, l, 0) == l)
break;
}
*plLength = l - *plStart;
#else
// we cut whatever they ask us to....
#endif
// Make a copy of the section being cut out
if (ppResult) {
// This will make cut start at key frame if it needs to
hr = this->Copy(plStart, plLength, ppResult);
if (hr != AVIERR_OK)
return hr;
}
lLength = *plLength;
lStart = *plStart;
#ifndef KEYALWAYS
if (!this->fFullFrames &&
lStart + lLength < AVIStreamLength(this) &&
AVIStreamFindSample(this, lStart + lLength, 0) != lStart + lLength) {
DPF("Cut: Converting stream to full frames\n");
this->fFullFrames = TRUE;
this->sinfo.dwFormatChangeCount++;
this->sinfo.fccHandler = 0;
}
#endif
// Now do the actual cut
hr = ResolveEdits(lStart, &p, &l, &edit, FALSE);
if (hr != NOERROR)
return hr;
if (this->edits[edit].lStart + this->edits[edit].lLength > l + lLength) {
// The part cut out is entirely within this edit.
if (this->edits[edit].lStart == l) {
// The part cut out is the start of this edit
this->edits[edit].lStart = l + lLength;
this->edits[edit].lLength -= lLength;
} else {
hr = AllocEditSpace(edit, 1);
if (hr == AVIERR_OK) {
this->edits[edit] = this->edits[edit+1];
if (this->edits[edit].pavi)
AVIStreamAddRef(this->edits[edit].pavi);
this->edits[edit].lStart = this->edits[edit+1].lStart;
this->edits[edit].lLength = l - this->edits[edit].lStart;
this->edits[edit+1].lStart = l + lLength;
this->edits[edit+1].lLength -= lLength +
this->edits[edit].lLength;
}
}
} else if (this->edits[edit].lStart + this->edits[edit].lLength == l + lLength) {
// The part cut out is the end of this edit
this->edits[edit].lLength = l - this->edits[edit].lStart;
} else {
LONG lTemp = lLength;
// We're cutting out more than this one edit.
// First, cut out the rest of this edit.
lTemp -= this->edits[edit].lStart + this->edits[edit].lLength - l;
this->edits[edit].lLength = l - this->edits[edit].lStart;
edit++;
// As long as subsequent edits are still shorter than the cut,
// kill them..
while (edit < this->cedits &&
this->edits[edit].lLength < lTemp) {
lTemp -= this->edits[edit].lLength;
this->edits[edit].lLength = 0;
edit++;
}
if (edit < this->cedits) {
this->edits[edit].lStart += lTemp;
this->edits[edit].lLength -= lTemp;
}
}
if (hr == AVIERR_OK) {
this->sinfo.dwLength -= lLength;
this->sinfo.dwEditCount++;
CheckEditList();
} else {
if (ppResult)
AVIStreamRelease(*ppResult);
}
return hr;
}
//
// Copy:
//
// Takes start, length to cut out
//
// returns actual start, length cut, along with new stream
//
//
STDMETHODIMP CEditStream::Copy(LONG FAR *plStart, LONG FAR *plLength, PAVISTREAM FAR * ppResult)
{
PAVISTREAM p1;
LONG l1;
LONG edit1;
PAVISTREAM p2;
LONG l2;
LONG edit2;
LONG l;
CEditStream FAR *p;
HRESULT hr;
LONG lStart, lLength;
l1 = AVIStreamLength(this);
// If start, length < 0, pick some defaults
if (*plStart < 0)
*plStart = 0;
if (*plLength < 0)
*plLength = l1 - *plStart;
// Make sure the start position is within range
if (*plStart > l1) {
if (ppResult)
*ppResult = 0;
return ResultFromScode(AVIERR_BADPARAM);
}
// Make sure the length is within range
if (*plStart + *plLength > l1)
*plLength = l1 - *plStart;
#ifdef KEYALWAYS
// Make copy start at key frame
lStart = AVIStreamFindSample(this, *plStart, 0);
*plLength += *plStart - lStart;
*plStart = lStart;
#endif
lLength = *plLength;
lStart = *plStart;
p = NewEditStream(NULL);
*ppResult = (PAVISTREAM) p;
if (!p)
return ResultFromScode(AVIERR_MEMORY);
hmemcpy(&p->sinfo, &this->sinfo, sizeof(p->sinfo));
if (lLength <= 0)
lLength = (LONG) (p->sinfo.dwLength + p->sinfo.dwStart) - lStart;
hr = ResolveEdits(lStart, &p1, &l1, &edit1, FALSE);
hr = ResolveEdits(lStart + lLength, &p2, &l2, &edit2, TRUE);
if (edit1 == edit2) {
p->edits[0].pavi = p1;
if (p1)
AVIStreamAddRef(p1);
p->edits[0].lStart = l1;
p->edits[0].lLength = lLength;
} else {
hr = p->AllocEditSpace(1, edit2 - edit1);
for (l = 0; l <= edit2 - edit1; l++) {
if (l == 0) {
p->edits[l].pavi = p1;
if (p1)
AVIStreamAddRef(p1);
p->edits[l].lStart = l1;
p->edits[l].lLength = this->edits[edit1].lStart +
this->edits[edit1].lLength - l1;
} else if (l < edit2 - edit1) {
p->edits[l] = this->edits[l+edit1];
if (p->edits[l].pavi)
AVIStreamAddRef(p->edits[l].pavi);
} else {
p->edits[l] = this->edits[edit2];
if (p->edits[l].pavi)
AVIStreamAddRef(p->edits[l].pavi);
p->edits[l].lLength = l2 - p->edits[l].lStart;
}
}
p->PossiblyRemoveEdit(edit2 - edit1);
p->PossiblyRemoveEdit(0);
}
#ifndef KEYALWAYS
l1 = AVIStreamFindSample(p->edits[0].pavi, p->edits[0].lStart, 0);
DPF("edit starts at %ld, key at %ld\n", p->edits[0].lStart, l1);
if (l1 != p->edits[0].lStart) {
p->fFullFrames = TRUE;
DPF("Copy: Converting new stream to full frames\n");
}
#endif
AVIStreamInfo(this->edits[0].pavi, &p->sinfo, sizeof(p->sinfo));
p->sinfo.dwStart = 0;
p->sinfo.dwLength = (DWORD) lLength;
p->sinfo.dwEditCount = 0;
p->sinfo.dwFormatChangeCount = 0;
if (p->fFullFrames)
p->sinfo.fccHandler = 0;
p->CheckEditList();
CheckEditList();
return AVIERR_OK;
}
/**************************************************************************
* @doc INTERNAL DRAWDIB
*
* @api BOOL | DibEq | This function compares two dibs.
*
* @parm LPBITMAPINFOHEADER lpbi1 | Pointer to one bitmap.
* this DIB is assumed to have the colors after the BITMAPINFOHEADER
*
* @parm LPBITMAPINFOHEADER | lpbi2 | Pointer to second bitmap.
* this DIB is assumed to have the colors after biSize bytes.
*
* @rdesc Returns TRUE if bitmaps are identical, FALSE otherwise.
*
**************************************************************************/
inline BOOL DibEq(LPBITMAPINFOHEADER lpbi1, LPBITMAPINFOHEADER lpbi2)
{
return
lpbi1->biCompression == lpbi2->biCompression &&
lpbi1->biSize == lpbi2->biSize &&
lpbi1->biWidth == lpbi2->biWidth &&
lpbi1->biHeight == lpbi2->biHeight &&
lpbi1->biBitCount == lpbi2->biBitCount;
}
BOOL AreVideoStreamsCompatible(PAVISTREAM ps1, PAVISTREAM ps2)
{
LONG cb1, cb2;
BITMAPINFOHEADER bih1, bih2;
AVIStreamReadFormat(ps1, AVIStreamStart(ps1), NULL, &cb1);
AVIStreamReadFormat(ps2, AVIStreamStart(ps2), NULL, &cb2);
if (cb1 != cb2)
return FALSE;
cb1 = sizeof(bih1);
cb2 = sizeof(bih2);
AVIStreamReadFormat(ps1, AVIStreamStart(ps1), &bih1, &cb1);
AVIStreamReadFormat(ps2, AVIStreamStart(ps2), &bih2, &cb2);
if (DibEq(&bih1, &bih2))
return TRUE;
return FALSE;
}
BOOL AreAudioStreamsCompatible(PAVISTREAM ps1, PAVISTREAM ps2)
{
LONG cb1, cb2;
LPVOID lpf;
BOOL f;
AVIStreamReadFormat(ps1, AVIStreamStart(ps1), NULL, &cb1);
AVIStreamReadFormat(ps2, AVIStreamStart(ps2), NULL, &cb2);
if (cb1 != cb2)
return FALSE;
lpf = GlobalAllocPtr(GHND, cb1 + cb2);
if (!lpf)
return FALSE; // !!!
AVIStreamReadFormat(ps1, AVIStreamStart(ps1), lpf, &cb1);
AVIStreamReadFormat(ps2, AVIStreamStart(ps2), (BYTE FAR *)lpf + cb1, &cb2);
f = !_fmemcmp(lpf, (BYTE FAR *)lpf + cb1, (UINT) cb1);
GlobalFreePtr(lpf);
return f;
}
//
// Paste:
//
// Takes stream to paste, along with start and length within that stream,
// and also target stream and position within the stream to do the paste.
//
// Returns position and length pasted.
//
STDMETHODIMP CEditStream::Paste(LONG FAR *plPos, LONG FAR *plLength, PAVISTREAM pstream, LONG lStart, LONG lLength)
{
PAVISTREAM p;
LONG l;
LONG edit;
HRESULT hr;
LONG lPos;
CEditStream FAR *pnew;
AVISTREAMINFO strinfo;
AVIStreamInfo(pstream, &strinfo, sizeof(strinfo));
if (this->sinfo.fccType == 0) {
AVIStreamInfo(pstream, &this->sinfo, sizeof(this->sinfo));
this->sinfo.dwLength = 0;
this->sinfo.dwStart = *plPos;
}
if (*plPos > (LONG) (sinfo.dwLength + sinfo.dwStart)) {
// !!! We should handle this case....
return ResultFromScode(AVIERR_BADPARAM);
}
#ifdef KEYALWAYS
// Make paste go before a key frame...
*plPos = AVIStreamFindSample(this, *plPos, 0);
#endif
lPos = *plPos;
if (strinfo.fccType != this->sinfo.fccType) {
DPF("Paste: Incompatible stream types!\n");
return ResultFromScode(AVIERR_UNSUPPORTED);
}
if (lLength <= 0 || ((lStart + lLength) >
(LONG) (strinfo.dwStart + strinfo.dwLength))) {
if (lStart >= (LONG) (strinfo.dwLength + strinfo.dwStart))
return ResultFromScode(AVIERR_BADPARAM);
lLength = (LONG) (strinfo.dwLength + strinfo.dwStart) - lStart;
}
if ((DWORD) lPos + (DWORD) lLength > 0x80000000) {
DPF("Paste result would be more than 2 billion frames!\n");
return ResultFromScode(AVIERR_MEMORY);
}
// !!! What if the frame rates don't match?
#define SIZEMISMATCH(rc1, rc2) \
(((rc1.right - rc1.left) != (rc2.right - rc2.left)) || \
((rc1.bottom - rc1.top) != (rc2.bottom - rc2.top)))
if (strinfo.fccType == streamtypeVIDEO &&
SIZEMISMATCH(strinfo.rcFrame, this->sinfo.rcFrame)) {
// !!! It would be nice if this worked.
DPF("Paste: Video streams are different sizes!\n");
return ResultFromScode(AVIERR_UNSUPPORTED);
}
if (this->sinfo.fccType == streamtypeAUDIO) {
if (!AreAudioStreamsCompatible((PAVISTREAM) this, pstream)) {
DPF("Paste: Audio streams are different formats!\n");
return ResultFromScode(AVIERR_UNSUPPORTED);
}
}
pnew = NULL;
pstream->QueryInterface(CLSID_EditStream, (LPVOID FAR *) &pnew);
#ifndef KEYALWAYS
if (this->sinfo.fccType == streamtypeVIDEO) {
if (!this->fFullFrames) {
if ((!AVIStreamIsKeyFrame(pstream, lStart) ||
(pnew && pnew->fFullFrames)) ||
((lPos < (LONG) (sinfo.dwLength + sinfo.dwStart)) &&
!AVIStreamIsKeyFrame((PAVISTREAM) this, lPos)) ||
!AreVideoStreamsCompatible((PAVISTREAM) this, pstream)) {
// !!! What if we're pasting, say, an 8-bit and a 32-bit
// movie together? Do we have to pick a common format
// to convert to?
CallGetFrame(this->edits[0].pavi, this->edits[0].lStart);
if (CallGetFrame(pstream, lStart) == NULL) {
DPF("Paste: Can't make a common format!\n");
if (pnew)
pnew->Release();
return ResultFromScode(AVIERR_BADFORMAT);
}
this->fFullFrames = TRUE;
DPF("Paste: Converting stream to full frames\n");
this->sinfo.dwFormatChangeCount++;
// ??? !!! Call get frame once, just so it's been done....
}
} else {
if (CallGetFrame(pstream, lStart) == NULL) {
DPF("Paste: Can't make a common format!\n");
if (pnew)
pnew->Release();
return ResultFromScode(AVIERR_BADFORMAT);
}
}
}
#endif
// Find where to do the paste...
hr = ResolveEdits(lPos, &p, &l, &edit, TRUE);
// Report back the size of what we pasted...
if (plLength)
*plLength = lLength;
if (pnew) {
LONG lNew;
// The inserted stream is itself an edit list; take advantage
// of this fact.
hr = AllocEditSpace(edit, 1 + pnew->cedits);
this->edits[edit].pavi = this->edits[edit + 1 + pnew->cedits].pavi;
if (this->edits[edit].pavi)
AVIStreamAddRef(this->edits[edit].pavi);
this->edits[edit].lStart = this->edits[edit + 1 + pnew->cedits].lStart;
this->edits[edit].lLength = l - this->edits[edit].lStart;
// !!! We're ignoring lStart and lLength!
for (lNew = 0; lNew < pnew->cedits; lNew++) {
this->edits[edit + 1 + lNew] = pnew->edits[lNew];
AVIStreamAddRef(pnew->edits[lNew].pavi);
}
this->edits[edit + pnew->cedits + 1].lStart = l;
this->edits[edit + pnew->cedits + 1].lLength -= this->edits[edit].lLength;
// Get rid of zero-length edits....
PossiblyRemoveEdit(edit + pnew->cedits + 1);
PossiblyRemoveEdit(edit);
this->sinfo.dwLength += lLength;
pnew->CheckEditList();
pnew->Release();
} else {
// Just insert the stream as a blob.
hr = AllocEditSpace(edit, 2);
this->edits[edit].pavi = this->edits[edit+2].pavi;
if (this->edits[edit].pavi)
AVIStreamAddRef(this->edits[edit].pavi);
this->edits[edit].lStart = this->edits[edit+2].lStart;
this->edits[edit].lLength = l - this->edits[edit+2].lStart;
this->edits[edit+ 1].pavi = pstream;
if (pstream)
AVIStreamAddRef(pstream);
this->edits[edit + 1].lStart = lStart;
this->edits[edit + 1].lLength = lLength;
this->edits[edit + 2].lStart = l;
this->edits[edit + 2].lLength -= this->edits[edit].lLength;
// No addref here, since the edit we're splitting had a ref already
this->sinfo.dwLength += lLength;
// Get rid of zero-length edits....
PossiblyRemoveEdit(edit + 2);
PossiblyRemoveEdit(edit);
}
CheckEditList();
this->sinfo.dwEditCount++;
return AVIERR_OK;
}
STDMETHODIMP CEditStream::Clone(PAVISTREAM FAR *ppResult)
{
CEditStream FAR * pnew;
HRESULT hr;
LONG l;
pnew = NewEditStream(NULL);
*ppResult = (PAVISTREAM) pnew;
if (!pnew)
return ResultFromScode(AVIERR_MEMORY);
if (this->cedits > 1) {
hr = pnew->AllocEditSpace(1, this->cedits - 1);
if (hr != NOERROR) {
// !!! Clean things up
return hr;
}
}
for (l = 0; l < this->cedits; l++) {
pnew->edits[l] = this->edits[l];
if (pnew->edits[l].pavi)
AVIStreamAddRef(pnew->edits[l].pavi);
}
pnew->sinfo = this->sinfo;
pnew->fFullFrames = this->fFullFrames;
pnew->CheckEditList();
return AVIERR_OK;
}
STDMETHODIMP CEditStream::SetInfo(AVISTREAMINFO FAR * lpInfo, LONG cbInfo)
{
if ((cbInfo < sizeof(AVISTREAMINFO)) ||
(IsBadReadPtr(lpInfo, sizeof(AVISTREAMINFO))))
return ResultFromScode(AVIERR_BADPARAM);
// Things we don't copy:
// fccType
// fccHandler
// dwFlags
// dwCaps
// dwLength
// dwInitialFrames
// dwSuggestedBufferSize
// dwSampleSize
// dwEditCount
// dwFormatChangeCount
this->sinfo.wPriority = lpInfo->wPriority;
this->sinfo.wLanguage = lpInfo->wLanguage;
this->sinfo.dwScale = lpInfo->dwScale;
this->sinfo.dwRate = lpInfo->dwRate;
this->sinfo.dwStart = lpInfo->dwStart; // !!! ???
this->sinfo.dwQuality = lpInfo->dwQuality;
this->sinfo.rcFrame = lpInfo->rcFrame;
if (lpInfo->szName[0])
_fmemcpy(this->sinfo.szName, lpInfo->szName, sizeof(this->sinfo.szName));
// The stream has been changed....
++this->sinfo.dwEditCount;
return NOERROR;
}
//
//
// Extra unimplemented functions.....
//
//
//
HRESULT STDMETHODCALLTYPE CEditStream::QueryInterface(REFIID riid, LPVOID FAR* ppvObj)
{
if (riid == IID_IUnknown)
*ppvObj = ((IUnknown FAR *) (IAVIStream FAR *) this);
else if (riid == IID_IAVIStream)
*ppvObj = ((IAVIStream FAR *) this);
else if (riid == IID_IAVIEditStream)
*ppvObj = ((IAVIEditStream FAR *) this);
#if 0
else if ((riid == IID_IMarshal) && CanMarshalSimply()) // !!!! Remove once fixed!
*ppvObj = ((IMarshal FAR *) this);
#endif
else if (riid == CLSID_EditStream)
*ppvObj = this;
else { // unsupported interface
*ppvObj = NULL;
return ResultFromScode(E_NOINTERFACE);
}
AddRef();
return NOERROR;
}
HRESULT STDMETHODCALLTYPE CEditStream::ReadData (DWORD fcc, LPVOID lp, LONG FAR *lpcb)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::SetFormat (LONG lPos, LPVOID lpFormat, LONG cbFormat)
{
// !!! We could set the whole format of the stream here, and do mapping....
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::WriteData (DWORD fcc, LPVOID lp, LONG cb)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Write (LONG lStart, LONG lSamples, LPVOID lpBuffer, LONG cbBuffer, DWORD dwFlags, LONG FAR *plSampWritten, LONG FAR *plBytesWritten)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Delete (LONG lStart, LONG lSamples)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT CEditStream::NewInstance(IUnknown FAR* pUnknownOuter,
REFIID riid,
LPVOID FAR* ppv)
{
CEditStream FAR * pedit;
HRESULT hr;
pedit = new CEditStream;
if (pedit)
(pedit->Create)(NULL, 0);
// !!! error check
hr = pedit->QueryInterface(riid, ppv);
if (FAILED(GetScode(hr)))
delete pedit;
return hr;
}
BOOL CEditStream::CanMarshalSimply()
{
LONG l;
LPUNKNOWN punk;
for (l = 0; l < this->cedits; l++) {
punk = NULL;
this->edits[l].pavi->QueryInterface(CLSID_AVISimpleUnMarshal,
(LPVOID FAR *) &punk);
if (!punk)
return FALSE;
punk->Release();
}
return TRUE;
}
STDMETHODIMP CEditStream::GetUnmarshalClass (REFIID riid, LPVOID pv,
DWORD dwDestContext, LPVOID pvDestContext,
DWORD mshlflags, LPCLSID pCid)
{
if (CanMarshalSimply()) {
DPF("UnMarshalClass called (simple)\n");
*pCid = CLSID_AVISimpleUnMarshal;
} else {
DPF("UnMarshalClass called (not simple)\n");
*pCid = CLSID_EditStream;
}
return NOERROR;
}
STDMETHODIMP CEditStream::GetMarshalSizeMax (REFIID riid, LPVOID pv,
DWORD dwDestContext, LPVOID pvDestContext,
DWORD mshlflags, LPDWORD pSize)
{
if (CanMarshalSimply())
*pSize = 4;
else
*pSize = 0;
return NOERROR;
}
STDMETHODIMP CEditStream::MarshalInterface (LPSTREAM pStm, REFIID riid,
LPVOID pv, DWORD dwDestContext, LPVOID pvDestContext,
DWORD mshlflags)
{
HRESULT hr;
if ((riid != IID_IAVIStream && riid != IID_IAVIFile && riid != IID_IUnknown))
return ResultFromScode(E_INVALIDARG);
if (CanMarshalSimply()) {
LPUNKNOWN pUnk = (LPUNKNOWN) (PAVISTREAM) this;
DPF("MarshalInterface called (simple): Marshalling %lx\n", (DWORD) pUnk);
if ((hr = pStm->Write(&pUnk, sizeof(pUnk), NULL)) == NOERROR)
AddRef();
} else {
DPF("MarshalInterface called (not simple)\n");
hr = ResultFromScode(AVIERR_UNSUPPORTED);
}
DPF("Returns %lx\n", (DWORD) (LPVOID) hr);
return hr;
}
STDMETHODIMP CEditStream::UnmarshalInterface (LPSTREAM pStm, REFIID riid,
LPVOID FAR* ppv)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
STDMETHODIMP CEditStream::ReleaseMarshalData (LPSTREAM pStm)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
STDMETHODIMP CEditStream::DisconnectObject (DWORD dwReserved)
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Reserved1()
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Reserved2()
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Reserved3()
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Reserved4()
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
HRESULT STDMETHODCALLTYPE CEditStream::Reserved5()
{
return ResultFromScode(AVIERR_UNSUPPORTED);
}
/*****************************************************************************
*
* dprintf() is called by the DPF macro if DEBUG is defined at compile time.
*
* The messages will be send to COM1: like any debug message. To
* enable debug output, add the following to WIN.INI :
*
* [debug]
* ICSAMPLE=1
*
****************************************************************************/
#ifdef DEBUG
#define MODNAME "EditStrm"
static BOOL fDebug = -1;
static void cdecl dprintf(LPSTR szFormat, ...)
{
char ach[128];
#ifdef WIN32
va_list va;
if (fDebug == -1)
fDebug = GetProfileIntA("Debug",MODNAME, FALSE);
if (!fDebug)
return;
va_start(va, szFormat);
if (szFormat[0] == '!')
ach[0]=0, szFormat++;
else
lstrcpyA(ach, MODNAME ": ");
wvsprintfA(ach+lstrlenA(ach),szFormat,(LPSTR)va);
va_end(va);
// lstrcatA(ach, "\r\r\n");
OutputDebugStringA(ach);
#else
if (fDebug == -1)
fDebug = GetProfileInt("Debug",MODNAME, FALSE);
if (!fDebug)
return;
if (szFormat[0] == '!')
ach[0]=0, szFormat++;
else
lstrcpy(ach, MODNAME ": ");
wvsprintf(ach+lstrlen(ach),szFormat,(LPSTR)(&szFormat+1));
// lstrcat(ach, "\r\r\n");
OutputDebugString(ach);
#endif
}
#endif