static int luaB_vec (lua_State *L) { float input[4] = { 0 }; switch (lua_gettop(L)) { case 4: input[3] = luaL_checknumber(L, 4); case 3: input[2] = luaL_checknumber(L, 3); case 2: input[1] = luaL_checknumber(L, 2); case 1: input[0] = luaL_checknumber(L, 1); break; default: luaL_error(L, "vec(...) takes 1 to 4 number arguments only"); } switch (lua_gettop(L)) { case 4: lua_pushvector4(L, input[0], input[1], input[2], input[3]); break; case 3: lua_pushvector3(L, input[0], input[1], input[2]); break; case 2: lua_pushvector2(L, input[0], input[1]); break; case 1: lua_pushnumber(L, input[0]); break; default:; } return 1; }
static int math_abs (lua_State *L) { lua_Number v; float x, y, z, w; switch (lua_type(L,1)) { case LUA_TNUMBER: if (lua_isinteger(L, 1)) { lua_Integer n = lua_tointeger(L, 1); if (n < 0) n = (lua_Integer)(0u - n); lua_pushinteger(L, n); } else { v = lua_tonumber(L,1); lua_pushnumber(L,l_mathop(fabs)(v)); } return 1; case LUA_TVECTOR2: lua_checkvector2(L,1,&x,&y); lua_pushvector2(L,fabsf(x),fabsf(y)); return 1; case LUA_TVECTOR3: lua_checkvector3(L,1,&x,&y,&z); lua_pushvector3(L,fabsf(x),fabsf(y),fabsf(z)); return 1; case LUA_TVECTOR4: lua_checkvector4(L,1,&x,&y,&z,&w); lua_pushvector4(L,fabsf(x),fabsf(y),fabsf(z),fabsf(w)); return 1; } luaL_error(L, "abs takes a number, integer, vector2, vector3, or vector4."); return 1; }
static int math_ceil (lua_State *L) { lua_Number v; float x, y, z, w; switch (lua_type(L,1)) { case LUA_TNUMBER: if (lua_isinteger(L, 1)) { lua_settop(L, 1); /* integer is its own ceil */ } else { v = lua_tonumber(L,1); pushnumint(L,ceil(v)); } return 1; case LUA_TVECTOR2: lua_checkvector2(L,1,&x,&y); lua_pushvector2(L,ceilf(x),ceilf(y)); return 1; case LUA_TVECTOR3: lua_checkvector3(L,1,&x,&y,&z); lua_pushvector3(L,ceilf(x),ceilf(y),ceilf(z)); return 1; case LUA_TVECTOR4: lua_checkvector4(L,1,&x,&y,&z,&w); lua_pushvector4(L,ceilf(x),ceilf(y),ceilf(z),ceilf(w)); return 1; } luaL_error(L, "ceil takes a number, integer, vector2, vector3, or vector4."); return 1; }
static int luaB_vector3 (lua_State *L) { float x,y,z; if (lua_gettop(L) != 3) luaL_error(L, "Invalid params, try vector3(n,n,n)"); x = (float)luaL_checknumber(L, 1); y = (float)luaL_checknumber(L, 2); z = (float)luaL_checknumber(L, 3); lua_pushvector3(L, x,y,z); return 1; }
static int luaB_cross (lua_State *L) { float x1, y1, z1; float x2, y2, z2; float ax, ay, az; if (lua_gettop(L) != 2) luaL_error(L, "Invalid params, try cross(v,v)"); lua_checkvector3(L, 1, &x1, &y1, &z1); lua_checkvector3(L, 2, &x2, &y2, &z2); cross3(x1,y1,z1, x2,y2,z2, &ax, &ay, &az); lua_pushvector3(L,ax,ay,az); return 1; }
static int math_clamp (lua_State *L) { if (lua_gettop(L)!=3) return luaL_error(L, "wrong number of arguments"); switch (lua_type(L,1)) { case LUA_TNUMBER: { lua_Number a, b, c; a = luaL_checknumber(L,1); b = luaL_checknumber(L,2); c = luaL_checknumber(L,3); if (a<b) a = b; if (a>c) a = c; lua_pushnumber(L, a); } break; case LUA_TVECTOR2: { float xa, ya; float xb, yb; float xc, yc; lua_checkvector2(L, 1, &xa, &ya); lua_checkvector2(L, 2, &xb, &yb); lua_checkvector2(L, 3, &xc, &yc); lua_pushvector2(L, do_clamp(xa,xb,xc), do_clamp(ya,yb,yc)); } break; case LUA_TVECTOR3: { float xa, ya, za; float xb, yb, zb; float xc, yc, zc; lua_checkvector3(L, 1, &xa, &ya, &za); lua_checkvector3(L, 2, &xb, &yb, &zb); lua_checkvector3(L, 3, &xc, &yc, &zc); lua_pushvector3(L, do_clamp(xa,xb,xc), do_clamp(ya,yb,yc), do_clamp(za,zb,zc)); } break; case LUA_TVECTOR4: { float xa, ya, za, wa; float xb, yb, zb, wb; float xc, yc, zc, wc; lua_checkvector4(L, 1, &xa, &ya, &za, &wa); lua_checkvector4(L, 2, &xb, &yb, &zb, &wb); lua_checkvector4(L, 3, &xc, &yc, &zc, &wc); lua_pushvector4(L, do_clamp(xa,xb,xc), do_clamp(ya,yb,yc), do_clamp(za,zb,zc), do_clamp(wa,wb,wc)); } break; default: return luaL_error(L, "clamp only works on number, vector2, vector3, vector4"); } return 1; }
static int luaB_norm (lua_State *L) { if (lua_gettop(L)==1 && lua_isvector3(L,1)) { float x, y, z; float len; lua_checkvector3(L, 1, &x, &y, &z); len = sqrtf(x*x + y*y + z*z); lua_pushvector3(L,x/len,y/len,z/len); return 1; } else if (lua_gettop(L)==1 && lua_isquat(L,1)) { float w, x, y, z; float qlen; lua_checkquat(L, 1, &w, &x, &y, &z); qlen = sqrtf(w*w + x*x + y*y + z*z); lua_pushquat(L,w/qlen,x/qlen,y/qlen,z/qlen); return 1; } else { luaL_error(L, "Invalid params, try norm(v) norm(q)"); return 0; } }
static int luaB_norm (lua_State *L) { if (lua_gettop(L)==1 && lua_isvector2(L,1)) { float x, y; float len; lua_checkvector2(L, 1, &x, &y); len = sqrtf(x*x + y*y); if (len == 0) luaL_error(L, "Cannot normalise vector2(0,0)"); lua_pushvector2(L,x/len,y/len); return 1; } else if (lua_gettop(L)==1 && lua_isvector3(L,1)) { float x, y, z; float len; lua_checkvector3(L, 1, &x, &y, &z); len = sqrtf(x*x + y*y + z*z); if (len == 0) luaL_error(L, "Cannot normalise vector3(0,0,0)"); lua_pushvector3(L,x/len,y/len,z/len); return 1; } else if (lua_gettop(L)==1 && lua_isvector4(L,1)) { float x, y, z, w; float len; lua_checkvector4(L, 1, &x, &y, &z, &w); len = sqrtf(x*x + y*y + z*z + w*w); if (len == 0) luaL_error(L, "Cannot normalise vector4(0,0,0,0)"); lua_pushvector4(L,x/len,y/len,z/len,w/len); return 1; } else if (lua_gettop(L)==1 && lua_isquat(L,1)) { float w, x, y, z; float qlen; lua_checkquat(L, 1, &w, &x, &y, &z); qlen = sqrtf(w*w + x*x + y*y + z*z); if (qlen == 0) luaL_error(L, "Cannot normalise quat(0,0,0,0)"); lua_pushquat(L,w/qlen,x/qlen,y/qlen,z/qlen); return 1; } else { return luaL_error(L, "Invalid arguments, try norm(v) or norm(q)."); } }
static int luaB_vector3 (lua_State *L) { float input[3]; if (!luaB_vectorn(L, 3, input)) luaL_error(L, "vector3(...) requires exactly 3 numbers"); lua_pushvector3(L, input[0], input[1], input[2]); return 1; }