lua_bullets.c

/* Functions for accessing bullet data from Lua.
 *
 * OWNER_SHIP = 1
 *
 * for i, b in ipairs(bullets) do
 *     if b.Used then
 *         if b.OwnerType == OWNER_SHIP and b.Owner == me then
 *             print("Bullet " .. i .. " is mine")
 *         else
 *             print("Bullet " .. i .. " is NOT mine")
 *         end
 *     end
 * end
 *
 * see also primary.h and secondary.h
 */
#include <string.h>
#include <lua.h>
#include <lauxlib.h>
#include "main.h" /* FIXME -- needed for primary.h and secondary.h */
#include "2dtextures.h" /* FIXME -- needed for primary.h and secondary.h */
#include "new3d.h" /* FIXME -- needed for primary.h and secondary.h */
#include "object.h" /* FIXME -- needed for networking.h */
#include "networking.h" /* FIXME -- needed for primary.h and secondary.h */
#include "primary.h"
#include "secondary.h"
#include "lua_vecmat.h"
#include "lua_weapons.h"

extern PRIMARYWEAPONBULLET PrimBulls[MAXPRIMARYWEAPONBULLETS];
extern SECONDARYWEAPONBULLET SecBulls[MAXSECONDARYWEAPONBULLETS];

static void pushprimbull(lua_State *L, u_int16_t index)
{
	int *id = lua_newuserdata(L, sizeof(int));
	*id = (int) index;
	luaL_getmetatable(L, "PRIMARYWEAPONBULLETIDX");
	lua_setmetatable(L, -2);
}

static void pushsecbull(lua_State *L, u_int16_t index)
{
	int *id = lua_newuserdata(L, sizeof(int));
	*id = (int) index;
	luaL_getmetatable(L, "SECONDARYWEAPONBULLETIDX");
	lua_setmetatable(L, -2);
}

static int pushentity(lua_State *L, u_int16_t type, u_int16_t index)
{
	const char *mt;
	int *id;

	switch (type)
	{
		case OWNER_SHIP: mt = "GLOBALSHIPIDX"; break;
		case OWNER_ENEMY: mt = "ENEMYIDX"; break;
		case OWNER_MINE: mt = "SECONDARYWEAPONBULLETIDX"; break;
		default:
			lua_pushnil(L);
			lua_pushliteral(L, "unknown target type");
			return 2;
	}
	id = lua_newuserdata(L, sizeof(int));
	*id = index;
	luaL_getmetatable(L, mt);
	lua_setmetatable(L, -2);
	return 1;
}

#define GETTABLEFORPRIM(index) do { \
	lua_getglobal(L, "weapons");    \
	lua_pushinteger(L, index + 1);  \
	lua_gettable(L, -2);            \
	lua_remove(L, -2);              \
} while (0)

#define FIELD(f, type) do {             \
	if (!strcmp(name, #f))              \
	{                                   \
		lua_push ## type(L, bullet->f); \
		return 1;                       \
	}                                   \
} while (0)
#define FIELDPTR(f, type) do {         \
	if (!strcmp(name, #f))             \
	{                                  \
		lua_push ## type(L, &bullet->f); \
		return 1;                      \
	}                                  \
} while (0)
static int luaprimbull_index(lua_State *L)
{
	PRIMARYWEAPONBULLET *bullet;
	const char *name;
	int *id;
	int bullidx = *((int *) luaL_checkudata(L, 1, "PRIMARYWEAPONBULLETIDX"));
	bullet = &PrimBulls[bullidx];
	name = luaL_checkstring(L, 2);
	/*if (!strcmp(name, "table")) -- TODO
	{
		lua_pushcfunction(L, luaprimbull_table);
		return 1;
	}
	else*/ if (!strcmp(name, "index") || !strcmp(name, "Index"))
	{
		lua_pushinteger(L, bullidx + 1);
		return 1;
	}
	else if (!strcmp(name, "category"))
	{
		lua_pushliteral(L, "primary");
		return 1;
	}
	FIELD(Used, boolean);
	if (!strcmp(name, "Next") && bullet->Next != (u_int16_t) -1)
	{
		pushprimbull(L, bullet->Next);
		return 1;
	}
	else if (!strcmp(name, "Prev") && bullet->Prev != (u_int16_t) -1)
	{
		pushprimbull(L, bullet->Prev);
		return 1;
	}
	else if (!strcmp(name, "Type"))
	{
		GETTABLEFORPRIM(bullet->Type);
		return 1;
	}
	FIELD(OwnerType, integer);
	if (!strcmp(name, "Owner"))
		return pushentity(L, bullet->OwnerType, bullet->Owner);
	/* FIELD(EnemyGun, ??); -- TODO */
	if (!strcmp(name, "Weapon"))
	{
		GETTABLEFORPRIM(bullet->Weapon);
		return 1;
	}
	FIELD(PowerLevel, integer);
	FIELD(TrojPower, number);
	FIELD(LifeCount, number);
	FIELD(Speed, number);
	FIELD(ColRadius, number);
	FIELD(ColType, integer);
	FIELDPTR(Offset, vector);
	FIELDPTR(Pos, vector);
	FIELDPTR(Dir, vector);
	FIELDPTR(LocalDir, vector);
	FIELDPTR(UpVector, vector);
	FIELDPTR(ColStart, vector);
	FIELD(ColDist, number);
	FIELD(ColFlag, integer);
	FIELDPTR(ColPoint, vector); /* VERT */
	FIELDPTR(ColPointNormal, vector); /* NORMAL */
	FIELD(ColGroup, integer);
	FIELD(GroupImIn, integer);
	FIELDPTR(Mat, matrix);
	FIELD(line, integer);
	FIELD(fmpoly, integer);
	FIELD(numfmpolys, integer);
	FIELD(poly, integer);
	FIELD(numpolys, integer);
	FIELD(light, integer);
	FIELD(lightsize, number);
	FIELD(r, number);
	FIELD(g, number);
	FIELD(b, number);
	FIELD(Bounces, integer);
	FIELD(TimeInterval, number);
	FIELD(TimeCount, number);
	FIELD(FirePoint, integer);
	FIELD(SpotFX, integer);
	FIELD(FramelagAddition, number);
	return luaL_argerror(L, 2, "unknown field name");
}

#define GETTABLEFORSEC(index) do {               \
	lua_getglobal(L, "weapons");               \
	lua_pushinteger(L, index + 1 + SECOFFSET); \
	lua_gettable(L, -2);                       \
	lua_remove(L, -2);                         \
} while (0)

#define SECFLAG(name) do {                                 \
	lua_pushboolean(L, bullet->Flags & SECFLAGS_ ## name); \
	lua_setfield(L, -2, #name);                            \
} while (0)

static int luasecbull_index(lua_State *L)
{
	SECONDARYWEAPONBULLET *bullet;
	const char *name;
	int *id;
	int bullidx = *((int *) luaL_checkudata(L, 1, "SECONDARYWEAPONBULLETIDX"));
	bullet = &SecBulls[bullidx];
	name = luaL_checkstring(L, 2);
	/*if (!strcmp(name, "table")) -- TODO
	{
		lua_pushcfunction(L, luasecbull_table);
		return 1;
	}
	else*/ if (!strcmp(name, "index") || !strcmp(name, "Index"))
	{
		lua_pushinteger(L, bullidx + 1 + MAXPRIMARYWEAPONBULLETS);
		return 1;
	}
	else if (!strcmp(name, "category"))
	{
		lua_pushliteral(L, "secondary");
		return 1;
	}
	FIELD(Used, boolean);
	if (!strcmp(name, "Next") && bullet->Next != (u_int16_t) -1)
	{
		pushsecbull(L, bullet->Next);
		return 1;
	}
	else if (!strcmp(name, "Prev") && bullet->Prev != (u_int16_t) -1)
	{
		pushsecbull(L, bullet->Prev);
		return 1;
	}
	else if (!strcmp(name, "NextInGroup") && bullet->NextInGroup != NULL)
	{
		pushsecbull(L, bullet->NextInGroup - SecBulls);
		return 1;
	}
	else if (!strcmp(name, "PrevInGroup") && bullet->PrevInGroup != NULL)
	{
		pushsecbull(L, bullet->PrevInGroup - SecBulls);
		return 1;
	}
	else if (!strcmp(name, "Flags"))
	{
		/* FIXME: not really consistent */
		lua_createtable(L, 0, 3);
		SECFLAG(CanShoot);
		SECFLAG(CanTarget);
		SECFLAG(NoIncoming);
		return 1;
	}
	FIELD(SecType, integer);
	if (!strcmp(name, "Type"))
	{
		GETTABLEFORSEC(bullet->Type);
		return 1;
	}
	FIELD(MoveType, integer);
	FIELD(State, integer);
	FIELD(OwnerType, integer);
	if (!strcmp(name, "Owner"))
		return pushentity(L, bullet->OwnerType, bullet->Owner);
	FIELD(ID, integer);
	if (!strcmp(name, "Weapon"))
	{
		GETTABLEFORSEC(bullet->Weapon);
		return 1;
	}
	FIELD(Lensflare, boolean);
	FIELD(LifeCount, number);
	FIELD(LifeSpan, number);
	FIELD(SpeedInc, number);
	FIELD(SpeedWanted, number);
	FIELD(Speed, number);
	FIELD(DropCount, number);
	FIELDPTR(DropVector, vector);
	FIELD(Size, number);
	FIELDPTR(Pos, vector);
	FIELDPTR(Offset, vector);
	FIELDPTR(StartPos, vector);
	FIELDPTR(StartDir, vector);
	FIELDPTR(StartMat, matrix);
	FIELDPTR(ColStart, vector);
	FIELD(ColDist, number);
	FIELD(ColFlag, integer);
	FIELD(ColGroup, integer);
	FIELDPTR(ColPoint, vector); /* VERT */
	FIELDPTR(ColPointNormal, vector); /* NORMAL */
	FIELD(GroupImIn, integer);
	FIELD(ModelNum, integer);
	FIELD(ModelIndex, integer);
	FIELDPTR(Mat, matrix);
	FIELD(fmpoly, integer);
	FIELD(numfmpolys, integer);
	FIELD(poly, integer);
	FIELD(numpolys, integer);
	FIELD(xsize, number);
	FIELD(ysize, number);
	FIELD(light, integer);
	FIELD(lightsize, number);
	FIELD(r, number);
	FIELD(g, number);
	FIELD(b, number);
	FIELD(TurnSpeed, number);
	FIELD(ViewCone, number);
	FIELD(TargetType, integer);
	if (!strcmp(name, "Target"))
		return pushentity(L, bullet->TargetType, bullet->Target);
	/* FIELDPTR(DirQuat, quat); -- TODO */
	FIELDPTR(DirVector, vector);
	FIELDPTR(UpVector, vector);
	FIELD(Shield, number);
	FIELD(Damage, number);
	FIELD(Ammo, integer);
	FIELD(NumBounces, integer);
	FIELD(NumOldPos, integer);
	FIELD(RetractPos, number);
	FIELD(ColRadius, number);
	FIELD(Interval, number);
	FIELD(Time, number);
	FIELD(FramelagAddition, number);
	return luaL_argerror(L, 2, "unknown field name");
}
#undef FIELDPTR
#undef FIELD

/* TODO: move to a common function file and export */
static bool isudatatype(lua_State *L, int index, const char *mt)
{
	bool ret;
	if (!lua_getmetatable(L, index))
		return false;
	lua_getfield(L, LUA_REGISTRYINDEX, mt);
	ret = lua_rawequal(L, -1, -2);
	lua_pop(L, 2);
	return ret;
}

static int luaprimbull_equals(lua_State *L)
{
	lua_pushboolean(L,
		isudatatype(L, 1, "PRIMARYWEAPONBULLETIDX") &&
		isudatatype(L, 2, "PRIMARYWEAPONBULLETIDX") &&
		*((int *) lua_touserdata(L, 1)) == *((int *) lua_touserdata(L, 2))
	);
	return 1;
}

static int luasecbull_equals(lua_State *L)
{
	lua_pushboolean(L,
		isudatatype(L, 1, "SECONDARYWEAPONBULLETIDX") &&
		isudatatype(L, 2, "SECONDARYWEAPONBULLETIDX") &&
		*((int *) lua_touserdata(L, 1)) == *((int *) lua_touserdata(L, 2))
	);
	return 1;
}

int luaopen_bullets(lua_State *L)
{
	static const luaL_Reg primbullmt[] = {
		{ "__index", luaprimbull_index },
		{ "__eq", luaprimbull_equals },
		{ NULL, NULL }
	}, secbullmt[] = {
		{ "__index", luasecbull_index },
		{ "__eq", luasecbull_equals },
		{ NULL, NULL }
	};
	int i;

	luaL_newmetatable(L, "PRIMARYWEAPONBULLETIDX");
	luaL_register(L, NULL, primbullmt);
	lua_pop(L, 1);
	luaL_newmetatable(L, "SECONDARYWEAPONBULLETIDX");
	luaL_register(L, NULL, secbullmt);
	lua_pop(L, 1);
	lua_createtable(L, MAXPRIMARYWEAPONBULLETS + MAXSECONDARYWEAPONBULLETS, 0);
	for (i=0; i<MAXPRIMARYWEAPONBULLETS; i++)
	{
		lua_pushinteger(L, i+1);
		pushprimbull(L, i);
		lua_settable(L, -3);
	}
	for (i=0; i<MAXSECONDARYWEAPONBULLETS; i++)
	{
		lua_pushinteger(L, i+1 + MAXPRIMARYWEAPONBULLETS);
		pushsecbull(L, i);
		lua_settable(L, -3);
	}
	lua_setglobal(L, "bullets");
	return 0;
}