Use make_shared<>.
[dcpomatic.git] / src / tools / dcpomatic.cc
index 0cd0bac330c85796ef1c00743aab78e4276eba00..6c789632a2dff6e46c73f10fb91be54985dfe2b0 100644 (file)
@@ -1,19 +1,20 @@
 /*
-    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
@@ -41,6 +42,7 @@
 #include "lib/config.h"
 #include "lib/util.h"
 #include "lib/video_content.h"
+#include "lib/content.h"
 #include "lib/version.h"
 #include "lib/signal_manager.h"
 #include "lib/log.h"
@@ -55,6 +57,7 @@
 #include "lib/content_factory.h"
 #include "lib/compose.hpp"
 #include "lib/cinema_kdms.h"
+#include "lib/dcpomatic_socket.h"
 #include <dcp/exceptions.h>
 #include <wx/generic/aboutdlgg.h>
 #include <wx/stdpaths.h>
@@ -68,6 +71,8 @@
 #include <ApplicationServices/ApplicationServices.h>
 #endif
 #include <boost/filesystem.hpp>
+#include <boost/noncopyable.hpp>
+#include <boost/make_shared.hpp>
 #include <iostream>
 #include <fstream>
 #include <sstream>
@@ -88,8 +93,10 @@ using std::list;
 using std::exception;
 using boost::shared_ptr;
 using boost::dynamic_pointer_cast;
+using boost::optional;
+using boost::make_shared;
 
-class FilmChangedDialog
+class FilmChangedDialog : public boost::noncopyable
 {
 public:
        FilmChangedDialog (string name)
@@ -100,7 +107,11 @@ public:
                        /// TRANSLATORS: this is the heading for a dialog box, which tells the user that the current
                        /// project (Film) has been changed since it was last saved.
                        _("Film changed"),
-                       wxYES_NO | wxYES_DEFAULT | wxICON_QUESTION
+                       wxYES_NO | wxCANCEL | wxYES_DEFAULT | wxICON_QUESTION
+                       );
+
+               _dialog->SetYesNoCancelLabels (
+                       _("Save film and close"), _("Close without saving film"), _("Don't close")
                        );
        }
 
@@ -115,9 +126,6 @@ public:
        }
 
 private:
-       /* Not defined */
-       FilmChangedDialog (FilmChangedDialog const &);
-
        wxMessageDialog* _dialog;
 };
 
@@ -138,6 +146,7 @@ enum {
        ID_content_scale_to_fit_width = 100,
        ID_content_scale_to_fit_height,
        ID_jobs_make_dcp,
+       ID_jobs_make_dcp_batch,
        ID_jobs_make_kdms,
        ID_jobs_make_self_dkdm,
        ID_jobs_send_dcp_to_tms,
@@ -149,7 +158,8 @@ enum {
        ID_tools_restore_default_preferences,
        ID_help_report_a_problem,
        /* IDs for shortcuts (with no associated menu item) */
-       ID_add_file
+       ID_add_file,
+       ID_remove
 };
 
 class DOMFrame : public wxFrame
@@ -204,6 +214,7 @@ public:
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::content_scale_to_fit_height, this), ID_content_scale_to_fit_height);
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_make_dcp, this),           ID_jobs_make_dcp);
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_make_kdms, this),          ID_jobs_make_kdms);
+               Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_make_dcp_batch, this),     ID_jobs_make_dcp_batch);
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_make_self_dkdm, this),     ID_jobs_make_self_dkdm);
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_send_dcp_to_tms, this),    ID_jobs_send_dcp_to_tms);
                Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::jobs_show_dcp, this),           ID_jobs_show_dcp);
@@ -243,10 +254,12 @@ public:
 
                overall_panel->SetSizer (main_sizer);
 
-               wxAcceleratorEntry accel[1];
+               wxAcceleratorEntry accel[2];
                accel[0].Set (wxACCEL_CTRL, static_cast<int>('A'), ID_add_file);
+               accel[1].Set (wxACCEL_NORMAL, WXK_DELETE, ID_remove);
                Bind (wxEVT_MENU, boost::bind (&ContentPanel::add_file_clicked, _film_editor->content_panel()), ID_add_file);
-               wxAcceleratorTable accel_table (1, accel);
+               Bind (wxEVT_MENU, boost::bind (&DOMFrame::remove_clicked, this, _1), ID_remove);
+               wxAcceleratorTable accel_table (2, accel);
                SetAcceleratorTable (accel_table);
 
                /* Instantly save any config changes when using the DCP-o-matic GUI */
@@ -255,9 +268,16 @@ public:
                UpdateChecker::instance()->StateChanged.connect (boost::bind (&DOMFrame::update_checker_state_changed, this));
        }
 
+       void remove_clicked (wxCommandEvent& ev)
+       {
+               if (_film_editor->content_panel()->remove_clicked (true)) {
+                       ev.Skip ();
+               }
+       }
+
        void new_film (boost::filesystem::path path)
        {
-               shared_ptr<Film> film (new Film (path));
+               shared_ptr<Film> film = make_shared<Film> (path);
                film->write_metadata ();
                film->set_name (path.filename().generic_string());
                set_film (film);
@@ -266,9 +286,7 @@ public:
        void load_film (boost::filesystem::path file)
        try
        {
-               maybe_save_then_delete_film ();
-
-               shared_ptr<Film> film (new Film (file));
+               shared_ptr<Film> film = make_shared<Film> (file);
                list<string> const notes = film->read_metadata ();
 
                if (film->state_version() == 4) {
@@ -342,8 +360,9 @@ private:
                                return;
                        }
 
-                       maybe_save_then_delete_film ();
-                       new_film (d->get_path ());
+                       if (maybe_save_then_delete_film ()) {
+                               new_film (d->get_path ());
+                       }
                }
 
                d->Destroy ();
@@ -368,7 +387,7 @@ private:
                        }
                }
 
-               if (r == wxID_OK) {
+               if (r == wxID_OK && maybe_save_then_delete_film()) {
                        load_film (wx_to_std (c->GetPath ()));
                }
 
@@ -384,7 +403,7 @@ private:
        {
                vector<boost::filesystem::path> history = Config::instance()->history ();
                int n = event.GetId() - ID_file_history;
-               if (n >= 0 && n < static_cast<int> (history.size ())) {
+               if (n >= 0 && n < static_cast<int> (history.size ()) && maybe_save_then_delete_film()) {
                        load_film (history[n]);
                }
        }
@@ -405,7 +424,19 @@ private:
 
        void tools_restore_default_preferences ()
        {
-               Config::restore_defaults ();
+               wxMessageDialog* d = new wxMessageDialog (
+                       0,
+                       _("Are you sure you want to restore preferences to their defaults?  This cannot be undone."),
+                       _("Restore default preferences"),
+                       wxYES_NO | wxYES_DEFAULT | wxICON_QUESTION
+                       );
+
+               int const r = d->ShowModal ();
+               d->Destroy ();
+
+               if (r == wxID_YES) {
+                       Config::restore_defaults ();
+               }
        }
 
        void jobs_make_dcp ()
@@ -461,14 +492,14 @@ private:
                                        );
                        } else {
                                JobManager::instance()->add (
-                                       shared_ptr<Job> (new SendKDMEmailJob (
-                                                                _film->name(),
-                                                                _film->dcp_name(),
-                                                                d->from(),
-                                                                d->until(),
-                                                                CinemaKDMs::collect (screen_kdms),
-                                                                _film->log()
-                                                                ))
+                                       boost::make_shared<SendKDMEmailJob> (
+                                               _film->name(),
+                                               _film->dcp_name(),
+                                               d->from(),
+                                               d->until(),
+                                               CinemaKDMs::collect (screen_kdms),
+                                               _film->log()
+                                               )
                                        );
                        }
                } catch (dcp::NotEncryptedError& e) {
@@ -482,6 +513,47 @@ private:
                d->Destroy ();
        }
 
+       void jobs_make_dcp_batch ()
+       {
+               if (!_film) {
+                       return;
+               }
+
+               _film->write_metadata ();
+
+               /* i = 0; try to connect via socket
+                  i = 1; try again, and then try to start the batch converter
+                  i = 2 onwards; try again.
+               */
+               for (int i = 0; i < 8; ++i) {
+                       try {
+                               boost::asio::io_service io_service;
+                               boost::asio::ip::tcp::resolver resolver (io_service);
+                               boost::asio::ip::tcp::resolver::query query ("127.0.0.1", raw_convert<string> (Config::instance()->server_port_base() + 2));
+                               boost::asio::ip::tcp::resolver::iterator endpoint_iterator = resolver.resolve (query);
+                               Socket socket (5);
+                               socket.connect (*endpoint_iterator);
+                               string s = _film->directory().string ();
+                               socket.write (s.length() + 1);
+                               socket.write ((uint8_t *) s.c_str(), s.length() + 1);
+                               /* OK\0 */
+                               uint8_t ok[3];
+                               socket.read (ok, 3);
+                               return;
+                       } catch (exception& e) {
+
+                       }
+
+                       if (i == 1) {
+                               start_batch_converter (wx_to_std (wxStandardPaths::Get().GetExecutablePath()));
+                       }
+
+                       dcpomatic_sleep (1);
+               }
+
+               error_dialog (this, _("Could not find batch converter."));
+       }
+
        void jobs_make_self_dkdm ()
        {
                if (!_film) {
@@ -494,8 +566,9 @@ private:
                        return;
                }
 
+               optional<dcp::EncryptedKDM> kdm;
                try {
-                       dcp::EncryptedKDM kdm = _film->make_kdm (
+                       kdm = _film->make_kdm (
                                Config::instance()->decryption_chain()->leaf(),
                                vector<dcp::Certificate> (),
                                d->cpl (),
@@ -503,9 +576,6 @@ private:
                                dcp::LocalTime ("2112-01-01T01:00:00+00:00"),
                                dcp::MODIFIED_TRANSITIONAL_1
                                );
-
-                       string const name = tidy_for_filename(_film->name()) + "_DKDM.kdm.xml";
-                       kdm.as_xml (d->directory() / name);
                } catch (dcp::NotEncryptedError& e) {
                        error_dialog (this, _("CPL's content is not encrypted."));
                } catch (exception& e) {
@@ -514,22 +584,33 @@ private:
                        error_dialog (this, _("An unknown exception occurred."));
                }
 
+               if (kdm) {
+                       if (d->internal ()) {
+                               vector<dcp::EncryptedKDM> dkdms = Config::instance()->dkdms ();
+                               dkdms.push_back (kdm.get());
+                               Config::instance()->set_dkdms (dkdms);
+                       } else {
+                               boost::filesystem::path path = d->directory() / (_film->dcp_name(false) + "_DKDM.xml");
+                               kdm->as_xml (path);
+                       }
+               }
+
                d->Destroy ();
        }
 
        void content_scale_to_fit_width ()
        {
-               VideoContentList vc = _film_editor->content_panel()->selected_video ();
-               for (VideoContentList::iterator i = vc.begin(); i != vc.end(); ++i) {
-                       (*i)->scale_and_crop_to_fit_width ();
+               ContentList vc = _film_editor->content_panel()->selected_video ();
+               for (ContentList::iterator i = vc.begin(); i != vc.end(); ++i) {
+                       (*i)->video->scale_and_crop_to_fit_width ();
                }
        }
 
        void content_scale_to_fit_height ()
        {
-               VideoContentList vc = _film_editor->content_panel()->selected_video ();
-               for (VideoContentList::iterator i = vc.begin(); i != vc.end(); ++i) {
-                       (*i)->scale_and_crop_to_fit_height ();
+               ContentList vc = _film_editor->content_panel()->selected_video ();
+               for (ContentList::iterator i = vc.begin(); i != vc.end(); ++i) {
+                       (*i)->video->scale_and_crop_to_fit_height ();
                }
        }
 
@@ -646,13 +727,33 @@ private:
                        return;
                }
 
+               if (_film && _film->dirty ()) {
+
+                       FilmChangedDialog* dialog = new FilmChangedDialog (_film->name ());
+                       int const r = dialog->run ();
+                       delete dialog;
+
+                       switch (r) {
+                       case wxID_NO:
+                               /* Don't save and carry on to close */
+                               break;
+                       case wxID_YES:
+                               /* Save and carry on to close */
+                               _film->write_metadata ();
+                               break;
+                       case wxID_CANCEL:
+                               /* Veto the event and stop */
+                               ev.Veto ();
+                               return;
+                       }
+               }
+
                /* We don't want to hear about any more configuration changes, since they
                   cause the File menu to be altered, which itself will be deleted around
                   now (without, as far as I can see, any way for us to find out).
                */
                _config_changed_connection.disconnect ();
 
-               maybe_save_then_delete_film ();
                ev.Skip ();
        }
 
@@ -691,10 +792,13 @@ private:
                }
        }
 
-       void maybe_save_then_delete_film ()
+       /** @return true if the operation that called this method
+        *  should continue, false to abort it.
+        */
+       bool maybe_save_then_delete_film ()
        {
                if (!_film) {
-                       return;
+                       return true;
                }
 
                if (_film->dirty ()) {
@@ -705,10 +809,13 @@ private:
                        case wxID_YES:
                                _film->write_metadata ();
                                break;
+                       case wxID_CANCEL:
+                               return false;
                        }
                }
 
                _film.reset ();
+               return true;
        }
 
        void add_item (wxMenu* menu, wxString text, int id, int sens)
@@ -750,6 +857,7 @@ private:
 
                wxMenu* jobs_menu = new wxMenu;
                add_item (jobs_menu, _("&Make DCP\tCtrl-M"), ID_jobs_make_dcp, NEEDS_FILM | NOT_DURING_DCP_CREATION);
+               add_item (jobs_menu, _("Make DCP in &batch converter\tCtrl-B"), ID_jobs_make_dcp_batch, NEEDS_FILM | NOT_DURING_DCP_CREATION);
                add_item (jobs_menu, _("Make &KDMs...\tCtrl-K"), ID_jobs_make_kdms, NEEDS_FILM);
                add_item (jobs_menu, _("Make DKDM for DCP-o-matic..."), ID_jobs_make_self_dkdm, NEEDS_FILM);
                add_item (jobs_menu, _("&Send DCP to TMS"), ID_jobs_send_dcp_to_tms, NEEDS_FILM | NOT_DURING_DCP_CREATION | NEEDS_CPL);
@@ -883,7 +991,7 @@ private:
 
                wxSplashScreen* splash = 0;
                try {
-                       if (!Config::have_existing ()) {
+                       if (!Config::have_existing ("config.xml")) {
                                wxBitmap bitmap;
                                boost::filesystem::path p = shared_path () / "splash.png";
                                if (bitmap.LoadFile (std_to_wx (p.string ()), wxBITMAP_TYPE_PNG)) {
@@ -1014,7 +1122,7 @@ private:
                        error_dialog (
                                0,
                                wxString::Format (
-                                       _("An exception occurred: %s.\n\n") + " " + REPORT_PROBLEM,
+                                       _("An exception occurred: %s.\n\n") + REPORT_PROBLEM,
                                        std_to_wx (e.what ())
                                        )
                                );