#include "avicodecrestrictions.h"

#include "tconvert.h"

#include <windows.h></windows.h>

#include <vfw.h></vfw.h>

namespace

{

HIC getCodec(const wstring &codecName, int &bpp)

{

	HIC hic = 0;

	ICINFO icinfo;

	memset(&icinfo, 0, sizeof(ICINFO));

	bool found = false;

	char descr[2048], name[2048];

	DWORD fccType = 0;

	BITMAPINFO inFmt;

	memset(&inFmt, 0, sizeof(BITMAPINFO));

	inFmt.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

	inFmt.bmiHeader.biWidth = inFmt.bmiHeader.biHeight = 100;

	inFmt.bmiHeader.biPlanes = 1;

	inFmt.bmiHeader.biCompression = BI_RGB;

	for (bpp = 32; (bpp >= 24) && !found; bpp -= 8) {

		//find the codec.

		inFmt.bmiHeader.biBitCount = bpp;

		for (int i = 0; ICInfo(fccType, i, &icinfo); i++) {

			hic = ICOpen(icinfo.fccType, icinfo.fccHandler, ICMODE_COMPRESS);

			ICGetInfo(hic, &icinfo, sizeof(ICINFO)); // Find out the compressor name

			WideCharToMultiByte(CP_ACP, 0, icinfo.szDescription, -1, descr, sizeof(descr), 0, 0);

			WideCharToMultiByte(CP_ACP, 0, icinfo.szName, -1, name, sizeof(name), 0, 0);

			string compressorName;

			compressorName = string(name) + " '" + toString(bpp) + "' " + string(descr);

			if (hic) {

				if (ICCompressQuery(hic, &inFmt, NULL) != ICERR_OK) {

					ICClose(hic);

					continue; // Skip this compressor if it can't handle the format.

				}

				if (toWideString(compressorName) == codecName) {

					found = true;

					break;

				}

				ICClose(hic);

			}

		}

		if (found)

			break;

	}

	return hic;

}

//-----------------------------------------------------------------------------

bool canWork(const HIC &hic, const TDimension &resolution, int bpp)

{

	int lx = resolution.lx, ly = resolution.ly;

	BITMAPINFO bi;

	bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

	bi.bmiHeader.biPlanes = 1;

	bi.bmiHeader.biCompression = BI_RGB;

	bi.bmiHeader.biXPelsPerMeter = 80;

	bi.bmiHeader.biYPelsPerMeter = 72;

	bi.bmiHeader.biClrUsed = 0;

	bi.bmiHeader.biClrImportant = 0;

	bi.bmiHeader.biBitCount = bpp;

	bi.bmiHeader.biWidth = lx;

	bi.bmiHeader.biHeight = ly;

	return ICERR_OK == ICCompressQuery(hic, &bi.bmiHeader, NULL);

}

}

//-----------------------------------------------------------------------------

void AviCodecRestrictions::getRestrictions(const wstring &codecName, QString &restrictions)

{

	restrictions.clear();

	if (codecName == L"Uncompressed") {

		restrictions = QObject::tr("No restrictions for uncompressed avi video");

		return;

	}

	//find the codec

	int bpp;

	HIC hic = getCodec(codecName, bpp);

	if (!hic) {

		restrictions = QObject::tr("It is not possible to communicate with the codec.\n Probably the codec cannot work correctly.");

		return;

	}

	BITMAPINFO bi;

	bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

	bi.bmiHeader.biPlanes = 1;

	bi.bmiHeader.biCompression = BI_RGB;

	bi.bmiHeader.biXPelsPerMeter = 80;

	bi.bmiHeader.biYPelsPerMeter = 72;

	bi.bmiHeader.biClrUsed = 0;

	bi.bmiHeader.biClrImportant = 0;

	int lx = 640, ly = 480;

	bi.bmiHeader.biWidth = lx;

	bi.bmiHeader.biHeight = ly;

	// Loop until we can find a width, height, and depth that works!

	int i;

	// check the x lenght

	bi.bmiHeader.biBitCount = bpp;

	for (i = 3; i >= 0; i--) {

		bi.bmiHeader.biWidth = lx + (1 << i);

		bi.bmiHeader.biSizeImage = ((bi.bmiHeader.biWidth * bi.bmiHeader.biBitCount + 31) / 32) * 4 * ly;

		if (ICERR_OK != ICCompressQuery(hic, &bi.bmiHeader, NULL))

			break;

	}

	if (i >= 0)

		restrictions = QObject::tr("video width must be a multiple of %1").arg(QString::number(1 << (i + 1)));

	// check the y lenght

	bi.bmiHeader.biWidth = 640;

	for (i = 3; i >= 0; i--) {

		bi.bmiHeader.biHeight = ly + (1 << i);

		bi.bmiHeader.biSizeImage = ((lx * bi.bmiHeader.biBitCount + 31) / 32) * 4 * bi.bmiHeader.biHeight;

		if (ICERR_OK != ICCompressQuery(hic, &bi.bmiHeader, NULL))

			break;

	}

	if (i >= 0)

		restrictions = restrictions + "\n" + QObject::tr("video lenght must be a multiple of %1").arg(QString::number(1 << (i + 1)));

	ICClose(hic);

	if (restrictions.isEmpty())

		restrictions = QObject::tr("No restrictions for this codec");

	else

		restrictions.prepend(QObject::tr("Resolution restrictions:") + "\n");

}

//-----------------------------------------------------------------------------

bool AviCodecRestrictions::canWriteMovie(const wstring &codecName, const TDimension &resolution)

{

	if (codecName == L"Uncompressed") {

		return true;

	}

	//find the codec

	int bpp;

	HIC hic = getCodec(codecName, bpp);

	if (!hic)

		return false;

	bool test = canWork(hic, resolution, bpp);

	ICClose(hic);

	return test;

}

//-----------------------------------------------------------------------------

bool AviCodecRestrictions::canBeConfigured(const wstring &codecName)

{

	if (codecName == L"Uncompressed")

		return false;

	//find the codec

	int bpp;

	HIC hic = getCodec(codecName, bpp);

	if (!hic)

		return false;

	bool test = ICQueryConfigure(hic);

	ICClose(hic);

	return test;

}

//-----------------------------------------------------------------------------

void AviCodecRestrictions::openConfiguration(const wstring &codecName, void *winId)

{

	if (codecName == L"Uncompressed")

		return;

	//find the codec

	int bpp;

	HIC hic = getCodec(codecName, bpp);

	if (!hic)

		return;

	ICConfigure(hic, winId);

	ICClose(hic);

}

//-----------------------------------------------------------------------------

QMap<wstring, bool=""> AviCodecRestrictions::getUsableCodecs(const TDimension &resolution)</wstring,>

{

	QMap<wstring, bool=""> codecs;</wstring,>

	HIC hic = 0;

	ICINFO icinfo;

	memset(&icinfo, 0, sizeof(ICINFO));

	char descr[2048], name[2048];

	DWORD fccType = 0;

	BITMAPINFO inFmt;

	memset(&inFmt, 0, sizeof(BITMAPINFO));

	inFmt.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

	inFmt.bmiHeader.biWidth = inFmt.bmiHeader.biHeight = 100;

	inFmt.bmiHeader.biPlanes = 1;

	inFmt.bmiHeader.biCompression = BI_RGB;

	int bpp;

	for (bpp = 32; (bpp >= 24); bpp -= 8) {

		//find the codec.

		inFmt.bmiHeader.biBitCount = bpp;

		for (int i = 0; ICInfo(fccType, i, &icinfo); i++) {

			hic = ICOpen(icinfo.fccType, icinfo.fccHandler, ICMODE_COMPRESS);

			ICGetInfo(hic, &icinfo, sizeof(ICINFO)); // Find out the compressor name

			WideCharToMultiByte(CP_ACP, 0, icinfo.szDescription, -1, descr, sizeof(descr), 0, 0);

			WideCharToMultiByte(CP_ACP, 0, icinfo.szName, -1, name, sizeof(name), 0, 0);

			wstring compressorName;

			compressorName = toWideString(string(name) + " '" + toString(bpp) + "' " + string(descr));

			if (hic) {

				if (ICCompressQuery(hic, &inFmt, NULL) != ICERR_OK) {

					ICClose(hic);

					continue; // Skip this compressor if it can't handle the format.

				}

				codecs[compressorName] = canWork(hic, resolution, bpp);

				ICClose(hic);

			}

		}

	}

	return codecs;

}