forked from hkc/cc-stuff
Added video+audio player
This commit is contained in:
parent
a00dca3be1
commit
f6b91af2ee
GPG Key ID:
F0F6CFE11CDB0960
|
|
@ -0,0 +1,222 @@
|
|||
local args = { ... }
|
||||
local dfpwm = require("cc.audio.dfpwm")
|
||||
local ccpi = require("ccpi")
|
||||
|
||||
local monitor = peripheral.find("monitor")
|
||||
local speakers = {
|
||||
l = peripheral.find("speaker"),
|
||||
r = nil
|
||||
}
|
||||
|
||||
local delay = 0
|
||||
local loading_concurrency = 8
|
||||
local buffer_size = 8192
|
||||
|
||||
while args[1] ~= nil and string.sub(args[1], 1, 1) == "-" do
|
||||
local k = table.remove(args, 1):sub(2)
|
||||
if k == "m" or k == "monitor" then
|
||||
monitor = peripheral.wrap(table.remove(args, 1))
|
||||
elseif k == "l" or k == "speaker-left" then
|
||||
speakers.l = peripheral.wrap(table.remove(args, 1))
|
||||
elseif k == "r" or k == "speaker-right" then
|
||||
speakers.l = peripheral.wrap(table.remove(args, 1))
|
||||
elseif k == "d" or k == "delay" then
|
||||
delay = tonumber(table.remove(args, 1))
|
||||
elseif k == "t" or k == "threads" then
|
||||
loading_concurrency = tonumber(table.remove(args, 1))
|
||||
elseif k == "b" or k == "buffer-size" then
|
||||
buffer_size = tonumber(table.remove(args, 1))
|
||||
end
|
||||
end
|
||||
|
||||
if not monitor then
|
||||
printError("No monitor connected or invalid one specified")
|
||||
return
|
||||
end
|
||||
|
||||
if not speakers.l then
|
||||
printError("No speaker connected or invalid one specified")
|
||||
return
|
||||
end
|
||||
|
||||
if #args < 3 then
|
||||
printError("Usage: video [-m MONITOR] [-l SPK_L] [-r SPK_R] [-d FRAME_T] <N_FRAMES> <VIDEO_TEMPLATE> <LEFT_CHANNEL> [RIGHT_CHANNEL]")
|
||||
return
|
||||
end
|
||||
|
||||
print(string.format("Using monitor %s", peripheral.getName(monitor)))
|
||||
if speakers.r then
|
||||
print("Stereo sound: L=%s R=%s", peripheral.getName(speakers.l), peripheral.getName(speakers.r))
|
||||
else
|
||||
print("Mono sound: " .. peripheral.getName(speakers.l))
|
||||
end
|
||||
|
||||
local n_frames = tonumber(table.remove(args, 1))
|
||||
local video_url = table.remove(args, 1)
|
||||
local audio_url_l = table.remove(args, 1)
|
||||
local audio_url_r = #args > 0 and table.remove(args, 1) or nil
|
||||
|
||||
if not speakers.r and audio_url_r then
|
||||
printError("No right speaker found but right audio channel was specified")
|
||||
printError("Right channel will not be played")
|
||||
elseif speakers.r and not audio_url_r then
|
||||
printError("No URL for right channel was specified but right speaker is set")
|
||||
printError("Right speaker will remain silent")
|
||||
end
|
||||
|
||||
print("\n\n")
|
||||
local _, ty = term.getCursorPos()
|
||||
local tw, _ = term.getSize()
|
||||
|
||||
local function draw_bar(y, c1, c2, p, fmt, ...)
|
||||
local str = string.format(fmt, ...)
|
||||
local w1 = math.ceil(p * tw)
|
||||
local w2 = tw - w1
|
||||
|
||||
term.setCursorPos(1, y)
|
||||
term.setBackgroundColor(c1)
|
||||
term.write(str:sub(1, w1))
|
||||
local rem = w1 - #str
|
||||
if rem > 0 then
|
||||
term.write(string.rep(" ", rem))
|
||||
end
|
||||
|
||||
term.setBackgroundColor(c2)
|
||||
term.write(str:sub(w1 + 1, w1 + w2))
|
||||
|
||||
rem = math.min(tw - #str, w2)
|
||||
if rem > 0 then
|
||||
term.write(string.rep(" ", rem))
|
||||
end
|
||||
end
|
||||
|
||||
local function decode_s8(data)
|
||||
local buffer = {}
|
||||
for i = 1, #data do
|
||||
local v = string.byte(data, i)
|
||||
if bit32.band(v, 0x80) then
|
||||
v = bit32.bxor(v, 0x7F) - 128
|
||||
end
|
||||
buffer[i] = v
|
||||
end
|
||||
return buffer
|
||||
end
|
||||
|
||||
local frames = {}
|
||||
local audio_frames = { l = {}, r = {} }
|
||||
|
||||
local subthreads = {}
|
||||
local dl_channels = {
|
||||
l = assert(http.get(audio_url_l, nil, true)),
|
||||
r = audio_url_r and assert(http.get(audio_url_r, nil, true)) or nil,
|
||||
}
|
||||
|
||||
local n_audio_samples = math.ceil(dl_channels.l.seek("end") / buffer_size)
|
||||
dl_channels.l.seek("set", 0)
|
||||
|
||||
table.insert(subthreads, function()
|
||||
local chunk
|
||||
repeat
|
||||
chunk = dl_channels.l.read(buffer_size)
|
||||
table.insert(audio_frames.l, chunk or {})
|
||||
if (#audio_frames.l % 8) == 0 then os.sleep(0) end
|
||||
until not chunk or #chunk == 0
|
||||
end)
|
||||
|
||||
if dl_channels.r then
|
||||
table.insert(subthreads, function()
|
||||
local chunk
|
||||
repeat
|
||||
chunk = dl_channels.r.read(buffer_size)
|
||||
table.insert(audio_frames.r, chunk or {})
|
||||
if (#audio_frames.r % 8) == 0 then os.sleep(0) end
|
||||
until not chunk or #chunk == 0
|
||||
end)
|
||||
end
|
||||
|
||||
for i = 1, loading_concurrency do
|
||||
table.insert(subthreads, function()
|
||||
for ndx = i, n_frames, loading_concurrency do
|
||||
local req = assert(http.get(string.format(video_url, ndx), nil, true))
|
||||
local img = assert(ccpi.parse(req))
|
||||
frames[ndx] = img
|
||||
req.close()
|
||||
end
|
||||
end)
|
||||
end
|
||||
|
||||
table.insert(subthreads, function()
|
||||
while #frames ~= n_frames or #audio_frames.l < n_audio_samples do
|
||||
draw_bar(ty - 3, colors.blue, colors.gray, #frames / n_frames, "Loading video [%5d / %5d]", #frames, n_frames)
|
||||
draw_bar(ty - 2, colors.red, colors.gray, #audio_frames.l / n_audio_samples, "Loading audio [%5d / %5d]", #audio_frames.l, n_audio_samples)
|
||||
os.sleep(0.25)
|
||||
end
|
||||
print()
|
||||
end)
|
||||
|
||||
local playback_locked = true
|
||||
local playback_done = false
|
||||
|
||||
table.insert(subthreads, function()
|
||||
while #frames < 60 or #audio_frames.l < n_audio_samples do
|
||||
term.setCursorPos(1, ty - 1)
|
||||
term.setBackgroundColor(colors.gray)
|
||||
term.clearLine()
|
||||
term.write(string.format("Waiting for frames... (V:%d, A:%d)", #frames, #audio_frames.l))
|
||||
os.sleep(0.25)
|
||||
end
|
||||
playback_locked = false
|
||||
end)
|
||||
|
||||
table.insert(subthreads, function()
|
||||
while playback_locked do os.sleep(0) end -- spin
|
||||
end)
|
||||
|
||||
table.insert(subthreads, function()
|
||||
local is_dfpwm = ({ audio_url_l:find("%.dfpwm") })[2] == #audio_url_l
|
||||
local decode = use_dfpwm and dfpwm.make_decoder() or decode_s8
|
||||
while playback_locked do os.sleep(0) end -- spin
|
||||
|
||||
for i = 1, n_audio_samples do
|
||||
local buffer = decode(audio_frames.l[i])
|
||||
while not speakers.l.playAudio(buffer) do
|
||||
os.pullEvent("speaker_audio_empty")
|
||||
end
|
||||
end
|
||||
|
||||
playback_done = true
|
||||
end)
|
||||
|
||||
table.insert(subthreads, function()
|
||||
if not audio_url_r then return end
|
||||
local is_dfpwm = ({ audio_url_r:find("%.dfpwm") })[2] == #audio_url_r
|
||||
local decode = use_dfpwm and dfpwm.make_decoder() or decode_s8
|
||||
|
||||
while playback_locked do os.sleep(0) end -- spin
|
||||
|
||||
for i = 1, n_audio_samples do
|
||||
local buffer = decode(audio_frames.r[i])
|
||||
while not speakers.r.playAudio(buffer) do
|
||||
os.pullEvent("speaker_audio_empty")
|
||||
end
|
||||
end
|
||||
playback_done = true
|
||||
end)
|
||||
|
||||
table.insert(subthreads, function()
|
||||
while playback_locked do os.sleep(0) end -- spin
|
||||
|
||||
local start_t = os.clock()
|
||||
while not playback_done do
|
||||
local frame = math.floor((os.clock() - start_t) / math.max(0.05, delay))
|
||||
term.setCursorPos(1, ty - 1)
|
||||
term.setBackgroundColor(colors.gray)
|
||||
term.clearLine()
|
||||
term.write(string.format("Playing frame: %d/%d", frame + 1, #frames))
|
||||
ccpi.draw(frames[frame + 1], 1, 1, monitor)
|
||||
os.sleep(delay)
|
||||
end
|
||||
end)
|
||||
|
||||
parallel.waitForAll(table.unpack(subthreads))
|
||||
|
||||
Loading…
Reference in New Issue