213 satır
7.2 KiB

  1. /*
  2. ** $Id: lauxlib.h,v 1.120 2011/11/29 15:55:08 roberto Exp $
  3. ** Auxiliary functions for building Lua libraries
  4. ** See Copyright Notice in lua.h
  5. */
  6. #ifndef lauxlib_h
  7. #define lauxlib_h
  8. #include <stddef.h>
  9. #include <stdio.h>
  10. #include "lua.h"
  11. /* extra error code for `luaL_load' */
  12. #define LUA_ERRFILE (LUA_ERRERR+1)
  13. typedef struct luaL_Reg {
  14. const char *name;
  15. lua_CFunction func;
  16. } luaL_Reg;
  17. LUALIB_API void (luaL_checkversion_) (lua_State *L, lua_Number ver);
  18. #define luaL_checkversion(L) luaL_checkversion_(L, LUA_VERSION_NUM)
  19. LUALIB_API int (luaL_getmetafield) (lua_State *L, int obj, const char *e);
  20. LUALIB_API int (luaL_callmeta) (lua_State *L, int obj, const char *e);
  21. LUALIB_API const char *(luaL_tolstring) (lua_State *L, int idx, size_t *len);
  22. LUALIB_API int (luaL_argerror) (lua_State *L, int numarg, const char *extramsg);
  23. LUALIB_API const char *(luaL_checklstring) (lua_State *L, int numArg,
  24. size_t *l);
  25. LUALIB_API const char *(luaL_optlstring) (lua_State *L, int numArg,
  26. const char *def, size_t *l);
  27. LUALIB_API lua_Number (luaL_checknumber) (lua_State *L, int numArg);
  28. LUALIB_API lua_Number (luaL_optnumber) (lua_State *L, int nArg, lua_Number def);
  29. LUALIB_API lua_Integer (luaL_checkinteger) (lua_State *L, int numArg);
  30. LUALIB_API lua_Integer (luaL_optinteger) (lua_State *L, int nArg,
  31. lua_Integer def);
  32. LUALIB_API lua_Unsigned (luaL_checkunsigned) (lua_State *L, int numArg);
  33. LUALIB_API lua_Unsigned (luaL_optunsigned) (lua_State *L, int numArg,
  34. lua_Unsigned def);
  35. LUALIB_API void (luaL_checkstack) (lua_State *L, int sz, const char *msg);
  36. LUALIB_API void (luaL_checktype) (lua_State *L, int narg, int t);
  37. LUALIB_API void (luaL_checkany) (lua_State *L, int narg);
  38. LUALIB_API int (luaL_newmetatable) (lua_State *L, const char *tname);
  39. LUALIB_API void (luaL_setmetatable) (lua_State *L, const char *tname);
  40. LUALIB_API void *(luaL_testudata) (lua_State *L, int ud, const char *tname);
  41. LUALIB_API void *(luaL_checkudata) (lua_State *L, int ud, const char *tname);
  42. LUALIB_API void (luaL_where) (lua_State *L, int lvl);
  43. LUALIB_API int (luaL_error) (lua_State *L, const char *fmt, ...);
  44. LUALIB_API int (luaL_checkoption) (lua_State *L, int narg, const char *def,
  45. const char *const lst[]);
  46. LUALIB_API int (luaL_fileresult) (lua_State *L, int stat, const char *fname);
  47. LUALIB_API int (luaL_execresult) (lua_State *L, int stat);
  48. /* pre-defined references */
  49. #define LUA_NOREF (-2)
  50. #define LUA_REFNIL (-1)
  51. LUALIB_API int (luaL_ref) (lua_State *L, int t);
  52. LUALIB_API void (luaL_unref) (lua_State *L, int t, int ref);
  53. LUALIB_API int (luaL_loadfilex) (lua_State *L, const char *filename,
  54. const char *mode);
  55. #define luaL_loadfile(L,f) luaL_loadfilex(L,f,NULL)
  56. LUALIB_API int (luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz,
  57. const char *name, const char *mode);
  58. LUALIB_API int (luaL_loadstring) (lua_State *L, const char *s);
  59. LUALIB_API lua_State *(luaL_newstate) (void);
  60. LUALIB_API int (luaL_len) (lua_State *L, int idx);
  61. LUALIB_API const char *(luaL_gsub) (lua_State *L, const char *s, const char *p,
  62. const char *r);
  63. LUALIB_API void (luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
  64. LUALIB_API int (luaL_getsubtable) (lua_State *L, int idx, const char *fname);
  65. LUALIB_API void (luaL_traceback) (lua_State *L, lua_State *L1,
  66. const char *msg, int level);
  67. LUALIB_API void (luaL_requiref) (lua_State *L, const char *modname,
  68. lua_CFunction openf, int glb);
  69. /*
  70. ** ===============================================================
  71. ** some useful macros
  72. ** ===============================================================
  73. */
  74. #define luaL_newlibtable(L,l) \
  75. lua_createtable(L, 0, sizeof(l)/sizeof((l)[0]) - 1)
  76. #define luaL_newlib(L,l) (luaL_newlibtable(L,l), luaL_setfuncs(L,l,0))
  77. #define luaL_argcheck(L, cond,numarg,extramsg) \
  78. ((void)((cond) || luaL_argerror(L, (numarg), (extramsg))))
  79. #define luaL_checkstring(L,n) (luaL_checklstring(L, (n), NULL))
  80. #define luaL_optstring(L,n,d) (luaL_optlstring(L, (n), (d), NULL))
  81. #define luaL_checkint(L,n) ((int)luaL_checkinteger(L, (n)))
  82. #define luaL_optint(L,n,d) ((int)luaL_optinteger(L, (n), (d)))
  83. #define luaL_checklong(L,n) ((long)luaL_checkinteger(L, (n)))
  84. #define luaL_optlong(L,n,d) ((long)luaL_optinteger(L, (n), (d)))
  85. #define luaL_typename(L,i) lua_typename(L, lua_type(L,(i)))
  86. #define luaL_dofile(L, fn) \
  87. (luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
  88. #define luaL_dostring(L, s) \
  89. (luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
  90. #define luaL_getmetatable(L,n) (lua_getfield(L, LUA_REGISTRYINDEX, (n)))
  91. #define luaL_opt(L,f,n,d) (lua_isnoneornil(L,(n)) ? (d) : f(L,(n)))
  92. #define luaL_loadbuffer(L,s,sz,n) luaL_loadbufferx(L,s,sz,n,NULL)
  93. /*
  94. ** {======================================================
  95. ** Generic Buffer manipulation
  96. ** =======================================================
  97. */
  98. typedef struct luaL_Buffer {
  99. char *b; /* buffer address */
  100. size_t size; /* buffer size */
  101. size_t n; /* number of characters in buffer */
  102. lua_State *L;
  103. char initb[LUAL_BUFFERSIZE]; /* initial buffer */
  104. } luaL_Buffer;
  105. #define luaL_addchar(B,c) \
  106. ((void)((B)->n < (B)->size || luaL_prepbuffsize((B), 1)), \
  107. ((B)->b[(B)->n++] = (c)))
  108. #define luaL_addsize(B,s) ((B)->n += (s))
  109. LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
  110. LUALIB_API char *(luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
  111. LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
  112. LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
  113. LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
  114. LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
  115. LUALIB_API void (luaL_pushresultsize) (luaL_Buffer *B, size_t sz);
  116. LUALIB_API char *(luaL_buffinitsize) (lua_State *L, luaL_Buffer *B, size_t sz);
  117. #define luaL_prepbuffer(B) luaL_prepbuffsize(B, LUAL_BUFFERSIZE)
  118. /* }====================================================== */
  119. /*
  120. ** {======================================================
  121. ** File handles for IO library
  122. ** =======================================================
  123. */
  124. /*
  125. ** A file handle is a userdata with metatable 'LUA_FILEHANDLE' and
  126. ** initial structure 'luaL_Stream' (it may contain other fields
  127. ** after that initial structure).
  128. */
  129. #define LUA_FILEHANDLE "FILE*"
  130. typedef struct luaL_Stream {
  131. FILE *f; /* stream (NULL for incompletely created streams) */
  132. lua_CFunction closef; /* to close stream (NULL for closed streams) */
  133. } luaL_Stream;
  134. /* }====================================================== */
  135. /* compatibility with old module system */
  136. #if defined(LUA_COMPAT_MODULE)
  137. LUALIB_API void (luaL_pushmodule) (lua_State *L, const char *modname,
  138. int sizehint);
  139. LUALIB_API void (luaL_openlib) (lua_State *L, const char *libname,
  140. const luaL_Reg *l, int nup);
  141. #define luaL_register(L,n,l) (luaL_openlib(L,(n),(l),0))
  142. #endif
  143. #endif