luaScriptResourceManager.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2016 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
#pragma once

#include <vrfLua/vrfLuaDefines.h>
#include <vrfobjcore/scriptResourceManager.h>
#include <vlutil/vlFilename.h>
#include <lua.hpp>
#include <list>
#include <map>
#include <vlutil/vlSmartPointers.h>
#include <string>
#include <vrfutil/scriptedTaskMetaData.h>
#include <tbb/mutex.h>

class DT_DLL_vrfLua DtLuaScriptResourceManager : public DtScriptResourceManager
{
public:
   DtLuaScriptResourceManager();
   virtual ~DtLuaScriptResourceManager() override;

   virtual void initializeNumberOfStates(unsigned numberOfStates) override;

   virtual lua_State* getLuaState(int scriptType = DtScriptedTaskScriptedTask);

   virtual bool loadModule(lua_State* L, const std::string& moduleName);

   static lua_State* createLuaState(const std::vector<std::string>& scriptPaths, int scriptType = DtScriptedTaskScriptedTask);

   static bool initializeVRFLuaExtensions(lua_State* L, const std::vector<std::string>& scriptPaths, int scriptType = DtScriptedTaskScriptedTask);

   static DtScriptResourceManager* creator();

protected:

   virtual std::string findLuaModuleFile(const std::string& filename);

   virtual void saveBytecodeChunk(lua_State* L, const std::string& chunkName);

   typedef std::vector<char> DtLuaBytecodeChunk;
   typedef std::shared_ptr<DtLuaBytecodeChunk> DtLuaBytecodeChunkPtr;
   struct DtBytecodeRwInteface
   {
      DtLuaBytecodeChunkPtr bytecodeChunk;
      int loadChunkNum;
   };

   virtual void createLuaStates(unsigned numStates);

   static void openLuaLibs(lua_State* L);

   DtLuaScriptResourceManager(const DtLuaScriptResourceManager& other) = delete;
   DtLuaScriptResourceManager& operator=(const DtLuaScriptResourceManager& other) = delete;

   static int luaBytecodeSaveCb(lua_State *L, const void* p, size_t sz, void* ud);

   static const char* luaBytecodeLoadCb(lua_State *L, void *ud, size_t *size);

   std::list<lua_State*> myLuaStates;
   std::map<std::string, DtLuaBytecodeChunkPtr > myLuaBytecodeChunks;

   tbb::mutex myBytecodeMutex;
};