#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;
ScreensPanel::ScreensPanel (wxWindow* parent)
: wxPanel (parent, wxID_ANY)
- , _ignore_selection_change (false)
{
auto sizer = new wxBoxSizer (wxVERTICAL);
sizer->Add (_search, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP);
auto targets = new wxBoxSizer (wxHORIZONTAL);
- _targets = new TreeListCtrl (this);
+ _targets = new wxTreeListCtrl (this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTL_MULTIPLE | wxTL_3STATE | wxTL_NO_HEADER);
_targets->AppendColumn (wxT("foo"));
+ _targets->SetSortColumn (0);
+ _targets->SetItemComparator (&_comparator);
+
targets->Add (_targets, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_GAP);
add_cinemas ();
optional<wxTreeListItem>
-ScreensPanel::add_cinema (shared_ptr<Cinema> c)
+ScreensPanel::add_cinema (shared_ptr<Cinema> cinema)
{
auto search = wx_to_std (_search->GetValue ());
transform (search.begin(), search.end(), search.begin(), ::tolower);
if (!search.empty ()) {
- auto name = c->name;
+ auto name = cinema->name;
transform (name.begin(), name.end(), name.begin(), ::tolower);
if (name.find (search) == string::npos) {
return {};
}
}
- auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(c->name));
+ auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(cinema->name));
- _cinemas[id] = c;
+ _cinemas.push_back(make_pair(id, cinema));
+ _item_to_cinema[id] = cinema;
+ _cinema_to_item[cinema] = id;
- for (auto i: c->screens()) {
- add_screen (c, i);
+ for (auto screen: cinema->screens()) {
+ add_screen (cinema, screen);
}
- _targets->SetSortColumn (0);
-
return id;
}
optional<wxTreeListItem>
-ScreensPanel::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
+ScreensPanel::add_screen (shared_ptr<Cinema> cinema, shared_ptr<Screen> screen)
{
- auto i = _cinemas.begin();
- while (i != _cinemas.end() && i->second != c) {
- ++i;
- }
-
- if (i == _cinemas.end()) {
+ auto item = cinema_to_item(cinema);
+ if (!item) {
return {};
}
- _screens[_targets->AppendItem (i->first, std_to_wx (s->name))] = s;
- return i->first;
+ auto id = _targets->AppendItem(*item, std_to_wx(screen->name));
+
+ _screens.push_back(make_pair(id, screen));
+ _item_to_screen[id] = screen;
+ _screen_to_item[screen] = id;
+
+ return item;
}
{
auto d = new CinemaDialog (GetParent(), _("Add Cinema"));
if (d->ShowModal () == wxID_OK) {
- auto c = make_shared<Cinema>(d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute());
- Config::instance()->add_cinema (c);
- auto id = add_cinema (c);
+ auto cinema = make_shared<Cinema>(d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute());
+ Config::instance()->add_cinema (cinema);
+ auto id = add_cinema (cinema);
if (id) {
_targets->UnselectAll ();
_targets->Select (*id);
return;
}
- for (auto i: cinema->second->screens()) {
- if (i->name == d->name()) {
+ for (auto screen: cinema->second->screens()) {
+ if (screen->name == d->name()) {
error_dialog (
GetParent(),
wxString::Format (
return;
}
- auto s = *_selected_screens.begin();
+ auto edit_screen = *_selected_screens.begin();
+
+ auto d = new ScreenDialog (
+ GetParent(), _("Edit screen"),
+ edit_screen.second->name,
+ edit_screen.second->notes,
+ edit_screen.second->recipient,
+ edit_screen.second->recipient_file,
+ edit_screen.second->trusted_devices
+ );
- auto d = new ScreenDialog (GetParent(), _("Edit screen"), s.second->name, s.second->notes, s.second->recipient, s.second->recipient_file, s.second->trusted_devices);
if (d->ShowModal() != wxID_OK) {
d->Destroy ();
return;
}
- auto c = s.second->cinema;
- for (auto i: c->screens ()) {
- if (i != s.second && i->name == d->name()) {
+ auto cinema = edit_screen.second->cinema;
+ for (auto screen: cinema->screens()) {
+ if (screen != edit_screen.second && screen->name == d->name()) {
error_dialog (
GetParent(),
wxString::Format (
}
}
- s.second->name = d->name ();
- s.second->notes = d->notes ();
- s.second->recipient = d->recipient ();
- s.second->recipient_file = d->recipient_file ();
- s.second->trusted_devices = d->trusted_devices ();
- _targets->SetItemText (s.first, std_to_wx (d->name()));
+ edit_screen.second->name = d->name ();
+ edit_screen.second->notes = d->notes ();
+ edit_screen.second->recipient = d->recipient ();
+ edit_screen.second->recipient_file = d->recipient_file ();
+ edit_screen.second->trusted_devices = d->trusted_devices ();
+ _targets->SetItemText (edit_screen.first, std_to_wx(d->name()));
Config::instance()->changed (Config::CINEMAS);
d->Destroy ();
for (auto item = _targets->GetFirstItem(); item.IsOk(); item = _targets->GetNextItem(item)) {
if (_targets->GetCheckedState(item) == wxCHK_CHECKED) {
- auto screen_iter = _screens.find(item);
- if (screen_iter != _screens.end()) {
- output.push_back (screen_iter->second);
+ if (auto screen = item_to_screen(item)) {
+ output.push_back (screen);
}
}
}
return;
}
- wxTreeListItems s;
- _targets->GetSelections (s);
+ wxTreeListItems selection;
+ _targets->GetSelections (selection);
_selected_cinemas.clear ();
_selected_screens.clear ();
- for (size_t i = 0; i < s.size(); ++i) {
- auto j = _cinemas.find (s[i]);
- if (j != _cinemas.end ()) {
- _selected_cinemas[j->first] = j->second;
+ for (size_t i = 0; i < selection.size(); ++i) {
+ if (auto cinema = item_to_cinema(selection[i])) {
+ _selected_cinemas.push_back(make_pair(selection[i], cinema));
}
- auto k = _screens.find (s[i]);
- if (k != _screens.end ()) {
- _selected_screens[k->first] = k->second;
+ if (auto screen = item_to_screen(selection[i])) {
+ _selected_screens.push_back(make_pair(selection[i], screen));
}
}
void
ScreensPanel::add_cinemas ()
{
- for (auto i: Config::instance()->cinemas()) {
- add_cinema (i);
+ for (auto cinema: Config::instance()->cinemas()) {
+ add_cinema (cinema);
}
}
_cinemas.clear ();
_screens.clear ();
+ _item_to_cinema.clear ();
+ _cinema_to_item.clear ();
+ _item_to_screen.clear ();
+ _screen_to_item.clear ();
+
add_cinemas ();
_ignore_selection_change = true;
- for (auto const& i: _selected_cinemas) {
+ for (auto const& selection: _selected_cinemas) {
/* The wxTreeListItems will now be different, so we must search by cinema */
- auto j = _cinemas.begin ();
- while (j != _cinemas.end() && j->second != i.second) {
- ++j;
+ if (auto item = cinema_to_item(selection.second)) {
+ _targets->Select (*item);
}
+ }
- if (j != _cinemas.end()) {
- _targets->Select (j->first);
+ for (auto const& selection: _selected_screens) {
+ /* Likewise by screen */
+ if (auto item = screen_to_item(selection.second)) {
+ _targets->Select (*item);
}
}
- for (auto const& i: _selected_screens) {
- auto j = _screens.begin ();
- while (j != _screens.end() && j->second != i.second) {
- ++j;
- }
+ _ignore_selection_change = false;
- if (j != _screens.end()) {
- _targets->Select (j->first);
+ _ignore_check_change = true;
+
+ for (auto const& checked: _checked_screens) {
+ if (auto item = screen_to_item(checked.second)) {
+ _targets->CheckItem(*item, wxCHK_CHECKED);
+ setup_cinema_checked_state(*item);
}
}
- _ignore_selection_change = false;
+ _ignore_check_change = false;
+}
+
+
+void
+ScreensPanel::set_screen_checked (wxTreeListItem item, bool checked)
+{
+ auto current = std::find_if(
+ _checked_screens.begin(), _checked_screens.end(),
+ [item](pair<wxTreeListItem, shared_ptr<Screen>> const& screen) { return screen.first == item; }
+ );
+
+ if (current == _checked_screens.end() && checked) {
+ _checked_screens.push_back({item, item_to_screen(item)});
+ } else if (current != _checked_screens.end() && !checked) {
+ _checked_screens.erase(current);
+ }
+}
+
+
+void
+ScreensPanel::setup_cinema_checked_state (wxTreeListItem screen)
+{
+ auto cinema = _targets->GetItemParent(screen);
+ DCPOMATIC_ASSERT (cinema.IsOk());
+ int checked = 0;
+ int unchecked = 0;
+ for (auto child = _targets->GetFirstChild(cinema); child.IsOk(); child = _targets->GetNextSibling(child)) {
+ if (_targets->GetCheckedState(child) == wxCHK_CHECKED) {
+ ++checked;
+ } else {
+ ++unchecked;
+ }
+ }
+ if (checked == 0) {
+ _targets->CheckItem(cinema, wxCHK_UNCHECKED);
+ } else if (unchecked == 0) {
+ _targets->CheckItem(cinema, wxCHK_CHECKED);
+ } else {
+ _targets->CheckItem(cinema, wxCHK_UNDETERMINED);
+ }
}
void
ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
{
- if (_cinemas.find(ev.GetItem()) != _cinemas.end()) {
+ if (_ignore_check_change) {
+ return;
+ }
+
+ 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)) {
_targets->CheckItem(child, checked);
+ set_screen_checked(child, checked);
}
} else {
- /* Screen: set cinema to checked/unchecked/3state */
- auto parent = _targets->GetItemParent(ev.GetItem());
- DCPOMATIC_ASSERT (parent.IsOk());
- int checked = 0;
- int unchecked = 0;
- for (auto child = _targets->GetFirstChild(parent); child.IsOk(); child = _targets->GetNextSibling(child)) {
- if (_targets->GetCheckedState(child) == wxCHK_CHECKED) {
- ++checked;
- } else {
- ++unchecked;
- }
- }
- if (checked == 0) {
- _targets->CheckItem(parent, wxCHK_UNCHECKED);
- } else if (unchecked == 0) {
- _targets->CheckItem(parent, wxCHK_CHECKED);
- } else {
- _targets->CheckItem(parent, wxCHK_UNDETERMINED);
- }
+ set_screen_checked(ev.GetItem(), _targets->GetCheckedState(ev.GetItem()));
+ setup_cinema_checked_state(ev.GetItem());
}
ScreensChanged ();
}
+shared_ptr<Cinema>
+ScreensPanel::item_to_cinema (wxTreeListItem item) const
+{
+ auto iter = _item_to_cinema.find (item);
+ if (iter == _item_to_cinema.end()) {
+ return {};
+ }
-wxIMPLEMENT_DYNAMIC_CLASS (TreeListCtrl, wxTreeListCtrl);
+ return iter->second;
+}
-int
-TreeListCtrl::OnCompareItems (wxTreeListItem const& a, wxTreeListItem const& b)
+shared_ptr<Screen>
+ScreensPanel::item_to_screen (wxTreeListItem item) const
{
- return strcoll (wx_to_std(GetItemText(a)).c_str(), wx_to_std(GetItemText(b)).c_str());
+ auto iter = _item_to_screen.find (item);
+ if (iter == _item_to_screen.end()) {
+ return {};
+ }
+
+ return iter->second;
+}
+
+
+optional<wxTreeListItem>
+ScreensPanel::cinema_to_item (shared_ptr<Cinema> cinema) const
+{
+ auto iter = _cinema_to_item.find (cinema);
+ if (iter == _cinema_to_item.end()) {
+ return {};
+ }
+
+ return iter->second;
}
+optional<wxTreeListItem>
+ScreensPanel::screen_to_item (shared_ptr<Screen> screen) const
+{
+ auto iter = _screen_to_item.find (screen);
+ if (iter == _screen_to_item.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());
+ }
+}