mirror of
https://repo.dec05eba.com/gpu-screen-recorder
synced 2026-03-31 09:07:13 +09:00
62 lines
2.4 KiB
C
62 lines
2.4 KiB
C
#ifndef GSR_CAPTURE_CAPTURE_H
|
|
#define GSR_CAPTURE_CAPTURE_H
|
|
|
|
#include "../color_conversion.h"
|
|
#include <stdbool.h>
|
|
|
|
typedef struct AVCodecContext AVCodecContext;
|
|
typedef struct AVFrame AVFrame;
|
|
typedef void* VADisplay;
|
|
typedef struct _VADRMPRIMESurfaceDescriptor VADRMPRIMESurfaceDescriptor;
|
|
typedef struct gsr_cuda gsr_cuda;
|
|
typedef struct AVFrame AVFrame;
|
|
typedef struct CUgraphicsResource_st *CUgraphicsResource;
|
|
typedef struct CUarray_st *CUarray;
|
|
|
|
typedef struct gsr_capture gsr_capture;
|
|
|
|
struct gsr_capture {
|
|
/* These methods should not be called manually. Call gsr_capture_* instead */
|
|
int (*start)(gsr_capture *cap, AVCodecContext *video_codec_context, AVFrame *frame);
|
|
void (*tick)(gsr_capture *cap, AVCodecContext *video_codec_context); /* can be NULL */
|
|
bool (*should_stop)(gsr_capture *cap, bool *err); /* can be NULL */
|
|
int (*capture)(gsr_capture *cap, AVFrame *frame);
|
|
void (*capture_end)(gsr_capture *cap, AVFrame *frame); /* can be NULL */
|
|
void (*destroy)(gsr_capture *cap, AVCodecContext *video_codec_context);
|
|
|
|
void *priv; /* can be NULL */
|
|
bool started;
|
|
};
|
|
|
|
typedef struct gsr_capture_base gsr_capture_base;
|
|
|
|
struct gsr_capture_base {
|
|
unsigned int input_texture;
|
|
unsigned int target_textures[2];
|
|
unsigned int cursor_texture;
|
|
|
|
gsr_color_conversion color_conversion;
|
|
|
|
AVCodecContext *video_codec_context;
|
|
};
|
|
|
|
typedef struct {
|
|
gsr_cuda *cuda;
|
|
CUgraphicsResource *cuda_graphics_resources;
|
|
CUarray *mapped_arrays;
|
|
} gsr_cuda_context;
|
|
|
|
int gsr_capture_start(gsr_capture *cap, AVCodecContext *video_codec_context, AVFrame *frame);
|
|
void gsr_capture_tick(gsr_capture *cap, AVCodecContext *video_codec_context);
|
|
bool gsr_capture_should_stop(gsr_capture *cap, bool *err);
|
|
int gsr_capture_capture(gsr_capture *cap, AVFrame *frame);
|
|
void gsr_capture_end(gsr_capture *cap, AVFrame *frame);
|
|
/* Calls |gsr_capture_stop| as well */
|
|
void gsr_capture_destroy(gsr_capture *cap, AVCodecContext *video_codec_context);
|
|
|
|
bool gsr_capture_base_setup_vaapi_textures(gsr_capture_base *self, AVFrame *frame, gsr_egl *egl, VADisplay va_dpy, VADRMPRIMESurfaceDescriptor *prime, gsr_color_range color_range);
|
|
bool gsr_capture_base_setup_cuda_textures(gsr_capture_base *base, AVFrame *frame, gsr_cuda_context *cuda_context, gsr_egl *egl, gsr_color_range color_range, gsr_source_color source_color, bool hdr);
|
|
void gsr_capture_base_stop(gsr_capture_base *self, gsr_egl *egl);
|
|
|
|
#endif /* GSR_CAPTURE_CAPTURE_H */
|