add multiaxis assignment to lua next

Author: arch

contrib/Installer/assets/main.lua.next

@@ -9,6 +9,10 @@ updateCounter = 0
 scriptIdx = 0
 mtfgVersion = "0.0.0"
 status = "MTFG not running"
+multiaxis = false
+tmpFileName = "funscript_actions.json"
+scriptNames = {}
+scriptAssignment = {x={idx=1}, y={idx=1}, roll={idx=1}}
 
 function exists(file)
    return os.rename(file, file)
@@ -39,7 +43,7 @@ function start_funscript_generator()
     end
 
     scriptIdx = ofs.ActiveIdx()
-    local tmpFile = ofs.ExtensionDir() .. "/funscript_actions.json"
+    local tmpFile = ofs.ExtensionDir() .. "/" .. tmpFileName 
     local video = player.CurrentVideo()
     local script = ofs.Script(scriptIdx)
     local currentTimeMs = player.CurrentTime() * 1000
@@ -72,6 +76,11 @@ function start_funscript_generator()
     end
 
     table.insert(args, "--generator")
+    
+    if multiaxis then;
+        table.insert(args, "--multiaxis")
+    end
+
     table.insert(args, "-s")
     table.insert(args, tostring(currentTimeMs))
     table.insert(args, "-i")
@@ -95,7 +104,7 @@ end
 
 function import_funscript_generator_csv_result()
     status = "MTFG not running"
-    local tmpFile = ofs.ExtensionDir() .. "/funscript_actions.csv"
+    local tmpFile = ofs.ExtensionDir() .. "/" .. tmpFileName
     local f = io.open(tmpFile)
     if not f then
         print('Funscript Generator csv output file not found')
@@ -124,7 +133,7 @@ end
 
 function import_funscript_generator_json_result()
     status = "MTFG not running"
-    local tmpFile = ofs.ExtensionDir() .. "/funscript_actions.json"
+    local tmpFile = ofs.ExtensionDir() .. "/" .. tmpFileName
     local f = io.open(tmpFile)
     if not f then
         print('Funscript Generator json output file not found')
@@ -136,17 +145,36 @@ function import_funscript_generator_json_result()
     json_body = json.decode(content)
     actions = json_body["actions"]
 
-    script = ofs.Script(scriptIdx)
+    if multiaxis then
+        local i = 1
+        while ofs.Script(i) do
+            name = ofs.ScriptTitle(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)
+                        end
+                        ofs.Commit(script)
+                    end
+                end
+            end
+       	    i = i + 1
+        end
+    else
+        script = ofs.Script(scriptIdx)
 
-    for metric, actions_metric in pairs(actions) do
-        print('metric', metric)
-        for _, action in pairs(actions_metric) do
-            ofs.AddAction(script, action["at"], action["pos"], true)
+        for metric, actions_metric in pairs(actions) do
+            print('add ', metric, ' to ', ofs.ScriptTitle(scriptIdx))
+            for _, action in pairs(actions_metric) do
+                ofs.AddAction(script, action["at"], action["pos"], true)
+            end
         end
-    end
 
-    -- save changes
-    ofs.Commit(script)
+        -- save changes
+        ofs.Commit(script)
+    end
 
     -- delete json file
     os.remove(tmpFile)
@@ -276,15 +304,7 @@ function init()
     end
 end
 
-
-function update(delta)
-    updateCounter = updateCounter + 1
-    if processHandleMTFG and not ofs.IsProcessAlive(processHandleMTFG) then
-        print('funscript generator completed import result')
-        processHandleMTFG = nil
-        import_funscript_generator_json_result()
-    end
-    if math.fmod(updateCounter, 1000) == 1 then
+function update_logfile_exists()
         local logfile = ""
         if platform == "Windows" then
             logfile = "C:/Temp/funscript_editor.log"
@@ -298,6 +318,37 @@ function update(delta)
         else
             logfileExist = false
         end
+end
+
+
+function is_empty(s)
+  return s == nil or s == ''
+end
+
+
+function update_script_names()
+    local i = 1
+    scriptNames = {'ignore'}
+    while ofs.Script(i) do
+       name = ofs.ScriptTitle(i)
+       if not is_empty(name) then
+           table.insert(scriptNames, name)
+       end
+       i = i + 1
+    end
+end
+
+
+function update(delta)
+    updateCounter = updateCounter + 1
+    if processHandleMTFG and not ofs.IsProcessAlive(processHandleMTFG) then
+        print('funscript generator completed import result')
+        processHandleMTFG = nil
+        import_funscript_generator_json_result()
+    end
+    if math.fmod(updateCounter, 100) == 1 then
+        update_logfile_exists()
+        update_script_names()
     end
 end
 
@@ -350,6 +401,21 @@ function gui()
     end
 
     ofs.Separator()
+
+    multiaxis, valueChanged = ofs.Checkbox("multiaxis", multiaxis)
+
+    if multiaxis then
+        local comboNum = 1
+        for k,v in pairs(scriptAssignment) do
+              ofs.Text(k.." => ")
+              ofs.SameLine()
+              v.idx, _ = ofs.Combo("#"..tostring(comboNum), v.idx, scriptNames)
+              comboNum = comboNum + 1
+        end
+    end
+
+    ofs.Separator()
+
     ofs.Text("Post-Processing:")
     ofs.SameLine()
     if ofs.Button("Invert") then