nlua_munition.c

Go to the documentation of this file.

/*
 * See Licensing and Copyright notice in naev.h
 */
#include "naev.h"
 
#include "nlua_munition.h"
 
#include "array.h"
#include "nlua.h"
#include "nlua_faction.h"
#include "nlua_outfit.h"
#include "nlua_pilot.h"
#include "nlua_vec2.h"
#include "nluadef.h"
 
/* Prototypes. */
static Weapon *munition_get( LuaMunition *lm );
 
/* Munition metatable methods. */
static int munitionL_eq( lua_State *L );
static int munitionL_tostring( lua_State *L );
static int munitionL_exists( lua_State *L );
static int munitionL_clear( lua_State *L );
static int munitionL_getAll( lua_State *L );
static int munitionL_getInrange( lua_State *L );
static int munitionL_pos( lua_State *L );
static int munitionL_vel( lua_State *L );
static int munitionL_faction( lua_State *L );
static int munitionL_parent( lua_State *L );
static int munitionL_target( lua_State *L );
static int munitionL_outfit( lua_State *L );
static int munitionL_strength( lua_State *L );
static int munitionL_strengthSet( lua_State *L );
 
static const luaL_Reg munitionL_methods[] = {
   /* General. */
   { "__eq", munitionL_eq },
   { "__tostring", munitionL_tostring },
   { "exists", munitionL_exists },
   { "clear", munitionL_clear },
   { "getAll", munitionL_getAll },
   { "getInrange", munitionL_getInrange },
   /* Get properties. */
   { "pos", munitionL_pos },
   { "vel", munitionL_vel },
   { "faction", munitionL_faction },
   { "parent", munitionL_parent },
   { "target", munitionL_target },
   { "outfit", munitionL_outfit },
   { "strength", munitionL_strength },
   /* Set properties. */
   { "strengthSet", munitionL_strengthSet },
   /* End sentinal. */
   { 0, 0 },
}; 

int nlua_loadMunition( nlua_env env )
{
   nlua_register( env, MUNITION_METATABLE, munitionL_methods, 1 );
 
   /* Munition always loads ship and asteroid. */
   nlua_loadOutfit( env );
 
   return 0;
}

LuaMunition *lua_tomunition( lua_State *L, int ind )
{
   return ( (LuaMunition *)lua_touserdata( L, ind ) );
}

LuaMunition *luaL_checkmunition( lua_State *L, int ind )
{
   if ( lua_ismunition( L, ind ) )
      return lua_tomunition( L, ind );
   luaL_typerror( L, ind, MUNITION_METATABLE );
   return lua_tomunition(
      L, ind ); /* Just to shut compiler up, shouldn't be reached. */
}

Weapon *luaL_validmunition( lua_State *L, int ind )
{
   Weapon *w = munition_get( luaL_checkmunition( L, ind ) );
   if ( w == NULL ) {
      NLUA_ERROR( L, _( "Munition is invalid." ) );
      return NULL;
   }
   return w;
}

LuaMunition *lua_pushmunition( lua_State *L, const Weapon *w )
{
   const Weapon *weapon_stack = weapon_getStack();
   LuaMunition *lm = (LuaMunition *)lua_newuserdata( L, sizeof( LuaMunition ) );
   lm->id          = w->id;
   lm->idx         = w - weapon_stack;
   luaL_getmetatable( L, MUNITION_METATABLE );
   lua_setmetatable( L, -2 );
   return lm;
}

int lua_ismunition( lua_State *L, int ind )
{
   int ret;
 
   if ( lua_getmetatable( L, ind ) == 0 )
      return 0;
   lua_getfield( L, LUA_REGISTRYINDEX, MUNITION_METATABLE );
 
   ret = 0;
   if ( lua_rawequal( L, -1, -2 ) ) /* does it have the correct mt? */
      ret = 1;
 
   lua_pop( L, 2 ); /* remove both metatables */
   return ret;
}

static int munitionL_eq( lua_State *L )
{
   const LuaMunition *lm1 = luaL_checkmunition( L, 1 );
   const LuaMunition *lm2 = luaL_checkmunition( L, 2 );
   lua_pushboolean( L, lm1->id == lm2->id );
   return 1;
}
 
static Weapon *munition_get( LuaMunition *lm )
{
   Weapon *weapon_stack = weapon_getStack();
   Weapon *w;
 
   if ( ( lm->idx < (size_t)array_size( weapon_stack ) ) &&
        ( weapon_stack[lm->idx].id == lm->id ) )
      return &weapon_stack[lm->idx];
 
   w = weapon_getID( lm->id );
   if ( w == NULL )
      return NULL;
   lm->idx = w - weapon_stack; /* For next look ups. */
   return w;
}

static int munitionL_tostring( lua_State *L )
{
   LuaMunition *lm = luaL_checkmunition( L, 1 );
   Weapon      *w  = munition_get( lm );
   if ( w != NULL )
      lua_pushstring( L, _( w->outfit->name ) );
   else
      lua_pushstring( L, "(inexistent munition)" );
   return 1;
}

static int munitionL_exists( lua_State *L )
{
   LuaMunition *lm = luaL_checkmunition( L, 1 );
   lua_pushboolean( L, munition_get( lm ) != NULL );
   return 1;
}

static int munitionL_clear( lua_State *L )
{
   (void)L;
   weapon_clear();
   return 0;
}

static int munitionL_getAll( lua_State *L )
{
   const Weapon *weapon_stack = weapon_getStack();
   int           onlyhittable = lua_toboolean( L, 1 );
   int           n            = 1;
   lua_newtable( L );
   for ( int i = 0; i < array_size( weapon_stack ); i++ ) {
      const Weapon *w = &weapon_stack[i];
      if ( weapon_isFlag( w, WEAPON_FLAG_DESTROYED ) )
         continue;
      if ( onlyhittable && !weapon_isFlag( w, WEAPON_FLAG_HITTABLE ) )
         continue;
      lua_pushmunition( L, w );
      lua_rawseti( L, -2, n++ );
   }
   return 1;
}

static int weapon_isHostile( const Weapon *w, const Pilot *p )
{
   if ( p->id == w->parent )
      return 0;
 
   if ( ( w->target.type == TARGET_PILOT ) && ( w->target.u.id == p->id ) )
      return 1;
 
   /* Let hostiles hit player. */
   if ( p->faction == FACTION_PLAYER ) {
      const Pilot *parent = pilot_get( w->parent );
      if ( parent != NULL ) {
         if ( pilot_isHostile( parent ) )
            return 1;
      }
   }
 
   /* Hit non-allies. */
   if ( areEnemies( w->faction, p->faction ) )
      return 1;
 
   return 0;
}

static int munitionL_getInrange( lua_State *L )
{
   const Weapon  *weapon_stack = weapon_getStack();
   const IntList *qt;
   int            n     = 1;
   const vec2    *pos   = luaL_checkvector( L, 1 );
   double         range = luaL_checknumber( L, 2 );
   const Pilot   *p     = luaL_optpilot( L, 3, NULL );
   double         r2    = pow2( range );
   int            x, y, r;
 
   x = round( pos->x );
   y = round( pos->y );
   r = ceil( range );
 
   lua_newtable( L );
   qt = weapon_collideQuery( x - r, y - r, x + r, y + r );
   for ( int i = 0; i < il_size( qt ); i++ ) {
      const Weapon *w = &weapon_stack[il_get( qt, i, 0 )];
      if ( weapon_isFlag( w, WEAPON_FLAG_DESTROYED ) )
         continue;
      if ( ( p != NULL ) && !weapon_isHostile( w, p ) )
         continue;
      if ( vec2_dist2( &w->solid.pos, pos ) > r2 )
         continue;
      lua_pushmunition( L, w );
      lua_rawseti( L, -2, n++ );
   }
   return 1;
}

static int munitionL_pos( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushvector( L, w->solid.pos );
   return 1;
}

static int munitionL_vel( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushvector( L, w->solid.vel );
   return 1;
}

static int munitionL_faction( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushfaction( L, w->faction );
   return 1;
}

static int munitionL_parent( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushpilot( L, w->parent );
   return 1;
}

static int munitionL_target( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushpilot( L, w->parent );
   return 1;
}

static int munitionL_outfit( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushoutfit( L, w->outfit );
   return 1;
}

static int munitionL_strength( lua_State *L )
{
   const Weapon *w = luaL_validmunition( L, 1 );
   lua_pushnumber( L, w->strength );
   return 1;
}

static int munitionL_strengthSet( lua_State *L )
{
   Weapon *w        = luaL_validmunition( L, 1 );
   double  sb       = w->strength_base;
   w->strength_base = luaL_checknumber( L, 2 );
   w->strength *= w->strength_base / sb;
   return 1;
}