Lua函数之二

 

Lua中函数的两个重要特性：

1、函数和其他类型（如number、string）一样，可以存放在变量中，也可以存放在table中，可以作为函数的参数，还可以作为函数的返回值。

2、嵌套的函数可以访问其外部函数中的局部变量——闭包。

 

例如：

local foo = function(x) return x^2 end          -- 函数构造式local function foo(x) return x^2 end             -- 函数定义

第1种方法创建一个函数，并赋一个变量；

第2种方法其实是一种特例。

 

1、非全局函数

Lua中的函数可以作为全局变量，也可以作为局部变量。

将函数保存在一个局部变量时，得到一个局部函数：

local f = function(...)     ... endlocal g = function(...)     ...        f()    -- external local 'f' is visible here    ... end

 

 

函数作为table的域，这种情况下，必须注意函数和表语法：

第1种方式：

lib = {add = function(x,y) return x+y end, sub = function(x,y) return x-y  end}

 

第2种方式：

lib = {}lib.add = function(x,y) return x+y endlib.sub = function(x,y) return x-y  end

 

第3种方式：

lib = {}function lib.add(x,y) return x+y endfunction lib.sub(x,y) return x-y  end

 

调用方式：

lib.add(x,y)

 

2、尾调用

（tail recursion）是指函数最后一个动作是调用另外一个函数，例如

function f(x)         return g(x)                 // 尾调用end

 这种情况下，当被调用函数g结束时程序不需要返回到调用者f，所以尾调用之后程序不需要在栈中保留关于调用者的任何信息。

Lua解释器利用这个特性在处理尾调用时不使用额外的栈，那么尾调用递归的层次是可以无限制的。例如：

function foo (n)         if n > 0 then return foo(n-1) endend

不管n的值有多大，都不会发生栈溢出。

 

注意：下面的返回方法不是尾调用：

return g(x)+1   　　　　 -- must do the additionreturn x or g(x)    　　-- must adjust to 1 resultreturn (g(x))    　　　　-- must adjust to 1 result

 

3、闭包

闭包：当一个函数内部嵌套另一个函数定义时，内部的函数体可以访问外部函数的局部变量。

例如：

function foo()    local i = 0    return function() i = i + 1 return i endend c1 = foo()c2 = foo()print(c1())      -- 1print(c1())      -- 2print(c2())      -- 1

 

例子中foo函数的内部匿名函数可以访问foo的局部变量i，在匿名函数内部i是external local variable（也称作upvalue）。

c1和c2是建立在同一个函数上，但作用在同一个局部变量的不同实例上的两个不同的闭包。简单的说，闭包是一个匿名函数及其upvalues 。

 

4、迭代器

迭代器是一种支持指针类型的结构，它可以遍历集合的每一个元素。

在Lua中使用函数来描述迭代器，每次调用该函数就返回集合的下一个元素。

迭代器需要保留上一次成功调用的状态和下一次成功调用的状态，闭包提供的机制可以很容易实现这个任务。

 

闭包是一个内部函数，它可以访问一个或者多个外部函数的external local variable，每次闭包的成功调用后，这些upvalues都会保存他们当前的状态。

一个典型的闭包结构包含两个函数：一个是闭包自己，另一个是工厂（创建闭包的外部函数）。 

 

用闭包实现数组迭代器的例子：

function list_iter(t)    local i=0    local n=#t    return function()        i=i+1        if i<=n then return t[i] end    endend a={
   1,2,3}iter=list_iter(a)　　-- 创建迭代器，保存的外部局部变量(t,i,n)while true do    local element=iter()    if element==nil then break end    print(element)   end

上面设计的迭代器也可以适用于泛型for循环：

for element in list_iter(a) do    print(element)end

泛型for循环首先调用迭代工厂，并在内部保留迭代函数，因此我们不需要iter变量；然后在每一个新的迭代处调用迭代器函数，当迭代器返回nil时循环结束。

 

另一个例子，逐行遍历所有的单词：

function allwords()    local line = io.read()    local pos = 1    return function()        while line do            local s,e=string.find(line, "%w+", pos)            if s then                pos = e+1                return string.sub(line, s, e)            else                line=io.read()                pos=1            end        end        return nil    endend for word in allwords() do    print(word)end

 

上面提到的迭代器的名字可能有些误导，因为它并没有迭代，完成迭代功能的是for循环语句，也许更好的叫法应该是生成器（generator）。

下面是一个真正的迭代器的例子，它在内部完成了循环，这样我们使用迭代器的时候就不需要再使用循环了，迭代器仅接受一个函数作为参数，并且这个函数在迭代器内部被调用。

function allwords(f)    for l in io.lines() do        for w in string.gfind(l, "%w+") do            f(w)        end    endend allwords(print)

更一般的做法是使用匿名函数作为参数，下面的例子打印出单词"hello"出现的次数：

local count = 0allwords(function(w)         if w=="hello" then count=count+1 end         end )print(count)

 

 

5、泛型for的语义

泛型for在自己内部保存迭代函数，包括三个值：迭代函数、状态常量、控制变量。

for 
     
       in 
      
        do    end
      
     

其中，<var-list>是一个或多个逗号分隔的变量名列表，且第一个变量为控制变量，其值为nil时循环结束；

<exp-list>是以一个或多个逗号分隔的表达式列表，通常情况下exp-list只有一个值：迭代工厂的调用。

 

泛型for的执行过程：

1、初始化，计算in后面表达式的值，表达式应该返回三个值：迭代函数、状态常量、控制变量；

2、将状态常量和控制变量作为参数调用迭代函数；

3、将迭代函数返回值赋给变量列表；

4、如果控制变量为nil，循环结束；

5、回到第2步；

 

LUA标准库提供了几种迭代器，包括迭代文件每行的(io.lines)，迭代table元素的(pairs)，迭代数组元素的(ipairs)，迭代字符串中单词的(string.gmatch)等。

以iparis为例，实现如下：

function iter(a, i)    i = i + 1    local v=a[i]    if v then return i,v endend function ipairs(a)    return iter, a, 0end

当泛型for循环调用iparis(a)开始循环时，它获取三个值：迭代函数iter、状态常量a、控制变量初始值0；

然后调用iter(a,0)返回1,a[1]，第二次迭代调用iter(a,1)返回2,a[2]，...，直到第一个非nil元素。

 

pairs函数的实现可以利用next方法：

function pairs(t)    return next,t,nilend

 

也可以直接使用next方法来迭代一个table：

lst = {
   100,250,x="bj",37,"chen", 4,y="qi"}for k,v in next,lst do    print(k,v)end

 

OK，上文提到的iterator都是无状态的迭代器，每一次迭代，迭代函数都是用两个常量（状态常量和控制变量）的值作为参数被调用。

在有些情况下，迭代器需要保存多个状态信息而不是简单的状态常量和控制变量，最简单的方法是使用闭包，

还有一种方法是将所有的状态信息封装到table内，将table作为迭代器的状态常量。

 

allwords的另一个实现版本：

local iteratorfunction allwords()    local state={line=io.read(), pos=1}    return iterator, stateend function iterator(state)    while state.line do        local s,e=string.find(state.line, "%w+", state.pos)        if s then            state.pos = e+1            return string.sub(state.line, s,e)        else            state.line = io.read()            state.pos = 1        end    end    return nilend