1bf2b1bef87737957f55dc92f873b4d761b82b5e
[dcpomatic.git] / src / lib / video_mxf_decoder.cc
1 /*
2     Copyright (C) 2016-2017 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 using boost::optional;
33 using namespace dcpomatic;
34
35 VideoMXFDecoder::VideoMXFDecoder (shared_ptr<const Film> film, shared_ptr<const VideoMXFContent> content)
36         : Decoder (film)
37         , _content (content)
38 {
39         video.reset (new VideoDecoder (this, content));
40
41         shared_ptr<dcp::MonoPictureAsset> mono;
42         try {
43                 mono.reset (new dcp::MonoPictureAsset (_content->path(0)));
44         } catch (dcp::MXFFileError& e) {
45                 /* maybe it's stereo */
46         } catch (dcp::DCPReadError& e) {
47                 /* maybe it's stereo */
48         }
49
50         shared_ptr<dcp::StereoPictureAsset> stereo;
51         try {
52                 stereo.reset (new dcp::StereoPictureAsset (_content->path(0)));
53         } catch (dcp::MXFFileError& e) {
54                 if (!mono) {
55                         throw;
56                 }
57         } catch (dcp::DCPReadError& e) {
58                 if (!mono) {
59                         throw;
60                 }
61         }
62
63         if (mono) {
64                 _mono_reader = mono->start_read ();
65                 _size = mono->size ();
66         } else {
67                 _stereo_reader = stereo->start_read ();
68                 _size = stereo->size ();
69         }
70 }
71
72 bool
73 VideoMXFDecoder::pass ()
74 {
75         double const vfr = _content->active_video_frame_rate (film());
76         int64_t const frame = _next.frames_round (vfr);
77
78
79         if (frame >= _content->video->length()) {
80                 return true;
81         }
82
83         if (_mono_reader) {
84                 video->emit (
85                         film(),
86                         shared_ptr<ImageProxy> (
87                                 new J2KImageProxy (_mono_reader->get_frame(frame), _size, AV_PIX_FMT_XYZ12LE, optional<int>())
88                                 ),
89                         frame
90                         );
91         } else {
92                 video->emit (
93                         film(),
94                         shared_ptr<ImageProxy> (
95                                 new J2KImageProxy (_stereo_reader->get_frame(frame), _size, dcp::EYE_LEFT, AV_PIX_FMT_XYZ12LE, optional<int>())
96                                 ),
97                         frame
98                         );
99                 video->emit (
100                         film(),
101                         shared_ptr<ImageProxy> (
102                                 new J2KImageProxy (_stereo_reader->get_frame(frame), _size, dcp::EYE_RIGHT, AV_PIX_FMT_XYZ12LE, optional<int>())
103                                 ),
104                         frame
105                         );
106         }
107
108         _next += ContentTime::from_frames (1, vfr);
109         return false;
110 }
111
112 void
113 VideoMXFDecoder::seek (ContentTime t, bool accurate)
114 {
115         Decoder::seek (t, accurate);
116         _next = t;
117 }