filter/ffmpeg: implement method Flush()

This commit is contained in:
Max Kellermann 2024-10-30 22:46:07 +01:00
parent 2c881c63ba
commit fa29693550
2 changed files with 18 additions and 0 deletions

View File

@ -52,6 +52,8 @@ FfmpegFilter::ReadOutput()
std::span<const std::byte> std::span<const std::byte>
FfmpegFilter::FilterPCM(std::span<const std::byte> src) FfmpegFilter::FilterPCM(std::span<const std::byte> src)
{ {
assert(!flushed);
/* submit source data into the FFmpeg audio buffer source */ /* submit source data into the FFmpeg audio buffer source */
frame.Unref(); frame.Unref();
@ -78,3 +80,16 @@ FfmpegFilter::FilterPCM(std::span<const std::byte> src)
return ReadOutput(); return ReadOutput();
} }
std::span<const std::byte>
FfmpegFilter::Flush()
{
if (!flushed) {
if (int err = av_buffersrc_add_frame(&buffer_src, nullptr); err < 0)
throw MakeFfmpegError(err, "av_buffersrc_write_frame() failed");
flushed = true;
}
return ReadOutput();
}

View File

@ -30,6 +30,8 @@ class FfmpegFilter final : public Filter {
const size_t in_audio_frame_size; const size_t in_audio_frame_size;
const size_t out_audio_frame_size; const size_t out_audio_frame_size;
bool flushed = false;
public: public:
/** /**
* @param _graph a checked and configured AVFilterGraph * @param _graph a checked and configured AVFilterGraph
@ -46,6 +48,7 @@ public:
/* virtual methods from class Filter */ /* virtual methods from class Filter */
std::span<const std::byte> FilterPCM(std::span<const std::byte> src) override; std::span<const std::byte> FilterPCM(std::span<const std::byte> src) override;
std::span<const std::byte> Flush() override;
private: private:
std::span<const std::byte> ReadOutput(); std::span<const std::byte> ReadOutput();