X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fpicture_asset_writer.cc;h=c30be1bc4f35a78d033ac46f7b93a645db5feeba;hb=0d31c86d6dfad9f437f5613d41cace9cc5928474;hp=b5ced7606925852c71973f89840dd18a31ea2867;hpb=dddaa7abf05f182246a5b7d9b6ebc64c25d4bd86;p=libdcp.git diff --git a/src/picture_asset_writer.cc b/src/picture_asset_writer.cc index b5ced760..c30be1bc 100644 --- a/src/picture_asset_writer.cc +++ b/src/picture_asset_writer.cc @@ -1,247 +1,67 @@ /* - Copyright (C) 2012-2013 Carl Hetherington + Copyright (C) 2012-2021 Carl Hetherington - This program is free software; you can redistribute it and/or modify + 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. - This program is distributed in the hope that it will be useful, + 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 this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. - + along with libdcp. If not, see . + + 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. */ -#include "AS_DCP.h" -#include "KM_fileio.h" -#include "picture_asset_writer.h" -#include "exceptions.h" -#include "picture_asset.h" - -using std::istream; -using std::ostream; -using std::string; -using boost::shared_ptr; -using namespace libdcp; - -FrameInfo::FrameInfo (istream& s) -{ - s >> offset >> size >> hash; -} - -void -FrameInfo::write (ostream& s) -{ - s << offset << " " << size << " " << hash; -} - -PictureAssetWriter::PictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata) - : _asset (asset) - , _frames_written (0) - , _started (false) - , _finalized (false) - , _overwrite (overwrite) - , _metadata (metadata) -{ - -} - -struct ASDCPState -{ - ASDCPState () - : frame_buffer (4 * Kumu::Megabyte) - {} - - ASDCP::JP2K::CodestreamParser j2k_parser; - ASDCP::JP2K::FrameBuffer frame_buffer; - ASDCP::WriterInfo writer_info; - ASDCP::JP2K::PictureDescriptor picture_descriptor; -}; - -struct MonoPictureAssetWriter::MonoASDCPState : public ASDCPState -{ - ASDCP::JP2K::MXFWriter mxf_writer; -}; - -struct StereoPictureAssetWriter::StereoASDCPState : public ASDCPState -{ - ASDCP::JP2K::MXFSWriter mxf_writer; -}; - -/** @param a Asset to write to. `a' must not be deleted while - * this writer class still exists, or bad things will happen. +/** @file src/picture_asset_writer.cc + * @brief PictureAssetWriter and FrameInfo classes */ -MonoPictureAssetWriter::MonoPictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata) - : PictureAssetWriter (asset, overwrite, metadata) - , _state (new MonoPictureAssetWriter::MonoASDCPState) -{ -} -StereoPictureAssetWriter::StereoPictureAssetWriter (PictureAsset* asset, bool overwrite, MXFMetadata const & metadata) - : PictureAssetWriter (asset, overwrite, metadata) - , _state (new StereoPictureAssetWriter::StereoASDCPState) - , _next_eye (EYE_LEFT) -{ +#include "picture_asset_writer.h" +#include "exceptions.h" +#include "picture_asset.h" +#include +#include +#include +#include -} -template -void libdcp::start (PictureAssetWriter* writer, shared_ptr

state, Q* asset, uint8_t* data, int size) -{ - if (ASDCP_FAILURE (state->j2k_parser.OpenReadFrame (data, size, state->frame_buffer))) { - boost::throw_exception (MiscError ("could not parse J2K frame")); - } - - state->j2k_parser.FillPictureDescriptor (state->picture_descriptor); - state->picture_descriptor.EditRate = ASDCP::Rational (asset->edit_rate(), 1); - - asset->fill_writer_info (&state->writer_info, asset->uuid(), writer->_metadata); - - if (ASDCP_FAILURE (state->mxf_writer.OpenWrite ( - asset->path().string().c_str(), - state->writer_info, - state->picture_descriptor, - 16384, - writer->_overwrite) - )) { - - boost::throw_exception (MXFFileError ("could not open MXF file for writing", asset->path().string())); - } - - writer->_started = true; -} +using std::string; +using std::shared_ptr; +using namespace dcp; -void -MonoPictureAssetWriter::start (uint8_t* data, int size) -{ - libdcp::start (this, _state, _asset, data, size); -} -void -StereoPictureAssetWriter::start (uint8_t* data, int size) +PictureAssetWriter::PictureAssetWriter (PictureAsset* asset, boost::filesystem::path file, bool overwrite) + : AssetWriter (asset, file) + , _picture_asset (asset) + , _overwrite (overwrite) { - libdcp::start (this, _state, _asset, data, size); + asset->set_file (file); } -FrameInfo -MonoPictureAssetWriter::write (uint8_t* data, int size) -{ - 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")); - } - - uint64_t const before_offset = _state->mxf_writer.Tell (); - - string hash; - if (ASDCP_FAILURE (_state->mxf_writer.WriteFrame (_state->frame_buffer, 0, 0, &hash))) { - boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string())); - } - - ++_frames_written; - return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash); -} - -/** Write a frame for one eye. Frames must be written left, then right, then left etc. - * @param data JPEG2000 data. - * @param size Size of data. - */ FrameInfo -StereoPictureAssetWriter::write (uint8_t* data, int size) -{ - 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")); - } - - uint64_t const before_offset = _state->mxf_writer.Tell (); - - string hash; - if (ASDCP_FAILURE ( - _state->mxf_writer.WriteFrame ( - _state->frame_buffer, - _next_eye == EYE_LEFT ? ASDCP::JP2K::SP_LEFT : ASDCP::JP2K::SP_RIGHT, - 0, - 0, - &hash) - )) { - - boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string())); - } - - _next_eye = _next_eye == EYE_LEFT ? EYE_RIGHT : EYE_LEFT; - - return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash); -} - -void -MonoPictureAssetWriter::fake_write (int size) -{ - assert (_started); - assert (!_finalized); - - if (ASDCP_FAILURE (_state->mxf_writer.FakeWriteFrame (size))) { - boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string())); - } - - ++_frames_written; -} - -void -StereoPictureAssetWriter::fake_write (int size) -{ - assert (_started); - assert (!_finalized); - - if (ASDCP_FAILURE (_state->mxf_writer.FakeWriteFrame (size))) { - boost::throw_exception (MXFFileError ("error in writing video MXF", _asset->path().string())); - } - - _next_eye = _next_eye == EYE_LEFT ? EYE_RIGHT : EYE_LEFT; - ++_frames_written; -} - -void -MonoPictureAssetWriter::finalize () -{ - assert (!_finalized); - - if (ASDCP_FAILURE (_state->mxf_writer.Finalize())) { - boost::throw_exception (MXFFileError ("error in finalizing video MXF", _asset->path().string())); - } - - _finalized = true; - _asset->set_intrinsic_duration (_frames_written); - _asset->set_duration (_frames_written); -} - -void -StereoPictureAssetWriter::finalize () +PictureAssetWriter::write (Data const& data) { - assert (!_finalized); - - if (ASDCP_FAILURE (_state->mxf_writer.Finalize())) { - boost::throw_exception (MXFFileError ("error in finalizing video MXF", _asset->path().string())); - } - - _finalized = true; - _asset->set_intrinsic_duration (_frames_written / 2); - _asset->set_duration (_frames_written / 2); + return write (data.data(), data.size()); }