LibavMediaInfo.hpp

#pragma once
#include <Media/Libav.hpp>
 
#include <score_plugin_media_export.h>
 
#if SCORE_HAS_LIBAV
 
#include <QString>
 
#include <optional>
#include <stdexcept>
#include <string>
#include <vector>
 
extern "C" {
#include <libavformat/avformat.h>
}
 
namespace score::libav
{
 
struct SCORE_PLUGIN_MEDIA_EXPORT MediaInfo
{
  struct FormatContextDeleter
  {
    void operator()(AVFormatContext* ctx) const noexcept
    {
      if(ctx)
        avformat_close_input(&ctx);
    }
  };
  using FormatContextPtr = std::unique_ptr<AVFormatContext, FormatContextDeleter>;
 
  FormatContextPtr format_context;
  std::vector<AVMediaType> streams;
  std::optional<int64_t> duration_av; // in AV_TIME_BASE units; empty == unknown
 
  std::optional<double> duration_seconds() const noexcept
  {
    if(!duration_av)
      return std::nullopt;
    return *duration_av / double(AV_TIME_BASE);
  }
 
  bool has_video() const noexcept
  {
    for(auto t : streams)
      if(t == AVMEDIA_TYPE_VIDEO)
        return true;
    return false;
  }
 
  bool has_audio() const noexcept
  {
    for(auto t : streams)
      if(t == AVMEDIA_TYPE_AUDIO)
        return true;
    return false;
  }
};
 
SCORE_PLUGIN_MEDIA_EXPORT
std::optional<MediaInfo> probe(const QString& path);
 
SCORE_PLUGIN_MEDIA_EXPORT
MediaInfo probe_or_throw(const QString& path);
 
} // namespace score::libav
 
#endif