src/wx/content_menu.cc

   1 /*
   2     Copyright (C) 2013 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 "content_menu.h"
  30 #include "repeat_dialog.h"
  31 #include "wx_util.h"
  32
  33 using std::cout;
  34 using std::vector;
  35 using boost::shared_ptr;
  36 using boost::weak_ptr;
  37 using boost::dynamic_pointer_cast;
  38
  39 enum {
  40         ID_repeat = 1,
  41         ID_join,
  42         ID_find_missing,
  43         ID_remove
  44 };
  45
  46 ContentMenu::ContentMenu (shared_ptr<Film> f, wxWindow* p)
  47         : _menu (new wxMenu)
  48         , _film (f)
  49         , _parent (p)
  50 {
  51         _repeat = _menu->Append (ID_repeat, _("Repeat..."));
  52         _join = _menu->Append (ID_join, _("Join"));
  53         _find_missing = _menu->Append (ID_find_missing, _("Find missing..."));
  54         _menu->AppendSeparator ();
  55         _remove = _menu->Append (ID_remove, _("Remove"));
  56
  57         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::repeat, this), ID_repeat);
  58         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::join, this), ID_join);
  59         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::find_missing, this), ID_find_missing);
  60         _parent->Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&ContentMenu::remove, this), ID_remove);
  61 }
  62
  63 ContentMenu::~ContentMenu ()
  64 {
  65         delete _menu;
  66 }
  67
  68 void
  69 ContentMenu::popup (ContentList c, wxPoint p)
  70 {
  71         _content = c;
  72         _repeat->Enable (!_content.empty ());
  73
  74         int n = 0;
  75         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
  76                 if (dynamic_pointer_cast<FFmpegContent> (*i)) {
  77                         ++n;
  78                 }
  79         }
  80
  81         _join->Enable (n > 1);
  82
  83         _find_missing->Enable (_content.size() == 1 && !_content.front()->paths_valid ());
  84         _remove->Enable (!_content.empty ());
  85         _parent->PopupMenu (_menu, p);
  86 }
  87
  88 void
  89 ContentMenu::repeat ()
  90 {
  91         if (_content.empty ()) {
  92                 return;
  93         }
  94
  95         RepeatDialog* d = new RepeatDialog (_parent);
  96         if (d->ShowModal() != wxID_OK) {
  97                 d->Destroy ();
  98                 return;
  99         }
 100
 101         shared_ptr<const Film> film = _film.lock ();
 102         if (!film) {
 103                 return;
 104         }
 105
 106         film->playlist()->repeat (_content, d->number ());
 107         d->Destroy ();
 108
 109         _content.clear ();
 110 }
 111
 112 void
 113 ContentMenu::join ()
 114 {
 115         vector<shared_ptr<Content> > fc;
 116         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
 117                 shared_ptr<FFmpegContent> f = dynamic_pointer_cast<FFmpegContent> (*i);
 118                 if (f) {
 119                         fc.push_back (f);
 120                 }
 121         }
 122
 123         assert (fc.size() > 1);
 124
 125         shared_ptr<Film> film = _film.lock ();
 126         if (!film) {
 127                 return;
 128         }
 129
 130         try {
 131                 shared_ptr<FFmpegContent> joined (new FFmpegContent (film, fc));
 132                 for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
 133                         film->remove_content (*i);
 134                 }
 135                 film->add_content (joined);
 136         } catch (JoinError& e) {
 137                 error_dialog (_parent, std_to_wx (e.what ()));
 138         }
 139 }
 140
 141 void
 142 ContentMenu::remove ()
 143 {
 144         if (_content.empty ()) {
 145                 return;
 146         }
 147
 148         shared_ptr<const Film> film = _film.lock ();
 149         if (!film) {
 150                 return;
 151         }
 152
 153         film->playlist()->remove (_content);
 154
 155         _content.clear ();
 156 }
 157
 158 void
 159 ContentMenu::find_missing ()
 160 {
 161         if (_content.size() != 1) {
 162                 return;
 163         }
 164
 165         shared_ptr<const Film> film = _film.lock ();
 166         if (!film) {
 167                 return;
 168         }
 169
 170         shared_ptr<Content> content;
 171
 172         /* XXX: a bit nasty */
 173         shared_ptr<ImageContent> ic = dynamic_pointer_cast<ImageContent> (_content.front ());
 174         if (ic && !ic->still ()) {
 175                 wxDirDialog* d = new wxDirDialog (0, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
 176                 int const r = d->ShowModal ();
 177                 if (r == wxID_OK) {
 178                         content.reset (new ImageContent (film, boost::filesystem::path (wx_to_std (d->GetPath ()))));
 179                 }
 180                 d->Destroy ();
 181         } else {
 182                 wxFileDialog* d = new wxFileDialog (0, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"), wxFD_MULTIPLE);
 183                 int const r = d->ShowModal ();
 184                 if (r == wxID_OK) {
 185                         content = content_factory (film, wx_to_std (d->GetPath ()));
 186                 }
 187                 d->Destroy ();
 188         }
 189
 190         if (!content) {
 191                 return;
 192         }
 193
 194         shared_ptr<Job> j (new ExamineContentJob (film, content));
 195
 196         j->Finished.connect (
 197                 bind (
 198                         &ContentMenu::maybe_found_missing,
 199                         this,
 200                         boost::weak_ptr<Job> (j),
 201                         boost::weak_ptr<Content> (_content.front ()),
 202                         boost::weak_ptr<Content> (content)
 203                         )
 204                 );
 205
 206         JobManager::instance()->add (j);
 207 }
 208
 209 void
 210 ContentMenu::maybe_found_missing (weak_ptr<Job> j, weak_ptr<Content> oc, weak_ptr<Content> nc)
 211 {
 212         shared_ptr<Job> job = j.lock ();
 213         if (!job || !job->finished_ok ()) {
 214                 return;
 215         }
 216
 217         shared_ptr<Content> old_content = oc.lock ();
 218         shared_ptr<Content> new_content = nc.lock ();
 219         assert (old_content);
 220         assert (new_content);
 221
 222         if (new_content->digest() != old_content->digest()) {
 223                 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."));
 224                 return;
 225         }
 226
 227         old_content->set_path (new_content->path (0));
 228 }