getframe.c File Reference

#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "wingdi.h"
#include "winuser.h"
#include "vfw.h"
#include "avifile_private.h"
#include "wine/debug.h"

Include dependency graph for getframe.c:

Go to the source code of this file.

Classes
struct	_IGetFrameImpl

Typedefs
typedef struct _IGetFrameImpl	IGetFrameImpl

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (avifile)
static IGetFrameImpl *	impl_from_IGetFrame (IGetFrame *iface)
static void	AVIFILE_CloseCompressor (IGetFrameImpl *This)
static HRESULT WINAPI	IGetFrame_fnQueryInterface (IGetFrame *iface, REFIID refiid, LPVOID *obj)
static ULONG WINAPI	IGetFrame_fnAddRef (IGetFrame *iface)
static ULONG WINAPI	IGetFrame_fnRelease (IGetFrame *iface)
static LPVOID WINAPI	IGetFrame_fnGetFrame (IGetFrame *iface, LONG lPos)
static HRESULT WINAPI	IGetFrame_fnBegin (IGetFrame *iface, LONG lStart, LONG lEnd, LONG lRate)
static HRESULT WINAPI	IGetFrame_fnEnd (IGetFrame *iface)
static HRESULT WINAPI	IGetFrame_fnSetFormat (IGetFrame *iface, LPBITMAPINFOHEADER lpbiWanted, LPVOID lpBits, INT x, INT y, INT dx, INT dy)
PGETFRAME	AVIFILE_CreateGetFrame (PAVISTREAM pStream)

Variables
static const struct IGetFrameVtbl	igetframeVtbl

Typedef Documentation

IGetFrameImpl

typedef struct _IGetFrameImpl IGetFrameImpl

Function Documentation

AVIFILE_CloseCompressor()

static void AVIFILE_CloseCompressor ( IGetFrameImpl *  This )

static

Definition at line 77 of file getframe.c.

{
  if (This->lpInFormat != This->lpOutFormat) {
    HeapFree(GetProcessHeap(), 0, This->lpOutFormat);
    This->lpOutFormat = NULL;
  }
  HeapFree(GetProcessHeap(), 0, This->lpInFormat);
  This->lpInFormat = NULL;
  if (This->hic != NULL) {
    if (This->bResize)
      ICDecompressExEnd(This->hic);
    else
      ICDecompressEnd(This->hic);
    ICClose(This->hic);
    This->hic = NULL;
  }
}

Referenced by IGetFrame_fnGetFrame(), IGetFrame_fnRelease(), and IGetFrame_fnSetFormat().

AVIFILE_CreateGetFrame()

PGETFRAME AVIFILE_CreateGetFrame

(

PAVISTREAM

pStream

)

Definition at line 487 of file getframe.c.

{
  IGetFrameImpl *pg;
 
  /* check parameter */
  if (pStream == NULL)
    return NULL;
 
  pg = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IGetFrameImpl));
  if (pg != NULL) {
    pg->IGetFrame_iface.lpVtbl = &igetframeVtbl;
    pg->ref           = 1;
    pg->lCurrentFrame = -1;
    pg->pStream       = pStream;
    IAVIStream_AddRef(pStream);
  }
 
  return &pg->IGetFrame_iface;
}

Referenced by AVIStreamGetFrameOpen().

IGetFrame_fnAddRef()

static ULONG WINAPI IGetFrame_fnAddRef ( IGetFrame *  iface )

static

Definition at line 112 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
  ULONG ref = InterlockedIncrement(&This->ref);
 
  TRACE("(%p)\n", iface);
 
  return ref;
}

IGetFrame_fnBegin()

static HRESULT WINAPI IGetFrame_fnBegin ( IGetFrame *  iface, LONG  lStart, LONG  lEnd, LONG  lRate  )

static

Definition at line 260 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
 
  TRACE("(%p,%d,%d,%d)\n", iface, lStart, lEnd, lRate);
 
  This->bFixedStream = TRUE;
 
  return (IGetFrame_GetFrame(iface, lStart) ? AVIERR_OK : AVIERR_ERROR);
}

IGetFrame_fnEnd()

static HRESULT WINAPI IGetFrame_fnEnd ( IGetFrame *  iface )

static

Definition at line 272 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
 
  TRACE("(%p)\n", iface);
 
  This->bFixedStream = FALSE;
 
  return AVIERR_OK;
}

IGetFrame_fnGetFrame()

static LPVOID WINAPI IGetFrame_fnGetFrame ( IGetFrame *  iface, LONG  lPos  )

static

Definition at line 142 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
 
  LONG readBytes;
  LONG readSamples;
 
  TRACE("(%p,%d)\n", iface, lPos);
 
  /* We don't want negative start values! -- marks invalid buffer content */
  if (lPos < 0)
    return NULL;
 
  /* check state */
  if (This->pStream == NULL)
    return NULL;
  if (This->lpInFormat == NULL)
    return NULL;
 
  /* Could stream have changed? */
  if (! This->bFixedStream) {
    AVISTREAMINFOW sInfo;
 
    IAVIStream_Info(This->pStream, &sInfo, sizeof(sInfo));
 
    if (sInfo.dwEditCount != This->dwEditCount) {
      This->dwEditCount   = sInfo.dwEditCount;
      This->lCurrentFrame = -1;
    }
 
    if (sInfo.dwFormatChangeCount != This->dwFormatChangeCount) {
      /* stream has changed */
      if (This->lpOutFormat != NULL) {
    BITMAPINFOHEADER bi;
 
    bi = *This->lpOutFormat;
    AVIFILE_CloseCompressor(This);
 
    if (FAILED(IGetFrame_SetFormat(iface, &bi, NULL, 0, 0, -1, -1))) {
      if (FAILED(IGetFrame_SetFormat(iface, NULL, NULL, 0, 0, -1, -1)))
        return NULL;
    }
      } else if (FAILED(IGetFrame_SetFormat(iface, NULL, NULL, 0, 0, -1, -1)))
    return NULL;
    }
  }
 
  if (lPos != This->lCurrentFrame) {
    LONG lNext = IAVIStream_FindSample(This->pStream,lPos,FIND_KEY|FIND_PREV);
 
    if (lNext == -1)
      return NULL; /* frame doesn't exist */
    if (lNext <= This->lCurrentFrame && This->lCurrentFrame < lPos)
      lNext = This->lCurrentFrame + 1;
 
    for (; lNext <= lPos; lNext++) {
      /* new format for this frame? */
      if (This->bFormatChanges) {
    IAVIStream_ReadFormat(This->pStream, lNext,
                  This->lpInFormat, &This->cbInFormat);
    if (This->lpOutFormat != NULL) {
      if (This->lpOutFormat->biBitCount <= 8)
        ICDecompressGetPalette(This->hic, This->lpInFormat,
                   This->lpOutFormat);
    }
      }
 
      /* read input frame */
      while (FAILED(AVIStreamRead(This->pStream, lNext, 1, This->lpInBuffer,
                  This->cbInBuffer, &readBytes, &readSamples))) {
    /* not enough memory for input buffer? */
    readBytes = 0;
    if (FAILED(AVIStreamSampleSize(This->pStream, lNext, &readBytes)))
      return NULL; /* bad thing, but bad things will happen */
    if (readBytes <= 0) {
      ERR(": IAVIStream::Read doesn't return needed bytes!\n");
      return NULL;
    }
 
    /* IAVIStream::Read failed because of other reasons not buffersize? */
    if (This->cbInBuffer >= readBytes)
      break;
    This->cbInBuffer = This->cbInFormat + readBytes;
    This->lpInFormat = HeapReAlloc(GetProcessHeap(), 0, This->lpInFormat, This->cbInBuffer);
    if (This->lpInFormat == NULL)
      return NULL; /* out of memory */
    This->lpInBuffer = (BYTE*)This->lpInFormat + This->cbInFormat;
      }
 
      if (readSamples != 1) {
    ERR(": no frames read\n");
    return NULL;
      }
      if (readBytes != 0) {
    This->lpInFormat->biSizeImage = readBytes;
 
    /* nothing to decompress? */
    if (This->hic == NULL) {
      This->lCurrentFrame = lPos;
      return This->lpInFormat;
    }
 
    if (This->bResize) {
      ICDecompressEx(This->hic,0,This->lpInFormat,This->lpInBuffer,0,0,
             This->lpInFormat->biWidth,This->lpInFormat->biHeight,
             This->lpOutFormat,This->lpOutBuffer,This->x,This->y,
             This->dx,This->dy);
    } else {
      ICDecompress(This->hic, 0, This->lpInFormat, This->lpInBuffer,
               This->lpOutFormat, This->lpOutBuffer);
    }
      }
    } /* for (lNext < lPos) */
  } /* if (This->lCurrentFrame != lPos) */
 
  return (This->hic == NULL ? This->lpInFormat : This->lpOutFormat);
}

IGetFrame_fnQueryInterface()

static HRESULT WINAPI IGetFrame_fnQueryInterface ( IGetFrame *  iface, REFIID  refiid, LPVOID *  obj  )

static

Definition at line 95 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
 
  TRACE("(%p,%s,%p)\n", This, debugstr_guid(refiid), obj);
 
  if (IsEqualGUID(&IID_IUnknown, refiid) ||
      IsEqualGUID(&IID_IGetFrame, refiid)) {
    *obj = iface;
    IGetFrame_AddRef(iface);
    return S_OK;
  }
 
  return OLE_E_ENUM_NOMORE;
}

IGetFrame_fnRelease()

static ULONG WINAPI IGetFrame_fnRelease ( IGetFrame *  iface )

static

Definition at line 122 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
  ULONG ref = InterlockedDecrement(&This->ref);
 
  TRACE("(%p)\n", iface);
 
  if (!ref) {
    AVIFILE_CloseCompressor(This);
    if (This->pStream != NULL) {
      IAVIStream_Release(This->pStream);
      This->pStream = NULL;
    }
 
    HeapFree(GetProcessHeap(), 0, iface);
  }
 
  return ref;
}

IGetFrame_fnSetFormat()

static HRESULT WINAPI IGetFrame_fnSetFormat ( IGetFrame *  iface, LPBITMAPINFOHEADER  lpbiWanted, LPVOID  lpBits, INT  x, INT  y, INT  dx, INT  dy  )

static

Definition at line 283 of file getframe.c.

{
  IGetFrameImpl *This = impl_from_IGetFrame(iface);
 
  AVISTREAMINFOW     sInfo;
  LPBITMAPINFOHEADER lpbi         = lpbiWanted;
  BOOL               bBestDisplay = FALSE;
 
  TRACE("(%p,%p,%p,%d,%d,%d,%d)\n", iface, lpbiWanted, lpBits,
    x, y, dx, dy);
 
  if (This->pStream == NULL)
    return AVIERR_ERROR;
 
  if (lpbiWanted == (LPBITMAPINFOHEADER)AVIGETFRAMEF_BESTDISPLAYFMT) {
    lpbi = NULL;
    bBestDisplay = TRUE;
  }
 
  IAVIStream_Info(This->pStream, &sInfo, sizeof(sInfo));
  if (sInfo.fccType != streamtypeVIDEO)
    return AVIERR_UNSUPPORTED;
 
  This->bFormatChanges = (sInfo.dwFlags & AVISTREAMINFO_FORMATCHANGES) != 0;
  This->dwFormatChangeCount = sInfo.dwFormatChangeCount;
  This->dwEditCount         = sInfo.dwEditCount;
  This->lCurrentFrame       = -1;
 
  /* get input format from stream */
  if (This->lpInFormat == NULL) {
    HRESULT hr;
 
    This->cbInBuffer = (LONG)sInfo.dwSuggestedBufferSize;
    if (This->cbInBuffer == 0)
      This->cbInBuffer = 1024;
 
    IAVIStream_ReadFormat(This->pStream, sInfo.dwStart,
              NULL, &This->cbInFormat);
 
    This->lpInFormat = HeapAlloc(GetProcessHeap(), 0, This->cbInFormat + This->cbInBuffer);
    if (This->lpInFormat == NULL) {
      AVIFILE_CloseCompressor(This);
      return AVIERR_MEMORY;
    }
 
    hr = IAVIStream_ReadFormat(This->pStream, sInfo.dwStart, This->lpInFormat, &This->cbInFormat);
    if (FAILED(hr)) {
      AVIFILE_CloseCompressor(This);
      return hr;
    }
 
    This->lpInBuffer = ((LPBYTE)This->lpInFormat) + This->cbInFormat;
  }
 
  /* check input format */
  if (This->lpInFormat->biClrUsed == 0 && This->lpInFormat->biBitCount <= 8)
    This->lpInFormat->biClrUsed = 1u << This->lpInFormat->biBitCount;
  if (This->lpInFormat->biSizeImage == 0 &&
      This->lpInFormat->biCompression == BI_RGB) {
    This->lpInFormat->biSizeImage =
      DIBWIDTHBYTES(*This->lpInFormat) * This->lpInFormat->biHeight;
  }
 
  /* only to pass through? */
  if (This->lpInFormat->biCompression == BI_RGB && lpBits == NULL) {
    if (lpbi == NULL || 
    (lpbi->biCompression == BI_RGB &&
     lpbi->biWidth == This->lpInFormat->biWidth &&
     lpbi->biHeight == This->lpInFormat->biHeight &&
     lpbi->biBitCount == This->lpInFormat->biBitCount)) {
      This->lpOutFormat = This->lpInFormat;
      This->lpOutBuffer = DIBPTR(This->lpInFormat);
      return AVIERR_OK;
    }
  }
 
  /* need memory for output format? */
  if (This->lpOutFormat == NULL) {
    This->lpOutFormat =
      HeapAlloc(GetProcessHeap(), 0, sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD));
    if (This->lpOutFormat == NULL) {
      AVIFILE_CloseCompressor(This);
      return AVIERR_MEMORY;
    }
  }
 
  /* need handle to video compressor */
  if (This->hic == NULL) {
    FOURCC fccHandler;
 
    if (This->lpInFormat->biCompression == BI_RGB)
      fccHandler = comptypeDIB;
    else if (This->lpInFormat->biCompression == BI_RLE8)
      fccHandler = mmioFOURCC('R','L','E',' ');
    else
      fccHandler = sInfo.fccHandler;
 
    if (lpbi != NULL) {
      if (lpbi->biWidth == 0)
    lpbi->biWidth = This->lpInFormat->biWidth;
      if (lpbi->biHeight == 0)
    lpbi->biHeight = This->lpInFormat->biHeight;
    }
 
    This->hic = ICLocate(ICTYPE_VIDEO, fccHandler, This->lpInFormat, lpbi, ICMODE_DECOMPRESS);
    if (This->hic == NULL) {
      AVIFILE_CloseCompressor(This);
      return AVIERR_NOCOMPRESSOR;
    }
  }
 
  /* output format given? */
  if (lpbi != NULL) {
    /* check the given output format ... */
    if (lpbi->biClrUsed == 0 && lpbi->biBitCount <= 8)
      lpbi->biClrUsed = 1u << lpbi->biBitCount;
 
    /* ... and remember it */
    memcpy(This->lpOutFormat, lpbi,
       lpbi->biSize + lpbi->biClrUsed * sizeof(RGBQUAD));
    if (lpbi->biBitCount <= 8)
      ICDecompressGetPalette(This->hic, This->lpInFormat, This->lpOutFormat);
  } else {
    if (bBestDisplay) {
      ICGetDisplayFormat(This->hic, This->lpInFormat,
             This->lpOutFormat, 0, dx, dy);
    } else if (ICDecompressGetFormat(This->hic, This->lpInFormat,
                     This->lpOutFormat) < 0) {
      AVIFILE_CloseCompressor(This);
      return AVIERR_NOCOMPRESSOR;
    }
  }
 
  /* check output format */
  if (This->lpOutFormat->biClrUsed == 0 &&
      This->lpOutFormat->biBitCount <= 8)
    This->lpOutFormat->biClrUsed = 1u << This->lpOutFormat->biBitCount;
  if (This->lpOutFormat->biSizeImage == 0 &&
      This->lpOutFormat->biCompression == BI_RGB) {
    This->lpOutFormat->biSizeImage =
      DIBWIDTHBYTES(*This->lpOutFormat) * This->lpOutFormat->biHeight;
  }
 
  if (lpBits == NULL) {
    DWORD size = This->lpOutFormat->biClrUsed * sizeof(RGBQUAD);
 
    size += This->lpOutFormat->biSize + This->lpOutFormat->biSizeImage;
    This->lpOutFormat = HeapReAlloc(GetProcessHeap(), 0, This->lpOutFormat, size);
    if (This->lpOutFormat == NULL) {
      AVIFILE_CloseCompressor(This);
      return AVIERR_MEMORY;
    }
    This->lpOutBuffer = DIBPTR(This->lpOutFormat);
  } else
    This->lpOutBuffer = lpBits;
 
  /* for user size was irrelevant */
  if (dx == -1)
    dx = This->lpOutFormat->biWidth;
  if (dy == -1)
    dy = This->lpOutFormat->biHeight;
 
  /* need to resize? */
  if (x != 0 || y != 0) {
    if (dy == This->lpOutFormat->biHeight &&
        dx == This->lpOutFormat->biWidth)
      This->bResize = FALSE;
    else
      This->bResize = TRUE;
  }
 
  if (This->bResize) {
    This->x  = x;
    This->y  = y;
    This->dx = dx;
    This->dy = dy;
 
    if (ICDecompressExBegin(This->hic,0,This->lpInFormat,This->lpInBuffer,0,
                            0,This->lpInFormat->biWidth,
                            This->lpInFormat->biHeight,This->lpOutFormat,
                            This->lpOutBuffer, x, y, dx, dy) == ICERR_OK)
      return AVIERR_OK;
  } else if (ICDecompressBegin(This->hic, This->lpInFormat,
                               This->lpOutFormat) == ICERR_OK)
    return AVIERR_OK;
 
  AVIFILE_CloseCompressor(This);
 
  return AVIERR_COMPRESSOR;
}

impl_from_IGetFrame()

static IGetFrameImpl * impl_from_IGetFrame ( IGetFrame *  iface )

inlinestatic

Definition at line 72 of file getframe.c.

{
  return CONTAINING_RECORD(iface, IGetFrameImpl, IGetFrame_iface);
}

Referenced by IGetFrame_fnAddRef(), IGetFrame_fnBegin(), IGetFrame_fnEnd(), IGetFrame_fnGetFrame(), IGetFrame_fnQueryInterface(), IGetFrame_fnRelease(), and IGetFrame_fnSetFormat().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

avifile

)

Variable Documentation

igetframeVtbl

const struct IGetFrameVtbl igetframeVtbl

static

Initial value:

= {
  IGetFrame_fnQueryInterface,
  IGetFrame_fnAddRef,
  IGetFrame_fnRelease,
  IGetFrame_fnGetFrame,
  IGetFrame_fnBegin,
  IGetFrame_fnEnd,
  IGetFrame_fnSetFormat
}

Definition at line 477 of file getframe.c.

Referenced by AVIFILE_CreateGetFrame().