swaroop: fix crash when there's an incorrect ecinema KDM.
[dcpomatic.git] / src / wx / swaroop_controls.cc
index f93cc66f1eefe76f14ac6a057b11c4cb0cd0e8ff..5ce6c45fc7a637f6e6c1e99124d596f7698cb422 100644 (file)
@@ -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)
@@ -101,9 +104,6 @@ SwaroopControls::SwaroopControls (wxWindow* parent, shared_ptr<FilmViewer> viewe
 
        _v_sizer->Add (e_sizer, 1, wxEXPAND);
 
-       _log = new wxTextCtrl (this, wxID_ANY, wxT(""), wxDefaultPosition, wxSize(-1, 200), wxTE_READONLY | wxTE_MULTILINE);
-       _v_sizer->Add (_log, 0, wxALL | wxEXPAND, DCPOMATIC_SIZER_GAP);
-
        _play_button->Bind     (wxEVT_BUTTON, boost::bind(&SwaroopControls::play_clicked,  this));
        _pause_button->Bind    (wxEVT_BUTTON, boost::bind(&SwaroopControls::pause_clicked, this));
        _stop_button->Bind     (wxEVT_BUTTON, boost::bind(&SwaroopControls::stop_clicked,  this));
@@ -118,13 +118,17 @@ SwaroopControls::SwaroopControls (wxWindow* parent, shared_ptr<FilmViewer> viewe
 
        _content_view->update ();
        update_playlist_directory ();
+
+       _viewer->set_background_image (true);
 }
 
 void
 SwaroopControls::check_restart ()
 {
+       cout << "check_restart called\n";
        FILE* f = fopen_boost (Config::path("position"), "r");
        if (!f) {
+               cout << "could not open position file (" << errno << ")\n";
                return;
        }
 
@@ -133,10 +137,12 @@ SwaroopControls::check_restart ()
        int64_t time;
        fscanf (f, "%63s %d %ld", id, &index, &time);
 
+       cout << "Looking for playlist " << id << " to restart.\n";
+
        for (size_t i = 0; i < _playlists.size(); ++i) {
                if (_playlists[i].id() == id) {
-                       _selected_playlist = i;
-                       _selected_playlist_position = index;
+                       cout << "Found playlist " << id << "\n";
+                       select_playlist (i, index);
                        update_current_content ();
                        _viewer->seek (DCPTime(time), false);
                        _viewer->start ();
@@ -149,7 +155,8 @@ SwaroopControls::check_restart ()
 void
 SwaroopControls::viewer_position_changed ()
 {
-       if (!_selected_playlist || !_viewer->playing() || _viewer->position().get() % DCPTime::HZ) {
+       /* Write position every two minutes if we're playing */
+       if (!_selected_playlist || !_viewer->playing() || _viewer->position().get() % (2 * 60 * DCPTime::HZ)) {
                return;
        }
 
@@ -160,6 +167,10 @@ SwaroopControls::viewer_position_changed ()
                        + " " + dcp::raw_convert<string>(_viewer->position().get());
 
                checked_fwrite (p.c_str(), p.length(), f, Config::path("position"));
+#ifdef DCPOMATIC_LINUX
+               fflush (f);
+               fsync (fileno(f));
+#endif
                fclose (f);
        }
 }
@@ -170,8 +181,10 @@ SwaroopControls::started ()
        Controls::started ();
        _play_button->Enable (false);
        _pause_button->Enable (true);
+       _viewer->set_background_image (false);
 }
 
+/** Called when the viewer finishes a single piece of content, or it is explicitly stopped */
 void
 SwaroopControls::stopped ()
 {
@@ -180,6 +193,29 @@ SwaroopControls::stopped ()
        _pause_button->Enable (false);
 }
 
+void
+SwaroopControls::deselect_playlist ()
+{
+       long int const selected = _spl_view->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
+       if (selected != -1) {
+               _selected_playlist = boost::none;
+               _spl_view->SetItemState (selected, 0, wxLIST_STATE_SELECTED);
+       }
+       ResetFilm (shared_ptr<Film>(new Film(optional<boost::filesystem::path>())));
+}
+
+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 ()
 {
@@ -215,6 +251,9 @@ SwaroopControls::stop_clicked ()
                _selected_playlist_position = 0;
                update_current_content ();
        }
+       _viewer->set_background_image (true);
+       decrement_allowed_shows ();
+       deselect_playlist ();
 }
 
 bool
@@ -254,6 +293,11 @@ SwaroopControls::next_clicked ()
 void
 SwaroopControls::log (wxString s)
 {
+       optional<boost::filesystem::path> log = Config::instance()->player_activity_log_file();
+       if (!log) {
+               return;
+       }
+
        struct timeval time;
        gettimeofday (&time, 0);
        char buffer[64];
@@ -261,14 +305,10 @@ SwaroopControls::log (wxString s)
        struct tm* t = localtime (&sec);
        strftime (buffer, 64, "%c", t);
        wxString ts = std_to_wx(string(buffer)) + N_(": ");
-       _log->SetValue(_log->GetValue() + ts + s + "\n");
-
-       optional<boost::filesystem::path> log = Config::instance()->player_log_file();
-       if (!log) {
+       FILE* f = fopen_boost (*log, "a");
+       if (!f) {
                return;
        }
-
-       FILE* f = fopen_boost (*log, "a");
        fprintf (f, "%s%s\n", wx_to_std(ts).c_str(), wx_to_std(s).c_str());
        fclose (f);
 }
@@ -336,7 +376,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()) {
@@ -372,6 +412,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 ()
 {
@@ -384,8 +446,7 @@ SwaroopControls::spl_selection_changed ()
 
        if (_playlists[selected].missing()) {
                error_dialog (this, "This playlist cannot be loaded as some content is missing.");
-               _selected_playlist = boost::none;
-               _spl_view->SetItemState (selected, 0, wxLIST_STATE_SELECTED);
+               deselect_playlist ();
                return;
        }
 
@@ -394,6 +455,17 @@ 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);
+}
+
+void
+SwaroopControls::select_playlist (int selected, int position)
+{
        log (wxString::Format("load-playlist %s", std_to_wx(_playlists[selected].name()).data()));
 
        wxProgressDialog dialog (_("DCP-o-matic"), "Loading playlist and KDMs");
@@ -417,9 +489,26 @@ SwaroopControls::spl_selection_changed ()
                        }
                        if (dcp->needs_kdm()) {
                                /* We didn't get a KDM for this */
-                               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);
+                               error_dialog (this, "This playlist cannot be loaded as a KDM is missing or incorrect.");
+                               deselect_playlist ();
+                               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, "This playlist cannot be loaded as a KDM is missing or incorrect.");
+                                       deselect_playlist ();
+                                       return;
+                               }
+                       } else {
+                               error_dialog (this, "This playlist cannot be loaded as a KDM is missing or incorrect.");
+                               deselect_playlist ();
                                return;
                        }
                }
@@ -438,7 +527,7 @@ SwaroopControls::spl_selection_changed ()
        }
 
        _selected_playlist = selected;
-       _selected_playlist_position = 0;
+       _selected_playlist_position = position;
        dialog.Pulse ();
        reset_film ();
        dialog.Pulse ();
@@ -489,6 +578,7 @@ SwaroopControls::update_current_content ()
        reset_film ();
 }
 
+/** One piece of content in our SPL has finished playing */
 void
 SwaroopControls::viewer_finished ()
 {
@@ -500,12 +590,18 @@ SwaroopControls::viewer_finished ()
 
        _selected_playlist_position++;
        if (_selected_playlist_position < int(_playlists[*_selected_playlist].get().size())) {
+               /* Next piece of content on the SPL */
                update_current_content ();
                if (!stop) {
                        _viewer->start ();
                }
        } else {
+               /* Finished the whole SPL */
                _selected_playlist_position = 0;
+               _viewer->set_background_image (true);
                ResetFilm (shared_ptr<Film>(new Film(optional<boost::filesystem::path>())));
+               decrement_allowed_shows ();
+               _play_button->Enable (true);
+               _pause_button->Enable (false);
        }
 }