Use $HOME rather than hard-coded user name.
[dcpomatic.git] / src / lib / image_examiner.cc
index 2d150583a1186570e222fd4f15557a01097d487d..15a0b043d56543c8451cfd243c27204afc336f16 100644 (file)
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2021 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/>.
 
 */
 
-#include <iostream>
-#include <boost/lexical_cast.hpp>
-#include <Magick++.h>
+
+#include "compose.hpp"
+#include "config.h"
+#include "cross.h"
+#include "exceptions.h"
+#include "ffmpeg_image_proxy.h"
+#include "film.h"
+#include "image.h"
 #include "image_content.h"
 #include "image_examiner.h"
-#include "film.h"
 #include "job.h"
-#include "exceptions.h"
-#include "config.h"
+#include <dcp/openjpeg_image.h>
+#include <dcp/exceptions.h>
+#include <dcp/filesystem.h>
+#include <dcp/j2k_transcode.h>
+#include <iostream>
 
 #include "i18n.h"
 
+
 using std::cout;
 using std::list;
+using std::shared_ptr;
 using std::sort;
-using boost::shared_ptr;
-using boost::lexical_cast;
-using boost::bad_lexical_cast;
+using boost::optional;
+
 
-ImageExaminer::ImageExaminer (shared_ptr<const Film> film, shared_ptr<const ImageContent> content, shared_ptr<Job> job)
+ImageExaminer::ImageExaminer (shared_ptr<const Film> film, shared_ptr<const ImageContent> content, shared_ptr<Job>)
        : _film (film)
        , _image_content (content)
-       , _video_length (0)
 {
-       list<unsigned int> frames;
-       size_t const N = content->number_of_paths ();
-
-       for (size_t i = 0; i < N; ++i) {
-               boost::filesystem::path const p = content->path (i);
-               try {
-                       frames.push_back (lexical_cast<int> (p.stem().string()));
-               } catch (bad_lexical_cast &) {
-                       /* We couldn't turn that filename into a number; never mind */
+       auto path = content->path(0);
+       if (valid_j2k_file (path)) {
+               auto size = dcp::filesystem::file_size(path);
+               dcp::File f(path, "rb");
+               if (!f) {
+                       throw FileError ("Could not open file for reading", path);
                }
-               
-               if (!_video_size) {
-                       using namespace MagickCore;
-                       Magick::Image* image = new Magick::Image (p.string());
-                       _video_size = libdcp::Size (image->columns(), image->rows());
-                       delete image;
+               std::vector<uint8_t> buffer(size);
+               f.checked_read(buffer.data(), size);
+               f.close();
+               try {
+                       _video_size = dcp::decompress_j2k(buffer.data(), size, 0)->size();
+               } catch (dcp::ReadError& e) {
+                       throw DecodeError (String::compose (_("Could not decode JPEG2000 file %1 (%2)"), path, e.what ()));
                }
-       
-               job->set_progress (float (i) / N);
-       }
-
-       frames.sort ();
-       
-       if (N > 1 && frames.front() != 0 && frames.front() != 1) {
-               throw StringError (String::compose (_("first frame in moving image directory is number %1"), frames.front ()));
-       }
-
-       if (N > 1 && frames.back() != frames.size() && frames.back() != (frames.size() - 1)) {
-               throw StringError (String::compose (_("there are %1 images in the directory but the last one is number %2"), frames.size(), frames.back ()));
+       } else {
+               FFmpegImageProxy proxy(content->path(0));
+               _video_size = proxy.image(Image::Alignment::COMPACT).image->size();
        }
 
        if (content->still ()) {
-               _video_length = Config::instance()->default_still_length() * video_frame_rate();
+               _video_length = Config::instance()->default_still_length() * video_frame_rate().get_value_or (film->video_frame_rate ());
        } else {
                _video_length = _image_content->number_of_paths ();
        }
 }
 
-libdcp::Size
+
+optional<dcp::Size>
 ImageExaminer::video_size () const
 {
-       return _video_size.get ();
+       return _video_size;
 }
 
-float
+
+optional<double>
 ImageExaminer::video_frame_rate () const
 {
-       boost::shared_ptr<const Film> f = _film.lock ();
-       if (!f) {
-               return 24;
+       if (_image_content->video_frame_rate()) {
+               /* The content already knows what frame rate it should be */
+               return _image_content->video_frame_rate().get();
        }
 
-       return f->video_frame_rate ();
+       /* Don't know */
+       return {};
+}
+
+
+bool
+ImageExaminer::yuv () const
+{
+       /* We never convert ImageSource from YUV to RGB (though maybe sometimes we should)
+          so it makes sense to just say they are never YUV so the option of a conversion
+          to RGB is not offered.
+       */
+       return false;
 }