/**
* @brief Gets a planet.
*
* Possible values of param:
* - nil : Gets the current landed planet or nil if there is none.
* - bool : Gets a random planet.
* - faction : Gets random planet belonging to faction matching the number.
* - string : Gets the planet by name.
* - table : Gets random planet belonging to any of the factions in the
* table.
*
* @usage p,s = planet.get( "Anecu" ) -- Gets planet by name
* @usage p,s = planet.get( faction.get( "Empire" ) ) -- Gets random Empire planet
* @usage p,s = planet.get(true) -- Gets completely random planet
* @usage p,s = planet.get( { faction.get("Empire"), faction.get("Dvaered") } ) -- Random planet belonging to Empire or Dvaered
* @luaparam param See description.
* @luareturn Returns the planet and the system it belongs to.
* @luafunc get( param )
*/
static int planetL_get( lua_State *L )
{
int i;
int *factions;
int nfactions;
char **planets;
int nplanets;
const char *rndplanet;
LuaPlanet planet;
LuaSystem sys;
LuaFaction *f;
rndplanet = NULL;
nplanets = 0;
/* Get the landed planet */
if (lua_gettop(L) == 0) {
if (land_planet != NULL) {
planet.p = land_planet;
lua_pushplanet(L,planet);
sys.s = system_get( planet_getSystem(land_planet->name) );
lua_pushsystem(L,sys);
return 2;
}
NLUA_ERROR(L,"Attempting to get landed planet when player not landed.");
return 0; /* Not landed. */
}
/* If boolean return random. */
else if (lua_isboolean(L,1)) {
planet.p = planet_get( space_getRndPlanet() );
lua_pushplanet(L,planet);
sys.s = system_get( planet_getSystem(land_planet->name) );
lua_pushsystem(L,sys);
return 2;
}
/* Get a planet by faction */
else if (lua_isfaction(L,1)) {
f = lua_tofaction(L,1);
planets = space_getFactionPlanet( &nplanets, &f->f, 1 );
}
/* Get a planet by name */
else if (lua_isstring(L,1)) {
rndplanet = lua_tostring(L,1);
}
/* Get a planet from faction list */
else if (lua_istable(L,1)) {
/* Get table length and preallocate. */
nfactions = (int) lua_objlen(L,1);
factions = malloc( sizeof(int) * nfactions );
/* Load up the table. */
lua_pushnil(L);
i = 0;
while (lua_next(L, -2) != 0) {
f = lua_tofaction(L, -1);
factions[i++] = f->f;
lua_pop(L,1);
}
/* get the planets */
planets = space_getFactionPlanet( &nplanets, factions, nfactions );
free(factions);
}
else NLUA_INVALID_PARAMETER(); /* Bad Parameter */
/* No suitable planet found */
if ((rndplanet == NULL) && (nplanets == 0)) {
free(planets);
return 0;
}
/* Pick random planet */
else if (rndplanet == NULL) {
rndplanet = planets[RNG(0,nplanets-1)];
free(planets);
}
/* Push the planet */
planet.p = planet_get(rndplanet); /* The real planet */
lua_pushplanet(L,planet);
sys.s = system_get( planet_getSystem(rndplanet) );
lua_pushsystem(L,sys);
return 2;
}
static int planetL_getBackend( lua_State *L, int landable )
{
int i;
int *factions;
int nfactions;
char **planets;
int nplanets;
const char *rndplanet;
LuaSystem luasys;
LuaFaction f;
Planet *pnt;
StarSystem *sys;
char *sysname;
rndplanet = NULL;
planets = NULL;
nplanets = 0;
/* If boolean return random. */
if (lua_isboolean(L,1)) {
pnt = planet_get( space_getRndPlanet(landable, 0, NULL) );
lua_pushplanet(L,planet_index( pnt ));
luasys = system_index( system_get( planet_getSystem(pnt->name) ) );
lua_pushsystem(L,luasys);
return 2;
}
/* Get a planet by faction */
else if (lua_isfaction(L,1)) {
f = lua_tofaction(L,1);
planets = space_getFactionPlanet( &nplanets, &f, 1, landable );
}
/* Get a planet by name */
else if (lua_isstring(L,1)) {
rndplanet = lua_tostring(L,1);
if (landable) {
pnt = planet_get( rndplanet );
if (pnt == NULL) {
NLUA_ERROR(L, _("Planet '%s' not found in stack"), rndplanet);
return 0;
}
/* Check if can land. */
planet_updateLand( pnt );
if (!pnt->can_land)
return 0;
}
}
/* Get a planet from faction list */
else if (lua_istable(L,1)) {
/* Get table length and preallocate. */
nfactions = (int) lua_objlen(L,1);
factions = malloc( sizeof(int) * nfactions );
/* Load up the table. */
lua_pushnil(L);
i = 0;
while (lua_next(L, -2) != 0) {
if (lua_isfaction(L, -1))
factions[i++] = lua_tofaction(L, -1);
lua_pop(L,1);
}
/* get the planets */
planets = space_getFactionPlanet( &nplanets, factions, nfactions, landable );
free(factions);
}
else
NLUA_INVALID_PARAMETER(L); /* Bad Parameter */
/* No suitable planet found */
if ((rndplanet == NULL) && ((planets == NULL) || nplanets == 0))
return 0;
/* Pick random planet */
else if (rndplanet == NULL) {
planets = (char**) arrayShuffle( (void**)planets, nplanets );
for (i=0; i<nplanets; i++) {
if (landable) {
/* Check landing. */
pnt = planet_get( planets[i] );
if (pnt == NULL)
continue;
planet_updateLand( pnt );
if (!pnt->can_land)
continue;
}
rndplanet = planets[i];
break;
}
free(planets);
}
/* Push the planet */
pnt = planet_get(rndplanet); /* The real planet */
if (pnt == NULL) {
NLUA_ERROR(L, _("Planet '%s' not found in stack"), rndplanet);
return 0;
}
sysname = planet_getSystem(rndplanet);
if (sysname == NULL) {
NLUA_ERROR(L, _("Planet '%s' is not placed in a system"), rndplanet);
return 0;
}
sys = system_get( sysname );
if (sys == NULL) {
NLUA_ERROR(L, _("Planet '%s' can't find system '%s'"), rndplanet, sysname);
return 0;
}
lua_pushplanet(L,planet_index( pnt ));
luasys = system_index( sys );
lua_pushsystem(L,luasys);
return 2;
}
