Fix silly crash in shuffler.
[dcpomatic.git] / src / lib / shuffler.cc
1 /*
2     Copyright (C) 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 "shuffler.h"
22 #include "content_video.h"
23 #include "dcpomatic_assert.h"
24 #include <boost/foreach.hpp>
25 #include <iostream>
26
27 using std::make_pair;
28 using boost::weak_ptr;
29 using boost::shared_ptr;
30 using boost::optional;
31
32 struct Comparator
33 {
34         bool operator()(Shuffler::Store const & a, Shuffler::Store const & b) {
35                 if (a.second.frame != b.second.frame) {
36                         return a.second.frame < b.second.frame;
37                 }
38                 return a.second.eyes < b.second.eyes;
39         }
40 };
41
42 void
43 Shuffler::video (weak_ptr<Piece> weak_piece, ContentVideo video)
44 {
45         /* We should only ever see 3D_LEFT / 3D_RIGHT */
46         DCPOMATIC_ASSERT (video.eyes == EYES_LEFT || video.eyes == EYES_RIGHT);
47
48         shared_ptr<Piece> piece = weak_piece.lock ();
49         DCPOMATIC_ASSERT (piece);
50
51         if (!_last && video.eyes == EYES_LEFT) {
52                 /* We haven't seen anything since the last clear() and we have some eyes-left so assume everything is OK */
53                 Video (weak_piece, video);
54                 _last = video;
55                 return;
56         }
57
58         _store.push_back (make_pair (weak_piece, video));
59         _store.sort (Comparator());
60
61         while (true) {
62
63                 bool const store_front_in_sequence =
64                         !_store.empty() &&
65                         _last &&
66                         (
67                                 (_store.front().second.frame == _last->frame && _store.front().second.eyes == EYES_RIGHT && _last->eyes == EYES_LEFT) ||
68                                 (_store.front().second.frame == (_last->frame + 1) && _store.front().second.eyes == EYES_LEFT && _last->eyes == EYES_RIGHT)
69                                 );
70
71                 if (!store_front_in_sequence && _store.size() <= 8) {
72                         /* store_front_in_sequence means everything is ok; otherwise if the store is getting too big just
73                            start emitting things as best we can.  This can easily happen if, for example, there is only content
74                            for one eye in some part of the timeline.
75                         */
76                         break;
77                 }
78
79                 Video (_store.front().first, _store.front().second);
80                 _last = _store.front().second;
81                 _store.pop_front ();
82         }
83 }
84
85 void
86 Shuffler::clear ()
87 {
88         _store.clear ();
89         _last = optional<ContentVideo>();
90 }
91
92 void
93 Shuffler::flush ()
94 {
95         BOOST_FOREACH (Store i, _store) {
96                 Video (i.first, i.second);
97         }
98 }