adjust linux install

Author: arch

contrib/Installer/assets/main.lua

@@ -1,4 +1,5 @@
-json = require "json"
+json = require "json.lua"
+-- MTFG LUA Wrappper Version 2.0.0
 
 -- global var
 processHandleMTFG = nil
@@ -39,7 +40,7 @@ end
 
 platform = get_platform()
 
-function start_funscript_generator()
+function binding.start_funscript_generator()
     if processHandleMTFG then
         print('MTFG already running')
         return
@@ -56,12 +57,12 @@ function start_funscript_generator()
     print("currentScriptIdx: ", scriptIdx)
     print("currentTimeMs: ", currentTimeMs)
 
-    local next_action = ofs.ClosestActionAfter(script, currentTimeMs / 1000)
-    if next_action and script.actions[next_action].at < (currentTimeMs + 500) then
-        next_action = ofs.ClosestActionAfter(script, script.actions[next_action].at / 1000)
+    local next_action, _ = script:closestActionAfter(currentTimeMs / 1000.0)
+    if next_action and (next_action.at*1000) < (currentTimeMs + 500) then
+        next_action, _ = script:closestActionAfter(next_action.at)
     end
 
-    print("nextAction: ", next_action and tostring(script.actions[next_action].at) or "nil")
+    print("nextAction: ", next_action and tostring(next_action.at) or "nil")
 
     local cmd = ""
     local args = {}
@@ -97,47 +98,18 @@ function start_funscript_generator()
 
     if next_action then
         table.insert(args, "-e")
-        table.insert(args, tostring(script.actions[next_action].at))
+        table.insert(args, tostring(next_action.at*1000.0))
     end
 
     print("cmd: ", cmd)
     print("args: ", table.unpack(args))
 
-    processHandleMTFG = ofs.CreateProcess(cmd, table.unpack(args))
+    processHandleMTFG = Process.new(cmd, table.unpack(args))
 
     status = "MTFG running"
 end
 
 
-function import_funscript_generator_csv_result()
-    status = "MTFG not running"
-    local tmpFile = ofs.ExtensionDir() .. "/" .. tmpFileName
-    local f = io.open(tmpFile)
-    if not f then
-        print('Funscript Generator csv output file not found')
-        return
-    end
-
-    script = ofs.Script(scriptIdx)
-    local k = 1
-    for line in f:lines() do
-        -- first line is header
-        if k > 1 then
-            for at, pos in string.gmatch(line, "(%w+);(%w+)") do
-                ofs.AddAction(script, at, pos, true)
-            end
-        end
-        k = k + 1
-    end
-    f:close()
-
-    -- save changes
-    ofs.Commit(script)
-
-    -- delete csv file
-    os.remove(tmpFile)
-end
-
 function import_funscript_generator_json_result()
     status = "MTFG not running"
     local tmpFile = ofs.ExtensionDir() .. "/" .. tmpFileName
@@ -154,16 +126,23 @@ function import_funscript_generator_json_result()
 
     if multiaxis then
         local i = 1
-        while ofs.Script(i) do
-            name = ofs.ScriptTitle(i)
+        while i <= ofs.ScriptCount() do
+            name = ofs.ScriptName(i)
             for k,v in pairs(scriptAssignment) do
                 if name and name == scriptNames[v.idx] then
                     if actions[k] then
                         script = ofs.Script(i)
                         for _, action in pairs(actions[k]) do
-                            ofs.AddAction(script, action["at"], action["pos"], true)
+                            local closest_action, _ = script:closestAction(action["at"])
+                             local filtered = false
+                            if closest_action and math.abs(closest_action.at - (action["at"]/1000.0)) < 0.01 then
+                                filtered = true
+                            else
+                                local a = Action.new(action["at"]/1000.0, action["pos"], true)
+	                        table.insert(script.actions, a)
+                            end
                         end
-                        ofs.Commit(script)
+                        script:commit()
                     end
                 end
             end
@@ -173,14 +152,20 @@ function import_funscript_generator_json_result()
         script = ofs.Script(scriptIdx)
 
         for metric, actions_metric in pairs(actions) do
-            print('add ', metric, ' to ', ofs.ScriptTitle(scriptIdx))
+            print('add ', metric, ' to ', ofs.ScriptName(scriptIdx))
             for _, action in pairs(actions_metric) do
-                ofs.AddAction(script, action["at"], action["pos"], true)
+                local closest_action, _ = script:closestAction(action["at"]/1000.0)
+                local filtered = false
+                if closest_action and math.abs(closest_action.at - (action["at"]/1000.0)) < 0.01 then
+                    filtered = true
+                else
+                    local a = Action.new(action["at"]/1000.0, action["pos"], true)
+	            table.insert(script.actions, a)
+                end
             end
         end
 
-        -- save changes
-        ofs.Commit(script)
+        script:commit()
     end
 
     -- delete json file
@@ -195,7 +180,7 @@ function invert_selected()
             action.pos = clamp(100 - action.pos, 0, 100)
         end
     end
-    ofs.Commit(script)
+    script:commit()
 end
 
 
@@ -219,16 +204,16 @@ function align_bottom_points(align_value)
         if action.selected then
             local bottom_point = true
 
-            local next_action = ofs.ClosestActionAfter(script, action.at / 1000)
+            local next_action, _ = script:closestActionAfter(action.at / 1000)
             if next_action then
-                if script.actions[next_action].pos <= action.pos then
+                if next_action.pos <= action.pos then
                     bottom_point = false
                 end
             end
 
-            local prev_action = ofs.ClosestActionBefore(script, action.at / 1000)
+            local prev_action, _ = script:closestActionBefore(action.at / 1000)
             if prev_action then
-                if script.actions[prev_action].pos <= action.pos then
+                if prev_action.pos <= action.pos then
                     bottom_point = false
                 end
             end
@@ -239,7 +224,7 @@ function align_bottom_points(align_value)
         end
     end
 
-    ofs.Commit(script)
+    script:commit()
 end
 
 
@@ -263,16 +248,16 @@ function align_top_points(align_value)
         if action.selected then
             local top_point = true
 
-            local next_action = ofs.ClosestActionAfter(script, action.at / 1000)
+            local next_action, _ = script:closestActionAfter(action.at / 1000)
             if next_action then
-                if script.actions[next_action].pos >= action.pos then
+                if next_action.pos >= action.pos then
                     top_point = false
                 end
             end
 
-            local prev_action = ofs.ClosestActionBefore(script, action.at / 1000)
+            local prev_action, _ = script:closestActionBefore(action.at / 1000)
             if prev_action then
-                if script.actions[prev_action].pos >= action.pos then
+                if prev_action.pos >= action.pos then
                     top_point = false
                 end
             end
@@ -283,13 +268,12 @@ function align_top_points(align_value)
         end
     end
 
-    ofs.Commit(script)
+    script:commit()
 end
 
 
 function init()
     print("Detected OS: ", platform)
-    ofs.Bind("start_funscript_generator", "execute the funcript generator")
     local version_file_path = ""
     if platform == "Windows" then
         version_file_path = ofs.ExtensionDir().."\\funscript-editor\\funscript_editor\\VERSION.txt"
@@ -338,8 +322,9 @@ function update_script_names()
     scriptNamesCount = 0
     scriptNames = {'ignore'}
     scriptNamesCount = scriptNamesCount + 1
-    while ofs.Script(i) do
-        name = ofs.ScriptTitle(i)
+    local i = 1
+    while i <= ofs.ScriptCount() do
+        name = ofs.ScriptName(i)
         if not is_empty(name) then
             table.insert(scriptNames, name)
             scriptNamesCount = scriptNamesCount + 1
@@ -362,7 +347,7 @@ end
 
 function update(delta)
     updateCounter = updateCounter + 1
-    if processHandleMTFG and not ofs.IsProcessAlive(processHandleMTFG) then
+    if processHandleMTFG and not processHandleMTFG:alive() then
         print('funscript generator completed import result')
         processHandleMTFG = nil
         import_funscript_generator_json_result()
@@ -384,7 +369,7 @@ function gui()
     if not processHandleMTFG then
 
         if ofs.Button("Start MTFG") then
-            start_funscript_generator()
+            binding.start_funscript_generator()
         end
     else
         if ofs.Button("Kill MTFG") then
@@ -399,7 +384,7 @@ function gui()
     ofs.SameLine()
     if ofs.Button("Open Config") then
         if platform == "Windows" then
-            processHandleConfigDir = ofs.CreateProcess("explorer.exe", ofs.ExtensionDir().."\\funscript-editor\\funscript_editor\\config")
+            processHandleConfigDir = Process.new("explorer.exe", ofs.ExtensionDir().."\\funscript-editor\\funscript_editor\\config")
         else
             local cmd = '/usr/bin/dbus-send --session --print-reply --dest=org.freedesktop.FileManager1 --type=method_call /org/freedesktop/FileManager1 org.freedesktop.FileManager1.ShowItems array:string:"file://'
                 ..ofs.ExtensionDir()..'/Python-Funscript-Editor/funscript_editor/config/" string:""'
@@ -412,12 +397,12 @@ function gui()
         if platform == "Windows" then
             ofs.SameLine()
             if ofs.Button("Open Log") then
-                 processHandleLogFile = ofs.CreateProcess("notepad.exe", "C:/Temp/funscript_editor.log")
+                 processHandleLogFile = Process.new("notepad.exe", "C:/Temp/funscript_editor.log")
             end
         else
             ofs.SameLine()
             if ofs.Button("Open Log") then
-                processHandleLogFile = ofs.CreateProcess("/usr/bin/xdg-open", "/tmp/funscript_editor.log")
+                processHandleLogFile = Process.new("/usr/bin/xdg-open", "/tmp/funscript_editor.log")
             end
         end
     end
@@ -450,20 +435,23 @@ function gui()
 
     ofs.Separator()
 
-    ofs.Text("Post-Processing:")
-    ofs.SameLine()
-    if ofs.Button("Invert") then
-        invert_selected()
-    end
+    local enable_post_processing = false  -- broken
+    if enable_post_processing then
+        ofs.Text("Post-Processing:")
+        ofs.SameLine()
+        if ofs.Button("Invert") then
+            invert_selected()
+        end
 
-    ofs.SameLine()
-    if ofs.Button("Align Bottom Points") then
-        align_bottom_points(-1)
-    end
+        ofs.SameLine()
+        if ofs.Button("Align Bottom Points") then
+            align_bottom_points(-1)
+        end
 
-    ofs.SameLine()
-    if ofs.Button("Align Top Points") then
-        align_top_points(-1)
+        ofs.SameLine()
+        if ofs.Button("Align Top Points") then
+             align_top_points(-1)
+        end
     end
 
 end