Various tinkerings.
[libdcp.git] / src / stereo_picture_mxf.cc
1 /*
2     Copyright (C) 2012-2014 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 "AS_DCP.h"
21 #include "stereo_picture_mxf.h"
22 #include "stereo_picture_frame.h"
23 #include "exceptions.h"
24 #include "stereo_picture_mxf_writer.h"
25
26 using std::string;
27 using std::pair;
28 using std::make_pair;
29 using boost::shared_ptr;
30 using boost::dynamic_pointer_cast;
31 using namespace dcp;
32
33 StereoPictureMXF::StereoPictureMXF (boost::filesystem::path file)
34         : PictureMXF (file)
35 {
36         ASDCP::JP2K::MXFSReader reader;
37         Kumu::Result_t r = reader.OpenRead (file.string().c_str());
38         if (ASDCP_FAILURE (r)) {
39                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file.string(), r));
40         }
41         
42         ASDCP::JP2K::PictureDescriptor desc;
43         if (ASDCP_FAILURE (reader.FillPictureDescriptor (desc))) {
44                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
45         }
46
47         read_picture_descriptor (desc);
48 }
49
50 StereoPictureMXF::StereoPictureMXF (Fraction edit_rate)
51         : PictureMXF
52           (edit_rate)
53 {
54
55 }
56
57 shared_ptr<const StereoPictureFrame>
58 StereoPictureMXF::get_frame (int n) const
59 {
60         return shared_ptr<const StereoPictureFrame> (new StereoPictureFrame (file().string(), n));
61 }
62
63 shared_ptr<PictureMXFWriter>
64 StereoPictureMXF::start_write (boost::filesystem::path file, Standard standard, bool overwrite)
65 {
66         return shared_ptr<StereoPictureMXFWriter> (new StereoPictureMXFWriter (this, file, standard, overwrite));
67 }
68
69 bool
70 StereoPictureMXF::equals (shared_ptr<const Content> other, EqualityOptions opt, boost::function<void (NoteType, string)> note) const
71 {
72         if (!MXF::equals (other, opt, note)) {
73                 return false;
74         }
75
76         ASDCP::JP2K::MXFSReader reader_A;
77         Kumu::Result_t r = reader_A.OpenRead (file().string().c_str());
78         if (ASDCP_FAILURE (r)) {
79                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
80         }
81         
82         ASDCP::JP2K::MXFSReader reader_B;
83         r = reader_B.OpenRead (other->file().string().c_str());
84         if (ASDCP_FAILURE (r)) {
85                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
86         }
87         
88         ASDCP::JP2K::PictureDescriptor desc_A;
89         if (ASDCP_FAILURE (reader_A.FillPictureDescriptor (desc_A))) {
90                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
91         }
92         ASDCP::JP2K::PictureDescriptor desc_B;
93         if (ASDCP_FAILURE (reader_B.FillPictureDescriptor (desc_B))) {
94                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
95         }
96         
97         if (!descriptor_equals (desc_A, desc_B, note)) {
98                 return false;
99         }
100         
101         shared_ptr<const StereoPictureMXF> other_picture = dynamic_pointer_cast<const StereoPictureMXF> (other);
102         assert (other_picture);
103
104         for (int i = 0; i < _intrinsic_duration; ++i) {
105                 shared_ptr<const StereoPictureFrame> frame_A = get_frame (i);
106                 shared_ptr<const StereoPictureFrame> frame_B = other_picture->get_frame (i);
107                 
108                 if (!frame_buffer_equals (
109                             i, opt, note,
110                             frame_A->left_j2k_data(), frame_A->left_j2k_size(),
111                             frame_B->left_j2k_data(), frame_B->left_j2k_size()
112                             )) {
113                         return false;
114                 }
115                 
116                 if (!frame_buffer_equals (
117                             i, opt, note,
118                             frame_A->right_j2k_data(), frame_A->right_j2k_size(),
119                             frame_B->right_j2k_data(), frame_B->right_j2k_size()
120                             )) {
121                         return false;
122                 }
123         }
124
125         return true;
126 }