aspect_advancements.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2013 - 2024
    by Felix Bauer <fehlxb+wesnoth@gmail.com>
    Part of the Battle for Wesnoth Project https://www.wesnoth.org/
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY.
 
    See the COPYING file for more details.
*/
 
#include "ai/lua/aspect_advancements.hpp"
 
#include "log.hpp"                // for LOG_STREAM, logger, etc
#include "lua/wrapper_lauxlib.h"        // for luaL_ref, LUA_REFNIL, lua_isstring, etc
#include "map/location.hpp"             // for map_location
#include "serialization/string_utils.hpp"  // for split
#include "units/unit.hpp"
#include "units/map.hpp"    // for unit_map::const_iterator, etc
 
#include <string>                       // for string, char_traits, etc
#include <vector>                       // for vector
 
 
 
static lg::log_domain log_ai_engine_lua("ai/engine/lua");
#define LOG_LUA LOG_STREAM(info, log_ai_engine_lua)
#define ERR_LUA LOG_STREAM(err, log_ai_engine_lua)
 
namespace ai{
 
unit_advancements_aspect::unit_advancements_aspect():
        val_(), L_(),ref_()
{
}
 
unit_advancements_aspect::unit_advancements_aspect(lua_State* L, int n)
    : val_("Lua Function")
    , L_(L)
    , ref_()
{
    lua_settop(L, n);
 
    //on the top of the Lua-Stack is now the pointer to the function. Save it:
    ref_ = luaL_ref(L, LUA_REGISTRYINDEX);
 
}
 
unit_advancements_aspect::unit_advancements_aspect(const std::string& val):  val_(val), L_(), ref_()
{
}
 
unit_advancements_aspect::~unit_advancements_aspect()
{
    if(L_) {
        // Remove the function from the registry
        luaL_unref(L_, LUA_REGISTRYINDEX, ref_);
    }
}
 
const std::vector<std::string> unit_advancements_aspect::get_advancements(const unit_map::const_iterator& unit) const
{
 
    if(!unit.valid())
    {
        return std::vector<std::string>();
    }
 
    const std::string& unit_id = (*unit).id();
    const int unit_x = (*unit).get_location().wml_x();
    const int unit_y = (*unit).get_location().wml_y();
 
    LOG_LUA << "Entering unit_advancements_aspect::get_advancements() in instance " << this << " with unit " << unit_id <<  " on (x,y) = (" << unit_x << ", " << unit_y << ")";
 
    if(L_ == nullptr || ref_ == LUA_REFNIL)
    {
        //If we end up here, most likely the aspect don't use the lua-engine.
        //Just to make sure:
        if (val_ == "Lua Function")
        {
            return std::vector<std::string>();
        }
        return utils::split(val_);
    }
 
    //put the Pointer back on the Stack
    lua_rawgeti(L_, LUA_REGISTRYINDEX, ref_);
 
    if(lua_isstring(L_, -1))
    {
        return utils::split(lua_tostring(L_, -1));
    }
 
    if(!lua_isfunction(L_, -1))
    {
        ERR_LUA << "Can't evaluate advancement aspect: Value is neither a string nor a function.";
        return std::vector<std::string>();
    }
 
    //push parameter to the stack
    lua_pushinteger(L_, unit_x);
    lua_pushinteger(L_, unit_y);
 
    //To make unit_id a Parameter of the Lua function:
    //lua_pushfstring(L_, unit_id.c_str());
 
    //call function
    if(lua_pcall(L_, 2, 1, 0) != 0)
    {
        ERR_LUA << "LUA Error while evaluating advancements_aspect: " << lua_tostring(L_, -1);
        return std::vector<std::string>();
    }
    if (!lua_isstring(L_, -1))
    {
        ERR_LUA << "LUA Error while evaluating advancements_aspect: Function must return String ";
        return std::vector<std::string>();
    }
 
    //get result from Lua-Stack
    const std::string retval = std::string(lua_tostring(L_, -1));
    lua_pop(L_, 1);
 
    LOG_LUA << "Called Lua advancement function. Result was: \"" << retval << "\".";
 
    return utils::split(retval);
}
 
const std::string unit_advancements_aspect::get_value() const
{
    return val_;
}
}