func WaypointsFunc(me *game.Entity) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "Waypoints") {
return 0
}
g := me.Game()
L.NewTable()
count := 0
for _, wp := range g.Waypoints {
if wp.Side != me.Side() {
continue
}
count++
L.PushInteger(count)
L.NewTable()
L.PushString("Name")
L.PushString(wp.Name)
L.SetTable(-3)
L.PushString("Radius")
L.PushNumber(wp.Radius)
L.SetTable(-3)
L.PushString("Pos")
game.LuaPushPoint(L, int(wp.X), int(wp.Y))
L.SetTable(-3)
L.SetTable(-3)
}
return 1
}
}
// Performs a move action to the closest one of any of the specifed inputs
// points. The movement can be restricted to not spend more than a certain
// amount of ap.
// Format:
// success, p = DoMove(dsts, max_ap)
//
// Input:
// dsts - array[table[x,y]] - Array of all points that are acceptable
// destinations.
// max_ap - integer - Maxmium ap to spend while doing this move, if the
// required ap exceeds this the entity will still move
// as far as possible towards a destination.
//
// Output:
// success = bool - True iff the move made it to a position in dsts.
// p - table[x,y] - New position of this entity, or nil if the move failed.
func DoMoveFunc(a *Ai) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "DoMove", game.LuaArray, game.LuaInteger) {
return 0
}
me := a.ent
max_ap := L.ToInteger(-1)
L.Pop(1)
cur_ap := me.Stats.ApCur()
if max_ap > cur_ap {
max_ap = cur_ap
}
n := int(L.ObjLen(-1))
dsts := make([]int, n)[0:0]
for i := 1; i <= n; i++ {
L.PushInteger(i)
L.GetTable(-2)
x, y := game.LuaToPoint(L, -1)
dsts = append(dsts, me.Game().ToVertex(x, y))
L.Pop(1)
}
var move *actions.Move
var ok bool
for i := range me.Actions {
move, ok = me.Actions[i].(*actions.Move)
if ok {
break
}
}
if !ok {
// TODO: what to do here? This poor guy didn't have a move action :(
L.PushNil()
L.PushNil()
return 2
}
exec := move.AiMoveToPos(me, dsts, max_ap)
if exec != nil {
a.execs <- exec
<-a.pause
// TODO: Need to get a resolution
x, y := me.Pos()
v := me.Game().ToVertex(x, y)
complete := false
for i := range dsts {
if v == dsts[i] {
complete = true
break
}
}
L.PushBoolean(complete)
game.LuaPushPoint(L, x, y)
base.Log().Printf("Finished move")
} else {
base.Log().Printf("Didn't bother moving")
L.PushBoolean(true)
L.PushNil()
}
return 2
}
}
// Returns a list of all positions that the specified door can be opened and
// closed from.
// Format
// ps = doorPositions(d)
//
// Input:
// d - door - A door.
//
// Output:
// ps - array[table[x,y]] - List of all position this door can be opened
// and closed from.
func DoorPositionsFunc(a *Ai) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "DoorPositions", game.LuaDoor) {
return 0
}
room := game.LuaToRoom(L, a.ent.Game(), -1)
door := game.LuaToDoor(L, a.ent.Game(), -1)
if door == nil || room == nil {
game.LuaDoError(L, "DoorPositions: Specified an invalid door.")
return 0
}
var x, y, dx, dy int
switch door.Facing {
case house.FarLeft:
x = door.Pos
y = room.Size.Dy - 1
dx = 1
case house.FarRight:
x = room.Size.Dx - 1
y = door.Pos
dy = 1
case house.NearLeft:
x = -1
y = door.Pos
dy = 1
case house.NearRight:
x = door.Pos
y = -1
dx = 1
default:
game.LuaDoError(L, fmt.Sprintf("Found a door with a bad facing."))
}
L.NewTable()
count := 1
for i := 0; i < door.Width; i++ {
L.PushInteger(count*2 - 1)
game.LuaPushPoint(L, room.X+x+dx*i, room.Y+y+dy*i)
L.SetTable(-3)
L.PushInteger(count * 2)
game.LuaPushPoint(L, room.X+x+dx*i+dy, room.Y+y+dy*i+dx)
L.SetTable(-3)
count++
}
return 1
}
}
func (exec aoeExec) Push(L *lua.State, g *game.Game) {
exec.BasicActionExec.Push(L, g)
if L.IsNil(-1) {
return
}
L.PushString("Pos")
game.LuaPushPoint(L, exec.X, exec.Y)
L.SetTable(-3)
}
func (exec summonExec) Push(L *lua.State, g *game.Game) {
exec.BasicActionExec.Push(L, g)
if L.IsNil(-1) {
return
}
_, x, y := g.FromVertex(exec.Pos)
L.PushString("Pos")
game.LuaPushPoint(L, x, y)
L.SetTable(-3)
}
// Returns an array of all points that can be reached by walking from a
// specific location that end in a certain general area. Assumes that a 1x1
// unit is doing the walking.
// Format:
// points = AllPathablePoints(src, dst, min, max)
//
// Inputs:
// src - table[x,y] - Where the path starts.
// dst - table[x,y] - Another point near where the path should go.
// min - integer - Minimum distance from dst that the path should end at.
// max - integer - Maximum distance from dst that the path should end at.
//
// Outputs:
// points - array[table[x,y]]
func AllPathablePointsFunc(a *Ai) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "AllPathablePoints", game.LuaPoint, game.LuaPoint, game.LuaInteger, game.LuaInteger) {
return 0
}
min := L.ToInteger(-2)
max := L.ToInteger(-1)
x1, y1 := game.LuaToPoint(L, -4)
x2, y2 := game.LuaToPoint(L, -3)
a.ent.Game().DetermineLos(x2, y2, max, grid)
var dst []int
for x := x2 - max; x <= x2+max; x++ {
for y := y2 - max; y <= y2+max; y++ {
if x > x2-min && x < x2+min && y > y2-min && y < y2+min {
continue
}
if x < 0 || y < 0 || x >= len(grid) || y >= len(grid[0]) {
continue
}
if !grid[x][y] {
continue
}
dst = append(dst, a.ent.Game().ToVertex(x, y))
}
}
vis := 0
for i := range grid {
for j := range grid[i] {
if grid[i][j] {
vis++
}
}
}
base.Log().Printf("Visible: %d", vis)
graph := a.ent.Game().Graph(a.ent.Side(), true, nil)
src := []int{a.ent.Game().ToVertex(x1, y1)}
reachable := algorithm.ReachableDestinations(graph, src, dst)
L.NewTable()
base.Log().Printf("%d/%d reachable from (%d, %d) -> (%d, %d)", len(reachable), len(dst), x1, y1, x2, y2)
for i, v := range reachable {
_, x, y := a.ent.Game().FromVertex(v)
L.PushInteger(i + 1)
game.LuaPushPoint(L, x, y)
L.SetTable(-3)
}
return 1
}
}
func (exec *moveExec) Push(L *lua.State, g *game.Game) {
exec.BasicActionExec.Push(L, g)
if L.IsNil(-1) {
return
}
L.PushString("Path")
L.NewTable()
for i := range exec.Path {
L.PushInteger(i + 1)
_, x, y := g.FromVertex(exec.Path[i])
game.LuaPushPoint(L, x, y)
L.SetTable(-3)
}
L.SetTable(-3)
}
// Performs an aoe attack against centered at the specified position.
// Format:
// target = BestAoeAttackPos(attack, extra_dist, spec)
//
// Inputs:
// attack - string - Name of the attack to use.
// extra_dist - integer - Available distance to move before attacking.
// spec - string - One of the following values:
// "allies ok", "minions ok", "enemies only"
//
// Outputs:
// pos - table[x,y] - Position to place aoe for maximum results.
// hits - array[ents] - Visible entities that will be in the aoe
func BestAoeAttackPosFunc(a *Ai) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "BestAoeAttackPos", game.LuaString, game.LuaInteger, game.LuaString) {
return 0
}
me := a.ent
name := L.ToString(-3)
action := getActionByName(me, name)
if action == nil {
game.LuaDoError(L, fmt.Sprintf("Entity '%s' (id=%d) has no action named '%s'.", me.Name, me.Id, name))
return 0
}
attack, ok := action.(*actions.AoeAttack)
if !ok {
game.LuaDoError(L, fmt.Sprintf("Action '%s' is not an aoe attack.", name))
return 0
}
var spec actions.AiAoeTarget
switch L.ToString(-1) {
case "allies ok":
spec = actions.AiAoeHitAlliesOk
case "minions ok":
spec = actions.AiAoeHitMinionsOk
case "enemies only":
spec = actions.AiAoeHitNoAllies
default:
game.LuaDoError(L, fmt.Sprintf("'%s' is not a valid value of spec for BestAoeAttackPos().", L.ToString(-1)))
return 0
}
x, y, hits := attack.AiBestTarget(me, L.ToInteger(-2), spec)
game.LuaPushPoint(L, x, y)
L.NewTable()
for i := range hits {
L.PushInteger(i + 1)
game.LuaPushEntity(L, hits[i])
L.SetTable(-3)
}
return 2
}
}
// Returns a list of all positions inside the specified room.
// Format
// ps = roomPositions(r)
//
// Input:
// r - room - A room.
//
// Output:
// ps - array[table[x,y]] - List of all position inside the specified room.
func RoomPositionsFunc(a *Ai) lua.GoFunction {
return func(L *lua.State) int {
if !game.LuaCheckParamsOk(L, "roomPositions", game.LuaRoom) {
return 0
}
room := game.LuaToRoom(L, a.ent.Game(), -1)
if room == nil {
game.LuaDoError(L, "RoomPositions: Specified an invalid room.")
return 0
}
L.NewTable()
count := 1
for x := room.X; x < room.X+room.Size.Dx; x++ {
for y := room.Y; y < room.Y+room.Size.Dy; y++ {
L.PushInteger(count)
count++
game.LuaPushPoint(L, x, y)
L.SetTable(-3)
}
}
return 1
}
}