src/lib/encoder.cc

   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/encoder.h
  21  *  @brief Parent class for classes which can encode video and audio frames.
  22  */
  23
  24 #include "encoder.h"
  25 #include "util.h"
  26 #include "options.h"
  27
  28 using std::pair;
  29 using namespace boost;
  30
  31 int const Encoder::_history_size = 25;
  32
  33 /** @param f Film that we are encoding.
  34  *  @param o Options.
  35  */
  36 Encoder::Encoder (shared_ptr<const Film> f, shared_ptr<const Options> o)
  37         : _film (f)
  38         , _opt (o)
  39         , _just_skipped (false)
  40         , _last_frame (0)
  41 {
  42
  43 }
  44
  45
  46 /** @return an estimate of the current number of frames we are encoding per second,
  47  *  or 0 if not known.
  48  */
  49 float
  50 Encoder::current_frames_per_second () const
  51 {
  52         boost::mutex::scoped_lock lock (_history_mutex);
  53         if (int (_time_history.size()) < _history_size) {
  54                 return 0;
  55         }
  56
  57         struct timeval now;
  58         gettimeofday (&now, 0);
  59
  60         return _history_size / (seconds (now) - seconds (_time_history.back ()));
  61 }
  62
  63 /** @return true if the last frame to be processed was skipped as it already existed */
  64 bool
  65 Encoder::skipping () const
  66 {
  67         boost::mutex::scoped_lock (_history_mutex);
  68         return _just_skipped;
  69 }
  70
  71 /** @return Index of last frame to be successfully encoded */
  72 SourceFrame
  73 Encoder::last_frame () const
  74 {
  75         boost::mutex::scoped_lock (_history_mutex);
  76         return _last_frame;
  77 }
  78
  79 /** Should be called when a frame has been encoded successfully.
  80  *  @param n Source frame index.
  81  */
  82 void
  83 Encoder::frame_done (SourceFrame n)
  84 {
  85         boost::mutex::scoped_lock lock (_history_mutex);
  86         _just_skipped = false;
  87         _last_frame = n;
  88
  89         struct timeval tv;
  90         gettimeofday (&tv, 0);
  91         _time_history.push_front (tv);
  92         if (int (_time_history.size()) > _history_size) {
  93                 _time_history.pop_back ();
  94         }
  95 }
  96
  97 /** Called by a subclass when it has just skipped the processing
  98     of a frame because it has already been done.
  99 */
 100 void
 101 Encoder::frame_skipped ()
 102 {
 103         boost::mutex::scoped_lock lock (_history_mutex);
 104         _just_skipped = true;
 105 }
 106
 107 void
 108 Encoder::process_video (shared_ptr<const Image> i, SourceFrame f, boost::shared_ptr<Subtitle> s)
 109 {
 110         if (_opt->decode_video_skip != 0 && (f % _opt->decode_video_skip) != 0) {
 111                 return;
 112         }
 113
 114         if (_opt->video_decode_range) {
 115                 pair<SourceFrame, SourceFrame> const r = _opt->video_decode_range.get();
 116                 if (f < r.first || f >= r.second) {
 117                         return;
 118                 }
 119         }
 120
 121         do_process_video (i, f, s);
 122 }
 123
 124 void
 125 Encoder::process_audio (shared_ptr<const AudioBuffers> data, int64_t f)
 126 {
 127         if (_opt->audio_decode_range) {
 128
 129                 shared_ptr<AudioBuffers> trimmed (new AudioBuffers (*data.get ()));
 130
 131                 /* Range that we are encoding */
 132                 pair<int64_t, int64_t> required_range = _opt->audio_decode_range.get();
 133                 /* Range of this block of data */
 134                 pair<int64_t, int64_t> this_range (f, f + trimmed->frames());
 135
 136                 if (this_range.second < required_range.first || required_range.second < this_range.first) {
 137                         /* No part of this audio is within the required range */
 138                         return;
 139                 } else if (required_range.first >= this_range.first && required_range.first < this_range.second) {
 140                         /* Trim start */
 141                         int64_t const shift = required_range.first - this_range.first;
 142                         trimmed->move (shift, 0, trimmed->frames() - shift);
 143                         trimmed->set_frames (trimmed->frames() - shift);
 144                 } else if (required_range.second >= this_range.first && required_range.second < this_range.second) {
 145                         /* Trim end */
 146                         trimmed->set_frames (required_range.second - this_range.first);
 147                 }
 148
 149                 data = trimmed;
 150         }
 151
 152         do_process_audio (data);
 153 }