improve a-b

1 files changed, 23 insertions, 10 deletions

diff --git a/a-b.lua b/a-b.lua
index e05bd47..994361f 100644
--- a/a-b.lua
+++ b/a-b.lua
@@ -1,21 +1,34 @@
 script_name = "A-B"
-script_description = "makes checking pre-timing possible.\nCAUTION: Overwrites every selected line!"
+script_description = "makes checking pre-timing possible."
 script_author = "garret"
-script_version = "2021-04-03"
+script_version = "2021-07-10"
+
+function switch_number(i)
+    if i == "a" then
+        return "b"
+    elseif i == "b" then
+        return "a"
+    end
+end
+
 function main(sub, sel)
-    local i = 0
+    local i = "a"
     for si,li in ipairs(sel) do
         line = sub[li]
-        if i == 0 then
-            line.text = line.actor.." a"
-            i = 1
-        elseif i == 1 then
-            line.text = line.actor.." b"
-            i = 0
+        if line.actor == "" then
+            indicator = i
+        else
+        indicator = line.actor.." "..i
+        end
+        if line.text == "" then
+            line.text = indicator
+        elseif line.text:gsub("{[^}]-}","") == "" then
+            line.text = indicator.." "..line.text
         end
         sub[li] = line
+        i = switch_number(i)
     end
     aegisub.set_undo_point(script_name)
-    return sel
 end
+
 aegisub.register_macro(script_name, script_description, main)