libshit

wscript_lua (5275B)

---

 # -*- python -*-
 
 from waflib import Context
 def options(ctx):
     ctx.with_opt('lua', ['lua', 'system', 'none'],
                  default=getattr(Context.g_module, 'DEFAULT_LUA', 'none'))
 
     grp = ctx.get_option_group('configure options')
     grp.add_option('--lua-pc-name', action='store', default='lua5.3',
                    help='Lua pkg-config name (in case of --with-lua=system). '+
                    'Should be lua 5.3 or compatible with c++ exceptions')
     grp.add_option('--lua-dll', action='store_true', default=None,
                    help='Compile lua as dll (Windows only)')
 
     grp.add_option('--luac-mode', action='store',
                    help='Luac mode: copy, system-luajit, luac, system-luac, luac-wrapper')
 
 def chk_luac(ctx, modes):
     if not ctx.env.LUAC_MODE in modes:
         ctx.fatal('Invalid --luac-mode, valid modes: %s' % modes)
     if ctx.env.LUAC_MODE in ['system-luajit', 'system-luac', 'luac-wrapper']:
         if ctx.env.LUAC == []:
             ctx.fatal('Must specify LUAC env var')
 
 def configure(ctx):
     from waflib import Logs
     ctx.add_os_flags('LUACFLAGS')
     ctx.add_os_flags('LUAC')
 
     def no_lua(ctx, _):
         ctx.env.append_value('DEFINES_LUA', 'LIBSHIT_WITH_LUA=0')
         ctx.env.WITH_LUA = 'none'
     ctx.only_host_env(no_lua)
 
     ctx.env.LUAC_MODE = ctx.options.luac_mode
 
     def check_none(ctx):
         chk_luac(ctx, [[]])
 
     def check_system(ctx):
         ctx.check_cfg(package=ctx.options.lua_pc_name, args='--cflags --libs',
                       uselib_store='LUA')
         if not ctx.env.LUAC_MODE: ctx.env.LUAC_MODE = 'copy'
         chk_luac(ctx, ['copy', 'system-luajit', 'system-luac'])
 
     def check_common(ctx):
         if ctx.env.DEST_OS == 'win32':
             ctx.env.LUA_DLL = ctx.options.lua_dll
 
         if ctx.env['COMPILER_CXX'] != 'msvc':
             ctx.env.append_value('CFLAGS_LUA', '-fexceptions')
 
         if ctx.env.LUA_DLL:
             ctx.env.append_value('DEFINES_LUA', 'LUA_BUILD_AS_DLL')
         if not ctx.options.release:
             ctx.env.append_value('DEFINES_LUA', 'LUA_USE_APICHECK')
 
     def check_lua(ctx):
         check_common(ctx)
         ctx.recurse('lua', name='configure', once=False)
         chk_luac(ctx, ['copy', 'luac', 'system-luac', 'luac-wrapper'])
 
     ctx.with_chk(
         'lua', {'none': check_none, 'system': check_system,
                 'lua': check_lua}, define='LIBSHIT_WITH_LUA')
 
     if ctx.env.DEBUG and ctx.env.LUAC_MODE in ['system-luajit']:
         ctx.env.append_value('LUACFLAGS', '-g')
     elif not ctx.env.DEBUG and \
          ctx.env.LUAC_MODE in ['luac', 'system-luac', 'luac-wrapper']:
         ctx.env.append_value('LUACFLAGS', '-s')
 
 def build(ctx):
     if ctx.env.WITH_LUA == 'lua':
         ctx.recurse('lua', name='build', once=False)
 
 
 from waflib import Task, Utils
 from hashlib import sha256
 class bin2c(Task.Task):
     color = 'BLUE'
     map = {
         39: "'",
         92: "\\",
         7:  "a",
         8:  "b",
         9:  "t",
         10: "n",
         11: "v",
         12: "f",
         13: "r",
     }
 
     def __init__(self, *k, **kw):
         Task.Task.__init__(self, *k, **kw)
         if True:
             self.name_prefix = 'luaJIT_BC_'
             self.quote_name = lambda x: x
         else:
             self.name_prefix = 'BIN2C'
             self.quote_name = Utils.quote_define_name
 
     def run(self):
         inp = self.inputs[0]
         data = inp.read('rb')
         hash = sha256(data).hexdigest()
 
         name = self.name_prefix + self.quote_name(inp.change_ext('').name)
         out = """
 #ifndef %s_H_INCLUDED_%s
 #define %s_H_INCLUDED_%s
 #pragma once
 
 static constexpr const size_t %s_SIZE = %d;
 static const char %s[] = {
 """ % (name, hash, name, hash, name, len(data), name)
         i = 0
         for c in data:
             if i > 16:
                 out += "\n"
                 i = 0
             i += 1
             if c in self.map:
                 out += "'\\%s'," % self.map[c]
             elif c >= 32 and c < 127:
                 out += "'%s'," % chr(c)
             else:
                 out += "'\%o'," % int(c)
         out += "\n};\n\n#endif\n"
         self.outputs[0].write(out)
 
 from waflib.TaskGen import extension
 @extension('.lua')
 def lua_to_h(self, node):
     if not self.env.WITH_LUA:
         self.bld.fatal('Trying to compile .lua in build with no lua')
 
     out = node.change_ext('.lua.h')
     if self.env.LUAC_MODE == 'copy':
         self.create_task('bin2c', node, out)
     elif self.env.LUAC_MODE == 'system-luajit':
         self.create_task('luajit', node, out)
     elif self.env.LUAC_MODE == 'system-luac':
         self.luac_task(node, out, self.env.LUAC[0])
     elif self.env.LUAC_MODE == 'luac':
         luac = self.bld.get_tgen_by_name('luac').link_task.outputs[0]
         tsk = self.luac_task(node, out, luac.abspath())
         tsk.dep_nodes += [luac]
     elif self.env.LUAC_MODE == 'luac-wrapper':
         luac = self.bld.get_tgen_by_name('luac').link_task.outputs[0]
         tsk = self.luac_task(node, out, self.env.LUAC + [luac.abspath()])
         tsk.dep_nodes += [luac]
     else:
         self.bld.fatal('Unknown LUAC_MODE %s' % self.env.LUAC_MODE)
 
 # task .lua->.h
 class luajit(Task.Task):
     run_str = '${LUAC} -bt h ${LUACFLAGS} ${SRC} ${TGT}'
     color = 'BLUE'