mirror of
https://github.com/k4yt3x/video2x.git
synced 2024-12-28 06:59:11 +00:00
118 lines
2.3 KiB
C
118 lines
2.3 KiB
C
#ifndef LIBVIDEO2X_H
|
|
#define LIBVIDEO2X_H
|
|
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <time.h>
|
|
|
|
#ifdef _WIN32
|
|
#ifdef LIBVIDEO2X_EXPORTS
|
|
#define LIBVIDEO2X_API __declspec(dllexport)
|
|
#else
|
|
#define LIBVIDEO2X_API __declspec(dllimport)
|
|
#endif
|
|
#else
|
|
#define LIBVIDEO2X_API
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
|
|
// Enum to specify filter type
|
|
enum FilterType {
|
|
FILTER_LIBPLACEBO,
|
|
FILTER_REALESRGAN
|
|
};
|
|
|
|
// Enum to specify log level
|
|
enum Libvideo2xLogLevel {
|
|
LIBVIDEO2X_LOG_LEVEL_TRACE,
|
|
LIBVIDEO2X_LOG_LEVEL_DEBUG,
|
|
LIBVIDEO2X_LOG_LEVEL_INFO,
|
|
LIBVIDEO2X_LOG_LEVEL_WARNING,
|
|
LIBVIDEO2X_LOG_LEVEL_ERROR,
|
|
LIBVIDEO2X_LOG_LEVEL_CRITICAL,
|
|
LIBVIDEO2X_LOG_LEVEL_OFF
|
|
};
|
|
|
|
// Configuration for Libplacebo filter
|
|
struct LibplaceboConfig {
|
|
int out_width;
|
|
int out_height;
|
|
#ifdef _WIN32
|
|
const wchar_t *shader_path;
|
|
#else
|
|
const char *shader_path;
|
|
#endif
|
|
};
|
|
|
|
// Configuration for RealESRGAN filter
|
|
struct RealESRGANConfig {
|
|
int gpuid;
|
|
bool tta_mode;
|
|
int scaling_factor;
|
|
#ifdef _WIN32
|
|
const wchar_t *model_path;
|
|
#else
|
|
const char *model_path;
|
|
#endif
|
|
};
|
|
|
|
// Unified filter configuration
|
|
struct FilterConfig {
|
|
enum FilterType filter_type;
|
|
union {
|
|
struct LibplaceboConfig libplacebo;
|
|
struct RealESRGANConfig realesrgan;
|
|
} config;
|
|
};
|
|
|
|
// Encoder configuration
|
|
struct EncoderConfig {
|
|
int out_width;
|
|
int out_height;
|
|
bool copy_streams;
|
|
enum AVCodecID codec;
|
|
enum AVPixelFormat pix_fmt;
|
|
const char *preset;
|
|
int64_t bit_rate;
|
|
float crf;
|
|
};
|
|
|
|
// Video processing context
|
|
struct VideoProcessingContext {
|
|
int64_t processed_frames;
|
|
int64_t total_frames;
|
|
time_t start_time;
|
|
bool pause;
|
|
bool abort;
|
|
bool completed;
|
|
};
|
|
|
|
// C-compatible process_video function
|
|
LIBVIDEO2X_API int process_video(
|
|
#ifdef _WIN32
|
|
const wchar_t *in_fname,
|
|
const wchar_t *out_fname,
|
|
#else
|
|
const char *in_fname,
|
|
const char *out_fname,
|
|
#endif
|
|
enum Libvideo2xLogLevel log_level,
|
|
bool benchmark,
|
|
enum AVHWDeviceType hw_device_type,
|
|
const struct FilterConfig *filter_config,
|
|
struct EncoderConfig *encoder_config,
|
|
struct VideoProcessingContext *proc_ctx
|
|
);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // LIBVIDEO2X_H
|