mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-08-20 11:34:55 +08:00
Jimmy
a4e4a8afef
* encode/store audio as part of video file * better match write_audio() with write() * handle different FFmpeg versions, flush audio encoder, suppress encoder QAvg/info messages * use audio_buffer.size() instead of keeping track of size separately * no more for loops * save to qcam and rlog * assert audio support check * microphone --> soundPressure, audioData --> rawAudioData * deque much more efficient if buffer ever >> frame_size, ~ same performance for defaults * cleanup and fix time scaling * initialize audio separately and pass sample_rate in * update comments * ensure header is written before writing audio * buffer audio frame but do not process before header written * handle buffer overflow now that we are using as an actual buffer * spelling
42 lines
1.0 KiB
C++
42 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <deque>
|
|
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
#include <libavcodec/avcodec.h>
|
|
}
|
|
|
|
#include "cereal/messaging/messaging.h"
|
|
|
|
class VideoWriter {
|
|
public:
|
|
VideoWriter(const char *path, const char *filename, bool remuxing, int width, int height, int fps, cereal::EncodeIndex::Type codec);
|
|
void write(uint8_t *data, int len, long long timestamp, bool codecconfig, bool keyframe);
|
|
void write_audio(uint8_t *data, int len, long long timestamp, int sample_rate);
|
|
|
|
~VideoWriter();
|
|
|
|
private:
|
|
void initialize_audio(int sample_rate);
|
|
void encode_and_write_audio_frame(AVFrame* frame);
|
|
|
|
std::string vid_path, lock_path;
|
|
FILE *of = nullptr;
|
|
|
|
AVCodecContext *codec_ctx;
|
|
AVFormatContext *ofmt_ctx;
|
|
AVStream *out_stream;
|
|
|
|
bool audio_initialized = false;
|
|
bool header_written = false;
|
|
AVStream *audio_stream = nullptr;
|
|
AVCodecContext *audio_codec_ctx = nullptr;
|
|
AVFrame *audio_frame = nullptr;
|
|
uint64_t audio_pts = 0;
|
|
std::deque<float> audio_buffer;
|
|
|
|
bool remuxing;
|
|
};
|