X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Ftools%2Fdcpomatic_kdm.cc;h=256367977e988a67112b455936564cc00bd7e60a;hb=b7e65adf286ce20918797a06a910ededf8f07b7b;hp=f877ad21f8510d4848f4f162660e563440aca29b;hpb=f5bd787fb71466a6187050086b2890b9b0ac9294;p=dcpomatic.git diff --git a/src/tools/dcpomatic_kdm.cc b/src/tools/dcpomatic_kdm.cc index f877ad21f..256367977 100644 --- a/src/tools/dcpomatic_kdm.cc +++ b/src/tools/dcpomatic_kdm.cc @@ -22,6 +22,7 @@ #include "wx/about_dialog.h" #include "wx/dcpomatic_button.h" #include "wx/editable_list.h" +#include "wx/file_dialog.h" #include "wx/file_picker_ctrl.h" #include "wx/full_config_dialog.h" #include "wx/job_view_dialog.h" @@ -35,7 +36,9 @@ #include "wx/wx_signal_manager.h" #include "wx/wx_util.h" #include "lib/cinema.h" +#include "lib/collator.h" #include "lib/compose.hpp" +#include "lib/constants.h" #include "lib/config.h" #include "lib/cross.h" #include "lib/dcpomatic_log.h" @@ -44,17 +47,19 @@ #include "lib/file_log.h" #include "lib/job_manager.h" #include "lib/kdm_with_metadata.h" +#include "lib/scope_guard.h" #include "lib/screen.h" #include "lib/send_kdm_email_job.h" -#include "lib/util.h" #include #include #include #include LIBDCP_DISABLE_WARNINGS +#include #include #include #include +#include #include #include LIBDCP_ENABLE_WARNINGS @@ -62,6 +67,7 @@ LIBDCP_ENABLE_WARNINGS #include #endif #include +#include #ifdef check #undef check @@ -75,6 +81,7 @@ using std::map; using std::pair; using std::shared_ptr; using std::string; +using std::unordered_set; using std::vector; using boost::bind; using boost::optional; @@ -160,12 +167,42 @@ public: h = new StaticText (overall_panel, _("DKDM")); h->SetFont (subheading_font); right->Add (h, 0, wxTOP, DCPOMATIC_SIZER_Y_GAP * 2); + + _dkdm_search = new wxSearchCtrl(overall_panel, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, search_ctrl_height())); +#ifndef __WXGTK3__ + /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */ + _dkdm_search->ShowCancelButton (true); +#endif + + right->Add(_dkdm_search, 0, wxTOP | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + + class DKDMDropTarget : public wxFileDropTarget + { + public: + DKDMDropTarget(DOMFrame* frame) + : _frame(frame) + {} + + bool OnDropFiles(wxCoord, wxCoord, wxArrayString const& filenames) override + { + for (size_t i = 0; i < filenames.GetCount(); ++i) { + _frame->add_dkdm(boost::filesystem::path(wx_to_std(filenames[0]))); + } + + return true; + } + + private: + DOMFrame* _frame; + }; + auto dkdm_sizer = new wxBoxSizer (wxHORIZONTAL); _dkdm = new wxTreeCtrl ( overall_panel, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_HIDE_ROOT | wxTR_HAS_BUTTONS | wxTR_LINES_AT_ROOT ); - dkdm_sizer->Add (_dkdm, 1, wxEXPAND | wxALL, DCPOMATIC_SIZER_Y_GAP); - wxBoxSizer* dkdm_buttons = new wxBoxSizer(wxVERTICAL); + _dkdm->SetDropTarget(new DKDMDropTarget(this)); + dkdm_sizer->Add(_dkdm, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + auto dkdm_buttons = new wxBoxSizer(wxVERTICAL); _add_dkdm = new Button (overall_panel, _("Add...")); dkdm_buttons->Add (_add_dkdm, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); _add_dkdm_folder = new Button (overall_panel, _("Add folder...")); @@ -175,9 +212,9 @@ public: _export_dkdm = new Button (overall_panel, _("Export...")); dkdm_buttons->Add (_export_dkdm, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); dkdm_sizer->Add (dkdm_buttons, 0, wxEXPAND | wxALL, DCPOMATIC_SIZER_GAP); - right->Add (dkdm_sizer, 1, wxEXPAND | wxALL, DCPOMATIC_SIZER_Y_GAP); + right->Add (dkdm_sizer, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); - add_dkdm_view (Config::instance()->dkdms()); + update_dkdm_view(); h = new StaticText (overall_panel, _("Output")); h->SetFont (subheading_font); @@ -196,13 +233,16 @@ public: _screens->ScreensChanged.connect (boost::bind (&DOMFrame::setup_sensitivity, this)); _create->Bind (wxEVT_BUTTON, bind (&DOMFrame::create_kdms, this)); - _dkdm->Bind (wxEVT_TREE_SEL_CHANGED, boost::bind (&DOMFrame::setup_sensitivity, this)); + _dkdm->Bind(wxEVT_TREE_SEL_CHANGED, boost::bind(&DOMFrame::dkdm_selection_changed, this)); _dkdm->Bind (wxEVT_TREE_BEGIN_DRAG, boost::bind (&DOMFrame::dkdm_begin_drag, this, _1)); _dkdm->Bind (wxEVT_TREE_END_DRAG, boost::bind (&DOMFrame::dkdm_end_drag, this, _1)); + _dkdm->Bind(wxEVT_TREE_ITEM_EXPANDED, boost::bind(&DOMFrame::dkdm_expanded, this, _1)); + _dkdm->Bind(wxEVT_TREE_ITEM_COLLAPSED, boost::bind(&DOMFrame::dkdm_collapsed, this, _1)); _add_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::add_dkdm_clicked, this)); _add_dkdm_folder->Bind (wxEVT_BUTTON, bind (&DOMFrame::add_dkdm_folder_clicked, this)); _remove_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::remove_dkdm_clicked, this)); _export_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::export_dkdm_clicked, this)); + _dkdm_search->Bind(wxEVT_TEXT, boost::bind(&DOMFrame::dkdm_search_changed, this)); setup_sensitivity (); @@ -227,17 +267,17 @@ private: void help_about () { auto d = new AboutDialog (this); + ScopeGuard sg = [d]() { d->Destroy(); }; d->ShowModal (); - d->Destroy (); } void help_report_a_problem () { auto d = new ReportProblemDialog (this, shared_ptr()); + ScopeGuard sg = [d]() { d->Destroy(); }; if (d->ShowModal () == wxID_OK) { d->report (); } - d->Destroy (); } void setup_menu (wxMenuBar* m) @@ -280,8 +320,8 @@ private: ); } - /** @id if not 0 this is filled in with the wxTreeItemId of the selection */ - shared_ptr selected_dkdm (wxTreeItemId* id = 0) const + /** @id if not nullptr this is filled in with the wxTreeItemId of the selection */ + shared_ptr selected_dkdm (wxTreeItemId* id = nullptr) const { wxArrayTreeItemIds selections; _dkdm->GetSelections (selections); @@ -334,8 +374,8 @@ private: continue; } - dcp::LocalTime begin(_timing->from(), i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()); - dcp::LocalTime end(_timing->until(), i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()); + dcp::LocalTime begin(_timing->from(), dcp::UTCOffset(i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute())); + dcp::LocalTime end(_timing->until(), dcp::UTCOffset(i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute())); /* Make an empty KDM */ dcp::DecryptedKDM kdm ( @@ -417,6 +457,32 @@ private: _export_dkdm->Enable (sel.GetCount() > 0 && dkdm); } + void dkdm_selection_changed() + { + _selected_dkdm = selected_dkdm(); + setup_sensitivity(); + } + + void dkdm_expanded(wxTreeEvent& ev) + { + if (_ignore_expand) { + return; + } + + auto iter = _dkdm_id.find(ev.GetItem()); + if (iter != _dkdm_id.end()) { + _expanded_dkdm_groups.insert(iter->second); + } + } + + void dkdm_collapsed(wxTreeEvent& ev) + { + auto iter = _dkdm_id.find(ev.GetItem()); + if (iter != _dkdm_id.end()) { + _expanded_dkdm_groups.erase(iter->second); + } + } + void dkdm_begin_drag (wxTreeEvent& ev) { ev.Allow (); @@ -439,87 +505,129 @@ private: DCPOMATIC_ASSERT (from->second->parent ()); from->second->parent()->remove (from->second); - add_dkdm_model (from->second, group, dynamic_pointer_cast(to->second)); + add_dkdm(from->second, group, dynamic_pointer_cast(to->second)); - _dkdm->Delete (from->first); - _dkdm_id.erase (from->first); - add_dkdm_view (from->second, dynamic_pointer_cast(to->second) ? to->first : optional()); + update_dkdm_view(); } void add_dkdm_clicked () { - auto d = new wxFileDialog( - this, - _("Select DKDM file"), - wxEmptyString, - wxEmptyString, - wxT("XML files|*.xml|All files|*.*"), - wxFD_MULTIPLE - ); + auto dialog = new FileDialog(this, _("Select DKDM file"), wxT("XML files|*.xml|All files|*.*"), wxFD_MULTIPLE, "AddDKDMPath"); - if (d->ShowModal() == wxID_OK) { - auto chain = Config::instance()->decryption_chain(); - DCPOMATIC_ASSERT (chain->key()); - - wxArrayString paths; - d->GetPaths(paths); - for (unsigned int i = 0; i < paths.GetCount(); ++i) { - try { - dcp::EncryptedKDM ekdm(dcp::file_to_string(wx_to_std(paths[i]), MAX_KDM_SIZE)); - /* Decrypt the DKDM to make sure that we can */ - dcp::DecryptedKDM dkdm(ekdm, chain->key().get()); - - auto new_dkdm = make_shared(ekdm); - auto group = dynamic_pointer_cast (selected_dkdm()); - if (!group) { - group = Config::instance()->dkdms (); - } - add_dkdm_model (new_dkdm, group); - add_dkdm_view (new_dkdm); - } catch (dcp::KDMFormatError& e) { - error_dialog ( - this, - _("Could not read file as a KDM. Perhaps it is badly formatted, or not a KDM at all."), - std_to_wx(e.what()) - ); - return; - } catch (dcp::KDMDecryptionError &) { - error_dialog ( - this, - _("Could not decrypt the DKDM. Perhaps it was not created with the correct certificate.") - ); - } catch (dcp::MiscError& e) { - error_dialog ( - this, - _("Could not read file as a KDM. It is much too large. Make sure you are loading a DKDM (XML) file."), - std_to_wx(e.what()) - ); - } + ScopeGuard sg = [dialog]() { dialog->Destroy(); }; + + if (!dialog->show()) { + return; + } + + for (auto path: dialog->paths()) { + add_dkdm(path); + } + } + + void add_dkdm(boost::filesystem::path path) + { + auto chain = Config::instance()->decryption_chain(); + DCPOMATIC_ASSERT (chain->key()); + + try { + dcp::EncryptedKDM ekdm(dcp::file_to_string(path, MAX_KDM_SIZE)); + /* Decrypt the DKDM to make sure that we can */ + dcp::DecryptedKDM dkdm(ekdm, chain->key().get()); + + auto new_dkdm = make_shared(ekdm); + auto group = dynamic_pointer_cast (selected_dkdm()); + if (!group) { + group = Config::instance()->dkdms (); } + add_dkdm(new_dkdm, group); + } catch (dcp::KDMFormatError& e) { + error_dialog ( + this, + _("Could not read file as a KDM. Perhaps it is badly formatted, or not a KDM at all."), + std_to_wx(e.what()) + ); + return; + } catch (dcp::KDMDecryptionError &) { + error_dialog ( + this, + _("Could not decrypt the DKDM. Perhaps it was not created with the correct certificate.") + ); + } catch (dcp::MiscError& e) { + error_dialog ( + this, + _("Could not read file as a KDM. It is much too large. Make sure you are loading a DKDM (XML) file."), + std_to_wx(e.what()) + ); } - d->Destroy (); + + update_dkdm_view(); } void add_dkdm_folder_clicked () { auto d = new NewDKDMFolderDialog (this); - if (d->ShowModal() == wxID_OK) { - auto new_dkdm = make_shared(wx_to_std(d->get())); - auto parent = dynamic_pointer_cast(selected_dkdm()); - if (!parent) { - parent = Config::instance()->dkdms (); + ScopeGuard sg = [d]() { d->Destroy(); }; + if (d->ShowModal() != wxID_OK) { + return; + } + + auto new_dkdm = make_shared(wx_to_std(d->get())); + auto parent = dynamic_pointer_cast(selected_dkdm()); + if (!parent) { + parent = Config::instance()->dkdms (); + } + add_dkdm(new_dkdm, parent); + update_dkdm_view(); + } + + void update_dkdm_view() + { + _dkdm->DeleteAllItems(); + _dkdm_id.clear(); + add_dkdm_to_view(Config::instance()->dkdms()); + if (_selected_dkdm) { + auto selection_in_id_map = std::find_if(_dkdm_id.begin(), _dkdm_id.end(), [this](pair> const& entry) { + return entry.second == _selected_dkdm; + }); + if (selection_in_id_map != _dkdm_id.end()) { + _dkdm->SelectItem(selection_in_id_map->first); } - add_dkdm_model (new_dkdm, parent); - add_dkdm_view (new_dkdm); } - d->Destroy (); } - /** @param dkdm Thing to add. - * @param parent Parent group, or 0. - */ - void add_dkdm_view (shared_ptr base, optional previous = optional()) + /** @return true if this thing or any of its children match a search string */ + bool matches(shared_ptr base, string const& search) + { + if (search.empty()) { + return true; + } + + auto name = base->name(); + transform(name.begin(), name.end(), name.begin(), ::tolower); + if (name.find(search) != string::npos) { + return true; + } + + auto group = dynamic_pointer_cast(base); + if (!group) { + return false; + } + + auto const children = group->children(); + return std::any_of(children.begin(), children.end(), [this, search](shared_ptr child) { + return matches(child, search); + }); + } + + /** Add DKDMs to the view that match the current search */ + void add_dkdm_to_view(shared_ptr base) { + auto search = wx_to_std(_dkdm_search->GetValue()); + transform(search.begin(), search.end(), search.begin(), ::tolower); + + optional group_to_expand; + if (!base->parent()) { /* This is the root group */ _dkdm_id[_dkdm->AddRoot("root")] = base; @@ -527,26 +635,38 @@ private: /* Add base to the view */ wxTreeItemId added; auto parent_id = dkdm_to_id(base->parent()); - if (previous) { - added = _dkdm->InsertItem(parent_id, *previous, std_to_wx(base->name())); - } else { - added = _dkdm->AppendItem(parent_id, std_to_wx(base->name())); + added = _dkdm->AppendItem(parent_id, std_to_wx(base->name())); + /* Expand the group (later) if it matches the search or it was manually expanded */ + if (!search.empty() || _expanded_dkdm_groups.find(base) != _expanded_dkdm_groups.end()) { + group_to_expand = added; } - _dkdm->SortChildren(parent_id); _dkdm_id[added] = base; } /* Add children */ - auto g = dynamic_pointer_cast(base); - if (g) { - for (auto i: g->children()) { - add_dkdm_view (i); + auto group = dynamic_pointer_cast(base); + if (group) { + auto children = group->children(); + children.sort( + [this](shared_ptr a, shared_ptr b) { return _collator.compare(a->name(), b->name()) < 0; } + ); + + for (auto i: children) { + if (matches(i, search)) { + add_dkdm_to_view(i); + } } } + + if (group_to_expand) { + _ignore_expand = true; + _dkdm->Expand(*group_to_expand); + _ignore_expand = false; + } } /** @param group Group to add dkdm to */ - void add_dkdm_model (shared_ptr dkdm, shared_ptr group, shared_ptr previous = shared_ptr ()) + void add_dkdm(shared_ptr dkdm, shared_ptr group, shared_ptr previous = shared_ptr()) { group->add (dkdm, previous); /* We're messing with a Config-owned object here, so tell it that something has changed. @@ -602,19 +722,35 @@ private: this, _("Select DKDM File"), wxEmptyString, wxEmptyString, wxT("XML files (*.xml)|*.xml"), wxFD_SAVE | wxFD_OVERWRITE_PROMPT ); + ScopeGuard sg = [d]() { d->Destroy(); }; if (d->ShowModal() == wxID_OK) { dkdm->dkdm().as_xml(wx_to_std(d->GetPath())); } - d->Destroy (); + } + + void dkdm_search_changed() + { + update_dkdm_view(); } wxPreferencesEditor* _config_dialog; ScreensPanel* _screens; KDMTimingPanel* _timing; wxTreeCtrl* _dkdm; + wxSearchCtrl* _dkdm_search; typedef std::map> DKDMMap; DKDMMap _dkdm_id; + /* Keep a separate track of the selected DKDM so that when a search happens, and some things + * get removed from the view, we can restore the selection when they are re-added. + */ + shared_ptr _selected_dkdm; + /* Keep expanded groups for the same reason */ + unordered_set> _expanded_dkdm_groups; + /* true if we are "artificially" expanding a group because it contains something found + * in a search. + */ + bool _ignore_expand = false; wxButton* _add_dkdm; wxButton* _add_dkdm_folder; wxButton* _remove_dkdm; @@ -622,6 +758,7 @@ private: wxButton* _create; KDMOutputPanel* _output; JobViewDialog* _job_view; + Collator _collator; };