Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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 mpopts = require("mp.options")
- local options = {
- -- Defaults to shift+w
- keybind = "W",
- -- If empty, saves on the same directory of the playing video.
- -- A starting "~" will be replaced by the home dir.
- output_directory = "",
- 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 time, without milliseconds
- -- %M - "-audio", if audio is enabled, empty otherwise
- output_template = "%T-[%s-%e]%M",
- -- 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 = 0,
- -- 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 = 128,
- -- Sets the output format, from a few predefined ones.
- -- Currently we have webm-vp8 (libvpx/libvorbis), webm-vp9 (libvpx-vp9/libopus)
- -- and raw (rawvideo/pcm_s16le).
- output_format = "webm-vp9",
- 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 = "",
- -- Useful for flags that may impact output filesize, such as crf, qmin, qmax etc
- -- Won't be applied when strict_filesize_constraint is on.
- non_strict_additional_flags = "--ovcopts-add=crf=10,quality=best",
- -- 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
- }
- mpopts.read_options(options)
- 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 f = io.open(name, "r")
- if f ~= nil then
- io.close(f)
- return true
- end
- return false
- end
- local format_filename
- format_filename = function(startTime, endTime, videoFormat)
- 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 ''
- }
- 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(videoFormat.outputExtension)
- 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_mpv_path
- get_mpv_path = function()
- if not is_windows then
- return "mpv"
- end
- local pid = utils.getpid()
- local res = utils.subprocess({
- args = {
- "wmic",
- "process",
- "where",
- "processid=" .. tostring(pid),
- "get",
- "ExecutablePath",
- "/VALUE"
- }
- })
- local key_value = trim(res.stdout)
- return key_value:sub(string.len("ExecutablePath=") + 1)
- 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)
- if res.status ~= 0 then
- msg.verbose("Command failed! Reason: ", res.error, " Killed by us? ", res.killed_by_us and "yes" or "no")
- msg.verbose("Command stdout: ")
- msg.verbose(res.stdout)
- 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 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 formats = { }
- local Format
- do
- local _class_0
- local _base_0 = {
- getPreFilters = function(self)
- return { }
- end,
- getPostFilters = function(self)
- return { }
- end,
- getFlags = function(self)
- return { }
- end
- }
- _base_0.__index = _base_0
- _class_0 = setmetatable({
- __init = function(self)
- self.displayName = "Basic"
- self.supportsTwopass = true
- self.videoCodec = ""
- self.audioCodec = ""
- self.outputExtension = ""
- self.acceptsBitrate = true
- end,
- __base = _base_0,
- __name = "Format"
- }, {
- __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
- Format = _class_0
- end
- local RawVideo
- do
- local _class_0
- local _parent_0 = Format
- local _base_0 = {
- getColorspace = function(self)
- local csp = mp.get_property("colormatrix")
- local _exp_0 = csp
- if "bt.601" == _exp_0 then
- return "bt601"
- elseif "bt.709" == _exp_0 then
- return "bt709"
- elseif "bt.2020" == _exp_0 then
- return "bt2020"
- elseif "smpte-240m" == _exp_0 then
- return "smpte240m"
- else
- msg.info("Warning, unknown colorspace " .. tostring(csp) .. " detected, using bt.601.")
- return "bt601"
- end
- end,
- getPostFilters = function(self)
- return {
- "format=yuv444p16",
- "lavfi-scale=in_color_matrix=" .. self:getColorspace(),
- "format=bgr24"
- }
- end
- }
- _base_0.__index = _base_0
- setmetatable(_base_0, _parent_0.__base)
- _class_0 = setmetatable({
- __init = function(self)
- self.displayName = "Raw"
- self.supportsTwopass = false
- self.videoCodec = "rawvideo"
- self.audioCodec = "pcm_s16le"
- self.outputExtension = "avi"
- self.acceptsBitrate = false
- end,
- __base = _base_0,
- __name = "RawVideo",
- __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
- RawVideo = _class_0
- end
- formats["raw"] = RawVideo()
- local WebmVP8
- do
- local _class_0
- local _parent_0 = Format
- local _base_0 = {
- getPreFilters = function(self)
- local colormatrixFilter = {
- ["bt.709"] = "bt709",
- ["bt.2020"] = "bt2020",
- ["smpte-240m"] = "smpte240m"
- }
- local ret = { }
- local colormatrix = mp.get_property_native("video-params/colormatrix")
- if colormatrixFilter[colormatrix] then
- append(ret, {
- "lavfi-colormatrix=" .. tostring(colormatrixFilter[colormatrix]) .. ":bt601"
- })
- end
- return ret
- end,
- getFlags = function(self)
- return {
- "--ovcopts-add=threads=" .. tostring(options.libvpx_threads)
- }
- end
- }
- _base_0.__index = _base_0
- setmetatable(_base_0, _parent_0.__base)
- _class_0 = setmetatable({
- __init = function(self)
- self.displayName = "WebM"
- self.supportsTwopass = true
- self.videoCodec = "libvpx"
- self.audioCodec = "libvorbis"
- self.outputExtension = "webm"
- self.acceptsBitrate = true
- end,
- __base = _base_0,
- __name = "WebmVP8",
- __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
- WebmVP8 = _class_0
- end
- formats["webm-vp8"] = WebmVP8()
- local WebmVP9
- do
- local _class_0
- local _parent_0 = Format
- local _base_0 = {
- getFlags = function(self)
- return {
- "--ovcopts-add=threads=" .. tostring(options.libvpx_threads)
- }
- end
- }
- _base_0.__index = _base_0
- setmetatable(_base_0, _parent_0.__base)
- _class_0 = setmetatable({
- __init = function(self)
- self.displayName = "WebM (VP9)"
- self.supportsTwopass = true
- self.videoCodec = "libvpx-vp9"
- self.audioCodec = "libopus"
- self.outputExtension = "webm"
- self.acceptsBitrate = true
- end,
- __base = _base_0,
- __name = "WebmVP9",
- __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
- WebmVP9 = _class_0
- end
- formats["webm-vp9"] = WebmVP9()
- local MP4
- do
- local _class_0
- local _parent_0 = Format
- local _base_0 = { }
- _base_0.__index = _base_0
- setmetatable(_base_0, _parent_0.__base)
- _class_0 = setmetatable({
- __init = function(self)
- self.displayName = "MP4 (h264/AAC)"
- self.supportsTwopass = true
- self.videoCodec = "libx264"
- self.audioCodec = "aac"
- self.outputExtension = "mp4"
- self.acceptsBitrate = true
- end,
- __base = _base_0,
- __name = "MP4",
- __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
- MP4 = _class_0
- end
- formats["mp4"] = MP4()
- 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)
- self.visible = true
- self:observe_properties()
- self:add_keybinds()
- self:prepare()
- self:clear()
- return self:draw()
- end,
- hide = function(self)
- 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: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}'
- })
- 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 = { }
- for _, track in ipairs(mp.get_property_native("track-list")) do
- if track["selected"] and accepted[track["type"]] then
- active[#active + 1] = track
- end
- end
- return active
- end
- local get_color_conversion_filters
- get_color_conversion_filters = function()
- local colormatrixFilter = {
- ["bt.709"] = "bt709",
- ["bt.2020"] = "bt2020"
- }
- local ret = { }
- local colormatrix = mp.get_property_native("video-params/colormatrix")
- if options.video_codec == "libvpx" and colormatrixFilter[colormatrix] then
- append(ret, {
- "colormatrix=" .. tostring(colormatrixFilter[colormatrix]) .. ":bt601"
- })
- end
- return ret
- 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, "video-rotate")
- for _, track in ipairs(mp.get_property_native("track-list")) do
- if track["type"] == "sub" and track["external"] then
- append(ret, {
- "--sub-files-append=" .. tostring(track['external-filename'])
- })
- end
- 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 format = formats[options.output_format]
- 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 command = {
- get_mpv_path(),
- path,
- "--start=" .. seconds_to_time_string(startTime, false, true),
- "--end=" .. seconds_to_time_string(endTime, false, true),
- "--ovc=" .. tostring(format.videoCodec),
- "--oac=" .. tostring(format.audioCodec),
- "--loop-file=no"
- }
- local vid = -1
- local aid = -1
- local sid = -1
- for _, track in ipairs(get_active_tracks()) do
- local _exp_0 = track["type"]
- if "video" == _exp_0 then
- vid = track['id']
- elseif "audio" == _exp_0 then
- aid = track['id']
- elseif "sub" == _exp_0 then
- sid = track['id']
- end
- end
- append(command, {
- "--vid=" .. (vid >= 0 and tostring(vid) or "no"),
- "--aid=" .. (aid >= 0 and tostring(aid) or "no"),
- "--sid=" .. (sid >= 0 and tostring(sid) or "no")
- })
- append(command, get_playback_options())
- local filters = { }
- append(filters, format:getPreFilters())
- if options.apply_current_filters then
- apply_current_filters(filters)
- end
- if region and region:is_valid() then
- append(filters, {
- "lavfi-crop=" .. tostring(region.w) .. ":" .. tostring(region.h) .. ":" .. tostring(region.x) .. ":" .. tostring(region.y)
- })
- end
- append(filters, get_scale_filters())
- append(filters, format:getPostFilters())
- for _index_0 = 1, #filters do
- local f = filters[_index_0]
- append(command, {
- "--vf-add=" .. tostring(f)
- })
- end
- append(command, format:getFlags())
- if options.write_filename_on_metadata then
- append(command, get_metadata_flags())
- end
- if options.target_filesize > 0 and format.acceptsBitrate then
- local dT = endTime - startTime
- if options.strict_filesize_constraint then
- local video_kilobits = options.target_filesize * 8
- if aid >= 0 then
- video_kilobits = video_kilobits - dT * options.strict_audio_bitrate
- append(command, {
- "--oacopts-add=b=" .. tostring(options.strict_audio_bitrate) .. "k"
- })
- end
- video_kilobits = video_kilobits * options.strict_bitrate_multiplier
- local bitrate = math.floor(video_kilobits / dT)
- append(command, {
- "--ovcopts-add=b=" .. tostring(bitrate) .. "k",
- "--ovcopts-add=minrate=" .. tostring(bitrate) .. "k",
- "--ovcopts-add=maxrate=" .. tostring(bitrate) .. "k"
- })
- else
- local bitrate = math.floor(options.target_filesize * 8 / dT)
- append(command, {
- "--ovcopts-add=b=" .. tostring(bitrate) .. "k"
- })
- end
- elseif options.target_filesize <= 0 and format.acceptsBitrate then
- append(command, {
- "--ovcopts-add=b=0"
- })
- end
- for token in string.gmatch(options.additional_flags, "[^%s]+") do
- command[#command + 1] = token
- end
- if not options.strict_filesize_constraint then
- for token in string.gmatch(options.non_strict_additional_flags, "[^%s]+") do
- command[#command + 1] = token
- end
- end
- if options.twopass and format.supportsTwopass and not is_stream then
- local first_pass_cmdline
- do
- local _accum_0 = { }
- local _len_0 = 1
- for _index_0 = 1, #command do
- local arg = command[_index_0]
- _accum_0[_len_0] = arg
- _len_0 = _len_0 + 1
- end
- first_pass_cmdline = _accum_0
- end
- append(first_pass_cmdline, {
- "--ovcopts-add=flags=+pass1",
- "-of=" .. tostring(format.outputExtension),
- "-o=" .. tostring(get_null_path())
- })
- message("Starting first pass...")
- msg.verbose("First-pass command line: ", table.concat(first_pass_cmdline, " "))
- local res = run_subprocess({
- args = first_pass_cmdline,
- cancellable = false
- })
- if not res then
- message("First pass failed! Check the logs for details.")
- return
- end
- append(command, {
- "--ovcopts-add=flags=+pass2"
- })
- end
- 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 formatted_filename = format_filename(startTime, endTime, format)
- 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
- return message("Encoded successfully! Saved to\\N" .. tostring(bold(out_path)))
- else
- return message("Encode failed! Check the logs for details.")
- 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: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")
- ass:append(tostring(bold('ENTER:')) .. " confirm crop\\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 scaleHeightOpts = {
- possibleValues = {
- {
- -1,
- "no"
- },
- {
- 240
- },
- {
- 360
- },
- {
- 480
- },
- {
- 720
- },
- {
- 1080
- },
- {
- 1440
- },
- {
- 2160
- }
- }
- }
- local filesizeOpts = {
- step = 250,
- min = 0,
- altDisplayNames = {
- [0] = "0 (constant quality)"
- }
- }
- local formatIds = {
- "webm-vp8",
- "webm-vp9",
- "mp4",
- "raw"
- }
- local formatOpts = {
- possibleValues = (function()
- local _accum_0 = { }
- local _len_0 = 1
- for _index_0 = 1, #formatIds do
- local fId = formatIds[_index_0]
- _accum_0[_len_0] = {
- fId,
- formats[fId].displayName
- }
- _len_0 = _len_0 + 1
- end
- return _accum_0
- end)()
- }
- self.options = {
- {
- "output_format",
- Option("list", "Output Format", options.output_format, formatOpts)
- },
- {
- "twopass",
- Option("bool", "Two Pass", options.twopass)
- },
- {
- "apply_current_filters",
- Option("bool", "Apply Current Video Filters", options.apply_current_filters)
- },
- {
- "scale_height",
- Option("list", "Scale Height", options.scale_height, scaleHeightOpts)
- },
- {
- "strict_filesize_constraint",
- Option("bool", "Strict Filesize Constraint", options.strict_filesize_constraint)
- },
- {
- "write_filename_on_metadata",
- Option("bool", "Write Filename on Metadata", options.write_filename_on_metadata)
- },
- {
- "target_filesize",
- Option("int", "Target Filesize", options.target_filesize, filesizeOpts)
- }
- }
- 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,
- 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('WebM maker')) .. "\\N\\N")
- ass:append(tostring(bold('c:')) .. " crop\\N")
- ass:append(tostring(bold('1:')) .. " set start time (current is " .. tostring(seconds_to_time_string(self.startTime)) .. ")\\N")
- ass:append(tostring(bold('2:')) .. " set end time (current is " .. tostring(seconds_to_time_string(self.endTime)) .. ")\\N")
- ass:append(tostring(bold('o:')) .. " change encode options\\N")
- ass:append(tostring(bold('p:')) .. " preview\\N")
- ass:append(tostring(bold('e:')) .. " encode\\N\\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()
- return 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
- })
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement