src/tools/dcpomatic.cc

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