X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Faudio_encoder.h;h=49cead7c4851dee82097fb2caed462373283ba6e;hb=a0e57ec99e9dfdd7e79bcf21e8c996dc43b6cb49;hp=3a47f99afb0c3fe2cc792af12152312919017b39;hpb=1501c53153cb0daa846e4de7a73cfbfc797fd543;p=nageru

diff --git a/nageru/audio_encoder.h b/nageru/audio_encoder.h
index 3a47f99..49cead7 100644
--- a/nageru/audio_encoder.h
+++ b/nageru/audio_encoder.h
@@ -3,12 +3,14 @@
 #ifndef _AUDIO_ENCODER_H
 #define _AUDIO_ENCODER_H 1
 
+#include <assert.h>
 #include <stddef.h>
 #include <stdint.h>
 #include <string>
 #include <vector>
 
 extern "C" {
+#include <libavcodec/codec.h>
 #include <libavcodec/avcodec.h>
 #include <libavformat/avformat.h>
 #include <libswresample/swresample.h>
@@ -25,11 +27,13 @@ public:
 	~AudioEncoder();
 
 	void add_mux(Mux *mux) {  // Does not take ownership.
+		assert(mux != nullptr);
 		muxes.push_back(mux);
 	}
 	void encode_audio(const std::vector<float> &audio, int64_t audio_pts);
 	void encode_last_audio();
 
+	const AVCodec *get_codec() const { return ctx->codec; }
 	AVCodecParametersWithDeleter get_codec_parameters();
 
 private: