src/wx/content_menu.cc

   1 /*
   2     Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <wx/wx.h>
  21 #include <wx/dirdlg.h>
  22 #include "lib/playlist.h"
  23 #include "lib/film.h"
  24 #include "lib/image_content.h"
  25 #include "lib/content_factory.h"
  26 #include "lib/examine_content_job.h"
  27 #include "lib/job_manager.h"
  28 #include "lib/exceptions.h"
  29 #include "lib/dcp_content.h"
  30 #include "content_menu.h"
  31 #include "repeat_dialog.h"
  32 #include "wx_util.h"
  33 #include "timeline_video_content_view.h"
  34 #include "timeline_audio_content_view.h"
  35
  36 using std::cout;
  37 using std::vector;
  38 using boost::shared_ptr;
  39 using boost::weak_ptr;
  40 using boost::dynamic_pointer_cast;
  41
  42 enum {
  43         ID_repeat = 1,
  44         ID_join,
  45         ID_find_missing,
  46         ID_re_examine,
  47         ID_kdm,
  48         ID_remove
  49 };
  50
  51 ContentMenu::ContentMenu (wxWindow* p)
  52         : _menu (new wxMenu)
  53         , _parent (p)
  54 {
  55         _repeat = _menu->Append (ID_repeat, _("Repeat..."));
  56         _join = _menu->Append (ID_join, _("Join"));
  57         _find_missing = _menu->Append (ID_find_missing, _("Find missing..."));
  58         _re_examine = _menu->Append (ID_re_examine, _("Re-examine..."));
  59         _kdm = _menu->Append (ID_kdm, _("Add KDM..."));
  60         _menu->AppendSeparator ();
  61         _remove = _menu->Append (ID_remove, _("Remove"));
  62
  63         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::repeat, this), ID_repeat);
  64         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::join, this), ID_join);
  65         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::find_missing, this), ID_find_missing);
  66         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::re_examine, this), ID_re_examine);
  67         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::kdm, this), ID_kdm);
  68         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::remove, this), ID_remove);
  69 }
  70
  71 ContentMenu::~ContentMenu ()
  72 {
  73         delete _menu;
  74 }
  75
  76 void
  77 ContentMenu::popup (weak_ptr<Film> f, ContentList c, TimelineContentViewList v, wxPoint p)
  78 {
  79         _film = f;
  80         _content = c;
  81         _views = v;
  82         _repeat->Enable (!_content.empty ());
  83
  84         int n = 0;
  85         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
  86                 if (dynamic_pointer_cast<FFmpegContent> (*i)) {
  87                         ++n;
  88                 }
  89         }
  90
  91         _join->Enable (n > 1);
  92
  93         _find_missing->Enable (_content.size() == 1 && !_content.front()->paths_valid ());
  94         _re_examine->Enable (!_content.empty ());
  95
  96         if (_content.size() == 1) {
  97                 shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
  98                 _kdm->Enable (dcp && dcp->encrypted ());
  99         } else {
 100                 _kdm->Enable (false);
 101         }
 102
 103         _remove->Enable (!_content.empty ());
 104         _parent->PopupMenu (_menu, p);
 105 }
 106
 107 void
 108 ContentMenu::repeat ()
 109 {
 110         if (_content.empty ()) {
 111                 return;
 112         }
 113
 114         RepeatDialog* d = new RepeatDialog (_parent);
 115         if (d->ShowModal() != wxID_OK) {
 116                 d->Destroy ();
 117                 return;
 118         }
 119
 120         shared_ptr<const Film> film = _film.lock ();
 121         if (!film) {
 122                 return;
 123         }
 124
 125         film->playlist()->repeat (_content, d->number ());
 126         d->Destroy ();
 127
 128         _content.clear ();
 129         _views.clear ();
 130 }
 131
 132 void
 133 ContentMenu::join ()
 134 {
 135         vector<shared_ptr<Content> > fc;
 136         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
 137                 shared_ptr<FFmpegContent> f = dynamic_pointer_cast<FFmpegContent> (*i);
 138                 if (f) {
 139                         fc.push_back (f);
 140                 }
 141         }
 142
 143         DCPOMATIC_ASSERT (fc.size() > 1);
 144
 145         shared_ptr<Film> film = _film.lock ();
 146         if (!film) {
 147                 return;
 148         }
 149
 150         try {
 151                 shared_ptr<FFmpegContent> joined (new FFmpegContent (film, fc));
 152                 for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
 153                         film->remove_content (*i);
 154                 }
 155                 film->add_content (joined);
 156         } catch (JoinError& e) {
 157                 error_dialog (_parent, std_to_wx (e.what ()));
 158         }
 159 }
 160
 161 void
 162 ContentMenu::remove ()
 163 {
 164         if (_content.empty ()) {
 165                 return;
 166         }
 167
 168         shared_ptr<const Film> film = _film.lock ();
 169         if (!film) {
 170                 return;
 171         }
 172
 173         /* We are removing from the timeline if _views is not empty */
 174         bool handled = false;
 175         if (!_views.empty ()) {
 176                 /* Special case: we only remove FFmpegContent if its video view is selected;
 177                    if not, and its audio view is selected, we unmap the audio.
 178                 */
 179                 for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
 180                         shared_ptr<FFmpegContent> fc = dynamic_pointer_cast<FFmpegContent> (*i);
 181                         if (!fc) {
 182                                 continue;
 183                         }
 184
 185                         shared_ptr<TimelineVideoContentView> video;
 186                         shared_ptr<TimelineAudioContentView> audio;
 187
 188                         for (TimelineContentViewList::iterator i = _views.begin(); i != _views.end(); ++i) {
 189                                 shared_ptr<TimelineVideoContentView> v = dynamic_pointer_cast<TimelineVideoContentView> (*i);
 190                                 shared_ptr<TimelineAudioContentView> a = dynamic_pointer_cast<TimelineAudioContentView> (*i);
 191                                 if (v && v->content() == fc) {
 192                                         video = v;
 193                                 } else if (a && a->content() == fc) {
 194                                         audio = a;
 195                                 }
 196                         }
 197
 198                         if (!video && audio) {
 199                                 AudioMapping m = fc->audio_mapping ();
 200                                 m.unmap_all ();
 201                                 fc->set_audio_mapping (m);
 202                                 handled = true;
 203                         }
 204                 }
 205         }
 206
 207         if (!handled) {
 208                 film->playlist()->remove (_content);
 209         }
 210
 211         _content.clear ();
 212         _views.clear ();
 213 }
 214
 215 void
 216 ContentMenu::find_missing ()
 217 {
 218         if (_content.size() != 1) {
 219                 return;
 220         }
 221
 222         shared_ptr<const Film> film = _film.lock ();
 223         if (!film) {
 224                 return;
 225         }
 226
 227         shared_ptr<Content> content;
 228
 229         /* XXX: a bit nasty */
 230         shared_ptr<ImageContent> ic = dynamic_pointer_cast<ImageContent> (_content.front ());
 231         if (ic && !ic->still ()) {
 232                 wxDirDialog* d = new wxDirDialog (0, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
 233                 int const r = d->ShowModal ();
 234                 if (r == wxID_OK) {
 235                         content.reset (new ImageContent (film, boost::filesystem::path (wx_to_std (d->GetPath ()))));
 236                 }
 237                 d->Destroy ();
 238         } else {
 239                 wxFileDialog* d = new wxFileDialog (0, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"), wxFD_MULTIPLE);
 240                 int const r = d->ShowModal ();
 241                 if (r == wxID_OK) {
 242                         content = content_factory (film, wx_to_std (d->GetPath ()));
 243                 }
 244                 d->Destroy ();
 245         }
 246
 247         if (!content) {
 248                 return;
 249         }
 250
 251         shared_ptr<Job> j (new ExamineContentJob (film, content));
 252
 253         _job_connection = j->Finished.connect (
 254                 bind (
 255                         &ContentMenu::maybe_found_missing,
 256                         this,
 257                         boost::weak_ptr<Job> (j),
 258                         boost::weak_ptr<Content> (_content.front ()),
 259                         boost::weak_ptr<Content> (content)
 260                         )
 261                 );
 262
 263         JobManager::instance()->add (j);
 264 }
 265
 266 void
 267 ContentMenu::re_examine ()
 268 {
 269         shared_ptr<Film> film = _film.lock ();
 270         if (!film) {
 271                 return;
 272         }
 273
 274         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
 275                 film->examine_content (*i);
 276         }
 277 }
 278
 279 void
 280 ContentMenu::maybe_found_missing (weak_ptr<Job> j, weak_ptr<Content> oc, weak_ptr<Content> nc)
 281 {
 282         shared_ptr<Job> job = j.lock ();
 283         if (!job || !job->finished_ok ()) {
 284                 return;
 285         }
 286
 287         shared_ptr<Content> old_content = oc.lock ();
 288         shared_ptr<Content> new_content = nc.lock ();
 289         DCPOMATIC_ASSERT (old_content);
 290         DCPOMATIC_ASSERT (new_content);
 291
 292         if (new_content->digest() != old_content->digest()) {
 293                 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."));
 294                 return;
 295         }
 296
 297         old_content->set_path (new_content->path (0));
 298 }
 299
 300 void
 301 ContentMenu::kdm ()
 302 {
 303         DCPOMATIC_ASSERT (!_content.empty ());
 304         shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (_content.front ());
 305         DCPOMATIC_ASSERT (dcp);
 306
 307         wxFileDialog* d = new wxFileDialog (_parent, _("Select KDM"));
 308
 309         if (d->ShowModal() == wxID_OK) {
 310                 dcp->add_kdm (dcp::EncryptedKDM (dcp::file_to_string (wx_to_std (d->GetPath ()))));
 311                 shared_ptr<Film> film = _film.lock ();
 312                 DCPOMATIC_ASSERT (film);
 313                 film->examine_content (dcp);
 314         }
 315
 316         d->Destroy ();
 317 }