X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Ftools%2Fdcpomatic_kdm.cc;h=a0a3a59f8786aa348ebc562578f207dbb1923b3f;hb=332ad44023ff81ae44b15f2587f4d40e4d5ccff5;hp=21be65dc4321109d12225a866ef518c1a5813526;hpb=7beb2cf613c7d310493fa39f9ef611b35f8f3d58;p=dcpomatic.git diff --git a/src/tools/dcpomatic_kdm.cc b/src/tools/dcpomatic_kdm.cc index 21be65dc4..a0a3a59f8 100644 --- a/src/tools/dcpomatic_kdm.cc +++ b/src/tools/dcpomatic_kdm.cc @@ -1,66 +1,136 @@ /* Copyright (C) 2015 Carl Hetherington - 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 . */ #include "wx/config_dialog.h" #include "wx/about_dialog.h" #include "wx/report_problem_dialog.h" +#include "wx/file_picker_ctrl.h" #include "wx/wx_util.h" #include "wx/wx_signal_manager.h" #include "wx/screens_panel.h" #include "wx/kdm_timing_panel.h" #include "wx/kdm_output_panel.h" +#include "wx/job_view_dialog.h" +#include "wx/file_dialog_wrapper.h" +#include "wx/editable_list.h" #include "lib/config.h" #include "lib/util.h" +#include "lib/screen.h" +#include "lib/job_manager.h" +#include "lib/screen_kdm.h" +#include "lib/exceptions.h" +#include "lib/cinema_kdms.h" +#include "lib/send_kdm_email_job.h" +#include "lib/compose.hpp" +#include "lib/cinema.h" #include #include +#include #include #include #include +#ifdef __WXOSX__ +#include +#endif #include +#include #ifdef check #undef check #endif using std::exception; +using std::list; +using std::string; +using std::vector; using boost::shared_ptr; using boost::bind; +using boost::optional; enum { - ID_help_report_a_problem, + ID_help_report_a_problem = 1, +}; + +class KDMFileDialogWrapper : public FileDialogWrapper +{ +public: + KDMFileDialogWrapper (wxWindow* parent) + : FileDialogWrapper (parent, _("Select DKDM file")) + { + + } + + optional get () + { + try { + return dcp::EncryptedKDM (dcp::file_to_string (wx_to_std (_dialog->GetPath ()), MAX_KDM_SIZE)); + } catch (cxml::Error& e) { + error_dialog (_parent, wxString::Format ("This file does not look like a KDM (%s)", std_to_wx (e.what()).data())); + } + + return optional (); + } }; +static string +column (dcp::EncryptedKDM k) +{ + return String::compose ("%1 (%2)", k.content_title_text(), k.cpl_id()); +} + class DOMFrame : public wxFrame { public: DOMFrame (wxString const & title) - : wxFrame (NULL, -1, title) + : wxFrame (0, -1, title) , _config_dialog (0) + , _job_view (0) { +#if defined(DCPOMATIC_WINDOWS) + if (Config::instance()->win32_console ()) { + AllocConsole(); + + HANDLE handle_out = GetStdHandle(STD_OUTPUT_HANDLE); + int hCrt = _open_osfhandle((intptr_t) handle_out, _O_TEXT); + FILE* hf_out = _fdopen(hCrt, "w"); + setvbuf(hf_out, NULL, _IONBF, 1); + *stdout = *hf_out; + + HANDLE handle_in = GetStdHandle(STD_INPUT_HANDLE); + hCrt = _open_osfhandle((intptr_t) handle_in, _O_TEXT); + FILE* hf_in = _fdopen(hCrt, "r"); + setvbuf(hf_in, NULL, _IONBF, 128); + *stdin = *hf_in; + + std::cout << "DCP-o-matic KDM creator is starting." << "\n"; + } +#endif + wxMenuBar* bar = new wxMenuBar; setup_menu (bar); SetMenuBar (bar); - Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::file_exit, this), wxID_EXIT); - Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::edit_preferences, this), wxID_PREFERENCES); - Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::help_about, this), wxID_ABOUT); - Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&DOMFrame::help_report_a_problem, this), ID_help_report_a_problem); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::file_exit, this), wxID_EXIT); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::edit_preferences, this), wxID_PREFERENCES); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::help_about, this), wxID_ABOUT); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::help_report_a_problem, this), ID_help_report_a_problem); /* Use a panel as the only child of the Frame so that we avoid the dark-grey background on Windows. @@ -68,67 +138,74 @@ public: wxPanel* overall_panel = new wxPanel (this, wxID_ANY); wxBoxSizer* main_sizer = new wxBoxSizer (wxHORIZONTAL); - wxBoxSizer* vertical = new wxBoxSizer (wxVERTICAL); + wxBoxSizer* horizontal = new wxBoxSizer (wxHORIZONTAL); + wxBoxSizer* left = new wxBoxSizer (wxVERTICAL); + wxBoxSizer* right = new wxBoxSizer (wxVERTICAL); + + horizontal->Add (left, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_X_GAP * 2); + horizontal->Add (right, 1, wxEXPAND); wxFont subheading_font (*wxNORMAL_FONT); subheading_font.SetWeight (wxFONTWEIGHT_BOLD); wxStaticText* h = new wxStaticText (overall_panel, wxID_ANY, _("Screens")); h->SetFont (subheading_font); - vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL); + left->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); _screens = new ScreensPanel (overall_panel); - vertical->Add (_screens, 1, wxEXPAND | wxALL, DCPOMATIC_SIZER_Y_GAP); + left->Add (_screens, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + /// TRANSLATORS: translate the word "Timing" here; do not include the "KDM|" prefix h = new wxStaticText (overall_panel, wxID_ANY, S_("KDM|Timing")); h->SetFont (subheading_font); - vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); + right->Add (h, 0, wxALIGN_CENTER_VERTICAL, DCPOMATIC_SIZER_Y_GAP * 2); _timing = new KDMTimingPanel (overall_panel); - vertical->Add (_timing, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); + right->Add (_timing, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); h = new wxStaticText (overall_panel, wxID_ANY, _("DKDM")); h->SetFont (subheading_font); - vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); - wxSizer* dkdm = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP); - add_label_to_sizer (dkdm, overall_panel, _("DKDM file"), true); -#ifdef DCPOMATIC_USE_OWN_PICKER - _dkdm = new FilePicker (overall_panel, _("Select a DKDM XML file..."), "*.xml"); -#else - _dkdm = new wxFilePickerCtrl (overall_panel, wxID_ANY, wxEmptyString, _("Select a DKDM XML file..."), "*.xml", wxDefaultPosition, wxSize (300, -1)); -#endif - dkdm->Add (_dkdm, 1, wxEXPAND); - add_label_to_sizer (dkdm, overall_panel, _("Annotation"), true); - _annotation_text = new wxStaticText (overall_panel, wxID_ANY, wxT("")); - dkdm->Add (_annotation_text, 1, wxEXPAND); - add_label_to_sizer (dkdm, overall_panel, _("Content title"), true); - _content_title_text = new wxStaticText (overall_panel, wxID_ANY, wxT("")); - dkdm->Add (_content_title_text, 1, wxEXPAND); - add_label_to_sizer (dkdm, overall_panel, _("Issue date"), true); - _issue_date = new wxStaticText (overall_panel, wxID_ANY, wxT("")); - dkdm->Add (_issue_date, 1, wxEXPAND); - vertical->Add (dkdm, 0, wxALL, DCPOMATIC_SIZER_X_GAP); + right->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); + + vector columns; + columns.push_back (wx_to_std (_("CPL"))); + _dkdm = new EditableList ( + overall_panel, columns, bind (&DOMFrame::dkdms, this), bind (&DOMFrame::set_dkdms, this, _1), bind (&column, _1), false + ); + right->Add (_dkdm, 0, wxEXPAND | wxALL, DCPOMATIC_SIZER_Y_GAP); h = new wxStaticText (overall_panel, wxID_ANY, _("Output")); h->SetFont (subheading_font); - vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); + right->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); /* XXX: hard-coded non-interop here */ _output = new KDMOutputPanel (overall_panel, false); - vertical->Add (_output, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); + right->Add (_output, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); _create = new wxButton (overall_panel, wxID_ANY, _("Create KDMs")); - vertical->Add (_create, 0, wxALL, DCPOMATIC_SIZER_GAP); + right->Add (_create, 0, wxALL, DCPOMATIC_SIZER_GAP); - main_sizer->Add (vertical, 1, wxALL, DCPOMATIC_DIALOG_BORDER); + main_sizer->Add (horizontal, 1, wxALL | wxEXPAND, DCPOMATIC_DIALOG_BORDER); overall_panel->SetSizer (main_sizer); /* Instantly save any config changes when using a DCP-o-matic GUI */ Config::instance()->Changed.connect (boost::bind (&Config::write, Config::instance ())); _screens->ScreensChanged.connect (boost::bind (&DOMFrame::setup_sensitivity, this)); - _dkdm->Bind (wxEVT_COMMAND_FILEPICKER_CHANGED, bind (&DOMFrame::dkdm_changed, this)); - _create->Bind (wxEVT_COMMAND_BUTTON_CLICKED, bind (&DOMFrame::create_kdms, this)); + _create->Bind (wxEVT_BUTTON, bind (&DOMFrame::create_kdms, this)); + _dkdm->SelectionChanged.connect (boost::bind (&DOMFrame::setup_sensitivity, this)); + + setup_sensitivity (); } private: + vector dkdms () const + { + return Config::instance()->dkdms (); + } + + void set_dkdms (vector dkdms) + { + Config::instance()->set_dkdms (dkdms); + } + void file_exit () { /* false here allows the close handler to veto the close request */ @@ -191,41 +268,80 @@ private: m->Append (help, _("&Help")); } - void dkdm_changed () - { - try { - dcp::EncryptedKDM encrypted (dcp::file_to_string (wx_to_std (_dkdm->GetPath()))); - dcp::DecryptedKDM decrypted (encrypted, Config::instance()->decryption_chain()->key().get()); - _annotation_text->Enable (true); - _annotation_text->SetLabel (std_to_wx (decrypted.annotation_text ())); - _content_title_text->Enable (true); - _content_title_text->SetLabel (std_to_wx (decrypted.content_title_text ())); - _issue_date->Enable (true); - _issue_date->SetLabel (std_to_wx (decrypted.issue_date ())); - } catch (exception& e) { - error_dialog (this, wxString::Format (_("Could not load DKDM (%s)"), std_to_wx (e.what()).data())); - _dkdm->SetPath (wxT("")); - _annotation_text->SetLabel (wxT("")); - _annotation_text->Enable (false); - _content_title_text->SetLabel (wxT("")); - _content_title_text->Enable (false); - _issue_date->SetLabel (wxT("")); - _issue_date->Enable (false); - } - } - void create_kdms () { -#if 0 try { + if (!_dkdm->selection()) { + return; + } + + /* Decrypt the DKDM */ + dcp::DecryptedKDM decrypted (_dkdm->selection().get(), Config::instance()->decryption_chain()->key().get()); + + /* This is the signer for our new KDMs */ + shared_ptr signer = Config::instance()->signer_chain (); + if (!signer->valid ()) { + throw InvalidSignerError (); + } + + list screen_kdms; + BOOST_FOREACH (shared_ptr i, _screens->screens()) { + + if (!i->recipient) { + continue; + } + + /* Make an empty KDM */ + dcp::DecryptedKDM kdm ( + dcp::LocalTime (_timing->from(), i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()), + dcp::LocalTime (_timing->until(), i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()), + decrypted.annotation_text().get_value_or (""), + decrypted.content_title_text(), + dcp::LocalTime().as_string() + ); + + /* Add keys from the DKDM */ + BOOST_FOREACH (dcp::DecryptedKDMKey const & j, decrypted.keys()) { + kdm.add_key (j); + } + + /* Encrypt */ + screen_kdms.push_back (ScreenKDM (i, kdm.encrypt (signer, i->recipient.get(), i->trusted_devices, _output->formulation()))); + } + + dcp::NameFormat::Map name_values; + name_values['f'] = decrypted.content_title_text(); + name_values['b'] = dcp::LocalTime(_timing->from()).date() + " " + dcp::LocalTime(_timing->from()).time_of_day(); + name_values['e'] = dcp::LocalTime(_timing->until()).date() + " " + dcp::LocalTime(_timing->until()).time_of_day(); + if (_output->write_to()) { - write_kdm_files ( - _film, d->screens (), d->cpl(), _timing->from(), _timing->until(), _output->formulation(), _output->directory() + ScreenKDM::write_files (screen_kdms, _output->directory(), _output->name_format(), name_values); + /* XXX: proper plural form support in wxWidgets? */ + wxString s = screen_kdms.size() == 1 ? _("%d KDM written to %s") : _("%d KDMs written to %s"); + message_dialog ( + this, + wxString::Format (s, int(screen_kdms.size()), std_to_wx(_output->directory().string()).data()) ); } else { - JobManager::instance()->add ( - shared_ptr (new SendKDMEmailJob (_film, d->screens (), d->cpl (), d->from (), d->until (), d->formulation ())) - ); + string film_name = decrypted.annotation_text().get_value_or (""); + if (film_name.empty ()) { + film_name = decrypted.content_title_text (); + } + shared_ptr job (new SendKDMEmailJob ( + CinemaKDMs::collect (screen_kdms), + _output->name_format(), + name_values, + decrypted.content_title_text(), + shared_ptr () + )); + + JobManager::instance()->add (job); + if (_job_view) { + _job_view->Destroy (); + _job_view = 0; + } + _job_view = new JobViewDialog (this, _("Send KDM emails"), job); + _job_view->ShowModal (); } } catch (dcp::NotEncryptedError& e) { error_dialog (this, _("CPL's content is not encrypted.")); @@ -234,29 +350,22 @@ private: } catch (...) { error_dialog (this, _("An unknown exception occurred.")); } -#endif } void setup_sensitivity () { _screens->setup_sensitivity (); _output->setup_sensitivity (); - _create->Enable (!_screens->screens().empty()); + _create->Enable (!_screens->screens().empty() && _dkdm->selection()); } wxPreferencesEditor* _config_dialog; ScreensPanel* _screens; KDMTimingPanel* _timing; -#ifdef DCPOMATIC_USE_OWN_PICKER - FilePicker* _dkdm; -#else - wxFilePickerCtrl* _dkdm; -#endif - wxStaticText* _annotation_text; - wxStaticText* _content_title_text; - wxStaticText* _issue_date; + EditableList* _dkdm; wxButton* _create; KDMOutputPanel* _output; + JobViewDialog* _job_view; }; /** @class App @@ -277,7 +386,7 @@ private: { wxInitAllImageHandlers (); - SetAppName (_("DCP-o-matic KDM creator")); + SetAppName (_("DCP-o-matic KDM Creator")); if (!wxApp::OnInit()) { return false; @@ -313,7 +422,7 @@ private: */ Config::drop (); - _frame = new DOMFrame (_("DCP-o-matic KDM creator")); + _frame = new DOMFrame (_("DCP-o-matic KDM Creator")); SetTopWindow (_frame); _frame->Maximize (); _frame->Show ();