src/lib/moving_image_examiner.cc

   1 /*
   2     Copyright (C) 2013 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 #include <iostream>
  21 #include <boost/lexical_cast.hpp>
  22 #include <Magick++.h>
  23 #include "moving_image_content.h"
  24 #include "moving_image_examiner.h"
  25 #include "film.h"
  26 #include "job.h"
  27 #include "exceptions.h"
  28
  29 #include "i18n.h"
  30
  31 using std::cout;
  32 using std::list;
  33 using std::sort;
  34 using boost::shared_ptr;
  35 using boost::lexical_cast;
  36
  37 MovingImageExaminer::MovingImageExaminer (shared_ptr<const Film> film, shared_ptr<const MovingImageContent> content, shared_ptr<Job> job)
  38         : MovingImage (content)
  39         , _film (film)
  40         , _video_length (0)
  41 {
  42         list<unsigned int> frames;
  43         unsigned int files = 0;
  44
  45         for (boost::filesystem::directory_iterator i(content->path()); i != boost::filesystem::directory_iterator(); ++i) {
  46                 if (boost::filesystem::is_regular_file (i->path ())) {
  47                         ++files;
  48                 }
  49         }
  50
  51         int j = 0;
  52         for (boost::filesystem::directory_iterator i(content->path()); i != boost::filesystem::directory_iterator(); ++i) {
  53                 if (!boost::filesystem::is_regular_file (i->path ())) {
  54                         continue;
  55                 }
  56
  57                 if (valid_image_file (i->path ())) {
  58                         int n = lexical_cast<int> (i->path().stem().string());
  59                         frames.push_back (n);
  60                         _files.push_back (i->path().filename ());
  61
  62                         if (!_video_size) {
  63                                 using namespace MagickCore;
  64                                 Magick::Image* image = new Magick::Image (i->path().string());
  65                                 _video_size = libdcp::Size (image->columns(), image->rows());
  66                                 delete image;
  67                         }
  68                 }
  69
  70                 job->set_progress (float (j) / files);
  71                 ++j;
  72         }
  73
  74         frames.sort ();
  75         sort (_files.begin(), _files.end ());
  76
  77         if (frames.size() < 2) {
  78                 throw StringError (String::compose (_("only %1 file(s) found in moving image directory"), frames.size ()));
  79         }
  80
  81         if (frames.front() != 0 && frames.front() != 1) {
  82                 throw StringError (String::compose (_("first frame in moving image directory is number %1"), frames.front ()));
  83         }
  84
  85         if (frames.back() != frames.size() && frames.back() != (frames.size() - 1)) {
  86                 throw StringError (String::compose (_("there are %1 images in the directory but the last one is number %2"), frames.size(), frames.back ()));
  87         }
  88
  89         _video_length = frames.size ();
  90 }
  91
  92 libdcp::Size
  93 MovingImageExaminer::video_size () const
  94 {
  95         return _video_size.get ();
  96 }
  97
  98 int
  99 MovingImageExaminer::video_length () const
 100 {
 101         return _video_length;
 102 }
 103
 104 float
 105 MovingImageExaminer::video_frame_rate () const
 106 {
 107         return 24;
 108 }
 109