*/
+#include <inttypes.h>
+#include <stdint.h>
#include "AS_DCP.h"
#include "KM_fileio.h"
#include "picture_asset_writer.h"
using namespace libdcp;
FrameInfo::FrameInfo (istream& s)
+ : offset (0)
+ , size (0)
{
- s >> offset >> size >> hash;
-}
-
-void
-FrameInfo::write (ostream& s)
-{
- s << offset << " " << size << " " << hash;
-}
-
-
-PictureAssetWriter::PictureAssetWriter (bool overwrite, MXFMetadata const & metadata)
- : _frames_written (0)
- , _started (false)
- , _finalized (false)
- , _overwrite (overwrite)
- , _metadata (metadata)
-{
-
-}
+ s >> offset >> size;
-struct MonoPictureAssetWriter::ASDCPState
-{
- ASDCPState()
- : frame_buffer (4 * Kumu::Megabyte)
- {}
-
- ASDCP::JP2K::CodestreamParser j2k_parser;
- ASDCP::JP2K::FrameBuffer frame_buffer;
- ASDCP::JP2K::MXFWriter mxf_writer;
- ASDCP::WriterInfo writer_info;
- ASDCP::JP2K::PictureDescriptor picture_descriptor;
-};
-
-struct StereoPictureAssetWriter::ASDCPState
-{
- ASDCPState()
- : frame_buffer (4 * Kumu::Megabyte)
- {}
-
- ASDCP::JP2K::CodestreamParser j2k_parser;
- ASDCP::JP2K::SFrameBuffer frame_buffer;
- ASDCP::JP2K::MXFSWriter mxf_writer;
- ASDCP::WriterInfo writer_info;
- ASDCP::JP2K::PictureDescriptor picture_descriptor;
-};
-
-/** @param a Asset to write to. `a' must not be deleted while
- * this writer class still exists, or bad things will happen.
- */
-MonoPictureAssetWriter::MonoPictureAssetWriter (MonoPictureAsset* asset, bool overwrite, MXFMetadata const & metadata)
- : PictureAssetWriter (overwrite, metadata)
- , _state (new MonoPictureAssetWriter::ASDCPState)
- , _asset (asset)
-{
-
-}
-
-StereoPictureAssetWriter::StereoPictureAssetWriter (StereoPictureAsset* asset, bool overwrite, MXFMetadata const & metadata)
- : PictureAssetWriter (overwrite, metadata)
- , _state (new StereoPictureAssetWriter::ASDCPState)
- , _asset (asset)
-{
-
-}
-
-template <class P, class Q>
-void start (PictureAssetWriter* writer, shared_ptr<P> 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"));
+ if (!s.good ()) {
+ /* Make sure we zero these if something bad happened, otherwise
+ the caller might try to alloc lots of RAM.
+ */
+ offset = size = 0;
}
- 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;
+ s >> hash;
}
-void
-MonoPictureAssetWriter::start (uint8_t* data, int size)
+FrameInfo::FrameInfo (FILE* f)
{
- ::start (this, _state, _asset, data, size);
-}
+ fscanf (f, "%" PRId64, &offset);
+ fscanf (f, "%" PRId64, &size);
-void
-StereoPictureAssetWriter::start (uint8_t* data, int size)
-{
- ::start (this, _state, _asset, data, size);
-}
-
-template <class P, class Q>
-FrameInfo write (PictureAssetWriter* writer, shared_ptr<P> state, Q* asset, uint8_t* data, int size)
-{
- assert (!writer->_finalized);
-
- if (!writer->_started) {
- writer->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()));
+ if (ferror (f)) {
+ offset = size = 0;
}
- ++asset->_frames_written;
- return FrameInfo (before_offset, state->mxf_writer.Tell() - before_offset, hash);
-}
-
-FrameInfo
-MonoPictureAssetWriter::write (uint8_t* data, int size)
-{
- return ::write (this, _state, _asset, data, size);
-}
-
-FrameInfo
-StereoPictureAssetWriter::write (uint8_t* data, int size)
-{
- return ::write (this, _state, _asset, data, size);
+ char hash_buffer[128];
+ fscanf (f, "%s", hash_buffer);
+ hash = hash_buffer;
}
void
-MonoPictureAssetWriter::fake_write (int size)
+FrameInfo::write (ostream& s) const
{
- 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;
+ s << offset << " " << size << " " << hash;
}
void
-StereoPictureAssetWriter::fake_write (int size)
+FrameInfo::write (FILE* f) const
{
- 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;
+ fprintf (f, "%" PRId64 " %" PRId64 " %s", offset, size, hash.c_str ());
}
-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::PictureAssetWriter (PictureAsset* asset, bool overwrite)
+ : _asset (asset)
+ , _frames_written (0)
+ , _started (false)
+ , _finalized (false)
+ , _overwrite (overwrite)
{
- 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);
}