nulldc-360/nullDC/emitter/regalloc/ppc_fpregalloc.cpp

/*
	This is actually not used. Never was
*/
#include "ppc_fpregalloc.h"
#include <assert.h>

#define REG_ALLOC_COUNT (16)
//FR0 is reserved for math/temp
ppc_fpr_reg reg_to_alloc_fpr[16]=
{
	FR16,FR17,FR18,FR19,FR20,FR21,FR22,FR23,FR24,FR25,FR26,FR27,FR28,FR29,FR30,FR31
};
u32 alb=0;
u32 nalb=0;

struct fprinfo
{
	ppc_fpr_reg reg;
	bool Loaded;
	bool WritenBack;
};

class SimpleFPRRegAlloc:public FloatRegAllocator
{
	struct sort_temp
	{
		int cnt;
		int reg;
		bool no_load;
	};

	//ebx, ebp, esi, and edi are preserved

	//Yay bubble sort
	void bubble_sort(sort_temp numbers[] , int array_size)
	{
		int i, j;
		sort_temp temp;
		for (i = (array_size - 1); i >= 0; i--)
		{
			for (j = 1; j <= i; j++)
			{
				if (numbers[j-1].cnt < numbers[j].cnt)
				{
					temp = numbers[j-1];
					numbers[j-1] = numbers[j];
					numbers[j] = temp;
				}
			}
		}
	}



	ppc_block* ppce;
	fprinfo reginf[16];
	

	fprinfo* GetInfo(u32 reg)
	{
		reg-=fr_0;
		if (reg<16)
		{
			if (reginf[reg].reg!=ERROR_REG)
			{
				return &reginf[reg];
			}
		}
		return 0;
	}
	void ensure_valid(u32 reg)
	{
		if (reg>=fr_0 && reg<=fr_15)
			__noop;
		else
			assert(false); 
	}
	bool DoAlloc;
	
	//methods needed
	//
	//DoAllocation		: do allocation on the block
	virtual void DoAllocation(BasicBlock* block,ppc_block* ppce)
	{
		this->ppce=ppce;

		for (int i=0;i<16;i++)
		{
			reginf[i].reg=ERROR_REG;
			reginf[i].Loaded=false;
			reginf[i].WritenBack=false;

			if (i<REG_ALLOC_COUNT)
			{
				reginf[i].Loaded=true;
				reginf[i].reg=reg_to_alloc_fpr[i];
			}
		}
	}
	//BeforeEmit		: generate any code needed before the main emittion begins (other register allocators may have emited code tho)
	virtual void BeforeEmit()
	{
	}
	//BeforeTrail		: generate any code needed after the main emittion has ended (other register allocators may emit code after that tho)
	virtual void BeforeTrail()
	{
		for (int i=0;i<16;i++)
		{
			if (IsRegAllocated(i+fr_0))
			{
				GetRegister(FR0,i+fr_0,RA_DEFAULT);
			}
		}
	}
	//AfterTrail		: generate code after the native block end (after the ret) , can be used to emit helper functions (other register allocators may emit code after that tho)
	virtual void AfterTrail()
	{

	}
	//IsRegAllocated	: *couh* yea .. :P
	virtual bool IsRegAllocated(u32 sh4_reg)
	{
		return GetInfo(sh4_reg)!=0;
	}
	//Carefull w/ register state , we may need to implement state push/pop
	//GetRegister		: Get the register , needs flag for mode
	virtual ppc_fpr_reg GetRegister(ppc_fpr_reg d_reg,u32 reg,u32 mode)
	{
		ensure_valid(reg);
        
        ppce->vectorWriteBack(reg);
        
		if (IsRegAllocated(reg))
		{
			fprinfo* r1=  GetInfo(reg);
			if (r1->Loaded==false)
			{
				if ((mode & RA_NODATA)==0)
				{
					u32 * rp = (u32*) GetRegPtr(reg);
					ppce->emitLoadFloat(r1->reg,rp);
					r1->WritenBack=true;//data on reg is same w/ data on mem
				}
				else
					r1->WritenBack=false;//data on reg is not same w/ data on mem
			}
			
			//we reg is now on sse reg :)
			r1->Loaded=true;

			if (mode & RA_FORCE)
			{
				if ((mode & RA_NODATA)==0)
					EMIT_FMR(ppce,d_reg,r1->reg);
				return d_reg;
			}
			else
			{
				return r1->reg;
			}

		}
		else
		{
			if ((mode & RA_NODATA)==0){
				u32 * rp = (u32*) GetRegPtr(reg);
				ppce->emitLoadFloat(d_reg,rp);
			}
			return d_reg;
		}
//		__debugbreak(); 
		//return XMM_Error;
	}
	//Save registers
	virtual void SaveRegister(u32 reg,ppc_fpr_reg from)
	{
		ensure_valid(reg);
		if (IsRegAllocated(reg))
		{
			fprinfo* r1=  GetInfo(reg);
			r1->Loaded=true;
			r1->WritenBack=false;
			
			if (r1->reg!=from)
				EMIT_FMR(ppce,r1->reg,from);
		}
		else
		{
			u32 * rp = (u32*) GetRegPtr(reg);
			ppce->emitStoreFloat(rp,from);			
		}
	
        ppce->vectorReload(reg);
    }
	
	virtual void SaveRegister(u32 reg,float* from)
	{
		ensure_valid(reg);
		if (IsRegAllocated(reg))
		{
			fprinfo* r1=  GetInfo(reg);
			r1->Loaded=true;
			r1->WritenBack=false;
	
			u32 * rp = (u32*) from;
			ppce->emitLoadFloat(r1->reg,rp);
		}
		else
		{
			u32 * rp = (u32*) from;
			ppce->emitLoadFloat(FR0,rp);
			rp = (u32*) GetRegPtr(reg);
			ppce->emitStoreFloat(rp,FR0);
		}
	
        ppce->vectorReload(reg);
	}
	//FlushRegister		: write reg to reg location , and reload it on next use that needs reloading
	virtual void FlushRegister(u32 reg)
	{
		ensure_valid(reg);
		if (IsRegAllocated(reg))
		{
			WriteBackRegister(reg);
            ppce->vectorWriteBack(reg);
			ReloadRegister(reg);
            ppce->vectorReload(reg);
		}
	}
	virtual void FlushRegister_xmm(ppc_fpr_reg reg)
	{
		for (int i=0;i<16;i++)
		{
			fprinfo* r1=  GetInfo(fr_0+i);
			if (r1!=0 && r1->reg==reg)
			{
				FlushRegister(fr_0+i);
			}
		}
	}
	virtual void FlushRegCache()
	{
		for (int i=0;i<16;i++)
			FlushRegister(fr_0+i);
	}
	//WriteBackRegister	: write reg to reg location
	virtual void WriteBackRegister(u32 reg)
	{
		ensure_valid(reg);
		if (IsRegAllocated(reg))
		{
			fprinfo* r1=  GetInfo(reg);
			if (r1->Loaded)
			{
				if (r1->WritenBack==false)
				{
					u32 * rp = (u32*) GetRegPtr(reg);
					ppce->emitStoreFloat(rp,r1->reg);					
					r1->WritenBack=true;
				}
			}
		}
	}
	//ReloadRegister	: read reg from reg location , discard old result
	virtual void ReloadRegister(u32 reg)
	{
		ensure_valid(reg);
		if (IsRegAllocated(reg))
		{
			fprinfo* r1=  GetInfo(reg);
			r1->Loaded=false;
		}
	}
	virtual void SaveRegisterGPR(u32 to,ppc_gpr_reg from)
	{
		bool ra=IsRegAllocated(to);

		u32 * rp = (u32*) GetRegPtr(to);
		ppce->emitStore32(rp,from);
		
		if (ra) ReloadRegister(to);
	}
	virtual void LoadRegisterGPR(ppc_gpr_reg to,u32 from)
	{
		if (IsRegAllocated(from)) WriteBackRegister(from);
		
		u32 * rp = (u32*) GetRegPtr(from);
		ppce->emitLoad32(to,rp);
	}
};

FloatRegAllocator * GetFloatAllocator()
{
	return new SimpleFPRRegAlloc();
}