src/lib/examine_content_job.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/examine_content_job.cc
  21  *  @brief A class to run through content at high speed to find its length.
  22  */
  23
  24 #include <boost/filesystem.hpp>
  25 #include "examine_content_job.h"
  26 #include "options.h"
  27 #include "decoder_factory.h"
  28 #include "decoder.h"
  29 #include "imagemagick_encoder.h"
  30 #include "transcoder.h"
  31 #include "log.h"
  32 #include "film.h"
  33 #include "video_decoder.h"
  34
  35 using std::string;
  36 using std::vector;
  37 using std::pair;
  38 using boost::shared_ptr;
  39
  40 ExamineContentJob::ExamineContentJob (shared_ptr<Film> f, shared_ptr<Job> req)
  41         : Job (f, req)
  42 {
  43
  44 }
  45
  46 ExamineContentJob::~ExamineContentJob ()
  47 {
  48 }
  49
  50 string
  51 ExamineContentJob::name () const
  52 {
  53         if (_film->name().empty ()) {
  54                 return "Examine content";
  55         }
  56
  57         return String::compose ("Examine content of %1", _film->name());
  58 }
  59
  60 void
  61 ExamineContentJob::run ()
  62 {
  63         float progress_remaining = 1;
  64
  65         /* Set the film's length to either
  66            a) a length judged by running through the content or
  67            b) the length from a decoder's header.
  68         */
  69
  70         if (!_film->trust_content_header()) {
  71                 /* Decode the content to get an accurate length */
  72
  73                 /* We don't want to use any existing length here, as progress
  74                    will be messed up.
  75                 */
  76                 _film->unset_length ();
  77
  78                 shared_ptr<Options> o (new Options ("", "", ""));
  79                 o->out_size = Size (512, 512);
  80                 o->apply_crop = false;
  81                 o->decode_audio = false;
  82
  83                 descend (0.5);
  84
  85                 pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > decoders = decoder_factory (_film, o, this);
  86
  87                 set_progress_unknown ();
  88                 while (!decoders.first->pass()) {
  89                         /* keep going */
  90                 }
  91
  92                 _film->set_length (decoders.first->video_frame());
  93
  94                 _film->log()->log (String::compose ("Video length examined as %1 frames", _film->length().get()));
  95
  96                 ascend ();
  97
  98                 progress_remaining -= 0.5;
  99
 100         } else {
 101
 102                 /* Get a quick decoder to get the content's length from its header.
 103                    It would have been nice to just use the thumbnail transcoder's decoder,
 104                    but that's a bit fiddly, and this isn't too expensive.
 105                 */
 106
 107                 shared_ptr<Options> o (new Options ("", "", ""));
 108                 o->out_size = Size (1024, 1024);
 109                 pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > d = decoder_factory (_film, o, 0);
 110                 _film->set_length (d.first->length());
 111
 112                 _film->log()->log (String::compose ("Video length obtained from header as %1 frames", _film->length().get()));
 113         }
 114
 115         /* Now make thumbnails for it */
 116
 117         descend (progress_remaining);
 118
 119         try {
 120                 shared_ptr<Options> o (new Options (_film->dir ("thumbs"), ".png", ""));
 121                 o->out_size = _film->size ();
 122                 o->apply_crop = false;
 123                 o->decode_audio = false;
 124                 o->decode_video_skip = _film->length().get() / 128;
 125                 o->decode_subtitles = true;
 126                 shared_ptr<ImageMagickEncoder> e (new ImageMagickEncoder (_film, o));
 127                 Transcoder w (_film, o, this, e);
 128                 w.go ();
 129
 130                 /* Now set the film's length from the transcoder's decoder, since we
 131                    went to all the trouble of going through the content.
 132                 */
 133
 134                 _film->set_length (w.video_decoder()->video_frame());
 135
 136         } catch (std::exception& e) {
 137
 138                 ascend ();
 139                 set_progress (1);
 140                 set_error (e.what ());
 141                 set_state (FINISHED_ERROR);
 142                 return;
 143
 144         }
 145
 146         string const tdir = _film->dir ("thumbs");
 147         vector<SourceFrame> thumbs;
 148
 149         for (boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (tdir); i != boost::filesystem::directory_iterator(); ++i) {
 150
 151                 /* Aah, the sweet smell of progress */
 152 #if BOOST_FILESYSTEM_VERSION == 3
 153                 string const l = boost::filesystem::path(*i).leaf().generic_string();
 154 #else
 155                 string const l = i->leaf ();
 156 #endif
 157
 158                 size_t const d = l.find (".png");
 159                 size_t const t = l.find (".tmp");
 160                 if (d != string::npos && t == string::npos) {
 161                         thumbs.push_back (atoi (l.substr (0, d).c_str()));
 162                 }
 163         }
 164
 165         sort (thumbs.begin(), thumbs.end());
 166         _film->set_thumbs (thumbs);
 167
 168         ascend ();
 169         set_progress (1);
 170         set_state (FINISHED_OK);
 171 }