swaroop: set controls to 'stopped' when a playlist finishes.

[dcpomatic.git] / src / wx / swaroop_controls.cc

diff --git a/src/wx/swaroop_controls.cc b/src/wx/swaroop_controls.cc
index 78419a08c3b8e0ef08ca0bf017b61ebf5e71c59f..99b1fd0647531ae89c570f9e62c9338b354b2cf9 100644 (file)
--- 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,8 @@
 #include "lib/cross.h"
 #include "lib/scoped_temporary.h"
 #include "lib/internet.h"
+#include "lib/ffmpeg_content.h"
+#include "lib/compose.hpp"
 #include <dcp/raw_convert.h>
 #include <dcp/exceptions.h>
 #include <wx/listctrl.h>
@@ -41,6 +43,7 @@ using std::sort;
 using boost::shared_ptr;
 using boost::dynamic_pointer_cast;
 using boost::optional;
+using namespace dcpomatic;
 
 SwaroopControls::SwaroopControls (wxWindow* parent, shared_ptr<FilmViewer> viewer)
        : Controls (parent, viewer, false)
@@ -191,6 +194,18 @@ SwaroopControls::stopped ()
        _pause_button->Enable (false);
 }
 
+void
+SwaroopControls::decrement_allowed_shows ()
+{
+       if (_selected_playlist) {
+               SPL& spl = _playlists[*_selected_playlist];
+               spl.decrement_allowed_shows();
+               if (spl.path()) {
+                       spl.write (*spl.path());
+               }
+       }
+}
+
 void
 SwaroopControls::play_clicked ()
 {
@@ -227,6 +242,7 @@ SwaroopControls::stop_clicked ()
                update_current_content ();
        }
        _viewer->set_background_image (true);
+       decrement_allowed_shows ();
 }
 
 bool
@@ -348,7 +364,7 @@ SwaroopControls::get_kdm_from_url (shared_ptr<DCPContent> dcp)
        string url = Config::instance()->kdm_server_url();
        boost::algorithm::replace_all (url, "{CPL}", *dcp->cpl());
        optional<dcp::EncryptedKDM> kdm;
-       if (dcp->cpl() && !get_from_url(url, false, temp)) {
+       if (dcp->cpl() && !get_from_url(url, false, false, temp)) {
                try {
                        kdm = dcp::EncryptedKDM (dcp::file_to_string(temp.file()));
                        if (kdm->cpl_id() != dcp->cpl()) {
@@ -384,6 +400,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 ()
 {
@@ -406,6 +444,11 @@ SwaroopControls::spl_selection_changed ()
                return;
        }
 
+       if (!_playlists[selected].have_allowed_shows()) {
+               error_dialog (this, "There are no more allowed shows of this playlist.");
+               return;
+       }
+
        select_playlist (selected, 0);
 }
 
@@ -441,6 +484,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 ();
@@ -526,5 +586,8 @@ SwaroopControls::viewer_finished ()
                _selected_playlist_position = 0;
                _viewer->set_background_image (true);
                ResetFilm (shared_ptr<Film>(new Film(optional<boost::filesystem::path>())));
+               stopped ();
        }
+
+       decrement_allowed_shows ();
 }