Seems to kind-of build video MXFs.

[dcpomatic.git] / src / lib / make_mxf_job.cc

diff --git a/src/lib/make_mxf_job.cc b/src/lib/make_mxf_job.cc
index 34f115e9811367ea71b5cbc69d6259c6829220c0..d66ea3e1e9b36f88700271b5a75c85041a926992 100644 (file)
--- a/src/lib/make_mxf_job.cc
+++ b/src/lib/make_mxf_job.cc
@@ -24,6 +24,7 @@
 #include <iostream>
 #include <boost/filesystem.hpp>
 #include "AS_DCP.h"
+#include "KM_fileio.h"
 #include "make_mxf_job.h"
 #include "film.h"
 #include "film_state.h"
@@ -101,7 +102,7 @@ MakeMXFJob::run ()
                break;
        }
 
-       std::list<std::string> files;
+       list<string> files;
         for (filesystem::directory_iterator i = filesystem::directory_iterator (dir); i != filesystem::directory_iterator(); ++i) {
                files.push_back (filesystem::path (*i).string());
        }
@@ -110,6 +111,8 @@ MakeMXFJob::run ()
                throw EncodeError ("no input files found for MXF");
        }
 
+       files.sort ();
+
        ASDCP::EssenceType_t essence_type;
        if (ASDCP_FAILURE (ASDCP::RawEssenceType (files.front().c_str(), essence_type))) {
                throw EncodeError ("could not work out type for MXF");
@@ -117,11 +120,11 @@ MakeMXFJob::run ()
 
        switch (essence_type) {
        case ASDCP::ESS_JPEG_2000:
-               /* XXX */
+               j2k (files, _fs->file ("video.mxf"));
                break;
        case ASDCP::ESS_PCM_24b_48k:
        case ASDCP::ESS_PCM_24b_96k:
-               /* XXX */
+               wav (files, _fs->file ("audio.mxf"));
                break;
        default:
                throw EncodeError ("unknown essence type");
@@ -129,3 +132,70 @@ MakeMXFJob::run ()
        
        set_progress (1);
 }
+
+void
+MakeMXFJob::wav (list<string> const & files, string const & mxf)
+{
+
+}
+
+void
+MakeMXFJob::j2k (list<string> const & files, string const & mxf)
+{
+       /* Arbitrarily assume that the J2K MXF will take 90% of the time */
+       descend (0.9);
+
+       ASDCP::JP2K::CodestreamParser j2k_parser;
+       ASDCP::JP2K::FrameBuffer frame_buffer (4 * Kumu::Megabyte);
+       if (ASDCP_FAILURE (j2k_parser.OpenReadFrame (files.front().c_str(), frame_buffer))) {
+               throw EncodeError ("could not open J2K file for reading");
+       }
+       
+       ASDCP::JP2K::PictureDescriptor picture_desc;
+       j2k_parser.FillPictureDescriptor (picture_desc);
+       /* XXX: we round for DCP: not sure if this is right */
+       picture_desc.EditRate = ASDCP::Rational (rintf (_fs->frames_per_second), 1);
+       
+       ASDCP::WriterInfo writer_info;
+       fill_writer_info (&writer_info);
+       
+       ASDCP::JP2K::MXFWriter mxf_writer;
+       if (ASDCP_FAILURE (mxf_writer.OpenWrite (mxf.c_str(), writer_info, picture_desc))) {
+               throw EncodeError ("could not open MXF for writing");
+       }
+
+       int j = 0;
+       for (list<string>::const_iterator i = files.begin(); i != files.end(); ++i) {
+               if (ASDCP_FAILURE (j2k_parser.OpenReadFrame (i->c_str(), frame_buffer))) {
+                       throw EncodeError ("could not open J2K file for reading");
+               }
+
+               /* XXX: passing 0 to WriteFrame ok? */
+               if (ASDCP_FAILURE (mxf_writer.WriteFrame (frame_buffer, 0, 0))) {
+                       throw EncodeError ("error in writing video MXF");
+               }
+               
+               ++j;
+               set_progress (float (j) / files.size ());
+       }
+       
+       if (ASDCP_FAILURE (mxf_writer.Finalize())) {
+               throw EncodeError ("error in finalising video MXF");
+       }
+       
+       ascend ();
+}
+
+void
+MakeMXFJob::fill_writer_info (ASDCP::WriterInfo* writer_info)
+{
+       writer_info->ProductVersion = DVDOMATIC_VERSION;
+       writer_info->CompanyName = "dvd-o-matic";
+       writer_info->ProductName = "dvd-o-matic";
+
+       /* set the label type */
+       writer_info->LabelSetType = ASDCP::LS_MXF_INTEROP;
+
+       /* generate a random UUID for this essence */
+       Kumu::GenRandomUUID (writer_info->AssetUUID);
+}