Tidying.
[libdcp.git] / src / stereo_picture_asset_writer.cc
index b4b0ad524e607a6650440e324bde20c3484bd908..e02e83605de241592d7cabee94cdaadfb87588ec 100644 (file)
@@ -1,63 +1,85 @@
 /*
-    Copyright (C) 2012-2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
-    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 <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 "AS_DCP.h"
-#include "KM_fileio.h"
+
+/** @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::istream;
-using std::ostream;
+
 using std::string;
-using boost::shared_ptr;
-using namespace libdcp;
+using std::shared_ptr;
+using namespace dcp;
+
 
 struct StereoPictureAssetWriter::ASDCPState : public ASDCPStateBase
 {
        ASDCP::JP2K::MXFSWriter mxf_writer;
 };
 
-StereoPictureAssetWriter::StereoPictureAssetWriter (PictureAsset* asset, bool overwrite)
-       : PictureAssetWriter (asset, overwrite)
+
+StereoPictureAssetWriter::StereoPictureAssetWriter (PictureAsset* mxf, boost::filesystem::path file, bool overwrite)
+       : PictureAssetWriter (mxf, file, overwrite)
        , _state (new StereoPictureAssetWriter::ASDCPState)
-       , _next_eye (EYE_LEFT)
 {
-       _state->encryption_context = asset->encryption_context ();
+
 }
 
+
 void
-StereoPictureAssetWriter::start (uint8_t* data, int size)
+StereoPictureAssetWriter::start (uint8_t const * data, int size)
 {
-       libdcp::start (this, _state, _asset, data, 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));
 }
 
-/** 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)
+StereoPictureAssetWriter::write (uint8_t const * data, int size)
 {
-       assert (!_finalized);
+       DCP_ASSERT (!_finalized);
 
        if (!_started) {
                start (data, size);
@@ -70,50 +92,56 @@ StereoPictureAssetWriter::write (uint8_t* data, int size)
        uint64_t const before_offset = _state->mxf_writer.Tell ();
 
        string hash;
-       Kumu::Result_t r = _state->mxf_writer.WriteFrame (
+       auto r = _state->mxf_writer.WriteFrame (
                _state->frame_buffer,
-               _next_eye == EYE_LEFT ? ASDCP::JP2K::SP_LEFT : ASDCP::JP2K::SP_RIGHT,
-               _state->encryption_context,
-               0,
+               _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", _asset->path().string(), r));
+               boost::throw_exception (MXFFileError ("error in writing video MXF", _file.string(), r));
        }
 
-       _next_eye = _next_eye == EYE_LEFT ? EYE_RIGHT : EYE_LEFT;
+       _next_eye = _next_eye == Eye::LEFT ? Eye::RIGHT : Eye::LEFT;
+
+       if (_next_eye == Eye::LEFT) {
+               ++_frames_written;
+       }
 
-       ++_frames_written;
        return FrameInfo (before_offset, _state->mxf_writer.Tell() - before_offset, hash);
 }
 
+
 void
 StereoPictureAssetWriter::fake_write (int size)
 {
-       assert (_started);
-       assert (!_finalized);
+       DCP_ASSERT (_started);
+       DCP_ASSERT (!_finalized);
 
-       Kumu::Result_t 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", _asset->path().string(), r));
+       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;
-       ++_frames_written;
+       _next_eye = _next_eye == Eye::LEFT ? Eye::RIGHT : Eye::LEFT;
+       if (_next_eye == Eye::LEFT) {
+               ++_frames_written;
+       }
 }
 
-void
+
+bool
 StereoPictureAssetWriter::finalize ()
 {
-       assert (!_finalized);
-       
-       Kumu::Result_t r = _state->mxf_writer.Finalize();
-       if (ASDCP_FAILURE (r)) {
-               boost::throw_exception (MXFFileError ("error in finalizing video MXF", _asset->path().string(), r));
+       if (_started) {
+               auto r = _state->mxf_writer.Finalize();
+               if (ASDCP_FAILURE(r)) {
+                       boost::throw_exception (MXFFileError("error in finalizing video MXF", _file.string(), r));
+               }
        }
 
-       _finalized = true;
-       _asset->set_intrinsic_duration (_frames_written / 2);
-       _asset->set_duration (_frames_written / 2);
+       _picture_asset->_intrinsic_duration = _frames_written;
+       return PictureAssetWriter::finalize ();
 }