Click here to Skip to main content
Click here to Skip to main content

Extracting AVI Frames

, , 5 Nov 2004
Rate this:
Please Sign up or sign in to vote.
Introducing AVI functions for extracting frames of an AVI movie and saving them in bitmap files.

Introduction

In the last project that I worked on, the project required to extract frames of an AVI movie and save them as bitmaps files. Here I want to share my experiences in this project with you.

AVIFile Functions

The best way to obtain information about AVI format and learn more about the internal behavior of the file, is going directly to MSDN as I did. The AVI related functions are inside vfw.h header file. This means you must include this header and add vfw32.lib library file to your project.

Here is the list of functions that I used in the project and a brief information about them (taken from MSDN):

void AVIFileInit(void); The AVIFileInit function initializes the AVIFile library.
void AVIFileOpen(PAVIFILE* ppfile, LPCTSTR szFile, UINT mode, CLSID pclsidHandler); The AVIFileOpen function opens an AVI file and returns the address of a file interface used to access it.
void AVIFileInfo(PAVIFILE pfile, AVIFILEINFO* pfi, LONG lSize); The AVIFileInfo function obtains information about an AVI file.
void AVIFileGetStream(PAVIFILE pfile, PAVISTREAM* ppavi, DWORD fccType, LONG lParam); The AVIFileGetStream function returns the address of a stream interface that is associated with a specified AVI file.
LONG AVIStreamStart(PAVISTREAM pavi); The AVIStreamStart function returns the starting sample number for the stream.
LONG AVIStreamLength(PAVISTREAM pavi); The AVIStreamLength function returns the length of the stream.
PGETFRAME AVIStreamGetFrameOpen(PAVISTREAM pavi, LPBITMAPINFOHEADER lpbiWanted); The AVIStreamGetFrameOpen function prepares to decompress video frames from the specified video stream.
LPVOID AVIStreamGetFrame(PGETFRAME pgf, LONG lPos); The AVIStreamGetFrame function returns the address of a decompressed video frame.
void AVIStreamGetFrameClose(PGETFRAME pget); The AVIStreamGetFrameClose function releases resources used to decompress video frames.
LONG AVIStreamRelease(PAVISTREAM pavi); The AVIStreamRelease function decrements the reference count of an AVI stream interface handle, and closes the stream if the count reaches zero.
void AVIFileExit(void); The AVIFileExit function exits the AVIFile library and decrements the reference count for the library.

Now it's time to write some code:

BOOL ExtractAVIFrames(CString szFileName)
{
    AVIFileInit();

    PAVIFILE avi;
    int res=AVIFileOpen(&avi, szFileName, OF_READ, NULL);

    if (res!=AVIERR_OK)
    {
        //an error occures
        if (avi!=NULL)
            AVIFileRelease(avi);
        
        return FALSE;
    }

    AVIFILEINFO avi_info;
    AVIFileInfo(avi, &avi_info, sizeof(AVIFILEINFO));

    CString szFileInfo;
    szFileInfo.Format("Dimention: %dx%d\n"
                      "Length: %d frames\n"
                      "Max bytes per second: %d\n"
                      "Samples per second: %d\n"
                      "Streams: %d\n"
                      "File Type: %d", avi_info.dwWidth,
                            avi_info.dwHeight,
                            avi_info.dwLength,
                            avi_info.dwMaxBytesPerSec,
                            (DWORD) (avi_info.dwRate / avi_info.dwScale),
                            avi_info.dwStreams,
                            avi_info.szFileType);

    AfxMessageBox(szFileInfo, MB_ICONINFORMATION | MB_OK);

    PAVISTREAM pStream;
    res=AVIFileGetStream(avi, &pStream, streamtypeVIDEO /*video stream*/, 
                                               0 /*first stream*/);

    if (res!=AVIERR_OK)
    {
        if (pStream!=NULL)
            AVIStreamRelease(pStream);

        AVIFileExit();
        return FALSE;
    }

    //do some task with the stream
    int iNumFrames;
    int iFirstFrame;

    iFirstFrame=AVIStreamStart(pStream);
    if (iFirstFrame==-1)
    {
        //Error getteing the frame inside the stream

        if (pStream!=NULL)
            AVIStreamRelease(pStream);

        AVIFileExit();
        return FALSE;
    }

    iNumFrames=AVIStreamLength(pStream);
    if (iNumFrames==-1)
    {
        //Error getteing the number of frames inside the stream
        
        if (pStream!=NULL)
            AVIStreamRelease(pStream);
        
        AVIFileExit();
        return FALSE;
    }

    //getting bitmap from frame
    BITMAPINFOHEADER bih;
    ZeroMemory(&bih, sizeof(BITMAPINFOHEADER));

    bih.biBitCount=24;    //24 bit per pixel
    bih.biClrImportant=0;
    bih.biClrUsed = 0;
    bih.biCompression = BI_RGB;
    bih.biPlanes = 1;
    bih.biSize = 40;
    bih.biXPelsPerMeter = 0;
    bih.biYPelsPerMeter = 0;
    //calculate total size of RGBQUAD scanlines (DWORD aligned)
    bih.biSizeImage = (((bih.biWidth * 3) + 3) & 0xFFFC) * bih.biHeight ;

    PGETFRAME pFrame;
    pFrame=AVIStreamGetFrameOpen(pStream, 
           NULL/*(BITMAPINFOHEADER*) AVIGETFRAMEF_BESTDISPLAYFMT*/ /*&bih*/);
    
    //Get the first frame
    int index=0;
    for (int i=iFirstFrame; i<iNumFrames; i++)
    {
        index= i-iFirstFrame;

        BYTE* pDIB = (BYTE*) AVIStreamGetFrame(pFrame, index);
        
        CreateFromPackedDIBPointer(pDIB, index);
    }

    AVIStreamGetFrameClose(pFrame);

    //close the stream after finishing the task
    if (pStream!=NULL)
        AVIStreamRelease(pStream);

    AVIFileExit();

    return TRUE;
}

The only one function that I must describe more about is: CreateFromPackedDIBPointer(). This function takes a pointer returned from AVIStreamGetFrame() function and creates a bitmap from it. As you know, the AVIStreamGetFrame() returns a pointer to DIB information about the frame. We take this pointer and create a bitmap from it.

BOOL CreateFromPackedDIBPointer(LPBYTE pDIB, int iFrame)
{
    ASSERT(pDIB!=NULL);

    //Creates a full-color (no palette) DIB from a pointer to a
    //full-color memory DIB

    //get the BitmapInfoHeader
    BITMAPINFOHEADER bih;
    RtlMoveMemory(&bih.biSize, pDIB, sizeof(BITMAPINFOHEADER));

    //now get the bitmap bits
    if (bih.biSizeImage < 1)
    {
        return FALSE;
    }

    BYTE* Bits=new BYTE[bih.biSizeImage];

    RtlMoveMemory(Bits, pDIB + sizeof(BITMAPINFOHEADER), bih.biSizeImage);

    //and BitmapInfo variable-length UDT
    BYTE memBitmapInfo[40];
    RtlMoveMemory(memBitmapInfo, &bih, sizeof(bih));

    BITMAPFILEHEADER bfh;
    bfh.bfType=19778;    //BM header
    bfh.bfSize=55 + bih.biSizeImage;
    bfh.bfReserved1=0;
    bfh.bfReserved2=0;
    bfh.bfOffBits=sizeof(BITMAPINFOHEADER) + sizeof(BITMAPFILEHEADER); //54
    
    CString FileName;
    FileName.Format("Frame-%05d.bmp", iFrame);
    
    FILE* fp=fopen(FileName, "wb");
    if (fp!=NULL)
    {
        fwrite(&bfh, sizeof(bfh), 1, fp);
        fwrite(&memBitmapInfo, sizeof(memBitmapInfo), 1, fp);
        fwrite(Bits, bih.biSizeImage, 1, fp);
        fclose(fp);
    }
    else
    {
        TRACE0(_T("Error writing the bitmap file"));
        return FALSE;
    }

    delete [] Bits;
    return TRUE;
}

Enjoy!

License

This article has no explicit license attached to it but may contain usage terms in the article text or the download files themselves. If in doubt please contact the author via the discussion board below.

A list of licenses authors might use can be found here

Share

About the Authors

A. Riazi
CEO Misbah3Com
Iran (Islamic Republic Of) Iran (Islamic Republic Of)
I was born in Shiraz, a very beautiful famous city in Iran. I started programming when I was 12 years old with GWBASIC. Since now, I worked with various programming languages from Basic, Foxpro, C/C++, Visual Basic, Pascal to MATLAB and now Visual C++.
I graduated from Iran University of Science & Technology in Communication Eng., and now work as a system programmer for a telecommunication industry.
I wrote several programs and drivers for Synthesizers, Power Amplifiers, GPIB, GPS devices, Radio cards, Data Acqusition cards and so many related devices.
I'm author of several books like Learning C (primary and advanced), Learning Visual Basic, API application for VB, Teach Yourself Object Oriented Programming (OOP) and etc.
I'm winner of January, May, August 2003 and April 2005 best article of month competetion, my articles are:

You can see list of my articles, by clicking here


Shafiee
Web Developer
Iran (Islamic Republic Of) Iran (Islamic Republic Of)
No Biography provided

Comments and Discussions

 
GeneralPlease Help me.. Pinmemberatans22-Jun-07 0:15 
GeneralRe: Please Help me.. PinmemberJohotech22-Jun-07 0:31 
GeneralRe: Please Help me.. Pinmemberatans24-Jun-07 16:39 
GeneralRe: Please Help me.. PinmemberJohotech25-Jun-07 7:34 
QuestionMatlab question Pinmembermarihan9-Jun-07 3:46 
QuestionHow can I do the same with file MPEG 1 and MPEG 2 Pinmemberhieunguyenit8-May-07 23:52 
GeneralLinking Error.......Need Help PinmemberGandhi_amit4-May-07 3:11 
GeneralRe: Linking Error.......Need Help PinmemberA. Riazi6-May-07 3:20 
This is because the code depends on MFC. Just replace CString with char* or remove it.
 
Best regards,
A. Riazi

Generalready project PinmemberDUNAKAN8-Apr-07 21:24 
Questionbyte* pDIB Question PinmemberFateJH8-Apr-07 15:39 
Generalsegmentation Pinmembermadhanblore7-Mar-07 4:40 
QuestionAVIStreamGetFrameOpen returns NULL Pinmemberbacer16-Oct-06 21:13 
QuestionAVIStreamGetFrame return NULL? Pinmemberbpadams9-Aug-06 5:59 
Answersame problem Pinmemberli.chen0016-Aug-06 4:32 
AnswerDIY - discussion Pinmemberli.chen0017-Aug-06 1:10 
Questionhow to create an HBITMAP [modified] Pinmemberdinhhuy24-Jun-06 6:44 
AnswerRe: how to create an HBITMAP PinmemberA. Riazi24-Jun-06 8:50 
General64-bit build of this SDK Pinmemberalqasrawi11-Jun-06 5:11 
Questionextract single frame as HBITMAP instead? Pinmembermound26-May-06 10:12 
GeneralSaving bmp's Pinmembershin_san11-May-06 23:26 
AnswerRe: Saving bmp's PinmemberA. Riazi13-May-06 21:42 
QuestionAVI length problem? Pinmemberwesb6_93-Apr-06 20:22 
AnswerRe: AVI length problem? PinmemberA. Riazi3-Apr-06 20:49 
AnswerRe: AVI length problem? Pinmembertgowtham11-Dec-06 7:00 
Generalsyntax error Pinmemberwongcl16-Mar-06 8:05 

General General    News News    Suggestion Suggestion    Question Question    Bug Bug    Answer Answer    Joke Joke    Rant Rant    Admin Admin   

Use Ctrl+Left/Right to switch messages, Ctrl+Up/Down to switch threads, Ctrl+Shift+Left/Right to switch pages.

| Advertise | Privacy | Terms of Use | Mobile
Web02 | 2.8.1411023.1 | Last Updated 5 Nov 2004
Article Copyright 2004 by A. Riazi, Shafiee
Everything else Copyright © CodeProject, 1999-2014
Layout: fixed | fluid