Hand-apply bbfb370d7de28ec1e8f307865cc6253bb5d4366e from master; quicker digest calcu...
[dcpomatic.git] / src / wx / content_menu.cc
index 60a95a9f26ed8c5166849f114ee7cb3c87220ba1..3bdaf259110bf0e0fdf9b204c5059f86ace3d54d 100644 (file)
 */
 
 #include <wx/wx.h>
+#include <wx/dirdlg.h>
 #include "lib/playlist.h"
 #include "lib/film.h"
+#include "lib/image_content.h"
+#include "lib/content_factory.h"
+#include "lib/examine_content_job.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"
 
 using std::cout;
+using std::vector;
 using boost::shared_ptr;
+using boost::weak_ptr;
+using boost::dynamic_pointer_cast;
 
 enum {
        ID_repeat = 1,
+       ID_join,
+       ID_find_missing,
+       ID_re_examine,
+       ID_kdm,
        ID_remove
 };
 
-ContentMenu::ContentMenu (shared_ptr<Film> f, wxWindow* p)
+ContentMenu::ContentMenu (wxWindow* p)
        : _menu (new wxMenu)
-       , _film (f)
        , _parent (p)
 {
-       _menu->Append (ID_repeat, _("Repeat..."));
+       _repeat = _menu->Append (ID_repeat, _("Repeat..."));
+       _join = _menu->Append (ID_join, _("Join"));
+       _find_missing = _menu->Append (ID_find_missing, _("Find missing..."));
+       _re_examine = _menu->Append (ID_re_examine, _("Re-examine..."));
+       _kdm = _menu->Append (ID_kdm, _("Add KDM..."));
        _menu->AppendSeparator ();
-       _menu->Append (ID_remove, _("Remove"));
+       _remove = _menu->Append (ID_remove, _("Remove"));
 
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, &ContentMenu::repeat, this, ID_repeat);
-       _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, &ContentMenu::remove, this, ID_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 ()
@@ -50,35 +72,90 @@ ContentMenu::~ContentMenu ()
 }
 
 void
-ContentMenu::popup (ContentList c, wxPoint p)
+ContentMenu::popup (weak_ptr<Film> f, ContentList c, wxPoint p)
 {
+       _film = f;
        _content = c;
+       _repeat->Enable (!_content.empty ());
+
+       int n = 0;
+       for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
+               if (dynamic_pointer_cast<FFmpegContent> (*i)) {
+                       ++n;
+               }
+       }
+       
+       _join->Enable (n > 1);
+       
+       _find_missing->Enable (_content.size() == 1 && !_content.front()->paths_valid ());
+       _re_examine->Enable (!_content.empty ());
+
+       if (_content.size() == 1) {
+               shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
+               _kdm->Enable (dcp && dcp->encrypted ());
+       } else {
+               _kdm->Enable (false);
+       }
+       
+       _remove->Enable (!_content.empty ());
        _parent->PopupMenu (_menu, p);
 }
 
 void
-ContentMenu::repeat (wxCommandEvent &)
+ContentMenu::repeat ()
 {
        if (_content.empty ()) {
                return;
        }
                
-       RepeatDialog d (_parent);
-       d.ShowModal ();
+       RepeatDialog* d = new RepeatDialog (_parent);
+       if (d->ShowModal() != wxID_OK) {
+               d->Destroy ();
+               return;
+       }
 
        shared_ptr<const Film> film = _film.lock ();
        if (!film) {
                return;
        }
 
-       film->playlist()->repeat (_content, d.number ());
-       d.Destroy ();
+       film->playlist()->repeat (_content, d->number ());
+       d->Destroy ();
 
        _content.clear ();
 }
 
 void
-ContentMenu::remove (wxCommandEvent &)
+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);
+               if (f) {
+                       fc.push_back (f);
+               }
+       }
+
+       DCPOMATIC_ASSERT (fc.size() > 1);
+
+       shared_ptr<Film> film = _film.lock ();
+       if (!film) {
+               return;
+       }
+
+       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);
+       } catch (JoinError& e) {
+               error_dialog (_parent, std_to_wx (e.what ()));
+       }
+}
+
+void
+ContentMenu::remove ()
 {
        if (_content.empty ()) {
                return;
@@ -94,3 +171,106 @@ ContentMenu::remove (wxCommandEvent &)
        _content.clear ();
 }
 
+void
+ContentMenu::find_missing ()
+{
+       if (_content.size() != 1) {
+               return;
+       }
+
+       shared_ptr<const Film> film = _film.lock ();
+       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 ()) {
+               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 ()))));
+               }
+               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 ()));
+               }
+               d->Destroy ();
+       }
+
+       if (!content) {
+               return;
+       }
+
+       shared_ptr<Job> j (new ExamineContentJob (film, content));
+       
+       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);
+}
+
+void
+ContentMenu::re_examine ()
+{
+       shared_ptr<Film> film = _film.lock ();
+       if (!film) {
+               return;
+       }
+
+       for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
+               film->examine_content (*i);
+       }
+}
+
+void
+ContentMenu::maybe_found_missing (weak_ptr<Job> j, weak_ptr<Content> oc, weak_ptr<Content> nc)
+{
+       shared_ptr<Job> job = j.lock ();
+       if (!job || !job->finished_ok ()) {
+               return;
+       }
+
+       shared_ptr<Content> old_content = oc.lock ();
+       shared_ptr<Content> new_content = nc.lock ();
+       DCPOMATIC_ASSERT (old_content);
+       DCPOMATIC_ASSERT (new_content);
+
+       if (new_content->digest() != old_content->digest()) {
+               error_dialog (0, _("The content file(s) you specified are not the same as those that are missing.  Either try again with the correct content file or remove the missing content."));
+               return;
+       }
+
+       old_content->set_path (new_content->path (0));
+}
+
+void
+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) {
+               dcp->add_kdm (dcp::EncryptedKDM (dcp::file_to_string (wx_to_std (d->GetPath ()))));
+               shared_ptr<Film> film = _film.lock ();
+               DCPOMATIC_ASSERT (film);
+               film->examine_content (dcp);
+       }
+       
+       d->Destroy ();
+}