mirror of
https://github.com/k4yt3x/video2x.git
synced 2024-12-28 06:59:11 +00:00
627f3d84a4
* feat: add RIFE files and processor/interpolator abstractions * feat: add `rife` as processor option * feat: add frame interpolation math except first frame * feat: complete motion interpolation and add scene detection * feat: improve Vulkan device validation * fix: fix casting issues and variable names * refactor: improve error-checking; add abstractions and factories * refactor: improve readability of the frames processor * docs: update changelog Signed-off-by: k4yt3x <i@k4yt3x.com>
45 lines
937 B
C++
45 lines
937 B
C++
#ifndef ENCODER_H
|
|
#define ENCODER_H
|
|
|
|
#include <cstdint>
|
|
#include <filesystem>
|
|
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
#include <libavutil/pixdesc.h>
|
|
}
|
|
|
|
#include "libvideo2x/libvideo2x.h"
|
|
|
|
class Encoder {
|
|
public:
|
|
Encoder();
|
|
~Encoder();
|
|
|
|
int init(
|
|
AVBufferRef *hw_ctx,
|
|
const std::filesystem::path &out_fpath,
|
|
AVFormatContext *ifmt_ctx,
|
|
AVCodecContext *dec_ctx,
|
|
EncoderConfig *encoder_config,
|
|
const ProcessorConfig *processor_config,
|
|
int in_vstream_idx
|
|
);
|
|
|
|
int write_frame(AVFrame *frame, int64_t frame_idx);
|
|
int flush();
|
|
|
|
AVCodecContext *get_encoder_context() const;
|
|
AVFormatContext *get_format_context() const;
|
|
int *get_stream_map() const;
|
|
int get_output_video_stream_index() const;
|
|
|
|
private:
|
|
AVFormatContext *ofmt_ctx_;
|
|
AVCodecContext *enc_ctx_;
|
|
int out_vstream_idx_;
|
|
int *stream_map_;
|
|
};
|
|
|
|
#endif // ENCODER_H
|