1799 lines
52 KiB
Lua
1799 lines
52 KiB
Lua
local mp = require("mp")
|
||
local assdraw = require("mp.assdraw")
|
||
local msg = require("mp.msg")
|
||
local utils = require("mp.utils")
|
||
local mpopts = require("mp.options")
|
||
local options = {
|
||
-- Defaults to shift+c
|
||
keybind = "C",
|
||
-- If empty, saves on the same directory of the playing video.
|
||
-- A starting "~" will be replaced by the home dir.
|
||
-- This field is delimited by double-square-brackets - [[ and ]] - instead of
|
||
-- quotes, because Windows users might run into a issue when using
|
||
-- backslashes as a path separator. Examples of valid inputs for this field
|
||
-- would be: [[]] (the default, empty value), [[C:\Users\John]] (on Windows),
|
||
-- and [[/home/john]] (on Unix-like systems eg. Linux).
|
||
output_directory = [[~/Videos]],
|
||
run_detached = false,
|
||
-- Template string for the output file
|
||
-- %f - Filename, with extension
|
||
-- %F - Filename, without extension
|
||
-- %T - Media title, if it exists, or filename, with extension (useful for some streams, such as YouTube).
|
||
-- %s, %e - Start and end time, with milliseconds
|
||
-- %S, %E - Start and end time, without milliseconds
|
||
-- %M - "-audio", if audio is enabled, empty otherwise
|
||
-- %R - "-(height)p", where height is the video's height, or scale_height, if it's enabled.
|
||
output_template = "%F-[%s-%e]",
|
||
-- Scale video to a certain height, keeping the aspect ratio. -1 disables it.
|
||
-- scale_height = -1,
|
||
-- Target filesize, in kB. This will be used to calculate the bitrate
|
||
-- used on the encode. If this is set to <= 0, the video bitrate will be set
|
||
-- to 0, which might enable constant quality modes, depending on the
|
||
-- video codec that's used (VP8 and VP9, for example).
|
||
-- target_filesize = 2500,
|
||
-- If true, will use stricter flags to ensure the resulting file doesn't
|
||
-- overshoot the target filesize. Not recommended, as constrained quality
|
||
-- mode should work well, unless you're really having trouble hitting
|
||
-- the target size.
|
||
-- strict_filesize_constraint = false,
|
||
-- strict_bitrate_multiplier = 0.95,
|
||
-- In kilobits.
|
||
-- strict_audio_bitrate = 64,
|
||
-- Sets the output format, from a few predefined ones.
|
||
-- Currently we have webm-vp8 (libvpx/libvorbis), webm-vp9 (libvpx-vp9/libvorbis)
|
||
-- and raw (rawvideo/pcm_s16le).
|
||
-- output_format = "webm-vp8",
|
||
-- twopass = false,
|
||
-- If set, applies the video filters currently used on the playback to the encode.
|
||
-- apply_current_filters = true,
|
||
-- If set, writes the video's filename to the "Title" field on the metadata.
|
||
write_filename_on_metadata = false,
|
||
-- Set the number of encoding threads, for codecs libvpx and libvpx-vp9
|
||
-- libvpx_threads = 4,
|
||
-- additional_flags = "",
|
||
-- Constant Rate Factor (CRF). The value meaning and limits may change,
|
||
-- from codec to codec. Set to -1 to disable.
|
||
crf = 15,
|
||
-- Useful for flags that may impact output filesize, such as qmin, qmax etc
|
||
-- Won't be applied when strict_filesize_constraint is on.
|
||
-- non_strict_additional_flags = "",
|
||
-- Display the encode progress, in %. Requires run_detached to be disabled.
|
||
-- On Windows, it shows a cmd popup. "auto" will display progress on non-Windows platforms.
|
||
display_progress = "auto",
|
||
-- The font size used in the menu. Isn't used for the notifications (started encode, finished encode etc)
|
||
font_size = 28,
|
||
margin = 10,
|
||
message_duration = 5,
|
||
encoding_profile = "enc-f-mp4", -- from default encoding-profiles.conf
|
||
audio = true,
|
||
burn_subtitles = false
|
||
}
|
||
|
||
mpopts.read_options(options)
|
||
local base64_chars='ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'
|
||
|
||
-- encoding
|
||
function base64_encode(data)
|
||
return ((data:gsub('.', function(x)
|
||
local r,b='',x:byte()
|
||
for i=8,1,-1 do r=r..(b%2^i-b%2^(i-1)>0 and '1' or '0') end
|
||
return r;
|
||
end)..'0000'):gsub('%d%d%d?%d?%d?%d?', function(x)
|
||
if (#x < 6) then return '' end
|
||
local c=0
|
||
for i=1,6 do c=c+(x:sub(i,i)=='1' and 2^(6-i) or 0) end
|
||
return base64_chars:sub(c+1,c+1)
|
||
end)..({ '', '==', '=' })[#data%3+1])
|
||
end
|
||
|
||
-- decoding
|
||
function base64_decode(data)
|
||
data = string.gsub(data, '[^'..base64_chars..'=]', '')
|
||
return (data:gsub('.', function(x)
|
||
if (x == '=') then return '' end
|
||
local r,f='',(base64_chars:find(x)-1)
|
||
for i=6,1,-1 do r=r..(f%2^i-f%2^(i-1)>0 and '1' or '0') end
|
||
return r;
|
||
end):gsub('%d%d%d?%d?%d?%d?%d?%d?', function(x)
|
||
if (#x ~= 8) then return '' end
|
||
local c=0
|
||
for i=1,8 do c=c+(x:sub(i,i)=='1' and 2^(8-i) or 0) end
|
||
return string.char(c)
|
||
end))
|
||
end
|
||
local bold
|
||
bold = function(text)
|
||
return "{\\b1}" .. tostring(text) .. "{\\b0}"
|
||
end
|
||
local message
|
||
message = function(text, duration)
|
||
local ass = mp.get_property_osd("osd-ass-cc/0")
|
||
ass = ass .. text
|
||
return mp.osd_message(ass, duration or options.message_duration)
|
||
end
|
||
local append
|
||
append = function(a, b)
|
||
for _, val in ipairs(b) do
|
||
a[#a + 1] = val
|
||
end
|
||
return a
|
||
end
|
||
local seconds_to_time_string
|
||
seconds_to_time_string = function(seconds, no_ms, full)
|
||
if seconds < 0 then
|
||
return "unknown"
|
||
end
|
||
local ret = ""
|
||
if not (no_ms) then
|
||
ret = string.format(".%03d", seconds * 1000 % 1000)
|
||
end
|
||
ret = string.format("%02d:%02d%s", math.floor(seconds / 60) % 60, math.floor(seconds) % 60, ret)
|
||
if full or seconds > 3600 then
|
||
ret = string.format("%d:%s", math.floor(seconds / 3600), ret)
|
||
end
|
||
return ret
|
||
end
|
||
local seconds_to_path_element
|
||
seconds_to_path_element = function(seconds, no_ms, full)
|
||
local time_string = seconds_to_time_string(seconds, no_ms, full)
|
||
local _
|
||
time_string, _ = time_string:gsub(":", ".")
|
||
return time_string
|
||
end
|
||
local file_exists
|
||
file_exists = function(name)
|
||
local info, err = utils.file_info(name)
|
||
if info ~= nil then
|
||
return true
|
||
end
|
||
return false
|
||
end
|
||
local format_filename
|
||
format_filename = function(startTime, endTime, extension)
|
||
local replaceTable = {
|
||
["%%f"] = mp.get_property("filename"),
|
||
["%%F"] = mp.get_property("filename/no-ext"),
|
||
["%%s"] = seconds_to_path_element(startTime),
|
||
["%%S"] = seconds_to_path_element(startTime, true),
|
||
["%%e"] = seconds_to_path_element(endTime),
|
||
["%%E"] = seconds_to_path_element(endTime, true),
|
||
["%%T"] = mp.get_property("media-title"),
|
||
["%%M"] = (mp.get_property_native('aid') and not mp.get_property_native('mute')) and '-audio' or '',
|
||
["%%R"] = (options.scale_height ~= -1) and "-" .. tostring(options.scale_height) .. "p" or "-" .. tostring(mp.get_property_native('height')) .. "p"
|
||
}
|
||
local filename = options.output_template
|
||
for format, value in pairs(replaceTable) do
|
||
local _
|
||
filename, _ = filename:gsub(format, value)
|
||
end
|
||
local _
|
||
filename, _ = filename:gsub("[<>:\"/\\|?*]", "")
|
||
return tostring(filename) .. "." .. tostring(extension)
|
||
end
|
||
local parse_directory
|
||
parse_directory = function(dir)
|
||
local home_dir = os.getenv("HOME")
|
||
if not home_dir then
|
||
home_dir = os.getenv("USERPROFILE")
|
||
end
|
||
if not home_dir then
|
||
local drive = os.getenv("HOMEDRIVE")
|
||
local path = os.getenv("HOMEPATH")
|
||
if drive and path then
|
||
home_dir = utils.join_path(drive, path)
|
||
else
|
||
msg.warn("Couldn't find home dir.")
|
||
home_dir = ""
|
||
end
|
||
end
|
||
local _
|
||
dir, _ = dir:gsub("^~", home_dir)
|
||
return dir
|
||
end
|
||
local is_windows = type(package) == "table" and type(package.config) == "string" and package.config:sub(1, 1) == "\\"
|
||
local trim
|
||
trim = function(s)
|
||
return s:match("^%s*(.-)%s*$")
|
||
end
|
||
local get_null_path
|
||
get_null_path = function()
|
||
if file_exists("/dev/null") then
|
||
return "/dev/null"
|
||
end
|
||
return "NUL"
|
||
end
|
||
local run_subprocess
|
||
run_subprocess = function(params)
|
||
local res = utils.subprocess(params)
|
||
msg.verbose("Command stdout: ")
|
||
msg.verbose(res.stdout)
|
||
if res.status ~= 0 then
|
||
msg.verbose("Command failed! Reason: ", res.error, " Killed by us? ", res.killed_by_us and "yes" or "no")
|
||
return false
|
||
end
|
||
return true
|
||
end
|
||
local shell_escape
|
||
shell_escape = function(args)
|
||
local ret = { }
|
||
for i, a in ipairs(args) do
|
||
local s = tostring(a)
|
||
if string.match(s, "[^A-Za-z0-9_/:=-]") then
|
||
if is_windows then
|
||
s = '"' .. string.gsub(s, '"', '"\\""') .. '"'
|
||
else
|
||
s = "'" .. string.gsub(s, "'", "'\\''") .. "'"
|
||
end
|
||
end
|
||
table.insert(ret, s)
|
||
end
|
||
local concat = table.concat(ret, " ")
|
||
if is_windows then
|
||
concat = '"' .. concat .. '"'
|
||
end
|
||
return concat
|
||
end
|
||
local run_subprocess_popen
|
||
run_subprocess_popen = function(command_line)
|
||
local command_line_string = shell_escape(command_line)
|
||
command_line_string = command_line_string .. " 2>&1"
|
||
msg.verbose("run_subprocess_popen: running " .. tostring(command_line_string))
|
||
return io.popen(command_line_string)
|
||
end
|
||
local calculate_scale_factor
|
||
calculate_scale_factor = function()
|
||
local baseResY = 720
|
||
local osd_w, osd_h = mp.get_osd_size()
|
||
return osd_h / baseResY
|
||
end
|
||
local should_display_progress
|
||
should_display_progress = function()
|
||
if options.display_progress == "auto" then
|
||
return not is_windows
|
||
end
|
||
return options.display_progress
|
||
end
|
||
local reverse
|
||
reverse = function(list)
|
||
local _accum_0 = { }
|
||
local _len_0 = 1
|
||
local _max_0 = 1
|
||
for _index_0 = #list, _max_0 < 0 and #list + _max_0 or _max_0, -1 do
|
||
local element = list[_index_0]
|
||
_accum_0[_len_0] = element
|
||
_len_0 = _len_0 + 1
|
||
end
|
||
return _accum_0
|
||
end
|
||
local get_pass_logfile_path
|
||
get_pass_logfile_path = function(encode_out_path)
|
||
return tostring(encode_out_path) .. "-video-pass1.log"
|
||
end
|
||
local starts_with
|
||
starts_with = function(str, start)
|
||
return string.sub(str, 1, #start) == start
|
||
end
|
||
local encoding_profiles
|
||
do
|
||
local _accum_0 = { }
|
||
local _len_0 = 1
|
||
local _list_0 = (mp.get_property_native('profile-list'))
|
||
for _index_0 = 1, #_list_0 do
|
||
local p = _list_0[_index_0]
|
||
if starts_with(p['name'], 'enc-') then
|
||
_accum_0[_len_0] = p
|
||
_len_0 = _len_0 + 1
|
||
end
|
||
end
|
||
encoding_profiles = _accum_0
|
||
end
|
||
local get_profile_desc
|
||
get_profile_desc = function(name)
|
||
for _index_0 = 1, #encoding_profiles do
|
||
local p = encoding_profiles[_index_0]
|
||
if name == p['name'] then
|
||
return p['profile-desc']
|
||
end
|
||
end
|
||
end
|
||
local dimensions_changed = true
|
||
local _video_dimensions = { }
|
||
local get_video_dimensions
|
||
get_video_dimensions = function()
|
||
if not (dimensions_changed) then
|
||
return _video_dimensions
|
||
end
|
||
local video_params = mp.get_property_native("video-out-params")
|
||
if not video_params then
|
||
return nil
|
||
end
|
||
dimensions_changed = false
|
||
local keep_aspect = mp.get_property_bool("keepaspect")
|
||
local w = video_params["w"]
|
||
local h = video_params["h"]
|
||
local dw = video_params["dw"]
|
||
local dh = video_params["dh"]
|
||
if mp.get_property_number("video-rotate") % 180 == 90 then
|
||
w, h = h, w
|
||
dw, dh = dh, dw
|
||
end
|
||
_video_dimensions = {
|
||
top_left = { },
|
||
bottom_right = { },
|
||
ratios = { }
|
||
}
|
||
local window_w, window_h = mp.get_osd_size()
|
||
if keep_aspect then
|
||
local unscaled = mp.get_property_native("video-unscaled")
|
||
local panscan = mp.get_property_number("panscan")
|
||
local fwidth = window_w
|
||
local fheight = math.floor(window_w / dw * dh)
|
||
if fheight > window_h or fheight < h then
|
||
local tmpw = math.floor(window_h / dh * dw)
|
||
if tmpw <= window_w then
|
||
fheight = window_h
|
||
fwidth = tmpw
|
||
end
|
||
end
|
||
local vo_panscan_area = window_h - fheight
|
||
local f_w = fwidth / fheight
|
||
local f_h = 1
|
||
if vo_panscan_area == 0 then
|
||
vo_panscan_area = window_h - fwidth
|
||
f_w = 1
|
||
f_h = fheight / fwidth
|
||
end
|
||
if unscaled or unscaled == "downscale-big" then
|
||
vo_panscan_area = 0
|
||
if unscaled or (dw <= window_w and dh <= window_h) then
|
||
fwidth = dw
|
||
fheight = dh
|
||
end
|
||
end
|
||
local scaled_width = fwidth + math.floor(vo_panscan_area * panscan * f_w)
|
||
local scaled_height = fheight + math.floor(vo_panscan_area * panscan * f_h)
|
||
local split_scaling
|
||
split_scaling = function(dst_size, scaled_src_size, zoom, align, pan)
|
||
scaled_src_size = math.floor(scaled_src_size * 2 ^ zoom)
|
||
align = (align + 1) / 2
|
||
local dst_start = math.floor((dst_size - scaled_src_size) * align + pan * scaled_src_size)
|
||
if dst_start < 0 then
|
||
dst_start = dst_start + 1
|
||
end
|
||
local dst_end = dst_start + scaled_src_size
|
||
if dst_start >= dst_end then
|
||
dst_start = 0
|
||
dst_end = 1
|
||
end
|
||
return dst_start, dst_end
|
||
end
|
||
local zoom = mp.get_property_number("video-zoom")
|
||
local align_x = mp.get_property_number("video-align-x")
|
||
local pan_x = mp.get_property_number("video-pan-x")
|
||
_video_dimensions.top_left.x, _video_dimensions.bottom_right.x = split_scaling(window_w, scaled_width, zoom, align_x, pan_x)
|
||
local align_y = mp.get_property_number("video-align-y")
|
||
local pan_y = mp.get_property_number("video-pan-y")
|
||
_video_dimensions.top_left.y, _video_dimensions.bottom_right.y = split_scaling(window_h, scaled_height, zoom, align_y, pan_y)
|
||
else
|
||
_video_dimensions.top_left.x = 0
|
||
_video_dimensions.bottom_right.x = window_w
|
||
_video_dimensions.top_left.y = 0
|
||
_video_dimensions.bottom_right.y = window_h
|
||
end
|
||
_video_dimensions.ratios.w = w / (_video_dimensions.bottom_right.x - _video_dimensions.top_left.x)
|
||
_video_dimensions.ratios.h = h / (_video_dimensions.bottom_right.y - _video_dimensions.top_left.y)
|
||
return _video_dimensions
|
||
end
|
||
local set_dimensions_changed
|
||
set_dimensions_changed = function()
|
||
dimensions_changed = true
|
||
end
|
||
local monitor_dimensions
|
||
monitor_dimensions = function()
|
||
local properties = {
|
||
"keepaspect",
|
||
"video-out-params",
|
||
"video-unscaled",
|
||
"panscan",
|
||
"video-zoom",
|
||
"video-align-x",
|
||
"video-pan-x",
|
||
"video-align-y",
|
||
"video-pan-y",
|
||
"osd-width",
|
||
"osd-height"
|
||
}
|
||
for _, p in ipairs(properties) do
|
||
mp.observe_property(p, "native", set_dimensions_changed)
|
||
end
|
||
end
|
||
local clamp
|
||
clamp = function(min, val, max)
|
||
if val <= min then
|
||
return min
|
||
end
|
||
if val >= max then
|
||
return max
|
||
end
|
||
return val
|
||
end
|
||
local clamp_point
|
||
clamp_point = function(top_left, point, bottom_right)
|
||
return {
|
||
x = clamp(top_left.x, point.x, bottom_right.x),
|
||
y = clamp(top_left.y, point.y, bottom_right.y)
|
||
}
|
||
end
|
||
local VideoPoint
|
||
do
|
||
local _class_0
|
||
local _base_0 = {
|
||
set_from_screen = function(self, sx, sy)
|
||
local d = get_video_dimensions()
|
||
local point = clamp_point(d.top_left, {
|
||
x = sx,
|
||
y = sy
|
||
}, d.bottom_right)
|
||
self.x = math.floor(d.ratios.w * (point.x - d.top_left.x) + 0.5)
|
||
self.y = math.floor(d.ratios.h * (point.y - d.top_left.y) + 0.5)
|
||
end,
|
||
to_screen = function(self)
|
||
local d = get_video_dimensions()
|
||
return {
|
||
x = math.floor(self.x / d.ratios.w + d.top_left.x + 0.5),
|
||
y = math.floor(self.y / d.ratios.h + d.top_left.y + 0.5)
|
||
}
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
_class_0 = setmetatable({
|
||
__init = function(self)
|
||
self.x = -1
|
||
self.y = -1
|
||
end,
|
||
__base = _base_0,
|
||
__name = "VideoPoint"
|
||
}, {
|
||
__index = _base_0,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
VideoPoint = _class_0
|
||
end
|
||
local Region
|
||
do
|
||
local _class_0
|
||
local _base_0 = {
|
||
is_valid = function(self)
|
||
return self.x > -1 and self.y > -1 and self.w > -1 and self.h > -1
|
||
end,
|
||
set_from_points = function(self, p1, p2)
|
||
self.x = math.min(p1.x, p2.x)
|
||
self.y = math.min(p1.y, p2.y)
|
||
self.w = math.abs(p1.x - p2.x)
|
||
self.h = math.abs(p1.y - p2.y)
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
_class_0 = setmetatable({
|
||
__init = function(self)
|
||
self.x = -1
|
||
self.y = -1
|
||
self.w = -1
|
||
self.h = -1
|
||
end,
|
||
__base = _base_0,
|
||
__name = "Region"
|
||
}, {
|
||
__index = _base_0,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
Region = _class_0
|
||
end
|
||
local make_fullscreen_region
|
||
make_fullscreen_region = function()
|
||
local r = Region()
|
||
local d = get_video_dimensions()
|
||
local a = VideoPoint()
|
||
local b = VideoPoint()
|
||
local xa, ya
|
||
do
|
||
local _obj_0 = d.top_left
|
||
xa, ya = _obj_0.x, _obj_0.y
|
||
end
|
||
a:set_from_screen(xa, ya)
|
||
local xb, yb
|
||
do
|
||
local _obj_0 = d.bottom_right
|
||
xb, yb = _obj_0.x, _obj_0.y
|
||
end
|
||
b:set_from_screen(xb, yb)
|
||
r:set_from_points(a, b)
|
||
return r
|
||
end
|
||
local Page
|
||
do
|
||
local _class_0
|
||
local _base_0 = {
|
||
add_keybinds = function(self)
|
||
if not self.keybinds then
|
||
return
|
||
end
|
||
for key, func in pairs(self.keybinds) do
|
||
mp.add_forced_key_binding(key, key, func, {
|
||
repeatable = true
|
||
})
|
||
end
|
||
end,
|
||
remove_keybinds = function(self)
|
||
if not self.keybinds then
|
||
return
|
||
end
|
||
for key, _ in pairs(self.keybinds) do
|
||
mp.remove_key_binding(key)
|
||
end
|
||
end,
|
||
observe_properties = function(self)
|
||
self.sizeCallback = function()
|
||
return self:draw()
|
||
end
|
||
local properties = {
|
||
"keepaspect",
|
||
"video-out-params",
|
||
"video-unscaled",
|
||
"panscan",
|
||
"video-zoom",
|
||
"video-align-x",
|
||
"video-pan-x",
|
||
"video-align-y",
|
||
"video-pan-y",
|
||
"osd-width",
|
||
"osd-height"
|
||
}
|
||
for _index_0 = 1, #properties do
|
||
local p = properties[_index_0]
|
||
mp.observe_property(p, "native", self.sizeCallback)
|
||
end
|
||
end,
|
||
unobserve_properties = function(self)
|
||
if self.sizeCallback then
|
||
mp.unobserve_property(self.sizeCallback)
|
||
self.sizeCallback = nil
|
||
end
|
||
end,
|
||
clear = function(self)
|
||
local window_w, window_h = mp.get_osd_size()
|
||
mp.set_osd_ass(window_w, window_h, "")
|
||
return mp.osd_message("", 0)
|
||
end,
|
||
prepare = function(self)
|
||
return nil
|
||
end,
|
||
dispose = function(self)
|
||
return nil
|
||
end,
|
||
show = function(self)
|
||
if self.visible then
|
||
return
|
||
end
|
||
self.visible = true
|
||
self:observe_properties()
|
||
self:add_keybinds()
|
||
self:prepare()
|
||
self:clear()
|
||
return self:draw()
|
||
end,
|
||
hide = function(self)
|
||
if not self.visible then
|
||
return
|
||
end
|
||
self.visible = false
|
||
self:unobserve_properties()
|
||
self:remove_keybinds()
|
||
self:clear()
|
||
return self:dispose()
|
||
end,
|
||
setup_text = function(self, ass)
|
||
local scale = calculate_scale_factor()
|
||
local margin = options.margin * scale
|
||
ass:append("{\\an7}")
|
||
ass:pos(margin, margin)
|
||
return ass:append("{\\fs" .. tostring(options.font_size * scale) .. "}")
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
_class_0 = setmetatable({
|
||
__init = function() end,
|
||
__base = _base_0,
|
||
__name = "Page"
|
||
}, {
|
||
__index = _base_0,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
Page = _class_0
|
||
end
|
||
local EncodeWithProgress
|
||
do
|
||
local _class_0
|
||
local _parent_0 = Page
|
||
local _base_0 = {
|
||
draw = function(self)
|
||
local progress = 100 * ((self.currentTime - self.startTime) / self.duration)
|
||
local progressText = string.format("%d%%", progress)
|
||
local window_w, window_h = mp.get_osd_size()
|
||
local ass = assdraw.ass_new()
|
||
ass:new_event()
|
||
self:setup_text(ass)
|
||
ass:append("Encoding (" .. tostring(bold(progressText)) .. ")\\N")
|
||
return mp.set_osd_ass(window_w, window_h, ass.text)
|
||
end,
|
||
parseLine = function(self, line)
|
||
local matchTime = string.match(line, "Encode time[-]pos: ([0-9.]+)")
|
||
local matchExit = string.match(line, "Exiting... [(]([%a ]+)[)]")
|
||
if matchTime == nil and matchExit == nil then
|
||
return
|
||
end
|
||
if matchTime ~= nil and tonumber(matchTime) > self.currentTime then
|
||
self.currentTime = tonumber(matchTime)
|
||
end
|
||
if matchExit ~= nil then
|
||
self.finished = true
|
||
self.finishedReason = matchExit
|
||
end
|
||
end,
|
||
startEncode = function(self, command_line)
|
||
local copy_command_line
|
||
do
|
||
local _accum_0 = { }
|
||
local _len_0 = 1
|
||
for _index_0 = 1, #command_line do
|
||
local arg = command_line[_index_0]
|
||
_accum_0[_len_0] = arg
|
||
_len_0 = _len_0 + 1
|
||
end
|
||
copy_command_line = _accum_0
|
||
end
|
||
append(copy_command_line, {
|
||
'--term-status-msg=Encode time-pos: ${=time-pos}\\n'
|
||
})
|
||
self:show()
|
||
local processFd = run_subprocess_popen(copy_command_line)
|
||
for line in processFd:lines() do
|
||
msg.verbose(string.format('%q', line))
|
||
self:parseLine(line)
|
||
self:draw()
|
||
end
|
||
processFd:close()
|
||
self:hide()
|
||
if self.finishedReason == "End of file" then
|
||
return true
|
||
end
|
||
return false
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
setmetatable(_base_0, _parent_0.__base)
|
||
_class_0 = setmetatable({
|
||
__init = function(self, startTime, endTime)
|
||
self.startTime = startTime
|
||
self.endTime = endTime
|
||
self.duration = endTime - startTime
|
||
self.currentTime = startTime
|
||
end,
|
||
__base = _base_0,
|
||
__name = "EncodeWithProgress",
|
||
__parent = _parent_0
|
||
}, {
|
||
__index = function(cls, name)
|
||
local val = rawget(_base_0, name)
|
||
if val == nil then
|
||
local parent = rawget(cls, "__parent")
|
||
if parent then
|
||
return parent[name]
|
||
end
|
||
else
|
||
return val
|
||
end
|
||
end,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
if _parent_0.__inherited then
|
||
_parent_0.__inherited(_parent_0, _class_0)
|
||
end
|
||
EncodeWithProgress = _class_0
|
||
end
|
||
local get_active_tracks
|
||
get_active_tracks = function()
|
||
local accepted = {
|
||
video = true,
|
||
audio = not mp.get_property_bool("mute"),
|
||
sub = mp.get_property_bool("sub-visibility")
|
||
}
|
||
local active = {
|
||
video = { },
|
||
audio = { },
|
||
sub = { }
|
||
}
|
||
for _, track in ipairs(mp.get_property_native("track-list")) do
|
||
if track["selected"] and accepted[track["type"]] then
|
||
local count = #active[track["type"]]
|
||
active[track["type"]][count + 1] = track
|
||
end
|
||
end
|
||
return active
|
||
end
|
||
local append_track
|
||
append_track = function(out, track)
|
||
local external_flag = {
|
||
["audio"] = "audio-file",
|
||
["sub"] = "sub-file"
|
||
}
|
||
local internal_flag = {
|
||
["video"] = "vid",
|
||
["audio"] = "aid",
|
||
["sub"] = "sid"
|
||
}
|
||
if track['external'] and string.len(track['external-filename']) <= 2048 then
|
||
return append(out, {
|
||
"--" .. tostring(external_flag[track['type']]) .. "=" .. tostring(track['external-filename'])
|
||
})
|
||
else
|
||
return append(out, {
|
||
"--" .. tostring(internal_flag[track['type']]) .. "=" .. tostring(track['id'])
|
||
})
|
||
end
|
||
end
|
||
local append_audio_tracks
|
||
append_audio_tracks = function(out, tracks)
|
||
local internal_tracks = { }
|
||
for _index_0 = 1, #tracks do
|
||
local track = tracks[_index_0]
|
||
if track['external'] then
|
||
append_track(out, track)
|
||
else
|
||
append(internal_tracks, {
|
||
track
|
||
})
|
||
end
|
||
end
|
||
if #internal_tracks > 1 then
|
||
local filter_string = ""
|
||
for _index_0 = 1, #internal_tracks do
|
||
local track = internal_tracks[_index_0]
|
||
filter_string = filter_string .. "[aid" .. tostring(track['id']) .. "]"
|
||
end
|
||
filter_string = filter_string .. "amix[ao]"
|
||
return append(out, {
|
||
"--lavfi-complex=" .. tostring(filter_string)
|
||
})
|
||
else
|
||
if #internal_tracks == 1 then
|
||
return append_track(out, internal_tracks[1])
|
||
end
|
||
end
|
||
end
|
||
local get_scale_filters
|
||
get_scale_filters = function()
|
||
if options.scale_height > 0 then
|
||
return {
|
||
"lavfi-scale=-2:" .. tostring(options.scale_height)
|
||
}
|
||
end
|
||
return { }
|
||
end
|
||
local append_property
|
||
append_property = function(out, property_name, option_name)
|
||
option_name = option_name or property_name
|
||
local prop = mp.get_property(property_name)
|
||
if prop and prop ~= "" then
|
||
return append(out, {
|
||
"--" .. tostring(option_name) .. "=" .. tostring(prop)
|
||
})
|
||
end
|
||
end
|
||
local append_list_options
|
||
append_list_options = function(out, property_name, option_prefix)
|
||
option_prefix = option_prefix or property_name
|
||
local prop = mp.get_property_native(property_name)
|
||
if prop then
|
||
for _index_0 = 1, #prop do
|
||
local value = prop[_index_0]
|
||
append(out, {
|
||
"--" .. tostring(option_prefix) .. "-append=" .. tostring(value)
|
||
})
|
||
end
|
||
end
|
||
end
|
||
local get_playback_options
|
||
get_playback_options = function()
|
||
local ret = { }
|
||
append_property(ret, "sub-ass-override")
|
||
append_property(ret, "sub-ass-force-style")
|
||
append_property(ret, "sub-auto")
|
||
append_property(ret, "sub-delay")
|
||
append_property(ret, "video-rotate")
|
||
append_property(ret, "ytdl-format")
|
||
return ret
|
||
end
|
||
local get_speed_flags
|
||
get_speed_flags = function()
|
||
local ret = { }
|
||
local speed = mp.get_property_native("speed")
|
||
if speed ~= 1 then
|
||
append(ret, {
|
||
"--vf-add=setpts=PTS/" .. tostring(speed),
|
||
"--af-add=atempo=" .. tostring(speed),
|
||
"--sub-speed=1/" .. tostring(speed)
|
||
})
|
||
end
|
||
return ret
|
||
end
|
||
local get_metadata_flags
|
||
get_metadata_flags = function()
|
||
local title = mp.get_property("filename/no-ext")
|
||
return {
|
||
"--oset-metadata=title=%" .. tostring(string.len(title)) .. "%" .. tostring(title)
|
||
}
|
||
end
|
||
local apply_current_filters
|
||
apply_current_filters = function(filters)
|
||
local vf = mp.get_property_native("vf")
|
||
msg.verbose("apply_current_filters: got " .. tostring(#vf) .. " currently applied.")
|
||
for _index_0 = 1, #vf do
|
||
local _continue_0 = false
|
||
repeat
|
||
local filter = vf[_index_0]
|
||
msg.verbose("apply_current_filters: filter name: " .. tostring(filter['name']))
|
||
if filter["enabled"] == false then
|
||
_continue_0 = true
|
||
break
|
||
end
|
||
local str = filter["name"]
|
||
local params = filter["params"] or { }
|
||
for k, v in pairs(params) do
|
||
str = str .. ":" .. tostring(k) .. "=%" .. tostring(string.len(v)) .. "%" .. tostring(v)
|
||
end
|
||
append(filters, {
|
||
str
|
||
})
|
||
_continue_0 = true
|
||
until true
|
||
if not _continue_0 then
|
||
break
|
||
end
|
||
end
|
||
end
|
||
local encode
|
||
encode = function(region, startTime, endTime)
|
||
local path = mp.get_property("path")
|
||
if not path then
|
||
message("No file is being played")
|
||
return
|
||
end
|
||
local is_stream = not file_exists(path)
|
||
local cutStartTime = startTime
|
||
if is_stream and mp.get_property('file-format') == 'hls' then
|
||
path = utils.join_path(options.output_directory, 'cache_dump.ts')
|
||
mp.command_native({
|
||
'dump-cache',
|
||
seconds_to_time_string(startTime, false, true),
|
||
seconds_to_time_string((endTime + 5), false, true),
|
||
path
|
||
})
|
||
cutStartTime = 0
|
||
end
|
||
local command = {
|
||
"mpv",
|
||
path,
|
||
"--start=" .. seconds_to_time_string(cutStartTime, false, true),
|
||
"--end=" .. seconds_to_time_string(endTime, false, true),
|
||
"--profile=" .. tostring(options.encoding_profile),
|
||
"--loop-file=no"
|
||
}
|
||
if region and region:is_valid() then
|
||
append(command, {
|
||
"--vf-add=lavfi-crop=" .. tostring(region.w) .. ":" .. tostring(region.h) .. ":" .. tostring(region.x) .. ":" .. tostring(region.y)
|
||
})
|
||
end
|
||
if options.write_filename_on_metadata then
|
||
append(command, get_metadata_flags())
|
||
end
|
||
append(command, {
|
||
"--ovcopts-add=crf=" .. tostring(options.crf)
|
||
})
|
||
local dir = ""
|
||
if is_stream then
|
||
dir = parse_directory("~")
|
||
else
|
||
local _
|
||
dir, _ = utils.split_path(path)
|
||
end
|
||
if options.output_directory ~= "" then
|
||
dir = parse_directory(options.output_directory)
|
||
end
|
||
local extension
|
||
for _index_0 = 1, #encoding_profiles do
|
||
local _continue_0 = false
|
||
repeat
|
||
local p = encoding_profiles[_index_0]
|
||
if p['name'] ~= options.encoding_profile then
|
||
_continue_0 = true
|
||
break
|
||
end
|
||
local _list_0 = p['options']
|
||
for _index_1 = 1, #_list_0 do
|
||
local _continue_1 = false
|
||
repeat
|
||
local o = _list_0[_index_1]
|
||
if o['key'] ~= 'of' then
|
||
_continue_1 = true
|
||
break
|
||
end
|
||
extension = o['value']
|
||
_continue_1 = true
|
||
until true
|
||
if not _continue_1 then
|
||
break
|
||
end
|
||
end
|
||
_continue_0 = true
|
||
until true
|
||
if not _continue_0 then
|
||
break
|
||
end
|
||
end
|
||
local formatted_filename = format_filename(startTime, endTime, extension)
|
||
local out_path = utils.join_path(dir, formatted_filename)
|
||
append(command, {
|
||
"--o=" .. tostring(out_path)
|
||
})
|
||
msg.info("Encoding to", out_path)
|
||
msg.verbose("Command line:", table.concat(command, " "))
|
||
if options.run_detached then
|
||
message("Started encode, process was detached.")
|
||
return utils.subprocess_detached({
|
||
args = command
|
||
})
|
||
else
|
||
local res = false
|
||
if not should_display_progress() then
|
||
message("Started encode...")
|
||
res = run_subprocess({
|
||
args = command,
|
||
cancellable = false
|
||
})
|
||
else
|
||
local ewp = EncodeWithProgress(startTime, endTime)
|
||
res = ewp:startEncode(command)
|
||
end
|
||
if res then
|
||
message("Encoded successfully! Saved to\\N" .. tostring(bold(out_path)))
|
||
else
|
||
message("Encode failed! Check the logs for details.")
|
||
end
|
||
os.remove(get_pass_logfile_path(out_path))
|
||
if is_stream then
|
||
return os.remove(path)
|
||
end
|
||
end
|
||
end
|
||
local CropPage
|
||
do
|
||
local _class_0
|
||
local _parent_0 = Page
|
||
local _base_0 = {
|
||
reset = function(self)
|
||
local dimensions = get_video_dimensions()
|
||
local xa, ya
|
||
do
|
||
local _obj_0 = dimensions.top_left
|
||
xa, ya = _obj_0.x, _obj_0.y
|
||
end
|
||
self.pointA:set_from_screen(xa, ya)
|
||
local xb, yb
|
||
do
|
||
local _obj_0 = dimensions.bottom_right
|
||
xb, yb = _obj_0.x, _obj_0.y
|
||
end
|
||
self.pointB:set_from_screen(xb, yb)
|
||
if self.visible then
|
||
return self:draw()
|
||
end
|
||
end,
|
||
setPointA = function(self)
|
||
local posX, posY = mp.get_mouse_pos()
|
||
self.pointA:set_from_screen(posX, posY)
|
||
if self.visible then
|
||
return self:draw()
|
||
end
|
||
end,
|
||
setPointB = function(self)
|
||
local posX, posY = mp.get_mouse_pos()
|
||
self.pointB:set_from_screen(posX, posY)
|
||
if self.visible then
|
||
return self:draw()
|
||
end
|
||
end,
|
||
cancel = function(self)
|
||
self:hide()
|
||
return self.callback(false, nil)
|
||
end,
|
||
finish = function(self)
|
||
local region = Region()
|
||
region:set_from_points(self.pointA, self.pointB)
|
||
self:hide()
|
||
return self.callback(true, region)
|
||
end,
|
||
draw_box = function(self, ass)
|
||
local region = Region()
|
||
region:set_from_points(self.pointA:to_screen(), self.pointB:to_screen())
|
||
local d = get_video_dimensions()
|
||
ass:new_event()
|
||
ass:append("{\\an7}")
|
||
ass:pos(0, 0)
|
||
ass:append('{\\bord0}')
|
||
ass:append('{\\shad0}')
|
||
ass:append('{\\c&H000000&}')
|
||
ass:append('{\\alpha&H77}')
|
||
ass:draw_start()
|
||
ass:rect_cw(d.top_left.x, d.top_left.y, region.x, region.y + region.h)
|
||
ass:rect_cw(region.x, d.top_left.y, d.bottom_right.x, region.y)
|
||
ass:rect_cw(d.top_left.x, region.y + region.h, region.x + region.w, d.bottom_right.y)
|
||
ass:rect_cw(region.x + region.w, region.y, d.bottom_right.x, d.bottom_right.y)
|
||
return ass:draw_stop()
|
||
end,
|
||
draw = function(self)
|
||
local window = { }
|
||
window.w, window.h = mp.get_osd_size()
|
||
local ass = assdraw.ass_new()
|
||
self:draw_box(ass)
|
||
ass:new_event()
|
||
self:setup_text(ass)
|
||
ass:append(tostring(bold('Crop:')) .. "\\N")
|
||
ass:append(tostring(bold('1:')) .. " change point A (" .. tostring(self.pointA.x) .. ", " .. tostring(self.pointA.y) .. ")\\N")
|
||
ass:append(tostring(bold('2:')) .. " change point B (" .. tostring(self.pointB.x) .. ", " .. tostring(self.pointB.y) .. ")\\N")
|
||
ass:append(tostring(bold('r:')) .. " reset to whole screen\\N")
|
||
ass:append(tostring(bold('ESC:')) .. " cancel crop\\N")
|
||
local width, height = math.abs(self.pointA.x - self.pointB.x), math.abs(self.pointA.y - self.pointB.y)
|
||
ass:append(tostring(bold('ENTER:')) .. " confirm crop (" .. tostring(width) .. "x" .. tostring(height) .. ")\\N")
|
||
return mp.set_osd_ass(window.w, window.h, ass.text)
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
setmetatable(_base_0, _parent_0.__base)
|
||
_class_0 = setmetatable({
|
||
__init = function(self, callback, region)
|
||
self.pointA = VideoPoint()
|
||
self.pointB = VideoPoint()
|
||
self.keybinds = {
|
||
["1"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.setPointA
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["2"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.setPointB
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["r"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.reset
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["ESC"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.cancel
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["ENTER"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.finish
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)()
|
||
}
|
||
self:reset()
|
||
self.callback = callback
|
||
if region and region:is_valid() then
|
||
self.pointA.x = region.x
|
||
self.pointA.y = region.y
|
||
self.pointB.x = region.x + region.w
|
||
self.pointB.y = region.y + region.h
|
||
end
|
||
end,
|
||
__base = _base_0,
|
||
__name = "CropPage",
|
||
__parent = _parent_0
|
||
}, {
|
||
__index = function(cls, name)
|
||
local val = rawget(_base_0, name)
|
||
if val == nil then
|
||
local parent = rawget(cls, "__parent")
|
||
if parent then
|
||
return parent[name]
|
||
end
|
||
else
|
||
return val
|
||
end
|
||
end,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
if _parent_0.__inherited then
|
||
_parent_0.__inherited(_parent_0, _class_0)
|
||
end
|
||
CropPage = _class_0
|
||
end
|
||
local Option
|
||
do
|
||
local _class_0
|
||
local _base_0 = {
|
||
hasPrevious = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
return true
|
||
elseif "int" == _exp_0 then
|
||
if self.opts.min then
|
||
return self.value > self.opts.min
|
||
else
|
||
return true
|
||
end
|
||
elseif "list" == _exp_0 then
|
||
return self.value > 1
|
||
end
|
||
end,
|
||
hasNext = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
return true
|
||
elseif "int" == _exp_0 then
|
||
if self.opts.max then
|
||
return self.value < self.opts.max
|
||
else
|
||
return true
|
||
end
|
||
elseif "list" == _exp_0 then
|
||
return self.value < #self.opts.possibleValues
|
||
end
|
||
end,
|
||
leftKey = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
self.value = not self.value
|
||
elseif "int" == _exp_0 then
|
||
self.value = self.value - self.opts.step
|
||
if self.opts.min and self.opts.min > self.value then
|
||
self.value = self.opts.min
|
||
end
|
||
elseif "list" == _exp_0 then
|
||
if self.value > 1 then
|
||
self.value = self.value - 1
|
||
end
|
||
end
|
||
end,
|
||
rightKey = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
self.value = not self.value
|
||
elseif "int" == _exp_0 then
|
||
self.value = self.value + self.opts.step
|
||
if self.opts.max and self.opts.max < self.value then
|
||
self.value = self.opts.max
|
||
end
|
||
elseif "list" == _exp_0 then
|
||
if self.value < #self.opts.possibleValues then
|
||
self.value = self.value + 1
|
||
end
|
||
end
|
||
end,
|
||
getValue = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
return self.value
|
||
elseif "int" == _exp_0 then
|
||
return self.value
|
||
elseif "list" == _exp_0 then
|
||
local value, _
|
||
do
|
||
local _obj_0 = self.opts.possibleValues[self.value]
|
||
value, _ = _obj_0[1], _obj_0[2]
|
||
end
|
||
return value
|
||
end
|
||
end,
|
||
setValue = function(self, value)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
self.value = value
|
||
elseif "int" == _exp_0 then
|
||
self.value = value
|
||
elseif "list" == _exp_0 then
|
||
local set = false
|
||
for i, possiblePair in ipairs(self.opts.possibleValues) do
|
||
local possibleValue, _
|
||
possibleValue, _ = possiblePair[1], possiblePair[2]
|
||
if possibleValue == value then
|
||
set = true
|
||
self.value = i
|
||
break
|
||
end
|
||
end
|
||
if not set then
|
||
return msg.warn("Tried to set invalid value " .. tostring(value) .. " to " .. tostring(self.displayText) .. " option.")
|
||
end
|
||
end
|
||
end,
|
||
getDisplayValue = function(self)
|
||
local _exp_0 = self.optType
|
||
if "bool" == _exp_0 then
|
||
return self.value and "yes" or "no"
|
||
elseif "int" == _exp_0 then
|
||
if self.opts.altDisplayNames and self.opts.altDisplayNames[self.value] then
|
||
return self.opts.altDisplayNames[self.value]
|
||
else
|
||
return tostring(self.value)
|
||
end
|
||
elseif "list" == _exp_0 then
|
||
local value, displayValue
|
||
do
|
||
local _obj_0 = self.opts.possibleValues[self.value]
|
||
value, displayValue = _obj_0[1], _obj_0[2]
|
||
end
|
||
return displayValue or value
|
||
end
|
||
end,
|
||
draw = function(self, ass, selected)
|
||
if selected then
|
||
ass:append(tostring(bold(self.displayText)) .. ": ")
|
||
else
|
||
ass:append(tostring(self.displayText) .. ": ")
|
||
end
|
||
if self:hasPrevious() then
|
||
ass:append("◀ ")
|
||
end
|
||
ass:append(self:getDisplayValue())
|
||
if self:hasNext() then
|
||
ass:append(" ▶")
|
||
end
|
||
return ass:append("\\N")
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
_class_0 = setmetatable({
|
||
__init = function(self, optType, displayText, value, opts)
|
||
self.optType = optType
|
||
self.displayText = displayText
|
||
self.opts = opts
|
||
self.value = 1
|
||
return self:setValue(value)
|
||
end,
|
||
__base = _base_0,
|
||
__name = "Option"
|
||
}, {
|
||
__index = _base_0,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
Option = _class_0
|
||
end
|
||
local EncodeOptionsPage
|
||
do
|
||
local _class_0
|
||
local _parent_0 = Page
|
||
local _base_0 = {
|
||
getCurrentOption = function(self)
|
||
return self.options[self.currentOption][2]
|
||
end,
|
||
leftKey = function(self)
|
||
(self:getCurrentOption()):leftKey()
|
||
return self:draw()
|
||
end,
|
||
rightKey = function(self)
|
||
(self:getCurrentOption()):rightKey()
|
||
return self:draw()
|
||
end,
|
||
prevOpt = function(self)
|
||
self.currentOption = math.max(1, self.currentOption - 1)
|
||
return self:draw()
|
||
end,
|
||
nextOpt = function(self)
|
||
self.currentOption = math.min(#self.options, self.currentOption + 1)
|
||
return self:draw()
|
||
end,
|
||
confirmOpts = function(self)
|
||
for _, optPair in ipairs(self.options) do
|
||
local optName, opt
|
||
optName, opt = optPair[1], optPair[2]
|
||
options[optName] = opt:getValue()
|
||
end
|
||
self:hide()
|
||
return self.callback(true)
|
||
end,
|
||
cancelOpts = function(self)
|
||
self:hide()
|
||
return self.callback(false)
|
||
end,
|
||
draw = function(self)
|
||
local window_w, window_h = mp.get_osd_size()
|
||
local ass = assdraw.ass_new()
|
||
ass:new_event()
|
||
self:setup_text(ass)
|
||
ass:append(tostring(bold('Options:')) .. "\\N\\N")
|
||
for i, optPair in ipairs(self.options) do
|
||
local opt = optPair[2]
|
||
opt:draw(ass, self.currentOption == i)
|
||
end
|
||
ass:append("\\N▲ / ▼: navigate\\N")
|
||
ass:append(tostring(bold('ENTER:')) .. " confirm options\\N")
|
||
ass:append(tostring(bold('ESC:')) .. " cancel\\N")
|
||
return mp.set_osd_ass(window_w, window_h, ass.text)
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
setmetatable(_base_0, _parent_0.__base)
|
||
_class_0 = setmetatable({
|
||
__init = function(self, callback)
|
||
self.callback = callback
|
||
self.currentOption = 1
|
||
local crfOpts = {
|
||
step = 1,
|
||
min = -1,
|
||
altDisplayNames = {
|
||
[-1] = "disabled"
|
||
}
|
||
}
|
||
local profileOpts = {
|
||
possibleValues = (function()
|
||
local _accum_0 = { }
|
||
local _len_0 = 1
|
||
for _index_0 = 1, #encoding_profiles do
|
||
local p = encoding_profiles[_index_0]
|
||
_accum_0[_len_0] = {
|
||
p['name'],
|
||
p['profile-desc']
|
||
}
|
||
_len_0 = _len_0 + 1
|
||
end
|
||
return _accum_0
|
||
end)()
|
||
}
|
||
self.options = {
|
||
{
|
||
"encoding_profile",
|
||
Option("list", "Encoding profile", options.encoding_profile, profileOpts)
|
||
},
|
||
{
|
||
"crf",
|
||
Option("int", "CRF", options.crf, crfOpts)
|
||
}
|
||
}
|
||
self.keybinds = {
|
||
["LEFT"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.leftKey
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["RIGHT"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.rightKey
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["UP"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.prevOpt
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["DOWN"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.nextOpt
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["ENTER"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.confirmOpts
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["ESC"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.cancelOpts
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)()
|
||
}
|
||
end,
|
||
__base = _base_0,
|
||
__name = "EncodeOptionsPage",
|
||
__parent = _parent_0
|
||
}, {
|
||
__index = function(cls, name)
|
||
local val = rawget(_base_0, name)
|
||
if val == nil then
|
||
local parent = rawget(cls, "__parent")
|
||
if parent then
|
||
return parent[name]
|
||
end
|
||
else
|
||
return val
|
||
end
|
||
end,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
if _parent_0.__inherited then
|
||
_parent_0.__inherited(_parent_0, _class_0)
|
||
end
|
||
EncodeOptionsPage = _class_0
|
||
end
|
||
local PreviewPage
|
||
do
|
||
local _class_0
|
||
local _parent_0 = Page
|
||
local _base_0 = {
|
||
prepare = function(self)
|
||
local vf = mp.get_property_native("vf")
|
||
vf[#vf + 1] = {
|
||
name = "sub"
|
||
}
|
||
if self.region:is_valid() then
|
||
vf[#vf + 1] = {
|
||
name = "crop",
|
||
params = {
|
||
w = tostring(self.region.w),
|
||
h = tostring(self.region.h),
|
||
x = tostring(self.region.x),
|
||
y = tostring(self.region.y)
|
||
}
|
||
}
|
||
end
|
||
mp.set_property_native("vf", vf)
|
||
if self.startTime > -1 and self.endTime > -1 then
|
||
mp.set_property_native("ab-loop-a", self.startTime)
|
||
mp.set_property_native("ab-loop-b", self.endTime)
|
||
mp.set_property_native("time-pos", self.startTime)
|
||
end
|
||
return mp.set_property_native("pause", false)
|
||
end,
|
||
dispose = function(self)
|
||
mp.set_property("ab-loop-a", "no")
|
||
mp.set_property("ab-loop-b", "no")
|
||
for prop, value in pairs(self.originalProperties) do
|
||
mp.set_property_native(prop, value)
|
||
end
|
||
end,
|
||
draw = function(self)
|
||
local window_w, window_h = mp.get_osd_size()
|
||
local ass = assdraw.ass_new()
|
||
ass:new_event()
|
||
self:setup_text(ass)
|
||
ass:append("Press " .. tostring(bold('ESC')) .. " to exit preview.\\N")
|
||
return mp.set_osd_ass(window_w, window_h, ass.text)
|
||
end,
|
||
cancel = function(self)
|
||
self:hide()
|
||
return self.callback()
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
setmetatable(_base_0, _parent_0.__base)
|
||
_class_0 = setmetatable({
|
||
__init = function(self, callback, region, startTime, endTime)
|
||
self.callback = callback
|
||
self.originalProperties = {
|
||
["vf"] = mp.get_property_native("vf"),
|
||
["time-pos"] = mp.get_property_native("time-pos"),
|
||
["pause"] = mp.get_property_native("pause")
|
||
}
|
||
self.keybinds = {
|
||
["ESC"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.cancel
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)()
|
||
}
|
||
self.region = region
|
||
self.startTime = startTime
|
||
self.endTime = endTime
|
||
self.isLoop = false
|
||
end,
|
||
__base = _base_0,
|
||
__name = "PreviewPage",
|
||
__parent = _parent_0
|
||
}, {
|
||
__index = function(cls, name)
|
||
local val = rawget(_base_0, name)
|
||
if val == nil then
|
||
local parent = rawget(cls, "__parent")
|
||
if parent then
|
||
return parent[name]
|
||
end
|
||
else
|
||
return val
|
||
end
|
||
end,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
if _parent_0.__inherited then
|
||
_parent_0.__inherited(_parent_0, _class_0)
|
||
end
|
||
PreviewPage = _class_0
|
||
end
|
||
local MainPage
|
||
do
|
||
local _class_0
|
||
local _parent_0 = Page
|
||
local _base_0 = {
|
||
setStartTime = function(self)
|
||
self.startTime = mp.get_property_number("time-pos")
|
||
if self.visible then
|
||
self:clear()
|
||
return self:draw()
|
||
end
|
||
end,
|
||
setEndTime = function(self)
|
||
self.endTime = mp.get_property_number("time-pos")
|
||
if self.visible then
|
||
self:clear()
|
||
return self:draw()
|
||
end
|
||
end,
|
||
setupStartAndEndTimes = function(self)
|
||
if mp.get_property_native("duration") then
|
||
self.startTime = 0
|
||
self.endTime = mp.get_property_native("duration")
|
||
else
|
||
self.startTime = -1
|
||
self.endTime = -1
|
||
end
|
||
if self.visible then
|
||
self:clear()
|
||
return self:draw()
|
||
end
|
||
end,
|
||
draw = function(self)
|
||
local window_w, window_h = mp.get_osd_size()
|
||
local ass = assdraw.ass_new()
|
||
ass:new_event()
|
||
self:setup_text(ass)
|
||
ass:append(tostring(bold('Profile:')) .. " " .. tostring(get_profile_desc(options.encoding_profile)) .. "\\N")
|
||
ass:append(tostring(bold('CRF:')) .. " " .. tostring(options.crf) .. "\\N")
|
||
ass:append(tostring(bold('Start time:')) .. " " .. tostring(seconds_to_time_string(self.startTime)) .. "\\N")
|
||
ass:append(tostring(bold('End time:')) .. " " .. tostring(seconds_to_time_string(self.endTime)) .. "\\N")
|
||
if self.region.x > 0 and self.region.y > 0 then
|
||
ass:append(tostring(bold('Crop:')) .. " " .. tostring(self.region.x) .. "×" .. tostring(self.region.y))
|
||
end
|
||
ass:append("\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N\\N")
|
||
ass:append(tostring(bold('C:')) .. " crop\\N")
|
||
ass:append(tostring(bold('1:')) .. " set start time\\N")
|
||
ass:append(tostring(bold('2:')) .. " set end time\\N")
|
||
ass:append(tostring(bold('O:')) .. " change options\\N")
|
||
ass:append(tostring(bold('P:')) .. " preview\\N")
|
||
ass:append(tostring(bold('E:')) .. " encode\\N")
|
||
ass:append(tostring(bold('ESC:')) .. " close\\N")
|
||
return mp.set_osd_ass(window_w, window_h, ass.text)
|
||
end,
|
||
onUpdateCropRegion = function(self, updated, newRegion)
|
||
if updated then
|
||
self.region = newRegion
|
||
end
|
||
return self:show()
|
||
end,
|
||
crop = function(self)
|
||
self:hide()
|
||
local cropPage = CropPage((function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.onUpdateCropRegion
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(), self.region)
|
||
return cropPage:show()
|
||
end,
|
||
onOptionsChanged = function(self, updated)
|
||
return self:show()
|
||
end,
|
||
changeOptions = function(self)
|
||
self:hide()
|
||
local encodeOptsPage = EncodeOptionsPage((function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.onOptionsChanged
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)())
|
||
return encodeOptsPage:show()
|
||
end,
|
||
onPreviewEnded = function(self)
|
||
return self:show()
|
||
end,
|
||
preview = function(self)
|
||
self:hide()
|
||
local previewPage = PreviewPage((function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.onPreviewEnded
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(), self.region, self.startTime, self.endTime)
|
||
return previewPage:show()
|
||
end,
|
||
encode = function(self)
|
||
self:hide()
|
||
if self.startTime < 0 then
|
||
message("No start time, aborting")
|
||
return
|
||
end
|
||
if self.endTime < 0 then
|
||
message("No end time, aborting")
|
||
return
|
||
end
|
||
if self.startTime >= self.endTime then
|
||
message("Start time is ahead of end time, aborting")
|
||
return
|
||
end
|
||
return encode(self.region, self.startTime, self.endTime)
|
||
end
|
||
}
|
||
_base_0.__index = _base_0
|
||
setmetatable(_base_0, _parent_0.__base)
|
||
_class_0 = setmetatable({
|
||
__init = function(self)
|
||
self.keybinds = {
|
||
["c"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.crop
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["1"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.setStartTime
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["2"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.setEndTime
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["o"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.changeOptions
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["p"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.preview
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["e"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.encode
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)(),
|
||
["ESC"] = (function()
|
||
local _base_1 = self
|
||
local _fn_0 = _base_1.hide
|
||
return function(...)
|
||
return _fn_0(_base_1, ...)
|
||
end
|
||
end)()
|
||
}
|
||
self.startTime = -1
|
||
self.endTime = -1
|
||
self.region = Region()
|
||
end,
|
||
__base = _base_0,
|
||
__name = "MainPage",
|
||
__parent = _parent_0
|
||
}, {
|
||
__index = function(cls, name)
|
||
local val = rawget(_base_0, name)
|
||
if val == nil then
|
||
local parent = rawget(cls, "__parent")
|
||
if parent then
|
||
return parent[name]
|
||
end
|
||
else
|
||
return val
|
||
end
|
||
end,
|
||
__call = function(cls, ...)
|
||
local _self_0 = setmetatable({}, _base_0)
|
||
cls.__init(_self_0, ...)
|
||
return _self_0
|
||
end
|
||
})
|
||
_base_0.__class = _class_0
|
||
if _parent_0.__inherited then
|
||
_parent_0.__inherited(_parent_0, _class_0)
|
||
end
|
||
MainPage = _class_0
|
||
end
|
||
monitor_dimensions()
|
||
local mainPage = MainPage()
|
||
mp.add_key_binding(options.keybind, "display-webm-encoder", (function()
|
||
local _base_0 = mainPage
|
||
local _fn_0 = _base_0.show
|
||
return function(...)
|
||
return _fn_0(_base_0, ...)
|
||
end
|
||
end)(), {
|
||
repeatable = false
|
||
})
|
||
return mp.register_event("file-loaded", (function()
|
||
local _base_0 = mainPage
|
||
local _fn_0 = _base_0.setupStartAndEndTimes
|
||
return function(...)
|
||
return _fn_0(_base_0, ...)
|
||
end
|
||
end)())
|