Keep maps of wxTreeListItem to Cinema/Screen.

[dcpomatic.git] / src / wx / screens_panel.cc

diff --git a/src/wx/screens_panel.cc b/src/wx/screens_panel.cc
index 2075515f353fe46edae7eb3b02ec13ae0064508f..8cdba50d446bf15f4599f0a0aacb067e309d19f5 100644 (file)
--- a/src/wx/screens_panel.cc
+++ b/src/wx/screens_panel.cc
@@ -27,6 +27,11 @@
 #include "lib/cinema.h"
 #include "lib/config.h"
 #include "lib/screen.h"
+#include <unicode/putil.h>
+#include <unicode/ucol.h>
+#include <unicode/uiter.h>
+#include <unicode/utypes.h>
+#include <unicode/ustring.h>
 
 
 using std::cout;
@@ -138,6 +143,7 @@ ScreensPanel::add_cinema (shared_ptr<Cinema> cinema)
        auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(cinema->name));
 
        _cinemas.push_back(make_pair(id, cinema));
+       _item_to_cinema[id] = cinema;
 
        for (auto screen: cinema->screens()) {
                add_screen (cinema, screen);
@@ -159,7 +165,11 @@ ScreensPanel::add_screen (shared_ptr<Cinema> cinema, shared_ptr<Screen> screen)
                return {};
        }
 
-       _screens.push_back(make_pair(_targets->AppendItem(cinema_iter->first, std_to_wx(screen->name)), screen));
+       auto id = _targets->AppendItem(cinema_iter->first, std_to_wx(screen->name));
+
+       _screens.push_back(make_pair(id, screen));
+       _item_to_screen[id] = screen;
+
        return cinema_iter->first;
 }
 
@@ -375,9 +385,8 @@ ScreensPanel::screens () const
 
        for (auto item = _targets->GetFirstItem(); item.IsOk(); item = _targets->GetNextItem(item)) {
                if (_targets->GetCheckedState(item) == wxCHK_CHECKED) {
-                       auto screen_iter = screen_by_tree_list_item(item);
-                       if (screen_iter != _screens.end()) {
-                               output.push_back (screen_iter->second);
+                       if (auto screen = item_to_screen(item)) {
+                               output.push_back (screen);
                        }
                }
        }
@@ -407,13 +416,11 @@ ScreensPanel::selection_changed ()
        _selected_screens.clear ();
 
        for (size_t i = 0; i < selection.size(); ++i) {
-               auto cinema = cinema_by_tree_list_item(selection[i]);
-               if (cinema != _cinemas.end ()) {
-                       _selected_cinemas.push_back(*cinema);
+               if (auto cinema = item_to_cinema(selection[i])) {
+                       _selected_cinemas.push_back(make_pair(selection[i], cinema));
                }
-               auto screen = screen_by_tree_list_item(selection[i]);
-               if (screen != _screens.end ()) {
-                       _selected_screens.push_back(*screen);
+               if (auto screen = item_to_screen(selection[i])) {
+                       _selected_screens.push_back(make_pair(selection[i], screen));
                }
        }
 
@@ -437,6 +444,8 @@ ScreensPanel::search_changed ()
        _cinemas.clear ();
        _screens.clear ();
 
+       _item_to_cinema.clear ();
+       _item_to_screen.clear ();
        add_cinemas ();
 
        _ignore_selection_change = true;
@@ -471,8 +480,7 @@ ScreensPanel::search_changed ()
 void
 ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
 {
-       auto cinema_iter = cinema_by_tree_list_item(ev.GetItem());
-       if (cinema_iter != _cinemas.end()) {
+       if (item_to_cinema(ev.GetItem())) {
                /* Cinema: check/uncheck all children */
                auto const checked = _targets->GetCheckedState(ev.GetItem());
                for (auto child = _targets->GetFirstChild(ev.GetItem()); child.IsOk(); child = _targets->GetNextSibling(child)) {
@@ -504,22 +512,61 @@ ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
 }
 
 
-ScreensPanel::Cinemas::iterator
-ScreensPanel::cinema_by_tree_list_item (wxTreeListItem item)
+shared_ptr<Cinema>
+ScreensPanel::item_to_cinema (wxTreeListItem item) const
 {
-       return std::find_if(
-               _cinemas.begin(), _cinemas.end(),
-               [item](pair<wxTreeListItem, shared_ptr<Cinema>> const& s) { return s.first == item; }
-               );
+       auto iter = _item_to_cinema.find (item);
+       if (iter == _item_to_cinema.end()) {
+               return {};
+       }
+
+       return iter->second;
 }
 
 
-ScreensPanel::Screens::const_iterator
-ScreensPanel::screen_by_tree_list_item (wxTreeListItem item) const
+shared_ptr<Screen>
+ScreensPanel::item_to_screen (wxTreeListItem item) const
 {
-       return std::find_if(
-               _screens.begin(), _screens.end(),
-               [item](pair<wxTreeListItem, shared_ptr<Screen>> const& s) { return s.first == item; }
-               );
+       auto iter = _item_to_screen.find (item);
+       if (iter == _item_to_screen.end()) {
+               return {};
+       }
+
+       return iter->second;
 }
 
+
+ScreensPanel::Comparator::Comparator ()
+{
+       UErrorCode status = U_ZERO_ERROR;
+       _collator = ucol_open(nullptr, &status);
+       if (_collator) {
+               ucol_setAttribute(_collator, UCOL_NORMALIZATION_MODE, UCOL_ON, &status);
+               ucol_setAttribute(_collator, UCOL_STRENGTH, UCOL_PRIMARY, &status);
+               ucol_setAttribute(_collator, UCOL_ALTERNATE_HANDLING, UCOL_SHIFTED, &status);
+       }
+}
+
+ScreensPanel::Comparator::~Comparator ()
+{
+       if (_collator) {
+               ucol_close (_collator);
+       }
+}
+
+int
+ScreensPanel::Comparator::Compare (wxTreeListCtrl* tree_list, unsigned, wxTreeListItem a, wxTreeListItem b)
+{
+       auto utf8_a = wx_to_std(tree_list->GetItemText(a));
+       auto utf8_b = wx_to_std(tree_list->GetItemText(b));
+       if (_collator) {
+               UErrorCode error = U_ZERO_ERROR;
+               boost::scoped_array<uint16_t> utf16_a(new uint16_t[utf8_a.size() + 1]);
+               u_strFromUTF8(reinterpret_cast<UChar*>(utf16_a.get()), utf8_a.size() + 1, nullptr, utf8_a.c_str(), -1, &error);
+               boost::scoped_array<uint16_t> utf16_b(new uint16_t[utf8_b.size() + 1]);
+               u_strFromUTF8(reinterpret_cast<UChar*>(utf16_b.get()), utf8_b.size() + 1, nullptr, utf8_b.c_str(), -1, &error);
+               return ucol_strcoll(_collator, reinterpret_cast<UChar*>(utf16_a.get()), -1, reinterpret_cast<UChar*>(utf16_b.get()), -1);
+       } else {
+               return strcoll(utf8_a.c_str(), utf8_b.c_str());
+       }
+}