AviFile.h

#pragma once

#ifndef __AVIFILE_H
#define __AVIFILE_H

#include <vfw.h>
#include <TChar.h>

class CAviFile
{
    HDC                 m_hAviDC;
    HANDLE              m_hHeap;
    LPVOID              m_lpBits;                   // Useful for holding the bitmap content, if any
    LONG                m_lSample;                  // Keeps track of the current Frame Index
    PAVIFILE            m_pAviFile;
    PAVISTREAM          m_pAviStream;
    PAVISTREAM          m_pAviCompressedStream;
    AVISTREAMINFO       m_AviStreamInfo;
    AVICOMPRESSOPTIONS  m_AviCompressOptions;
    DWORD               m_dwFrameRate;              // Frames Per Second Rate (FPS)
    DWORD               m_dwFCCHandler;             // Video Codec FourCC   
    TCHAR               m_szFileName[MAX_PATH];     // Holds the Output Movie File Name
    TCHAR               m_szErrMsg[MAX_PATH];       // Holds the Last Error Message, if any
    
    int                 m_nAppendFuncSelector;      //0=Dummy   1=FirstTime 2=Usual

    HRESULT AppendFrameFirstTime(HBITMAP );
    HRESULT AppendFrameUsual(HBITMAP);
    HRESULT AppendDummy(HBITMAP);
    HRESULT (CAviFile::*pAppendFrame[3])(HBITMAP hBitmap);

    HRESULT AppendFrameFirstTime(int, int, LPVOID,int );
    HRESULT AppendFrameUsual(int, int, LPVOID,int );
    HRESULT AppendDummy(int, int, LPVOID,int );
    HRESULT (CAviFile::*pAppendFrameBits[3])(int, int, LPVOID, int);

    /// Takes care of creating the memory, streams, compression options etc. required for the movie
    HRESULT InitMovieCreation(int nFrameWidth, int nFrameHeight, int nBitsPerPixel);

    /// Takes care of releasing the memory and movie related handles
    void ReleaseMemory();

    /// Sets the Error Message
    void SetErrorMessage(LPCTSTR lpszErrMsg);

public:
    /// <Summary>
    /// Constructor accepts the filename, video code and frame rate settings
    /// as parameters.
    /// lpszFileName: Name of the output movie file to create
    /// dwCodec: FourCC of the Video Codec to be used for compression
    /// dwFrameRate: The Frames Per Second (FPS) setting to be used for the movie
    /// </Summary>
    /// <Remarks>
    /// The default Codec used here is MPG4. To use a different codec, pass its Fourcc
    /// value as the input parameter for dwCodec. 
    /// For example,
    /// pass mmioFOURCC('D','I','V','X') to use DIVX codec, or
    /// pass mmioFOURCC('I','V','5','0') to use IV50 codec etc...
    ///
    /// Also, you can pass just 0 to avoid the codecs altogether. In that case, Frames 
    /// would be saved as they are without any compression; However, the output movie file
    /// size would be very huge in that case.
    ///
    /// Finally, make sure you have the codec installed on the machine before 
    /// passing its Fourcc here.
    /// </Remarks>
    CAviFile(LPCTSTR lpszFileName=_T("Output.avi"), 
            DWORD dwCodec = mmioFOURCC('M','P','G','4'),
            DWORD dwFrameRate = 1);

    /// <Summary> 
    /// Destructor closes the movie file and flushes all the frames
    /// </Summary>
    ~CAviFile(void);

    /// </Summary>
    /// Inserts the given HBitmap into the movie as a new Frame at the end.
    /// </Summary>
    HRESULT AppendNewFrame(HBITMAP hBitmap);

    /// </Summary>
    /// Inserts the given bitmap bits into the movie as a new Frame at the end.
    /// The width, height and nBitsPerPixel are the width, height and bits per pixel
    /// of the bitmap pointed to by the input pBits.
    /// </Summary>
    HRESULT AppendNewFrame(int nWidth, int nHeight, LPVOID pBits,int nBitsPerPixel);

    /// <Summary>
    /// Returns the last error message, if any.
    /// </Summary>
    LPCTSTR GetLastErrorMessage() const {   return m_szErrMsg;  }
};

#endif