initial commitHEADmaster

1 files changed, 158 insertions, 0 deletions

diff --git a/mpv/scripts/autodeint.lua b/mpv/scripts/autodeint.lua
new file mode 100644
index 0000000..16f56ea
--- /dev/null
+++ b/mpv/scripts/autodeint.lua
@@ -0,0 +1,158 @@
+-- From: https://github.com/mpv-player/mpv/tree/master/TOOLS/lua
+-- (with slight modifications)
+--
+-- This script uses the lavfi idet filter to automatically insert the
+-- appropriate deinterlacing filter based on a short section of the
+-- currently playing video.
+--
+-- It registers the key-binding ctrl+d, which when pressed, inserts the filters
+-- ``vf=lavfi=idet,pullup,vf=lavfi=idet``. After 4 seconds, it removes these
+-- filters and decides whether the content is progressive, interlaced, or
+-- telecined and the interlacing field dominance.
+--
+-- Based on this information, it may set mpv's ``deinterlace`` property (which
+-- usually inserts the yadif filter), or insert the ``pullup`` filter if the
+-- content is telecined.  It also sets mpv's ``field-dominance`` property.
+--
+-- OPTIONS:
+-- The default detection time may be overridden by adding
+--
+-- --script-opts=autodeint.detect_seconds=<number of seconds>
+--
+-- to mpv's arguments. This may be desirable to allow idet more
+-- time to collect data.
+--
+-- To see counts of the various types of frames for each detection phase,
+-- the verbosity can be increased with
+--
+-- --msg-level autodeint=v
+--
+-- This script requires a recent version of ffmpeg for which the idet
+-- filter adds the required metadata.
+
+require "mp.msg"
+
+script_name = mp.get_script_name()
+detect_label = string.format("%s-detect", script_name)
+pullup_label = string.format("%s", script_name)
+ivtc_detect_label = string.format("%s-ivtc-detect", script_name)
+
+-- number of seconds to gather cropdetect data
+detect_seconds = tonumber(mp.get_opt(string.format("%s.detect_seconds", script_name)))
+if not detect_seconds then
+    detect_seconds = 4
+end
+
+function del_filter_if_present(label)
+    -- necessary because mp.command('vf del @label:filter') raises an
+    -- error if the filter doesn't exist
+    local vfs = mp.get_property_native("vf")
+
+    for i,vf in pairs(vfs) do
+        if vf["label"] == label then
+            table.remove(vfs, i)
+            mp.set_property_native("vf", vfs)
+            return true
+        end
+    end
+    return false
+end
+
+function start_detect()
+    -- exit if detection is already in progress
+    if timer then
+        mp.msg.warn("already detecting!")
+        mp.osd_message("autodeint: already detecting!")
+        return
+    end
+
+    mp.set_property("deinterlace","no")
+    del_filter_if_present(pullup_label)
+
+    -- insert the detection filter
+    local cmd = string.format('vf add @%s:lavfi=graph="idet",@%s:pullup,@%s:lavfi=graph="idet"',
+                              detect_label, pullup_label, ivtc_detect_label)
+    if not mp.command(cmd) then
+        mp.msg.error("failed to insert detection filters")
+        mp.osd_message("autodeint: failed to insert detection filters")
+        return
+    end
+
+    -- wait to gather data
+    mp.osd_message("autodeint: starting detection")
+    timer = mp.add_timeout(detect_seconds, select_filter)
+end
+
+function stop_detect()
+    del_filter_if_present(detect_label)
+    del_filter_if_present(ivtc_detect_label)
+    timer = nil
+end
+
+progressive, interlaced_tff, interlaced_bff, interlaced = 0, 1, 2, 3, 4
+
+function judge(label)
+    -- get the metadata
+    local result = mp.get_property_native(string.format("vf-metadata/%s", label))
+    -- filter might have been removed by third party
+    if not result or next(result) == nil then
+        return progressive
+    end
+    num_tff          = tonumber(result["lavfi.idet.multiple.tff"])
+    num_bff          = tonumber(result["lavfi.idet.multiple.bff"])
+    num_progressive  = tonumber(result["lavfi.idet.multiple.progressive"])
+    num_undetermined = tonumber(result["lavfi.idet.multiple.undetermined"])
+    num_interlaced   = num_tff + num_bff
+    num_determined   = num_interlaced + num_progressive
+
+    mp.msg.verbose(label .. " progressive    = "..num_progressive)
+    mp.msg.verbose(label .. " interlaced-tff = "..num_tff)
+    mp.msg.verbose(label .. " interlaced-bff = "..num_bff)
+    mp.msg.verbose(label .. " undetermined   = "..num_undetermined)
+
+    if num_determined < num_undetermined then
+        mp.msg.warn("majority undetermined frames")
+    end
+    if num_progressive > 20*num_interlaced then
+        return progressive
+    elseif num_tff > 10*num_bff then
+        return interlaced_tff
+    elseif num_bff > 10*num_tff then
+        return interlaced_bff
+    else
+        return interlaced
+    end
+end
+
+function select_filter()
+    -- handle the first detection filter results
+    verdict = judge(detect_label)
+    if verdict == progressive then
+        mp.msg.info("progressive: doing nothing")
+        mp.osd_message("autodeint: no deinterlacing required")
+        stop_detect()
+        return
+    elseif verdict == interlaced_tff then
+        mp.set_property("field-dominance", "top")
+    elseif verdict == interlaced_bff then
+        mp.set_property("field-dominance", "bottom")
+    elseif verdict == interlaced then
+        mp.set_property("field-dominance", "auto")
+    end
+
+    -- handle the ivtc detection filter results
+    verdict = judge(ivtc_detect_label)
+    if verdict == progressive then
+        mp.msg.info(string.format("telecinied with %s field dominance: using pullup", mp.get_property("field-dominance")))
+        mp.osd_message("autodeint: using pullup")
+        stop_detect()
+    else
+        mp.msg.info(string.format("interlaced with %s field dominance: setting deinterlace property", mp.get_property("field-dominance")))
+        del_filter_if_present(pullup_label)
+        mp.osd_message(string.format("autodeint: setting deinterlace property (%s)", mp.get_property("field-dominance")))
+        mp.set_property("deinterlace", "yes")
+        stop_detect()
+    end
+end
+
+mp.add_key_binding("ctrl+d", script_name, start_detect)