src/lib/decoder.h

   1 /*
   2     Copyright (C) 2012 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 /** @file  src/decoder.h
  21  *  @brief Parent class for decoders of content.
  22  */
  23
  24 #ifndef DVDOMATIC_DECODER_H
  25 #define DVDOMATIC_DECODER_H
  26
  27 #include <vector>
  28 #include <string>
  29 #include <stdint.h>
  30 #include <boost/shared_ptr.hpp>
  31 #include <boost/signals2.hpp>
  32 #include "util.h"
  33 #include "stream.h"
  34
  35 class Job;
  36 class Options;
  37 class Image;
  38 class Log;
  39 class DelayLine;
  40 class TimedSubtitle;
  41 class Subtitle;
  42 class Film;
  43 class FilterGraph;
  44
  45 /** @class Decoder.
  46  *  @brief Parent class for decoders of content.
  47  *
  48  *  These classes can be instructed run through their content
  49  *  (by calling ::go), and they emit signals when video or audio data is ready for something else
  50  *  to process.
  51  */
  52 class Decoder
  53 {
  54 public:
  55         Decoder (boost::shared_ptr<Film>, boost::shared_ptr<const Options>, Job *);
  56         virtual ~Decoder ();
  57
  58         /* Methods to query our input video */
  59
  60         /** @return video frames per second, or 0 if unknown */
  61         virtual float frames_per_second () const = 0;
  62         /** @return native size in pixels */
  63         virtual Size native_size () const = 0;
  64
  65         virtual int time_base_numerator () const = 0;
  66         virtual int time_base_denominator () const = 0;
  67         virtual int sample_aspect_ratio_numerator () const = 0;
  68         virtual int sample_aspect_ratio_denominator () const = 0;
  69
  70         void process_begin ();
  71         virtual bool pass () = 0;
  72         void process_end ();
  73         void go ();
  74
  75         SourceFrame video_frame () const {
  76                 return _video_frame;
  77         }
  78
  79         virtual void set_audio_stream (boost::optional<AudioStream>);
  80         virtual void set_subtitle_stream (boost::optional<SubtitleStream>);
  81
  82         boost::optional<AudioStream> audio_stream () const {
  83                 return _audio_stream;
  84         }
  85
  86         boost::optional<SubtitleStream> subtitle_stream () const {
  87                 return _subtitle_stream;
  88         }
  89
  90         std::vector<AudioStream> audio_streams () const {
  91                 return _audio_streams;
  92         }
  93
  94         std::vector<SubtitleStream> subtitle_streams () const {
  95                 return _subtitle_streams;
  96         }
  97
  98         /** Emitted when a video frame is ready.
  99          *  First parameter is the frame within the source.
 100          *  Second parameter is its index within the content.
 101          *  Third parameter is either 0 or a subtitle that should be on this frame.
 102          */
 103         boost::signals2::signal<void (boost::shared_ptr<Image>, SourceFrame, boost::shared_ptr<Subtitle>)> Video;
 104
 105         /** Emitted when some audio data is ready */
 106         boost::signals2::signal<void (boost::shared_ptr<AudioBuffers>, int64_t)> Audio;
 107
 108 protected:
 109
 110         virtual PixelFormat pixel_format () const = 0;
 111
 112         void process_video (AVFrame *);
 113         void process_audio (boost::shared_ptr<AudioBuffers>);
 114         void process_subtitle (boost::shared_ptr<TimedSubtitle>);
 115         void repeat_last_video ();
 116
 117         /** our Film */
 118         boost::shared_ptr<Film> _film;
 119         /** our options */
 120         boost::shared_ptr<const Options> _opt;
 121         /** associated Job, or 0 */
 122         Job* _job;
 123
 124         boost::optional<AudioStream> _audio_stream;
 125         boost::optional<SubtitleStream> _subtitle_stream;
 126
 127         std::vector<AudioStream> _audio_streams;
 128         std::vector<SubtitleStream> _subtitle_streams;
 129
 130 private:
 131         void emit_video (boost::shared_ptr<Image>, boost::shared_ptr<Subtitle>);
 132         void emit_audio (boost::shared_ptr<AudioBuffers>);
 133
 134         SourceFrame _video_frame;
 135         int64_t _audio_frame;
 136
 137         std::list<boost::shared_ptr<FilterGraph> > _filter_graphs;
 138
 139         DelayLine* _delay_line;
 140         int _delay_in_frames;
 141
 142         boost::shared_ptr<TimedSubtitle> _timed_subtitle;
 143
 144         boost::shared_ptr<Image> _last_image;
 145         boost::shared_ptr<Subtitle> _last_subtitle;
 146 };
 147
 148 #endif