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 #include "ffmpeg_stream.h"
  21 #include "dcpomatic_assert.h"
  22 #include <dcp/raw_convert.h>
  23 #include <libxml++/libxml++.h>
  24 extern "C" {
  25 #include <libavformat/avformat.h>
  26 }
  27
  28 using std::string;
  29 using dcp::raw_convert;
  30
  31 FFmpegStream::FFmpegStream (cxml::ConstNodePtr node)
  32         : name (node->string_child ("Name"))
  33         , _id (node->number_child<int> ("Id"))
  34 {
  35
  36 }
  37
  38 void
  39 FFmpegStream::as_xml (xmlpp::Node* root) const
  40 {
  41         root->add_child("Name")->add_child_text (name);
  42         root->add_child("Id")->add_child_text (raw_convert<string> (_id));
  43 }
  44
  45 bool
  46 FFmpegStream::uses_index (AVFormatContext const * fc, int index) const
  47 {
  48         size_t i = 0;
  49         while (i < fc->nb_streams) {
  50                 if (fc->streams[i]->id == _id) {
  51                         return int (i) == index;
  52                 }
  53                 ++i;
  54         }
  55
  56         return false;
  57 }
  58
  59 AVStream *
  60 FFmpegStream::stream (AVFormatContext const * fc) const
  61 {
  62         size_t i = 0;
  63         while (i < fc->nb_streams) {
  64                 if (fc->streams[i]->id == _id) {
  65                         return fc->streams[i];
  66                 }
  67                 ++i;
  68         }
  69
  70         DCPOMATIC_ASSERT (false);
  71         return 0;
  72 }