Brandon W
58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
#include <vector>
|
|
|
|
#include "global.h"
|
|
|
|
#include "EnumHelper.h"
|
|
#include "GameplayHelpers.h"
|
|
#include "GameState.h"
|
|
#include "LuaManager.h"
|
|
#include "ThemeManager.h"
|
|
#include "Style.h"
|
|
|
|
std::vector<NotefieldMargins> GetNotefieldMargins() {
|
|
LuaReference margin_func;
|
|
std::vector<NotefieldMargins> margins(2);
|
|
|
|
THEME->GetMetric("ScreenGameplay", "MarginFunction", margin_func);
|
|
if (margin_func.GetLuaType() != LUA_TFUNCTION)
|
|
{
|
|
LuaHelpers::ReportScriptErrorFmt("MarginFunction metric for %s must be a function.", "ScreenGameplay");
|
|
return margins;
|
|
}
|
|
|
|
// Setup the lua.
|
|
Lua* lua_ptr = LUA->Get();
|
|
margin_func.PushSelf(lua_ptr);
|
|
lua_createtable(lua_ptr, 0, 0);
|
|
int next_player_slot = 1;
|
|
FOREACH_EnabledPlayer(pn)
|
|
{
|
|
Enum::Push(lua_ptr, pn);
|
|
lua_rawseti(lua_ptr, -2, next_player_slot);
|
|
++next_player_slot;
|
|
}
|
|
|
|
Enum::Push(lua_ptr, GAMESTATE->GetCurrentStyle(PLAYER_INVALID)->m_StyleType);
|
|
RString err = "Error running MarginFunction: ";
|
|
|
|
// Run the lua code.
|
|
if (LuaHelpers::RunScriptOnStack(lua_ptr, /*Error=*/err, /*Args=*/2, /*ReturnValues*/3, /*ReportOnError*/true))
|
|
{
|
|
std::string margin_error_msg = "Margin value must be a number.";
|
|
|
|
// Pull the values off the lua stack. Note that the lua function is
|
|
// returning the margins of P1 + P2 combined. Therefore, we need to use
|
|
// the center to tell where P1's margin ends and P2's begins, if
|
|
// applicable.
|
|
margins[PLAYER_1].left = SafeFArg(lua_ptr, -3, margin_error_msg, 40);
|
|
float center = SafeFArg(lua_ptr, -2, margin_error_msg, 80);
|
|
margins[PLAYER_1].right = center / 2.0f;
|
|
|
|
margins[PLAYER_2].left = center / 2.0f;
|
|
margins[PLAYER_2].right = SafeFArg(lua_ptr, -1, margin_error_msg, 40);
|
|
}
|
|
lua_settop(lua_ptr, 0);
|
|
LUA->Release(lua_ptr);
|
|
return margins;
|
|
}
|