X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=audio_encoder.h;h=7bdb02a3657d61675e3ea66354f58ea17e35015a;hb=1062c5403b57859c219558e736564a3d0bbecfd5;hp=c49cf9f761a0e3b0ebf3263242e460d46eafdcdb;hpb=8e3e40723b72bbf0c233167cee60b4959d6b08e7;p=nageru

diff --git a/audio_encoder.h b/audio_encoder.h
index c49cf9f..7bdb02a 100644
--- a/audio_encoder.h
+++ b/audio_encoder.h
@@ -3,6 +3,7 @@
 #ifndef _AUDIO_ENCODER_H
 #define _AUDIO_ENCODER_H 1
 
+#include <memory>
 #include <string>
 #include <vector>
 
@@ -12,17 +13,21 @@ extern "C" {
 #include <libavutil/frame.h>
 }
 
+#include "ffmpeg_raii.h"
 #include "mux.h"
 
 class AudioEncoder {
 public:
-	AudioEncoder(const std::string &codec_name, int bit_rate, const std::vector<Mux *> &muxes);
+	AudioEncoder(const std::string &codec_name, int bit_rate, const AVOutputFormat *oformat);
 	~AudioEncoder();
 
+	void add_mux(Mux *mux) {  // Does not take ownership.
+		muxes.push_back(mux);
+	}
 	void encode_audio(const std::vector<float> &audio, int64_t audio_pts);
 	void encode_last_audio();
 
-	const AVCodec *get_codec() { return ctx->codec; }
+	AVCodecParametersWithDeleter get_codec_parameters();
 
 private:
 	void encode_audio_one_frame(const float *audio, size_t num_samples, int64_t audio_pts);