首页 > 代码库 > Lua中的函数

Lua中的函数

【前言】

Lua中的函数和C++中的函数的含义是一致的,Lua中的函数格式如下:

function MyFunc(param)
     -- Do something
end

在调用函数时,也需要将对应的参数放在一对圆括号中,即使调用函数时没有参数,也必须写出一对空括号。对于这个规则只有一种特殊的例外情况:一个函数若只有一个参数,并且此参数是一个字符串或table构造式,那么圆括号便可以省略掉。看以下代码:

print "Hello World"          --> print("Hello World")等价
print [[a multi-line
          message]]          -->print([[a multi-line
                              -->               message]]) 等价
-- f是一个函数
f{x=10, y=20}               -->f({x=10, y=20}) 等价

上面代码的一些简便写法,如果不熟悉的话,在阅读别人的代码时,就会是一头雾水。

一个函数定义具有一个名称、一系列的参数和一个函数体。函数定义时,所定义的参数的使用方式与局部变量非常相似,它们是由调用函数时的“实际参数”初始化的。调用函数时提供的实参数量可以与形参数量不同。Lua会自动调整实参的数量,以匹配参数表的要求,若“实参多余形参,则舍弃多余的实参;若实参不足,则多余的形参初始化为nil”。这个与接下来要介绍的多重返回值非常相似。

 【多重返回值】

这个应该是Lua的一个特征吧。允许函数返回多个结果,只需要在return关键字后列出所有的返回值即可。以下根据带来来说明情况:

function foo0() end                         -- 无返回值
function foo1() return "a" end          -- 返回一个结果
function foo2() return "a", "b" end     -- 返回两个结果

-- 在多重赋值时,如果一个函数调用是最后,或仅有的一个表达式,
-- 那么Lua会保留其尽可能多的返回值,用于匹配赋值变量
x, y = foo2()               -- x = "a", y = "b"
x = foo2()                    -- x = "a", "b"被丢弃
x, y, z = 10, foo2()     -- x = 10, y = "a", z = "b"

-- 如果一个函数没有返回值或者没有足够多的返回值,那么Lua会用
-- nil来补充缺失的值
x, y = foo0()               -- x = nil, y = nil
x, y = foo1()               -- x = "a", y = nil
x, y, z = foo2()          -- x = "a", y = "b", z = nil

-- 如果一个函数调用不是一系列表达式的最后一个元素,那么将只产生一个值:
x, y = foo2(), 20          -- x = "a", y = 20
x, y = foo0(), 20, 30     -- x = nil, y = 20, 30则被丢弃

-- table构造式可以完整的接收一个函数调用的所有结果,即不会有任何数量
-- 方面的调整
local t = {foo0()}          -- t = {}(一个空的table)
local t = {foo1()}          -- t = {"a"}
local t = {foo2()}          -- t = {"a", "b"}

-- 但是,对于上述的行为,只有当一个函数调用作为最后一个元素时才会发生,
-- 而在其他位置上的函数调用总是只产生一个结果值
local t = {foo0(), foo2(), 4}          -- t[1] = nil, t[2] = "a", t[3] = 4

-- 我们也可以在一个函数中,使用return返回另一个函数的返回值
function MyFunc()          -- 返回a
     return foo1()          -- 注:这里是return foo1(),而不是return (foo1())
end

-- return foo1()和return (foo1())是两个完全不同的意思
-- 将一个函数调用放入一对圆括号中,从而迫使它只返回一个结果
print((foo0()))          -- nil
print((foo1()))          -- a
print((foo2()))          -- a

【变长参数】

在C语言中,函数可以接受不同数量的实参,Lua中的函数也可以接受不同数量的实参,例如以下代码:

-- 打印所有的参数
function VarArguments(...)
     for i, v in ipairs{...} do
          print(v)
     end
end

VarArguments(1, 2, 3)

参数表中的3个点(…)表示该函数可接受不同数量的实参。当这个函数被调用时,它的所有参数都会被收集到一起。这部分收集起来的实参称为这个函数的“变长参数”。一个函数要访问它的变长参数时,仍需要用到3个点(…)。但不同的是,此时这3个点是作为一个表达式来使用的。在上例中,表达式{…}表示一个由所有变长参数构成的数组。在C语言中使用变长参数需要注意的问题,在Lua中同样需要注意。

通常一个函数在遍历其变长参数时只需要使用表达式{…},这就像访问一个table一样,访问所有的变长参数。然而在某些特殊的情况下,变长参数中可能会包含一些故意传入的nil,那么此时就需要用select来访问变长参数了。调用select时,必须传入一个固定实参selector和一系列变长参数。如果selector为数字n,那么select返回它的第n个可变实参以及其后面的所有参数;否则selector只能为字符串“#”,这样select会返回变长参数的总数,请看以下代码:

for i = 1, select(#, ...) do
    local arg = select(i, ...) -- 得到第i个参数
    -- Do something else
end

select(‘#’, …)会返回所有变长参数的总数,其中包括nil(还记得table.maxn么?)对于Lua 5.0版本来说,变长参数则有另外一套机制。声明函数的语法是一样的,也是将3个点作为最后一个参数。但Lua 5.0没有提供“…”表达式。而是通过一个隐含的局部table变量“arg”来接受所有的变长参数。这个table还有一个名为“n”的字段,用来记录变长参数的总数,例如以下代码:

function MyFunc(a, b, ...)
     print(arg.n)
end

MyFunc(1, 2, 3, 4, 5)     -->3

这套旧机制的缺点在于,每当程序调用了一个具有变长参数的函数时,都会创建一个新的table。而在新机制中,只有在需要时才会去创建这个用于变长参数访问的table。这里只是对这个方法进行简单介绍,别在阅读别人的代码时,看不懂!!!

【深入讨论函数】

在Lua中,函数与其它传统类型的值具有相同的权利。函数可以存储到变量或table中,也可以作为实参传递给其它函数,还可以作为其它函数的返回值。在Lua中有一个容易混淆的概念是,函数与所有其它值一样都是匿名的,即它们都没有名称。当讨论一个函数名时,实际上是在讨论一个持有某函数的变量,例如以下代码:

-- 我们经常这样定义函数
function foo(x) return 2 * x end

-- 实际上,这只是一种“语法糖”而已;
-- 上述代码只是下面代码的一种简化书写形式
foo = function (x) return 2 * x end

实际上,一个函数定义实际就是一条语句(更准确地说是一条赋值语句),这条语句创建了一种类型为“函数”的值,并将这个值赋予一个变量。由于函数在Lua中就是一个普通的值,所以不仅可以将其存储在全局变量中,还可以存储在局部变量甚至table的字段中。

【内嵌函数】

若将一个函数写在另一个函数之内,那么这个位于内部的函数便可以访问外部函数中的局部变量,这个特征叫做“词法域”。我们来看看下面一段有趣的代码:

function newCounter()
     local i = 0
     return function () -- 匿名函数
          i = i + 1
          return i
     end
end

c1 = newCounter()
print(c1())     -->输出什么?
print(c1())     -->又输出什么?

如果你很明白上面的输出,很明白上面的代码,那么闭合函数这一小节就不需要阅读了。在上述代码中,有一个变量i,对于函数newCounter来说,i是一个局部变量,但是对于匿名函数来说,当它访问这个i时,i既不是全局变量,也不是局部变量,对于我们来说,我们称这样的变量为一个“非局部的变量”。下面这段代码也是同样的道理:

function newCounter(i)
     return function () -- 匿名函数
          i = i + 1
          return i
     end
end

c1 = newCounter(10)
print(c1())     -->输出什么?
print(c1())     -->又输出什么?

匿名函数访问了一个“非局部的变量”i,该变量用于保持一个计数器。乍一看,由于创建变量i的函数,也就是newCounter已经返回,所以之后每次调用匿名函数时,i都应该是已经超出了作用范围。但是,Lua会以closure的概念来正确地处理这种情况。在这里简单的讲,一个closure就是一个函数加上该函数所需访问的所有“非局部的变量”。如果再次调用newCounter,那么它会创建一个新的局部变量i,从而将得到一个新的closure。在后续的总结中,专门总结一篇关于Lua中的闭包的博文,敬请期待。

【非全局的函数】

由于函数和普通变量一样,所以函数不仅可以存储在全局变量中,还可以存储在table的字段中,或局部变量中。我们可以把函数存在一个table中,比如以下代码:

Lib = {}
Lib.foo = function (x, y) return x + y end
Lib.goo = function (x, y) return x - y end

只要将一个函数存储在一个局部变量中,就得到了一个“局部函数”,也就是说这个函数只能在某个特定的作用域内才有效。我们可以这样定义一个局部的函数:

local f = function (<参数>)
     <函数体>
end
-- Lua还提供另一种特殊的“语法糖”
local function f (<参数>)
     <函数体>
end

有的时候,我们需要进行函数的前置声明,比如以下代码:

local f, g

function f()
     <一些其它操作>
     g()
end

function g()
     <一些其它操作>
     f()
end

 

Lua中的函数