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