videoclip/videoclip.lua
2020-10-23 16:10:29 +03:00

414 lines
11 KiB
Lua
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

local mpopt = require('mp.options')
local utils = require('mp.utils')
-- Options can be changed here or in a separate config file.
-- Config path: ~/.config/mpv/script-opts/videoclip.conf
local config = {
-- absolute paths
-- relative paths (e.g. ~ for home dir) do NOT work.
video_folder_path = string.format('%s/Videos/', os.getenv("HOME") or os.getenv('USERPROFILE')),
audio_folder_path = string.format('%s/Music/', os.getenv("HOME") or os.getenv('USERPROFILE')),
font_size = 24,
audio_bitrate = '32k',
-- The range of the CRF scale is 051, where 0 is lossless,
-- 23 is the default, and 51 is worst quality possible.
-- Insane values like 9999 still work but produce the worst quality.
video_quality = 23,
-- Use the slowest preset that you have patience for.
-- https://trac.ffmpeg.org/wiki/Encode/H.264
preset = 'faster',
video_width = -2,
video_height = 480,
}
mpopt.read_options(config, 'videoclip')
local menu
local encoder
local OSD
local Timings
local allowed_presets = {
ultrafast = true,
superfast = true,
veryfast = true,
faster = true,
fast = true,
medium = true,
slow = true,
slower = true,
veryslow = true,
}
------------------------------------------------------------
-- utility functions
function string:endswith(suffix)
return suffix == "" or self:sub(-#suffix) == suffix
end
local function remove_extension(filename)
return filename:gsub('%.%w+$', '')
end
local function remove_text_in_brackets(str)
return str:gsub('%b[]', '')
end
local function remove_special_characters(str)
return str:gsub('[%c%p%s]', '')
end
local function human_readable_time(seconds)
if type(seconds) ~= 'number' or seconds < 0 then
return 'empty'
end
local parts = {}
parts.h = math.floor(seconds / 3600)
parts.m = math.floor(seconds / 60) % 60
parts.s = math.floor(seconds % 60)
parts.ms = math.floor((seconds * 1000) % 1000)
local ret = string.format("%02dm%02ds%03dms", parts.m, parts.s, parts.ms)
if parts.h > 0 then
ret = string.format('%dh%s', parts.h, ret)
end
return ret
end
local function construct_filename()
local filename = mp.get_property("filename") -- filename without path
filename = remove_extension(filename)
filename = remove_text_in_brackets(filename)
filename = remove_special_characters(filename)
filename = string.format(
'%s_(%s-%s)',
filename,
human_readable_time(menu.timings['start']),
human_readable_time(menu.timings['end'])
)
return filename
end
local function subprocess(args)
return mp.command_native {
name = "subprocess",
playback_only = false,
capture_stdout = true,
capture_stderr = true,
args = args
}
end
------------------------------------------------------------
-- provides interface for creating audio/video clips
encoder = {}
encoder.create_videoclip = function(clip_filename, muted)
local clip_path = utils.join_path(config.video_folder_path, clip_filename .. '.mp4')
return subprocess {
'mpv',
mp.get_property('path'),
'--loop-file=no',
'--no-ocopy-metadata',
'--no-sub',
'--audio-channels=2',
'--oac=libopus',
'--ovc=libx264',
'--oacopts-add=vbr=on',
'--oacopts-add=application=voip',
'--oacopts-add=compression_level=10',
table.concat { '--start=', menu.timings['start'] },
table.concat { '--end=', menu.timings['end'] },
table.concat { '--aid=', muted and 'no' or mp.get_property("aid") }, -- track number
table.concat { '--volume=', mp.get_property('volume') },
table.concat { '--oacopts-add=b=', config.audio_bitrate },
table.concat { '--ovcopts-add=crf=', config.video_quality },
table.concat { '--ovcopts-add=preset=', config.preset },
table.concat { '--vf-add=scale=', config.video_width, ':', config.video_height },
table.concat { '-o=', clip_path }
}
end
encoder.create_audioclip = function(clip_filename)
local clip_path = utils.join_path(config.audio_folder_path, clip_filename .. '.ogg')
return subprocess {
'mpv',
mp.get_property('path'),
'--loop-file=no',
'--no-ocopy-metadata',
'--no-sub',
'--audio-channels=2',
'--video=no',
'--oac=libopus',
'--oacopts-add=vbr=on',
'--oacopts-add=application=voip',
'--oacopts-add=compression_level=10',
table.concat { '--start=', menu.timings['start'] },
table.concat { '--end=', menu.timings['end'] },
table.concat { '--volume=', mp.get_property('volume') },
table.concat { '--aid=', mp.get_property("aid") }, -- track number
table.concat { '--oacopts-add=b=', config.audio_bitrate },
table.concat { '-o=', clip_path }
}
end
encoder.create_clip = function(clip_type)
menu.close();
if clip_type == nil then
return
end
if not menu.timings:validate() then
mp.osd_message("Wrong timings. Aborting.", 2)
return
end
local clip_filename = construct_filename()
mp.osd_message("Please wait...", 9999)
local ret
local location
if clip_type:match('video') ~= nil then
location = config.video_folder_path
local muted = clip_type:match('mute') ~= nil
ret = encoder.create_videoclip(clip_filename, muted)
else
location = config.audio_folder_path
ret = encoder.create_audioclip(clip_filename)
end
if ret.status ~= 0 or string.match(ret.stdout, "could not open") then
mp.osd_message(string.format("Error: couldn't create the clip.\nDoes %s exist?", location), 5)
else
mp.osd_message(string.format("Clip saved to %s.", location), 2)
end
menu.timings:reset()
end
------------------------------------------------------------
-- Menu interface
local Menu = {}
Menu.__index = Menu
function Menu:new(parent)
local o = {
parent = parent,
overlay = parent and parent.overlay or mp.create_osd_overlay('ass-events'),
keybinds = { },
}
return setmetatable(o, self)
end
function Menu:overlay_draw(text)
self.overlay.data = text
self.overlay:update()
end
function Menu:open()
if self.parent then
self.parent:close()
end
for _, val in pairs(self.keybinds) do
mp.add_key_binding(val.key, val.key, val.fn)
end
self:update()
end
function Menu:close()
for _, val in pairs(self.keybinds) do
mp.remove_key_binding(val.key)
end
if self.parent then
self.parent:open()
else
self.overlay:remove()
end
end
function Menu:update()
local osd = OSD:new():size(config.font_size):align(4)
osd:append('Dummy menu.'):newline()
self:overlay_draw(osd:get_text())
end
------------------------------------------------------------
-- main menu
menu = {}
menu.overlay = mp.create_osd_overlay('ass-events')
menu.overlay_draw = function(text)
menu.overlay.data = text
menu.overlay:update()
end
menu.keybinds = {
{ key = 's', fn = function() menu.set_time('start') end },
{ key = 'e', fn = function() menu.set_time('end') end },
{ key = 'S', fn = function() menu.set_time_sub('start') end },
{ key = 'E', fn = function() menu.set_time_sub('end') end },
{ key = 'c', fn = function() encoder.create_clip('video') end },
{ key = 'm', fn = function() encoder.create_clip('video_muted') end },
{ key = 'a', fn = function() encoder.create_clip('audio') end },
{ key = 'o', fn = function() mp.commandv('run', 'xdg-open', 'https://streamable.com/') end },
{ key = 'ESC', fn = function() menu.close() end },
}
menu.set_time = function(property)
menu.timings[property] = mp.get_property_number('time-pos')
menu.update()
end
menu.set_time_sub = function(property)
local sub_delay = mp.get_property_native("sub-delay")
local time_pos = mp.get_property_number(string.format("sub-%s", property))
if time_pos == nil then
mp.osd_message("Warning: No subtitles visible.", 2)
return
end
menu.timings[property] = time_pos + sub_delay
menu.update()
end
menu.update = function()
local osd = OSD:new():size(config.font_size):align(4)
osd:bold('Clip creator'):newline():newline()
osd:bold('Start time: '):append(human_readable_time(menu.timings['start'])):newline()
osd:bold('End time: '):append(human_readable_time(menu.timings['end'])):newline()
osd:newline()
osd:bold('Bindings:'):newline()
osd:tab():bold('s: '):append('Set start time'):newline()
osd:tab():bold('e: '):append('Set end time'):newline()
osd:newline()
osd:tab():bold('S: '):append('Set start time based on subtitles'):newline()
osd:tab():bold('E: '):append('Set end time based on subtitles'):newline()
osd:newline()
osd:tab():bold('o: '):append('Open `streamable.com`'):newline()
osd:tab():bold('ESC: '):append('Close'):newline()
osd:newline()
osd:bold('Create clip:'):newline()
osd:tab():bold('c: '):append('video clip'):newline()
osd:tab():bold('m: '):append('video clip (silent)'):newline()
osd:tab():bold('a: '):append('audio clip'):newline()
menu.overlay_draw(osd:get_text())
end
menu.close = function()
for _, val in pairs(menu.keybinds) do
mp.remove_key_binding(val.key)
end
menu.overlay:remove()
end
menu.open = function()
menu.timings = Timings:new()
for _, val in pairs(menu.keybinds) do
mp.add_key_binding(val.key, val.key, val.fn)
end
menu.update()
end
------------------------------------------------------------
-- Helper class for styling OSD messages
-- http://docs.aegisub.org/3.2/ASS_Tags/
OSD = {}
OSD.__index = OSD
function OSD:new()
return setmetatable({ text = {} }, self)
end
function OSD:append(s)
table.insert(self.text, s)
return self
end
function OSD:bold(s)
return self:append(string.format([[{\b1}%s{\b0}]], s))
end
function OSD:italics(s)
return self:append('{\\i1}'):append(s):append('{\\i0}')
end
function OSD:newline()
return self:append([[\N]])
end
function OSD:tab()
return self:append([[\h\h\h\h]])
end
function OSD:size(size)
return self:append(string.format([[{\fs%s}]], size))
end
function OSD:align(number)
return self:append(string.format([[{\an%s}]], number))
end
function OSD:get_text()
return table.concat(self.text)
end
------------------------------------------------------------
-- Timings class
Timings = {
['start'] = -1,
['end'] = -1,
}
function Timings:new(o)
o = o or {}
setmetatable(o, self)
self.__index = self
return o
end
function Timings:reset()
self['start'] = -1
self['end'] = -1
end
function Timings:validate()
return self['start'] >= 0 and self['start'] < self['end']
end
------------------------------------------------------------
-- Validate config
if not config.audio_bitrate:endswith('k') then
config.audio_bitrate = config.audio_bitrate .. 'k'
end
if not allowed_presets[config.preset] then
config.preset = 'faster'
end
------------------------------------------------------------
-- Finally, set an 'entry point' in mpv
mp.add_key_binding('c', 'videoclip-menu-open', menu.open)