diff options
Diffstat (limited to 'src/stereo_j2k_picture_asset_writer.cc')
| -rw-r--r-- | src/stereo_j2k_picture_asset_writer.cc | 160 |
1 files changed, 160 insertions, 0 deletions
diff --git a/src/stereo_j2k_picture_asset_writer.cc b/src/stereo_j2k_picture_asset_writer.cc new file mode 100644 index 00000000..6ee271bc --- /dev/null +++ b/src/stereo_j2k_picture_asset_writer.cc @@ -0,0 +1,160 @@ +/* + Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net> + + This file is part of libdcp. + + libdcp is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + libdcp is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with libdcp. If not, see <http://www.gnu.org/licenses/>. + + In addition, as a special exception, the copyright holders give + permission to link the code of portions of this program with the + OpenSSL library under certain conditions as described in each + individual source file, and distribute linked combinations + including the two. + + You must obey the GNU General Public License in all respects + for all of the code used other than OpenSSL. If you modify + file(s) with this exception, you may extend this exception to your + version of the file(s), but you are not obligated to do so. If you + do not wish to do so, delete this exception statement from your + version. If you delete this exception statement from all source + files in the program, then also delete it here. +*/ + + +/** @file src/stereo_picture_asset_writer.cc + * @brief StereoPictureAssetWriter class + */ + + +#include "stereo_picture_asset_writer.h" +#include "exceptions.h" +#include "dcp_assert.h" +#include "picture_asset.h" +#include "crypto_context.h" +#include <asdcp/AS_DCP.h> +#include <asdcp/KM_fileio.h> + + +#include "picture_asset_writer_common.cc" + + +using std::string; +using std::shared_ptr; +using namespace dcp; + + +struct StereoPictureAssetWriter::ASDCPState : public ASDCPStateBase +{ + ASDCP::JP2K::MXFSWriter mxf_writer; +}; + + +StereoPictureAssetWriter::StereoPictureAssetWriter (PictureAsset* mxf, boost::filesystem::path file, bool overwrite) + : PictureAssetWriter (mxf, file, overwrite) + , _state (new StereoPictureAssetWriter::ASDCPState) +{ + +} + + +StereoPictureAssetWriter::~StereoPictureAssetWriter() +{ + try { + /* Last-resort finalization to close the file, at least */ + if (!_finalized) { + _state->mxf_writer.Finalize(); + } + } catch (...) {} +} + + +void +StereoPictureAssetWriter::start (uint8_t const * data, int size) +{ + dcp::start (this, _state, _picture_asset, data, size); + _picture_asset->set_frame_rate (Fraction (_picture_asset->edit_rate().numerator * 2, _picture_asset->edit_rate().denominator)); +} + + +FrameInfo +StereoPictureAssetWriter::write (uint8_t const * data, int size) +{ + DCP_ASSERT (!_finalized); + + if (!_started) { + start (data, size); + } + + if (ASDCP_FAILURE (_state->j2k_parser.OpenReadFrame (data, size, _state->frame_buffer))) { + boost::throw_exception (MiscError ("could not parse J2K frame")); + } + + _state->frame_buffer.PlaintextOffset(0); + + uint64_t const before_offset = _state->mxf_writer.Tell (); + + string hash; + auto r = _state->mxf_writer.WriteFrame ( + _state->frame_buffer, + _next_eye == Eye::LEFT ? ASDCP::JP2K::SP_LEFT : ASDCP::JP2K::SP_RIGHT, + _crypto_context->context(), + _crypto_context->hmac(), + &hash + ); + + if (ASDCP_FAILURE (r)) { + boost::throw_exception (MXFFileError ("error in writing video MXF", _file.string(), r)); + } + + _next_eye = _next_eye == Eye::LEFT ? Eye::RIGHT : Eye::LEFT; + + if (_next_eye == Eye::LEFT) { + ++_frames_written; + } + + return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash); +} + + +void +StereoPictureAssetWriter::fake_write (int size) +{ + DCP_ASSERT (_started); + DCP_ASSERT (!_finalized); + + auto r = _state->mxf_writer.FakeWriteFrame (size, _next_eye == Eye::LEFT ? ASDCP::JP2K::SP_LEFT : ASDCP::JP2K::SP_RIGHT); + if (ASDCP_FAILURE(r)) { + boost::throw_exception (MXFFileError("error in writing video MXF", _file.string(), r)); + } + + _next_eye = _next_eye == Eye::LEFT ? Eye::RIGHT : Eye::LEFT; + if (_next_eye == Eye::LEFT) { + ++_frames_written; + } +} + + +bool +StereoPictureAssetWriter::finalize () +{ + if (_started) { + auto r = _state->mxf_writer.Finalize(); + if (ASDCP_FAILURE(r)) { + boost::throw_exception (MXFFileError("error in finalizing video MXF", _file.string(), r)); + } + } + + _picture_asset->_intrinsic_duration = _frames_written; + return PictureAssetWriter::finalize (); +} |