#include <common/bitmap.hpp>
#include <common/file.hpp>

#include <jpgimg/jpgimg.hpp>

#include <avifile/avifile.hpp>
#include <avifile/mainhdr.hpp>
#include <avifile/vidc.hpp>
#include <avifile/riffhdr.hpp>

bool proto1(void);
bool proto2(const String &pathAVIFile);
bool createAVIFile(const String &pathAVIFile,Block<Bitmap> &images);
bool createAVIFile(const String &pathAVIFile,Block<SmartPointer<JPGImage> >&images);
bool createAVI(const String &pathAVIFile);

int PASCAL WinMain(HINSTANCE /*hInstance*/,HINSTANCE /*hPrevInstance*/,LPSTR lpszCmdLine,int /*nCmdShow*/)
{
  Block<SmartPointer<JPGImage> > images;

	images.insert(&SmartPointer<JPGImage>());
	images[0]=new JPGImage();
	images[0].disposition(PointerDisposition::Delete);
	images[0]->decode("d:\\im000256.jpg");

	createAVIFile("d:\\foo.avi",images);

//  return proto2("c:\\winnt\\clock.avi");
//  return proto2("d:\\foo.avi");
//  return proto2("d:\\avi\\capture2.avi");

//   return createAVI("d:\\foo.avi"); 
	return 0;
}

bool createAVI(const String &pathAVIFile)
{
  Block<Bitmap> bitmaps;
	bitmaps.insert(&Bitmap());
	Bitmap &bitmap=bitmaps[0];
 	bitmap="d:\\work\\avifile\\frm00011.bmp";

//  return createAVIFile(pathAVIFile,bitmaps); 
  return createAVIFile(pathAVIFile,bitmaps); 
}

bool createAVIFile(const String &pathAVIFile,Block<SmartPointer<JPGImage> >&images)
{
  PAVIFILE pFile;
  PAVISTREAM pStream;
  AVISTREAMINFO aviStreamInfo;
	HRESULT hResult;

  pFile=0;
	pStream=0;
  AVIFileInit();
  hResult=AVIFileOpen(&pFile,pathAVIFile,OF_WRITE|OF_CREATE,NULL);
  if(hResult!=AVIERR_OK)return false;
	JPGImage &image=*images[0];
	::memset(&aviStreamInfo,0,sizeof(AVISTREAMINFO));
	aviStreamInfo.fccType=streamtypeVIDEO;
	aviStreamInfo.fccHandler=0;
	aviStreamInfo.dwScale=1;
	aviStreamInfo.dwRate=1;
	aviStreamInfo.dwSuggestedBufferSize=image.getBitmapInfo().sizeImage();
	SetRect(&aviStreamInfo.rcFrame,0,0,image.width(),image.height());
	hResult=AVIFileCreateStream(pFile,&pStream,&aviStreamInfo);
	if(hResult==AVIERR_OK)
	{
    hResult=AVIStreamSetFormat(pStream,0,&image.getBitmapInfo().getBITMAPINFOHEADER(),
		  image.getBitmapInfo().getBITMAPINFOHEADER().biSize+(image.getBitmapInfo().getBITMAPINFOHEADER().biClrUsed*sizeof(RGBQUAD)));
		if(hResult==AVIERR_OK)
		{
		  for(int index=0;index<images.size();index++)
			{
			  JPGImage &image=*images[index];
				GlobalData<BYTE> rawData;
				image.getRawData(rawData);
			  hResult=AVIStreamWrite(pStream,index,1,
				&rawData[0],
				rawData.size(),
				AVIIF_KEYFRAME,0,0);
        if(hResult!=AVIERR_OK)break;
			}
		}
	}
  if(pStream)AVIStreamClose(pStream);
	if(pFile)AVIFileClose(pFile);
	AVIFileExit();
	return true;
}

bool createAVIFile(const String &pathAVIFile,Block<Bitmap> &images)
{
  PAVIFILE pFile;
  PAVISTREAM pStream;
  AVISTREAMINFO aviStreamInfo;
	HRESULT hResult;

  pFile=0;
	pStream=0;
  AVIFileInit();
  hResult=AVIFileOpen(&pFile,pathAVIFile,OF_WRITE|OF_CREATE,NULL);
  if(hResult!=AVIERR_OK)return false;
	Bitmap bitmap=images[0];
	::memset(&aviStreamInfo,0,sizeof(AVISTREAMINFO));
	aviStreamInfo.fccType=streamtypeVIDEO;
	aviStreamInfo.fccHandler=0;
	aviStreamInfo.dwScale=1;
	aviStreamInfo.dwRate=1;
	aviStreamInfo.dwSuggestedBufferSize=bitmap.getInfoPtr()->bmiHeader.biSizeImage;
	SetRect(&aviStreamInfo.rcFrame,0,0,bitmap.width(),bitmap.height());
	hResult=AVIFileCreateStream(pFile,&pStream,&aviStreamInfo);
	if(hResult==AVIERR_OK)
	{
    hResult=AVIStreamSetFormat(pStream,0,&bitmap.getInfoPtr()->bmiHeader,bitmap.getInfoPtr()->bmiHeader.biSize+(bitmap.getInfoPtr()->bmiHeader.biClrUsed*sizeof(RGBQUAD)));
		if(hResult==AVIERR_OK)
		{
		  for(int index=0;index<images.size();index++)
			{
			  Bitmap &bitmap=images[index];
			  hResult=AVIStreamWrite(pStream,index,1,
				bitmap.ptrData(),
				bitmap.getInfoPtr()->bmiHeader.biSizeImage,
				AVIIF_KEYFRAME,0,0);
        if(hResult!=AVIERR_OK)break;
			}
		}
	}
  if(pStream)AVIStreamClose(pStream);
	if(pFile)AVIFileClose(pFile);
	AVIFileExit();
	return true;
}


bool proto2(const String &pathAVIFile)
{
	VideoCodec videoCodec;
	ICMOpen icmOpen;
	ICMInfo icmInfo;
	ICMDecompress icmDecompress;
  WORD streamCount;
	WORD frameCount(0);

	AVIFile aviFile(pathAVIFile);
	AVIStreamData &aviStreamData=(AVIStreamData&)aviFile;
	if(0==(streamCount=aviStreamData.size()))return FALSE;
  ::OutputDebugString(String("FCCType:")+aviStreamData[0].fccType().toString()+String("\n"));
  ::OutputDebugString(String("FCCHandler:")+aviStreamData[0].fccHandler().toString()+String("\n"));

  FOURCC driverName("mrle");
  Block<DriverName> driverNames;
  videoCodec.enumerateDrivers(driverNames);
	for(int index=0;index<driverNames.size();index++)
	{
	  ::OutputDebugString(driverNames[index].driverName()+String("\n"));
	}

	if(!(VideoCodec::OkResult==videoCodec.openDriver(driverName)))return false;
  icmOpen.fccType(aviStreamData[0].fccType());
	icmOpen.fccHandler(aviStreamData[0].fccHandler());
	icmOpen.flags(ICMOpen::DeCompress);
	icmOpen.version(1);
	videoCodec.driverLoad();
	videoCodec.driverEnable();
	videoCodec.driverProc(icmOpen);
	videoCodec.driverProc(icmInfo);
//	videoCodec.icmConfigure(::GetFocus());

	for(short streamIndex=0;streamIndex<streamCount;streamIndex++)
	{
		if(!StreamType::Video==aviStreamData[streamIndex].streamType())continue;
		AVIBitmap &aviBitmap=aviStreamData[streamIndex].getBitmap();
		AVIBitmap dstBitmap(aviBitmap);
		dstBitmap.compression(BI_RGB);
		dstBitmap.sizeImage(dstBitmap.width()*dstBitmap.height());
		dstBitmap.colorUsed(aviBitmap.colorUsed());

		if(VideoCodec::FailResult==videoCodec.canDecompress(aviBitmap))continue;
		WORD movieCount(((AVIMovieHeader&)aviFile).size());
		for(short movieIndex=0;movieIndex<movieCount;movieIndex++)
		{
			MovieData &srcMovieData=((AVIMovieHeader&)aviFile)[movieIndex];
			if(!((srcMovieData.streamNumber()==streamIndex)&&
					 (MovieData::DIBCompressed==srcMovieData.dataType()||
						MovieData::IV==srcMovieData.dataType()||
						MovieData::DIBUnCompressed==srcMovieData.dataType())))continue;
			String pathFileName;
			MovieData dstMovieData;
			dstMovieData.dataType(srcMovieData.dataType());
			dstMovieData.streamNumber(srcMovieData.streamNumber());
			::sprintf(pathFileName,"FRM%05d.BMP",frameCount++);
			if(VideoCodec::OkResult!=videoCodec.decompressFrame(aviBitmap,dstBitmap,srcMovieData,dstMovieData))
			{
			  ::printf("Error decompressing frame\n");
				continue;
			}
			Bitmap imageBitmap(pathFileName,(BitmapInfo&)dstBitmap,(GlobalData<BYTE>&)dstMovieData);
			imageBitmap.saveBitmap();
    }
	}
	videoCodec.closeDriver();
	return FALSE;
}

bool proto1(void)
{
	VideoCodec videoCodec;
	ICMOpen icmOpen;
	ICMInfo icmInfo;
	ICMDecompress icmDecompress;
  WORD streamCount;
	WORD frameCount(0);
	String commandLine;

//	if(!(VideoCodec::OkResult==videoCodec.openDriver(FOURCC("I420"))))return FALSE;

//	lpszCmdLine="D:\\CAPTURE1.AVI";  
	commandLine="D:\\CAPTURE2.AVI";  
//  commandLine="c:\\winnt\\clock.avi";

	AVIFile aviFile(commandLine);
	AVIStreamData &aviStreamData=(AVIStreamData&)aviFile;
	if(0==(streamCount=aviStreamData.size()))return FALSE;
  ::OutputDebugString(String("FCCType:")+aviStreamData[0].fccType().toString()+String("\n"));
  ::OutputDebugString(String("FCCHandler:")+aviStreamData[0].fccHandler().toString()+String("\n"));

  icmOpen.fccType(aviStreamData[0].fccType());
//	icmOpen.fccType(FOURCC("vidc"));
	icmOpen.fccHandler(aviStreamData[0].fccHandler());
//  icmOpen.fccType(aviStreamData[0].fccType());
//	icmOpen.fccHandler(FOURCC("DIVX"));

	icmOpen.flags(ICMOpen::DeCompress);
	icmOpen.version(1);
	videoCodec.driverLoad();
	videoCodec.driverEnable();
	videoCodec.driverProc(icmOpen);
	videoCodec.driverProc(icmInfo);
//	videoCodec.icmConfigure(::GetFocus());
	for(short streamIndex=0;streamIndex<streamCount;streamIndex++)
	{
		if(!StreamType::Video==aviStreamData[streamIndex].streamType())continue;
		AVIBitmap &aviBitmap=(AVIBitmap&)(aviStreamData[streamIndex]);
		AVIBitmap dstBitmap(aviBitmap);
		dstBitmap.compression(BI_RGB);
		dstBitmap.sizeImage(dstBitmap.width()*dstBitmap.height());
		dstBitmap.colorUsed(0);

//		if(VideoCodec::FailResult==videoCodec.canDecompress(aviBitmap))continue;
		WORD movieCount(((AVIMovieHeader&)aviFile).size());
		for(short movieIndex=0;movieIndex<movieCount;movieIndex++)
		{
			MovieData &srcMovieData=((AVIMovieHeader&)aviFile)[movieIndex];
			if(!((srcMovieData.streamNumber()==streamIndex)&&
					 (MovieData::DIBCompressed==srcMovieData.dataType()||
						MovieData::IV==srcMovieData.dataType()||
						MovieData::DIBUnCompressed==srcMovieData.dataType())))continue;


      File outfile;
			outfile.open("d:\\frame.dvsd","wb");
			outfile.write((char*)&srcMovieData[0],srcMovieData.size());

			String pathFileName;
			MovieData dstMovieData;
			dstMovieData.dataType(srcMovieData.dataType());
			dstMovieData.streamNumber(srcMovieData.streamNumber());
			::sprintf(pathFileName,"FRM%05d.BMP",frameCount++);

			  Bitmap imageBitmap(pathFileName,(BitmapInfo&)aviBitmap,(GlobalData<BYTE>&)srcMovieData);
			  imageBitmap.saveBitmap();

			if(VideoCodec::OkResult!=videoCodec.decompressFrame(aviBitmap,dstBitmap,srcMovieData,dstMovieData))
			{
			  ::printf("Error decompressing frame\n");
				continue;
			}
//			Bitmap imageBitmap(pathFileName,(BitmapInfo&)dstBitmap,(GlobalData<BYTE>&)dstMovieData);
//			imageBitmap.saveBitmap();
    }
	}
	videoCodec.closeDriver();
	return FALSE;
}