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