Rescale.hpp

#pragma once
 
#include <Media/Libav.hpp>
#if SCORE_HAS_LIBAV
 
#include <Video/DecoderThreading.hpp>
#include <Video/FrameQueue.hpp>
#include <Video/VideoInterface.hpp>
 
extern "C" {
struct SwsContext;
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}
 
#include <QDebug>
 
#include <score_plugin_media_export.h>
 
#include <atomic>
namespace Video
{
class SCORE_PLUGIN_MEDIA_EXPORT Rescale
{
public:
  Rescale() = default;
  ~Rescale();
  Rescale(const Rescale&) = delete;
  Rescale(Rescale&&) = delete;
  Rescale& operator=(const Rescale&) = delete;
  Rescale& operator=(Rescale&&) = delete;
 
  operator bool() const noexcept { return m_rescale; }
 
  void open(const VideoMetadata& src);
  void close();
  void rescale(FrameQueue& m_frames, AVFramePointer& frame, ReadFrame& read);
 
private:
  const VideoMetadata* m_src{};
  SwsContext* m_rescale{};
  AVPixelFormat m_rescaleFormat{};
  int m_srcWidth{};
  int m_srcHeight{};
};
 
struct SCORE_PLUGIN_MEDIA_EXPORT DecoderConfiguration
{
  AVPixelFormat hardwareAcceleration{AV_PIX_FMT_NONE};
  std::string decoder;
  int threads{};
  DecodeUseCase useCase{DecodeUseCase::Playback};
  int graphicsApi{};    // score::gfx::GraphicsApi (avoid header dependency)
  uint32_t gpuVendorId{}; // PCI vendor ID of the render GPU (0 = unknown)
  bool useAVCodec{true};
  bool ignorePTS{false};
};
 
struct SCORE_PLUGIN_MEDIA_EXPORT LibAVDecoder
{
  ReadFrame enqueue_frame(const AVPacket* pkt) noexcept;
  std::pair<AVBufferRef*, const AVCodec*> open_hwdec(const AVCodec&) noexcept;
 
  int init_codec_context(
      const AVCodec* codec, AVBufferRef* hw_dev_ctx, const AVStream* stream,
      std::function<void(AVCodecContext&)> setup);
  bool open_codec_context(
      VideoInterface& self, const AVStream* stream,
      std::function<void(AVCodecContext&)> setup);
  void init_scaler(VideoInterface& self) noexcept;
  void load_packet_in_frame(const AVPacket& packet, AVFrame& frame);
 
  ReadFrame read_one_frame(AVPacket& packet);
  ReadFrame read_one_frame_raw(AVPacket& packet);
  ReadFrame read_one_frame_avcodec(AVPacket& packet);
 
  DecoderConfiguration m_conf;
 
  AVFormatContext* m_formatContext{};
  AVStream* m_avstream{};
  const AVCodec* m_codec{};
  AVCodecContext* m_codecContext{};
 
  FrameQueue m_frames;
  Rescale m_rescale;
  MissingTimestamps m_missingTimestamps;
  int64_t m_ptsLead{};
  int64_t m_skipBefore{AV_NOPTS_VALUE};
  AVFrame* m_skipCandidate{};
  int64_t m_skipLandingPts{AV_NOPTS_VALUE};
  void landSkip(AVFrame* f) noexcept;
  void dropSkipCandidate() noexcept;
  ReadFrame takeSkipCandidate() noexcept;
  std::atomic_bool m_finished{};
};
 
}
#endif