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
 243                 Bind (wxEVT_MENU_OPEN, boost::bind (&Frame::menu_opened, this, _1));
 244                 Bind (wxEVT_CLOSE_WINDOW, boost::bind (&Frame::close, this, _1));
 245
 246                 /* Use a panel as the only child of the Frame so that we avoid
 247                    the dark-grey background on Windows.
 248                 */
 249                 wxPanel* overall_panel = new wxPanel (this, wxID_ANY);
 250
 251                 film_editor = new FilmEditor (film, overall_panel);
 252                 film_viewer = new FilmViewer (film, overall_panel);
 253                 JobManagerView* job_manager_view = new JobManagerView (overall_panel, static_cast<JobManagerView::Buttons> (0));
 254
 255                 wxBoxSizer* right_sizer = new wxBoxSizer (wxVERTICAL);
 256                 right_sizer->Add (film_viewer, 2, wxEXPAND | wxALL, 6);
 257                 right_sizer->Add (job_manager_view, 1, wxEXPAND | wxALL, 6);
 258
 259                 wxBoxSizer* main_sizer = new wxBoxSizer (wxHORIZONTAL);
 260                 main_sizer->Add (film_editor, 1, wxEXPAND | wxALL, 6);
 261                 main_sizer->Add (right_sizer, 2, wxEXPAND | wxALL, 6);
 262
 263                 set_menu_sensitivity ();
 264
 265                 film_editor->FileChanged.connect (bind (&Frame::file_changed, this, _1));
 266                 if (film) {
 267                         file_changed (film->directory ());
 268                 } else {
 269                         file_changed ("");
 270                 }
 271
 272                 JobManager::instance()->ActiveJobsChanged.connect (boost::bind (set_menu_sensitivity));
 273
 274                 set_film ();
 275                 overall_panel->SetSizer (main_sizer);
 276         }
 277
 278 private:
 279
 280         void menu_opened (wxMenuEvent& ev)
 281         {
 282                 if (ev.GetMenu() != jobs_menu) {
 283                         return;
 284                 }
 285
 286                 bool const have_dcp = film && film->have_dcp();
 287                 jobs_menu->Enable (ID_jobs_send_dcp_to_tms, have_dcp);
 288                 jobs_menu->Enable (ID_jobs_show_dcp, have_dcp);
 289         }
 290
 291         void set_film ()
 292         {
 293                 film_viewer->set_film (film);
 294                 film_editor->set_film (film);
 295                 set_menu_sensitivity ();
 296         }
 297
 298         void file_changed (string f)
 299         {
 300                 stringstream s;
 301                 s << wx_to_std (_("DCP-o-matic"));
 302                 if (!f.empty ()) {
 303                         s << " - " << f;
 304                 }
 305
 306                 SetTitle (std_to_wx (s.str()));
 307         }
 308
 309         void file_new ()
 310         {
 311                 NewFilmDialog* d = new NewFilmDialog (this);
 312                 int const r = d->ShowModal ();
 313
 314                 if (r == wxID_OK) {
 315
 316                         if (boost::filesystem::is_directory (d->get_path()) && !boost::filesystem::is_empty(d->get_path())) {
 317                                 if (!confirm_dialog (
 318                                             this,
 319                                             std_to_wx (
 320                                                     String::compose (wx_to_std (_("The directory %1 already exists and is not empty.  "
 321                                                                                   "Are you sure you want to use it?")),
 322                                                                      d->get_path().c_str())
 323                                                     )
 324                                             )) {
 325                                         return;
 326                                 }
 327                         } else if (boost::filesystem::is_regular_file (d->get_path())) {
 328                                 error_dialog (
 329                                         this,
 330                                         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())
 331                                         );
 332                                 return;
 333                         }
 334
 335                         maybe_save_then_delete_film ();
 336                         film.reset (new Film (d->get_path ()));
 337                         film->write_metadata ();
 338                         film->log()->set_level (log_level);
 339                         film->set_name (boost::filesystem::path (d->get_path()).filename().generic_string());
 340                         set_film ();
 341                 }
 342
 343                 d->Destroy ();
 344         }
 345
 346         void file_open ()
 347         {
 348                 wxDirDialog* c = new wxDirDialog (this, _("Select film to open"), wxStandardPaths::Get().GetDocumentsDir(), wxDEFAULT_DIALOG_STYLE | wxDD_DIR_MUST_EXIST);
 349                 int r;
 350                 while (1) {
 351                         r = c->ShowModal ();
 352                         if (r == wxID_OK && c->GetPath() == wxStandardPaths::Get().GetDocumentsDir()) {
 353                                 error_dialog (this, _("You did not select a folder.  Make sure that you select a folder before clicking Open."));
 354                         } else {
 355                                 break;
 356                         }
 357                 }
 358
 359                 if (r == wxID_OK) {
 360                         maybe_save_then_delete_film ();
 361                         try {
 362                                 film.reset (new Film (wx_to_std (c->GetPath ())));
 363                                 film->read_metadata ();
 364                                 film->log()->set_level (log_level);
 365                                 set_film ();
 366                         } catch (std::exception& e) {
 367                                 wxString p = c->GetPath ();
 368                                 wxCharBuffer b = p.ToUTF8 ();
 369                                 error_dialog (this, wxString::Format (_("Could not open film at %s (%s)"), p.data(), std_to_wx (e.what()).data()));
 370                         }
 371                 }
 372
 373                 c->Destroy ();
 374         }
 375
 376         void file_save ()
 377         {
 378                 film->write_metadata ();
 379         }
 380
 381         void file_properties ()
 382         {
 383                 PropertiesDialog* d = new PropertiesDialog (this, film);
 384                 d->ShowModal ();
 385                 d->Destroy ();
 386         }
 387
 388         void file_exit ()
 389         {
 390                 if (!should_close ()) {
 391                         return;
 392                 }
 393
 394                 maybe_save_then_delete_film ();
 395                 Close (true);
 396         }
 397
 398         void edit_preferences ()
 399         {
 400                 ConfigDialog* d = new ConfigDialog (this);
 401                 d->ShowModal ();
 402                 d->Destroy ();
 403                 Config::instance()->write ();
 404         }
 405
 406         void jobs_make_dcp ()
 407         {
 408                 JobWrapper::make_dcp (this, film);
 409         }
 410
 411         void jobs_send_dcp_to_tms ()
 412         {
 413                 film->send_dcp_to_tms ();
 414         }
 415
 416         void jobs_show_dcp ()
 417         {
 418 #ifdef __WXMSW__
 419                 string d = film->directory();
 420                 wstring w;
 421                 w.assign (d.begin(), d.end());
 422                 ShellExecute (0, L"open", w.c_str(), 0, 0, SW_SHOWDEFAULT);
 423 #else
 424                 int r = system ("which nautilus");
 425                 if (WEXITSTATUS (r) == 0) {
 426                         r = system (string ("nautilus " + film->directory()).c_str ());
 427                         if (WEXITSTATUS (r)) {
 428                                 error_dialog (this, _("Could not show DCP (could not run nautilus)"));
 429                         }
 430                 } else {
 431                         int r = system ("which konqueror");
 432                         if (WEXITSTATUS (r) == 0) {
 433                                 r = system (string ("konqueror " + film->directory()).c_str ());
 434                                 if (WEXITSTATUS (r)) {
 435                                         error_dialog (this, _("Could not show DCP (could not run konqueror)"));
 436                                 }
 437                         }
 438                 }
 439 #endif
 440         }
 441
 442         void help_about ()
 443         {
 444                 AboutDialog* d = new AboutDialog (this);
 445                 d->ShowModal ();
 446                 d->Destroy ();
 447         }
 448
 449         bool should_close ()
 450         {
 451                 if (!JobManager::instance()->work_to_do ()) {
 452                         return true;
 453                 }
 454
 455                 wxMessageDialog* d = new wxMessageDialog (
 456                         0,
 457                         _("There are unfinished jobs; are you sure you want to quit?"),
 458                         _("Unfinished jobs"),
 459                         wxYES_NO | wxYES_DEFAULT | wxICON_QUESTION
 460                         );
 461
 462                 bool const r = d->ShowModal() == wxID_YES;
 463                 d->Destroy ();
 464                 return r;
 465         }
 466
 467         void close (wxCloseEvent& ev)
 468         {
 469                 if (!should_close ()) {
 470                         ev.Veto ();
 471                         return;
 472                 }
 473
 474                 ev.Skip ();
 475         }
 476 };
 477
 478 #if wxMINOR_VERSION == 9
 479 static const wxCmdLineEntryDesc command_line_description[] = {
 480         { wxCMD_LINE_OPTION, "l", "log", "set log level (silent, verbose or timing)", wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL },
 481         { wxCMD_LINE_SWITCH, "n", "new", "create new film", wxCMD_LINE_VAL_NONE, wxCMD_LINE_PARAM_OPTIONAL },
 482         { wxCMD_LINE_PARAM, 0, 0, "film to load or create", wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_MULTIPLE | wxCMD_LINE_PARAM_OPTIONAL },
 483         { wxCMD_LINE_NONE, "", "", "", wxCmdLineParamType (0), 0 }
 484 };
 485 #else
 486 static const wxCmdLineEntryDesc command_line_description[] = {
 487         { wxCMD_LINE_OPTION, wxT("l"), wxT("log"), wxT("set log level (silent, verbose or timing)"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL },
 488         { wxCMD_LINE_SWITCH, wxT("n"), wxT("new"), wxT("create new film"), wxCMD_LINE_VAL_NONE, wxCMD_LINE_PARAM_OPTIONAL },
 489         { wxCMD_LINE_PARAM, 0, 0, wxT("film to load or create"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_MULTIPLE | wxCMD_LINE_PARAM_OPTIONAL },
 490         { wxCMD_LINE_NONE, wxT(""), wxT(""), wxT(""), wxCmdLineParamType (0), 0 }
 491 };
 492 #endif
 493
 494 class App : public wxApp
 495 {
 496         bool OnInit ()
 497         try
 498         {
 499                 if (!wxApp::OnInit()) {
 500                         return false;
 501                 }
 502
 503 #ifdef DCPOMATIC_LINUX
 504                 unsetenv ("UBUNTU_MENUPROXY");
 505 #endif
 506
 507 #ifdef __WXOSX__
 508                 ProcessSerialNumber serial;
 509                 GetCurrentProcess (&serial);
 510                 TransformProcessType (&serial, kProcessTransformToForegroundApplication);
 511 #endif
 512
 513                 wxInitAllImageHandlers ();
 514
 515                 /* Enable i18n; this will create a Config object
 516                    to look for a force-configured language.  This Config
 517                    object will be wrong, however, because dcpomatic_setup
 518                    hasn't yet been called and there aren't any scalers, filters etc.
 519                    set up yet.
 520                 */
 521                 dcpomatic_setup_i18n ();
 522
 523                 /* Set things up, including scalers / filters etc.
 524                    which will now be internationalised correctly.
 525                 */
 526                 dcpomatic_setup ();
 527
 528                 /* Force the configuration to be re-loaded correctly next
 529                    time it is needed.
 530                 */
 531                 Config::drop ();
 532
 533                 if (!film_to_load.empty() && boost::filesystem::is_directory (film_to_load)) {
 534                         try {
 535                                 film.reset (new Film (film_to_load));
 536                                 film->read_metadata ();
 537                                 film->log()->set_level (log_level);
 538                         } catch (exception& e) {
 539                                 error_dialog (0, std_to_wx (String::compose (wx_to_std (_("Could not load film %1 (%2)")), film_to_load, e.what())));
 540                         }
 541                 }
 542
 543                 if (!film_to_create.empty ()) {
 544                         film.reset (new Film (film_to_create));
 545                         film->write_metadata ();
 546                         film->log()->set_level (log_level);
 547                         film->set_name (boost::filesystem::path (film_to_create).filename().generic_string ());
 548                 }
 549
 550                 Frame* f = new Frame (_("DCP-o-matic"));
 551                 SetTopWindow (f);
 552                 f->Maximize ();
 553                 f->Show ();
 554
 555                 ui_signaller = new wxUISignaller (this);
 556                 this->Bind (wxEVT_IDLE, boost::bind (&App::idle, this));
 557
 558                 return true;
 559         }
 560         catch (exception& e)
 561         {
 562                 error_dialog (0, wxString::Format ("DCP-o-matic could not start: %s", e.what ()));
 563                 return true;
 564         }
 565
 566         void OnInitCmdLine (wxCmdLineParser& parser)
 567         {
 568                 parser.SetDesc (command_line_description);
 569                 parser.SetSwitchChars (wxT ("-"));
 570         }
 571
 572         bool OnCmdLineParsed (wxCmdLineParser& parser)
 573         {
 574                 if (parser.GetParamCount() > 0) {
 575                         if (parser.Found (wxT ("new"))) {
 576                                 film_to_create = wx_to_std (parser.GetParam (0));
 577                         } else {
 578                                 film_to_load = wx_to_std (parser.GetParam(0));
 579                         }
 580                 }
 581
 582                 wxString log;
 583                 if (parser.Found (wxT ("log"), &log)) {
 584                         log_level = wx_to_std (log);
 585                 }
 586
 587                 return true;
 588         }
 589
 590         void idle ()
 591         {
 592                 ui_signaller->ui_idle ();
 593         }
 594 };
 595
 596 IMPLEMENT_APP (App)