/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/

#include "hsConfig.h"

#if HS_BUILD_FOR_WIN32
#define WIN32_EXTRA_LEAN
#define WIN32_LEAN_AND_MEAN
#ifndef _WINDOWS_H_	// redundant include guard to minimize compile times
#define _WINDOWS_H_
#include <windows.h>
#endif // _WINDOWS_H_
#include "vfw.h"
#endif // HS_BUILD_FOR_WIN32

#include "hsTypes.h"
#include "plLayerAVI.h"
#include "../plGImage/plMipmap.h"

#if HS_BUILD_FOR_WIN32
class plAVIFileInfo
{
public:
	plAVIFileInfo() : fAVIStream(nil), fGetFrame(0) {}
	IAVIStream*					fAVIStream;
	AVISTREAMINFO				fAVIStreamInfo;
	PGETFRAME					fGetFrame;      // Where in the stream to get the next frame
};
#else // HS_BUILD_FOR_WIN32
struct plAVIFileInfo
{
};
#endif // HS_BUILD_FOR_WIN32

static hsBool						ICopySourceToTexture24(BITMAPINFO* bmi, plMipmap* t);
static hsBool						ICopySourceToTexture16(BITMAPINFO* bmi, plMipmap* t);


plLayerAVI::plLayerAVI()
{
	fAVIInfo = TRACKED_NEW plAVIFileInfo;
}

plLayerAVI::~plLayerAVI()
{
	ICloseMovie();

	delete fAVIInfo;
}


hsBool plLayerAVI::IInit()
{
#if HS_BUILD_FOR_WIN32
	int ret = AVIStreamOpenFromFile( &fAVIInfo->fAVIStream, 
								fMovieName,													
								streamtypeVIDEO, 
								0, 
								OF_READ,
								NULL);

	if (ret)
		return ISetFault("Error opening AVI");

	if( !(fAVIInfo->fGetFrame = AVIStreamGetFrameOpen(fAVIInfo->fAVIStream, NULL)) )
		return ISetFault("Error positioning AVI");

	if( AVIStreamInfo(fAVIInfo->fAVIStream, &fAVIInfo->fAVIStreamInfo, sizeof(AVISTREAMINFO)) )
		return ISetFault("Error getting AVI info");

	BITMAPINFO* bmi;
	if( !(bmi = (BITMAPINFO*)AVIStreamGetFrame(fAVIInfo->fGetFrame, fCurrentFrame >= 0 ? fCurrentFrame : 0)) )
		return ISetFault("Can't get first frame");
	ISetSize(bmi->bmiHeader.biWidth, bmi->bmiHeader.biHeight);

	Int32 endFrame = fAVIInfo->fAVIStreamInfo.dwLength-1;
	hsScalar length = float(endFrame) * float(fAVIInfo->fAVIStreamInfo.dwScale) 
						/ float(fAVIInfo->fAVIStreamInfo.dwRate);
	ISetLength(length);
#endif

	return false;
}

Int32 plLayerAVI::ISecsToFrame(hsScalar secs)
{
	float timeScale = float(fAVIInfo->fAVIStreamInfo.dwRate) / float(fAVIInfo->fAVIStreamInfo.dwScale);
	return Int32(secs * timeScale + 0.5f);
}

hsBool plLayerAVI::IGetCurrentFrame()
{
	if( !fAVIInfo->fAVIStream )
		IInit();

	ICheckBitmap();
	
	BITMAPINFO* bmi;
	if( !(bmi = (BITMAPINFO*)AVIStreamGetFrame(fAVIInfo->fGetFrame, fCurrentFrame)) )
		return ISetFault("Can't get frame");

	switch(bmi->bmiHeader.biBitCount)
	{
	case 16:
		return ICopySourceToTexture16(bmi, plMipmap::ConvertNoRef(GetTexture()));
		break;
	case 24:
		return ICopySourceToTexture24(bmi, plMipmap::ConvertNoRef(GetTexture()));
		break;
	default:
		return ISetFault("Unknown AVI color depth");
	}
	return true;
}

static hsBool ICopySourceToTexture16(BITMAPINFO* bmi, plMipmap* b)
{
	hsAssert( b != nil, "nil mipmap passed to ICopySourceToTexture16()" );

	UInt16* pSrc = (UInt16*)( bmi->bmiHeader.biSize + (BYTE*)bmi );

	UInt32* pix = (UInt32*)b->GetImage();
	pix += b->GetWidth() * b->GetHeight();

	int width = bmi->bmiHeader.biWidth;
	int height = bmi->bmiHeader.biHeight;

	int useHeight = hsMinimum(height, b->GetHeight());
	int useWidth = hsMinimum(width, b->GetWidth());
	int i;
	for( i = 0; i < useHeight; i++ )
	{
		UInt16* src = pSrc;
		pSrc += width;

		pix -= b->GetWidth();
		UInt32* tPix = pix;
		int j;
		for( j = 0; j < useWidth; j++ )
		{
			*tPix = ((*src & 0x001f) << 3) // up 3 
				| ((*src & 0x03e0) << 6) // down 5 up 3 up 8
				| ((*src & 0x7c00) << 9) // down 10 up 3 up 16
				| (0xff << 24); // alpha
			src++;
			++tPix;
		}
	}
	
	return false;
}

static hsBool ICopySourceToTexture24(BITMAPINFO* bmi, plMipmap* b)
{
	hsAssert( b != nil, "nil mipmap passed to ICopySourceToTexture24()" );

	unsigned char* pSrc = (unsigned char*)( bmi->bmiHeader.biSize + (BYTE*)bmi );

	hsRGBAColor32* pix = (hsRGBAColor32*)b->GetImage();
	pix += b->GetWidth() * b->GetHeight();

	int width = bmi->bmiHeader.biWidth;
	int height = bmi->bmiHeader.biHeight;

	int useHeight = hsMinimum(height, b->GetHeight());
	int useWidth = hsMinimum(width, b->GetWidth());
	int i;

	for( i = 0; i < useHeight; i++ )
	{
		unsigned char* src = pSrc;
		pSrc += width * 3;

		pix -= b->GetWidth();
		hsRGBAColor32* tPix = pix;
		int j;
		for( j = 0; j < useWidth; j++ )
		{
			tPix->b = *src++;
			tPix->g = *src++;
			tPix->r = *src++;
			++tPix;
		}
	}
	
	return false;
}

hsBool plLayerAVI::ICloseMovie()
{
	if( fAVIInfo->fGetFrame )
		AVIStreamGetFrameClose(fAVIInfo->fGetFrame);
	fAVIInfo->fGetFrame = 0;

	if( fAVIInfo->fAVIStream )
		AVIStreamRelease(fAVIInfo->fAVIStream);

	fAVIInfo->fAVIStream = nil;

	return false;
}

hsBool plLayerAVI::IRelease()
{
	ICloseMovie();

	return false;
}