src/mono_picture_asset.cc

   1 /*
   2     Copyright (C) 2012-2021 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
  35 /** @file  src/mono_picture_asset.cc
  36  *  @brief MonoPictureAsset class
  37  */
  38
  39
  40 #include "compose.hpp"
  41 #include "dcp_assert.h"
  42 #include "equality_options.h"
  43 #include "exceptions.h"
  44 #include "filesystem.h"
  45 #include "mono_picture_asset.h"
  46 #include "mono_picture_asset_reader.h"
  47 #include "mono_picture_asset_writer.h"
  48 #include "mono_picture_frame.h"
  49 #include <asdcp/AS_DCP.h>
  50 #include <asdcp/KM_fileio.h>
  51
  52
  53 using std::dynamic_pointer_cast;
  54 using std::list;
  55 using std::make_shared;
  56 using std::pair;
  57 using std::shared_ptr;
  58 using std::string;
  59 using std::vector;
  60 #if BOOST_VERSION >= 106100
  61 using namespace boost::placeholders;
  62 #endif
  63 using namespace dcp;
  64
  65
  66 MonoPictureAsset::MonoPictureAsset (boost::filesystem::path file)
  67         : PictureAsset (file)
  68 {
  69         ASDCP::JP2K::MXFReader reader;
  70         auto r = reader.OpenRead(dcp::filesystem::fix_long_path(file).string().c_str());
  71         if (ASDCP_FAILURE(r)) {
  72                 boost::throw_exception (MXFFileError("could not open MXF file for reading", file.string(), r));
  73         }
  74
  75         ASDCP::JP2K::PictureDescriptor desc;
  76         if (ASDCP_FAILURE (reader.FillPictureDescriptor(desc))) {
  77                 boost::throw_exception (ReadError("could not read video MXF information"));
  78         }
  79
  80         read_picture_descriptor (desc);
  81
  82         ASDCP::WriterInfo info;
  83         if (ASDCP_FAILURE (reader.FillWriterInfo (info))) {
  84                 boost::throw_exception (ReadError("could not read video MXF information"));
  85         }
  86
  87         _id = read_writer_info (info);
  88 }
  89
  90
  91 MonoPictureAsset::MonoPictureAsset (Fraction edit_rate, Standard standard)
  92         : PictureAsset (edit_rate, standard)
  93 {
  94
  95 }
  96
  97
  98 static void
  99 storing_note_handler (list<pair<NoteType, string>>& notes, NoteType t, string s)
 100 {
 101         notes.push_back (make_pair (t, s));
 102 }
 103
 104
 105 bool
 106 MonoPictureAsset::equals(shared_ptr<const Asset> other, EqualityOptions const& opt, NoteHandler note) const
 107 {
 108         if (!dynamic_pointer_cast<const MonoPictureAsset>(other)) {
 109                 return false;
 110         }
 111
 112         ASDCP::JP2K::MXFReader reader_A;
 113         DCP_ASSERT (_file);
 114         auto r = reader_A.OpenRead(dcp::filesystem::fix_long_path(*_file).string().c_str());
 115         if (ASDCP_FAILURE(r)) {
 116                 boost::throw_exception (MXFFileError("could not open MXF file for reading", _file->string(), r));
 117         }
 118
 119         ASDCP::JP2K::MXFReader reader_B;
 120         DCP_ASSERT (other->file ());
 121         r = reader_B.OpenRead(dcp::filesystem::fix_long_path(*other->file()).string().c_str());
 122         if (ASDCP_FAILURE (r)) {
 123                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", other->file()->string(), r));
 124         }
 125
 126         ASDCP::JP2K::PictureDescriptor desc_A;
 127         if (ASDCP_FAILURE (reader_A.FillPictureDescriptor (desc_A))) {
 128                 boost::throw_exception (ReadError ("could not read video MXF information"));
 129         }
 130         ASDCP::JP2K::PictureDescriptor desc_B;
 131         if (ASDCP_FAILURE (reader_B.FillPictureDescriptor (desc_B))) {
 132                 boost::throw_exception (ReadError ("could not read video MXF information"));
 133         }
 134
 135         if (!descriptor_equals (desc_A, desc_B, note)) {
 136                 return false;
 137         }
 138
 139         auto other_picture = dynamic_pointer_cast<const MonoPictureAsset> (other);
 140         DCP_ASSERT (other_picture);
 141
 142         bool result = true;
 143
 144         auto reader = start_read ();
 145         auto other_reader = other_picture->start_read ();
 146
 147 #ifdef LIBDCP_OPENMP
 148 #pragma omp parallel for
 149 #endif
 150
 151         for (int i = 0; i < _intrinsic_duration; ++i) {
 152                 if (i >= other_picture->intrinsic_duration()) {
 153                         result = false;
 154                 }
 155
 156                 if (result || opt.keep_going) {
 157
 158                         auto frame_A = reader->get_frame (i);
 159                         auto frame_B = other_reader->get_frame (i);
 160
 161                         list<pair<NoteType, string>> notes;
 162
 163                         if (!frame_buffer_equals (
 164                                     i, opt, bind (&storing_note_handler, boost::ref(notes), _1, _2),
 165                                     frame_A->data(), frame_A->size(),
 166                                     frame_B->data(), frame_B->size()
 167                                     )) {
 168                                 result = false;
 169                         }
 170
 171 #ifdef LIBDCP_OPENMP
 172 #pragma omp critical
 173 #endif
 174                         {
 175                                 note (NoteType::PROGRESS, String::compose("Compared video frame %1 of %2", i, _intrinsic_duration));
 176                                 for (auto const& i: notes) {
 177                                         note (i.first, i.second);
 178                                 }
 179                         }
 180                 }
 181         }
 182
 183         return result;
 184 }
 185
 186
 187 shared_ptr<PictureAssetWriter>
 188 MonoPictureAsset::start_write(boost::filesystem::path file, Behaviour behaviour)
 189 {
 190         /* Can't use make_shared here as the MonoPictureAssetWriter constructor is private */
 191         return shared_ptr<MonoPictureAssetWriter>(new MonoPictureAssetWriter(this, file, behaviour == Behaviour::OVERWRITE_EXISTING));
 192 }
 193
 194 shared_ptr<MonoPictureAssetReader>
 195 MonoPictureAsset::start_read () const
 196 {
 197         /* Can't use make_shared here as the MonoPictureAssetReader constructor is private */
 198         return shared_ptr<MonoPictureAssetReader>(new MonoPictureAssetReader(this, key(), standard()));
 199
 200 }
 201
 202 string
 203 MonoPictureAsset::cpl_node_name () const
 204 {
 205         return "MainPicture";
 206 }