Code/Tools/RC/ResourceCompilerPC/LuaCompiler.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "ConvertContext.h"
   5 #include "IConfig.h"
   6 #include "FileUtil.h"
   7 #include "UpToDateFileHelpers.h"
   8
   9 #include "LuaCompiler.h"
  10
  11 extern "C"
  12 {
  13 #include <Lua/src/lua.h>
  14 #include <Lua/src/lauxlib.h>
  15
  16 #include <Lua/src/ldo.h>
  17 #include <Lua/src/lfunc.h>
  18 #include <Lua/src/lmem.h>
  19 #include <Lua/src/lobject.h>
  20 #include <Lua/src/lopcodes.h>
  21 #include <Lua/src/lstring.h>
  22 #include <Lua/src/lundump.h>
  23 }
  24
  25 extern "C"
  26 {
  27         float script_frand0_1()
  28         {
  29                 return rand() / static_cast<float>(RAND_MAX);
  30         }
  31
  32         void script_randseed(uint seed)
  33         {
  34                 srand(seed);
  35         }
  36 }
  37
  38 // Shamelessly stolen from luac.c and modified a bit to support rc usage
  39
  40 #define toproto(L,i) (clvalue(L->top+(i))->l.p)
  41
  42 static const Proto* combine(lua_State* L)
  43 {
  44         return toproto(L,-1);
  45 }
  46
  47 static int writer(lua_State* L, const void* p, size_t size, void* u)
  48 {
  49         UNUSED(L);
  50         return (fwrite(p,size,1,(FILE*)u)!=1) && (size!=0);
  51 }
  52
  53 static int pmain(lua_State* L)
  54 {
  55         LuaCompiler* pCompiler = reinterpret_cast<LuaCompiler*>(lua_touserdata(L, 1));
  56         const Proto* f;
  57         const char* filename = pCompiler->GetInFilename();
  58         if (luaL_loadfile(L,filename)!=0)
  59         {
  60                 RCLogError(lua_tostring(L,-1));
  61                 return 1;
  62         }
  63
  64         f = combine(L);
  65         if (pCompiler->IsDumping())
  66         {
  67                 const char* outputFilename = pCompiler->GetOutFilename();
  68
  69                 FILE* D = fopen(outputFilename, "wb");
  70                 if (D == NULL)
  71                 {
  72                         RCLogError("Cannot open %s", outputFilename);
  73                         return 1;
  74                 }
  75
  76                 lua_lock(L);
  77                 luaU_dump(L, f, writer, D, (int) pCompiler->IsStripping(), (int) pCompiler->IsBigEndian());
  78                 lua_unlock(L);
  79
  80                 if (ferror(D))
  81                 {
  82                         RCLogError("Cannot write to %s", outputFilename);
  83                         return 1;
  84                 }
  85
  86                 if (fclose(D))
  87                 {
  88                         RCLogError("Cannot close %s", outputFilename);
  89                         return 1;
  90                 }
  91         }
  92
  93         return 0;
  94 }
  95
  96 //////////////////////////////////////////////////////////////////////////
  97 LuaCompiler::LuaCompiler()
  98         : m_bIsDumping(true)
  99         , m_bIsStripping(true)
 100         , m_bIsBigEndian(false)
 101 {
 102 }
 103
 104 //////////////////////////////////////////////////////////////////////////
 105 LuaCompiler::~LuaCompiler()
 106 {
 107 }
 108
 109 ////////////////////////////////////////////////////////////
 110 string LuaCompiler::GetOutputFileNameOnly() const
 111 {
 112         return PathUtil::RemoveExtension(m_CC.sourceFileNameOnly) + ".lua";
 113 }
 114
 115 ////////////////////////////////////////////////////////////
 116 string LuaCompiler::GetOutputPath() const
 117 {
 118         return PathUtil::Make(m_CC.GetOutputFolder(), GetOutputFileNameOnly());
 119 }
 120
 121 //////////////////////////////////////////////////////////////////////////
 122 bool LuaCompiler::Process()
 123 {
 124         string sourceFile = m_CC.GetSourcePath();
 125         string outputFile = GetOutputPath();
 126         std::replace(sourceFile.begin(), sourceFile.end(), '/', '\\');
 127         std::replace(outputFile.begin(), outputFile.end(), '/', '\\');
 128
 129         if (!m_CC.bForceRecompiling && UpToDateFileHelpers::FileExistsAndUpToDate(GetOutputPath(), m_CC.GetSourcePath()))
 130         {
 131                 // The file is up-to-date
 132                 m_CC.pRC->AddInputOutputFilePair(m_CC.GetSourcePath(), GetOutputPath());
 133                 return true;
 134         }
 135
 136         bool ok = false;
 137
 138         const bool isPlatformBigEndian = m_CC.pRC->GetPlatformInfo(m_CC.platform)->bBigEndian;
 139
 140         m_bIsDumping = true;
 141         m_bIsStripping = true;
 142         m_bIsBigEndian = isPlatformBigEndian;
 143         m_sInFilename = sourceFile;
 144         m_sOutFilename = outputFile;
 145
 146         lua_State* L = lua_open();
 147
 148         if (L)
 149         {
 150                 if (lua_cpcall(L, pmain, this) == 0)
 151                 {
 152                         ok = true;
 153                 }
 154                 else
 155                 {
 156                         RCLogError(lua_tostring(L,-1));
 157                 }
 158
 159                 lua_close(L);
 160         }
 161         else
 162         {
 163                 RCLogError("Not enough memory for lua state");
 164         }
 165
 166         if (ok)
 167         {
 168                 if (!UpToDateFileHelpers::SetMatchingFileTime(GetOutputPath(), m_CC.GetSourcePath()))
 169                 {
 170                         return false;
 171                 }
 172                 m_CC.pRC->AddInputOutputFilePair(m_CC.GetSourcePath(), GetOutputPath());
 173         }
 174
 175         return ok;
 176 }