create_lua_library.py 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856
  1. import sys
  2. import CppHeaderParser
  3. import os
  4. import errno
  5. import re
  6. from zipfile import *
  7. import fnmatch
  8. import re
  9. def mkdir_p(path): # Same effect as mkdir -p, create dir and all necessary parent dirs
  10. try:
  11. os.makedirs(path)
  12. except OSError as e:
  13. if e.errno == errno.EEXIST: # Dir already exists; not really an error
  14. pass
  15. else: raise
  16. def template_returnPtrLookupArray(prefix, className, ptr):
  17. out = "%sif %s == nil then return nil end\n" % (prefix, ptr)
  18. out += "%sfor i=1,count(%s) do\n" % (prefix, ptr)
  19. out += "%s\tlocal __c = _G[%s](\"__skip_ptr__\")\n" % (prefix, className.replace("*", ""))
  20. out += "%s\t__c.__ptr = %s[i]\n" % (prefix, ptr)
  21. out += "%s\t%s[i] = __c\n" % (prefix, ptr)
  22. out += "%send\n" % (prefix)
  23. out += "%sreturn %s\n" % (prefix,ptr)
  24. return out
  25. # Note we expect className to be a valid string
  26. def template_returnPtrLookup(prefix, className, ptr):
  27. out = "%sif %s == nil then return nil end\n" % (prefix, ptr)
  28. out += "%slocal __c = _G[%s](\"__skip_ptr__\")\n" % (prefix, className.replace("*", ""))
  29. out += "%s__c.__ptr = %s\n" % (prefix, ptr)
  30. out += "%sreturn __c\n" % (prefix)
  31. return out
  32. def template_quote(str):
  33. return "\"%s\"" % str;
  34. def cleanDocs(docs):
  35. return docs.replace("/*", "").replace("*/", "").replace("*", "").replace("\n", "").replace("\r", "").replace("::", ".").replace("\t", "")
  36. def toLuaType(t):
  37. return t.replace("void", "nil").replace("int", "Integer").replace("bool", "Boolean").replace("*", "")
  38. # FIXME: Some "unsigned int *" functions are still being generated on the polycode API?
  39. def typeFilter(ty):
  40. ty = ty.replace("Polycode::", "")
  41. ty = ty.replace("std::", "")
  42. ty = ty.replace("const", "")
  43. ty = ty.replace("inline", "")
  44. ty = ty.replace("static", "")
  45. ty = ty.replace("virtual", "")
  46. ty = ty.replace("&", "")
  47. ty = re.sub(r'^.*\sint\s*$', 'int', ty) # eg "unsigned int"
  48. ty = re.sub(r'^.*\schar\s*$', 'char', ty) # eg "unsigned int"
  49. ty = re.sub(r'^.*\slong\s*$', 'int', ty)
  50. ty = re.sub(r'^.*\swchar_t\s*$', 'int', ty)
  51. ty = re.sub(r'^.*\sshort\s*$', 'int', ty)
  52. ty = re.sub(r'^.*\sfloat\s*$', 'Number', ty)
  53. ty = re.sub(r'^.*\sdouble\s*$', 'Number', ty) # eg "long double"
  54. ty = ty.replace("unsigned", "int")
  55. ty = ty.replace("long", "int")
  56. ty = ty.replace("float", "Number")
  57. ty = ty.replace("double", "Number")
  58. ty = ty.replace(" ", "") # Not very safe!
  59. return ty
  60. def createLUABindings(inputPath, prefix, mainInclude, libSmallName, libName, apiPath, apiClassPath, includePath, sourcePath, luaDocPath, inheritInModuleFiles):
  61. wrappersHeaderOut = "" # Def: Global C++ *LUAWrappers.h
  62. cppRegisterOut = "" # Def: Global C++ *LUA.cpp
  63. cppLoaderOut = "" # Def: Global C++ *LUA.cpp
  64. luaDocOut = ""
  65. luaIndexOut = "" # Def: Global Lua everything-gets-required-from-this-file file
  66. # Header boilerplate for wrappersHeaderOut and cppRegisterOut
  67. cppRegisterOut += "#include \"%sLUA.h\"\n" % (prefix)
  68. cppRegisterOut += "#include \"%sLUAWrappers.h\"\n" % (prefix)
  69. cppRegisterOut += "#include \"PolyCoreServices.h\"\n\n"
  70. cppRegisterOut += "using namespace Polycode;\n\n"
  71. cppRegisterOut += "int luaopen_%s(lua_State *L) {\n" % (prefix)
  72. if prefix != "Polycode" and prefix != "Physics2D" and prefix != "Physics3D" and prefix != "UI":
  73. cppRegisterOut += "CoreServices *inst = (CoreServices*) *((PolyBase**)lua_touserdata(L, 1));\n"
  74. cppRegisterOut += "CoreServices::setInstance(inst);\n"
  75. cppRegisterOut += "\tstatic const struct luaL_reg %sLib [] = {" % (libSmallName)
  76. wrappersHeaderOut += "#pragma once\n\n"
  77. wrappersHeaderOut += "extern \"C\" {\n\n"
  78. wrappersHeaderOut += "#include <stdio.h>\n"
  79. wrappersHeaderOut += "#include \"lua.h\"\n"
  80. wrappersHeaderOut += "#include \"lualib.h\"\n"
  81. wrappersHeaderOut += "#include \"lauxlib.h\"\n"
  82. wrappersHeaderOut += "#undef near\n"
  83. wrappersHeaderOut += "#undef far\n"
  84. wrappersHeaderOut += "} // extern \"C\" \n\n"
  85. luaDocOut += "<?xml version=\"1.0\" ?>\n"
  86. luaDocOut += "<docs>\n"
  87. luaDocOut += "<classes>\n"
  88. # Get list of headers to create bindings from
  89. inputPathIsDir = os.path.isdir(inputPath)
  90. if inputPathIsDir:
  91. files = os.listdir(inputPath)
  92. else:
  93. files = []
  94. with open(inputPath) as f:
  95. for line in f.readlines():
  96. files.append(line.strip()) # Strip whitespace, path/
  97. filteredFiles = []
  98. for fileName in files:
  99. if inputPathIsDir:
  100. fileName = "%s/%s" % (inputPath, fileName)
  101. if os.path.isdir(fileName):
  102. continue
  103. head, tail = os.path.split(fileName)
  104. ignore = ["PolyTween", "PolyTweenManager", "PolyGLSLProgram", "PolyGLSLShader", "PolyGLSLShaderModule", "PolyWinCore", "PolyCocoaCore", "PolyAGLCore", "PolySDLCore", "Poly_iPhone", "PolyGLES1Renderer", "PolyGLRenderer", "tinyxml", "tinystr", "OpenGLCubemap", "PolyiPhoneCore", "PolyGLES1Texture", "PolyGLTexture", "PolyGLVertexBuffer", "PolyThreaded", "PolyGLHeaders", "GLee", "PolyPeer", "PolySocket", "PolyClient", "PolyServer", "PolyServerWorld", "OSFILE", "OSFileEntry", "OSBasics", "PolyLogger", "PolyFontGlyphSheet"]
  105. if tail.split(".")[1] == "h" and tail.split(".")[0] not in ignore:
  106. filteredFiles.append(fileName)
  107. wrappersHeaderOut += "#include \"%s\"\n" % (tail)
  108. wrappersHeaderOut += "\nusing namespace std;\n\n"
  109. wrappersHeaderOut += "\nnamespace Polycode {\n\n"
  110. # list of classes that don't get the garbage collection in their meta table
  111. disable_gc = ["Entity","SceneLabel", "SceneMesh", "Scene", "Texture", "Image", "Camera", "SceneParticleEmitter", "Mesh", "Vertex", "Polygon", "Polycode::Polygon", "Material", "ScenePrimitive", "SceneLine", "SceneLight", "SceneSound", "SceneImage", "SceneEntity", "SceneEntityInstance", "SceneSprite"]
  112. # Special case: If we are building the Polycode library itself, inject the LuaEventHandler class.
  113. # Note: so that event callbacks can work, any object inheriting from EventHandler will secretly
  114. # be modified to actually inherit from LuaEventHandler instead.
  115. if prefix == "Polycode":
  116. wrappersHeaderOut += "class LuaEventHandler : public EventHandler {\n"
  117. wrappersHeaderOut += "public:\n"
  118. wrappersHeaderOut += " LuaEventHandler() : EventHandler() {}\n"
  119. wrappersHeaderOut += " void handleEvent(Event *e) {\n"
  120. wrappersHeaderOut += " lua_getfield (L, LUA_GLOBALSINDEX, \"__customError\");\n"
  121. wrappersHeaderOut += " int errH = lua_gettop(L);\n"
  122. wrappersHeaderOut += " lua_getfield(L, LUA_GLOBALSINDEX, \"__handleEvent\");\n"
  123. wrappersHeaderOut += " lua_rawgeti( L, LUA_REGISTRYINDEX, wrapperIndex );\n"
  124. wrappersHeaderOut += " PolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  125. wrappersHeaderOut += " *userdataPtr = (PolyBase*)e;\n"
  126. wrappersHeaderOut += " lua_pcall(L, 2, 0, errH);\n"
  127. wrappersHeaderOut += " lua_settop(L, 0);\n"
  128. wrappersHeaderOut += " }\n"
  129. wrappersHeaderOut += " int wrapperIndex;\n"
  130. wrappersHeaderOut += " lua_State *L;\n"
  131. wrappersHeaderOut += "};\n\n"
  132. # Iterate, process each input file
  133. for fileName in filteredFiles:
  134. # "Package owned" classes that ship with Polycode
  135. inheritInModule = ["PhysicsGenericConstraint", "PhysicsHingeConstraint", "PhysicsPointToPointConstraint", "PhysicsConstraint", "PhysicsEntity", "CollisionScene", "CollisionEntity", "UIElement", "UIWindow", "UIMenuItem", "UIImage", "UIRect"]
  136. # A file or comma-separated list of files can be given to specify classes which are "package owned"
  137. # and should not be inherited out of Polycode/. The files should contain one class name per line,
  138. # and the class name may be prefixed with a path (which will be ignored).
  139. if inheritInModuleFiles:
  140. for moduleFileName in inheritInModuleFiles.split(","):
  141. with open(moduleFileName) as f:
  142. for line in f.readlines():
  143. inheritInModule.append(line.strip().split("/",1)[-1]) # Strip whitespace, path/
  144. print("Parsing %s" % fileName)
  145. try: # One input file parse.
  146. f = open(fileName) # Def: Input file handle
  147. contents = f.read().replace("_PolyExport", "") # Def: Input file contents, strip out "_PolyExport"
  148. cppHeader = CppHeaderParser.CppHeader(contents, "string") # Def: Input file contents, parsed structure
  149. ignore_classes = ["PolycodeShaderModule", "Object", "Threaded", "OpenGLCubemap", "PolyBase", "Matrix4::union "]
  150. # Iterate, check each class in this file.
  151. for ckey in cppHeader.classes:
  152. print(">> Parsing class %s" % ckey)
  153. c = cppHeader.classes[ckey] # Def: The class structure
  154. luaClassBindingOut = "" # Def: The local lua file to generate for this class.
  155. inherits = False
  156. parentClass = ""
  157. if len(c["inherits"]) > 0: # Does this class have parents?
  158. if c["inherits"][0]["class"] not in ignore_classes:
  159. if c["inherits"][0]["class"] in inheritInModule: # Parent class is in this module
  160. luaClassBindingOut += "require \"%s/%s\"\n\n" % (prefix, c["inherits"][0]["class"])
  161. else: # Parent class is in Polycore
  162. luaClassBindingOut += "require \"Polycode/%s\"\n\n" % (c["inherits"][0]["class"])
  163. luaClassBindingOut += "class \"%s\" (%s)\n\n" % (ckey, c["inherits"][0]["class"])
  164. parentClass = c["inherits"][0]["class"]
  165. inherits = True
  166. if inherits == False: # Class does not have parents
  167. luaClassBindingOut += "class \"%s\"\n\n" % ckey
  168. if ckey in ignore_classes:
  169. print("INGORING class %s" % ckey)
  170. continue
  171. #if len(c["methods"]["public"]) < 2: # Used to, this was a continue.
  172. # print("Warning: Lua-binding class with less than two methods")
  173. # continue # FIXME: Remove this, move any non-compileable classes into ignore_classes
  174. extendString = ""
  175. if len(c["inherits"]) > 0:
  176. if c["inherits"][0]["class"] != "PolyBase":
  177. extendString = " extends=\"%s\"" % (c["inherits"][0]["class"])
  178. luaDocOut += "\t<class name=\"%s\"%s>\n" % (ckey, extendString)
  179. if 'doxygen' in c:
  180. luaDocOut += "\t\t<desc><![CDATA[%s]]></desc>\n" % (cleanDocs(c['doxygen']))
  181. if ckey in disable_gc:
  182. luaDocOut += "\t\t<class_notes>NOTE: %s instances are not automatically garbage collected.</class_notes>\n" % (ckey)
  183. parsed_methods = [] # Def: List of discovered methods
  184. ignore_methods = ["readByte32", "readByte16", "getCustomEntitiesByType", "Core", "Renderer", "Shader", "Texture", "handleEvent", "secondaryHandler", "getSTLString", "readInt"]
  185. luaClassBindingOut += "\n\n"
  186. luaDocOut += "\t\t<static_members>\n"
  187. classProperties = [] # Def: List of found property structures ("properties" meaning "data members")
  188. for pp in c["properties"]["public"]:
  189. pp["type"] = pp["type"].replace("Polycode::", "")
  190. pp["type"] = pp["type"].replace("std::", "")
  191. if pp["type"].find("POLYIGNORE") != -1:
  192. continue
  193. if pp["type"].find("static ") != -1: # If static. FIXME: Static doesn't work?
  194. if "defaltValue" in pp: # FIXME: defaltValue is misspelled.
  195. defaltValue = pp["defaltValue"]
  196. # The "Default Value" is more or less a literal C++ string. This causes a problem:
  197. # Frequently we say static const int A = 1; static const int B = A + 1.
  198. # Put in a one-off hack to ensure namespacing works in this special case.
  199. if re.match(r'\s*[a-zA-Z_][a-zA-Z0-9_]*\s*\+', defaltValue):
  200. defaltValue = "%s.%s" % (ckey, defaltValue)
  201. luaClassBindingOut += "%s.%s = %s\n" % (ckey, pp["name"], defaltValue)
  202. luaDocOut += "\t\t\t<static_member name=\"%s\" type=\"%s\" value=\"%s\">\n" % (pp["name"], toLuaType(typeFilter(pp["type"])), pp["defaltValue"])
  203. if 'doxygen' in pp:
  204. luaDocOut += "\t\t\t\t<desc><![CDATA[%s]]></desc>\n" % (cleanDocs(pp['doxygen']))
  205. luaDocOut += "\t\t\t</static_member>\n"
  206. else: # FIXME: Nonstatic method ? variable ?? found.
  207. #there are some bugs in the class parser that cause it to return junk
  208. if pp["type"].find("vector") == -1 and pp["name"] != "setScale" and pp["name"] != "setPosition" and pp["name"] != "BUFFER_CACHE_PRECISION" and not pp["name"].isdigit():
  209. classProperties.append(pp)
  210. luaDocOut += "\t\t</static_members>\n"
  211. # Iterate over properties, creating getters
  212. pidx = 0 # Def: Count of properties processed so far
  213. # TODO: Remove or generalize ParticleEmitter special casing. These lines are marked with #SPEC
  214. luaDocOut += "\t\t<members>\n"
  215. numGetVars = 0
  216. if len(classProperties) > 0: # If there are properties, add index lookup to the metatable
  217. luaClassBindingOut += "function %s:__getvar(name)\n" % ckey
  218. # Iterate over property structures, creating if/else clauses for each.
  219. # TODO: Could a table be more appropriate for
  220. for pp in classProperties:
  221. if pp["name"] == "" or pp["array"] == 1:
  222. continue
  223. numGetVars = numGetVars + 1
  224. pp["type"] = typeFilter(pp["type"])
  225. if pidx == 0:
  226. luaClassBindingOut += "\tif name == \"%s\" then\n" % (pp["name"])
  227. else:
  228. luaClassBindingOut += "\telseif name == \"%s\" then\n" % (pp["name"])
  229. # Generate Lua side of binding:
  230. # If type is a primitive such as Number/String/int/bool
  231. if pp["type"] == "PolyKEY" or pp["type"] == "Number" or pp["type"] == "String" or pp["type"] == "int" or pp["type"] == "bool":
  232. luaClassBindingOut += "\t\treturn %s.%s_get_%s(self.__ptr)\n" % (libName, ckey, pp["name"])
  233. # If type is a particle emitter, specifically #SPEC
  234. elif (ckey == "ScreenParticleEmitter" or ckey == "SceneParticleEmitter") and pp["name"] == "emitter":
  235. luaClassBindingOut += "\t\tlocal ret = %s(\"__skip_ptr__\")\n" % (pp["type"])
  236. luaClassBindingOut += "\t\tret.__ptr = self.__ptr\n"
  237. luaClassBindingOut += "\t\treturn ret\n"
  238. # If type is a class
  239. else:
  240. luaClassBindingOut += "\t\tlocal retVal = %s.%s_get_%s(self.__ptr)\n" % (libName, ckey, pp["name"])
  241. luaClassBindingOut += template_returnPtrLookup("\t\t", template_quote(pp["type"]), "retVal")
  242. luaDocOut += "\t\t\t<member name=\"%s\" type=\"%s\">\n" % (pp["name"], toLuaType(typeFilter(pp["type"])))
  243. if 'doxygen' in pp:
  244. luaDocOut += "\t\t\t\t<desc><![CDATA[%s]]></desc>\n" % (cleanDocs(pp['doxygen']))
  245. luaDocOut += "\t\t\t</member>\n"
  246. # Generate C++ side of binding:
  247. if not ((ckey == "ScreenParticleEmitter" or ckey == "SceneParticleEmitter") and pp["name"] == "emitter"): #SPEC
  248. cppRegisterOut += "\t\t{\"%s_get_%s\", %s_%s_get_%s},\n" % (ckey, pp["name"], libName, ckey, pp["name"])
  249. wrappersHeaderOut += "static int %s_%s_get_%s(lua_State *L) {\n" % (libName, ckey, pp["name"])
  250. wrappersHeaderOut += "\tluaL_checktype(L, 1, LUA_TUSERDATA);\n"
  251. wrappersHeaderOut += "\t%s *inst = (%s*) *((PolyBase**)lua_touserdata(L, 1));\n" % (ckey, ckey)
  252. outfunc = "this_shouldnt_happen"
  253. retFunc = ""
  254. if pp["type"] == "Number":
  255. outfunc = "lua_pushnumber"
  256. if pp["type"] == "String":
  257. outfunc = "lua_pushstring"
  258. retFunc = ".c_str()"
  259. if pp["type"] == "int" or pp["type"] == "PolyKEY":
  260. outfunc = "lua_pushinteger"
  261. if pp["type"] == "bool":
  262. outfunc = "lua_pushboolean"
  263. if pp["type"] == "Number" or pp["type"] == "String" or pp["type"] == "int" or pp["type"] == "bool" or pp["type"] == "PolyKEY":
  264. wrappersHeaderOut += "\t%s(L, inst->%s%s);\n" % (outfunc, pp["name"], retFunc)
  265. else:
  266. if pp["type"].find("*") != -1:
  267. wrappersHeaderOut += "\tif(!inst->%s%s) {\n" % (pp["name"], retFunc)
  268. wrappersHeaderOut += "\t\tlua_pushnil(L);\n"
  269. wrappersHeaderOut += "\t} else {\n"
  270. wrappersHeaderOut += "\t\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  271. wrappersHeaderOut += "\t\t*userdataPtr = (PolyBase*)inst->%s%s;\n" % (pp["name"], retFunc)
  272. wrappersHeaderOut += "\t}\n"
  273. else:
  274. wrappersHeaderOut += "\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  275. wrappersHeaderOut += "\t*userdataPtr = (PolyBase*)&inst->%s%s;\n" % (pp["name"], retFunc)
  276. wrappersHeaderOut += "\treturn 1;\n"
  277. wrappersHeaderOut += "}\n\n"
  278. # Success
  279. pidx = pidx + 1
  280. if numGetVars != 0:
  281. luaClassBindingOut += "\tend\n"
  282. if inherits:
  283. luaClassBindingOut += "\tif %s[\"__getvar\"] ~= nil then\n" % (parentClass)
  284. luaClassBindingOut += "\t\treturn %s.__getvar(self, name)\n" % (parentClass)
  285. luaClassBindingOut += "\tend\n"
  286. luaClassBindingOut += "end\n"
  287. luaDocOut += "\t\t</members>\n"
  288. luaClassBindingOut += "\n\n"
  289. # Iterate over properties again, creating setters
  290. pidx = 0 # Def: Count of
  291. if len(classProperties) > 0: # If there are properties, add index setter to the metatable
  292. luaClassBindingOut += "function %s:__setvar(name,value)\n" % ckey
  293. for pp in classProperties:
  294. if pp["name"] == "" or pp["array"] == 1:
  295. continue
  296. pp["type"] = typeFilter(pp["type"])
  297. # If type is a primitive: Create lua and C++ sides at the same time.
  298. if pp["type"] == "Number" or pp["type"] == "String" or pp["type"] == "int" or pp["type"] == "bool" or pp["type"] == "PolyKEY":
  299. if pidx == 0:
  300. luaClassBindingOut += "\tif name == \"%s\" then\n" % (pp["name"])
  301. else:
  302. luaClassBindingOut += "\telseif name == \"%s\" then\n" % (pp["name"])
  303. luaClassBindingOut += "\t\t%s.%s_set_%s(self.__ptr, value)\n" % (libName, ckey, pp["name"])
  304. luaClassBindingOut += "\t\treturn true\n"
  305. cppRegisterOut += "\t\t{\"%s_set_%s\", %s_%s_set_%s},\n" % (ckey, pp["name"], libName, ckey, pp["name"])
  306. wrappersHeaderOut += "static int %s_%s_set_%s(lua_State *L) {\n" % (libName, ckey, pp["name"])
  307. wrappersHeaderOut += "\tluaL_checktype(L, 1, LUA_TUSERDATA);\n"
  308. wrappersHeaderOut += "\t%s *inst = (%s*) *((PolyBase**)lua_touserdata(L, 1));\n" % (ckey, ckey)
  309. outfunc = "this_shouldnt_happen"
  310. outfuncsuffix = ""
  311. if pp["type"] == "Number":
  312. outfunc = "lua_tonumber"
  313. if pp["type"] == "String":
  314. outfunc = "lua_tostring"
  315. if pp["type"] == "int":
  316. outfunc = "lua_tointeger"
  317. if pp["type"] == "PolyKEY":
  318. outfunc = "(PolyKEY)lua_tointeger"
  319. if pp["type"] == "bool":
  320. outfunc = "lua_toboolean"
  321. outfuncsuffix = " != 0"
  322. wrappersHeaderOut += "\t%s param = %s(L, 2)%s;\n" % (pp["type"], outfunc, outfuncsuffix)
  323. wrappersHeaderOut += "\tinst->%s = param;\n" % (pp["name"])
  324. wrappersHeaderOut += "\treturn 0;\n"
  325. wrappersHeaderOut += "}\n\n"
  326. pidx = pidx + 1 # Success
  327. else:
  328. if pp["type"].find("*") == -1 and pp["type"].find("static") == -1:
  329. if pidx == 0:
  330. luaClassBindingOut += "\tif name == \"%s\" then\n" % (pp["name"])
  331. else:
  332. luaClassBindingOut += "\telseif name == \"%s\" then\n" % (pp["name"])
  333. luaClassBindingOut += "\t\t%s.%s_set_%s(self.__ptr, value.__ptr)\n" % (libName, ckey, pp["name"])
  334. luaClassBindingOut += "\t\treturn true\n"
  335. cppRegisterOut += "\t\t{\"%s_set_%s\", %s_%s_set_%s},\n" % (ckey, pp["name"], libName, ckey, pp["name"])
  336. wrappersHeaderOut += "static int %s_%s_set_%s(lua_State *L) {\n" % (libName, ckey, pp["name"])
  337. wrappersHeaderOut += "\tluaL_checktype(L, 1, LUA_TUSERDATA);\n"
  338. wrappersHeaderOut += "\t%s *inst = (%s*) *((PolyBase**)lua_touserdata(L, 1));\n" % (ckey, ckey)
  339. wrappersHeaderOut += "\tluaL_checktype(L, 2, LUA_TUSERDATA);\n"
  340. wrappersHeaderOut += "\t%s *argInst = (%s*) *((PolyBase**)lua_touserdata(L, 2));\n" % (typeFilter(pp["type"]), typeFilter(pp["type"]))
  341. wrappersHeaderOut += "\tinst->%s = *argInst;\n" % (pp["name"])
  342. wrappersHeaderOut += "\treturn 0;\n"
  343. wrappersHeaderOut += "}\n\n"
  344. pidx = pidx + 1 # Success
  345. # Notice: Setters for object types are not created.
  346. if pidx != 0:
  347. luaClassBindingOut += "\tend\n"
  348. if inherits:
  349. luaClassBindingOut += "\tif %s[\"__setvar\"] ~= nil then\n" % (parentClass)
  350. luaClassBindingOut += "\t\treturn %s.__setvar(self, name, value)\n" % (parentClass)
  351. luaClassBindingOut += "\telse\n"
  352. luaClassBindingOut += "\t\treturn false\n"
  353. luaClassBindingOut += "\tend\n"
  354. else:
  355. luaClassBindingOut += "\treturn false\n"
  356. luaClassBindingOut += "end\n"
  357. # Iterate over methods
  358. luaClassBindingOut += "\n\n"
  359. luaDocOut += "\t\t<methods>\n"
  360. for pm in c["methods"]["public"]:
  361. # Skip argument-overloaded methods and operators.
  362. # TODO: Instead of skipping arguemnt overloads, have special behavior.
  363. # TODO: Instead of skipping operators, add to metatable.
  364. if pm["name"] in parsed_methods or pm["name"].find("operator") > -1 or pm["rtnType"].find("POLYIGNORE") > -1 or pm["name"] in ignore_methods:
  365. continue
  366. # Skip destructors and methods which return templates.
  367. # TODO: Special-case certain kind of vector<>s?
  368. if pm["name"] == "~"+ckey or pm["name"] == "CoreServices":
  369. continue
  370. staticString = ""
  371. if pm["rtnType"].find("static ") != -1:
  372. staticString = " static=\"true\""
  373. if pm["rtnType"].find("std::vector") > -1:
  374. vectorReturnClass = pm["rtnType"].replace("std::vector<", "").replace(">","").replace(" ", "")
  375. luaDocOut += "\t\t\t<method name=\"%s\" return_array=\"true\" return_type=\"%s\"%s>\n" % (pm["name"], toLuaType(typeFilter(vectorReturnClass).replace("*", "")), staticString)
  376. else:
  377. luaDocOut += "\t\t\t<method name=\"%s\" return_type=\"%s\"%s>\n" % (pm["name"], toLuaType(typeFilter(pm["rtnType"].replace("*", ""))), staticString)
  378. docs = None
  379. if 'doxygen' in pm:
  380. if pm['doxygen'].find("@return") > -1:
  381. docs = cleanDocs(pm['doxygen']).split("@return")[0].split("@param")
  382. else:
  383. docs = cleanDocs(pm['doxygen']).split("@param")
  384. luaDocOut += "\t\t\t\t<desc><![CDATA[%s]]></desc>\n" % (docs[0])
  385. if len(pm["parameters"]) > 0:
  386. luaDocOut += "\t\t\t\t<params>\n"
  387. paramIndex = 0
  388. for param in pm["parameters"]:
  389. if "name" in param:
  390. if not "type" in param:
  391. continue
  392. if param["type"] == "0":
  393. continue
  394. if param["type"].find("vector<") != -1:
  395. vectorClass = param["type"].replace("std::vector<", "").replace(">","").replace(" ", "")
  396. luaDocOut += "\t\t\t\t\t<param name=\"%s\" param_array=\"true\" type=\"%s\">\n" % (param["name"], toLuaType(vectorClass.replace("*","")))
  397. else:
  398. luaDocOut += "\t\t\t\t\t<param name=\"%s\" type=\"%s\">\n" % (param["name"], toLuaType(typeFilter(param["type"]).replace("*","")))
  399. if docs != None:
  400. if len(docs) > paramIndex+1:
  401. cdoc = docs[paramIndex+1].split()
  402. cdoc.pop(0)
  403. luaDocOut += "\t\t\t\t\t\t<desc><![CDATA[%s]]></desc>\n" % (" ".join(cdoc).replace("\n", ""))
  404. luaDocOut += "\t\t\t\t\t</param>\n"
  405. paramIndex = paramIndex + 1
  406. luaDocOut += "\t\t\t\t</params>\n"
  407. luaDocOut += "\t\t\t</method>\n"
  408. basicType = False
  409. voidRet = False
  410. vectorReturn = False
  411. vectorReturnClass = ""
  412. # Def: True if method takes a lua_State* as argument (i.e.: no preprocessing by us)
  413. rawMethod = len(pm["parameters"]) > 0 and pm["parameters"][0].get("type","").find("lua_State") > -1
  414. # Basic setup, C++ side: Add function to registry and start building wrapper function.
  415. if pm["name"] == ckey: # It's a constructor
  416. cppRegisterOut += "\t\t{\"%s\", %s_%s},\n" % (ckey, libName, ckey)
  417. wrappersHeaderOut += "static int %s_%s(lua_State *L) {\n" % (libName, ckey)
  418. idx = 1 # Def: Current stack depth (TODO: Figure out, is this correct?)
  419. else: # It's not a constructor
  420. cppRegisterOut += "\t\t{\"%s_%s\", %s_%s_%s},\n" % (ckey, pm["name"], libName, ckey, pm["name"])
  421. wrappersHeaderOut += "static int %s_%s_%s(lua_State *L) {\n" % (libName, ckey, pm["name"])
  422. # Skip static methods (TODO: Figure out, why is this being done here?). # FIXME
  423. if pm["rtnType"].find("static ") == -1:
  424. wrappersHeaderOut += "\tluaL_checktype(L, 1, LUA_TUSERDATA);\n"
  425. wrappersHeaderOut += "\t%s *inst = (%s*) *((PolyBase**)lua_touserdata(L, 1));\n" % (ckey, ckey)
  426. idx = 2
  427. else:
  428. idx = 1
  429. if rawMethod:
  430. wrappersHeaderOut += "\treturn inst->%s(L);\n" % (pm["name"])
  431. else:
  432. # Generate C++ side parameter pushing
  433. paramlist = []
  434. lparamlist = []
  435. for param in pm["parameters"]:
  436. if not "type" in param:
  437. continue
  438. if param["type"] == "0":
  439. continue
  440. param["type"] = typeFilter(param["type"])
  441. param["name"] = param["name"].replace("end", "_end").replace("repeat", "_repeat")
  442. if"type" in param:
  443. luatype = "LUA_TUSERDATA"
  444. checkfunc = "lua_isuserdata"
  445. if param["type"].find("*") > -1:
  446. luafunc = "(%s) *((PolyBase**)lua_touserdata" % (param["type"].replace("Polygon", "Polycode::Polygon").replace("Rectangle", "Polycode::Rectangle"))
  447. elif param["type"].find("&") > -1:
  448. luafunc = "*(%s*) *((PolyBase**)lua_touserdata" % (param["type"].replace("const", "").replace("&", "").replace("Polygon", "Polycode::Polygon").replace("Rectangle", "Polycode::Rectangle"))
  449. else:
  450. luafunc = "*(%s*) *((PolyBase**)lua_touserdata" % (param["type"].replace("Polygon", "Polycode::Polygon").replace("Rectangle", "Polycode::Rectangle"))
  451. lend = ".__ptr"
  452. luafuncsuffix = ")"
  453. if param["type"] == "int" or param["type"] == "unsigned int" or param["type"] == "short":
  454. luafunc = "lua_tointeger"
  455. luatype = "LUA_TNUMBER"
  456. checkfunc = "lua_isnumber"
  457. luafuncsuffix = ""
  458. lend = ""
  459. if param["type"] == "PolyKEY":
  460. luafunc = "(PolyKEY)lua_tointeger"
  461. luatype = "LUA_TNUMBER"
  462. checkfunc = "lua_isnumber"
  463. luafuncsuffix = ""
  464. lend = ""
  465. if param["type"] == "bool":
  466. luafunc = "lua_toboolean"
  467. luatype = "LUA_TBOOLEAN"
  468. checkfunc = "lua_isboolean"
  469. luafuncsuffix = " != 0"
  470. lend = ""
  471. if param["type"] == "Number" or param["type"] == "float" or param["type"] == "double":
  472. luatype = "LUA_TNUMBER"
  473. luafunc = "lua_tonumber"
  474. checkfunc = "lua_isnumber"
  475. luafuncsuffix = ""
  476. lend = ""
  477. if param["type"] == "String":
  478. luatype = "LUA_TSTRING"
  479. luafunc = "lua_tostring"
  480. checkfunc = "lua_isstring"
  481. luafuncsuffix = ""
  482. lend = ""
  483. param["type"] = param["type"].replace("Polygon", "Polycode::Polygon").replace("Rectangle", "Polycode::Rectangle")
  484. if "defaltValue" in param:
  485. if checkfunc != "lua_isuserdata" or (checkfunc == "lua_isuserdata" and param["defaltValue"] == "NULL"):
  486. #param["defaltValue"] = param["defaltValue"].replace(" 0f", ".0f")
  487. param["defaltValue"] = param["defaltValue"].replace(": :", "::")
  488. #param["defaltValue"] = param["defaltValue"].replace("0 ", "0.")
  489. param["defaltValue"] = re.sub(r'([0-9]+) ([0-9])+', r'\1.\2', param["defaltValue"])
  490. wrappersHeaderOut += "\t%s %s;\n" % (param["type"], param["name"])
  491. wrappersHeaderOut += "\tif(%s(L, %d)) {\n" % (checkfunc, idx)
  492. wrappersHeaderOut += "\t\t%s = %s(L, %d)%s;\n" % (param["name"], luafunc, idx, luafuncsuffix)
  493. wrappersHeaderOut += "\t} else {\n"
  494. wrappersHeaderOut += "\t\t%s = %s;\n" % (param["name"], param["defaltValue"])
  495. wrappersHeaderOut += "\t}\n"
  496. else:
  497. wrappersHeaderOut += "\tluaL_checktype(L, %d, %s);\n" % (idx, luatype);
  498. if param["type"] == "String":
  499. wrappersHeaderOut += "\t%s %s = String(%s(L, %d));\n" % (param["type"], param["name"], luafunc, idx)
  500. else:
  501. wrappersHeaderOut += "\t%s %s = %s(L, %d)%s;\n" % (param["type"], param["name"], luafunc, idx,luafuncsuffix)
  502. else:
  503. wrappersHeaderOut += "\tluaL_checktype(L, %d, %s);\n" % (idx, luatype);
  504. if param["type"] == "String":
  505. wrappersHeaderOut += "\t%s %s = String(%s(L, %d));\n" % (param["type"], param["name"], luafunc, idx)
  506. else:
  507. wrappersHeaderOut += "\t%s %s = %s(L, %d)%s;\n" % (param["type"], param["name"], luafunc, idx, luafuncsuffix)
  508. paramlist.append(param["name"])
  509. lparamlist.append(param["name"]+lend)
  510. idx = idx +1 # Param parse success-- mark the increased stack
  511. # Generate C++-side method call / generate return value
  512. if pm["name"] == ckey: # If constructor
  513. if ckey == "EventHandler": # See LuaEventHandler above
  514. wrappersHeaderOut += "\tLuaEventHandler *inst = new LuaEventHandler();\n"
  515. wrappersHeaderOut += "\tinst->wrapperIndex = luaL_ref(L, LUA_REGISTRYINDEX );\n"
  516. wrappersHeaderOut += "\tinst->L = L;\n"
  517. else:
  518. wrappersHeaderOut += "\t%s *inst = new %s(%s);\n" % (ckey, ckey, ", ".join(paramlist))
  519. wrappersHeaderOut += "\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  520. wrappersHeaderOut += "\t*userdataPtr = (PolyBase*)inst;\n"
  521. wrappersHeaderOut += "\tluaL_getmetatable(L, \"%s.%s\");\n" % (libName, ckey)
  522. wrappersHeaderOut += "\tlua_setmetatable(L, -2);\n"
  523. wrappersHeaderOut += "\treturn 1;\n"
  524. else: #If non-constructor
  525. if pm["rtnType"].find("static ") == -1: # If non-static
  526. call = "inst->%s(%s)" % (pm["name"], ", ".join(paramlist))
  527. else: # If static (FIXME: Why doesn't this work?)
  528. call = "%s::%s(%s)" % (ckey, pm["name"], ", ".join(paramlist))
  529. #check if returning a template
  530. if pm["rtnType"].find("<") > -1:
  531. #if returning a vector, convert to lua table
  532. if pm["rtnType"].find("std::vector") > -1:
  533. vectorReturnClass = pm["rtnType"].replace("std::vector<", "").replace(">","").replace(" ", "")
  534. if vectorReturnClass.find("&") == -1 and vectorReturnClass.find("*") > -1: #FIXME: return references to std::vectors and basic types
  535. vectorReturn = True
  536. wrappersHeaderOut += "\tstd::vector<%s> retVector = %s;\n" % (vectorReturnClass,call)
  537. wrappersHeaderOut += "\tlua_newtable(L);\n"
  538. wrappersHeaderOut += "\tfor(int i=0; i < retVector.size(); i++) {\n"
  539. wrappersHeaderOut += "\t\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  540. wrappersHeaderOut += "\t\t*userdataPtr = (PolyBase*)retVector[i];\n"
  541. wrappersHeaderOut += "\t\tlua_rawseti(L, -2, i+1);\n"
  542. wrappersHeaderOut += "\t}\n"
  543. wrappersHeaderOut += "\treturn 1;\n"
  544. else:
  545. wrappersHeaderOut += "\treturn 0;\n"
  546. # else If void-typed:
  547. elif pm["rtnType"] == "void" or pm["rtnType"] == "static void" or pm["rtnType"] == "virtual void" or pm["rtnType"] == "inline void":
  548. wrappersHeaderOut += "\t%s;\n" % (call)
  549. basicType = True
  550. voidRet = True
  551. vectorReturn = False
  552. wrappersHeaderOut += "\treturn 0;\n" # 0 arguments returned
  553. else: # If there is a return value:
  554. # What type is the return value? Default to pointer
  555. outfunc = "this_shouldnt_happen"
  556. retFunc = ""
  557. basicType = False
  558. vectorReturn = False
  559. if pm["rtnType"] == "Number" or pm["rtnType"] == "inline Number":
  560. outfunc = "lua_pushnumber"
  561. basicType = True
  562. if pm["rtnType"] == "String" or pm["rtnType"] == "static String": # TODO: Path for STL strings?
  563. outfunc = "lua_pushstring"
  564. basicType = True
  565. retFunc = ".c_str()"
  566. if pm["rtnType"] == "int" or pm["rtnType"] == "unsigned int" or pm["rtnType"] == "static int" or pm["rtnType"] == "size_t" or pm["rtnType"] == "static size_t" or pm["rtnType"] == "long" or pm["rtnType"] == "unsigned int" or pm["rtnType"] == "static long" or pm["rtnType"] == "short" or pm["rtnType"] == "PolyKEY" or pm["rtnType"] == "wchar_t":
  567. outfunc = "lua_pushinteger"
  568. basicType = True
  569. if pm["rtnType"] == "bool" or pm["rtnType"] == "static bool" or pm["rtnType"] == "virtual bool":
  570. outfunc = "lua_pushboolean"
  571. basicType = True
  572. if pm["rtnType"].find("*") > -1: # Returned var is definitely a pointer.
  573. wrappersHeaderOut += "\tPolyBase *ptrRetVal = (PolyBase*)%s%s;\n" % (call, retFunc)
  574. wrappersHeaderOut += "\tif(ptrRetVal == NULL) {\n"
  575. wrappersHeaderOut += "\t\tlua_pushnil(L);\n"
  576. wrappersHeaderOut += "\t} else {\n"
  577. wrappersHeaderOut += "\t\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  578. wrappersHeaderOut += "\t\t*userdataPtr = ptrRetVal;\n"
  579. wrappersHeaderOut += "\t}\n"
  580. elif basicType == True: # Returned var has been flagged as a recognized primitive type
  581. wrappersHeaderOut += "\t%s(L, %s%s);\n" % (outfunc, call, retFunc)
  582. else: # Some static object is being returned. Convert it to a pointer, then return that.
  583. className = pm["rtnType"].replace("const", "").replace("&", "").replace("inline", "").replace("virtual", "").replace("static", "")
  584. if className == "Polygon": # Deal with potential windows.h conflict
  585. className = "Polycode::Polygon"
  586. if className == "Rectangle":
  587. className = "Polycode::Rectangle"
  588. if className == "Polycode : : Rectangle":
  589. className = "Polycode::Rectangle"
  590. wrappersHeaderOut += "\t%s *retInst = new %s();\n" % (className, className)
  591. wrappersHeaderOut += "\t*retInst = %s;\n" % (call)
  592. wrappersHeaderOut += "\tPolyBase **userdataPtr = (PolyBase**)lua_newuserdata(L, sizeof(PolyBase*));\n"
  593. wrappersHeaderOut += "\tluaL_getmetatable(L, \"%s.%s\");\n" % (libName, className)
  594. wrappersHeaderOut += "\tlua_setmetatable(L, -2);\n"
  595. wrappersHeaderOut += "\t*userdataPtr = (PolyBase*)retInst;\n"
  596. wrappersHeaderOut += "\treturn 1;\n"
  597. wrappersHeaderOut += "}\n\n" # Close out C++ generation
  598. # Now generate the Lua side method.
  599. if rawMethod:
  600. luaClassBindingOut += "function %s:%s(...)\n" % (ckey, pm["name"])
  601. luaClassBindingOut += "\treturn %s.%s_%s(self.__ptr, ...)\n" % (libName, ckey, pm["name"])
  602. luaClassBindingOut += "end\n"
  603. elif pm["name"] == ckey: # Constructors
  604. luaClassBindingOut += "function %s:%s(...)\n" % (ckey, ckey)
  605. luaClassBindingOut += "\tlocal arg = {...}\n"
  606. if inherits:
  607. luaClassBindingOut += "\tif type(arg[1]) == \"table\" and count(arg) == 1 then\n"
  608. luaClassBindingOut += "\t\tif \"\"..arg[1].__classname == \"%s\" then\n" % (c["inherits"][0]["class"])
  609. luaClassBindingOut += "\t\t\tself.__ptr = arg[1].__ptr\n"
  610. luaClassBindingOut += "\t\t\treturn\n"
  611. luaClassBindingOut += "\t\tend\n"
  612. luaClassBindingOut += "\tend\n"
  613. luaClassBindingOut += "\tfor k,v in pairs(arg) do\n"
  614. luaClassBindingOut += "\t\tif type(v) == \"table\" then\n"
  615. luaClassBindingOut += "\t\t\tif v.__ptr ~= nil then\n"
  616. luaClassBindingOut += "\t\t\t\targ[k] = v.__ptr\n"
  617. luaClassBindingOut += "\t\t\tend\n"
  618. luaClassBindingOut += "\t\tend\n"
  619. luaClassBindingOut += "\tend\n"
  620. luaClassBindingOut += "\tif self.__ptr == nil and arg[1] ~= \"__skip_ptr__\" then\n"
  621. if ckey == "EventHandler": # See LuaEventHandler above
  622. luaClassBindingOut += "\t\tself.__ptr = %s.%s(self)\n" % (libName, ckey)
  623. else:
  624. luaClassBindingOut += "\t\tself.__ptr = %s.%s(unpack(arg))\n" % (libName, ckey)
  625. luaClassBindingOut += "\tend\n"
  626. luaClassBindingOut += "end\n\n"
  627. else: # Non-constructors.
  628. if pm["rtnType"].find("static ") == -1: # Non-static method
  629. luaClassBindingOut += "function %s:%s(%s)\n" % (ckey, pm["name"], ", ".join(paramlist))
  630. if len(lparamlist):
  631. luaClassBindingOut += "\tlocal retVal = %s.%s_%s(self.__ptr, %s)\n" % (libName, ckey, pm["name"], ", ".join(lparamlist))
  632. else:
  633. luaClassBindingOut += "\tlocal retVal = %s.%s_%s(self.__ptr)\n" % (libName, ckey, pm["name"])
  634. else: # Static method
  635. luaClassBindingOut += "function %s.%s(%s)\n" % (ckey, pm["name"], ", ".join(paramlist))
  636. if len(lparamlist):
  637. luaClassBindingOut += "\tlocal retVal = %s.%s_%s(%s)\n" % (libName, ckey, pm["name"], ", ".join(lparamlist))
  638. else:
  639. luaClassBindingOut += "\tlocal retVal = %s.%s_%s()\n" % (libName, ckey, pm["name"])
  640. if not voidRet: # Was there a return value?
  641. if basicType == True: # Yes, a primitive
  642. luaClassBindingOut += "\treturn retVal\n"
  643. else: # Yes, a pointer was returned
  644. if vectorReturn == True:
  645. className = vectorReturnClass.replace("*", "")
  646. luaClassBindingOut += template_returnPtrLookupArray("\t",template_quote(className),"retVal")
  647. else:
  648. className = pm["rtnType"].replace("const", "").replace("&", "").replace("inline", "").replace("virtual", "").replace("static", "").replace("*","").replace(" ", "")
  649. luaClassBindingOut += template_returnPtrLookup("\t",template_quote(className),"retVal")
  650. luaClassBindingOut += "end\n\n" # Close out Lua generation
  651. parsed_methods.append(pm["name"]) # Method parse success
  652. luaDocOut += "\t\t</methods>\n"
  653. # With methods out of the way, do some final cleanup:
  654. # user pointer metatable creation in C++
  655. cppLoaderOut += "\n\tluaL_newmetatable(L, \"%s.%s\");\n" % (libName, ckey)
  656. if ckey not in disable_gc:
  657. cppLoaderOut += "\tlua_pushstring(L, \"__gc\");\n"
  658. cppLoaderOut += "\tlua_pushcfunction(L, %s_delete_%s);\n" % (libName, ckey)
  659. cppLoaderOut += "\tlua_settable(L, -3);\n"
  660. cppLoaderOut +="\tlua_pop(L, 1);\n"
  661. # Delete method (C++ side)
  662. cppRegisterOut += "\t\t{\"delete_%s\", %s_delete_%s},\n" % (ckey, libName, ckey)
  663. wrappersHeaderOut += "static int %s_delete_%s(lua_State *L) {\n" % (libName, ckey)
  664. wrappersHeaderOut += "\tluaL_checktype(L, 1, LUA_TUSERDATA);\n"
  665. wrappersHeaderOut += "\tPolyBase **inst = (PolyBase**)lua_touserdata(L, 1);\n"
  666. wrappersHeaderOut += "\tdelete ((%s*) *inst);\n" % (ckey)
  667. wrappersHeaderOut += "\t*inst = NULL;\n"
  668. wrappersHeaderOut += "\treturn 0;\n"
  669. wrappersHeaderOut += "}\n\n"
  670. # Delete method (Lua side)
  671. luaClassBindingOut += "function %s:__delete()\n" % (ckey)
  672. luaClassBindingOut += "\tif self then %s.delete_%s(self.__ptr) end\n" % (libName, ckey)
  673. luaClassBindingOut += "end\n"
  674. # Add class to lua index file
  675. luaIndexOut += "require \"%s/%s\"\n" % (prefix, ckey)
  676. # Write lua file
  677. mkdir_p(apiClassPath)
  678. if ckey != "EventDispatcher":
  679. fout = open("%s/%s.lua" % (apiClassPath, ckey), "w")
  680. fout.write(luaClassBindingOut)
  681. luaDocOut += "\t</class>\n"
  682. except CppHeaderParser.CppParseError as e: # One input file parse; failed.
  683. print(e)
  684. sys.exit(1)
  685. luaDocOut += "</classes>\n"
  686. luaDocOut += "</docs>\n"
  687. # Footer boilerplate for wrappersHeaderOut and cppRegisterOut.
  688. wrappersHeaderOut += "} // namespace Polycode\n"
  689. cppRegisterOut += "\t\t{NULL, NULL}\n"
  690. cppRegisterOut += "\t};\n"
  691. cppRegisterOut += "\tluaL_openlib(L, \"%s\", %sLib, 0);\n" % (libName, libSmallName)
  692. cppRegisterOut += cppLoaderOut
  693. cppRegisterOut += "\treturn 1;\n"
  694. cppRegisterOut += "}"
  695. cppRegisterHeaderOut = "" # Def: Global C++ *LUA.h
  696. cppRegisterHeaderOut += "#pragma once\n"
  697. cppRegisterHeaderOut += "#include <%s>\n" % (mainInclude)
  698. cppRegisterHeaderOut += "extern \"C\" {\n"
  699. cppRegisterHeaderOut += "#include <stdio.h>\n"
  700. cppRegisterHeaderOut += "#include \"lua.h\"\n"
  701. cppRegisterHeaderOut += "#include \"lualib.h\"\n"
  702. cppRegisterHeaderOut += "#include \"lauxlib.h\"\n"
  703. cppRegisterHeaderOut += "int _PolyExport luaopen_%s(lua_State *L);\n" % (prefix)
  704. cppRegisterHeaderOut += "}\n"
  705. # Write out global files
  706. mkdir_p(includePath)
  707. mkdir_p(apiPath)
  708. mkdir_p(sourcePath)
  709. fout = open("%s/%sLUA.h" % (includePath, prefix), "w")
  710. fout.write(cppRegisterHeaderOut)
  711. if luaDocPath is None:
  712. luaDocPath = "../../../Documentation/Lua/xml"
  713. if luaDocPath != "-":
  714. fout = open("%s/%s.xml" % (luaDocPath, prefix), "w")
  715. fout.write(luaDocOut)
  716. fout = open("%s/%s.lua" % (apiPath, prefix), "w")
  717. fout.write(luaIndexOut)
  718. fout = open("%s/%sLUAWrappers.h" % (includePath, prefix), "w")
  719. fout.write(wrappersHeaderOut)
  720. fout = open("%s/%sLUA.cpp" % (sourcePath, prefix), "w")
  721. fout.write(cppRegisterOut)
  722. # Create .pak zip archive
  723. pattern = '*.lua'
  724. os.chdir(apiPath)
  725. if libName == "Polycore":
  726. with ZipFile("api.pak", 'w') as myzip:
  727. for root, dirs, files in os.walk("."):
  728. for filename in fnmatch.filter(files, pattern):
  729. myzip.write(os.path.join(root, filename))
  730. else:
  731. with ZipFile("%s.pak" % (libName), 'w') as myzip:
  732. for root, dirs, files in os.walk("."):
  733. for filename in fnmatch.filter(files, pattern):
  734. myzip.write(os.path.join(root, filename))
  735. if len(sys.argv) < 10:
  736. print ("Usage:\n%s [input path] [prefix] [main include] [lib small name] [lib name] [api path] [api class-path] [include path] [source path] [lua doc path (optional) (or - for omit)] [inherit-in-module-file path (optional)]" % (sys.argv[0]))
  737. sys.exit(1)
  738. else:
  739. createLUABindings(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4], sys.argv[5], sys.argv[6], sys.argv[7], sys.argv[8], sys.argv[9], sys.argv[10] if len(sys.argv)>10 else None, sys.argv[11] if len(sys.argv)>11 else None)