unit.cpp

#include "stdafx.h"
#include "unit.h"
#include <math.h>
#include "player.h"

void (__cdecl *unitSetFollow)(bigUnitStruct* Me,void *Him);
void (__cdecl *unitBecomePet)(bigUnitStruct* Me,void *Him);
void (__cdecl *noxFrozen) (bigUnitStruct* Unit,int Num);
void (__cdecl *noxUnFrozen) (bigUnitStruct* Unit,int Num);
void (__cdecl *unitMove)(bigUnitStruct *Who,noxPoint *Pt);
void (__cdecl *unitActivate) (bigUnitStruct *Unit);
void (__cdecl *dropAllItems)(bigUnitStruct*Unit);
void (__cdecl *noxUnitHunt) (bigUnitStruct *Unit);
void (__cdecl *noxAgressionLevel) (bigUnitStruct *Unit,float *level);
void (__cdecl *noxMirrorShot) (bigUnitStruct *Shot,void *);
int (__cdecl *noxUnitTestBuff) (bigUnitStruct *Unit,int buffN);

void (__cdecl *inventoryPut)(bigUnitStruct *Who,bigUnitStruct *What,int A);

/// ñïèñîê ñîçäàííûõ â òåêóùåì êàäðå íà ñåðâåðå
bigUnitStruct **unitCreatedList;

extern const char *(__cdecl *noxThingNameByType)(int Type);
extern int (__cdecl *noxGetUnitsInRect)(FloatRect *Rect, void (__cdecl *SomeFn)(void *Unit, void *Arg), void *Arg);

struct UnitAndEye_s
{
	float unitX,unitY;
	float unitViewX,unitViewY;
};

int (__cdecl *noxUnknown535250)(UnitAndEye_s *Ptr,int Unk1,int Unk2,int Unk3);//bool
void (__cdecl *unitSetDecayTime)(void *Unit,int Time);//bool
void (__cdecl *noxDeleteUnit)(void *Unit);

void* (__cdecl *noxUnitDefByName)(const char *Name);//Âîçâðàùàåò þíèòäåô

void (__cdecl* noxCreateAt)(unitBigStructPtr Obj,unitBigStructPtr ParentUnit, float X,float Y);

namespace
{

	int getUnitCoordL(lua_State *L)
	{
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *Unit=(bigUnitStruct*)lua_touserdata(L,1);
		if(Unit==NULL)
			return 0;
		lua_pushnumber(L,Unit->unitX);
		lua_pushnumber(L,Unit->unitY);
		return 2;
	}
	void unitPickImpl(void *Unit, void *L_)
	{
		void **LL=(void **)L_;
		*LL=Unit;
	}
	int unitPickL(lua_State *L)
	{
		if (
			(lua_type(L,1)!=LUA_TNUMBER)||
			(lua_type(L,2)!=LUA_TNUMBER)||
			(lua_type(L,3)!=LUA_TNUMBER) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		void *P=0;
		float x=lua_tonumber(L,1),y=lua_tonumber(L,2),d=lua_tonumber(L,3);
		if (d<=0)
			d=1;
		noxGetUnitsInRect(
				&FloatRect(x-d,y-d,x+d,y+d)
				,&unitPickImpl,&P);
		if (P)
			lua_pushlightuserdata(L,P);
		else
			lua_pushnil(L);
		return 1;
	}
	int createObjectL(lua_State *L)
	{
		if (
			(lua_type(L,1)!=LUA_TSTRING)||
			(lua_type(L,2)!=LUA_TNUMBER)||
			(lua_type(L,3)!=LUA_TNUMBER) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *T=
			(bigUnitStruct *)objectCreateByName(lua_tostring(L,1));
		if (!T)
			return 0;
		noxCreateAt(T,0,lua_tonumber(L,2),lua_tonumber(L,3));
		if (T->Class  & clImmobile)
		{
			BYTE Buf[256],*Out=Buf;
			int Size=16;
			netUniPacket(upNewStatic,Out,Size);
			memcpy(Out,&T->thingType,4);Out+=4;
			memcpy(Out,&T->unitX,4);Out+=4;
			memcpy(Out,&T->unitY,4);Out+=4;
			memcpy(Out,&T->netCode,4);Out+=4;
			netSendAll(Buf,Out-Buf);
		}
		lua_pushlightuserdata(L,T);
		return 1;
	}
	int createObjectIn(lua_State *L)
	{
		if (
			(lua_type(L,1)!=LUA_TSTRING)||
			(lua_type(L,2)!=LUA_TLIGHTUSERDATA)
			)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *T=
			(bigUnitStruct *)objectCreateByName(lua_tostring(L,1));
		if (!T)
			return 0;
		inventoryPut((bigUnitStruct*)lua_touserdata(L,2),T,1);
		lua_pushlightuserdata(L,T);
		return 1;
	}
	int deleteUnitL(lua_State *L)
	{
		if ((lua_type(L,1)!=LUA_TLIGHTUSERDATA) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *P=(bigUnitStruct*)lua_touserdata(L,1);
		if(P!=0)
		{
			if (P->Class & clImmobile)
			{
				BYTE Buf[256],*Out=Buf;
				int Size=4;
				netUniPacket(upDelStatic,Out,Size);
				memcpy(Out,&P->netCode,4);Out+=4;
				netSendAll(Buf,Out-Buf);
			}
			noxDeleteObject(P);
		}
		return 0;
	}
	int unitSetFollowL(lua_State*L)
	{
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA)||(lua_type(L,2)!=LUA_TLIGHTUSERDATA) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *P1,*P2;
		P1=(bigUnitStruct*)lua_touserdata(L,1);P2=(bigUnitStruct*)lua_touserdata(L,2);
		if(P1==0 || P2==0)
			return 0;
		unitSetFollow(P1,P2);
		return 0;
	}
	int unitBecomePetL(lua_State*L)
	{
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA)||(lua_type(L,2)!=LUA_TLIGHTUSERDATA) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *P1,*P2;
		P1=(bigUnitStruct*)lua_touserdata(L,1);P2=(bigUnitStruct*)lua_touserdata(L,2);
		if(P1==0 || P2==0)
			return 0;
		unitBecomePet(P2,P1);/// Âûçûâàåòñÿ íàîáîðîò
		return 0;
	}

	int unitHuntL(lua_State*L)
	{
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		noxUnitHunt((bigUnitStruct*)lua_touserdata(L,1));
		return 0;
	}

	int unitCopyUcL(lua_State*L)
	{
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		BYTE *P=(BYTE*)lua_touserdata(L,1);
		int TT=*((WORD*)(P+0x4));
		if(TT==0)
			return 0;
//		char D[100];
///		wsprintf(D,"tt:%x",TT);
//		printI(D);
		const char*Name=noxThingNameByType(TT);
//		wsprintf(D,"tname:%s",Name);
//		printI(D);
//		return 0;
		BYTE *UnitDef=(BYTE*)noxUnitDefByName(Name);
		int DataSize=*((int*)(UnitDef+0xC4));
		void *Data=noxAlloc(DataSize);
		memcpy(Data,*((void**)(P+0x2EC)),DataSize);
		lua_pushlightuserdata(L,Data);
		return 1;
	}
	int unitDecayL(lua_State*L)
	{	/// äîáàâèòü â ñïèñîê ðàñïàäà
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA) ||
			 (lua_type(L,2)!=LUA_TNUMBER) )
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		unitSetDecayTime(lua_touserdata(L,1),lua_tointeger(L,2));
		return 0;
	}
	int unitMoveL(lua_State*L)
	{	/// äîáàâèòü â ñïèñîê ðàñïàäà
		if ( (lua_type(L,1)!=LUA_TLIGHTUSERDATA) ||
			 (lua_type(L,2)!=LUA_TNUMBER) ||
			 (lua_type(L,3)!=LUA_TNUMBER) ) 
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		noxPoint Pt(lua_tonumber(L,2),lua_tonumber(L,3));
		unitMove((bigUnitStruct*)lua_touserdata(L,1),&Pt);
		return 0;
	}
	
	int unitFreezL(lua_State*L)
	{
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		noxFrozen((bigUnitStruct*)lua_touserdata(L,1),1);
		return 1;
	}

	int unitAgressionLevel(lua_State*L)
	{
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA || lua_type(L,2)!=LUA_TNUMBER)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		float lev=lua_tonumber(L,2);
		if (lev<0 || lev>1)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		noxAgressionLevel((bigUnitStruct*)lua_touserdata(L,1),&lev);
		return 0;
	}

	int unitUnFreezL(lua_State*L)
	{
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		noxUnFrozen((bigUnitStruct*)lua_touserdata(L,1),1);
		return 1;
	}
	int unitDropAll(lua_State*L)
	{
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		dropAllItems((bigUnitStruct*)lua_touserdata(L,1));
		return 1;
	}
	int unitInventoryPut(lua_State*L)
	{
		if (
			(lua_type(L,1)!=LUA_TLIGHTUSERDATA)||
			(lua_type(L,2)!=LUA_TLIGHTUSERDATA)
			)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		inventoryPut((bigUnitStruct*)lua_touserdata(L,1),(bigUnitStruct*)lua_touserdata(L,2),1);
		return 0;
	}
	int unitSpeedL(lua_State*L)
	{	
		if (lua_type(L,1)!=LUA_TLIGHTUSERDATA)
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		bigUnitStruct *P=(bigUnitStruct*)lua_touserdata(L,1);		
		lua_pushnumber(L,P->velX);
		lua_pushnumber(L,P->velY);
		if (lua_gettop(L)<5)
		{
			return 2;
		}
		if( (lua_type(L,2)!=LUA_TNUMBER) ||
			(lua_type(L,3)!=LUA_TNUMBER) ) 
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		if((lua_gettop(L)>5)&& lua_toboolean(L,4))
		{
			float a=lua_tonumber(L,3),v=lua_tonumber(L,2);
			P->velX=v*a;
			P->velY=-v*a;
		}
		else
		{
			P->velX=lua_tonumber(L,2);
			P->velY=lua_tonumber(L,3);
		}
		unitActivate(P);
		return 2;
	}

	int unitTestBuff(lua_State*L)
	{
		if ((lua_type(L,1)!=LUA_TLIGHTUSERDATA)||
			(lua_type(L,2)!=LUA_TNUMBER))
		{
			lua_pushstring(L,"wrong args!");
			lua_error_(L);
		}
		if (noxUnitTestBuff((bigUnitStruct*)lua_touserdata(L,1),lua_tonumber(L,2))==1)
			lua_pushboolean(L,true);
		else
			lua_pushboolean(L,false);
		return 1;
	}
}
float (__cdecl *getFloatByName)(const char *Name);
void unitInit()
{
	ASSIGN(getFloatByName,0x00419D40);
	ASSIGN(unitCreatedList,0x00750710);
	ASSIGN(unitSetFollow,0x5158C0);
	ASSIGN(unitBecomePet,0x4E7B00);
	ASSIGN(noxUnknown535250,0x535250);
	ASSIGN(unitSetDecayTime,0x00511660);
	ASSIGN(noxUnitDefByName,0x004E3830);
	ASSIGN(noxCreateAt,0x004DAA50);
	ASSIGN(unitMove,0x004E7010);
	ASSIGN(unitActivate,0x00537610);
	ASSIGN(noxFrozen,0x004E79C0);
	ASSIGN(noxUnFrozen,0x004E7A60);
	ASSIGN(inventoryPut,0x004F3070);
	ASSIGN(dropAllItems,0x004EDA40);
	ASSIGN(noxUnitHunt,0x5449D0); 
	ASSIGN(noxAgressionLevel,0x00515980);
	ASSIGN(noxMirrorShot,0x004E0A70);
	ASSIGN(noxUnitTestBuff,0x004FF350);

	registerserver("unitSetFollow",&unitSetFollowL);
	registerserver("unitBecomePet",&unitBecomePetL);
	registerserver("unitDecay",&unitDecayL);
	registerserver("unitDelete",&deleteUnitL);
	registerserver("unitCopyUc",&unitCopyUcL);
	registerserver("createObject",&createObjectL);
	registerserver("unitPos",&getUnitCoordL);
	registerserver("unitFreeze",&unitFreezL);
	registerserver("unitUnFreeze",&unitUnFreezL);
	registerserver("unitPick",&unitPickL);
	registerserver("unitHunt",&unitHuntL);
	registerserver("unitSetAgression",&unitAgressionLevel);
	registerserver("unitTestBuff",&unitTestBuff);

	registerserver("unitMove",&unitMoveL);
	registerserver("unitSpeed",&unitSpeedL);
	registerserver("unitDropAll",&unitDropAll);
	registerserver("unitInventoryPut",&unitInventoryPut);
	registerserver("createObjectIn",&createObjectIn);
}