recalculator.lua

-- Example: Set to 120%, check fscx and fscy, and all values for fscx/y will be increased by 20% for selected lines.
-- With "multiply/add more with each line" and fscx100 you'll get 120, 140, 160, 180 for consecutive lines.
-- Alternative 2nd value allows for a different value for all Y things (fscY, Ybord, Yshad, frY, faY, all Y coordinates) + fad2, t2.
--   It will be used as Multiply or Add depending on the button you press.

script_name="Recalculator"
script_description="recalculates sizes of things"
script_author="unanimated"
script_version="2.11"

-- SETTINGS: type the names of checkboxes you want checked by default as you see them in the GUI, separated by commas

checked="fscx,fscy,anchor clip"
default_rounding=2

-- SETTINGS END

function calc(num)
    if pressed=="Multiply" then num=round(num+num*count) end
    if pressed=="Add" then num=round(num+(res.add*linec))end
    if neg==0 and num<0 then num=0 end
    return num
end

function calc2(num)
    if pressed=="Multiply" then num=round(num+num*altcount) end
    if pressed=="Add" then num=round(num+(alt*linec))end
    if neg==0 and num<0 then num=0 end
    return num
end

function round(num) num=math.floor(num*rnd+0.5)/rnd return num end

function multiply(subs, sel)
    c=(res.pc-100)/100
    oc=res.pc/100
    rnd=10^res.rnd
    if res.alt then ac=res.altval/100 else ac=oc end
    if pressed=="Multiply" then alt=(res.altval-100)/100 else alt=res.altval end
    if res.mov1 or res.mov2 or res.mov3 or res.mov4 then move=1 else move=0 end
    if res.clipx or res.clipy then clip=1 else clip=0 end
    for x, i in ipairs(sel) do
        if res.byline then count=x*c linec=x altcount=x*alt else count=c linec=1 altcount=alt end
	if not res.alt then altcount=count alt=res.add end
	line=subs[i]
	text=line.text
	styleref=stylechk(line.style)
	if not text:match("^{\\") then text="{\\}"..text end
	
		scx=styleref.scale_x
	if res.fscx and not text:match("\\fscx") then text=text:gsub("^({\\[^}]*)}","%1\\fscx"..scx.."}") end
		scy=styleref.scale_y
	if res.fscy and not text:match("\\fscy") then text=text:gsub("^({\\[^}]*)}","%1\\fscy"..scy.."}") end
		fsize=styleref.fontsize
	if res.fs and not text:match("\\fs%d") then text=text:gsub("^({\\[^}]*)}","%1\\fs"..fsize.."}") end 
		brdr=styleref.outline
	if res.bord and not text:match("\\bord") and brdr~=0 then text=text:gsub("^({\\[^}]*)}","%1\\bord"..brdr.."}") end
		shdw=styleref.shadow
	if res.shad and not text:match("\\shad") and shdw~=0 then text=text:gsub("^({\\[^}]*)}","%1\\shad"..shdw.."}") end
		spac=styleref.spacing
	if res.fsp and not text:match("\\fsp") and spac~=0 then text=text:gsub("^({\\[^}]*)}","%1\\fsp"..spac.."}") end
	
	if res.fscx then neg=0 text=text:gsub("\\fscx([%d%.]+)",function(a) return "\\fscx"..calc(tonumber(a)) end) end
	if res.fscy then neg=0 text=text:gsub("\\fscy([%d%.]+)",function(a) return "\\fscy"..calc2(tonumber(a)) end) end
	if res.fs then neg=0 text=text:gsub("\\fs([%d%.]+)",function(a) return "\\fs"..calc(tonumber(a)) end) end
	if res.fsp then neg=1 text=text:gsub("\\fsp([%d%.%-]+)",function(a) return "\\fsp"..calc(tonumber(a)) end) end
	if res.bord then neg=0 text=text:gsub("\\bord([%d%.]+)",function(a) return "\\bord"..calc(tonumber(a)) end) end
	if res.shad then neg=0 text=text:gsub("\\shad([%d%.]+)",function(a) return "\\shad"..calc(tonumber(a)) end) end
	if res.blur then neg=0 text=text:gsub("\\blur([%d%.]+)",function(a) return "\\blur"..calc(tonumber(a)) end) end
	if res.be then neg=0 text=text:gsub("\\be([%d%.]+)",function(a) return "\\be"..calc(tonumber(a)) end) end
	if res.xbord then neg=0 text=text:gsub("\\xbord([%d%.]+)",function(a) return "\\xbord"..calc(tonumber(a)) end) end
	if res.ybord then neg=0 text=text:gsub("\\ybord([%d%.]+)",function(a) return "\\ybord"..calc2(tonumber(a)) end) end
	if res.xshad then neg=1 text=text:gsub("\\xshad([%d%.%-]+)",function(a) return "\\xshad"..calc(tonumber(a)) end) end
	if res.yshad then neg=1 text=text:gsub("\\yshad([%d%.%-]+)",function(a) return "\\yshad"..calc2(tonumber(a)) end) end
	if res.frx then neg=1 text=text:gsub("\\frx([%d%.%-]+)",function(a) return "\\frx"..calc(tonumber(a)) end) end
	if res.fry then neg=1 text=text:gsub("\\fry([%d%.%-]+)",function(a) return "\\fry"..calc2(tonumber(a)) end) end
	if res.frz then neg=1 text=text:gsub("\\frz([%d%.%-]+)",function(a) return "\\frz"..calc(tonumber(a)) end) end
	if res.fax then neg=1 text=text:gsub("\\fax([%d%.%-]+)",function(a) return "\\fax"..calc(tonumber(a)) end) end
	if res.fay then neg=1 text=text:gsub("\\fay([%d%.%-]+)",function(a) return "\\fay"..calc2(tonumber(a)) end) end
	if res.kara then neg=0 text=text:gsub("{(\\[^}]-)}{(\\[^}]-)}","{%1%2}")
		:gsub("^({[^}]-\\[Kk][fo]?)([%d%.]+)",function(a,b) return a..calc(tonumber(b)) end) end
	
	if res.posx then neg=1 text=text:gsub("\\pos%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\pos("..calc(tonumber(a))..","..b..")" end) end
	if res.posy then neg=1 text=text:gsub("\\pos%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\pos("..a..","..calc2(tonumber(b))..")" end) end
	
	if move==1 then neg=1 text=text:gsub("\\move%(([%d%.%-]+),([%d%.%-]+),([%d%.%-]+),([%d%.%-]+)",function(a,b,c,d) 
		if res.mov1 then a=calc(tonumber(a)) end
		if res.mov2 then b=calc2(tonumber(b)) end
		if res.mov3 then c=calc(tonumber(c)) end
		if res.mov4 then d=calc2(tonumber(d)) end
	return "\\move("..a..","..b..","..c..","..d end) end
	
	if res.orgx then neg=1 text=text:gsub("\\org%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\org("..calc(tonumber(a))..","..b..")" end) end
	if res.orgy then neg=1 text=text:gsub("\\org%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\org("..a..","..calc2(tonumber(b))..")" end) end
	
	if res.fad1 then neg=0 text=text:gsub("\\fad%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\fad("..calc(tonumber(a))..","..b..")" end) end
	if res.fad2 then neg=0 text=text:gsub("\\fad%(([%d%.%-]+),([%d%.%-]+)%)",function(a,b) 
	return "\\fad("..a..","..calc2(tonumber(b))..")" end) end
	
	if clip==1 then neg=1
		if res.anchor and pressed=="Multiply" then m=1/oc m2=1/ac
		  text=text:gsub("clip%(([%d%.%-]+),([%d%.%-]+),([%d%.%-]+),([%d%.%-]+)",
		    function(a,b,c,d) x=0 y=0
		      if res.clipx then x=(a+c)/2-((a+c)/2)*m end
		      if res.clipy then y=(b+d)/2-((b+d)/2)*m2 end
		    return "clip("..a-x..","..b-y..","..c-x..","..d-y end)
		  if text:match("clip%(m [%d%a%s%-%.]+%)") then
		    ctext=text:match("clip%(m ([%d%a%s%-%.]+)%)")
		    c1,c2=ctext:match("([%d%-%.]+)%s([%d%-%.]+)")
		    x=0 y=0
		    if res.clipx then x=c1-c1*m end
		    if res.clipy then y=c2-c2*m2 end
		    ctext2=ctext:gsub("([%d%-%.]+)%s([%d%-%.]+)",function(a,b) return a-x.." "..b-y end)
		    ctext=ctext:gsub("%-","%%-")
		    text=text:gsub("clip%(m "..ctext,"clip(m "..ctext2)
	          end
		end
	      text=text:gsub("clip%(([%d%.%-]+),([%d%.%-]+),([%d%.%-]+),([%d%.%-]+)%)",function(a,b,c,d)
		if res.clipx then a=calc(tonumber(a)) c=calc(tonumber(c)) end
		if res.clipy then b=calc2(tonumber(b)) d=calc2(tonumber(d)) end
	      return "clip("..a..","..b..","..c..","..d..")" end) 
	      if text:match("clip%(m [%d%a%s%-%.]+%)") then
		ctext=text:match("clip%(m ([%d%a%s%-%.]+)%)")
		ctext2=ctext:gsub("([%d%-%.]+)%s([%d%-%.]+)",function(a,b)
		  if res.clipx then a=calc(tonumber(a)) end
		  if res.clipy then b=calc2(tonumber(b)) end
		return a.." "..b end)
		ctext=ctext:gsub("%-","%%-")
		text=text:gsub(ctext,ctext2)
	      end
	end
	    
	if res.drawx or res.drawy then neg=1
	      if text:match("\\p[1-9]") and text:match("}m [%d%a%s%-%.]+") then
	      dtext=text:match("}m ([%d%a%s%-%.]+)")
	      dtext2=dtext:gsub("([%d%-%.]+)%s([%d%-%.]+)",function(a,b) 
		if res.drawx then xx=math.floor(calc(tonumber(a))+0.5) else xx=a end
		if res.drawy then yy=math.floor(calc2(tonumber(b))+0.5) else yy=b end
		return xx.." "..yy end)
	      dtext=dtext:gsub("%-","%%-")
	      text=text:gsub(dtext,dtext2)
	      end
	end
	    
	if res.ttim1 then neg=1 text=text:gsub("\\t%(([%d%.%-]+),([%d%.%-]+),",function(a,b) 
	return "\\t("..calc(tonumber(a))..","..b.."," end) end
	if res.ttim2 then neg=1 text=text:gsub("\\t%(([%d%.%-]+),([%d%.%-]+),",function(a,b) 
	return "\\t("..a..","..calc2(tonumber(b)).."," end) end
	    
	text=text:gsub("\\\\","\\")
	text=text:gsub("\\}","}")
	text=text:gsub("{}","")
	line.text=text
        subs[i]=line
    end
end

function styleget(subs)
    styles={}
    for i=1,#subs do
        if subs[i].class=="style" then
	    table.insert(styles,subs[i])
	end
	if subs[i].class=="dialogue" then break end
    end
end

function stylechk(stylename)
    for i=1,#styles do
	if stylename==styles[i].name then
	    styleref=styles[i]
	    if styles[i].name=="Default" then defaref=styles[i] end
	    break
	end
    end
    return styleref
end

function recalculator(subs, sel)
	dialog_config=
	{
	    {x=0,y=0,width=2,height=1,class="label",label="Change values to:",},
	    {x=2,y=0,width=3,height=1,class="floatedit",name="pc",value=100,min=0,hint="Multiply"},
	    {x=5,y=0,width=1,height=1,class="label",label="%",},
	    
	    {x=0,y=1,width=2,height=1,class="label",label="Increase values by:",},
	    {x=2,y=1,width=3,height=1,class="floatedit",name="add",value=0,hint="Add (use negative to subtract)"},
	    
	    {x=2,y=2,width=3,height=1,class="floatedit",name="altval",value=0,hint="Multiply/Add based on button pressed"},
	    {x=0,y=2,width=2,height=1,class="checkbox",name="alt",label="Alternative 2nd value",value=false,hint="Affects fscy, ybord, yshad, fry, fay, all Y coordinates, fad2, t2"},
	    
	    {x=0,y=3,width=2,height=1,class="label",label="Rounding:",},
	    {x=2,y=3,width=3,height=1,class="intedit",name="rnd",value=default_rounding,min=0,hint="How many decimals should be allowed"},
	    
	    {x=0,y=4,width=1,height=1,class="checkbox",name="fscx",label="fscx",value=true,},
	    {x=1,y=4,width=1,height=1,class="checkbox",name="fscy",label="fscy",value=true,},
	    {x=2,y=4,width=1,height=1,class="checkbox",name="fs",label="fs",value=false,},
	    {x=3,y=4,width=1,height=1,class="checkbox",name="fsp",label="fsp",value=false,},
	    {x=4,y=4,width=1,height=1,class="checkbox",name="blur",label="blur",value=false,},
	    {x=5,y=4,width=1,height=1,class="checkbox",name="be",label="be",value=false,},

	    {x=0,y=5,width=1,height=1,class="checkbox",name="bord",label="bord",value=false,},
	    {x=1,y=5,width=1,height=1,class="checkbox",name="shad",label="shad",value=false,},
	    {x=2,y=5,width=1,height=1,class="checkbox",name="xbord",label="xbord",value=false,},
	    {x=3,y=5,width=1,height=1,class="checkbox",name="ybord",label="ybord",value=false,},
	    {x=4,y=5,width=1,height=1,class="checkbox",name="xshad",label="xshad",value=false,},
	    {x=5,y=5,width=1,height=1,class="checkbox",name="yshad",label="yshad",value=false,},

	    {x=0,y=6,width=1,height=1,class="checkbox",name="frz",label="frz",value=false,},
	    {x=1,y=6,width=1,height=1,class="checkbox",name="frx",label="frx",value=false,},
	    {x=2,y=6,width=1,height=1,class="checkbox",name="fry",label="fry",value=false,},
	    {x=3,y=6,width=1,height=1,class="checkbox",name="fax",label="fax",value=false,},
	    {x=4,y=6,width=1,height=1,class="checkbox",name="fay",label="fay",value=false,},
	    {x=5,y=6,width=1,height=1,class="checkbox",name="kara",label="kara",value=false,hint="k/kf/ko. only the first one in the line."},

	    {x=0,y=7,width=1,height=1,class="checkbox",name="posx",label="pos x",value=false,},
	    {x=1,y=7,width=1,height=1,class="checkbox",name="posy",label="pos y",value=false,},
	    {x=2,y=7,width=1,height=1,class="checkbox",name="orgx",label="org x",value=false,},
	    {x=3,y=7,width=1,height=1,class="checkbox",name="orgy",label="org y",value=false,},
	    {x=4,y=7,width=1,height=1,class="checkbox",name="fad1",label="fad 1",value=false,},
	    {x=5,y=7,width=1,height=1,class="checkbox",name="fad2",label="fad 2",value=false,},

	    {x=0,y=8,width=1,height=1,class="checkbox",name="mov1",label="move1",value=false,},
	    {x=1,y=8,width=1,height=1,class="checkbox",name="mov2",label="move2",value=false,},
	    {x=2,y=8,width=1,height=1,class="checkbox",name="mov3",label="move3",value=false,},
	    {x=3,y=8,width=1,height=1,class="checkbox",name="mov4",label="move4",value=false,},

	    {x=4,y=8,width=2,height=1,class="checkbox",name="allpos",label="all pos/move/org",value=false,hint="same as all 8 checkboxes. \naffects only existing tags."},

	    {x=0,y=9,width=1,height=1,class="checkbox",name="clipx",label="clip x",value=false,},
	    {x=1,y=9,width=1,height=1,class="checkbox",name="clipy",label="clip y",value=false,},
	    {x=2,y=9,width=1,height=1,class="checkbox",name="drawx",label="draw x",value=false,},
	    {x=3,y=9,width=1,height=1,class="checkbox",name="drawy",label="draw y",value=false,},
	    {x=4,y=9,width=1,height=1,class="checkbox",name="ttim1",label="\\t 1",value=false,hint="\\t timecode 1"},
	    {x=5,y=9,width=1,height=1,class="checkbox",name="ttim2",label="\\t 2",value=false,hint="\\t timecode 2"},

	    {x=0,y=10,width=4,height=1,class="checkbox",name="byline",label="multiply/add more with each line",value=false,},
	    {x=4,y=10,width=2,height=1,class="checkbox",name="anchor",label="anchor clip",value=false,hint="anchor clip with Multiply"},
	} 	
	    chk=","..checked..","
	    chk=chk:gsub(" *, *",",") :gsub("\t","\\t")
	    for key,val in ipairs(dialog_config) do
		if val.class=="checkbox" and chk:match(","..val.label..",") then val.value=true end
	    end
	repeat
	  if pressed=="Clear" then
	    for key,val in ipairs(dialog_config) do
		if val.class=="checkbox" and val.name~="anchor" then val.value=false end
		if val.name=="anchor" then val.value=res.anchor end
		if val.name=="alt" then val.value=res.alt end
		if val.class:match("edit") then val.value=res[val.name] end
	    end
	  end
	pressed, res=aegisub.dialog.display(dialog_config,
		{"Multiply","Add","Clear","Cancel"},{ok='Multiply',cancel='Cancel'})
	until pressed~="Clear"
	if pressed=="Cancel" then    aegisub.cancel() end
	if res.allpos then 
		res.posx=true res.posy=true res.orgx=true res.orgy=true
		res.mov1=true res.mov2=true res.mov3=true res.mov4=true
	end
	if pressed=="Multiply" or pressed=="Add" then  styleget(subs)  multiply(subs, sel) end
	aegisub.set_undo_point(script_name)
	return sel
end

aegisub.register_macro(script_name, script_description, recalculator)