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
  34 using std::string;
  35 using std::vector;
  36 using boost::shared_ptr;
  37
  38 ExamineContentJob::ExamineContentJob (shared_ptr<Film> f, shared_ptr<Job> req)
  39         : Job (f, req)
  40 {
  41
  42 }
  43
  44 ExamineContentJob::~ExamineContentJob ()
  45 {
  46 }
  47
  48 string
  49 ExamineContentJob::name () const
  50 {
  51         if (_film->name().empty ()) {
  52                 return "Examine content";
  53         }
  54
  55         return String::compose ("Examine content of %1", _film->name());
  56 }
  57
  58 void
  59 ExamineContentJob::run ()
  60 {
  61         /* Decode the content to get an accurate length */
  62
  63         /* We don't want to use any existing length here, as progress
  64            will be messed up.
  65         */
  66         _film->unset_length ();
  67
  68         shared_ptr<Options> o (new Options ("", "", ""));
  69         o->out_size = Size (512, 512);
  70         o->apply_crop = false;
  71         o->decode_audio = false;
  72
  73         descend (0.5);
  74
  75         _decoder = decoder_factory (_film, o, this);
  76         _decoder->go ();
  77
  78         _film->set_length (_decoder->video_frame());
  79
  80         _film->log()->log (String::compose ("Video length is %1 frames", _film->length()));
  81
  82         ascend ();
  83
  84         /* Now make thumbnails for it */
  85
  86         descend (0.5);
  87
  88         try {
  89                 o.reset (new Options (_film->dir ("thumbs"), ".png", ""));
  90                 o->out_size = _film->size ();
  91                 o->apply_crop = false;
  92                 o->decode_audio = false;
  93                 if (_film->length() > 0) {
  94                         o->decode_video_skip = _film->length().get() / 128;
  95                 } else {
  96                         o->decode_video_skip = 0;
  97                 }
  98                 o->decode_subtitles = true;
  99                 shared_ptr<ImageMagickEncoder> e (new ImageMagickEncoder (_film, o));
 100                 Transcoder w (_film, o, this, e);
 101                 w.go ();
 102
 103         } catch (std::exception& e) {
 104
 105                 ascend ();
 106                 set_progress (1);
 107                 set_error (e.what ());
 108                 set_state (FINISHED_ERROR);
 109                 return;
 110
 111         }
 112
 113         string const tdir = _film->dir ("thumbs");
 114         vector<SourceFrame> thumbs;
 115
 116         for (boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (tdir); i != boost::filesystem::directory_iterator(); ++i) {
 117
 118                 /* Aah, the sweet smell of progress */
 119 #if BOOST_FILESYSTEM_VERSION == 3
 120                 string const l = boost::filesystem::path(*i).leaf().generic_string();
 121 #else
 122                 string const l = i->leaf ();
 123 #endif
 124
 125                 size_t const d = l.find (".png");
 126                 size_t const t = l.find (".tmp");
 127                 if (d != string::npos && t == string::npos) {
 128                         thumbs.push_back (atoi (l.substr (0, d).c_str()));
 129                 }
 130         }
 131
 132         sort (thumbs.begin(), thumbs.end());
 133         _film->set_thumbs (thumbs);
 134
 135         ascend ();
 136         set_progress (1);
 137         set_state (FINISHED_OK);
 138 }