Move everything into a separate futatabi/ subdir, for the upcoming merge with Futatabi.

[nageru] / nageru / audio_encoder.h

diff --git a/nageru/audio_encoder.h b/nageru/audio_encoder.h
new file mode 100644 (file)
index 0000000..93adbaf
--- /dev/null
+++ b/nageru/audio_encoder.h
@@ -0,0 +1,47 @@
+// A class to encode audio (using ffmpeg) and send it to a Mux.
+
+#ifndef _AUDIO_ENCODER_H
+#define _AUDIO_ENCODER_H 1
+
+#include <stddef.h>
+#include <stdint.h>
+#include <string>
+#include <vector>
+
+extern "C" {
+#include <libavcodec/avcodec.h>
+#include <libavformat/avformat.h>
+#include <libavresample/avresample.h>
+#include <libavutil/frame.h>
+}
+
+#include "ffmpeg_raii.h"
+
+class Mux;
+
+class AudioEncoder {
+public:
+       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();
+
+       AVCodecParametersWithDeleter get_codec_parameters();
+
+private:
+       void encode_audio_one_frame(const float *audio, size_t num_samples, int64_t audio_pts);
+
+       std::vector<float> audio_queue;
+       int64_t last_pts = 0;  // The first pts after all audio we've encoded.
+
+       AVCodecContext *ctx;
+       AVAudioResampleContext *resampler;
+       AVFrame *audio_frame = nullptr;
+       std::vector<Mux *> muxes;
+};
+
+#endif  // !defined(_AUDIO_ENCODER_H)