Various tweaks; work on read_dcp example.
[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         ASDCP::WriterInfo info;
50         if (ASDCP_FAILURE (reader.FillWriterInfo (info))) {
51                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
52         }
53
54         read_writer_info (info);
55 }
56
57 StereoPictureMXF::StereoPictureMXF (Fraction edit_rate)
58         : PictureMXF
59           (edit_rate)
60 {
61
62 }
63
64 shared_ptr<const StereoPictureFrame>
65 StereoPictureMXF::get_frame (int n) const
66 {
67         return shared_ptr<const StereoPictureFrame> (new StereoPictureFrame (file().string(), n));
68 }
69
70 shared_ptr<PictureMXFWriter>
71 StereoPictureMXF::start_write (boost::filesystem::path file, Standard standard, bool overwrite)
72 {
73         return shared_ptr<StereoPictureMXFWriter> (new StereoPictureMXFWriter (this, file, standard, overwrite));
74 }
75
76 bool
77 StereoPictureMXF::equals (shared_ptr<const Content> other, EqualityOptions opt, boost::function<void (NoteType, string)> note) const
78 {
79         if (!MXF::equals (other, opt, note)) {
80                 return false;
81         }
82
83         ASDCP::JP2K::MXFSReader reader_A;
84         Kumu::Result_t r = reader_A.OpenRead (file().string().c_str());
85         if (ASDCP_FAILURE (r)) {
86                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
87         }
88         
89         ASDCP::JP2K::MXFSReader reader_B;
90         r = reader_B.OpenRead (other->file().string().c_str());
91         if (ASDCP_FAILURE (r)) {
92                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
93         }
94         
95         ASDCP::JP2K::PictureDescriptor desc_A;
96         if (ASDCP_FAILURE (reader_A.FillPictureDescriptor (desc_A))) {
97                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
98         }
99         ASDCP::JP2K::PictureDescriptor desc_B;
100         if (ASDCP_FAILURE (reader_B.FillPictureDescriptor (desc_B))) {
101                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
102         }
103         
104         if (!descriptor_equals (desc_A, desc_B, note)) {
105                 return false;
106         }
107         
108         shared_ptr<const StereoPictureMXF> other_picture = dynamic_pointer_cast<const StereoPictureMXF> (other);
109         assert (other_picture);
110
111         for (int i = 0; i < _intrinsic_duration; ++i) {
112                 shared_ptr<const StereoPictureFrame> frame_A = get_frame (i);
113                 shared_ptr<const StereoPictureFrame> frame_B = other_picture->get_frame (i);
114                 
115                 if (!frame_buffer_equals (
116                             i, opt, note,
117                             frame_A->left_j2k_data(), frame_A->left_j2k_size(),
118                             frame_B->left_j2k_data(), frame_B->left_j2k_size()
119                             )) {
120                         return false;
121                 }
122                 
123                 if (!frame_buffer_equals (
124                             i, opt, note,
125                             frame_A->right_j2k_data(), frame_A->right_j2k_size(),
126                             frame_B->right_j2k_data(), frame_B->right_j2k_size()
127                             )) {
128                         return false;
129                 }
130         }
131
132         return true;
133 }