Splitting an AVI File into a Multipage TIFF File Using ltmmSampleTarget for C
// define helper macros for using interfaces under C
#define COBJMACROS
#include "stdafx.h"
// include the LEAD Multimedia TOOLKIT header
#include "ltmm.h"
#include "l_bitmap.h"
// include amvideo.h for VIDEOINFOHEADER, available in the Visual Studio 6.0 or the DirectX SDK
#include <amvideo.h>
// includes for string handling
#include <tchar.h>
#include <wchar.h>
#include <string.h>
#include <stdio.h>
/////////////////////////////////////////////////////////////////
// AviToTIFF
// Creates a multipage TIFF file from an AVI file
// pszAviFile - source file name
// pszOutputFile - the name of the output TIF file
// nMaxFrames - maximum number of pages (frames) to append to the file
//
HRESULT AviToTIFF(LPCWSTR pszAviFile, L_CHAR L_FAR* pszOutputFile, int nMaxFrames)
{
HRESULT hr;
IltmmConvert* pConvert = NULL;
IltmmSampleTarget* pSampleTarget = NULL;
IltmmMediaTypeDisp* pMediaType = NULL;
IltmmMediaSampleDisp* pMediaSample;
VARIANT varFormat;
VARIANT varBuffer;
VIDEOINFOHEADER* pVIH;
void* pBuffer;
BSTR bstr;
int n;
BITMAPHANDLE Bitmap;
L_INT nRet;
// initialize COM library
hr = CoInitialize(NULL);
if(FAILED(hr))
goto error;
// create the convert object
hr = CoCreateInstance(&CLSID_ltmmConvert, NULL, CLSCTX_INPROC_SERVER, &IID_IltmmConvert, (void**) &pConvert);
if(FAILED(hr))
goto error;
// create the target object
hr = CoCreateInstance(&CLSID_ltmmSampleTarget, NULL, CLSCTX_INPROC_SERVER, &IID_IltmmSampleTarget, (void**) &pSampleTarget);
if(FAILED(hr))
goto error;
// create the media type object
hr = CoCreateInstance(&CLSID_ltmmMediaType, NULL, CLSCTX_INPROC_SERVER, &IID_IltmmMediaTypeDisp, (void**) &pMediaType);
if(FAILED(hr))
goto error;
// set type to MEDIATYPE_Video
bstr = SysAllocString(L"{73646976-0000-0010-8000-00AA00389B71}");
hr = IltmmMediaTypeDisp_put_Type (pMediaType, bstr);
SysFreeString(bstr);
if(FAILED(hr))
goto error;
// set subtype to MEDIASUBTYPE_RGB24
bstr = SysAllocString(L"{e436eb7d-524f-11ce-9f53-0020af0ba770}");
hr = IltmmMediaTypeDisp_put_Subtype (pMediaType, bstr);
SysFreeString(bstr);
if(FAILED(hr))
goto error;
// set the accepted media type
hr = IltmmSampleTarget_SetAcceptedMediaType(pSampleTarget, pMediaType);
if(FAILED(hr))
goto error;
// release the media type, since we don't need it anymore
IltmmMediaTypeDisp_Release(pMediaType);
pMediaType = NULL;
// set the convert input file name
bstr = SysAllocString(pszAviFile);
hr = IltmmConvert_put_SourceFile(pConvert, bstr);
SysFreeString(bstr);
if(FAILED(hr))
goto error;
// set the convert object target
hr = IltmmConvert_put_TargetObject(pConvert, (IUnknown*)pSampleTarget);
if(FAILED(hr))
goto error;
// start the conversion
hr = IltmmConvert_StartConvert(pConvert);
if(FAILED(hr))
goto error;
// get the connected media type
hr = IltmmSampleTarget_GetConnectedMediaType(pSampleTarget, &pMediaType);
if(FAILED(hr))
goto error;
// get the VIDEOINFOHEADER
hr = IltmmMediaTypeDisp_get_Format(pMediaType, &varFormat);
if(FAILED(hr))
goto error;
hr = SafeArrayAccessData(V_ARRAY(&varFormat), (void**) &pVIH);
if(FAILED(hr))
{
VariantClear(&varFormat);
goto error;
}
// set the shell bitmap handle to match the sample data
nRet = L_CreateBitmap(&Bitmap, TYPE_USER,
pVIH->bmiHeader.biWidth, abs(pVIH->bmiHeader.biHeight),
pVIH->bmiHeader.biBitCount, ORDER_BGR, NULL,
pVIH->bmiHeader.biHeight > 0 ? BOTTOM_LEFT : TOP_LEFT,
NULL, pVIH->bmiHeader.biSizeImage);
if(nRet != SUCCESS)
{
hr = FAILURE;
// make sure the bitmap handle is initialized so L_FreeBitmap doesn't crash
L_InitBitmap(&Bitmap, 1, 1, 1);
goto converterror;
}
for(n = 0; n < 1 * nMaxFrames; n++)
{
// fetch a sample
hr = IltmmSampleTarget_GetSample(pSampleTarget, 1000, &pMediaSample);
if(FAILED(hr) || !pMediaSample)
break;
// access the image bits
hr = IltmmMediaSampleDisp_get_Buffer(pMediaSample, &varBuffer);
if(FAILED(hr))
{
IltmmMediaSampleDisp_Release(pMediaSample);
IltmmConvert_StopConvert(pConvert);
goto converterror;
}
hr = SafeArrayAccessData(V_ARRAY(&varBuffer), &pBuffer);
if(FAILED(hr))
{
VariantClear(&varBuffer);
IltmmMediaSampleDisp_Release(pMediaSample);
IltmmConvert_StopConvert(pConvert);
goto converterror;
}
// set the data pointer for the bitmap to be the sample
nRet = L_SetBitmapDataPointer(&Bitmap, (L_UCHAR L_FAR*)pBuffer, pVIH->bmiHeader.biSizeImage);
if(nRet == SUCCESS)
// append a JPEG-compressed page to the output file
nRet = L_SaveFile(pszOutputFile, &Bitmap, FILE_LEAD1JTIF, 0, 25, SAVEFILE_MULTIPAGE, NULL, NULL, NULL);
// release the sample
SafeArrayUnaccessData(V_ARRAY(&varBuffer));
VariantClear(&varBuffer);
IltmmMediaSampleDisp_Release(pMediaSample);
if(nRet != SUCCESS)
{
// an error occured - abort the process
hr = E_FAIL;
IltmmConvert_StopConvert(pConvert);
goto converterror;
}
}
// stop
hr = IltmmConvert_StopConvert(pConvert);
if(FAILED(hr))
goto converterror;
// cleanup and exit
hr = S_OK;
converterror:
SafeArrayUnaccessData(V_ARRAY(&varFormat));
VariantClear(&varFormat);
error:
if(pConvert)
IltmmConvert_Release(pConvert);
if(pSampleTarget)
IltmmSampleTarget_Release(pSampleTarget);
if(pMediaType)
IltmmMediaType_Release(pMediaType);
CoUninitialize();
return hr;
}
int main(int argc, char* argv[])
{
AviToTIFF(L"c:\\count.avi", "c:\\multi.tif", 100);
return 0;
}