mantarimay/Aegisub
1
0
Fork 0
forked from mia/Aegisub
Code Pull requests Activity

Make subs.delete accept a table of rows to delete

Browse source 
Lua has a limit to how many arguments can be pushed onto the stack, so
when deleting very large numbers of lines at once unpacking a table can
fail.
This commit is contained in:
Thomas Goyne 2014-04-15 20:47:59 -07:00
parent c685ae4aea
commit 162e119afa
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/auto4_lua_assfile.cpp
View file

@ -448,14 +448,25 @@ namespace Automation4 {
// get number of items to delete // get number of items to delete
int itemcount = lua_gettop(L); int itemcount = lua_gettop(L);
std::vector<int> ids; if (itemcount == 0) return;
ids.reserve(itemcount);
while (itemcount > 0) { std::vector<int> ids;
int n = luaL_checkint(L, itemcount); if (itemcount == 1 && lua_istable(L, 1)) {
luaL_argcheck(L, n > 0 && n <= (int)lines.size(), itemcount, "Out of range line index"); lua_pushvalue(L, 1);
ids.push_back(n - 1); lua_for_each(L, [&] {
--itemcount; int n = luaL_checkint(L, -1);
luaL_argcheck(L, n > 0 && n <= (int)lines.size(), 1, "Out of range line index");
ids.push_back(n - 1);
});
}
else {
ids.reserve(itemcount);
while (itemcount > 0) {
int n = luaL_checkint(L, itemcount);
luaL_argcheck(L, n > 0 && n <= (int)lines.size(), itemcount, "Out of range line index");
ids.push_back(n - 1);
--itemcount;
}
} }
sort(ids.begin(), ids.end()); sort(ids.begin(), ids.end());