Support MPEG2 compression.v1.9.6

1 files changed, 144 insertions, 0 deletions

diff --git a/src/mono_mpeg2_picture_asset_writer.cc b/src/mono_mpeg2_picture_asset_writer.cc
new file mode 100644
index 00000000..f8101c54
--- /dev/null
+++ b/src/mono_mpeg2_picture_asset_writer.cc
@@ -0,0 +1,144 @@
+/*
+    Copyright (C) 2024 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.
+*/
+
+
+#include "mono_mpeg2_picture_asset_writer.h"
+#include "mpeg2_picture_asset.h"
+#include "mpeg2_picture_asset_writer.h"
+
+
+using std::string;
+using namespace dcp;
+
+
+struct MonoMPEG2PictureAssetWriter::ASDCPState : public ASDCPMPEG2StateBase
+{
+	ASDCP::MPEG2::MXFWriter mxf_writer;
+};
+
+
+
+
+MonoMPEG2PictureAssetWriter::MonoMPEG2PictureAssetWriter(MPEG2PictureAsset* asset, boost::filesystem::path file, bool overwrite)
+	: MPEG2PictureAssetWriter(asset, file, overwrite)
+	, _state(new MonoMPEG2PictureAssetWriter::ASDCPState)
+{
+	asset->set_file(file);
+}
+
+
+MonoMPEG2PictureAssetWriter::~MonoMPEG2PictureAssetWriter()
+{
+	try {
+		/* Last-resort finalization to close the file, at least */
+		if (!_finalized) {
+			_state->mxf_writer.Finalize();
+		}
+	} catch (...) {}
+}
+
+
+void
+MonoMPEG2PictureAssetWriter::start(uint8_t const * data, int size)
+{
+	dcp::start(this, _state, _picture_asset, data, size);
+	_picture_asset->set_frame_rate (_picture_asset->edit_rate());
+}
+
+
+MPEG2FrameInfo
+MonoMPEG2PictureAssetWriter::write(uint8_t const * data, int size)
+{
+	DCP_ASSERT(!_finalized);
+
+	if (!_started) {
+		start(data, size);
+	}
+
+	ASDCP::MPEG2::FrameBuffer buffer;
+	buffer.SetData(const_cast<uint8_t*>(data), size);
+	buffer.Size(size);
+	buffer.PlaintextOffset(0);
+
+	auto const before_offset = _state->mxf_writer.Tell();
+
+	string hash;
+	auto const r = _state->mxf_writer.WriteFrame(buffer, _crypto_context->context(), _crypto_context->hmac(), &hash);
+	if (ASDCP_FAILURE(r)) {
+		boost::throw_exception(MXFFileError("error in writing video MXF", _file.string(), r));
+	}
+
+	++_frames_written;
+	return MPEG2FrameInfo(
+		before_offset,
+		_state->mxf_writer.Tell() - before_offset,
+		hash,
+		buffer.FrameType(),
+		buffer.GOPStart(),
+		buffer.ClosedGOP(),
+		buffer.TemporalOffset()
+		);
+}
+
+
+void
+MonoMPEG2PictureAssetWriter::fake_write(MPEG2FrameInfo const& info)
+{
+	DCP_ASSERT(_started);
+	DCP_ASSERT(!_finalized);
+
+	DCP_ASSERT(false);
+
+	auto r = _state->mxf_writer.FakeWriteFrame(info.size, info.type, info.gop_start, info.closed_gop, info.temporal_offset);
+	if (ASDCP_FAILURE(r)) {
+		boost::throw_exception(MXFFileError("error in writing video MXF", _file.string(), r));
+	}
+
+	++_frames_written;
+}
+
+
+bool
+MonoMPEG2PictureAssetWriter::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 MPEG2PictureAssetWriter::finalize();
+}
+