src/mono_picture_asset.cc

   1 /*
   2     Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of libdcp.
   5
   6     libdcp 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     libdcp 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 libdcp.  If not, see <http://www.gnu.org/licenses/>.
  18
  19     In addition, as a special exception, the copyright holders give
  20     permission to link the code of portions of this program with the
  21     OpenSSL library under certain conditions as described in each
  22     individual source file, and distribute linked combinations
  23     including the two.
  24
  25     You must obey the GNU General Public License in all respects
  26     for all of the code used other than OpenSSL.  If you modify
  27     file(s) with this exception, you may extend this exception to your
  28     version of the file(s), but you are not obligated to do so.  If you
  29     do not wish to do so, delete this exception statement from your
  30     version.  If you delete this exception statement from all source
  31     files in the program, then also delete it here.
  32 */
  33
  34 #include "mono_picture_asset.h"
  35 #include "mono_picture_asset_writer.h"
  36 #include "mono_picture_asset_reader.h"
  37 #include "exceptions.h"
  38 #include "dcp_assert.h"
  39 #include "mono_picture_frame.h"
  40 #include "compose.hpp"
  41 #include <asdcp/AS_DCP.h>
  42 #include <asdcp/KM_fileio.h>
  43
  44 using std::string;
  45 using std::vector;
  46 using std::list;
  47 using std::pair;
  48 using boost::shared_ptr;
  49 using boost::dynamic_pointer_cast;
  50 using namespace dcp;
  51
  52 MonoPictureAsset::MonoPictureAsset (boost::filesystem::path file)
  53         : PictureAsset (file)
  54 {
  55         ASDCP::JP2K::MXFReader reader;
  56         Kumu::Result_t r = reader.OpenRead (file.string().c_str());
  57         if (ASDCP_FAILURE (r)) {
  58                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file.string(), r));
  59         }
  60
  61         ASDCP::JP2K::PictureDescriptor desc;
  62         if (ASDCP_FAILURE (reader.FillPictureDescriptor (desc))) {
  63                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
  64         }
  65
  66         read_picture_descriptor (desc);
  67
  68         ASDCP::WriterInfo info;
  69         if (ASDCP_FAILURE (reader.FillWriterInfo (info))) {
  70                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
  71         }
  72
  73         _id = read_writer_info (info);
  74 }
  75
  76 MonoPictureAsset::MonoPictureAsset (Fraction edit_rate)
  77         : PictureAsset (edit_rate)
  78 {
  79
  80 }
  81
  82 static void
  83 storing_note_handler (list<pair<NoteType, string> >& notes, NoteType t, string s)
  84 {
  85         notes.push_back (make_pair (t, s));
  86 }
  87
  88 bool
  89 MonoPictureAsset::equals (shared_ptr<const Asset> other, EqualityOptions opt, NoteHandler note) const
  90 {
  91         if (!dynamic_pointer_cast<const MonoPictureAsset> (other)) {
  92                 return false;
  93         }
  94
  95         ASDCP::JP2K::MXFReader reader_A;
  96         DCP_ASSERT (_file);
  97         Kumu::Result_t r = reader_A.OpenRead (_file->string().c_str());
  98         if (ASDCP_FAILURE (r)) {
  99                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", _file->string(), r));
 100         }
 101
 102         ASDCP::JP2K::MXFReader reader_B;
 103         DCP_ASSERT (other->file ());
 104         r = reader_B.OpenRead (other->file()->string().c_str());
 105         if (ASDCP_FAILURE (r)) {
 106                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", other->file()->string(), r));
 107         }
 108
 109         ASDCP::JP2K::PictureDescriptor desc_A;
 110         if (ASDCP_FAILURE (reader_A.FillPictureDescriptor (desc_A))) {
 111                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
 112         }
 113         ASDCP::JP2K::PictureDescriptor desc_B;
 114         if (ASDCP_FAILURE (reader_B.FillPictureDescriptor (desc_B))) {
 115                 boost::throw_exception (DCPReadError ("could not read video MXF information"));
 116         }
 117
 118         if (!descriptor_equals (desc_A, desc_B, note)) {
 119                 return false;
 120         }
 121
 122         shared_ptr<const MonoPictureAsset> other_picture = dynamic_pointer_cast<const MonoPictureAsset> (other);
 123         DCP_ASSERT (other_picture);
 124
 125         bool result = true;
 126
 127 #ifdef LIBDCP_OPENMP
 128 #pragma omp parallel for
 129 #endif
 130
 131         shared_ptr<MonoPictureAssetReader> reader = start_read ();
 132         shared_ptr<MonoPictureAssetReader> other_reader = other_picture->start_read ();
 133
 134         for (int i = 0; i < _intrinsic_duration; ++i) {
 135                 if (i >= other_picture->intrinsic_duration()) {
 136                         result = false;
 137                 }
 138
 139                 if (result || opt.keep_going) {
 140
 141                         shared_ptr<const MonoPictureFrame> frame_A = reader->get_frame (i);
 142                         shared_ptr<const MonoPictureFrame> frame_B = other_reader->get_frame (i);
 143
 144                         list<pair<NoteType, string> > notes;
 145
 146                         if (!frame_buffer_equals (
 147                                     i, opt, bind (&storing_note_handler, boost::ref(notes), _1, _2),
 148                                     frame_A->j2k_data(), frame_A->j2k_size(),
 149                                     frame_B->j2k_data(), frame_B->j2k_size()
 150                                     )) {
 151                                 result = false;
 152                         }
 153
 154 #ifdef LIBDCP_OPENMP
 155 #pragma omp critical
 156 #endif
 157                         {
 158                                 note (DCP_PROGRESS, String::compose ("Compared video frame %1 of %2", i, _intrinsic_duration));
 159                                 for (list<pair<NoteType, string> >::const_iterator i = notes.begin(); i != notes.end(); ++i) {
 160                                         note (i->first, i->second);
 161                                 }
 162                         }
 163                 }
 164         }
 165
 166         return result;
 167 }
 168
 169 shared_ptr<PictureAssetWriter>
 170 MonoPictureAsset::start_write (boost::filesystem::path file, Standard standard, bool overwrite)
 171 {
 172         /* XXX: can't we use shared_ptr here? */
 173         return shared_ptr<MonoPictureAssetWriter> (new MonoPictureAssetWriter (this, file, standard, overwrite));
 174 }
 175
 176 shared_ptr<MonoPictureAssetReader>
 177 MonoPictureAsset::start_read () const
 178 {
 179         return shared_ptr<MonoPictureAssetReader> (new MonoPictureAssetReader (this, key ()));
 180 }
 181
 182 string
 183 MonoPictureAsset::cpl_node_name () const
 184 {
 185         return "MainPicture";
 186 }