#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