X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Fcontent_menu.cc;h=c5dc1606d790bb577a0a5ba5382b218febc45867;hb=c967096b5061e4bbb3a275d4daf04a2b9c084468;hp=1a409fa6cef18fb4168495ab3b58fb55c2ea8109;hpb=9a9f2cf479127bf9b3eff726f4f73103acae9b85;p=dcpomatic.git diff --git a/src/wx/content_menu.cc b/src/wx/content_menu.cc index 1a409fa6c..c5dc1606d 100644 --- a/src/wx/content_menu.cc +++ b/src/wx/content_menu.cc @@ -1,96 +1,450 @@ /* - Copyright (C) 2013 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 . */ -#include -#include "lib/playlist.h" -#include "lib/film.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 "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 "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, + /* 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 (shared_ptr f, wxWindow* p) +ContentMenu::ContentMenu (wxWindow* p) : _menu (new wxMenu) - , _film (f) , _parent (p) + , _pop_up_open (false) { - _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...")); + _properties = _menu->Append (ID_properties, _("Properties...")); + _re_examine = _menu->Append (ID_re_examine, _("Re-examine...")); _menu->AppendSeparator (); - _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); -} + _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")); -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 -ContentMenu::popup (ContentList c, wxPoint p) +ContentMenu::popup (weak_ptr film, ContentList c, TimelineContentViewList v, wxPoint p) { + _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; + 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 ()); + if (dcp) { + _kdm->Enable (dcp->encrypted ()); + _ov->Enable (dcp->needs_assets ()); + try { + DCPExaminer ex (dcp, true); + 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 -ContentMenu::repeat (wxCommandEvent &) +ContentMenu::repeat () { if (_content.empty ()) { return; } - - RepeatDialog d (_parent); - d.ShowModal (); - shared_ptr film = _film.lock (); + RepeatDialog* d = new RepeatDialog (_parent); + if (d->ShowModal() != wxID_OK) { + d->Destroy (); + return; + } + + shared_ptr film = _film.lock (); if (!film) { return; } - film->playlist()->repeat (_content, d.number ()); - d.Destroy (); + film->repeat_content (_content, d->number ()); + d->Destroy (); _content.clear (); + _views.clear (); } void -ContentMenu::remove (wxCommandEvent &) +ContentMenu::join () +{ + vector > fc; + BOOST_FOREACH (shared_ptr i, _content) { + shared_ptr f = dynamic_pointer_cast (i); + if (f) { + fc.push_back (f); + } + } + + DCPOMATIC_ASSERT (fc.size() > 1); + + shared_ptr film = _film.lock (); + if (!film) { + return; + } + + try { + 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 ())); + } +} + +void +ContentMenu::remove () { if (_content.empty ()) { return; } - shared_ptr film = _film.lock (); + shared_ptr 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. + */ + BOOST_FOREACH (shared_ptr i, _content) { + shared_ptr fc = dynamic_pointer_cast (i); + if (!fc) { + continue; + } + + shared_ptr video; + shared_ptr audio; + + 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) { + 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 +ContentMenu::find_missing () +{ + if (_content.size() != 1) { + return; + } + + shared_ptr film = _film.lock (); + if (!film) { + return; + } + + /* XXX: a bit nasty */ + shared_ptr ic = dynamic_pointer_cast (_content.front ()); + 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); + 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); + r = d->ShowModal (); + path = wx_to_std (d->GetPath ()); + d->Destroy (); + } + + 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; + } + + 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 +ContentMenu::re_examine () +{ + shared_ptr film = _film.lock (); + if (!film) { + return; + } + + BOOST_FOREACH (shared_ptr i, _content) { + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, i))); + } +} + +void +ContentMenu::maybe_found_missing (weak_ptr j, weak_ptr oc, weak_ptr nc) +{ + shared_ptr job = j.lock (); + if (!job || !job->finished_ok ()) { + return; + } + + shared_ptr old_content = oc.lock (); + shared_ptr 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_paths (new_content->paths()); +} + +void +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_ov (wx_to_std (d->GetPath ())); + shared_ptr film = _film.lock (); + DCPOMATIC_ASSERT (film); + JobManager::instance()->add (shared_ptr (new ExamineContentJob (film, dcp))); + } + + d->Destroy (); +} + +void +ContentMenu::properties () +{ + 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, true); + 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))); +}