BOOST_FOREACH.
[dcpomatic.git] / src / wx / content_menu.cc
index 749337b7518674c03b35ac2451272ebd84c26e12..d93fd31b2ce023c6730c7ba67d46389c711281ae 100644 (file)
@@ -1,24 +1,30 @@
 /*
-    Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2020 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
-#include <wx/wx.h>
-#include <wx/dirdlg.h>
+#include "content_menu.h"
+#include "repeat_dialog.h"
+#include "wx_util.h"
+#include "timeline_video_content_view.h"
+#include "timeline_audio_content_view.h"
+#include "content_properties_dialog.h"
+#include "content_advanced_dialog.h"
 #include "lib/playlist.h"
 #include "lib/film.h"
 #include "lib/image_content.h"
 #include "lib/job_manager.h"
 #include "lib/exceptions.h"
 #include "lib/dcp_content.h"
-#include "content_menu.h"
-#include "repeat_dialog.h"
-#include "wx_util.h"
+#include "lib/dcp_examiner.h"
+#include "lib/ffmpeg_content.h"
+#include "lib/audio_content.h"
+#include "lib/config.h"
+#include "lib/copy_dcp_details_to_film.h"
+#include <dcp/cpl.h>
+#include <dcp/exceptions.h>
+#include <dcp/decrypted_kdm.h>
+#include <wx/wx.h>
+#include <wx/dirdlg.h>
+#include <iostream>
 
 using std::cout;
 using std::vector;
-using boost::shared_ptr;
-using boost::weak_ptr;
-using boost::dynamic_pointer_cast;
+using std::exception;
+using std::list;
+using std::shared_ptr;
+using std::weak_ptr;
+using std::dynamic_pointer_cast;
+using boost::optional;
+#if BOOST_VERSION >= 106100
+using namespace boost::placeholders;
+#endif
+
 
 enum {
-       ID_repeat = 1,
+       /* Start at 256 so we can have IDs on _cpl_menu from 1 to 255 */
+       ID_repeat = 256,
        ID_join,
        ID_find_missing,
+       ID_properties,
+       ID_advanced,
        ID_re_examine,
        ID_kdm,
+       ID_ov,
+       ID_choose_cpl,
+       ID_set_dcp_settings,
        ID_remove
 };
 
 ContentMenu::ContentMenu (wxWindow* p)
        : _menu (new wxMenu)
        , _parent (p)
+       , _pop_up_open (false)
 {
        _repeat = _menu->Append (ID_repeat, _("Repeat..."));
        _join = _menu->Append (ID_join, _("Join"));
        _find_missing = _menu->Append (ID_find_missing, _("Find missing..."));
+       _properties = _menu->Append (ID_properties, _("Properties..."));
+       _advanced = _menu->Append (ID_advanced, _("Advanced settings..."));
        _re_examine = _menu->Append (ID_re_examine, _("Re-examine..."));
+       _menu->AppendSeparator ();
        _kdm = _menu->Append (ID_kdm, _("Add KDM..."));
+       _ov = _menu->Append (ID_ov, _("Add OV..."));
+       _cpl_menu = new wxMenu ();
+       _choose_cpl = _menu->Append (ID_choose_cpl, _("Choose CPL..."), _cpl_menu);
+       _set_dcp_settings = _menu->Append (ID_set_dcp_settings, _("Set project DCP settings from this DCP"));
        _menu->AppendSeparator ();
        _remove = _menu->Append (ID_remove, _("Remove"));
 
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::repeat, this), ID_repeat);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::join, this), ID_join);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::find_missing, this), ID_find_missing);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::re_examine, this), ID_re_examine);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::kdm, this), ID_kdm);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::remove, this), ID_remove);
-}
-
-ContentMenu::~ContentMenu ()
-{
-       delete _menu;
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::repeat, this), ID_repeat);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::join, this), ID_join);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::find_missing, this), ID_find_missing);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::properties, this), ID_properties);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::advanced, this), ID_advanced);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::re_examine, this), ID_re_examine);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::kdm, this), ID_kdm);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::ov, this), ID_ov);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::set_dcp_settings, this), ID_set_dcp_settings);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::remove, this), ID_remove);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::cpl_selected, this, _1), 1, ID_repeat - 1);
 }
 
 void
-ContentMenu::popup (weak_ptr<Film> f, ContentList c, wxPoint p)
+ContentMenu::popup (weak_ptr<Film> film, ContentList c, TimelineContentViewList v, wxPoint p)
 {
-       _film = f;
+       _film = film;
        _content = c;
+       _views = v;
+
+       int const N = _cpl_menu->GetMenuItemCount();
+       for (int i = 1; i <= N; ++i) {
+               _cpl_menu->Delete (i);
+       }
+
        _repeat->Enable (!_content.empty ());
 
        int n = 0;
-       for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
-               if (dynamic_pointer_cast<FFmpegContent> (*i)) {
+       for (auto i: _content) {
+               if (dynamic_pointer_cast<FFmpegContent> (i)) {
                        ++n;
                }
        }
-       
+
        _join->Enable (n > 1);
-       
+
        _find_missing->Enable (_content.size() == 1 && !_content.front()->paths_valid ());
+       _properties->Enable (_content.size() == 1);
+       _advanced->Enable (_content.size() == 1);
        _re_examine->Enable (!_content.empty ());
 
        if (_content.size() == 1) {
                shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
-               _kdm->Enable (dcp && dcp->encrypted ());
+               if (dcp) {
+                       _kdm->Enable (dcp->encrypted ());
+                       _ov->Enable (dcp->needs_assets ());
+                       _set_dcp_settings->Enable (static_cast<bool>(dcp));
+                       try {
+                               DCPExaminer ex (dcp, true);
+                               list<shared_ptr<dcp::CPL> > cpls = ex.cpls ();
+                               _choose_cpl->Enable (cpls.size() > 1);
+                               /* We can't have 0 as a menu item ID on OS X */
+                               int id = 1;
+                               for (auto i: cpls) {
+                                       wxMenuItem* item = _cpl_menu->AppendRadioItem (
+                                               id++,
+                                               wxString::Format (
+                                                       "%s (%s)",
+                                                       std_to_wx(i->annotation_text()).data(),
+                                                       std_to_wx(i->id()).data()
+                                                       )
+                                               );
+                                       item->Check (dcp->cpl() && dcp->cpl() == i->id());
+                               }
+                       } catch (dcp::ReadError &) {
+                               /* The DCP is probably missing */
+                       } catch (dcp::KDMDecryptionError &) {
+                               /* We have an incorrect KDM */
+                       } catch (KDMError &) {
+                               /* We have an incorrect KDM */
+                       }
+               } else {
+                       _kdm->Enable (false);
+                       _ov->Enable (false);
+                       _choose_cpl->Enable (false);
+                       _set_dcp_settings->Enable (false);
+               }
        } else {
                _kdm->Enable (false);
+               _set_dcp_settings->Enable (false);
        }
-       
+
        _remove->Enable (!_content.empty ());
+
+       _pop_up_open = true;
        _parent->PopupMenu (_menu, p);
+       _pop_up_open = false;
+}
+
+
+void
+ContentMenu::set_dcp_settings ()
+{
+       shared_ptr<Film> film = _film.lock ();
+       if (!film) {
+               return;
+       }
+
+       DCPOMATIC_ASSERT (_content.size() == 1);
+       shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent>(_content.front());
+       DCPOMATIC_ASSERT (dcp);
+       copy_dcp_details_to_film (dcp, film);
 }
 
+
 void
 ContentMenu::repeat ()
 {
        if (_content.empty ()) {
                return;
        }
-               
+
        RepeatDialog* d = new RepeatDialog (_parent);
        if (d->ShowModal() != wxID_OK) {
                d->Destroy ();
                return;
        }
 
-       shared_ptr<const Film> film = _film.lock ();
+       shared_ptr<Film> film = _film.lock ();
        if (!film) {
                return;
        }
 
-       film->playlist()->repeat (_content, d->number ());
+       film->repeat_content (_content, d->number ());
        d->Destroy ();
 
        _content.clear ();
+       _views.clear ();
 }
 
 void
 ContentMenu::join ()
 {
        vector<shared_ptr<Content> > fc;
-       for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
-               shared_ptr<FFmpegContent> f = dynamic_pointer_cast<FFmpegContent> (*i);
+       for (auto i: _content) {
+               shared_ptr<FFmpegContent> f = dynamic_pointer_cast<FFmpegContent> (i);
                if (f) {
                        fc.push_back (f);
                }
@@ -144,11 +242,9 @@ ContentMenu::join ()
        }
 
        try {
-               shared_ptr<FFmpegContent> joined (new FFmpegContent (film, fc));
-               for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
-                       film->remove_content (*i);
-               }
-               film->add_content (joined);
+               shared_ptr<FFmpegContent> joined (new FFmpegContent(fc));
+               film->remove_content (_content);
+               film->examine_and_add_content (joined);
        } catch (JoinError& e) {
                error_dialog (_parent, std_to_wx (e.what ()));
        }
@@ -161,14 +257,51 @@ ContentMenu::remove ()
                return;
        }
 
-       shared_ptr<const Film> film = _film.lock ();
+       shared_ptr<Film> film = _film.lock ();
        if (!film) {
                return;
        }
 
-       film->playlist()->remove (_content);
+       /* We are removing from the timeline if _views is not empty */
+       bool handled = false;
+       if (!_views.empty ()) {
+               /* Special case: we only remove FFmpegContent if its video view is selected;
+                  if not, and its audio view is selected, we unmap the audio.
+               */
+               for (auto i: _content) {
+                       shared_ptr<FFmpegContent> fc = dynamic_pointer_cast<FFmpegContent> (i);
+                       if (!fc) {
+                               continue;
+                       }
+
+                       shared_ptr<TimelineVideoContentView> video;
+                       shared_ptr<TimelineAudioContentView> audio;
+
+                       for (auto j: _views) {
+                               shared_ptr<TimelineVideoContentView> v = dynamic_pointer_cast<TimelineVideoContentView> (j);
+                               shared_ptr<TimelineAudioContentView> a = dynamic_pointer_cast<TimelineAudioContentView> (j);
+                               if (v && v->content() == fc) {
+                                       video = v;
+                               } else if (a && a->content() == fc) {
+                                       audio = a;
+                               }
+                       }
+
+                       if (!video && audio) {
+                               AudioMapping m = fc->audio->mapping ();
+                               m.unmap_all ();
+                               fc->audio->set_mapping (m);
+                               handled = true;
+                       }
+               }
+       }
+
+       if (!handled) {
+               film->remove_content (_content);
+       }
 
        _content.clear ();
+       _views.clear ();
 }
 
 void
@@ -182,44 +315,55 @@ ContentMenu::find_missing ()
        if (!film) {
                return;
        }
-       
-       shared_ptr<Content> content;
 
        /* XXX: a bit nasty */
        shared_ptr<ImageContent> ic = dynamic_pointer_cast<ImageContent> (_content.front ());
-       if (ic && !ic->still ()) {
+       shared_ptr<DCPContent> dc = dynamic_pointer_cast<DCPContent> (_content.front ());
+
+       int r = wxID_CANCEL;
+       boost::filesystem::path path;
+
+       if ((ic && !ic->still ()) || dc) {
                wxDirDialog* d = new wxDirDialog (0, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
-               int const r = d->ShowModal ();
-               if (r == wxID_OK) {
-                       content.reset (new ImageContent (film, boost::filesystem::path (wx_to_std (d->GetPath ()))));
-               }
+               r = d->ShowModal ();
+               path = wx_to_std (d->GetPath ());
                d->Destroy ();
        } else {
                wxFileDialog* d = new wxFileDialog (0, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"), wxFD_MULTIPLE);
-               int const r = d->ShowModal ();
-               if (r == wxID_OK) {
-                       content = content_factory (film, wx_to_std (d->GetPath ()));
-               }
+               r = d->ShowModal ();
+               path = wx_to_std (d->GetPath ());
                d->Destroy ();
        }
 
-       if (!content) {
+       list<shared_ptr<Content> > content;
+
+       if (r == wxID_OK) {
+               if (dc) {
+                       content.push_back (shared_ptr<DCPContent>(new DCPContent(path)));
+               } else {
+                       content = content_factory (path);
+               }
+       }
+
+       if (content.empty ()) {
                return;
        }
 
-       shared_ptr<Job> j (new ExamineContentJob (film, content));
-       
-       _job_connection = j->Finished.connect (
-               bind (
-                       &ContentMenu::maybe_found_missing,
-                       this,
-                       boost::weak_ptr<Job> (j),
-                       boost::weak_ptr<Content> (_content.front ()),
-                       boost::weak_ptr<Content> (content)
-                       )
-               );
-       
-       JobManager::instance()->add (j);
+       for (auto i: content) {
+               shared_ptr<Job> j (new ExamineContentJob (film, i));
+
+               j->Finished.connect (
+                       bind (
+                               &ContentMenu::maybe_found_missing,
+                               this,
+                               std::weak_ptr<Job> (j),
+                               std::weak_ptr<Content> (_content.front ()),
+                               std::weak_ptr<Content> (i)
+                               )
+                       );
+
+               JobManager::instance()->add (j);
+       }
 }
 
 void
@@ -230,8 +374,8 @@ ContentMenu::re_examine ()
                return;
        }
 
-       for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
-               film->examine_content (*i);
+       for (auto i: _content) {
+               JobManager::instance()->add (shared_ptr<Job> (new ExamineContentJob (film, i)));
        }
 }
 
@@ -253,7 +397,7 @@ ContentMenu::maybe_found_missing (weak_ptr<Job> j, weak_ptr<Content> oc, weak_pt
                return;
        }
 
-       old_content->set_path (new_content->path (0));
+       old_content->set_paths (new_content->paths());
 }
 
 void
@@ -262,15 +406,122 @@ ContentMenu::kdm ()
        DCPOMATIC_ASSERT (!_content.empty ());
        shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
        DCPOMATIC_ASSERT (dcp);
-       
+
        wxFileDialog* d = new wxFileDialog (_parent, _("Select KDM"));
-               
+
+       if (d->ShowModal() == wxID_OK) {
+               optional<dcp::EncryptedKDM> kdm;
+               try {
+                       kdm = dcp::EncryptedKDM (dcp::file_to_string(wx_to_std(d->GetPath()), MAX_KDM_SIZE));
+               } catch (exception& e) {
+                       error_dialog (_parent, _("Could not load KDM"), std_to_wx(e.what()));
+                       d->Destroy ();
+                       return;
+               }
+
+               /* Try to decrypt it to get an early preview of any errors */
+               try {
+                       decrypt_kdm_with_helpful_error (*kdm);
+               } catch (KDMError& e) {
+                       error_dialog (_parent, std_to_wx(e.summary()), std_to_wx(e.detail()));
+                       return;
+               } catch (exception& e) {
+                       error_dialog (_parent, e.what());
+                       return;
+               }
+
+               DCPExaminer ex (dcp, true);
+
+               bool kdm_matches_any_cpl = false;
+               for (auto i: ex.cpls()) {
+                       if (i->id() == kdm->cpl_id()) {
+                               kdm_matches_any_cpl = true;
+                       }
+               }
+
+               bool kdm_matches_selected_cpl = dcp->cpl() || kdm->cpl_id() == dcp->cpl().get();
+
+               if (!kdm_matches_any_cpl) {
+                       error_dialog (_parent, _("This KDM was not made for this DCP.  You will need a different one."));
+                       return;
+               }
+
+               if (!kdm_matches_selected_cpl && kdm_matches_any_cpl) {
+                       message_dialog (_parent, _("This KDM was made for one of the CPLs in this DCP, but not the currently selected one.  To play the currently-selected CPL you will need a different KDM."));
+               }
+
+               dcp->add_kdm (*kdm);
+
+               shared_ptr<Film> film = _film.lock ();
+               DCPOMATIC_ASSERT (film);
+               JobManager::instance()->add (shared_ptr<Job> (new ExamineContentJob (film, dcp)));
+       }
+
+       d->Destroy ();
+}
+
+void
+ContentMenu::ov ()
+{
+       DCPOMATIC_ASSERT (!_content.empty ());
+       shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
+       DCPOMATIC_ASSERT (dcp);
+
+       wxDirDialog* d = new wxDirDialog (_parent, _("Select OV"));
+
        if (d->ShowModal() == wxID_OK) {
-               dcp->add_kdm (dcp::EncryptedKDM (dcp::file_to_string (wx_to_std (d->GetPath ()))));
+               dcp->add_ov (wx_to_std (d->GetPath ()));
                shared_ptr<Film> film = _film.lock ();
                DCPOMATIC_ASSERT (film);
-               film->examine_content (dcp);
+               JobManager::instance()->add (shared_ptr<Job> (new ExamineContentJob (film, dcp)));
        }
-       
+
+       d->Destroy ();
+}
+
+void
+ContentMenu::properties ()
+{
+       shared_ptr<Film> film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+       ContentPropertiesDialog* d = new ContentPropertiesDialog (_parent, film, _content.front());
+       d->ShowModal ();
        d->Destroy ();
 }
+
+
+void
+ContentMenu::advanced ()
+{
+       ContentAdvancedDialog* d = new ContentAdvancedDialog (_parent, _content.front());
+       d->ShowModal ();
+       d->Destroy ();
+}
+
+
+void
+ContentMenu::cpl_selected (wxCommandEvent& ev)
+{
+       if (!_pop_up_open) {
+               return;
+       }
+
+       DCPOMATIC_ASSERT (!_content.empty ());
+       shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
+       DCPOMATIC_ASSERT (dcp);
+
+       DCPExaminer ex (dcp, true);
+       list<shared_ptr<dcp::CPL> > cpls = ex.cpls ();
+       DCPOMATIC_ASSERT (ev.GetId() > 0);
+       DCPOMATIC_ASSERT (ev.GetId() <= int (cpls.size()));
+
+       list<shared_ptr<dcp::CPL> >::const_iterator i = cpls.begin ();
+       for (int j = 0; j < ev.GetId() - 1; ++j) {
+               ++i;
+       }
+
+       dcp->set_cpl ((*i)->id ());
+       shared_ptr<Film> film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+       JobManager::instance()->add (shared_ptr<Job> (new ExamineContentJob (film, dcp)));
+}