src/lib/ffmpeg_stream.cc

   1 /*
   2     Copyright (C) 2013-2014 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 extern "C" {
  21 #include <libavformat/avformat.h>
  22 }
  23 #include <libxml++/libxml++.h>
  24 #include <dcp/raw_convert.h>
  25 #include "ffmpeg_stream.h"
  26
  27 using std::string;
  28 using dcp::raw_convert;
  29
  30 FFmpegStream::FFmpegStream (cxml::ConstNodePtr node)
  31         : name (node->string_child ("Name"))
  32         , _id (node->number_child<int> ("Id"))
  33 {
  34
  35 }
  36
  37 void
  38 FFmpegStream::as_xml (xmlpp::Node* root) const
  39 {
  40         root->add_child("Name")->add_child_text (name);
  41         root->add_child("Id")->add_child_text (raw_convert<string> (_id));
  42 }
  43
  44 bool
  45 FFmpegStream::uses_index (AVFormatContext const * fc, int index) const
  46 {
  47         size_t i = 0;
  48         while (i < fc->nb_streams) {
  49                 if (fc->streams[i]->id == _id) {
  50                         return int (i) == index;
  51                 }
  52                 ++i;
  53         }
  54
  55         return false;
  56 }
  57
  58 AVStream *
  59 FFmpegStream::stream (AVFormatContext const * fc) const
  60 {
  61         size_t i = 0;
  62         while (i < fc->nb_streams) {
  63                 if (fc->streams[i]->id == _id) {
  64                         return fc->streams[i];
  65                 }
  66                 ++i;
  67         }
  68
  69         assert (false);
  70         return 0;
  71 }