git.sesse.net Git - casparcg/blob - modules/ffmpeg/producer/util/util.h

   1 /*
   2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
   3 *
   4 * This file is part of CasparCG (www.casparcg.com).
   5 *
   6 * CasparCG is free software: you can redistribute it and/or modify
   7 * it under the terms of the GNU General Public License as published by
   8 * the Free Software Foundation, either version 3 of the License, or
   9 * (at your option) any later version.
  10 *
  11 * CasparCG is distributed in the hope that it will be useful,
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 * GNU General Public License for more details.
  15 *
  16 * You should have received a copy of the GNU General Public License
  17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
  18 *
  19 * Author: Robert Nagy, ronag89@gmail.com
  20 */
  21
  22 #pragma once
  23
  24 #include <common/forward.h>
  25 #include <common/memory.h>
  26
  27 #include <core/video_format.h>
  28 #include <core/frame/pixel_format.h>
  29 #include <core/mixer/audio/audio_mixer.h>
  30 #include <core/fwd.h>
  31
  32 #include <boost/rational.hpp>
  33
  34 #include <array>
  35
  36 #if defined(_MSC_VER)
  37 #pragma warning (push)
  38 #pragma warning (disable : 4244)
  39 #endif
  40 extern "C"
  41 {
  42 #include <libavutil/avutil.h>
  43 }
  44 #if defined(_MSC_VER)
  45 #pragma warning (pop)
  46 #endif
  47
  48 struct AVFrame;
  49 struct AVFormatContext;
  50 struct AVPacket;
  51 struct AVRational;
  52 struct AVCodecContext;
  53
  54 namespace caspar { namespace ffmpeg {
  55
  56 // Utils
  57
  58 core::field_mode                                        get_mode(const AVFrame& frame);
  59 core::mutable_frame                                     make_frame(const void* tag, const spl::shared_ptr<AVFrame>& decoded_frame, core::frame_factory& frame_factory, const core::audio_channel_layout& channel_layout);
  60 spl::shared_ptr<AVFrame>                        make_av_frame(core::mutable_frame& frame);
  61 spl::shared_ptr<AVFrame>                        make_av_frame(std::array<uint8_t*, 4> data, const core::pixel_format_desc& pix_desc);
  62
  63 core::pixel_format_desc                         pixel_format_desc(PixelFormat pix_fmt, int width, int height);
  64
  65 spl::shared_ptr<AVPacket> create_packet();
  66 spl::shared_ptr<AVFrame>  create_frame();
  67
  68 spl::shared_ptr<AVCodecContext> open_codec(AVFormatContext& context, AVMediaType type, int& index, bool single_threaded);
  69 spl::shared_ptr<AVFormatContext> open_input(const std::wstring& filename);
  70
  71 bool is_sane_fps(AVRational time_base);
  72 AVRational fix_time_base(AVRational time_base);
  73
  74 std::shared_ptr<AVFrame> flush();
  75
  76 double read_fps(AVFormatContext& context, double fail_value);
  77 boost::rational<int> read_framerate(AVFormatContext& context, const boost::rational<int>& fail_value);
  78
  79 std::wstring print_mode(int width, int height, double fps, bool interlaced);
  80
  81 std::wstring probe_stem(const std::wstring& stem, bool only_video);
  82 bool is_valid_file(const std::wstring& filename, bool only_video);
  83 bool try_get_duration(const std::wstring filename, std::int64_t& duration, boost::rational<std::int64_t>& time_base);
  84
  85 core::audio_channel_layout get_audio_channel_layout(int num_channels, std::uint64_t layout, const std::wstring& channel_layout_spec);
  86
  87 // av_get_default_channel_layout does not work for layouts not predefined in ffmpeg. This is needed to support > 8 channels.
  88 std::int64_t create_channel_layout_bitmask(int num_channels);
  89
  90 }}