summaryrefslogtreecommitdiff
path: root/scripts/_vamp_audio_to_midi.lua
blob: e74aac59ebac4118ca5685057ccd9fb4fc2676bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
ardour { ["type"] = "EditorAction", name = "Vamp Audio to MIDI",
description = "analyze audio from selected audio region to selected midi region" }

function factory () return function ()
	local sel = Editor:get_selection ()
	local sr = Session:nominal_frame_rate ()
	local tm = Session:tempo_map ()
	local vamp = ARDOUR.LuaAPI.Vamp ("libardourvampplugins:qm-transcription", sr)
	local ar, mr
	for r in sel.regions:regionlist ():iter () do
		if r:to_midiregion():isnil() then
			ar = r
		else
			mr = r:to_midiregion()
		end
	end
	assert (ar and mr)
	assert (mr:position () >= mr:start ())

	local a_off = ar:position ()
	local m_off = tm:exact_beat_at_frame (mr:position () - mr:start (), 0)

	vamp:analyze (ar:to_readable (), 0, nil)
	local fl = vamp:plugin ():getRemainingFeatures ():at (0)
	if fl and fl:size() > 0 then
		local mm = mr: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_beat_at_frame (a_off + ft, 0)
			local be = tm:exact_beat_at_frame (a_off + ft + fd, 0)

			local pos = Evoral.Beats (bs - m_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
end end