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

   1 /*
   2 * Copyright 2013 Sveriges Television AB http://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/memory.h>
  25
  26 #include <boost/rational.hpp>
  27 #include <boost/noncopyable.hpp>
  28
  29 #include <string>
  30 #include <vector>
  31
  32 #if defined(_MSC_VER)
  33 #pragma warning (push)
  34 #pragma warning (disable : 4244)
  35 #endif
  36 extern "C"
  37 {
  38 #include <libavutil/samplefmt.h>
  39 }
  40 #if defined(_MSC_VER)
  41 #pragma warning (pop)
  42 #endif
  43
  44 struct AVFrame;
  45
  46 namespace caspar { namespace ffmpeg {
  47
  48 struct audio_input_pad
  49 {
  50         boost::rational<int>    time_base;
  51         int                                             sample_rate;
  52         AVSampleFormat                  sample_fmt;
  53         std::int64_t                    audio_channel_layout;
  54
  55         audio_input_pad(
  56                         boost::rational<int> time_base,
  57                         int sample_rate,
  58                         AVSampleFormat sample_fmt,
  59                         std::int64_t audio_channel_layout)
  60                 : time_base(std::move(time_base))
  61                 , sample_rate(sample_rate)
  62                 , sample_fmt(sample_fmt)
  63                 , audio_channel_layout(audio_channel_layout)
  64         {
  65         }
  66 };
  67
  68 struct audio_output_pad
  69 {
  70         std::vector<int>                        sample_rates;
  71         std::vector<AVSampleFormat>     sample_fmts;
  72         std::vector<std::int64_t>       audio_channel_layouts;
  73
  74         audio_output_pad(
  75                         std::vector<int> sample_rates,
  76                         std::vector<AVSampleFormat> sample_fmts,
  77                         std::vector<std::int64_t> audio_channel_layouts)
  78                 : sample_rates(std::move(sample_rates))
  79                 , sample_fmts(std::move(sample_fmts))
  80                 , audio_channel_layouts(std::move(audio_channel_layouts))
  81         {
  82         }
  83 };
  84
  85 class audio_filter : boost::noncopyable
  86 {
  87 public:
  88         audio_filter(
  89                         std::vector<audio_input_pad> input_pads,
  90                         std::vector<audio_output_pad> output_pads,
  91                         const std::string& filtergraph);
  92         audio_filter(audio_filter&& other);
  93         audio_filter& operator=(audio_filter&& other);
  94
  95         void push(int input_pad_id, const std::shared_ptr<AVFrame>& frame);
  96         std::shared_ptr<AVFrame> poll(int output_pad_id);
  97         std::vector<spl::shared_ptr<AVFrame>> poll_all(int output_pad_id);
  98
  99         std::wstring filter_str() const;
 100 private:
 101         struct implementation;
 102         spl::shared_ptr<implementation> impl_;
 103 };
 104
 105 }}