AviFile.h

#pragma once

#ifndef __AVIFILE_H
#define __AVIFILE_H

#include <vfw.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(/*LPCSTR	lpszFileName = _T("Output.avi"),*/ const wchar_t* lpszFileName,
			DWORD dwCodec = 0,//mmioFOURCC('M','P','G','4'),
			DWORD dwFrameRate = 30);

	/// <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