/*************************/
/*** Actor Sprite Tool ***/
/*************************/

#include "stdio.h"
#include <misc.hpp>
#include <conio.h>
#include <iostream.h>
#include <vector>
#include <DaveLib.h>
#include <io.h>
#include <SprSet.h>
#include "MkActor.h"

#include	<pak.h>

using namespace std;
//#define	OutputTGA

// This has got really messy :o(
 
//***************************************************************************
vector<CMkActor>	ActorList;
int			TPBase=-1,TPWidth=-1,TPHeight=-1;
int			DupThresh;
GString		TPOutStr;

GString		RootPath;
GString		SpritePath;
GString		OutPath;
GString		IncludePath;

float		Aspect=512.0f/320.0f;

//***************************************************************************
char * CycleCommands(char *String,int Num)
{
char	Text[2048],*TextPtr;
int		Count;


	if (String[0]=='-' || String[0]=='/')
		{
		GString		TpStr;
		switch (String[1])
			{
				// Switches 
			case 'o':
				OutPath = CheckFileString(String);
				OutPath.Upper();
				OutPath.Append('\\');
				break;
			case 'i':
				IncludePath = CheckFileString(String);
				IncludePath.Upper();
				IncludePath.Append('\\');
				break;
			case 'd':
				DebugOn =true;
				break;
			case 's':
				SpritePath=CheckFileString(String);
				SpritePath.Upper();
				SpritePath.Append('\\');
				break;
			case 'r':
				RootPath=CheckFileString(String);
				RootPath.Upper();
				RootPath.Append('\\');
				break;
			case 'a':
				TpStr= CheckFileString(String);
				Aspect=(float)atof(TpStr);
				break;
			case 'c':
				TpStr= CheckFileString(String);
				DupThresh=(int)atol(TpStr);
				break;
			case 't':
				TpStr= CheckFileString(String);
				TextPtr=Text;
				strcpy(TextPtr,TpStr);
				Count=ZeroAndCountCommas(TextPtr);
				if (Count!=2) GObject::Error(ERR_FATAL,"Problem with option %s\n",String);
				TPBase=atol(TextPtr);
				TextPtr+=strlen(TextPtr)+1;
				TPWidth=atol(TextPtr);
				TextPtr+=strlen(TextPtr)+1;
				TPHeight=atol(TextPtr);
				break;
			default:
				GObject::Error(ERR_FATAL,"Unknown switch %s",String);
				break;
			}
		} 
	else
		{
			GString	Name=String;
			Name.Upper();
			ActorList.push_back(CMkActor(Name,RootPath,SpritePath));
		}

	return(String);
}

//***************************************************************************

void Usage(char *ErrStr)
{
	printf("\nMkActor (usese scripts): by Dave\n");
	printf("Usage: MkActor <file> [ <file>.. ] [ switches.. ]\n");
	printf("Switches:\n");
	printf("   -o:[FILE]       Set output Dir\n");
	printf("   -p:nn           Root Dir\n");
	printf("   -s:nn           Sprite Dir\n");
	printf("   -t:p,w,h        Set TPage No,Width,Height\n");
	printf("   -d:             Enable Debug output\n");
	printf("   -a:             Aspect Ratio\n");
	GObject::Error(ERR_FATAL,ErrStr);
}

//***************************************************************************
int	main (int argc, char *argv[])
{
int		i,ListSize;

		CommandLine(argc,argv,CycleCommands);
		if (OutPath.Empty()) Usage("No Output File Set\n");

		ListSize=ActorList.size();
		for (i=0; i<ListSize; i++)
		{
			ActorList[i].Load();
			ActorList[i].Process();
			ActorList[i].Write();
			ActorList[i].CleanUp();
		}

		return 0;
}

//***************************************************************************
//***************************************************************************
//***************************************************************************
CMkActor::CMkActor(GString &ActorName,GString &ActorPath,GString &SpritePath)
{
		Name=ActorName;
		ActorDir=ActorPath+ActorName;
		ActorDir.Append('\\');
		SpriteDir=ActorDir+SpritePath;
		OutFile=OutPath+ActorName;
		IncFile=IncludePath+"ACTOR_"+Name;
		
		TexGrab.SetTPage(TPBase,TPWidth,TPHeight);
		TexGrab.SetOutFile(OutFile+".Tex");
		TexGrab.SetDebugOut(OutFile+".Lbm");
		TexGrab.SetIncFile(IncFile+"_TEX.h");
		TexGrab.NoSort();
		TexGrab.AnimatedHeadersOnly(true);
		TexGrab.DontOutputBoxes(true);
		TexGrab.AllowRotate(false);

		DupCount=0;
		MaxW=0;
		MaxH=0;
		BlankCount=0;

}

//***************************************************************************
void	CMkActor::CleanUp()
{
int		i,ListSize=BmpList.size();
	
		for (i=0; i<ListSize; i++)
		{
			if (BmpList[i].RGB) free(BmpList[i].RGB);
			if (BmpList[i].Pak) free(BmpList[i].Pak);
			if (BmpList[i].Psx) free(BmpList[i].Psx);
		}
}


//***************************************************************************
void	CMkActor::Load()
{
		ReadScript(ActorDir+"AnimList.Txt",InAnimList);
}

//***************************************************************************
void	CMkActor::ReadScript(const char *Filename,vector<GString> &List)
{
char	*Script,*Ptr;
int		Size;

		File=fopen(Filename,"rb");
		if (!File) return;
		fseek(File,0,SEEK_END);
		Size=ftell(File);
		fseek(File,0,SEEK_SET);
		Script=(char*)malloc(Size+1);
		ASSERT(Script);
		// Load It
		fread(Script,Size,1,File);
		fclose(File);

// Extract Names
		Ptr=Script;
		Script[Size]=0;
		while (*Ptr)
		{
			GString	Str;

			while (*Ptr=='\n' || *Ptr=='\r'|| *Ptr==' ') *Ptr++; // Skip gaff
			if (*Ptr)
			{
				if (*Ptr=='#')
				{ // Skip commented lines
					while (*Ptr!='\n' && *Ptr) Ptr++;
				}
				else
				{ // Read data
					while (*Ptr!=' ' && *Ptr!='\n' && *Ptr!='\r' && *Ptr)
					{
						Str.Append(*Ptr++);
					}
					List.push_back(Str);
				}
			}
		}

		free(Script);
}

//***************************************************************************
void	CMkActor::BuildFrameList()
{
int		i,ListSize=InAnimList.size();
char	AnimName[256];
		AnimList.resize(ListSize);
		for (i=0; i<ListSize; i++)
		{
			sAnim	&ThisAnim=AnimList[i];
			sprintf(AnimName,"%s",InAnimList[i]);

			ThisAnim.Name=&AnimName[0];

			FindFrames(ThisAnim);
		}
}

//***************************************************************************
void	CMkActor::FindFrames(sAnim &ThisAnim)
{
GString		Filename;
_finddata_t Find;
long		FileHandle;
int			Error=0;
GString		AnimName=Name+"_"+ThisAnim.Name;
GString		DirName=AnimName+"\\";
GString		BaseName=DirName+AnimName;

int			CheckLen=strlen(BaseName);
			BaseName.Upper();

			Filename=SpriteDir+BaseName+"*.bmp";

			if( (FileHandle= _findfirst( Filename, &Find)) == -1L )
			{
				GObject::Error(ERR_WARNING,"Cant find Anim %s.\n",ThisAnim.Name);
				return;
			}

			while (Error==0)
			{
				sFrame	NewFrame;
				NewFrame.Filename=DirName+Find.name;
				char	c=NewFrame.Filename[CheckLen];
				if (c>='0' && c<='9')
				{
					ThisAnim.Frames.push_back(NewFrame);
				}

				Error=_findnext( FileHandle, &Find);
			}
			_findclose( FileHandle);
}

//***************************************************************************
void	CMkActor::LoadFrameList()
{
int		i,ListSize=AnimList.size();
		
		for (i=0; i<ListSize; i++)
		{
			sAnim	&ThisAnim=AnimList[i];
			int		FrameCount=ThisAnim.Frames.size();
			for (int f=0; f<FrameCount; f++)
			{
				printf("%s - Load Anim %2d/%2d\tFrame %2d/%2d   \r",Name,i+1,ListSize,f,FrameCount);
				LoadBmp(ThisAnim.Frames[f]);
			}
		}
		printf("\t\t\t\t\t\t\r");
}


//***************************************************************************
void	CMkActor::MakePsxGfx(sBmp &Bmp)
{
// capture blank frames
		if (Bmp.Frm.IsBlank())
		{
			Bmp.Psx=0;
			Bmp.PsxSize=0;
			BlankCount++;
			return;
		}
		
// Copied from SprSet 
int		nfW,nfH,nfLineWidthBytes,nfAreaBytes;
Frame	Frm=Bmp.Frm;

		nfW=Frm.GetWidth();
		nfH=Frm.GetHeight();
		nfLineWidthBytes=GU_AlignVal(nfW,2)/2;
		nfAreaBytes=nfLineWidthBytes*nfH;
		Bmp.PsxSize=nfAreaBytes;

		if (nfW>MaxW) MaxW=nfW;
		if (nfH>MaxH) MaxH=nfH;

		Bmp.Psx=(u8*)malloc(nfAreaBytes+16);
		ASSERT(Bmp.Psx);

		for (int y=0;y<nfH;y++)
		{
			for (int x=0;x<nfW;x++)
			{
				u8 * PixAddr;
				u8  ScrNib;

				ScrNib=Frm.SeeData()[y*nfW+x]&0xf;

				PixAddr=&Bmp.Psx[(x/2)+(nfLineWidthBytes*y)];

				if (PixAddr>=&Bmp.Psx[nfAreaBytes]) ASSERT(!"");
				if ((x&1))
				{
					*PixAddr&=0x0f;
					*PixAddr|=ScrNib<<4;
				}
				else
				{
					*PixAddr&=0xf0;
					*PixAddr|=ScrNib;
				}
			}
		}
}

//***************************************************************************
int		CMkActor::FindDup(sBmp &Bmp)
{
int			i,BmpListSize=BmpList.size();
int			Size=Bmp.Frm.GetWidth()*Bmp.Frm.GetHeight();

			Bmp.RGB=(u8*)malloc(Size*3);
			ASSERT(Bmp.RGB);
			Bmp.Frm.MakeRGB(Bmp.RGB);
// Check for dups (Broken at the mo, ah well)
// Gen Chksum
/*
u8		*RGB=Bmp.RGB;
		Bmp.ChkR=Bmp.ChkG=Bmp.ChkB=0;
		for (i=0; i<Size; i++)
		{
			Bmp.ChkR+=*RGB++;
			Bmp.ChkG+=*RGB++;
			Bmp.ChkB+=*RGB++;
		}
*/
// Find existing
		for (i=0; i<BmpListSize; i++)
		{
			sBmp	&ThisBmp=BmpList[i];
			ASSERT(Bmp.RGB);
			ASSERT(ThisBmp.RGB);
			if (IsImageSame(ThisBmp,Bmp)) 
			{
				free(Bmp.RGB);
				return(i);
			}
		}

		return(-1);
}

//***************************************************************************
bool	CMkActor::IsImageSame(sBmp &Bmp0,sBmp &Bmp1)
{
int		W0,H0,W1,H1,Size;
u8		*RGB0,*RGB1;
int		DiffCount=0;
//		if (Bmp0.ChkR!=Bmp1.ChkR || Bmp0.ChkG!=Bmp1.ChkG || Bmp0.ChkB!=Bmp1.ChkB) return(false);
		W0=Bmp0.Frm.GetWidth();
		H0=Bmp0.Frm.GetHeight();
		W1=Bmp1.Frm.GetWidth();
		H1=Bmp1.Frm.GetHeight();

		if (W0!=W1 || H0!=H1) return(false);
		Size=W0*H0;
		RGB0=Bmp0.RGB;
		RGB1=Bmp1.RGB;
		for (int i=0; i<Size; i++)
		{
//			if (*RGB0++!=*RGB1++) return(false);
			if (*RGB0++!=*RGB1++ ||	// R
				*RGB0++!=*RGB1++ ||	// G
				*RGB0++!=*RGB1++)	// B
			{
				DiffCount++;
			}

		}
		if (DiffCount<=DupThresh)
			return(true);
		else
			return(false);
}

//***************************************************************************
void	CMkActor::CheckAndShrinkFrame(sBmp &Bmp,const char *FrameName)
{
// Check Colors
int		ColorCount=Bmp.Frm.GetNumOfCols();
		if (ColorCount>16)
		{
				GObject::Error(ERR_FATAL,"%s has %i colors.\n",Name,ColorCount);
		}
// Check Palette
		if (BmpList.size())
		{
			Palette	&BasePal=BmpList[0].Frm.GetPal();
			Palette	&ThisPal=Bmp.Frm.GetPal();
			int	Count=__min(BasePal.GetNumOfCols(),ThisPal.GetNumOfCols());

			for (int p=1;p<Count; p++)
			{
				if (BasePal[p].GetR()!=ThisPal[p].GetR() ||
					BasePal[p].GetG()!=ThisPal[p].GetG() ||
					BasePal[p].GetB()!=ThisPal[p].GetB())
				{
					GObject::Error(ERR_WARNING,"Palette MisMatch : %s\n",FrameName);
					break;
				}
			}
		}

		Bmp.OrigW=Bmp.Frm.GetWidth();
		Bmp.OrigH=Bmp.Frm.GetHeight();

Rect	BBox;
		BBox=Bmp.Frm.FindBoundingRect();
		BBox.W=GU_AlignVal(BBox.W,4);
		Bmp.Frm.Crop(BBox);
		Bmp.CrossHairX=BBox.X-(Bmp.OrigW/2);//+)/2;
		Bmp.CrossHairY=-(Bmp.OrigH-BBox.Y);

}

//***************************************************************************
void	CMkActor::LoadBmp(sFrame &ThisFrame)
{
GString	Filename=SpriteDir+ThisFrame.Filename;
int		BmpListSize=BmpList.size();
sBmp	NewBmp;

// Init & Load Frame
		NewBmp.Frm.LoadBMP(Filename);
		NewBmp.RGB=0;
		NewBmp.Pak=0;
		NewBmp.Psx=0;

		CheckAndShrinkFrame(NewBmp,ThisFrame.Filename);
		ThisFrame.XOfs=NewBmp.CrossHairX;
		ThisFrame.YOfs=NewBmp.CrossHairY;

// Check Dups
int		Idx=FindDup(NewBmp);

		if (Idx!=-1)
		{
			DupCount++;
			ThisFrame.FrameIdx=Idx;
			return;
		}
// Its unique, add to list
		BmpList.push_back(NewBmp);

		MakePsxGfx(BmpList[BmpListSize]);

#if	_DEBUG && defined(OutputTGA)
void	WriteTGA(GString Name,Frame Frm);
		WriteTGA(Name,NewBmp.Frm);
#endif

		ThisFrame.FrameIdx=BmpListSize;
}

//***************************************************************************
void	WriteTGA(GString Name,Frame Frm)
{
u8		*TGA=(u8*)malloc(Frm.GetWidth()*Frm.GetHeight()*3);
		ASSERT(TGA);
		Frm.FlipY();
		Frm.MakeRGB(TGA);

		char	OutName[256];
		sprintf(OutName,"\\x\\%s.tga",Name);
		SaveTGA(OutName,Frm.GetWidth(),Frm.GetHeight(),TGA,true);
		free(TGA);
}

//***************************************************************************
//***************************************************************************
//***************************************************************************
void	CMkActor::Process()
{
		BuildFrameList();
		LoadFrameList();
		ProcessFrames();
}

//***************************************************************************
void	CMkActor::ProcessFrames()
{
int		i,ListSize=BmpList.size();

		TotalIn=0;		
		TotalOut=0;		
		for (i=0; i<ListSize; i++)
		{
			sBmp	&ThisBmp=BmpList[i];

			printf("%s - Processing Frame %2d\\%2d\r",Name,i+1,ListSize);
			if (ThisBmp.Psx)
			{
				ThisBmp.PakSize=PAK_findPakSize(ThisBmp.Psx,ThisBmp.PsxSize);
				ThisBmp.Pak=(u8*)malloc(ThisBmp.PakSize);
				ASSERT(ThisBmp.Pak);
				PAK_doPak(ThisBmp.Pak,ThisBmp.Psx,ThisBmp.PsxSize);
			}
			else
			{ // Blank Frame
				ThisBmp.Pak=0;
				ThisBmp.PakSize=0;
			}

			TotalIn+=ThisBmp.PsxSize;
			TotalOut+=ThisBmp.PakSize;
		}
		printf("\t\t\t\t\t\r");
}

//***************************************************************************
//***************************************************************************
//***************************************************************************
void	CMkActor::Write()
{
GString			OutName=OutFile+".SBK";

		File=fopen(OutName,"wb");

// Write Dummy Hdr
		fwrite(&FileHdr,1,sizeof(sSpriteAnimBank),File);
		PadFile(File);
// Write Palette
		FileHdr.Palette=(u8*)WritePalette();
		FileHdr.Clut=0;
		PadFile(File);
// Write AnimList
		FileHdr.AnimList=(sSpriteAnim*)WriteAnimList();
		PadFile(File);
// Write FrameList
		FileHdr.FrameList=(sSpriteFrameGfx*)WriteFrameList();
//		PadFile(File);

// Rewrite Header
		fseek(File, 0, SEEK_SET);
		fwrite(&FileHdr,1,sizeof(sSpriteAnimBank),File);
		fclose(File);

		WriteInclude();
		printf("%s. D=%i B=%i - WH=%i %i - In=%i Out=%i Sv=%i \n",Name,DupCount,BlankCount,FileHdr.MaxW,FileHdr.MaxH,TotalIn,TotalOut,TotalIn-TotalOut);

}

//***************************************************************************
int		CMkActor::WritePalette()
{
int			Pos=ftell(File);
Palette		&PCPalette=BmpList[0].Frm.GetPal();
SprPal		PsxPalette;
int			i,Count;
vector<u16>	OutPal;

		PsxPalette=PCPalette;
		PsxPalette.SetPlusColZero(true);

		Count=PsxPalette.GetNumOfCols();
		if (Count<16) Count=16;
		PsxPalette.MakePSXPal(OutPal);
		for (i=0; i<Count; i++)
		{
			u16	ThisCol=OutPal[i];
			fwrite(&ThisCol,1,sizeof(u16),File);
		}
		return(Pos);
}

//***************************************************************************
int		CMkActor::WriteAnimList()
{
int		Pos=ftell(File);
int		i,AnimCount=AnimList.size();
vector<sSpriteAnim>	Hdrs;

// Write Dummy Hdrs
		FileHdr.AnimCount=AnimCount;
		Hdrs.resize(AnimCount);
		for (i=0; i<AnimCount; i++)
		{
			fwrite(&Hdrs[i],1,sizeof(sSpriteAnim),File);
		}
		PadFile(File);
		
// Write Frame Lists
		for (i=0; i<AnimCount; i++)
		{
			sAnim	&ThisAnim=AnimList[i];
			int		f,FrameCount=ThisAnim.Frames.size();

			Hdrs[i].FrameCount=FrameCount;
			Hdrs[i].Anim=(sSpriteFrame*)ftell(File);

			for (f=0; f<FrameCount; f++)
			{
				sFrame			&ThisFrame=ThisAnim.Frames[f];
				sSpriteFrame	OutFrame;
				
				OutFrame.FrameIdx=ThisFrame.FrameIdx;
				OutFrame.XOfs=ThisFrame.XOfs;
				OutFrame.YOfs=ThisFrame.YOfs;
				fwrite(&OutFrame,1,sizeof(sSpriteFrame),File);
			}
			PadFile(File);
			
		}

// ReWrite Headers
int		SavePos=ftell(File);
		fseek(File,Pos,SEEK_SET);
		for (i=0; i<AnimCount; i++)
		{
			fwrite(&Hdrs[i],1,sizeof(sSpriteAnim),File);
		}
		fseek(File,SavePos,SEEK_SET);

		return(Pos);
}

//***************************************************************************
int		CMkActor::WriteFrameList()
{
int		Pos=ftell(File);
int		i,FrameCount=BmpList.size();
vector<sSpriteFrameGfx>	Hdrs;

		FileHdr.MaxW=MaxW;
		FileHdr.MaxH=MaxH;
		FileHdr.FrameCount=FrameCount;

// Write Dummy Hdrs
		Hdrs.resize(FrameCount);

		for (i=0; i<FrameCount; i++)
		{
			fwrite(&Hdrs[i],1,sizeof(sSpriteFrameGfx),File);
		}
		PadFile(File);
		
// Write Frame Lists
		for (i=0; i<FrameCount; i++)
		{
			sBmp	&ThisBmp=BmpList[i];

			if (ThisBmp.Pak)
			{
				Hdrs[i].PAKSpr=(u8*)ftell(File);
				Hdrs[i].W=ThisBmp.Frm.GetWidth();
				Hdrs[i].H=ThisBmp.Frm.GetHeight();
				// aspect
				float	aW=Hdrs[i].W*Aspect;
				int		dW=(int)aW-Hdrs[i].W;
				int		X0=dW/2;
				int		X1=dW-X0;

				Hdrs[i].AspectX0=X0;
				Hdrs[i].AspectX1=X1;
//	printf("%i %i %i\n",Hdrs[i].W,X0,X1);
				fwrite(ThisBmp.Pak,1,ThisBmp.PakSize,File);
			}
			else
			{
				Hdrs[i].PAKSpr=0;
				Hdrs[i].W=0;
				Hdrs[i].H=0;
			}
		}
// ReWrite Headers
int		SavePos=ftell(File);
		fseek(File,Pos,SEEK_SET);
		for (i=0; i<FrameCount; i++)
		{
			fwrite(&Hdrs[i],1,sizeof(sSpriteFrameGfx),File);
		}
		fseek(File,SavePos,SEEK_SET);

		return(Pos);
}

//***************************************************************************
void	CMkActor::WriteInclude()
{
GString	Filename=IncFile+"_Anim.h";

		File=fopen(Filename,"wt");

		fprintf(File,"// %s Anim Header\n",Name);
		fprintf(File,"\n");
		fprintf(File,"#ifndef\t__ANIM_%s_HEADER__\n",Name);
		fprintf(File,"#define\t__ANIM_%s_HEADER__\n",Name);
		fprintf(File,"\n");
		fprintf(File,"\n");
		fprintf(File,"enum\tANIM_%s_LIST\n",Name);
		fprintf(File,"{\n");

int		ListSize=AnimList.size();
		for (int i=0; i<ListSize; i++)
		{
			sAnim	&ThisAnim=AnimList[i];
			GString	AnimName=ThisAnim.Name;
			AnimName.Upper();
			fprintf(File,"\tANIM_%s_%s",Name,AnimName);
			if (i==0)
			{
			fprintf(File,"=0");
			}
		fprintf(File,",\n");
		}

		fprintf(File,"\tNUM_ANIM_%s\n",Name);
		fprintf(File,"};\n");
		fprintf(File,"\n");
		fprintf(File,"#endif\n");

		fclose(File);
}