git.sesse.net Git - nageru/blob - ffmpeg_raii.h

   1 #ifndef _FFMPEG_RAII_H
   2 #define _FFMPEG_RAII_H 1
   3
   4 // Some helpers to make RAII versions of FFmpeg objects.
   5 // The cleanup functions don't interact all that well with unique_ptr,
   6 // so things get a bit messy and verbose, but overall it's worth it to ensure
   7 // we never leak things by accident in error paths.
   8 //
   9 // This does not cover any of the types that can actually be declared as
  10 // a unique_ptr with no helper functions for deleter.
  11
  12 #include <memory>
  13
  14 struct AVCodec;
  15 struct AVCodecContext;
  16 struct AVCodecParameters;
  17 struct AVDictionary;
  18 struct AVFormatContext;
  19 struct AVFrame;
  20 struct AVInputFormat;
  21 struct SwsContext;
  22 typedef struct AVIOInterruptCB AVIOInterruptCB;
  23
  24 // AVFormatContext
  25 struct avformat_close_input_unique {
  26         void operator() (AVFormatContext *format_ctx) const;
  27 };
  28
  29 typedef std::unique_ptr<AVFormatContext, avformat_close_input_unique>
  30         AVFormatContextWithCloser;
  31
  32 AVFormatContextWithCloser avformat_open_input_unique(
  33         const char *pathname, AVInputFormat *fmt,
  34         AVDictionary **options);
  35
  36 AVFormatContextWithCloser avformat_open_input_unique(
  37         const char *pathname, AVInputFormat *fmt,
  38         AVDictionary **options,
  39         const AVIOInterruptCB &interrupt_cb);
  40
  41
  42 // AVCodecContext
  43 struct avcodec_free_context_unique {
  44         void operator() (AVCodecContext *ctx) const;
  45 };
  46
  47 typedef std::unique_ptr<AVCodecContext, avcodec_free_context_unique>
  48         AVCodecContextWithDeleter;
  49
  50 AVCodecContextWithDeleter avcodec_alloc_context3_unique(const AVCodec *codec);
  51
  52
  53 // AVCodecParameters
  54 struct avcodec_parameters_free_unique {
  55         void operator() (AVCodecParameters *codec_par) const;
  56 };
  57
  58 typedef std::unique_ptr<AVCodecParameters, avcodec_parameters_free_unique>
  59         AVCodecParametersWithDeleter;
  60
  61
  62 // AVFrame
  63 struct av_frame_free_unique {
  64         void operator() (AVFrame *frame) const;
  65 };
  66
  67 typedef std::unique_ptr<AVFrame, av_frame_free_unique>
  68         AVFrameWithDeleter;
  69
  70 AVFrameWithDeleter av_frame_alloc_unique();
  71
  72 // SwsContext
  73 struct sws_free_context_unique {
  74         void operator() (SwsContext *context) const;
  75 };
  76
  77 typedef std::unique_ptr<SwsContext, sws_free_context_unique>
  78         SwsContextWithDeleter;
  79
  80 #endif  // !defined(_FFMPEG_RAII_H)