github-actions[bot]
f8e13417f2
date: 2026-08-15T14:20:10 master commit: d48cfa730bf21593b684efeaf3fd8940695e1dea
44 lines
1.1 KiB
C++
44 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <deque>
|
|
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
#include <libavcodec/avcodec.h>
|
|
}
|
|
|
|
#include "openpilot/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 set_metadata(const char *key, const char *value);
|
|
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);
|
|
void process_remaining_audio();
|
|
|
|
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;
|
|
};
|