Try to rationalise handling of urn:uuid: prefixes.
[libdcp.git] / src / reel.cc
1 /*
2     Copyright (C) 2014-2015 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include "reel.h"
21 #include "util.h"
22 #include "picture_asset.h"
23 #include "mono_picture_asset.h"
24 #include "stereo_picture_asset.h"
25 #include "sound_asset.h"
26 #include "subtitle_asset.h"
27 #include "reel_mono_picture_asset.h"
28 #include "reel_stereo_picture_asset.h"
29 #include "reel_sound_asset.h"
30 #include "reel_subtitle_asset.h"
31 #include "decrypted_kdm_key.h"
32 #include "decrypted_kdm.h"
33 #include "interop_subtitle_asset.h"
34 #include <libxml++/nodes/element.h>
35
36 using std::string;
37 using std::list;
38 using std::cout;
39 using std::max;
40 using boost::shared_ptr;
41 using boost::dynamic_pointer_cast;
42 using namespace dcp;
43
44 Reel::Reel (boost::shared_ptr<const cxml::Node> node)
45         : Object (remove_urn_uuid (node->string_child ("Id")))
46 {
47         shared_ptr<cxml::Node> asset_list = node->node_child ("AssetList");
48
49         shared_ptr<cxml::Node> main_picture = asset_list->optional_node_child ("MainPicture");
50         if (main_picture) {
51                 _main_picture.reset (new ReelMonoPictureAsset (main_picture));
52         }
53
54         shared_ptr<cxml::Node> main_stereoscopic_picture = asset_list->optional_node_child ("MainStereoscopicPicture");
55         if (main_stereoscopic_picture) {
56                 _main_picture.reset (new ReelStereoPictureAsset (main_stereoscopic_picture));
57         }
58
59         shared_ptr<cxml::Node> main_sound = asset_list->optional_node_child ("MainSound");
60         if (main_sound) {
61                 _main_sound.reset (new ReelSoundAsset (main_sound));
62         }
63
64         shared_ptr<cxml::Node> main_subtitle = asset_list->optional_node_child ("MainSubtitle");
65         if (main_subtitle) {
66                 _main_subtitle.reset (new ReelSubtitleAsset (main_subtitle));
67         }
68
69         node->ignore_child ("AnnotationText");
70         node->done ();
71 }
72
73 void
74 Reel::write_to_cpl (xmlpp::Element* node, Standard standard) const
75 {
76         xmlpp::Element* reel = node->add_child ("Reel");
77         reel->add_child("Id")->add_child_text ("urn:uuid:" + make_uuid());
78         xmlpp::Element* asset_list = reel->add_child ("AssetList");
79
80         if (_main_picture && dynamic_pointer_cast<ReelMonoPictureAsset> (_main_picture)) {
81                 /* Mono pictures come before other stuff... */
82                 _main_picture->write_to_cpl (asset_list, standard);
83         }
84
85         if (_main_sound) {
86                 _main_sound->write_to_cpl (asset_list, standard);
87         }
88
89         if (_main_subtitle) {
90                 _main_subtitle->write_to_cpl (asset_list, standard);
91         }
92
93         if (_main_picture && dynamic_pointer_cast<ReelStereoPictureAsset> (_main_picture)) {
94                 /* ... but stereo pictures must come after */
95                 _main_picture->write_to_cpl (asset_list, standard);
96         }
97 }
98
99 bool
100 Reel::equals (boost::shared_ptr<const Reel> other, EqualityOptions opt, NoteHandler note) const
101 {
102         if ((_main_picture && !other->_main_picture) || (!_main_picture && other->_main_picture)) {
103                 note (DCP_ERROR, "Reel: assets differ");
104                 return false;
105         }
106
107         if (_main_picture && !_main_picture->equals (other->_main_picture, opt, note)) {
108                 return false;
109         }
110
111         if ((_main_sound && !other->_main_sound) || (!_main_sound && other->_main_sound)) {
112                 note (DCP_ERROR, "Reel: assets differ");
113                 return false;
114         }
115
116         if (_main_sound && !_main_sound->equals (other->_main_sound, opt, note)) {
117                 return false;
118         }
119
120         if ((_main_subtitle && !other->_main_subtitle) || (!_main_subtitle && other->_main_subtitle)) {
121                 note (DCP_ERROR, "Reel: assets differ");
122                 return false;
123         }
124
125         if (_main_subtitle && !_main_subtitle->equals (other->_main_subtitle, opt, note)) {
126                 return false;
127         }
128
129         return true;
130 }
131
132 bool
133 Reel::encrypted () const
134 {
135         return ((_main_picture && _main_picture->encrypted ()) || (_main_sound && _main_sound->encrypted ()));
136 }
137
138 void
139 Reel::add (DecryptedKDM const & kdm)
140 {
141         list<DecryptedKDMKey> keys = kdm.keys ();
142
143         for (list<DecryptedKDMKey>::iterator i = keys.begin(); i != keys.end(); ++i) {
144                 if (i->id() == _main_picture->key_id()) {
145                         _main_picture->asset()->set_key (i->key ());
146                 }
147                 if (i->id() == _main_sound->key_id()) {
148                         _main_sound->asset()->set_key (i->key ());
149                 }
150         }
151 }
152
153 void
154 Reel::add (shared_ptr<ReelAsset> asset)
155 {
156         shared_ptr<ReelPictureAsset> p = dynamic_pointer_cast<ReelPictureAsset> (asset);
157         shared_ptr<ReelSoundAsset> so = dynamic_pointer_cast<ReelSoundAsset> (asset);
158         shared_ptr<ReelSubtitleAsset> su = dynamic_pointer_cast<ReelSubtitleAsset> (asset);
159         if (p) {
160                 _main_picture = p;
161         } else if (so) {
162                 _main_sound = so;
163         } else if (su) {
164                 _main_subtitle = su;
165         }
166 }
167
168 void
169 Reel::resolve_refs (list<shared_ptr<Asset> > assets)
170 {
171         if (_main_picture) {
172                 _main_picture->asset_ref().resolve (assets);
173         }
174
175         if (_main_sound) {
176                 _main_sound->asset_ref().resolve (assets);
177         }
178
179         if (_main_subtitle) {
180                 _main_subtitle->asset_ref().resolve (assets);
181
182                 /* Interop subtitle handling is all special cases */
183                 shared_ptr<InteropSubtitleAsset> iop = dynamic_pointer_cast<InteropSubtitleAsset> (_main_subtitle->asset_ref().asset ());
184                 if (iop) {
185                         iop->resolve_fonts (assets);
186                 }
187         }
188 }
189
190 int64_t
191 Reel::duration () const
192 {
193         int64_t d = 0;
194
195         if (_main_picture) {
196                 d = max (d, _main_picture->duration ());
197         }
198         if (_main_sound) {
199                 d = max (d, _main_sound->duration ());
200         }
201         if (_main_subtitle) {
202                 d = max (d, _main_subtitle->duration ());
203         }
204
205         return d;
206 }