Extract common code out into kdm_for_screen()
[dcpomatic.git] / src / lib / examine_content_job.cc
index 93333605b8e6a6983fc63fd668f420bd426fae35..3be38863d46e4204bf364da3989ee4b1d7e0fe19 100644 (file)
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
-/** @file  src/examine_content_job.cc
- *  @brief A class to run through content at high speed to find its length.
- */
-
 #include <boost/filesystem.hpp>
 #include "examine_content_job.h"
-#include "options.h"
-#include "decoder_factory.h"
-#include "decoder.h"
-#include "imagemagick_encoder.h"
-#include "transcoder.h"
 #include "log.h"
+#include "content.h"
 #include "film.h"
-#include "video_decoder.h"
+#include <iostream>
+
+#include "i18n.h"
 
 using std::string;
-using std::vector;
-using std::pair;
+using std::cout;
 using boost::shared_ptr;
 
-ExamineContentJob::ExamineContentJob (shared_ptr<Film> f, shared_ptr<Job> req)
-       : Job (f, req)
+ExamineContentJob::ExamineContentJob (shared_ptr<const Film> film, shared_ptr<Content> c)
+       : Job (film)
+       , _content (c)
 {
 
 }
 
 ExamineContentJob::~ExamineContentJob ()
 {
+       stop_thread ();
 }
 
 string
 ExamineContentJob::name () const
 {
-       if (_film->name().empty ()) {
-               return "Examine content";
-       }
-       
-       return String::compose ("Examine content of %1", _film->name());
+       return _("Examining content");
+}
+
+string
+ExamineContentJob::json_name () const
+{
+       return N_("examine_content");
 }
 
 void
 ExamineContentJob::run ()
 {
-       float progress_remaining = 1;
-
-       /* Set the film's length to either
-          a) a length judged by running through the content or
-          b) the length from a decoder's header.
-       */
-
-       if (!_film->trust_content_header()) {
-               /* Decode the content to get an accurate length */
-               
-               /* We don't want to use any existing length here, as progress
-                  will be messed up.
-               */
-               _film->unset_length ();
-               
-               shared_ptr<Options> o (new Options ("", "", ""));
-               o->out_size = Size (512, 512);
-               o->apply_crop = false;
-               o->decode_audio = false;
-               
-               descend (0.5);
-               
-               pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > decoders = decoder_factory (_film, o, this);
-               
-               set_progress_unknown ();
-               while (!decoders.first->pass()) {
-                       /* keep going */
-               }
-               
-               _film->set_length (decoders.first->video_frame());
-               
-               _film->log()->log (String::compose ("Video length examined as %1 frames", _film->length().get()));
-               
-               ascend ();
-               
-               progress_remaining -= 0.5;
-               
-       } else {
-
-               /* Get a quick decoder to get the content's length from its header.
-                  It would have been nice to just use the thumbnail transcoder's decoder,
-                  but that's a bit fiddly, and this isn't too expensive.
-               */
-               
-               shared_ptr<Options> o (new Options ("", "", ""));
-               o->out_size = Size (1024, 1024);
-               pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > d = decoder_factory (_film, o, 0);
-               _film->set_length (d.first->length());
-       
-               _film->log()->log (String::compose ("Video length obtained from header as %1 frames", _film->length().get()));
-       }
-
-       /* Now make thumbnails for it */
-
-       descend (progress_remaining);
-
-       try {
-               shared_ptr<Options> o (new Options (_film->dir ("thumbs"), ".png", ""));
-               o->out_size = _film->size ();
-               o->apply_crop = false;
-               o->decode_audio = false;
-               o->decode_video_skip = _film->length().get() / 128;
-               o->decode_subtitles = true;
-               shared_ptr<ImageMagickEncoder> e (new ImageMagickEncoder (_film, o));
-               Transcoder w (_film, o, this, e);
-               w.go ();
-
-               /* Now set the film's length from the transcoder's decoder, since we
-                  went to all the trouble of going through the content.
-               */
-
-               _film->set_length (w.video_decoder()->video_frame());
-               
-       } catch (std::exception& e) {
-
-               ascend ();
-               set_progress (1);
-               set_error (e.what ());
-               set_state (FINISHED_ERROR);
-               return;
-               
-       }
-
-       string const tdir = _film->dir ("thumbs");
-       vector<SourceFrame> thumbs;
-
-       for (boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (tdir); i != boost::filesystem::directory_iterator(); ++i) {
-
-               /* Aah, the sweet smell of progress */
-#if BOOST_FILESYSTEM_VERSION == 3              
-               string const l = boost::filesystem::path(*i).leaf().generic_string();
-#else
-               string const l = i->leaf ();
-#endif
-               
-               size_t const d = l.find (".png");
-               size_t const t = l.find (".tmp");
-               if (d != string::npos && t == string::npos) {
-                       thumbs.push_back (atoi (l.substr (0, d).c_str()));
-               }
-       }
-
-       sort (thumbs.begin(), thumbs.end());
-       _film->set_thumbs (thumbs);     
-
-       ascend ();
+       _content->examine (_film, shared_from_this());
        set_progress (1);
        set_state (FINISHED_OK);
 }