Expanding LuaSlice.

Author: nicolasnoble

src/lua/fileffi-cdef.lua

@@ -89,8 +89,11 @@ LuaFile* dupFile(LuaFile*);
 
 LuaFile* zReader(LuaFile*, int64_t size, bool raw);
 
+LuaSlice* createEmptySlice();
 uint64_t getSliceSize(LuaSlice*);
 const void* getSliceData(LuaSlice*);
+void* getSliceMutableData(LuaSlice*);
+void resizeSlice(LuaSlice* slice, uint32_t size);
 void destroySlice(LuaSlice*);
 
 LuaFile* mem4g();

src/lua/fileffimeta.lua

@@ -27,12 +27,28 @@ local sliceMeta = {
             return buffer[index]
         elseif index == 'data' then
             return C.getSliceData(slice._wrapper)
+        elseif index == 'mutable' then
+            return C.getSliceMutableData(slice._wrapper)
         elseif index == 'size' then
             return tonumber(C.getSliceSize(slice._wrapper))
+        elseif index == 'resize' then
+            return function(slice, size)
+                C.resizeSlice(slice._wrapper, size)
+            end
         end
         error('Unknown index `' .. index .. '` for LuaSlice')
     end,
-    __newindex = function(slice, index, value) end,
+    __newindex = function(slice, index, value)
+        if type(index) == 'number' and index >= 0 and index < C.getSliceSize(slice._wrapper) then
+            local data = C.getSliceMutableData(slice._wrapper)
+            local buffer = ffi.cast('uint8_t*', data)
+            buffer[index] = value
+        elseif index == 'size' then
+            C.resizeSlice(slice._wrapper, value)
+        else
+            error('Unknown or immutable index `' .. index .. '` for LuaSlice')
+        end
+    end,
 }
 
 local function createSliceWrapper(wrapper)
@@ -71,5 +87,6 @@ local LuaBuffer = ffi.metatype('LuaBuffer', bufferMeta)
 
 Support.File._LuaBuffer = LuaBuffer
 Support.File._createSliceWrapper = createSliceWrapper
+Support.File.createEmptySlice = function() return createSliceWrapper(C.createEmptySlice()) end
 
 -- )EOF"

src/lua/luafile.cc

@@ -167,10 +167,11 @@ LuaFile* zReader(LuaFile* wrapper, int64_t size, bool raw) {
                            : new PCSX::ZReader(wrapper->file, size));
 }
 
+PCSX::Slice* createEmptySlice() { return new PCSX::Slice(); }
 uint64_t getSliceSize(PCSX::Slice* slice) { return slice->size(); }
-
 const void* getSliceData(PCSX::Slice* slice) { return slice->data(); }
-
+void* getSliceMutableData(PCSX::Slice* slice) { return slice->mutableData(); }
+void resizeSlice(PCSX::Slice* slice, uint32_t size) { slice->resize(size); }
 void destroySlice(PCSX::Slice* slice) { delete slice; }
 
 int readFileUserData(PCSX::Lua L) {
@@ -343,8 +344,11 @@ static void registerAllSymbols(PCSX::Lua L) {
 
     REGISTER(L, zReader);
 
+    REGISTER(L, createEmptySlice);
     REGISTER(L, getSliceSize);
     REGISTER(L, getSliceData);
+    REGISTER(L, getSliceMutableData);
+    REGISTER(L, resizeSlice);
     REGISTER(L, destroySlice);
 
     REGISTER(L, mem4g);