Forward-port raw_convert precision parameter from v0.
[libdcp.git] / src / mono_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 "mono_picture_mxf.h"
21 #include "mono_picture_mxf_writer.h"
22 #include "AS_DCP.h"
23 #include "KM_fileio.h"
24 #include "exceptions.h"
25 #include "mono_picture_frame.h"
26 #include "compose.hpp"
27
28 using std::string;
29 using std::vector;
30 using boost::shared_ptr;
31 using boost::dynamic_pointer_cast;
32 using namespace dcp;
33
34 MonoPictureMXF::MonoPictureMXF (boost::filesystem::path file)
35         : PictureMXF (file)
36 {
37         ASDCP::JP2K::MXFReader reader;
38         Kumu::Result_t r = reader.OpenRead (file.string().c_str());
39         if (ASDCP_FAILURE (r)) {
40                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file.string(), r));
41         }
42         
43         ASDCP::JP2K::PictureDescriptor desc;
44         if (ASDCP_FAILURE (reader.FillPictureDescriptor (desc))) {
45                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
46         }
47
48         read_picture_descriptor (desc);
49         
50         ASDCP::WriterInfo info;
51         if (ASDCP_FAILURE (reader.FillWriterInfo (info))) {
52                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
53         }
54
55         read_writer_info (info);
56 }
57
58 MonoPictureMXF::MonoPictureMXF (Fraction edit_rate)
59         : PictureMXF (edit_rate)
60 {
61         
62 }
63
64 shared_ptr<const MonoPictureFrame>
65 MonoPictureMXF::get_frame (int n) const
66 {
67         return shared_ptr<const MonoPictureFrame> (new MonoPictureFrame (_file, n, _decryption_context));
68 }
69
70 bool
71 MonoPictureMXF::equals (shared_ptr<const Content> other, EqualityOptions opt, boost::function<void (NoteType, string)> note) const
72 {
73         if (!MXF::equals (other, opt, note)) {
74                 return false;
75         }
76
77         ASDCP::JP2K::MXFReader reader_A;
78         Kumu::Result_t r = reader_A.OpenRead (_file.string().c_str());
79         if (ASDCP_FAILURE (r)) {
80                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", _file.string(), r));
81         }
82         
83         ASDCP::JP2K::MXFReader reader_B;
84         r = reader_B.OpenRead (other->file().string().c_str());
85         if (ASDCP_FAILURE (r)) {
86                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", other->file().string(), r));
87         }
88         
89         ASDCP::JP2K::PictureDescriptor desc_A;
90         if (ASDCP_FAILURE (reader_A.FillPictureDescriptor (desc_A))) {
91                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
92         }
93         ASDCP::JP2K::PictureDescriptor desc_B;
94         if (ASDCP_FAILURE (reader_B.FillPictureDescriptor (desc_B))) {
95                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
96         }
97         
98         if (!descriptor_equals (desc_A, desc_B, note)) {
99                 return false;
100         }
101
102         shared_ptr<const MonoPictureMXF> other_picture = dynamic_pointer_cast<const MonoPictureMXF> (other);
103         assert (other_picture);
104
105         for (int i = 0; i < _intrinsic_duration; ++i) {
106                 if (i >= other_picture->intrinsic_duration()) {
107                         return false;
108                 }
109                 
110                 note (DCP_PROGRESS, String::compose ("Comparing video frame %1 of %2", i, _intrinsic_duration));
111                 shared_ptr<const MonoPictureFrame> frame_A = get_frame (i);
112                 shared_ptr<const MonoPictureFrame> frame_B = other_picture->get_frame (i);
113                 
114                 if (!frame_buffer_equals (
115                             i, opt, note,
116                             frame_A->j2k_data(), frame_A->j2k_size(),
117                             frame_B->j2k_data(), frame_B->j2k_size()
118                             )) {
119                         return false;
120                 }
121         }
122
123         return true;
124 }
125
126 shared_ptr<PictureMXFWriter>
127 MonoPictureMXF::start_write (boost::filesystem::path file, Standard standard, bool overwrite)
128 {
129         /* XXX: can't we use shared_ptr here? */
130         return shared_ptr<MonoPictureMXFWriter> (new MonoPictureMXFWriter (this, file, standard, overwrite));
131 }
132
133 string
134 MonoPictureMXF::cpl_node_name () const
135 {
136         return "MainPicture";
137 }