mở rộng Lua: kiểm tra số tham số được chuyển đến hàm

Question

Tôi muốn tạo hàm Lua mới.

Tôi có thể sử dụng chức năng với tham số (Tôi đang theo dõi this link) để đọc các tham số chức năng.

static int idiv(lua_State *L) { 
    int n1 = lua_tointeger(L, 1); /* first argument */ 
    int n2 = lua_tointeger(L, 2); /* second argument */ 
    int q = n1/n2; int r = n1 % n2; 
    lua_pushinteger(L, q); /* first return value */ 
    lua_pushinteger(L, r); /* second return value */ 
    return 2; /* return two values */ 
} 

Tôi muốn biết nếu có một cách nào để biết số lượng các tham số truyền cho một chức năng, để in một tin nhắn nếu người dùng không gọi hàm với hai tham số.

Tôi muốn thực hiện chức năng khi người dùng viết

idiv(3, 4) 

và in một lỗi khi

idiv(2) 
idiv(3,4,5) 
and so on... 

Answer 1

Bạn có thể sử dụng lua_gettop() để xác định số lượng đối số truyền cho một hàm C Lua:

int lua_gettop (lua_State *L);
Trả về chỉ mục của phần tử trên cùng trong ngăn xếp. Vì các chỉ số bắt đầu ở mức 1, kết quả này bằng với số phần tử trong ngăn xếp (và vì vậy 0 có nghĩa là một ngăn xếp trống).

static int idiv(lua_State *L) { 
    if (lua_gettop(L) != 2) { 
    return luaL_error(L, "expecting exactly 2 arguments"); 
    } 
    int n1 = lua_tointeger(L, 1); /* first argument */ 
    int n2 = lua_tointeger(L, 2); /* second argument */ 
    int q = n1/n2; int r = n1 % n2; 
    lua_pushinteger(L, q); /* first return value */ 
    lua_pushinteger(L, r); /* second return value */ 
    return 2; /* return two values */ 
}