Rearrange checking (and re-examining) content.
[dcpomatic.git] / src / tools / dcpomatic_batch.cc
index 64eb46af2c2813de29acc96a04e534fe147e3e70..29f79ff080bf840276773c1159c12f3078c21fc1 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
 
 */
 
-#include "wx/wx_util.h"
+
 #include "wx/about_dialog.h"
-#include "wx/wx_signal_manager.h"
+#include "wx/dcpomatic_button.h"
+#include "wx/full_config_dialog.h"
 #include "wx/job_manager_view.h"
-#include "wx/config_dialog.h"
 #include "wx/servers_list_dialog.h"
-#include "lib/version.h"
+#include "wx/wx_signal_manager.h"
+#include "wx/wx_util.h"
 #include "lib/compose.hpp"
 #include "lib/config.h"
-#include "lib/util.h"
+#include "lib/dcpomatic_socket.h"
 #include "lib/film.h"
+#include "lib/job.h"
 #include "lib/job_manager.h"
-#include "lib/dcpomatic_socket.h"
+#include "lib/transcode_job.h"
+#include "lib/util.h"
+#include "lib/version.h"
 #include <wx/aboutdlg.h>
-#include <wx/stdpaths.h>
 #include <wx/cmdline.h>
 #include <wx/preferences.h>
+#include <wx/splash.h>
+#include <wx/stdpaths.h>
 #include <wx/wx.h>
 #include <iostream>
+#include <set>
 
+
+using std::cout;
+using std::dynamic_pointer_cast;
 using std::exception;
+using std::list;
+using std::make_shared;
+using std::set;
+using std::shared_ptr;
 using std::string;
-using std::cout;
-using boost::shared_ptr;
-using boost::thread;
 using boost::scoped_array;
+using boost::thread;
+#if BOOST_VERSION >= 106100
+using namespace boost::placeholders;
+#endif
+
+
+static list<boost::filesystem::path> films_to_load;
 
-static boost::optional<boost::filesystem::path> film_to_load;
 
 enum {
        ID_file_add_film = 1,
@@ -53,11 +69,12 @@ enum {
        ID_help_about
 };
 
+
 void
 setup_menu (wxMenuBar* m)
 {
-       wxMenu* file = new wxMenu;
-       file->Append (ID_file_add_film, _("&Add Film..."));
+       auto file = new wxMenu;
+       file->Append (ID_file_add_film, _("&Add Film...\tCtrl-A"));
 #ifdef DCPOMATIC_OSX
        file->Append (wxID_EXIT, _("&Exit"));
 #else
@@ -67,14 +84,14 @@ setup_menu (wxMenuBar* m)
 #ifdef DCPOMATIC_OSX
        file->Append (wxID_PREFERENCES, _("&Preferences...\tCtrl-P"));
 #else
-       wxMenu* edit = new wxMenu;
+       auto edit = new wxMenu;
        edit->Append (wxID_PREFERENCES, _("&Preferences...\tCtrl-P"));
 #endif
 
-       wxMenu* tools = new wxMenu;
+       auto tools = new wxMenu;
        tools->Append (ID_tools_encoding_servers, _("Encoding servers..."));
 
-       wxMenu* help = new wxMenu;
+       auto help = new wxMenu;
        help->Append (ID_help_about, _("About"));
 
        m->Append (file, _("&File"));
@@ -85,56 +102,120 @@ setup_menu (wxMenuBar* m)
        m->Append (help, _("&Help"));
 }
 
+
 class DOMFrame : public wxFrame
 {
 public:
-       DOMFrame (wxString const & title)
-               : wxFrame (NULL, -1, title)
-               , _sizer (new wxBoxSizer (wxVERTICAL))
-               , _config_dialog (0)
-               , _servers_list_dialog (0)
+       explicit DOMFrame (wxString const & title)
+               : wxFrame (nullptr, -1, title)
+               , _sizer (new wxBoxSizer(wxVERTICAL))
        {
-               wxMenuBar* bar = new wxMenuBar;
+               auto bar = new wxMenuBar;
                setup_menu (bar);
                SetMenuBar (bar);
 
-               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::file_add_film, this),    ID_file_add_film);
-               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::file_quit, this),        wxID_EXIT);
-               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::edit_preferences, this), wxID_PREFERENCES);
-               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::tools_encoding_servers, this), ID_tools_encoding_servers);
-               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::help_about, this),       ID_help_about);
+               Config::instance()->Changed.connect (boost::bind (&DOMFrame::config_changed, this, _1));
 
-               wxPanel* panel = new wxPanel (this);
-               wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::file_add_film, this),    ID_file_add_film);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::file_quit, this),        wxID_EXIT);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::edit_preferences, this), wxID_PREFERENCES);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::tools_encoding_servers, this), ID_tools_encoding_servers);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::help_about, this),       ID_help_about);
+
+               auto panel = new wxPanel (this);
+               auto s = new wxBoxSizer (wxHORIZONTAL);
                s->Add (panel, 1, wxEXPAND);
                SetSizer (s);
 
-               JobManagerView* job_manager_view = new JobManagerView (panel, false);
+               auto job_manager_view = new JobManagerView (panel, true);
                _sizer->Add (job_manager_view, 1, wxALL | wxEXPAND, 6);
 
-               wxSizer* buttons = new wxBoxSizer (wxHORIZONTAL);
-               wxButton* add = new wxButton (panel, wxID_ANY, _("Add Film..."));
-               add->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&DOMFrame::add_film, this));
+               auto buttons = new wxBoxSizer (wxHORIZONTAL);
+               auto add = new Button (panel, _("Add Film..."));
+               add->Bind (wxEVT_BUTTON, boost::bind(&DOMFrame::add_film, this));
                buttons->Add (add, 1, wxALL, 6);
+               _pause = new Button (panel, _("Pause"));
+               _pause->Bind (wxEVT_BUTTON, boost::bind(&DOMFrame::pause, this));
+               buttons->Add (_pause, 1, wxALL, 6);
+               _resume = new Button (panel, _("Resume"));
+               _resume->Bind (wxEVT_BUTTON, boost::bind(&DOMFrame::resume, this));
+               buttons->Add (_resume, 1, wxALL, 6);
+
+               setup_sensitivity ();
 
                _sizer->Add (buttons, 0, wxALL, 6);
 
                panel->SetSizer (_sizer);
 
-               Bind (wxEVT_CLOSE_WINDOW, boost::bind (&DOMFrame::close, this, _1));
-               Bind (wxEVT_SIZE, boost::bind (&DOMFrame::sized, this, _1));
+               Bind (wxEVT_CLOSE_WINDOW, boost::bind(&DOMFrame::close, this, _1));
+               Bind (wxEVT_SIZE, boost::bind(&DOMFrame::sized, this, _1));
+       }
+
+       void setup_sensitivity ()
+       {
+               _pause->Enable (!JobManager::instance()->paused());
+               _resume->Enable (JobManager::instance()->paused());
+       }
+
+       void pause ()
+       {
+               JobManager::instance()->pause();
+               setup_sensitivity ();
+       }
+
+       void resume ()
+       {
+               JobManager::instance()->resume();
+               setup_sensitivity ();
        }
 
        void start_job (boost::filesystem::path path)
        {
                try {
-                       shared_ptr<Film> film (new Film (path));
+                       auto film = make_shared<Film>(path);
                        film->read_metadata ();
-                       film->make_dcp ();
+
+                       double total_required;
+                       double available;
+                       bool can_hard_link;
+
+                       film->should_be_enough_disk_space (total_required, available, can_hard_link);
+
+                       set<shared_ptr<const Film>> films;
+
+                       for (auto i: JobManager::instance()->get()) {
+                               films.insert (i->film());
+                       }
+
+                       for (auto i: films) {
+                               double progress = 0;
+                               for (auto j: JobManager::instance()->get()) {
+                                       if (i == j->film() && dynamic_pointer_cast<TranscodeJob>(j)) {
+                                               progress = j->progress().get_value_or(0);
+                                       }
+                               }
+
+                               double required;
+                               i->should_be_enough_disk_space (required, available, can_hard_link);
+                               total_required += (1 - progress) * required;
+                       }
+
+                       if ((total_required - available) > 1) {
+                               if (!confirm_dialog (
+                                           this,
+                                           wxString::Format(
+                                                   _("The DCPs for this film and the films already in the queue will take up about %.1f GB.  The "
+                                                     "disks that you are using only have %.1f GB available.  Do you want to add this film to the queue anyway?"),
+                                                   total_required, available))) {
+                                       return;
+                               }
+                       }
+
+                       film->make_dcp (TranscodeJob::ChangedBehaviour::STOP);
                } catch (std::exception& e) {
-                       wxString p = std_to_wx (path.string ());
-                       wxCharBuffer b = p.ToUTF8 ();
-                       error_dialog (this, wxString::Format (_("Could not open film at %s (%s)"), p.data(), std_to_wx (e.what()).data()));
+                       auto p = std_to_wx (path.string ());
+                       auto b = p.ToUTF8 ();
+                       error_dialog (this, wxString::Format(_("Could not open film at %s"), p.data()), std_to_wx(e.what()));
                }
        }
 
@@ -147,11 +228,11 @@ private:
 
        bool should_close ()
        {
-               if (!JobManager::instance()->work_to_do ()) {
+               if (!JobManager::instance()->work_to_do()) {
                        return true;
                }
 
-               wxMessageDialog* d = new wxMessageDialog (
+               auto d = new wxMessageDialog (
                        0,
                        _("There are unfinished jobs; are you sure you want to quit?"),
                        _("Unfinished jobs"),
@@ -165,7 +246,7 @@ private:
 
        void close (wxCloseEvent& ev)
        {
-               if (!should_close ()) {
+               if (!should_close()) {
                        ev.Veto ();
                        return;
                }
@@ -180,7 +261,7 @@ private:
 
        void file_quit ()
        {
-               if (should_close ()) {
+               if (should_close()) {
                        Close (true);
                }
        }
@@ -188,7 +269,7 @@ private:
        void edit_preferences ()
        {
                if (!_config_dialog) {
-                       _config_dialog = create_config_dialog ();
+                       _config_dialog = create_full_config_dialog ();
                }
                _config_dialog->Show (this);
        }
@@ -204,16 +285,16 @@ private:
 
        void help_about ()
        {
-               AboutDialog* d = new AboutDialog (this);
+               auto d = new AboutDialog (this);
                d->ShowModal ();
                d->Destroy ();
        }
 
        void add_film ()
        {
-               wxDirDialog* c = new wxDirDialog (this, _("Select film to open"), wxStandardPaths::Get().GetDocumentsDir(), wxDEFAULT_DIALOG_STYLE | wxDD_DIR_MUST_EXIST);
+               auto c = new wxDirDialog (this, _("Select film to open"), wxStandardPaths::Get().GetDocumentsDir(), wxDEFAULT_DIALOG_STYLE | wxDD_DIR_MUST_EXIST);
                if (_last_parent) {
-                       c->SetPath (std_to_wx (_last_parent.get().string ()));
+                       c->SetPath (std_to_wx(_last_parent.get().string()));
                }
 
                int r;
@@ -235,22 +316,56 @@ private:
                c->Destroy ();
        }
 
+       void config_changed (Config::Property what)
+       {
+               /* Instantly save any config changes when using the DCP-o-matic GUI */
+               if (what == Config::CINEMAS) {
+                       try {
+                               Config::instance()->write_cinemas();
+                       } catch (exception& e) {
+                               error_dialog (
+                                       this,
+                                       wxString::Format(
+                                               _("Could not write to cinemas file at %s.  Your changes have not been saved."),
+                                               std_to_wx (Config::instance()->cinemas_file().string()).data()
+                                               )
+                                       );
+                       }
+               } else {
+                       try {
+                               Config::instance()->write_config();
+                       } catch (exception& e) {
+                               error_dialog (
+                                       this,
+                                       wxString::Format(
+                                               _("Could not write to config file at %s.  Your changes have not been saved."),
+                                               std_to_wx (Config::instance()->cinemas_file().string()).data()
+                                               )
+                                       );
+                       }
+               }
+       }
+
        boost::optional<boost::filesystem::path> _last_parent;
        wxSizer* _sizer;
-       wxPreferencesEditor* _config_dialog;
-       ServersListDialog* _servers_list_dialog;
+       wxPreferencesEditor* _config_dialog = nullptr;
+       ServersListDialog* _servers_list_dialog = nullptr;
+       wxButton* _pause;
+       wxButton* _resume;
 };
 
+
 static const wxCmdLineEntryDesc command_line_description[] = {
        { wxCMD_LINE_PARAM, 0, 0, "film to load", wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_MULTIPLE | wxCMD_LINE_PARAM_OPTIONAL },
        { wxCMD_LINE_NONE, "", "", "", wxCmdLineParamType (0), 0 }
 };
 
+
 class JobServer : public Server
 {
 public:
-       JobServer (DOMFrame* frame)
-               : Server (Config::instance()->server_port_base() + 2)
+       explicit JobServer (DOMFrame* frame)
+               : Server (BATCH_JOB_PORT)
                , _frame (frame)
        {}
 
@@ -258,11 +373,11 @@ public:
        {
                try {
                        int const length = socket->read_uint32 ();
-                       scoped_array<char> buffer (new char[length]);
-                       socket->read (reinterpret_cast<uint8_t*> (buffer.get()), length);
+                       scoped_array<char> buffer(new char[length]);
+                       socket->read (reinterpret_cast<uint8_t*>(buffer.get()), length);
                        string s (buffer.get());
                        _frame->start_job (s);
-                       socket->write (reinterpret_cast<uint8_t const *> ("OK"), 3);
+                       socket->write (reinterpret_cast<uint8_t const *>("OK"), 3);
                } catch (...) {
 
                }
@@ -272,11 +387,20 @@ private:
        DOMFrame* _frame;
 };
 
+
 class App : public wxApp
 {
        bool OnInit ()
        {
+               wxInitAllImageHandlers ();
+
                SetAppName (_("DCP-o-matic Batch Converter"));
+               is_batch_converter = true;
+
+               Config::FailedToLoad.connect (boost::bind(&App::config_failed_to_load, this));
+               Config::Warning.connect (boost::bind(&App::config_warning, this, _1));
+
+               auto splash = maybe_show_splash ();
 
                if (!wxApp::OnInit()) {
                        return false;
@@ -306,27 +430,34 @@ class App : public wxApp
                */
                Config::drop ();
 
-               DOMFrame* f = new DOMFrame (_("DCP-o-matic Batch Converter"));
-               SetTopWindow (f);
-               f->Maximize ();
-               f->Show ();
+               _frame = new DOMFrame (_("DCP-o-matic Batch Converter"));
+               SetTopWindow (_frame);
+               _frame->Maximize ();
+               if (splash) {
+                       splash->Destroy ();
+               }
+               _frame->Show ();
 
-               JobServer* server = new JobServer (f);
+               auto server = new JobServer (_frame);
                new thread (boost::bind (&JobServer::run, server));
 
                signal_manager = new wxSignalManager (this);
                this->Bind (wxEVT_IDLE, boost::bind (&App::idle, this));
 
                shared_ptr<Film> film;
-               if (film_to_load && boost::filesystem::is_directory (film_to_load.get())) {
-                       try {
-                               film.reset (new Film (film_to_load.get()));
-                               film->read_metadata ();
-                               film->make_dcp ();
-                       } catch (exception& e) {
-                               error_dialog (
-                                       0, std_to_wx (String::compose (wx_to_std (_("Could not load film %1 (%2)")), film_to_load->string(), e.what()))
-                                       );
+               for (auto i: films_to_load) {
+                       if (boost::filesystem::is_directory(i)) {
+                               try {
+                                       film = make_shared<Film>(i);
+                                       film->read_metadata ();
+                                       film->make_dcp (TranscodeJob::ChangedBehaviour::EXAMINE_THEN_STOP);
+                               } catch (exception& e) {
+                                       error_dialog (
+                                               0,
+                                               std_to_wx(String::compose(wx_to_std(_("Could not load film %1")), i.string())),
+                                               std_to_wx(e.what())
+                                               );
+                               }
                        }
                }
 
@@ -346,12 +477,25 @@ class App : public wxApp
 
        bool OnCmdLineParsed (wxCmdLineParser& parser)
        {
-               if (parser.GetParamCount() > 0) {
-                       film_to_load = wx_to_std (parser.GetParam(0));
+               for (size_t i = 0; i < parser.GetParamCount(); ++i) {
+                       films_to_load.push_back (wx_to_std(parser.GetParam(i)));
                }
 
                return true;
        }
+
+       void config_failed_to_load ()
+       {
+               message_dialog (_frame, _("The existing configuration failed to load.  Default values will be used instead.  These may take a short time to create."));
+       }
+
+       void config_warning (string m)
+       {
+               message_dialog (_frame, std_to_wx(m));
+       }
+
+       DOMFrame* _frame;
 };
 
+
 IMPLEMENT_APP (App)