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