Lua的Iterator以及函数心得

from http://sunxiunan.com/?p=1249

local tbl = {"one", "two", "three", ‘five’}

local count = #tbl

local inc = 0

function Iter(tableinput)

return function()

if inc > count then

return

end

inc = inc + 1

return tableinput[inc], inc

end

end

for elem, index in Iter(tbl) do

print(elem)

if index > 2 then

inc = 0

end

end

如果运行一下的话,不出意外一定会死循环。如果把函数Iter改一下就是我们常见的generic for了。

function Iter(tableinput)

local count2 = #tbl

local inc2 = 0

return function()

if inc2 > count2 then

return

end

inc2 = inc2 + 1

return tableinput[inc2], inc2

end

end

跟踪一下运行也可以发现,count2以及inc2这两行在Iter被for调用的时候,只会第一次运行而且只运行一次。接下来的调用中只会从return function这里开始。如果返回nil,那么for就停止循环。

我比较好奇的是,对于Iter在for循环的时候,很显然是保持了一个内部的堆栈不变,也就是说inc2和count2类似于c语言的static内部变量。但是学的还是一知半解,不知道如何dump出此时的堆栈状态。

关于函数对于局部变量,好像是会在栈上做一个缓存,看下面代码

stringb = "bbbbbbbbbbbbbbb"

do

local internalA = "print for first time"

function printStrA()

print(internalA)

end

function printStrB()

print(stringb)

end

end

do

local internalA = "!!!!!!!!!!!!!!!!!!!!"

stringb = "cccccccccccccccc"

printStrA()

printStrB()

end

internalA = "!!!!2222222222222"

printStrA()

stringb = "ddddddddddddddddddd"

printStrB()

当我们运行这段代码,可以发现

print for first time

cccccccccccccccc

print for first time

ddddddddddddddddddd

也就是对于printStrA,每次调用的结果都是一样的,而printStrB的结果会根据全局变量stringb的不同而不同。这一点值得以后的研究。

关于闭包(closures),PIL上有一个很有意思的例子,我做了一些扩展。

local x = function ()

local i = 0

print "aaa"

local function x2()

i = i + 1

print "bbb"

return i

end

return x2 — 这个地方不能是return x2(),否则返回的就是值而不是函数

end

x()

print "======="

x()

print "======="

local c = x()

print "======="

print(c())

print "======="

print(c())

print "======="

print(c())

print "======="

local c = x()

print "======="

print(c())

结果会是什么呢?

首先,很有意思的一点是,对于返回正确函数尾调用,一种情况下,return function()后面的代码不会运行,就是说当x()时,以及当local c=x()的时候。

我们如果想达到计数器的效果,一定要用local c=x(),然后反复调用c(),而不是调用x()。原因就是c实际上指向了x返回的匿名函数,这样每次运行c,实际上运行的就是那个匿名函数。另外一个有趣的结论是,对于返回正确尾调用的函数,尾调用实际上还没有运行。

运行的结果就是下面这样,你猜对了么?

aaa

=======

aaa

=======

aaa

=======

bbb

1

=======

bbb

2

=======

bbb

3

=======

aaa

=======

bbb

1

关于尾调用, return g(x)是尾调用,而return (g(x))就不是,原因是return (Fun()) — This expression is a single function call surrounded in parentheses, which is different than the required single function call (the parentheses adjust Fun to one return value).