src/picture_asset_writer.cc

   1 /*
   2     Copyright (C) 2012-2013 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 "KM_fileio.h"
  22 #include "picture_asset_writer.h"
  23 #include "exceptions.h"
  24 #include "picture_asset.h"
  25
  26 using std::istream;
  27 using std::ostream;
  28 using std::string;
  29 using boost::shared_ptr;
  30 using namespace libdcp;
  31
  32 FrameInfo::FrameInfo (istream& s)
  33         : offset (0)
  34         , size (0)
  35 {
  36         s >> offset >> size;
  37
  38         if (!s.good ()) {
  39                 /* Make sure we zero these if something bad happened, otherwise
  40                    the caller might try to alloc lots of RAM.
  41                 */
  42                 offset = size = 0;
  43         }
  44
  45         s >> hash;
  46 }
  47
  48 void
  49 FrameInfo::write (ostream& s)
  50 {
  51         s << offset << " " << size << " " << hash;
  52 }
  53
  54
  55 PictureAssetWriter::PictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata)
  56         : _asset (asset)
  57         , _frames_written (0)
  58         , _started (false)
  59         , _finalized (false)
  60         , _overwrite (overwrite)
  61         , _metadata (metadata)
  62 {
  63
  64 }
  65
  66 struct ASDCPStateBase
  67 {
  68         ASDCPStateBase ()
  69                 : frame_buffer (4 * Kumu::Megabyte)
  70         {}
  71
  72         ASDCP::JP2K::CodestreamParser j2k_parser;
  73         ASDCP::JP2K::FrameBuffer frame_buffer;
  74         ASDCP::WriterInfo writer_info;
  75         ASDCP::JP2K::PictureDescriptor picture_descriptor;
  76 };
  77
  78 struct MonoPictureAssetWriter::ASDCPState : public ASDCPStateBase
  79 {
  80         ASDCP::JP2K::MXFWriter mxf_writer;
  81 };
  82
  83 struct StereoPictureAssetWriter::ASDCPState : public ASDCPStateBase
  84 {
  85         ASDCP::JP2K::MXFSWriter mxf_writer;
  86 };
  87
  88 /** @param a Asset to write to.  `a' must not be deleted while
  89  *  this writer class still exists, or bad things will happen.
  90  */
  91 MonoPictureAssetWriter::MonoPictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata)
  92         : PictureAssetWriter (asset, overwrite, metadata)
  93         , _state (new MonoPictureAssetWriter::ASDCPState)
  94 {
  95
  96 }
  97
  98 StereoPictureAssetWriter::StereoPictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata)
  99         : PictureAssetWriter (asset, overwrite, metadata)
 100         , _state (new StereoPictureAssetWriter::ASDCPState)
 101         , _next_eye (EYE_LEFT)
 102 {
 103
 104 }
 105
 106 template <class P, class Q>
 107 void libdcp::start (PictureAssetWriter* writer, shared_ptr<P> state, Q* asset, uint8_t* data, int size)
 108 {
 109         if (ASDCP_FAILURE (state->j2k_parser.OpenReadFrame (data, size, state->frame_buffer))) {
 110                 boost::throw_exception (MiscError ("could not parse J2K frame"));
 111         }
 112
 113         state->j2k_parser.FillPictureDescriptor (state->picture_descriptor);
 114         state->picture_descriptor.EditRate = ASDCP::Rational (asset->edit_rate(), 1);
 115
 116         asset->fill_writer_info (&state->writer_info, asset->uuid(), writer->_metadata);
 117
 118         if (ASDCP_FAILURE (state->mxf_writer.OpenWrite (
 119                                    asset->path().string().c_str(),
 120                                    state->writer_info,
 121                                    state->picture_descriptor,
 122                                    16384,
 123                                    writer->_overwrite)
 124                     )) {
 125
 126                 boost::throw_exception (MXFFileError ("could not open MXF file for writing", asset->path().string()));
 127         }
 128
 129         writer->_started = true;
 130 }
 131
 132 void
 133 MonoPictureAssetWriter::start (uint8_t* data, int size)
 134 {
 135         libdcp::start (this, _state, _asset, data, size);
 136 }
 137
 138 void
 139 StereoPictureAssetWriter::start (uint8_t* data, int size)
 140 {
 141         libdcp::start (this, _state, _asset, data, size);
 142 }
 143
 144 FrameInfo
 145 MonoPictureAssetWriter::write (uint8_t* data, int size)
 146 {
 147         assert (!_finalized);
 148
 149         if (!_started) {
 150                 start (data, size);
 151         }
 152
 153         if (ASDCP_FAILURE (_state->j2k_parser.OpenReadFrame (data, size, _state->frame_buffer))) {
 154                 boost::throw_exception (MiscError ("could not parse J2K frame"));
 155         }
 156
 157         uint64_t const before_offset = _state->mxf_writer.Tell ();
 158
 159         string hash;
 160         if (ASDCP_FAILURE (_state->mxf_writer.WriteFrame (_state->frame_buffer, 0, 0, &hash))) {
 161                 boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string()));
 162         }
 163
 164         ++_frames_written;
 165         return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash);
 166 }
 167
 168 /** Write a frame for one eye.  Frames must be written left, then right, then left etc.
 169  *  @param data JPEG2000 data.
 170  *  @param size Size of data.
 171  */
 172 FrameInfo
 173 StereoPictureAssetWriter::write (uint8_t* data, int size)
 174 {
 175         assert (!_finalized);
 176
 177         if (!_started) {
 178                 start (data, size);
 179         }
 180
 181         if (ASDCP_FAILURE (_state->j2k_parser.OpenReadFrame (data, size, _state->frame_buffer))) {
 182                 boost::throw_exception (MiscError ("could not parse J2K frame"));
 183         }
 184
 185         uint64_t const before_offset = _state->mxf_writer.Tell ();
 186
 187         string hash;
 188         if (ASDCP_FAILURE (
 189                     _state->mxf_writer.WriteFrame (
 190                             _state->frame_buffer,
 191                             _next_eye == EYE_LEFT ? ASDCP::JP2K::SP_LEFT : ASDCP::JP2K::SP_RIGHT,
 192                             0,
 193                             0,
 194                             &hash)
 195                     )) {
 196
 197                 boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string()));
 198         }
 199
 200         _next_eye = _next_eye == EYE_LEFT ? EYE_RIGHT : EYE_LEFT;
 201
 202         return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash);
 203 }
 204
 205 void
 206 MonoPictureAssetWriter::fake_write (int size)
 207 {
 208         assert (_started);
 209         assert (!_finalized);
 210
 211         if (ASDCP_FAILURE (_state->mxf_writer.FakeWriteFrame (size))) {
 212                 boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string()));
 213         }
 214
 215         ++_frames_written;
 216 }
 217
 218 void
 219 StereoPictureAssetWriter::fake_write (int size)
 220 {
 221         assert (_started);
 222         assert (!_finalized);
 223
 224         if (ASDCP_FAILURE (_state->mxf_writer.FakeWriteFrame (size))) {
 225                 boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string()));
 226         }
 227
 228         _next_eye = _next_eye == EYE_LEFT ? EYE_RIGHT : EYE_LEFT;
 229         ++_frames_written;
 230 }
 231
 232 void
 233 MonoPictureAssetWriter::finalize ()
 234 {
 235         assert (!_finalized);
 236
 237         if (ASDCP_FAILURE (_state->mxf_writer.Finalize())) {
 238                 boost::throw_exception (MXFFileError ("error in finalizing video MXF", _asset->path().string()));
 239         }
 240
 241         _finalized = true;
 242         _asset->set_intrinsic_duration (_frames_written);
 243         _asset->set_duration (_frames_written);
 244 }
 245
 246 void
 247 StereoPictureAssetWriter::finalize ()
 248 {
 249         assert (!_finalized);
 250
 251         if (ASDCP_FAILURE (_state->mxf_writer.Finalize())) {
 252                 boost::throw_exception (MXFFileError ("error in finalizing video MXF", _asset->path().string()));
 253         }
 254
 255         _finalized = true;
 256         _asset->set_intrinsic_duration (_frames_written / 2);
 257         _asset->set_duration (_frames_written / 2);
 258 }