X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Fswaroop_controls.cc;h=d8019a41e7b840b4dc82a4bbee143ba167f59b48;hb=c1d3751723c5413eea4b4602ec5ecff5c961038d;hp=d557d09e9b8fe9190c7d7882fd455ccbcfaa8f2d;hpb=c4403784febdbdd42e9c32e67fadb147f11fe566;p=dcpomatic.git

diff --git a/src/wx/swaroop_controls.cc b/src/wx/swaroop_controls.cc
index d557d09e9..d8019a41e 100644
--- a/src/wx/swaroop_controls.cc
+++ b/src/wx/swaroop_controls.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2018 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2018-2019 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -29,6 +29,7 @@
 #include "lib/cross.h"
 #include "lib/scoped_temporary.h"
 #include "lib/internet.h"
+#include "lib/ffmpeg_content.h"
 #include <dcp/raw_convert.h>
 #include <dcp/exceptions.h>
 #include <wx/listctrl.h>
@@ -385,6 +386,28 @@ SwaroopControls::get_kdm_from_directory (shared_ptr<DCPContent> dcp)
 	return optional<dcp::EncryptedKDM>();
 }
 
+optional<EncryptedECinemaKDM>
+SwaroopControls::get_kdm_from_directory (shared_ptr<FFmpegContent> ffmpeg)
+{
+	using namespace boost::filesystem;
+	optional<path> kdm_dir = Config::instance()->player_kdm_directory();
+	if (!kdm_dir) {
+		return optional<EncryptedECinemaKDM>();
+	}
+	for (directory_iterator i = directory_iterator(*kdm_dir); i != directory_iterator(); ++i) {
+		try {
+			if (file_size(i->path()) < MAX_KDM_SIZE) {
+				EncryptedECinemaKDM kdm (dcp::file_to_string(i->path()));
+				if (kdm.id() == ffmpeg->id().get_value_or("")) {
+					return kdm;
+				}
+			}
+		} catch (std::exception& e) {
+			/* Hey well */
+		}
+	}
+	return optional<EncryptedECinemaKDM>();
+}
 void
 SwaroopControls::spl_selection_changed ()
 {
@@ -442,6 +465,23 @@ SwaroopControls::select_playlist (int selected, int position)
 				return;
 			}
 		}
+		shared_ptr<FFmpegContent> ffmpeg = dynamic_pointer_cast<FFmpegContent> (i.content);
+		if (ffmpeg && ffmpeg->encrypted()) {
+			optional<EncryptedECinemaKDM> kdm = get_kdm_from_directory (ffmpeg);
+			if (kdm) {
+				try {
+					ffmpeg->add_kdm (*kdm);
+					ffmpeg->examine (_film, shared_ptr<Job>());
+				} catch (KDMError& e) {
+					error_dialog (this, "Could not load KDM.");
+				}
+			} else {
+				error_dialog (this, "This playlist cannot be loaded as a KDM is missing.");
+				_selected_playlist = boost::none;
+				_spl_view->SetItemState (selected, 0, wxLIST_STATE_SELECTED);
+				return;
+			}
+		}
 	}
 
 	_current_spl_view->DeleteAllItems ();