613ec9c7311f8c162b34b7306c341f8c0ad4fdd7
[dcpomatic.git] / src / lib / video_decoder.cc
1 /*
2     Copyright (C) 2012-2018 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_decoder.h"
22 #include "image.h"
23 #include "raw_image_proxy.h"
24 #include "film.h"
25 #include "log.h"
26 #include "compose.hpp"
27 #include <boost/foreach.hpp>
28 #include <iostream>
29
30 #include "i18n.h"
31
32 using std::cout;
33 using std::list;
34 using std::max;
35 using std::back_inserter;
36 using boost::shared_ptr;
37 using boost::optional;
38
39 VideoDecoder::VideoDecoder (Decoder* parent, shared_ptr<const Content> c)
40         : DecoderPart (parent)
41         , _content (c)
42 {
43
44 }
45
46 /** Called by decoder classes when they have a video frame ready.
47  *  @param frame Frame index within the content; this does not take into account 3D
48  *  so for 3D_ALTERNATE this value goes:
49  *     0: frame 0 left
50  *     1: frame 0 right
51  *     2: frame 1 left
52  *     3: frame 1 right
53  *  and so on.
54  */
55 void
56 VideoDecoder::emit (shared_ptr<const Film> film, shared_ptr<const ImageProxy> image, Frame decoder_frame)
57 {
58         if (ignore ()) {
59                 return;
60         }
61
62         double const afr = _content->active_video_frame_rate(film);
63         VideoFrameType const vft = _content->video->frame_type();
64
65         Frame frame;
66         if (!_position) {
67                 /* This is the first data we have received since initialisation or seek.  Set
68                    the position based on the frame that was given.  After this first time
69                    we just cound frames, since (as with audio) it seems that ContentTimes
70                    are unreliable from FFmpegDecoder.  They are much better than audio times
71                    but still we get the occasional one which is duplicated.  In this case
72                    ffmpeg seems to carry on regardless, processing the video frame as normal.
73                    If we drop the frame with the duplicated timestamp we obviously lose sync.
74                 */
75                 _position = ContentTime::from_frames (decoder_frame, afr);
76                 if (vft == VIDEO_FRAME_TYPE_3D) {
77                         frame = decoder_frame;
78                         _last_emitted_eyes = EYES_RIGHT;
79                 } else if (vft == VIDEO_FRAME_TYPE_3D_ALTERNATE) {
80                         frame = decoder_frame / 2;
81                         _last_emitted_eyes = EYES_RIGHT;
82                 } else {
83                         frame = decoder_frame;
84                 }
85         } else {
86                 if (vft == VIDEO_FRAME_TYPE_3D || vft == VIDEO_FRAME_TYPE_3D_ALTERNATE) {
87                         DCPOMATIC_ASSERT (_last_emitted_eyes);
88                         if (_last_emitted_eyes.get() == EYES_RIGHT) {
89                                 frame = _position->frames_round(afr) + 1;
90                         } else {
91                                 frame = _position->frames_round(afr);
92                         }
93                 } else {
94                         frame = _position->frames_round(afr) + 1;
95                 }
96         }
97
98         switch (_content->video->frame_type ()) {
99         case VIDEO_FRAME_TYPE_2D:
100                 Data (ContentVideo (image, frame, EYES_BOTH, PART_WHOLE));
101                 break;
102         case VIDEO_FRAME_TYPE_3D:
103         {
104                 /* We should receive the same frame index twice for 3D; hence we know which
105                    frame this one is.
106                 */
107                 bool const same = (_last_emitted_frame && _last_emitted_frame.get() == frame);
108                 if (!same && _last_emitted_eyes && *_last_emitted_eyes == EYES_LEFT) {
109                         /* We just got a new frame index but the last frame was left-eye; it looks like
110                            this content is not really 3D.
111                         */
112                         boost::throw_exception (
113                                 DecodeError(
114                                         String::compose(
115                                                 _("The content file %1 is set as 3D but does not appear to contain 3D images.  Please set it to 2D.  "
116                                                   "You can still make a 3D DCP from this content by ticking the 3D option in the DCP video tab."),
117                                                 _content->path(0)
118                                                 )
119                                         )
120                                 );
121                 }
122                 Eyes const eyes = same ? EYES_RIGHT : EYES_LEFT;
123                 Data (ContentVideo (image, frame, eyes, PART_WHOLE));
124                 _last_emitted_frame = frame;
125                 _last_emitted_eyes = eyes;
126                 break;
127         }
128         case VIDEO_FRAME_TYPE_3D_ALTERNATE:
129         {
130                 DCPOMATIC_ASSERT (_last_emitted_eyes);
131                 Eyes const eyes = _last_emitted_eyes.get() == EYES_LEFT ? EYES_RIGHT : EYES_LEFT;
132                 Data (ContentVideo (image, frame, eyes, PART_WHOLE));
133                 _last_emitted_eyes = eyes;
134                 break;
135         }
136         case VIDEO_FRAME_TYPE_3D_LEFT_RIGHT:
137                 Data (ContentVideo (image, frame, EYES_LEFT, PART_LEFT_HALF));
138                 Data (ContentVideo (image, frame, EYES_RIGHT, PART_RIGHT_HALF));
139                 break;
140         case VIDEO_FRAME_TYPE_3D_TOP_BOTTOM:
141                 Data (ContentVideo (image, frame, EYES_LEFT, PART_TOP_HALF));
142                 Data (ContentVideo (image, frame, EYES_RIGHT, PART_BOTTOM_HALF));
143                 break;
144         case VIDEO_FRAME_TYPE_3D_LEFT:
145                 Data (ContentVideo (image, frame, EYES_LEFT, PART_WHOLE));
146                 break;
147         case VIDEO_FRAME_TYPE_3D_RIGHT:
148                 Data (ContentVideo (image, frame, EYES_RIGHT, PART_WHOLE));
149                 break;
150         default:
151                 DCPOMATIC_ASSERT (false);
152         }
153
154         _position = ContentTime::from_frames (frame, afr);
155 }
156
157 void
158 VideoDecoder::seek ()
159 {
160         _position = boost::none;
161         _last_emitted_frame.reset ();
162         _last_emitted_eyes.reset ();
163 }