13
0
livetrax/share/scripts/vamp_audio_to_midi.lua
Robin Gareus 925be49b2d
Remove redundant collectgarbage() calls from scripts
Since 8112d6472e garbage is automatically collected
after running a Lua GUI script.
2020-03-10 23:31:20 +01:00

110 lines
3.7 KiB
Lua

ardour {
["type"] = "EditorAction",
name = "Polyphonic Audio to MIDI",
license = "MIT",
author = "Ardour Team",
description = [[
Analyze audio from the selected audio region to a selected MIDI region.
A MIDI region on the target track will have to be created first (use the pen tool).
This script uses the Polyphonic Transcription VAMP plugin from Queen Mary Univ, London.
The plugin works best at 44.1KHz input sample rate, and is tuned for piano and guitar music. Velocity is not estimated.
]]
}
function factory () return function ()
local sel = Editor:get_selection ()
local sr = Session:nominal_sample_rate ()
local tm = Session:tempo_map ()
local vamp = ARDOUR.LuaAPI.Vamp ("libardourvampplugins:qm-transcription", sr)
local midi_region = nil
local audio_regions = {}
local start_time = Session:current_end_sample ()
local end_time = Session:current_start_sample ()
local max_pos = 0
local cur_pos = 0
for r in sel.regions:regionlist ():iter () do
if r:to_midiregion():isnil() then
local st = r:position()
local ln = r:length()
local et = st + ln
if st < start_time then
start_time = st
end
if et > end_time then
end_time = et
end
table.insert(audio_regions, r)
max_pos = max_pos + r:to_readable ():readable_length ()
else
midi_region = r:to_midiregion()
end
end
if #audio_regions == 0 then
LuaDialog.Message ("Polyphonic Audio to MIDI", "No source audio region(s) selected.\nAt least one audio-region to be analyzed need to be selected.", LuaDialog.MessageType.Error, LuaDialog.ButtonType.Close):run ()
return
end
if not midi_region then
LuaDialog.Message ("Polyphonic Audio to MIDI", "No target MIDI region selected.\nA MIDI region, ideally empty, and extending beyond the selected audio-region(s) needs to be selected.", LuaDialog.MessageType.Error, LuaDialog.ButtonType.Close):run ()
return
end
midi_region:set_initial_position(start_time)
midi_region:set_length(end_time - start_time, 0)
local pdialog = LuaDialog.ProgressWindow ("Audio to MIDI", true)
function progress (_, pos)
return pdialog:progress ((cur_pos + pos) / max_pos, "Analyzing")
end
for i,ar in pairs(audio_regions) do
local a_off = ar:position ()
local b_off = midi_region:quarter_note () - midi_region:start_beats ()
vamp:analyze (ar:to_readable (), 0, progress)
if pdialog:canceled () then
goto out
end
cur_pos = cur_pos + ar:to_readable ():readable_length ()
pdialog:progress (cur_pos / max_pos, "Generating MIDI")
local fl = vamp:plugin ():getRemainingFeatures ():at (0)
if fl and fl:size() > 0 then
local mm = midi_region:midi_source(0):model()
local midi_command = mm:new_note_diff_command ("Audio2Midi")
for f in fl:iter () do
local ft = Vamp.RealTime.realTime2Frame (f.timestamp, sr)
local fd = Vamp.RealTime.realTime2Frame (f.duration, sr)
local fn = f.values:at (0)
local bs = tm:exact_qn_at_sample (a_off + ft, 0)
local be = tm:exact_qn_at_sample (a_off + ft + fd, 0)
local pos = Evoral.Beats (bs - b_off)
local len = Evoral.Beats (be - bs)
local note = ARDOUR.LuaAPI.new_noteptr (1, pos, len, fn + 1, 0x7f)
midi_command:add (note)
end
mm:apply_command (Session, midi_command)
end
-- reset the plugin (prepare for next iteration)
vamp:reset ()
end
::out::
pdialog:done ();
end end
function icon (params) return function (ctx, width, height, fg)
local txt = Cairo.PangoLayout (ctx, "ArdourMono ".. math.ceil(width * .7) .. "px")
txt:set_text ("\u{2669}") -- quarter note symbol UTF8
local tw, th = txt:get_pixel_size ()
ctx:set_source_rgba (ARDOUR.LuaAPI.color_to_rgba (fg))
ctx:move_to (.5 * (width - tw), .5 * (height - th))
txt:show_in_cairo_context (ctx)
end end