Move encoding code from video encoder to encoder, since it also processes audio input

This commit is contained in:
dec05eba
2025-04-22 00:07:20 +02:00
parent 8b11abd404
commit 1fd30187fa
6 changed files with 223 additions and 189 deletions

44
include/encoder/encoder.h Normal file
View File

@@ -0,0 +1,44 @@
#ifndef GSR_ENCODER_H
#define GSR_ENCODER_H
#include "../replay_buffer.h"
#include <stdbool.h>
#include <stdint.h>
#include <stddef.h>
#include <pthread.h>
#define GSR_MAX_RECORDING_DESTINATIONS 128
typedef struct AVCodecContext AVCodecContext;
typedef struct AVFormatContext AVFormatContext;
typedef struct AVStream AVStream;
typedef struct {
size_t id;
AVCodecContext *codec_context;
AVFormatContext *format_context;
AVStream *stream;
int64_t start_pts;
bool has_received_keyframe;
} gsr_encoder_recording_destination;
typedef struct {
gsr_replay_buffer replay_buffer;
bool has_replay_buffer;
pthread_mutex_t file_write_mutex;
bool mutex_created;
gsr_encoder_recording_destination recording_destinations[GSR_MAX_RECORDING_DESTINATIONS];
size_t num_recording_destinations;
size_t recording_destination_id_counter;
} gsr_encoder;
bool gsr_encoder_init(gsr_encoder *self, size_t replay_buffer_num_packets);
void gsr_encoder_deinit(gsr_encoder *self);
void gsr_encoder_receive_packets(gsr_encoder *self, AVCodecContext *codec_context, int64_t pts, int stream_index);
/* Returns the id to the recording destination, or -1 on error */
size_t gsr_encoder_add_recording_destination(gsr_encoder *self, AVCodecContext *codec_context, AVFormatContext *format_context, AVStream *stream, int64_t start_pts);
bool gsr_encoder_remove_recording_destination(gsr_encoder *self, size_t id);
#endif /* GSR_ENCODER_H */

View File

@@ -2,27 +2,13 @@
#define GSR_ENCODER_VIDEO_H
#include "../../color_conversion.h"
#include "../../replay_buffer.h"
#include <stdbool.h>
#include <stdint.h>
#include <pthread.h>
#define GSR_MAX_RECORDING_DESTINATIONS 128
typedef struct gsr_video_encoder gsr_video_encoder;
typedef struct AVCodecContext AVCodecContext;
typedef struct AVFormatContext AVFormatContext;
typedef struct AVFrame AVFrame;
typedef struct AVStream AVStream;
typedef struct {
size_t id;
AVCodecContext *codec_context;
AVFormatContext *format_context;
AVStream *stream;
int64_t start_pts;
bool has_received_keyframe;
} gsr_video_encoder_recording_destination;
struct gsr_video_encoder {
bool (*start)(gsr_video_encoder *encoder, AVCodecContext *video_codec_context, AVFrame *frame);
@@ -33,24 +19,12 @@ struct gsr_video_encoder {
void *priv;
bool started;
gsr_replay_buffer replay_buffer;
bool has_replay_buffer;
pthread_mutex_t file_write_mutex;
gsr_video_encoder_recording_destination recording_destinations[GSR_MAX_RECORDING_DESTINATIONS];
size_t num_recording_destinations;
size_t recording_destination_id;
};
/* Set |replay_buffer_time_seconds| and |fps| to 0 to disable replay buffer */
bool gsr_video_encoder_start(gsr_video_encoder *encoder, AVCodecContext *video_codec_context, AVFrame *frame, size_t replay_buffer_num_packets);
bool gsr_video_encoder_start(gsr_video_encoder *encoder, AVCodecContext *video_codec_context, AVFrame *frame);
void gsr_video_encoder_destroy(gsr_video_encoder *encoder, AVCodecContext *video_codec_context);
void gsr_video_encoder_copy_textures_to_frame(gsr_video_encoder *encoder, AVFrame *frame, gsr_color_conversion *color_conversion);
void gsr_video_encoder_get_textures(gsr_video_encoder *encoder, unsigned int *textures, int *num_textures, gsr_destination_color *destination_color);
void gsr_video_encoder_receive_packets(gsr_video_encoder *encoder, AVCodecContext *codec_context, int64_t pts, int stream_index);
/* Returns the id to the recording destination, or -1 on error */
size_t gsr_video_encoder_add_recording_destination(gsr_video_encoder *encoder, AVCodecContext *codec_context, AVFormatContext *format_context, AVStream *stream, int64_t start_pts);
bool gsr_video_encoder_remove_recording_destination(gsr_video_encoder *encoder, size_t id);
#endif /* GSR_ENCODER_VIDEO_H */