shiftcut.lua

-- Basically an enhanced version of TPP + Shift Times.
-- It can not only add, but also cut leads; prevent adding leads on keyframes; fix overlaps; etc.

script_name="ShiftCut"
script_description="Time Machine."
script_author="unanimated"
script_version="2.6"


function style(subs)
styles={"All","All Default","Default+Alt","----------------"}
  for i=1, #subs do
    if subs[i].class=="style" then
      local st=subs[i]
      table.insert(styles,st.name)
    end
  end
  return styles
end

function runcheck()
	run=0
	if res.stail=="All" then run=1 end
	if res.stail=="All Default" and line.style:match("Defa") then run=1 end
	if res.stail=="Default+Alt" and line.style:match("Defa") then run=1 end
	if res.stail=="Default+Alt" and line.style:match("Alt") then run=1 end
	if res.stail==line.style then run=1 end
	if line.style==res.plustyle then run=1 end
	return run
end

--	Lead in		--
function cutin(subs, sel)
	for z, i in ipairs(sel) do
	    line=subs[i]
		inn=res.inn
		start=line.start_time
		endt=line.end_time
		prevline=subs[i-1]
		if prevline.class=="dialogue" then
		prevend=prevline.end_time end
		run=runcheck()
		
	    kfr=1
	    if res.holdkf then
		startf=ms2fr(start)
		for k,kf in ipairs(keyframes) do
		    if kf==startf then kfr=0 break end
		end
	    end
	    
	    if run==1 and kfr==1 then
		if res.cutin then	-- cut
		  if (start+inn)<endt then start=(start+inn) else start=endt end
		else
		  start=(start-inn)	-- add
		  if res.preventcut and prevline.class=="dialogue" and start<prevend then start=prevend end
		end
	    line.start_time=start
	    subs[i]=line
	    end
	end
end

--	Lead out	--
function cutout(subs, sel)
	for z, i in ipairs(sel) do
	    line=subs[i]
		ut=res.utt
		start=line.start_time
		endt=line.end_time
		if i<#subs then nextline=subs[i+1]
		nextart=nextline.start_time end
		run=runcheck()
		
	    kfr=1
	    if res.holdkf then
		endf=ms2fr(endt)
		for k,kf in ipairs(keyframes) do
		    if kf==endf then kfr=0 break end
		end
	    end
	    
	    if run==1 and kfr==1 then
		if res.cutout then	-- cut
		  if (endt-ut)>start then endt=(endt-ut) else endt=start end
		else
		  endt=(endt+ut)	-- add
		  if res.preventcut and i<#subs and endt>nextart and start<nextart then endt=nextart end
		end
	    line.end_time=endt
	    subs[i]=line
	    end
	end
end

--	Shifting	--
function shiift(subs, sel)
	for z, i in ipairs(sel) do
	    line=subs[i]
		shift=res.shifft
		start=line.start_time
		endt=line.end_time
		run=runcheck()
		
	    if run==1 then
		if not res.shit then
		start=(start-shift)
		endt=(endt-shift)
		else
		start=(start+shift)
		endt=(endt+shift)
		end
	    line.start_time=start
	    line.end_time=endt
	    subs[i]=line
	    end
	end
end

--	Linking		--
function linklines(subs, sel)
	marker=0	linx=0	overl=0
	for z, i in ipairs(sel) do
	    line=subs[i]
		lnk=res.link
		start=line.start_time
		endt=line.end_time
		s1=start e1=endt
		if i<#subs then nextline=subs[i+1]
		nextart=nextline.start_time nextf=ms2fr(nextart) end
		if marker==1 then start=start-diff2 end	-- link line 2
		if markover==1 then start=start+diffo end	-- overlap line 2
		marker=0
		markover=0
		run=runcheck()
	    if res.holdkf then
		endf=ms2fr(endt)
		for k,kf in ipairs(keyframes) do
		    if kf==endf or kf==nextf then run=0 break end
		end
	    end
		
	    if run==1 then
		-- linking
		if lnk>0 and nextart>endt and nextart-endt<lnk and z~=#sel then
		 gap=nextart-endt
		 diff=gap*res.bias
		 endt=endt+diff
		 diff2=gap-diff
		 marker=1
		end
		
		-- overlaps
		if res.over and endt>nextart and endt-nextart<res.overlap and z~=#sel then
		lap=endt-nextart
		diffo=lap*res.bios
		endt=nextart+diffo
		markover=1
		end
	    line.start_time=start
	    line.end_time=endt
	    if res.mark then if line.start_time~=s1 or line.end_time~=e1 then line.effect=line.effect.."[linked]" end end
	    if res.info and marker==1 then linx=linx+1 end
	    if res.info and markover==1 then overl=overl+1 end
	    subs[i]=line
	    end
	end
end

--	Snapping	--
function keyframesnap(subs, sel)
snapd=0
    if res.pres then kfsb,kfeb,kfsa,kfea=res.preset:match("(%d+),(%d+),(%d+),(%d+)") 
	kfsb=tonumber(kfsb)
	kfeb=tonumber(kfeb)
	kfsa=tonumber(kfsa)
	kfea=tonumber(kfea)
    end
    if not res.pres then kfsb=res.sb kfeb=res.eb kfsa=res.sa kfea=res.ea end
	for z, i in ipairs(sel) do aegisub.progress.title(string.format("Snapping Line %d/%d",z,#sel))
	    line=subs[i]
		run=runcheck()
		
	    if run==1 then

	    -- snapping to keyframes
		
		start=line.start_time		-- start time
		endt=line.end_time		-- end time
		startemp=start
		endtemp=endt
		if z~=#sel then nextline=subs[i+1]
		nextart=nextline.start_time end
		if z~=1 then prevline=subs[i-1]
		prevend=prevline.end_time end
		
		startf=ms2fr(start)		-- startframe
		endf=ms2fr(endt)		-- endframe
		
		diff=250
		diffe=250
		startkf=keyframes[1]
		endkf=keyframes[#keyframes]
		
		-- check for nearby keyframes
		for k,kf in ipairs(keyframes) do
		
			-- startframe
			if kf>=startf-kfsa and kf<=startf+kfsb then 
			tdiff=math.abs(startf-kf)
			if tdiff<=diff then diff=tdiff startkf=kf end
			startemp=fr2ms(startkf)
			
			stopstart=0
			if res.prevent and z~=1 and startemp<prevend and start>=prevend then stopstart=1 end
			if stopstart==0 then start=startemp end
			end
			
			-- endframe
			if kf>=endf-kfea and kf<=endf+kfeb then
			tdiff=math.abs(endf-kf)
			if tdiff<diffe then diffe=tdiff endkf=kf end
			endtemp=fr2ms(endkf)
			
			stopend=0
			if res.prevent and z~=#sel and endtemp>nextart and endkf-endf>kfsb then stopend=1 end
			if stopend==0 then endt=endtemp end
			end
			
		end
	    line.start_time=start
	    if endt-start>450 then line.end_time=endt end
	    startf2=ms2fr(line.start_time)	    endf2=ms2fr(line.end_time)
	    if res.mark then if startf2~=startf or endf2~=endf then line.effect=line.effect.."[snapped]" end end
	    if res.info then if startf2~=startf or endf2~=endf then snapd=snapd+1 end end
	    subs[i]=line
	    end
	end
end

function selectall(subs, sel)
sel={}
    for i=1,#subs do
	if subs[i].class=="dialogue" then table.insert(sel,i) end
    end
    return sel
end

--	Config Stuff	--
function saveconfig()
scconf="ShiftCut Config:\n\n"
  for key,val in ipairs(gui) do
    if val.class=="floatedit" or val.class=="dropdown" then
      scconf=scconf..val.name..":"..res[val.name].."\n"
    end
    if val.class=="checkbox" and val.name~="save" then
      scconf=scconf..val.name..":"..tf(res[val.name]).."\n"
    end
    if val.name=="preset" then
	prsts="keyframe presets:"
	for v=1,#val.items do
	    prsts=prsts..val.items[v]..":"
	end
	scconf=scconf..prsts.."\n"
    end
  end
shiftkonfig=aegisub.decode_path("?user").."\\shiftcut.conf"
file=io.open(shiftkonfig,"w")
file:write(scconf)
file:close()
press,rez=aegisub.dialog.display({{class="label",label="Config saved to:\n"..shiftkonfig}},{"OK","Add/Delete kf preset","Restore Defaults"},{close='OK'})
    if press=="Add/Delete kf preset" then
	ite=scconf:match("keyframe presets:(.-)\n")
	pressets=""
	for it in ite:gmatch("[^:]+") do
	pressets=pressets..it.."\n"
	end
	repeat
	if press=="Reset" then pressets="1,1,1,1\n2,2,2,2\n6,6,6,10\n6,6,8,12\n6,6,10,12\n6,10,8,12\n8,8,8,12\n0,0,0,10\n7,12,10,13" end
	press,rez=aegisub.dialog.display({{class="textbox",x=0,y=0,width=12,height=12,name="addpres",value=pressets}},
	{"Save","Reset"},{ok='Save'})
	until press~="Reset"
	newpres=rez.addpres.."\n"
	newpres=newpres:gsub("\n",":") :gsub("::",":")
	scconf=scconf:gsub(ite,newpres)
	file=io.open(shiftkonfig,"w")
	file:write(scconf)
	file:close()
	aegisub.dialog.display({{class="label",label="Saved"}},{"OK"},{close='OK'})
    end
    if press=="Restore Defaults" then
	file=io.open(shiftkonfig,"w")
	file:write("")
	file:close()
	aegisub.dialog.display({{class="label",label="Defaults restored"}},{"OK"},{close='OK'})
    end
end

function loadconfig()
sconfig=aegisub.decode_path("?user").."\\shiftcut.conf"
file=io.open(sconfig)
    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" then
	      if konf:match(val.name) then val.value=detf(konf:match(val.name..":(.-)\n")) end
	    end
	    if val.name=="preset" then ite=konf:match("keyframe presets:(.-)\n")
	      if ite~=nil then
	        val.items={}
	        for it in ite:gmatch("[^:]+") do
		  table.insert(val.items,it)
	        end
	      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)
BIAS={"0","0.1","0.2","0.3","0.4","0.5","0.6","0.7","0.8","0.9","1"}
kf_snap_presets={"1,1,1,1","2,2,2,2","6,6,6,10","6,6,8,12","6,6,10,12","6,10,8,12","8,8,8,12","0,0,0,10","7,12,10,13"}
	style(subs)
	gui=
	{
	    {x=0,y=0,width=2,height=1,class="label",label="ShiftCut v"..script_version },
	    {x=2,y=0,width=2,height=1,class="dropdown",name="slct",items={"Apply to selected","Apply to all lines"},value="Apply to selected"},
	    
	    {x=0,y=1,width=2,height=1,class="label",label="Styles to aply to:" },
	    {x=2,y=1,width=2,height=1,class="dropdown",name="stail",items=styles,value="All Default"},
	    {x=4,y=1,width=3,height=1,class="edit",name="plustyle",value="",hint="Additional style"},
	    
	    {x=4,y=0,width=4,height=1,class="checkbox",name="info",label="Info (link/snap)",value=false},
	    {x=8,y=0,width=2,height=1,class="checkbox",name="mark",label="Mark changed lines",value=false,
		hint="linking/snapping - marks changed lines in effect"},

	    -- shift
	    {x=0,y=2,width=1,height=1,class="label",label="SHIFT backward"},
	    {x=2,y=2,width=2,height=1,class="floatedit",name="shifft",value=0},
	    {x=4,y=2,width=1,height=1,class="label",label="ms"},
	    {x=5,y=2,width=2,height=1,class="checkbox",name="shit",label="Shift forward",value=false},

	    -- add/cut
	    {x=0,y=3,width=1,height=1,class="checkbox",name="IN",label="Add lead in"},
	    {x=2,y=3,width=2,height=1,class="floatedit",name="inn",value=0},
	    {x=4,y=3,width=1,height=1,class="label",label="ms"},
	    {x=5,y=3,width=2,height=1,class="checkbox",name="cutin",label="Cut lead in",value=false},
	    
	    {x=0,y=4,width=1,height=1,class="checkbox",name="OUT",label="Add lead out"},
	    {x=2,y=4,width=2,height=1,class="floatedit",name="utt",value=0},
	    {x=4,y=4,width=1,height=1,class="label",label="ms"},
	    {x=5,y=4,width=2,height=1,class="checkbox",name="cutout",label="Cut lead out",value=false},
	    
	    {x=0,y=5,width=3,height=1,class="checkbox",name="preventcut",
	    label="prevent overlaps from adding leads",value=true},
	    {x=3,y=5,width=4,height=1,class="checkbox",name="holdkf",
	    label="don't add leads on keyframes",value=false},

	    -- linking
	    {x=0,y=6,width=2,height=1,class="label",label="Line linking:  Max gap:"},
	    {x=2,y=6,width=2,height=1,class="floatedit",name="link",value=400,min=0},
	    {x=4,y=6,width=1,height=1,class="label",label="ms   "},
	    {x=5,y=6,width=1,height=1,class="label",label="Bias:  "},
	    {x=6,y=6,width=1,height=1,class="dropdown",name="bias",items=BIAS,value="0.8",hint="higher number=closer to 2nd line"},

	    -- overlaps
	    {x=0,y=7,width=2,height=1,class="checkbox",name="over",label="Fix overlaps up to:",value=false,
		hint="This is part of line linking.\nIf you want only overlaps, set linking gap to 0."},
	    {x=2,y=7,width=2,height=1,class="floatedit",name="overlap",value=500,min=0 },
	    {x=4,y=7,width=1,height=1,class="label",label="ms   "},
	    {x=5,y=7,width=1,height=1,class="label",label="Bias:  "},
	    {x=6,y=7,width=1,height=1,class="dropdown",name="bios",items=BIAS,value="0.5",hint="higher number=closer to 2nd line"},

	    -- keyframes
	    {x=8,y=1,width=1,height=1,class="label",label="Keyframes"},
	    
	    {x=8,y=2,width=1,height=1,class="label",label="Starts before:"},
	    {x=8,y=3,width=1,height=1,class="label",label="Ends before:"},
	    {x=8,y=4,width=1,height=1,class="label",label="Starts after:"},
	    {x=8,y=5,width=1,height=1,class="label",label="Ends after:"},
	    
	    {x=9,y=2,width=1,height=1,class="floatedit",name="sb",value=0,min=0,max=250,hint="frames, not ms"},
	    {x=9,y=3,width=1,height=1,class="floatedit",name="eb",value=0,min=0,max=250,hint="frames, not ms"},
	    {x=9,y=4,width=1,height=1,class="floatedit",name="sa",value=0,min=0,max=250,hint="frames, not ms"},
	    {x=9,y=5,width=1,height=1,class="floatedit",name="ea",value=0,min=0,max=250,hint="frames, not ms"},
	    
	    {x=8,y=6,width=1,height=1,class="checkbox",name="pres",label="Preset:",value=true},
	    {x=9,y=6,width=1,height=1,class="dropdown",name="preset",items=kf_snap_presets,value="6,10,8,12"},
	    
	    {x=8,y=7,width=2,height=1,class="checkbox",name="prevent",label="Prevent overlaps by snapping",value=true},
	}
	loadconfig()
	pressed, res=aegisub.dialog.display(gui,
		{"Lead in/out","Link Lines","Shift times","Snap to keyframes","Save config","Cancel"},{cancel='Cancel'})
	if res.slct=="Apply to all lines" then sel=selectall(subs, sel) end
	keyframes=aegisub.keyframes()
	ms2fr=aegisub.frame_from_ms
	fr2ms=aegisub.ms_from_frame
	
	if pressed=="Lead in/out" then 
	    if res.IN or res.cutin then cutin(subs, sel) end
	    if res.OUT or res.cutout then cutout(subs, sel) end
	end
	if pressed=="Shift times" then shiift(subs, sel) end
	if pressed=="Link Lines" then linklines(subs, sel) end
	if pressed=="Snap to keyframes" then keyframesnap(subs, sel) end
	if pressed=="Cancel" then aegisub.cancel() end
	if res.info then
	    if pressed=="Link Lines" then aegisub.log("\n Linked lines: "..linx) end
	    if pressed=="Link Lines" and res.over then aegisub.log("\n Overlaps fixed: "..overl) end
	    if pressed=="Snap to keyframes" then aegisub.log("\n Lines snapped to keyframes: "..snapd) end
	end
	if pressed=="Save config" then saveconfig() end
	
	return sel
end

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

aegisub.register_macro(script_name, script_description, shiftcut)