X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Fcontent_menu.cc;h=e5cadd7942870f90512d57f2320fc1cd2f2e7a88;hb=3044ef060894c62e8a1cef15ad14078001093982;hp=3cf9f23dcf636f2544fe7edf1668bb2fcee524b4;hpb=53eea12d3e0d7925d5949de92859bc358ee0adcc;p=dcpomatic.git diff --git a/src/wx/content_menu.cc b/src/wx/content_menu.cc index 3cf9f23dc..e5cadd794 100644 --- a/src/wx/content_menu.cc +++ b/src/wx/content_menu.cc @@ -1,19 +1,20 @@ /* - Copyright (C) 2013-2015 Carl Hetherington + Copyright (C) 2013-2016 Carl Hetherington - 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 . */ @@ -31,50 +32,65 @@ #include "lib/job_manager.h" #include "lib/exceptions.h" #include "lib/dcp_content.h" +#include "lib/dcp_examiner.h" +#include "lib/ffmpeg_content.h" +#include "lib/audio_content.h" +#include "lib/config.h" +#include +#include #include #include +#include +#include using std::cout; using std::vector; +using std::exception; +using std::list; using boost::shared_ptr; using boost::weak_ptr; using boost::dynamic_pointer_cast; 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_re_examine, ID_kdm, + ID_ov, + ID_choose_cpl, 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...")); _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); _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::properties, this), ID_properties); - _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::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::remove, this), ID_remove); + _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::cpl_selected, this, _1), 1, ID_repeat - 1); } void @@ -83,30 +99,70 @@ ContentMenu::popup (weak_ptr film, ContentList c, TimelineContentViewList _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 (*i)) { + BOOST_FOREACH (shared_ptr i, _content) { + if (dynamic_pointer_cast (i)) { ++n; } } - + _join->Enable (n > 1); - + _find_missing->Enable (_content.size() == 1 && !_content.front()->paths_valid ()); _properties->Enable (_content.size() == 1); _re_examine->Enable (!_content.empty ()); if (_content.size() == 1) { shared_ptr dcp = dynamic_pointer_cast (_content.front ()); - _kdm->Enable (dcp && dcp->encrypted ()); + if (dcp) { + _kdm->Enable (dcp->encrypted ()); + _ov->Enable (dcp->needs_assets ()); + try { + DCPExaminer ex (dcp); + list > 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; + BOOST_FOREACH (shared_ptr 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::DCPReadError &) { + /* 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); + } } else { _kdm->Enable (false); } - + _remove->Enable (!_content.empty ()); + + _pop_up_open = true; _parent->PopupMenu (_menu, p); + _pop_up_open = false; } void @@ -115,7 +171,7 @@ ContentMenu::repeat () if (_content.empty ()) { return; } - + RepeatDialog* d = new RepeatDialog (_parent); if (d->ShowModal() != wxID_OK) { d->Destroy (); @@ -138,8 +194,8 @@ void ContentMenu::join () { vector > fc; - for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) { - shared_ptr f = dynamic_pointer_cast (*i); + BOOST_FOREACH (shared_ptr i, _content) { + shared_ptr f = dynamic_pointer_cast (i); if (f) { fc.push_back (f); } @@ -153,11 +209,9 @@ ContentMenu::join () } try { - shared_ptr 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 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 ())); } @@ -181,18 +235,18 @@ ContentMenu::remove () /* 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 (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) { - shared_ptr fc = dynamic_pointer_cast (*i); + BOOST_FOREACH (shared_ptr i, _content) { + shared_ptr fc = dynamic_pointer_cast (i); if (!fc) { continue; } - + shared_ptr video; shared_ptr audio; - for (TimelineContentViewList::iterator i = _views.begin(); i != _views.end(); ++i) { - shared_ptr v = dynamic_pointer_cast (*i); - shared_ptr a = dynamic_pointer_cast (*i); + BOOST_FOREACH (shared_ptr j, _views) { + shared_ptr v = dynamic_pointer_cast (j); + shared_ptr a = dynamic_pointer_cast (j); if (v && v->content() == fc) { video = v; } else if (a && a->content() == fc) { @@ -201,9 +255,9 @@ ContentMenu::remove () } if (!video && audio) { - AudioMapping m = fc->audio_mapping (); + AudioMapping m = fc->audio->mapping (); m.unmap_all (); - fc->set_audio_mapping (m); + fc->audio->set_mapping (m); handled = true; } } @@ -228,44 +282,55 @@ ContentMenu::find_missing () if (!film) { return; } - - shared_ptr content; /* XXX: a bit nasty */ shared_ptr ic = dynamic_pointer_cast (_content.front ()); - if (ic && !ic->still ()) { + shared_ptr dc = dynamic_pointer_cast (_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 > content; + + if (r == wxID_OK) { + if (dc) { + content.push_back (shared_ptr(new DCPContent(path))); + } else { + content = content_factory (path); + } + } + + if (content.empty ()) { return; } - shared_ptr j (new ExamineContentJob (film, content)); - - _job_connection = j->Finished.connect ( - bind ( - &ContentMenu::maybe_found_missing, - this, - boost::weak_ptr (j), - boost::weak_ptr (_content.front ()), - boost::weak_ptr (content) - ) - ); - - JobManager::instance()->add (j); + BOOST_FOREACH (shared_ptr i, content) { + shared_ptr j (new ExamineContentJob (film, i)); + + j->Finished.connect ( + bind ( + &ContentMenu::maybe_found_missing, + this, + boost::weak_ptr (j), + boost::weak_ptr (_content.front ()), + boost::weak_ptr (i) + ) + ); + + JobManager::instance()->add (j); + } } void @@ -276,8 +341,8 @@ ContentMenu::re_examine () return; } - for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) { - film->examine_content (*i); + BOOST_FOREACH (shared_ptr i, _content) { + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, i))); } } @@ -299,7 +364,7 @@ ContentMenu::maybe_found_missing (weak_ptr j, weak_ptr oc, weak_pt return; } - old_content->set_path (new_content->path (0)); + old_content->set_paths (new_content->paths()); } void @@ -308,23 +373,78 @@ ContentMenu::kdm () DCPOMATIC_ASSERT (!_content.empty ()); shared_ptr dcp = dynamic_pointer_cast (_content.front ()); DCPOMATIC_ASSERT (dcp); - + wxFileDialog* d = new wxFileDialog (_parent, _("Select KDM")); - + + if (d->ShowModal() == wxID_OK) { + try { + dcp->add_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; + } + + shared_ptr film = _film.lock (); + DCPOMATIC_ASSERT (film); + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, dcp))); + } + + d->Destroy (); +} + +void +ContentMenu::ov () +{ + DCPOMATIC_ASSERT (!_content.empty ()); + shared_ptr dcp = dynamic_pointer_cast (_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.lock (); DCPOMATIC_ASSERT (film); - film->examine_content (dcp); + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, dcp))); } - + d->Destroy (); } void ContentMenu::properties () { - ContentPropertiesDialog* d = new ContentPropertiesDialog (_parent, _content.front ()); + shared_ptr film = _film.lock (); + DCPOMATIC_ASSERT (film); + ContentPropertiesDialog* d = new ContentPropertiesDialog (_parent, film, _content.front()); d->ShowModal (); d->Destroy (); } + +void +ContentMenu::cpl_selected (wxCommandEvent& ev) +{ + if (!_pop_up_open) { + return; + } + + DCPOMATIC_ASSERT (!_content.empty ()); + shared_ptr dcp = dynamic_pointer_cast (_content.front ()); + DCPOMATIC_ASSERT (dcp); + + DCPExaminer ex (dcp); + list > cpls = ex.cpls (); + DCPOMATIC_ASSERT (ev.GetId() > 0); + DCPOMATIC_ASSERT (ev.GetId() <= int (cpls.size())); + + list >::const_iterator i = cpls.begin (); + for (int j = 0; j < ev.GetId() - 1; ++j) { + ++i; + } + + dcp->set_cpl ((*i)->id ()); + shared_ptr film = _film.lock (); + DCPOMATIC_ASSERT (film); + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, dcp))); +}