e06f6023d54a8d35b1203f81eef72e823ab9ed0d
[dcpomatic.git] / src / lib / image_decoder.cc
1 /*
2     Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "image_content.h"
22 #include "image_decoder.h"
23 #include "video_decoder.h"
24 #include "image.h"
25 #include "ffmpeg_image_proxy.h"
26 #include "j2k_image_proxy.h"
27 #include "film.h"
28 #include "exceptions.h"
29 #include "video_content.h"
30 #include <boost/filesystem.hpp>
31 #include <iostream>
32
33 #include "i18n.h"
34
35 using std::cout;
36 using boost::shared_ptr;
37 using dcp::Size;
38
39 ImageDecoder::ImageDecoder (shared_ptr<const ImageContent> c, shared_ptr<Log> log)
40         : _image_content (c)
41         , _frame_video_position (0)
42 {
43         video.reset (new VideoDecoder (this, c, log));
44 }
45
46 bool
47 ImageDecoder::pass ()
48 {
49         if (_frame_video_position >= _image_content->video->length()) {
50                 return true;
51         }
52
53         if (!_image_content->still() || !_image) {
54                 /* Either we need an image or we are using moving images, so load one */
55                 boost::filesystem::path path = _image_content->path (_image_content->still() ? 0 : _frame_video_position);
56                 if (valid_j2k_file (path)) {
57                         AVPixelFormat pf;
58                         if (_image_content->video->colour_conversion()) {
59                                 /* We have a specified colour conversion: assume the image is RGB */
60                                 pf = AV_PIX_FMT_RGB48LE;
61                         } else {
62                                 /* No specified colour conversion: assume the image is XYZ */
63                                 pf = AV_PIX_FMT_XYZ12LE;
64                         }
65                         /* We can't extract image size from a JPEG2000 codestream without decoding it,
66                            so pass in the image content's size here.
67                         */
68                         _image.reset (new J2KImageProxy (path, _image_content->video->size(), pf));
69                 } else {
70                         _image.reset (new FFmpegImageProxy (path));
71                 }
72         }
73
74         video->emit (_image, _frame_video_position);
75         ++_frame_video_position;
76         return false;
77 }
78
79 void
80 ImageDecoder::seek (ContentTime time, bool accurate)
81 {
82         Decoder::seek (time, accurate);
83         _frame_video_position = time.frames_round (_image_content->active_video_frame_rate ());
84 }