Merge master.

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

diff --git a/src/lib/dcp_examiner.cc b/src/lib/dcp_examiner.cc
index 625276e180de13f54a92a05b16f5e453f27e8d08..1e4cc899df96df614fd6a9b1413d3101245247a4 100644 (file)
--- a/src/lib/dcp_examiner.cc
+++ b/src/lib/dcp_examiner.cc
@@ -22,21 +22,38 @@
 #include <dcp/reel.h>
 #include <dcp/reel_picture_asset.h>
 #include <dcp/reel_sound_asset.h>
+#include <dcp/mono_picture_mxf.h>
+#include <dcp/mono_picture_frame.h>
+#include <dcp/stereo_picture_mxf.h>
+#include <dcp/stereo_picture_frame.h>
 #include <dcp/sound_mxf.h>
 #include "dcp_examiner.h"
 #include "dcp_content.h"
 #include "exceptions.h"
+#include "image.h"
+#include "config.h"
 
 #include "i18n.h"
 
 using std::list;
+using std::cout;
 using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
 
 DCPExaminer::DCPExaminer (shared_ptr<const DCPContent> content)
+       : _video_length (0)
+       , _audio_length (0)
+       , _has_subtitles (false)
+       , _encrypted (false)
+       , _kdm_valid (false)
 {
        dcp::DCP dcp (content->directory ());
        dcp.read ();
 
+       if (content->kdm ()) {
+               dcp.add (dcp::DecryptedKDM (content->kdm().get(), Config::instance()->decryption_private_key ()));
+       }
+
        if (dcp.cpls().size() == 0) {
                throw DCPError ("No CPLs found in DCP");
        } else if (dcp.cpls().size() > 1) {
@@ -89,4 +106,31 @@ DCPExaminer::DCPExaminer (shared_ptr<const DCPContent> content)
                        _has_subtitles = true;
                }
        }
+
+       _encrypted = dcp.encrypted ();
+       _kdm_valid = true;
+       
+       /* Check that we can read the first picture frame */
+       try {
+               if (!dcp.cpls().empty () && !dcp.cpls().front()->reels().empty ()) {
+                       shared_ptr<dcp::PictureMXF> mxf = dcp.cpls().front()->reels().front()->main_picture()->mxf ();
+                       shared_ptr<dcp::MonoPictureMXF> mono = dynamic_pointer_cast<dcp::MonoPictureMXF> (mxf);
+                       shared_ptr<dcp::StereoPictureMXF> stereo = dynamic_pointer_cast<dcp::StereoPictureMXF> (mxf);
+                       
+                       shared_ptr<Image> image (new Image (PIX_FMT_RGB24, _video_size.get(), false));
+                       
+                       if (mono) {
+                               mono->get_frame(0)->rgb_frame (image->data()[0]);
+                       } else {
+                               stereo->get_frame(0)->rgb_frame (dcp::EYE_LEFT, image->data()[0]);
+                       }
+                       
+               }
+       } catch (dcp::DCPReadError& e) {
+               _kdm_valid = false;
+               if (_encrypted && content->kdm ()) {
+                       /* XXX: maybe don't use an exception for this */
+                       throw StringError (_("The KDM does not decrypt the DCP.  Perhaps it is targeted at the wrong CPL"));
+               }
+       }
 }