video2x/include/libvideo2x/decoder.h

#ifndef DECODER_H
#define DECODER_H

#include <filesystem>

extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}

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 enum AVPixelFormat hw_pix_fmt_;
    static enum AVPixelFormat
    get_hw_format(AVCodecContext *ctx, const enum AVPixelFormat *pix_fmts);

    AVFormatContext *fmt_ctx_;
    AVCodecContext *dec_ctx_;
    int in_vstream_idx_;
};

#endif  // DECODER_H