mirror of
https://github.com/k4yt3x/video2x.git
synced 2024-12-28 06:59:11 +00:00
35 lines
736 B
C++
35 lines
736 B
C++
#pragma once
|
|
|
|
#include <filesystem>
|
|
|
|
extern "C" {
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
}
|
|
|
|
namespace video2x {
|
|
namespace decoder {
|
|
|
|
class Decoder {
|
|
public:
|
|
Decoder();
|
|
~Decoder();
|
|
|
|
int init(AVHWDeviceType hw_type, AVBufferRef *hw_ctx, const std::filesystem::path &in_fpath);
|
|
|
|
AVFormatContext *get_format_context() const;
|
|
AVCodecContext *get_codec_context() const;
|
|
int get_video_stream_index() const;
|
|
|
|
private:
|
|
static AVPixelFormat hw_pix_fmt_;
|
|
static AVPixelFormat get_hw_format(AVCodecContext *ctx, const AVPixelFormat *pix_fmts);
|
|
|
|
AVFormatContext *fmt_ctx_;
|
|
AVCodecContext *dec_ctx_;
|
|
int in_vstream_idx_;
|
|
};
|
|
|
|
} // namespace decoder
|
|
} // namespace video2x
|