Lua教程(五):C/C++操作Lua數組和字串樣本_Lua

來源:互聯網
上載者:User

本文將介紹如何在C/C++裡面操作Lua的數組和字串類型,同時還會介紹如何在C/C++函數裡面儲存Lua狀態(registry和upvalue),而registry在使用C/C++自訂類型時非常有用,可以方便地為userdata指定metatable。

C/C++操作Lua數組

Lua數組Overview

在Lua裡面,數組只不過是key為整數的table而已。比如一個table為array = {12,”Hello”, “World”},它是一個數組,可以用下面的代碼來訪問它:

複製代碼 代碼如下:

print(array[1])  --這裡會輸出array的第一個元素12。
print(array[3]) --這裡會輸出array的第三個元素World

需要注意的一點就是:Lua的數組的下標是從1開始的。如果你使用下面的語句則會輸出nil值:

複製代碼 代碼如下:

print(array[0])  --輸出nil
print(array["1"])  --輸出nil(想想和array[1]的區別:一個是integer作為key,一個是字串做為key)

通用Table操作方法

之前我們在教程1中介紹了如何傳遞Table給Lua,以及在教程3中介紹了如何訪問Table的資料。因為數組也是Table,所以我們可以用同樣的方式來讀取數組。

讀取數組

假設我們的Lua Table為array = {“Hello”, 1, “World”, 23.2},那麼我們可以用下列函數來訪問它:

複製代碼 代碼如下:

void readLuaArray(lua_State *L)
{
    lua_settop(L,0); //這樣確保我們的array是放在當前棧的棧頂。
    lua_getglobal(L, "array");
    //如果前面不調用lua_settop(L,0),那我們必須要使用luaL_len(L,-1)
    int n = luaL_len(L, 1);   //luaL_len可以獲得table的元素個數
    for (int i = 1; i <= n; ++i) {
        lua_pushnumber(L, i);  //往棧裡面壓入i
        lua_gettable(L, -2);  //讀取table[i],table位於-2的位置。
        //lua_rawget(L, -2);  //lua_gettable也可以用lua_rawget來替換
        cout<<lua_tostring(L, -1)<<endl;
        lua_pop(L, 1);
    }
}

最後輸出的結果為:

複製代碼 代碼如下:

"Hello", 1, "World", 23.2

修改數組

現在我們如果想要修改這個數組,把每一個數組的元素都變成”hehe[i]”(i = 1-n),我們看看怎麼做。

複製代碼 代碼如下:

int writeLuaArray(lua_State *L)
{
    lua_settop(L, 0);
    lua_getglobal(L, "array");
    //確保第一個函數一個要是一個table
    luaL_checktype(L, 1, LUA_TTABLE);
    int n = luaL_len(L,1);
    for (int i = 1; i <= n; ++i) {
        lua_pushnumber(L, i);
        char buf[256];
        sprintf(buf, "hehe%d", i);
        lua_pushstring(L, buf);
//        lua_settable(L, -3);
        lua_rawset(L, -3);
    }
    return 0;
}
}

注意這裡的lua_rawset和lua_settable是等價的,只不過lua_rawset速度更快。 最後,我們在載入完Lua指令碼以後調用這兩個函數:

複製代碼 代碼如下:

writeLuaArray(L);
readLuaArray(L);

輸出結果為:

複製代碼 代碼如下:

readLuaArray: hehe1
readLuaArray: hehe2
readLuaArray: hehe3
readLuaArray: hehe4

專門的數組操作方法

因為數組一般在程式語言裡面都會被特殊對待,Lua也不例外,它的C API還提供另外一種更方便高效地方法來存取數組的元素。

複製代碼 代碼如下:

 void lua_rawgeti (lua_State *L, int index, int key);
 void lua_rawseti (lua_State *L, int index, int key);

這兩個函數後面兩個參數的意思分別是:index(table在棧中的索引),key(table中數組的索引,下標從1開始) 接下來,我會通過改造上面的樣本來示範這兩個API的用法。

讀取數組

因為lua_rawgeti(L,t,key)等價於:

複製代碼 代碼如下:

 lua_pushnumber(L, key);
 lua_rawget(L, t);

因此,我們的讀取代碼可以改寫成下面這樣:

複製代碼 代碼如下:

void readLuaArray(lua_State *L)
{
    lua_getglobal(L, "array");
    int n = luaL_len(L, -1);
    for (int i = 1; i <= n; ++i) {
        lua_rawgeti(L, 1, i);
        cout<<"readLuaArray: "<<lua_tostring(L, -1)<<endl;
        lua_pop(L, 1);
    }
}

修改數組

同理,lua_rawset(L,t,key)等價於

複製代碼 代碼如下:

lua_pushnumber(L,key); //此時的棧 table->value->key
lua_insert(L,-2);  //調用完後的棧: table->key->value (table[key]=value)
lua_rawset(L,t);

相應的修改數組的代碼可以修改為:

複製代碼 代碼如下:

int writeLuaArray(lua_State *L)
{
    lua_settop(L, 0);
    lua_getglobal(L, "array");
    //確保第一個函數一個要是一個table
    luaL_checktype(L, 1, LUA_TTABLE);
    int n = luaL_len(L,1);
    for (int i = 1; i <= n; ++i) {
        char buf[256];
        sprintf(buf, "hehe%d", i);
        lua_pushstring(L, buf);
        lua_rawseti(L, 1, i);
    }
    return 0;
}

C/C++操作Lua字串

基底字元串操作

Lua C API操作字串主要包含兩個操作:求子串(lua_pushlstring)和字串拼接(lua_concat). 例如,我們求一個字串s的子串[i,j],它可以表示為:

複製代碼 代碼如下:

lua_pushlstring(L, s + i, j - i + 1);

而lua_concat(L,n)則可以把當前棧頂的n個元素轉換成字串並拼接起來,最後把結果壓入棧頂。 比如,我們想定義一個函數mycontact(…,n)可以把n個字串拼接起來,n表示字串的個數,那麼我們的代碼可以寫成這樣:
複製代碼 代碼如下:

static int l_mycontact(lua_State* L){
    luaL_checktype(L, -1, LUA_TNUMBER);
    int n = lua_tonumber(L, -1);
    lua_pop(L, 1);
    lua_concat(L, n);
    return 1;
}

然後,我們需要註冊此函數到libs中去,最後在Lua裡面調用此函數:

複製代碼 代碼如下:

print(mylib.mycontact("zilong","shanren"," meng meng"," da",4))

輸出結果為:

複製代碼 代碼如下:

zilongshanren meng meng da

格式化輸出

當我們想要往Lua裡面寫入一個格式化字串時,可以使用函數

複製代碼 代碼如下:

const char *lua_pushfstring (lua_State *L, const char *fmt, ...);

另外,我們還可以使用luaL_Buffer,下面是PIL書中的樣本,把Lua字串轉換成大寫:

複製代碼 代碼如下:

 static int str_upper (lua_State *L) {
     size_t l;
     size_t i;
     luaL_Buffer b;
     const char *s = luaL_checklstring(L, 1, &l);  //從Lua棧中取出字串
     char *p = luaL_buffinitsize(L, &b, l); //分配一塊與取出字串同樣大小的緩衝區
     for (i = 0; i < l; i++)
       p[i] = toupper(uchar(s[i]));
     luaL_pushresultsize(&b, l);  //把緩衝區結果轉換為字串
     return 1;
}

更多的Lua Buffer操作函數如下:

複製代碼 代碼如下:

 void luaL_buffinit   (lua_State *L, luaL_Buffer *B);
 void luaL_addvalue   (luaL_Buffer *B);
 void luaL_addlstring (luaL_Buffer *B, const char *s, size_t l);
 void luaL_addstring  (luaL_Buffer *B, const char *s);
 void luaL_addchar    (luaL_Buffer *B, char c);
 void luaL_pushresult (luaL_Buffer *B);

關於每一個函數的用法和每一個參數的含義,大家可以去Lua的Reference Manual上去查看,本文就不贅述了。

儲存Lua狀態

在C函數裡面,當我們需要儲存函數裡面的一些狀態的時候,我們一般採用全域變數或者靜態變數的方式。但是,如果在與Lua互動時,這兩種方法都不可取。 原因有二: 1. C變數很難儲存各種各樣的Lua變數。 2. 當存在多個Lua棧的時候,就不生效了。 在Lua裡面有兩種方法來存在函數內的non-local資料:registry和upvalue.

Registry方式

Register是一個Lua的全域Table,只有在Lua的C API裡面可以訪問這個Table。它可以用來儲存多個Lua模組之間的資料。 訪問Register的方式一般為:

複製代碼 代碼如下:

 lua_getfield(L, LUA_REGISTRYINDEX, "Key");

我們需要提供一個LUA_REGISTRYINDEX的“虛擬索引”來標識它在Lua棧中的位置。我們在操作這個table的時候,最好是使用字串做為key,而不要使用數字來做為key。關於Registry更為實際的用法,我們會在下一篇文章中討論。

Upvalue方式

Upvalue主要用來儲存模組或者函數內部的一些私人的資料,它與C語言的靜態變數有點類似。具體的用法可以參考PIL

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.