Fix handling of incorrectly-recognised JPEG2000 files.
[dcpomatic.git] / src / lib / video_mxf_decoder.cc
1 /*
2     Copyright (C) 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 "video_mxf_decoder.h"
22 #include "video_decoder.h"
23 #include "video_mxf_content.h"
24 #include "j2k_image_proxy.h"
25 #include <dcp/mono_picture_asset.h>
26 #include <dcp/mono_picture_asset_reader.h>
27 #include <dcp/stereo_picture_asset.h>
28 #include <dcp/stereo_picture_asset_reader.h>
29 #include <dcp/exceptions.h>
30
31 using boost::shared_ptr;
32
33 VideoMXFDecoder::VideoMXFDecoder (shared_ptr<const VideoMXFContent> content, shared_ptr<Log> log)
34         : _content (content)
35 {
36         video.reset (new VideoDecoder (this, content, log));
37
38         shared_ptr<dcp::MonoPictureAsset> mono;
39         try {
40                 mono.reset (new dcp::MonoPictureAsset (_content->path(0)));
41         } catch (dcp::MXFFileError& e) {
42                 /* maybe it's stereo */
43         } catch (dcp::DCPReadError& e) {
44                 /* maybe it's stereo */
45         }
46
47         shared_ptr<dcp::StereoPictureAsset> stereo;
48         try {
49                 stereo.reset (new dcp::StereoPictureAsset (_content->path(0)));
50         } catch (dcp::MXFFileError& e) {
51                 if (!mono) {
52                         throw;
53                 }
54         } catch (dcp::DCPReadError& e) {
55                 if (!mono) {
56                         throw;
57                 }
58         }
59
60         if (mono) {
61                 _mono_reader = mono->start_read ();
62                 _size = mono->size ();
63         } else {
64                 _stereo_reader = stereo->start_read ();
65                 _size = stereo->size ();
66         }
67 }
68
69 bool
70 VideoMXFDecoder::pass (PassReason, bool)
71 {
72         double const vfr = _content->active_video_frame_rate ();
73         int64_t const frame = _next.frames_round (vfr);
74
75         if (frame >= _content->video->length()) {
76                 return true;
77         }
78
79         if (_mono_reader) {
80                 video->give (
81                         shared_ptr<ImageProxy> (new J2KImageProxy (_mono_reader->get_frame(frame), _size, AV_PIX_FMT_XYZ12LE)), frame
82                         );
83         } else {
84                 video->give (
85                         shared_ptr<ImageProxy> (new J2KImageProxy (_stereo_reader->get_frame(frame), _size, dcp::EYE_LEFT, AV_PIX_FMT_XYZ12LE)), frame
86                         );
87                 video->give (
88                         shared_ptr<ImageProxy> (new J2KImageProxy (_stereo_reader->get_frame(frame), _size, dcp::EYE_RIGHT, AV_PIX_FMT_XYZ12LE)), frame
89                         );
90         }
91
92         _next += ContentTime::from_frames (1, vfr);
93         return false;
94 }
95
96 void
97 VideoMXFDecoder::seek (ContentTime t, bool accurate)
98 {
99         video->seek (t, accurate);
100         _next = t;
101 }