blur-and-glow.lua

--[[	"Blur with Layers" creates layers with blur.
	It supports 2 borders, xbord, ybord, xshad, and yshad. It has basic support for transforms and \r.
	
	"Blur and Glow" - Same as above but with an extra layer for glow. Set blur amount and alpha for the glow.
	
	The "double border" option additionally lets you change the size and colour of the 2nd border.
	
	If blur is missing, default_blur is added by default. (see settings)
	
	"Bottom blur" allows you to use different blur for the lowest non-glow layer than for top layer(s).
	
	"fix \\1a for layers with border and fade" - Uses \1a&HFF& for the duration of a fade on layers with border.
		"transition" - for \fad(500,0) with transition 80ms you get \1a&HFF&\t(420,500,\1a&H00&).
	
	"only add glow" - will add glow to a line with a border, without messing with the primary / border. (Blur + Glow)
	
	"only add 2nd border" - will add 2nd border, without messing with the primary / first border. (Blur / Layers)
	
	"Fix fades" - Recalculates those \1a fades mentioned above. 
	Use this when you shift something like an episode title to a new episode and the duration of the sign is different.
	
	"Change layer" - raises or lowers layer for all selected lines by the same amount. [This is separate from the other functions.]

]]

script_name="Blur and Glow"
script_description="Add blur and/or glow to signs"
script_author="unanimated"
script_url1="http://unanimated.xtreemhost.com/ts/blur-and-glow.lua"
script_url2="https://raw.githubusercontent.com/unanimated/luaegisub/master/blur-and-glow.lua"
script_version="2.2"

default_blur="0.6"


function glow(subs,sel)
    if not res.rep then al=res.alfa bl=res.blur end
    if res.glowcol then glowc=res.glc:gsub("#(%x%x)(%x%x)(%x%x)","&H%3%2%1&") end
    if res.autod then if res.clr or res.bsize then res.double=true end end
    for i=#sel,1,-1 do
	if aegisub.progress.is_cancelled() then aegisub.cancel() end
	aegisub.progress.title(string.format("Glowing line: %d/%d",(#sel-i+1),#sel))
	line=subs[sel[i]]
	text=line.text
	if defaref~=nil and line.style=="Default" then styleref=defaref
	elseif lastref~=nil and laststyle==line.style then styleref=lastref
	else styleref=stylechk(line.style) end
	lastref=styleref	laststyle=line.style
	duration=line.end_time-line.start_time

	    -- get colors, border, shadow from style
	    stylinfo(text)
	    text=preprocess(text)
	    line.text=text

	if border~="0" or text:match("\\[xy]bord") then

	    -- WITH TWO BORDERS
	  if res["double"] then

		-- second border
	    line1=line
	    line1.text=text
	    line1.text=borderline2(line1.text)
	    line1.layer=line1.layer+1
	    subs.insert(sel[i]+1,line1)

		-- first border
	    line2=line
	    line2.text=text
	    line2.text=borderline(line2.text)
	    if shadow~="0" then line2.text=line2.text:gsub("^({\\[^}]+)}","%1\\shad0}") end
	    line2.text=line2.text:gsub("\\shad[%d%.]+","\\shad0")
	    line2.layer=line2.layer+1
	    subs.insert(sel[i]+2,line2)

		-- top line
	    line3=line
	    line3.text=text
	    line3.text=topline(line3.text)
	    line3.layer=line3.layer+1
	    subs.insert(sel[i]+3,line3)

		-- bottom / glow
	    text=borderline2(text)
	    text=glowlayer(text,"3c","3")
	    if res.botalpha and line.text:match("\\fad%(") then text=botalfa(text) end
	    line.layer=line.layer-3
	    line.text=text
	    sls=3

	  else
	    -- WITH ONE BORDER

		-- border
	    line2=line
	    if not res.onlyg then
	    line2.text=text
	    line2.text=borderline(line2.text)
	    end
	    line2.layer=line2.layer+1
	    subs.insert(sel[i]+1,line2)

		-- top line
	    line3=line
	    line3.layer=line3.layer+1
	    if not res.onlyg then
	    line3.text=text
	    line3.text=topline(line3.text)
	    subs.insert(sel[i]+2,line3)
	    end

		-- bottom / glow
	    text=glowlayer(text,"3c","3")
	    if res.botalpha and line.text:match("\\fad%(") then text=botalfa(text) end
	    line.layer=line.layer-2
	    line.text=text
	    sls=2

	  end

	else
	    -- WITHOUT BORDER

	    line2=line
	    line2.layer=line2.layer+1
	    subs.insert(sel[i]+1,line2)
	    text=glowlayer(text,"c","1")
	    line.layer=line.layer-1
	    line.text=text
	    sls=1

	end
	subs[sel[i]]=line
	for s=i,#sel do sel[s]=sel[s]+sls end
    end
    aegisub.progress.title("Blur & Glow: DONE")
    return sel
end

function layerblur(subs,sel)
    if res.autod then if res.clr or res.bsize then res.double=true end end
    for i=#sel,1,-1 do
	if aegisub.progress.is_cancelled() then aegisub.cancel() end
	aegisub.progress.title(string.format("Blurring line: %d/%d",(#sel-i+1),#sel))
	line=subs[sel[i]]
	text=line.text
	if defaref~=nil and line.style=="Default" then styleref=defaref
	elseif lastref~=nil and laststyle==line.style then styleref=lastref
	else styleref=stylechk(line.style) end
	lastref=styleref	laststyle=line.style
	duration=line.end_time-line.start_time

	    -- get colors, border, shadow from style
	    stylinfo(text)
	    text=preprocess(text)
	    line.text=text

		-- TWO BORDERS
	    if res["double"] then

		-- first border
	    line2=line
	    if not res.onlyb then
	    line2.text=text
	    line2.text=borderline(line2.text)
	    line2.text=line2.text:gsub("(\\[xy]?shad)[%d%.%-]+","%10")
	    end
	    line2.layer=line2.layer+1
	    subs.insert(sel[i]+1,line2)

		-- top line
	    line3=line
	    line3.layer=line3.layer+1
	    if not res.onlyb then
	    line3.text=text
	    line3.text=topline(line3.text)
	    subs.insert(sel[i]+2,line3)
	    end

		-- second border
	    text=borderline2(text)
	    line.layer=line.layer-2
	    line.text=text
	    sls=2

		-- ONE BORDER
	    else

		-- top line
	    line3=line
	    line3.text=text
	    line3.text=topline(line3.text)
	    line3.layer=line3.layer+1
	    subs.insert(sel[i]+1,line3)

		-- bottom line
	    text=borderline(text)
	    line.layer=line.layer-1
	    line.text=text
	    sls=1
	    end

	subs[sel[i]]=line
	for s=i,#sel do sel[s]=sel[s]+sls end
    end
    aegisub.progress.title("Blur: DONE")
end

function topline(txt)
    txt=txt
    :gsub("(\\t%([^%)]*)\\bord[%d%.]+","%1")
    :gsub("(\\t%([^%)]*)\\shad[%d%.]+","%1")
    :gsub("\\t%([^\\]*%)","")
    if not txt:match("^{[^}]-\\bord") then txt=txt:gsub("^{\\","{\\bord0\\") end
    txt=txt
    :gsub("\\bord[%d%.]+","\\bord0") 
    :gsub("(\\r[^}]-)}","%1\\bord0}")
    txt=txt:gsub("(\\[xy]bord)[%d%.]+","")    :gsub("(\\[xy]shad)[%d%.%-]+","")    :gsub("\\3c&H%x+&","")
    if shadow~="0" then txt=txt:gsub("^({\\[^}]+)}","%1\\shad0}") end
    txt=txt
    :gsub("\\shad[%d%.]+","\\shad0")
    :gsub("(\\r[^}]-)}","%1\\shad0}")
    :gsub("\\bord[%d%.%-]+([^}]-)(\\bord[%d%.%-]+)","%1%2")
    :gsub("\\shad[%d%.%-]+([^}]-)(\\shad[%d%.%-]+)","%1%2")
    :gsub("{}","")
    return txt
end

function borderline(txt)
    txt=txt:gsub("\\c&H%x+&","")
    -- transform check
    if txt:match("^{[^}]-\\t%([^%)]-\\3c") then
	pretrans=text:match("^{(\\[^}]-)\\t")
	if not pretrans:match("^{[^}]-\\3c") then txt=txt:gsub("^{\\","{\\c"..soutline.."\\") end
    end
    if not txt:match("^{[^}]-\\3c&[^}]-}") then
        txt=txt:gsub("^({\\[^}]+)}","%1\\c"..soutline.."}")
	:gsub("(\\r[^}]-)}","%1\\c"..routline.."}")
    end
    txt=txt:gsub("(\\3c)(&H%x+&)","%1%2\\c%2")
    :gsub("(\\r[^}]-)}","%1\\c"..routline.."}")
    :gsub("(\\r[^}]-\\3c)(&H%x+&)([^}]-)}","%1%2\\c%2%3")
    :gsub("\\c&H%x+&([^}]-)(\\c&H%x+&)",function(a,b) if not a:match("\\t") then return a..b end end)
    :gsub("{%*?}","")
    if res.bbl and not res.double then txt=txt:gsub("\\blur[%d%.]+","\\blur"..res.bblur) end
    if res.botalpha and txt:match("\\fad%(") then txt=botalfa(txt) end
    return txt
end

function borderline2(txt)
    outlinetwo=primary
    if res.clr then col3=res.c3:gsub("#(%x%x)(%x%x)(%x%x)","&H%3%2%1&") outlinetwo=col3 rimary=col3 end
    bordertwo=border
    if res.bsize then bordertwo=res.secbord end
    -- transform check
    if txt:match("^{[^}]-\\t%([^%)]-\\bord") then
	pretrans=text:match("^{(\\[^}]-)\\t")
	if not pretrans:match("^{[^}]-\\bord") then txt=txt:gsub("^{\\","{\\bord"..border.."\\") end
    end
    if not txt:match("^{[^}]-\\bord") then txt=txt:gsub("^{\\","{\\bord"..border.."\\")  end
    txt=txt:gsub("(\\r[^\\}]-)([\\}])","%1\\bord"..rbord.."%2")
    :gsub("(\\r[^\\}]-)\\bord[%d%.%-]+([^}]-)(\\bord[%d%.%-]+)","%1%2%3")
    :gsub("(\\bord)([%d%.]+)",function(a,b) if res.bsize then brd=bordertwo else brd=b end return a..b+brd end)
    :gsub("(\\[xy]bord)([%d%.]+)",function(a,b) return a..b+b end)
    :gsub("\\3c&H%x+&","")
    :gsub("^({\\[^}]+)}","%1\\3c"..outlinetwo.."}")
    :gsub("(\\3c)(&H%x+&)","%1"..outlinetwo)
    if res.clr then txt=txt:gsub("\\c&H%x+&([^}]-)}","\\c"..rimary.."\\3c"..outlinetwo.."%1}")
      else txt=txt:gsub("(\\c)(&H%x+&)([^}]-)}","%1%2%3\\3c%2}") end
    txt=txt:gsub("(\\r[^}]+)}","%1\\3c"..rimary.."}")
    :gsub("\\c&H%x+&([^}]-)(\\c&H%x+&)",function(a,b) if not a:match("\\t") then return a..b end end)
    :gsub("\\3c&H%x+&([^}]-)(\\3c&H%x+&)",function(a,b) if not a:match("\\t") then return a..b end end)
    :gsub("{%*?}","")
    if res.bbl and res.double then txt=txt:gsub("\\blur[%d%.]+","\\blur"..res.bblur) end
    if res.botalpha and txt:match("\\fad%(") then txt=botalfa(txt) end
    return txt
end

function glowlayer(txt,kol,alf)
    txt=txt:gsub("\\alpha&H(%x%x)&",function(a) if a>al then return "\\alpha&H"..a.."&" else return "\\alpha&H"..al.."&" end end)
    :gsub("\\"..alf.."a&H(%x%x)&",function(a) if a>al then return "\\"..alf.."a&H"..a.."&" else return "\\"..alf.."a&H"..al.."&" end end)
    :gsub("(\\blur)[%d%.]*([\\}])","%1"..bl.."%2")
    :gsub("(\\r[^}]-)}","%1\\alpha&H"..al.."&}")
    if not txt:match("^{[^}]-\\alpha") then txt=txt:gsub("^({\\[^}]-)}","%1\\alpha&H"..al.."&}") end
    if res.alfa=="00" then txt=txt:gsub("^({\\[^}]-)\\alpha&H00&","%1") end
    txt=txt:gsub("{%*?}","")
    if res.glowcol then
	if txt:match("^{\\[^}]-\\"..kol.."&") then txt=txt:gsub("\\"..kol.."&H%x+&","\\"..kol..glowc)
	else txt=txt:gsub("\\"..kol.."&H%x+&","\\"..kol..glowc) txt=txt:gsub("^({\\[^}]-)}","%1\\"..kol..glowc.."}")
	end
    end
    return txt
end

function botalfa(txt)
    fadin,fadout=txt:match("\\fad%((%d+)%,(%d+)")
    alfadin=res.alphade	alfadout=res.alphade
    if res.alphade=="max" then alfadin=fadin alfadout=fadout end
    if fadin==nil or fadout==nil then aegisub.log("\n ERROR: Failed to capture fade times from line:\n "..text) end
        if fadin~="0" then
	    txt=txt:gsub("^({\\[^}]-)}","%1\\1a&HFF&\\t("..fadin-alfadin..","..fadin..",\\1a&H00&)}")
        end
        if fadout~="0" then
	    txt=txt:gsub("^({\\[^}]-)}","%1\\t("..duration-fadout..","..duration-fadout+alfadout..",\\1a&HFF&)}")
        end
    return txt
end

function stylinfo(text)
    	startags=text:match("^{\\[^}]-}")
    	if startags==nil then startags="" end
    	startags=startags:gsub("\\t%([^%(%)]+%)","") :gsub("\\t%([^%(%)]-%([^%)]-%)[^%)]-%)","")
    	
    	primary=styleref.color1:gsub("H%x%x","H")
    	pri=startags:match("^{[^}]-\\c(&H%x+&)")
    	if pri~=nil then primary=pri end
    	
    	soutline=styleref.color3:gsub("H%x%x","H")
    	outline=soutline
    	out=startags:match("^{[^}]-\\3c(&H%x+&)")
    	if out~=nil then outline=out end
    	
    	border=tostring(styleref.outline)
    	bord=startags:match("^{[^}]-\\bord([%d%.]+)")
    	if bord~=nil then border=bord end
    	
    	shadow=tostring(styleref.shadow)
    	shad=startags:match("^{[^}]-\\shad([%d%.]+)")
    	if shad~=nil then shadow=shad end
    	
    	if text:match("\\r%a") then 
    	rstyle=text:match("\\r([^\\}]+)")
    	reref=stylechk(rstyle)
    	rimary=reref.color1:gsub("H%x%x","H")
    	routline=reref.color3:gsub("H%x%x","H")
    	rbord=tostring(reref.outline)
    	else routline=soutline rimary=primary rbord=border
    	end
end

function preprocess(text)
    if not text:match("^{\\") then text="{\\blur"..default_blur.."}"..text			-- default blur if no tags
	text=text:gsub("(\\r[^}]-)}","%1\\blur"..default_blur.."}")
    end
    if not text:match("\\blur") then text=text:gsub("^{\\","{\\blur"..default_blur.."\\")	-- default blur if missing in tags
	text=text:gsub("(\\r[^}]-)}","%1\\blur"..default_blur.."}")
    end
    if text:match("\\blur") and not text:match("^{[^}]*blur[^}]*}") then			-- add blur if missing in first tag block
	text=text:gsub("^{\\","{\\blur"..default_blur.."\\")
    end	
    if text:match("^({[^}]-\\t)[^}]-}") and not text:match("^{[^}]-\\3c[^}]-\\t") then	-- \t workaround
	text=text:gsub("^{\\","{\\3c"..soutline.."\\")
    end
    text=text:gsub("\\1c","\\c")
    return text
end

function fixfade(subs,sel)
    for i=#sel,1,-1 do
	line=subs[sel[i]]
	text=line.text
	styleref=stylechk(line.style)
	duration=line.end_time-line.start_time
		border=tostring(styleref.outline)
		bord=text:match("^{[^}]-\\bord([%d%.]+)")
		if bord~=nil then border=bord end
	
		if border~="0" and line.text:match("\\fad%(") then
		text=text:gsub("\\1a&[%w]+&","")
		text=text:gsub("\\t%([^%(%)]-%)","")
		text=botalfa(text)
		end
	line.text=text
	subs[sel[i]]=line
    end
end

function layeraise(subs,sel)
    for i=#sel,1,-1 do
	line=subs[sel[i]]
	    if line.layer+res["layer"]>=0 then
	    line.layer=line.layer+res["layer"] else
	    aegisub.dialog.display({{class="label",
	    label="You're dumb. Layers can't go below 0.",x=0,y=0,width=1,height=2}},{"OK"})
	    aegisub.cancel()
	    end
	subs[sel[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
	end
    end
    return styleref
end

function saveconfig()
bgconf="Blur & Glow config\n\n"
  for key,val in ipairs(GUI) do
    if val.class=="floatedit" or val.class=="dropdown" or val.class=="color" then
      bgconf=bgconf..val.name..":"..res[val.name].."\n"
    end
    if val.class=="checkbox" and val.name~="save" then
      bgconf=bgconf..val.name..":"..tf(res[val.name]).."\n"
    end
  end
blurkonfig=aegisub.decode_path("?user").."\\blurandglow.conf"
file=io.open(blurkonfig,"w")
file:write(bgconf)
file:close()
aegisub.dialog.display({{class="label",label="Config saved to:\n"..blurkonfig}},{"OK"},{close='OK'})
end

function loadconfig()
blurkonfig=aegisub.decode_path("?user").."\\blurandglow.conf"
file=io.open(blurkonfig)
    if file~=nil then
	konf=file:read("*all")
	io.close(file)
	  for key,val in ipairs(GUI) do
	    if val.class=="floatedit" or val.class=="checkbox" or val.class=="dropdown" or val.class=="color" then
	      if konf:match(val.name) then val.value=detf(konf:match(val.name..":(.-)\n")) end
	    end
	  end
    end
end

function tf(val)
    if val==true then ret="true" else ret="false" end
    return ret
end

function detf(txt)
    if txt=="true" then ret=true
    elseif txt=="false" then ret=false
    else ret=txt end
    return ret
end

function konfig(subs,sel)
GUI={
    --left
    {x=0,y=0,width=2,class="label",label="  =   Blur and Glow version "..script_version.."   =" },
    {x=0,y=1,width=1,class="label",label="Glow blur:" },
    {x=0,y=2,width=1,class="label",label="Glow alpha:" },
    
    {x=1,y=1,width=2,class="floatedit",name="blur",value=3 },
    {x=1,y=2,width=2,class="dropdown",name="alfa",
    items={"00","20","30","40","50","60","70","80","90","A0","B0","C0","D0","F0"},value="80" },

    {x=0,y=3,width=1,class="checkbox",name="glowcol",label="glow c.:",value=false,hint="glow colour"},
    {x=1,y=3,width=2,class="color",name="glc" },
    
    {x=0,y=4,width=5,class="checkbox",name="botalpha",label="fix \\1a for layers with border and fade --> transition:",
			value=true,hint="uses \\1a&HFF& for bottom layer during fade"},
    {x=5,y=4,width=1,class="dropdown",name="alphade",items={0,45,80,120,160,200,"max"},value=45 },
    {x=6,y=4,width=1,class="label",label="ms" },
    
    {x=0,y=5,width=4,class="checkbox",name="onlyg",label="only add glow (layers w/ border)",value=false},
    
    -- right
    {x=4,y=0,width=1,class="checkbox",name="double",label="double border",value=false },
    {x=5,y=0,width=2,class="checkbox",name="onlyb",label="only add 2nd border",value=false},
    
    {x=4,y=1,width=1,class="checkbox",name="bbl",label="bottom blur:",
	hint="Blur for bottom layer \n[not the glow layer] \nif different from top layer."},
    {x=5,y=1,width=2,class="floatedit",name="bblur",value=1 },
    
    {x=4,y=2,width=1,class="checkbox",name="bsize",label="2nd b. size:",
	hint="Size for 2nd border \n[counts from first border out] \nif different from the current border."},
    {x=5,y=2,width=2,class="floatedit",name="secbord",value=2 },
    
    {x=4,y=3,width=1,class="checkbox",name="clr",label="2nd b. colour:",hint="Colour for 2nd border \nif different from primary." },
    {x=5,y=3,width=2,class="color",name="c3" },
    
    {x=4,y=5,width=1,class="label",label="     Change layer:", },
    {x=5,y=5,width=1,class="dropdown",name="layer",
	items={"-5","-4","-3","-2","-1","+1","+2","+3","+4","+5"},value="+1" },
    
    {x=0,y=6,width=2,class="checkbox",name="rep",label="repeat with last settings",value=false},
    {x=4,y=6,width=1,class="checkbox",name="autod",label="auto double",value=true,
	hint="automatically use double border\nif 2nd colour or 2nd border size is checked"},
    {x=5,y=6,width=2,class="checkbox",name="save",label="save configuration",value=false},
} 
    loadconfig()
    buttons={"Blur / Layers","Blur + Glow","Fix fades","Change layer","cancel"}
    pressed, res=aegisub.dialog.display(GUI,buttons,{ok='Blur / Layers',close='cancel'})
    if pressed=="cancel" then aegisub.cancel() end
    if res.onlyg then res.double=false end
    if res.onlyb then res.double=true end
    if res.save then saveconfig()
    else
	if res.rep then res=lastres end
	styleget(subs)
	if pressed=="Blur / Layers" then layerblur(subs,sel) end
	if pressed=="Blur + Glow" then sel=glow(subs,sel) end
	if pressed=="Fix fades" then fixfade(subs,sel) end
	if pressed=="Change layer" then layeraise(subs,sel) end
    end
    if res.rep==false then lastres=res end
    return sel
end

function blurandglow(subs,sel)
    sel=konfig(subs,sel)
    aegisub.set_undo_point(script_name)
    return sel
end

aegisub.register_macro(script_name,script_description,blurandglow)