Add a Utils.Where function to lua for filtering collections
This commit is contained in:
@@ -9,6 +9,7 @@
|
|||||||
*/
|
*/
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
|
using System.Collections;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using Eluant;
|
using Eluant;
|
||||||
using OpenRA.Scripting;
|
using OpenRA.Scripting;
|
||||||
@@ -54,6 +55,20 @@ namespace OpenRA.Mods.Common.Scripting
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Desc("Returns the original collection filtered with the func.")]
|
||||||
|
public LuaTable Where(LuaValue[] collection, LuaFunction func)
|
||||||
|
{
|
||||||
|
var t = Context.CreateTable();
|
||||||
|
|
||||||
|
foreach (var c in collection)
|
||||||
|
using (var ret = func.Call(c))
|
||||||
|
using (var result = ret.FirstOrDefault())
|
||||||
|
if (result != null && result.ToBoolean())
|
||||||
|
t.Add(t.Count + 1, c);
|
||||||
|
|
||||||
|
return t;
|
||||||
|
}
|
||||||
|
|
||||||
[Desc("Returns the first n values from a collection.")]
|
[Desc("Returns the first n values from a collection.")]
|
||||||
public LuaValue[] Take(int n, LuaValue[] source)
|
public LuaValue[] Take(int n, LuaValue[] source)
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user