src/tools/dcpomatic.cc

   1 /*
   2     Copyright (C) 2012 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 <iostream>
  21 #include <fstream>
  22 #include <boost/filesystem.hpp>
  23 #ifdef __WXMSW__
  24 #include <shellapi.h>
  25 #endif
  26 #ifdef __WXOSX__
  27 #include <ApplicationServices/ApplicationServices.h>
  28 #endif
  29 #include <wx/generic/aboutdlgg.h>
  30 #include <wx/stdpaths.h>
  31 #include <wx/cmdline.h>
  32 #include "wx/film_viewer.h"
  33 #include "wx/film_editor.h"
  34 #include "wx/job_manager_view.h"
  35 #include "wx/config_dialog.h"
  36 #include "wx/job_wrapper.h"
  37 #include "wx/wx_util.h"
  38 #include "wx/new_film_dialog.h"
  39 #include "wx/properties_dialog.h"
  40 #include "wx/wx_ui_signaller.h"
  41 #include "wx/about_dialog.h"
  42 #include "lib/film.h"
  43 #include "lib/config.h"
  44 #include "lib/util.h"
  45 #include "lib/version.h"
  46 #include "lib/ui_signaller.h"
  47 #include "lib/log.h"
  48 #include "lib/job_manager.h"
  49 #include "lib/transcode_job.h"
  50
  51 using std::cout;
  52 using std::string;
  53 using std::wstring;
  54 using std::stringstream;
  55 using std::map;
  56 using std::make_pair;
  57 using std::list;
  58 using std::exception;
  59 using std::ofstream;
  60 using boost::shared_ptr;
  61 using boost::dynamic_pointer_cast;
  62
  63 static FilmEditor* film_editor = 0;
  64 static FilmViewer* film_viewer = 0;
  65 static shared_ptr<Film> film;
  66 static std::string log_level;
  67 static std::string film_to_load;
  68 static std::string film_to_create;
  69 static wxMenu* jobs_menu = 0;
  70
  71 static void set_menu_sensitivity ();
  72
  73 class FilmChangedDialog
  74 {
  75 public:
  76         FilmChangedDialog ()
  77         {
  78                 _dialog = new wxMessageDialog (
  79                         0,
  80                         wxString::Format (_("Save changes to film \"%s\" before closing?"), std_to_wx (film->name ()).data()),
  81                         _("Film changed"),
  82                         wxYES_NO | wxYES_DEFAULT | wxICON_QUESTION
  83                         );
  84         }
  85
  86         ~FilmChangedDialog ()
  87         {
  88                 _dialog->Destroy ();
  89         }
  90
  91         int run ()
  92         {
  93                 return _dialog->ShowModal ();
  94         }
  95
  96 private:
  97         /* Not defined */
  98         FilmChangedDialog (FilmChangedDialog const &);
  99
 100         wxMessageDialog* _dialog;
 101 };
 102
 103
 104 void
 105 maybe_save_then_delete_film ()
 106 {
 107         if (!film) {
 108                 return;
 109         }
 110
 111         if (film->dirty ()) {
 112                 FilmChangedDialog d;
 113                 switch (d.run ()) {
 114                 case wxID_NO:
 115                         break;
 116                 case wxID_YES:
 117                         film->write_metadata ();
 118                         break;
 119                 }
 120         }
 121
 122         film.reset ();
 123 }
 124
 125 #define ALWAYS                  0x0
 126 #define NEEDS_FILM              0x1
 127 #define NOT_DURING_DCP_CREATION 0x2
 128
 129 map<wxMenuItem*, int> menu_items;
 130
 131 void
 132 add_item (wxMenu* menu, wxString text, int id, int sens)
 133 {
 134         wxMenuItem* item = menu->Append (id, text);
 135         menu_items.insert (make_pair (item, sens));
 136 }
 137
 138 void
 139 set_menu_sensitivity ()
 140 {
 141         list<shared_ptr<Job> > jobs = JobManager::instance()->get ();
 142         list<shared_ptr<Job> >::iterator i = jobs.begin();
 143         while (i != jobs.end() && dynamic_pointer_cast<TranscodeJob> (*i) == 0) {
 144                 ++i;
 145         }
 146         bool const dcp_creation = (i != jobs.end ());
 147
 148         for (map<wxMenuItem*, int>::iterator j = menu_items.begin(); j != menu_items.end(); ++j) {
 149
 150                 bool enabled = true;
 151
 152                 if ((j->second & NEEDS_FILM) && film == 0) {
 153                         enabled = false;
 154                 }
 155
 156                 if ((j->second & NOT_DURING_DCP_CREATION) && dcp_creation) {
 157                         enabled = false;
 158                 }
 159
 160                 j->first->Enable (enabled);
 161         }
 162 }
 163
 164 enum {
 165         ID_file_new = 1,
 166         ID_file_open,
 167         ID_file_save,
 168         ID_file_properties,
 169         ID_jobs_make_dcp,
 170         ID_jobs_send_dcp_to_tms,
 171         ID_jobs_show_dcp,
 172 };
 173
 174 void
 175 setup_menu (wxMenuBar* m)
 176 {
 177         wxMenu* file = new wxMenu;
 178         add_item (file, _("New..."), ID_file_new, ALWAYS);
 179         add_item (file, _("&Open..."), ID_file_open, ALWAYS);
 180         file->AppendSeparator ();
 181         add_item (file, _("&Save"), ID_file_save, NEEDS_FILM);
 182         file->AppendSeparator ();
 183         add_item (file, _("&Properties..."), ID_file_properties, NEEDS_FILM);
 184 #ifndef __WXOSX__
 185         file->AppendSeparator ();
 186 #endif
 187
 188 #ifdef __WXOSX__
 189         add_item (file, _("&Exit"), wxID_EXIT, ALWAYS);
 190 #else
 191         add_item (file, _("&Quit"), wxID_EXIT, ALWAYS);
 192 #endif
 193
 194
 195 #ifdef __WXOSX__
 196         add_item (file, _("&Preferences..."), wxID_PREFERENCES, ALWAYS);
 197 #else
 198         wxMenu* edit = new wxMenu;
 199         add_item (edit, _("&Preferences..."), wxID_PREFERENCES, ALWAYS);
 200 #endif
 201
 202         jobs_menu = new wxMenu;
 203         add_item (jobs_menu, _("&Make DCP"), ID_jobs_make_dcp, NEEDS_FILM | NOT_DURING_DCP_CREATION);
 204         add_item (jobs_menu, _("&Send DCP to TMS"), ID_jobs_send_dcp_to_tms, NEEDS_FILM | NOT_DURING_DCP_CREATION);
 205         add_item (jobs_menu, _("S&how DCP"), ID_jobs_show_dcp, NEEDS_FILM | NOT_DURING_DCP_CREATION);
 206
 207         wxMenu* help = new wxMenu;
 208 #ifdef __WXOSX__
 209         add_item (help, _("About DCP-o-matic"), wxID_ABOUT, ALWAYS);
 210 #else
 211         add_item (help, _("About"), wxID_ABOUT, ALWAYS);
 212 #endif
 213
 214         m->Append (file, _("&File"));
 215 #ifndef __WXOSX__
 216         m->Append (edit, _("&Edit"));
 217 #endif
 218         m->Append (jobs_menu, _("&Jobs"));
 219         m->Append (help, _("&Help"));
 220 }
 221
 222 class Frame : public wxFrame
 223 {
 224 public:
 225         Frame (wxString const & title)
 226                 : wxFrame (NULL, -1, title)
 227         {
 228                 wxMenuBar* bar = new wxMenuBar;
 229                 setup_menu (bar);
 230                 SetMenuBar (bar);
 231
 232                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::file_new, this),             ID_file_new);
 233                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::file_open, this),            ID_file_open);
 234                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::file_save, this),            ID_file_save);
 235                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::file_properties, this),      ID_file_properties);
 236                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::file_exit, this),            wxID_EXIT);
 237                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::edit_preferences, this),     wxID_PREFERENCES);
 238                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::jobs_make_dcp, this),        ID_jobs_make_dcp);
 239                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::jobs_send_dcp_to_tms, this), ID_jobs_send_dcp_to_tms);
 240                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::jobs_show_dcp, this),        ID_jobs_show_dcp);
 241                 Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&Frame::help_about, this),           wxID_ABOUT);
 242                 Bind (wxEVT_MENU_OPEN, boost::bind (&Frame::menu_opened, this, _1));
 243
 244                 /* Use a panel as the only child of the Frame so that we avoid
 245                    the dark-grey background on Windows.
 246                 */
 247                 wxPanel* overall_panel = new wxPanel (this, wxID_ANY);
 248
 249                 film_editor = new FilmEditor (film, overall_panel);
 250                 film_viewer = new FilmViewer (film, overall_panel);
 251                 JobManagerView* job_manager_view = new JobManagerView (overall_panel, static_cast<JobManagerView::Buttons> (0));
 252
 253                 wxBoxSizer* right_sizer = new wxBoxSizer (wxVERTICAL);
 254                 right_sizer->Add (film_viewer, 2, wxEXPAND | wxALL, 6);
 255                 right_sizer->Add (job_manager_view, 1, wxEXPAND | wxALL, 6);
 256
 257                 wxBoxSizer* main_sizer = new wxBoxSizer (wxHORIZONTAL);
 258                 main_sizer->Add (film_editor, 1, wxEXPAND | wxALL, 6);
 259                 main_sizer->Add (right_sizer, 2, wxEXPAND | wxALL, 6);
 260
 261                 set_menu_sensitivity ();
 262
 263                 film_editor->FileChanged.connect (bind (&Frame::file_changed, this, _1));
 264                 if (film) {
 265                         file_changed (film->directory ());
 266                 } else {
 267                         file_changed ("");
 268                 }
 269
 270                 JobManager::instance()->ActiveJobsChanged.connect (boost::bind (set_menu_sensitivity));
 271
 272                 set_film ();
 273                 overall_panel->SetSizer (main_sizer);
 274         }
 275
 276 private:
 277
 278         void menu_opened (wxMenuEvent& ev)
 279         {
 280                 if (ev.GetMenu() != jobs_menu) {
 281                         return;
 282                 }
 283
 284                 bool const have_dcp = film && film->have_dcp();
 285                 jobs_menu->Enable (ID_jobs_send_dcp_to_tms, have_dcp);
 286                 jobs_menu->Enable (ID_jobs_show_dcp, have_dcp);
 287         }
 288
 289         void set_film ()
 290         {
 291                 film_viewer->set_film (film);
 292                 film_editor->set_film (film);
 293                 set_menu_sensitivity ();
 294         }
 295
 296         void file_changed (string f)
 297         {
 298                 stringstream s;
 299                 s << wx_to_std (_("DCP-o-matic"));
 300                 if (!f.empty ()) {
 301                         s << " - " << f;
 302                 }
 303
 304                 SetTitle (std_to_wx (s.str()));
 305         }
 306
 307         void file_new ()
 308         {
 309                 NewFilmDialog* d = new NewFilmDialog (this);
 310                 int const r = d->ShowModal ();
 311
 312                 if (r == wxID_OK) {
 313
 314                         if (boost::filesystem::is_directory (d->get_path()) && !boost::filesystem::is_empty(d->get_path())) {
 315                                 if (!confirm_dialog (
 316                                             this,
 317                                             std_to_wx (
 318                                                     String::compose (wx_to_std (_("The directory %1 already exists and is not empty.  "
 319                                                                                   "Are you sure you want to use it?")),
 320                                                                      d->get_path().c_str())
 321                                                     )
 322                                             )) {
 323                                         return;
 324                                 }
 325                         } else if (boost::filesystem::is_regular_file (d->get_path())) {
 326                                 error_dialog (
 327                                         this,
 328                                         String::compose (wx_to_std (_("%1 already exists as a file, so you cannot use it for a new film.")), d->get_path().c_str())
 329                                         );
 330                                 return;
 331                         }
 332
 333                         maybe_save_then_delete_film ();
 334                         film.reset (new Film (d->get_path ()));
 335                         film->write_metadata ();
 336                         film->log()->set_level (log_level);
 337                         film->set_name (boost::filesystem::path (d->get_path()).filename().generic_string());
 338                         set_film ();
 339                 }
 340
 341                 d->Destroy ();
 342         }
 343
 344         void file_open ()
 345         {
 346                 wxDirDialog* c = new wxDirDialog (this, _("Select film to open"), wxStandardPaths::Get().GetDocumentsDir(), wxDEFAULT_DIALOG_STYLE | wxDD_DIR_MUST_EXIST);
 347                 int r;
 348                 while (1) {
 349                         r = c->ShowModal ();
 350                         if (r == wxID_OK && c->GetPath() == wxStandardPaths::Get().GetDocumentsDir()) {
 351                                 error_dialog (this, _("You did not select a folder.  Make sure that you select a folder before clicking Open."));
 352                         } else {
 353                                 break;
 354                         }
 355                 }
 356
 357                 if (r == wxID_OK) {
 358                         maybe_save_then_delete_film ();
 359                         try {
 360                                 film.reset (new Film (wx_to_std (c->GetPath ())));
 361                                 film->read_metadata ();
 362                                 film->log()->set_level (log_level);
 363                                 set_film ();
 364                         } catch (std::exception& e) {
 365                                 wxString p = c->GetPath ();
 366                                 wxCharBuffer b = p.ToUTF8 ();
 367                                 error_dialog (this, wxString::Format (_("Could not open film at %s (%s)"), p.data(), std_to_wx (e.what()).data()));
 368                         }
 369                 }
 370
 371                 c->Destroy ();
 372         }
 373
 374         void file_save ()
 375         {
 376                 film->write_metadata ();
 377         }
 378
 379         void file_properties ()
 380         {
 381                 PropertiesDialog* d = new PropertiesDialog (this, film);
 382                 d->ShowModal ();
 383                 d->Destroy ();
 384         }
 385
 386         void file_exit ()
 387         {
 388                 maybe_save_then_delete_film ();
 389                 Close (true);
 390         }
 391
 392         void edit_preferences ()
 393         {
 394                 ConfigDialog* d = new ConfigDialog (this);
 395                 d->ShowModal ();
 396                 d->Destroy ();
 397                 Config::instance()->write ();
 398         }
 399
 400         void jobs_make_dcp ()
 401         {
 402                 JobWrapper::make_dcp (this, film);
 403         }
 404
 405         void jobs_send_dcp_to_tms ()
 406         {
 407                 film->send_dcp_to_tms ();
 408         }
 409
 410         void jobs_show_dcp ()
 411         {
 412 #ifdef __WXMSW__
 413                 string d = film->directory();
 414                 wstring w;
 415                 w.assign (d.begin(), d.end());
 416                 ShellExecute (0, L"open", w.c_str(), 0, 0, SW_SHOWDEFAULT);
 417 #else
 418                 int r = system ("which nautilus");
 419                 if (WEXITSTATUS (r) == 0) {
 420                         r = system (string ("nautilus " + film->directory()).c_str ());
 421                         if (WEXITSTATUS (r)) {
 422                                 error_dialog (this, _("Could not show DCP (could not run nautilus)"));
 423                         }
 424                 } else {
 425                         int r = system ("which konqueror");
 426                         if (WEXITSTATUS (r) == 0) {
 427                                 r = system (string ("konqueror " + film->directory()).c_str ());
 428                                 if (WEXITSTATUS (r)) {
 429                                         error_dialog (this, _("Could not show DCP (could not run konqueror)"));
 430                                 }
 431                         }
 432                 }
 433 #endif
 434         }
 435
 436         void help_about ()
 437         {
 438                 AboutDialog* d = new AboutDialog (this);
 439                 d->ShowModal ();
 440                 d->Destroy ();
 441         }
 442 };
 443
 444 #if wxMINOR_VERSION == 9
 445 static const wxCmdLineEntryDesc command_line_description[] = {
 446         { wxCMD_LINE_OPTION, "l", "log", "set log level (silent, verbose or timing)", wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL },
 447         { wxCMD_LINE_SWITCH, "n", "new", "create new film", wxCMD_LINE_VAL_NONE, wxCMD_LINE_PARAM_OPTIONAL },
 448         { wxCMD_LINE_PARAM, 0, 0, "film to load or create", wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_MULTIPLE | wxCMD_LINE_PARAM_OPTIONAL },
 449         { wxCMD_LINE_NONE, "", "", "", wxCmdLineParamType (0), 0 }
 450 };
 451 #else
 452 static const wxCmdLineEntryDesc command_line_description[] = {
 453         { wxCMD_LINE_OPTION, wxT("l"), wxT("log"), wxT("set log level (silent, verbose or timing)"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL },
 454         { wxCMD_LINE_SWITCH, wxT("n"), wxT("new"), wxT("create new film"), wxCMD_LINE_VAL_NONE, wxCMD_LINE_PARAM_OPTIONAL },
 455         { wxCMD_LINE_PARAM, 0, 0, wxT("film to load or create"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_MULTIPLE | wxCMD_LINE_PARAM_OPTIONAL },
 456         { wxCMD_LINE_NONE, wxT(""), wxT(""), wxT(""), wxCmdLineParamType (0), 0 }
 457 };
 458 #endif
 459
 460 class App : public wxApp
 461 {
 462         bool OnInit ()
 463         try
 464         {
 465                 if (!wxApp::OnInit()) {
 466                         return false;
 467                 }
 468
 469 #ifdef DCPOMATIC_LINUX
 470                 unsetenv ("UBUNTU_MENUPROXY");
 471 #endif
 472
 473 #ifdef __WXOSX__
 474                 ProcessSerialNumber serial;
 475                 GetCurrentProcess (&serial);
 476                 TransformProcessType (&serial, kProcessTransformToForegroundApplication);
 477 #endif
 478
 479                 wxInitAllImageHandlers ();
 480
 481                 /* Enable i18n; this will create a Config object
 482                    to look for a force-configured language.  This Config
 483                    object will be wrong, however, because dcpomatic_setup
 484                    hasn't yet been called and there aren't any scalers, filters etc.
 485                    set up yet.
 486                 */
 487                 dcpomatic_setup_i18n ();
 488
 489                 /* Set things up, including scalers / filters etc.
 490                    which will now be internationalised correctly.
 491                 */
 492                 dcpomatic_setup ();
 493
 494                 /* Force the configuration to be re-loaded correctly next
 495                    time it is needed.
 496                 */
 497                 Config::drop ();
 498
 499                 if (!film_to_load.empty() && boost::filesystem::is_directory (film_to_load)) {
 500                         try {
 501                                 film.reset (new Film (film_to_load));
 502                                 film->read_metadata ();
 503                                 film->log()->set_level (log_level);
 504                         } catch (exception& e) {
 505                                 error_dialog (0, std_to_wx (String::compose (wx_to_std (_("Could not load film %1 (%2)")), film_to_load, e.what())));
 506                         }
 507                 }
 508
 509                 if (!film_to_create.empty ()) {
 510                         film.reset (new Film (film_to_create));
 511                         film->write_metadata ();
 512                         film->log()->set_level (log_level);
 513                         film->set_name (boost::filesystem::path (film_to_create).filename().generic_string ());
 514                 }
 515
 516                 Frame* f = new Frame (_("DCP-o-matic"));
 517                 SetTopWindow (f);
 518                 f->Maximize ();
 519                 f->Show ();
 520
 521                 ui_signaller = new wxUISignaller (this);
 522                 this->Bind (wxEVT_IDLE, boost::bind (&App::idle, this));
 523
 524                 return true;
 525         }
 526         catch (exception& e)
 527         {
 528                 error_dialog (0, wxString::Format ("DCP-o-matic could not start: %s", e.what ()));
 529                 return true;
 530         }
 531
 532         void OnInitCmdLine (wxCmdLineParser& parser)
 533         {
 534                 parser.SetDesc (command_line_description);
 535                 parser.SetSwitchChars (wxT ("-"));
 536         }
 537
 538         bool OnCmdLineParsed (wxCmdLineParser& parser)
 539         {
 540                 if (parser.GetParamCount() > 0) {
 541                         if (parser.Found (wxT ("new"))) {
 542                                 film_to_create = wx_to_std (parser.GetParam (0));
 543                         } else {
 544                                 film_to_load = wx_to_std (parser.GetParam(0));
 545                         }
 546                 }
 547
 548                 wxString log;
 549                 if (parser.Found (wxT ("log"), &log)) {
 550                         log_level = wx_to_std (log);
 551                 }
 552
 553                 return true;
 554         }
 555
 556         void idle ()
 557         {
 558                 ui_signaller->ui_idle ();
 559         }
 560 };
 561
 562 IMPLEMENT_APP (App)