aboutsummaryrefslogtreecommitdiffstats
path: root/mpv/lua
diff options
context:
space:
mode:
Diffstat (limited to 'mpv/lua')
-rw-r--r--mpv/lua/convert_script.lua319
1 files changed, 319 insertions, 0 deletions
diff --git a/mpv/lua/convert_script.lua b/mpv/lua/convert_script.lua
new file mode 100644
index 0000000..04fa508
--- /dev/null
+++ b/mpv/lua/convert_script.lua
@@ -0,0 +1,319 @@
+-- README:
+-- original version by Zehkul https://gist.github.com/Zehkul/25ea7ae77b30af959be0
+-- needs: yad, libnotify (and --ytdl if you want to encode from streams)
+-- press any of alt + w, g or x to set the start frame
+-- press again to set the end frame and use
+-- alt + w to make a webm
+-- alt + g to make a gif
+-- alt + x to make a x264 encoded mkv
+
+-- Note: encoding a webm to a specific filesize will only work if audio is disabled
+-- and including subs will make them lose their styling
+
+local msg = require 'mp.msg'
+local opt = require 'mp.options'
+local mputils = require 'mp.utils'
+
+-- default options, convert_script.conf is read
+local options = {
+ bitrate_multiplier = 0.975, -- to make sure the file won’t go over the target file size, set it to 1 if you don’t care
+}
+
+read_options(options, "convert_script_webm")
+read_options(options, "convert_script_gif")
+read_options(options, "convert_script_x264")
+
+-----------------
+-- Main script --
+-----------------
+
+function set_timepos(func)
+
+ if timepos1 then
+
+ timepos2 = mp.get_property("time-pos")
+ timepos2_humanreadable = mp.get_property_osd("time-pos")
+
+ if tonumber(timepos1) > tonumber(timepos2) then
+
+ length = timepos1-timepos2
+ start = timepos2
+ start_humanreadable = timepos2_humanreadable
+ end_humanreadable = timepos1_humanreadable
+ msg.info("End frame set")
+
+ elseif tonumber(timepos2) > tonumber(timepos1) then
+
+ length = timepos2-timepos1
+ start = timepos1
+ start_humanreadable = timepos1_humanreadable
+ end_humanreadable = timepos2_humanreadable
+ msg.info("End frame set")
+
+ else
+
+ msg.error("Both frames are the same, ignoring the second one")
+ mp.osd_message("Both frames are the same, ignoring the second one")
+ timepos2 = nil
+ return
+
+ end
+
+ timepos1 = nil
+ func.call_gui()
+
+ else
+
+ timepos1 = mp.get_property("time-pos")
+ timepos1_humanreadable = mp.get_property_osd("time-pos")
+ msg.info("Start frame set")
+ mp.osd_message("Start frame set")
+
+ end
+
+end
+
+function convert_script_hotkey_call_webm ()
+
+ func = {
+ call_gui = call_gui_webm
+ }
+
+ set_timepos(func)
+
+end
+
+function convert_script_hotkey_call_gif ()
+
+ func = {
+ call_gui = call_gui_gif
+ }
+
+ set_timepos(func)
+
+end
+
+function convert_script_hotkey_call_x264 ()
+
+ func = {
+ call_gui = call_gui_x264
+ }
+
+ set_timepos(func)
+
+end
+
+------------
+-- Encode --
+------------
+
+function generate_filenames(ext)
+
+ local filename_ext = mp.get_property_osd("filename")
+ filename_ext = string.gsub(filename_ext, "'", "'\\''")
+ local filename = string.gsub(filename_ext, "%....$","")
+
+ if string.len(filename) > 230 then
+
+ filename = mp.get_property("options/title")
+ if filename == 'mpv - ${media-title}' or string.len(filename) > 230 then
+ filename = 'output'
+ end
+
+ end
+
+ local path = mp.get_property("path", "")
+ local dir, fil = mputils.split_path(path)
+
+ file_in = dir .. fil
+ file_out = dir .. filename .. "-out." .. ext
+
+ file_in = string.gsub(file_in, "'", "'\\''")
+ file_out = string.gsub(file_out, "'", "'\\''")
+
+ if string.sub(file_in,1,string.len("http")) == "http" then
+
+ file_in = mp.get_property("stream-open-filename")
+ file_out = os.getenv("HOME") .. "/" .. mp.get_property("media-title") .. "-out." .. ext
+ file_out = string.gsub(file_out, "'", "'\\''")
+
+ end
+
+end
+
+function encode_webm ()
+
+ generate_filenames("webm")
+
+ local full_command = '(ffmpeg -ss ' .. start .. ' -i \'' .. file_in .. '\' -t ' .. length .. ' -b:v ' .. bitrate .. bitsize .. ' -vf scale=' .. scale .. ':-1 -quality good -cpu-used 0 -pass 1 ' .. audio .. ' ' .. subs .. ' -f webm /dev/null -y'
+
+ full_command = full_command .. ' && ffmpeg -ss ' .. start .. ' -i \'' .. file_in .. '\' -t ' .. length .. ' -b:v ' .. bitrate .. bitsize .. ' -vf scale=' .. scale .. ':-1 -quality good -cpu-used 0 -pass 2 ' .. audio .. ' ' .. subs .. ' -f webm \'' .. file_out .. '\' -y'
+
+ full_command = full_command .. ') && notify-send "Encode done!"'
+
+ msg.info(full_command)
+ os.execute(full_command)
+
+end
+
+function encode_gif ()
+
+ generate_filenames("gif")
+
+ local full_command = '(ffmpeg -ss ' .. start .. ' -i \'' .. file_in .. '\' -t ' .. length .. ' -vf scale=' .. scale .. ':-1 -r ' .. fps .. ' \'' .. file_out .. '\' -y'
+
+ full_command = full_command .. ') && notify-send "Encode done!"'
+
+ msg.info(full_command)
+ os.execute(full_command)
+
+end
+
+function encode_x264 ()
+
+ generate_filenames("mkv")
+
+ local full_command = '(ffmpeg -ss ' .. start .. ' -i \'' .. file_in .. '\' -t ' .. length .. ' -c:v libx264 -vf scale=' .. sheight .. ':' .. swidth .. ' -preset slow -qp ' .. quality .. ' -c:a copy -c:s copy -map 0 \'' .. file_out .. '\' -y'
+
+ full_command = full_command .. ') && notify-send "Encode done! (' .. quality .. ')"'
+
+ msg.info(full_command)
+ os.execute(full_command)
+
+end
+
+---------
+-- GUI --
+---------
+
+function call_gui_webm ()
+
+ mp.resume_all()
+ local handle = io.popen('yad --title="Convert Script" --center --form --separator="\n" --field="Resize to:NUM" "720" --field="Don’t resize at all:CHK" "false" --field="Include audio:CHK" "false" --field="Include subs:CHK" "false" --field="Bitrate/Filesize (M):NUM" "3" --field="Filesize > Bitrate:CHK" "true" --button="gtk-cancel:2" --button="gtk-ok:0" && echo "$?"')
+ local yad = handle:read("*a")
+ handle:close()
+
+ if yad == "" then
+ return
+ end
+
+ local yad_table = {}
+
+ local i = 0
+ for k in string.gmatch(yad, "[%a%d]+") do
+ yad_table[i] = k
+ i = i + 1
+ end
+
+ if (yad_table[2] == "FALSE") then
+ scale = yad_table[0]
+ else
+ scale = "-1"
+ end
+
+ if yad_table[3] == "FALSE" then
+ audio = '-an'
+ else
+ audio = ""
+ end
+
+ if yad_table[4] == "FALSE" then
+ subs = '-sn'
+ else
+ subs = ""
+ end
+
+ if yad_table[7] == "TRUE" then
+ bitrate = math.floor(yad_table[5]*1024*8/length*options.bitrate_multiplier)
+ bitsize = "K"
+ else
+ bitrate = yad_table[5]
+ bitsize = "M"
+ end
+
+ if yad_table[8] == "0" then
+ encode_webm()
+ end
+
+end
+
+function call_gui_gif ()
+
+ mp.resume_all()
+ local handle = io.popen('yad --title="Convert Script" --center --form --separator="\n" --field="Resize to:NUM" "320" --field="Don’t resize at all:CHK" "false" --field="FPS:NUM" "24" --button="gtk-cancel:2" --button="gtk-ok:0" && echo "$?"')
+ local yad = handle:read("*a")
+ handle:close()
+
+ if yad == "" then
+ return
+ end
+
+ local yad_table = {}
+
+ local i = 0
+ for k in string.gmatch(yad, "[%a%d]+") do
+ yad_table[i] = k
+ i = i + 1
+ end
+
+ if (yad_table[2] == "FALSE") then
+ scale = yad_table[0]
+ else
+ scale = "-1"
+ end
+
+ fps = yad_table[3]
+
+ if yad_table[5] == "0" then
+ encode_gif()
+ end
+
+end
+
+function call_gui_x264 ()
+
+ mp.resume_all()
+ local handle = io.popen('yad --title="Convert Script" --center --form --separator="\n" --field="Resize height:NUM" "720" --field="Resize width:NUM" "480" --field="Don’t resize at all:CHK" "true" --field="Quality (0-51):NUM" "18" --button="gtk-cancel:2" --button="gtk-ok:0" && echo "$?"')
+ local yad = handle:read("*a")
+ handle:close()
+
+ if yad == "" then
+ return
+ end
+
+ local yad_table = {}
+
+ local i = 0
+ for k in string.gmatch(yad, "[%a%d]+") do
+ yad_table[i] = k
+ i = i + 1
+ end
+
+ if (yad_table[4] == "FALSE") then
+ sheight = yad_table[0]
+ else
+ sheight = "-1"
+ end
+
+ if (yad_table[4] == "FALSE") then
+ swidth = yad_table[2]
+ else
+ swidth = "-1"
+ end
+
+ if ((sheight ~= "-1" and swidth ~= "-1") and (tonumber(sheight)%2 ~= 0 or tonumber(swidth)%2 ~= 0)) then
+ os.execute('notify-send "' .. sheight ..':' .. swidth .. '"')
+ return
+ end
+
+ quality = yad_table[5];
+
+ if yad_table[7] == "0" then
+ encode_x264()
+ end
+
+end
+
+mp.add_key_binding("alt+w", "convert_script_webm", convert_script_hotkey_call_webm)
+mp.add_key_binding("alt+g", "convert_script_gif", convert_script_hotkey_call_gif)
+mp.add_key_binding("alt+x", "convert_script_x264", convert_script_hotkey_call_x264)