src/wx/screens_panel.cc

   1 /*
   2     Copyright (C) 2015-2022 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 #include "cinema_dialog.h"
  23 #include "dcpomatic_button.h"
  24 #include "screen_dialog.h"
  25 #include "screens_panel.h"
  26 #include "wx_util.h"
  27 #include "lib/cinema.h"
  28 #include "lib/config.h"
  29 #include "lib/screen.h"
  30 #include <unicode/putil.h>
  31 #include <unicode/ucol.h>
  32 #include <unicode/uiter.h>
  33 #include <unicode/utypes.h>
  34 #include <unicode/ustring.h>
  35
  36
  37 using std::cout;
  38 using std::make_pair;
  39 using std::make_shared;
  40 using std::map;
  41 using std::pair;
  42 using std::shared_ptr;
  43 using std::string;
  44 using std::vector;
  45 using boost::optional;
  46 using namespace dcpomatic;
  47
  48
  49 ScreensPanel::ScreensPanel (wxWindow* parent)
  50         : wxPanel (parent, wxID_ANY)
  51         , _ignore_selection_change (false)
  52 {
  53         auto sizer = new wxBoxSizer (wxVERTICAL);
  54
  55         _search = new wxSearchCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, search_ctrl_height()));
  56 #ifndef __WXGTK3__
  57         /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */
  58         _search->ShowCancelButton (true);
  59 #endif
  60         sizer->Add (_search, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP);
  61
  62         auto targets = new wxBoxSizer (wxHORIZONTAL);
  63         _targets = new wxTreeListCtrl (this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTL_MULTIPLE | wxTL_3STATE | wxTL_NO_HEADER);
  64         _targets->AppendColumn (wxT("foo"));
  65         _targets->SetSortColumn (0);
  66         _targets->SetItemComparator (&_comparator);
  67
  68         targets->Add (_targets, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_GAP);
  69
  70         add_cinemas ();
  71
  72         auto target_buttons = new wxBoxSizer (wxVERTICAL);
  73
  74         _add_cinema = new Button (this, _("Add Cinema..."));
  75         target_buttons->Add (_add_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  76         _edit_cinema = new Button (this, _("Edit Cinema..."));
  77         target_buttons->Add (_edit_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  78         _remove_cinema = new Button (this, _("Remove Cinema"));
  79         target_buttons->Add (_remove_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  80         _add_screen = new Button (this, _("Add Screen..."));
  81         target_buttons->Add (_add_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  82         _edit_screen = new Button (this, _("Edit Screen..."));
  83         target_buttons->Add (_edit_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  84         _remove_screen = new Button (this, _("Remove Screen"));
  85         target_buttons->Add (_remove_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  86
  87         targets->Add (target_buttons, 0, 0);
  88
  89         sizer->Add (targets, 1, wxEXPAND);
  90
  91         _search->Bind        (wxEVT_TEXT, boost::bind (&ScreensPanel::search_changed, this));
  92         _targets->Bind       (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
  93         _targets->Bind       (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
  94
  95         _add_cinema->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_cinema_clicked, this));
  96         _edit_cinema->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_cinema_clicked, this));
  97         _remove_cinema->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_cinema_clicked, this));
  98
  99         _add_screen->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_screen_clicked, this));
 100         _edit_screen->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_screen_clicked, this));
 101         _remove_screen->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_screen_clicked, this));
 102
 103         SetSizer (sizer);
 104 }
 105
 106
 107 ScreensPanel::~ScreensPanel ()
 108 {
 109         _targets->Unbind (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
 110         _targets->Unbind (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
 111 }
 112
 113
 114 void
 115 ScreensPanel::setup_sensitivity ()
 116 {
 117         bool const sc = _selected_cinemas.size() == 1;
 118         bool const ss = _selected_screens.size() == 1;
 119
 120         _edit_cinema->Enable (sc || ss);
 121         _remove_cinema->Enable (_selected_cinemas.size() >= 1);
 122
 123         _add_screen->Enable (sc || ss);
 124         _edit_screen->Enable (ss);
 125         _remove_screen->Enable (_selected_screens.size() >= 1);
 126 }
 127
 128
 129 optional<wxTreeListItem>
 130 ScreensPanel::add_cinema (shared_ptr<Cinema> cinema)
 131 {
 132         auto search = wx_to_std (_search->GetValue ());
 133         transform (search.begin(), search.end(), search.begin(), ::tolower);
 134
 135         if (!search.empty ()) {
 136                 auto name = cinema->name;
 137                 transform (name.begin(), name.end(), name.begin(), ::tolower);
 138                 if (name.find (search) == string::npos) {
 139                         return {};
 140                 }
 141         }
 142
 143         auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(cinema->name));
 144
 145         _cinemas.push_back(make_pair(id, cinema));
 146         _item_to_cinema[id] = cinema;
 147         _cinema_to_item[cinema] = id;
 148
 149         for (auto screen: cinema->screens()) {
 150                 add_screen (cinema, screen);
 151         }
 152
 153         return id;
 154 }
 155
 156
 157 optional<wxTreeListItem>
 158 ScreensPanel::add_screen (shared_ptr<Cinema> cinema, shared_ptr<Screen> screen)
 159 {
 160         auto item = cinema_to_item(cinema);
 161         if (!item) {
 162                 return {};
 163         }
 164
 165         auto id = _targets->AppendItem(*item, std_to_wx(screen->name));
 166
 167         _screens.push_back(make_pair(id, screen));
 168         _item_to_screen[id] = screen;
 169         _screen_to_item[screen] = id;
 170
 171         return item;
 172 }
 173
 174
 175 void
 176 ScreensPanel::add_cinema_clicked ()
 177 {
 178         auto d = new CinemaDialog (GetParent(), _("Add Cinema"));
 179         if (d->ShowModal () == wxID_OK) {
 180                 auto cinema = make_shared<Cinema>(d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute());
 181                 Config::instance()->add_cinema (cinema);
 182                 auto id = add_cinema (cinema);
 183                 if (id) {
 184                         _targets->UnselectAll ();
 185                         _targets->Select (*id);
 186                 }
 187         }
 188
 189         d->Destroy ();
 190 }
 191
 192
 193 optional<pair<wxTreeListItem, shared_ptr<Cinema>>>
 194 ScreensPanel::cinema_for_operation () const
 195 {
 196         if (_selected_cinemas.size() == 1) {
 197                 return make_pair(_selected_cinemas.begin()->first, _selected_cinemas.begin()->second);
 198         } else if (_selected_screens.size() == 1) {
 199                 return make_pair(_targets->GetItemParent(_selected_screens.begin()->first), _selected_screens.begin()->second->cinema);
 200         }
 201
 202         return {};
 203 }
 204
 205
 206 void
 207 ScreensPanel::edit_cinema_clicked ()
 208 {
 209         auto cinema = cinema_for_operation ();
 210         if (!cinema) {
 211                 return;
 212         }
 213
 214         auto d = new CinemaDialog (
 215                 GetParent(), _("Edit cinema"), cinema->second->name, cinema->second->emails, cinema->second->notes, cinema->second->utc_offset_hour(), cinema->second->utc_offset_minute()
 216                 );
 217
 218         if (d->ShowModal() == wxID_OK) {
 219                 cinema->second->name = d->name ();
 220                 cinema->second->emails = d->emails ();
 221                 cinema->second->notes = d->notes ();
 222                 cinema->second->set_utc_offset_hour (d->utc_offset_hour ());
 223                 cinema->second->set_utc_offset_minute (d->utc_offset_minute ());
 224                 _targets->SetItemText (cinema->first, std_to_wx(d->name()));
 225                 Config::instance()->changed (Config::CINEMAS);
 226         }
 227
 228         d->Destroy ();
 229 }
 230
 231
 232 void
 233 ScreensPanel::remove_cinema_clicked ()
 234 {
 235         if (_selected_cinemas.size() == 1) {
 236                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the cinema '%s'?"), std_to_wx(_selected_cinemas.begin()->second->name)))) {
 237                         return;
 238                 }
 239         } else {
 240                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d cinemas?"), int(_selected_cinemas.size())))) {
 241                         return;
 242                 }
 243         }
 244
 245         for (auto const& i: _selected_cinemas) {
 246                 Config::instance()->remove_cinema (i.second);
 247                 _targets->DeleteItem (i.first);
 248         }
 249
 250         selection_changed ();
 251 }
 252
 253
 254 void
 255 ScreensPanel::add_screen_clicked ()
 256 {
 257         auto cinema = cinema_for_operation ();
 258         if (!cinema) {
 259                 return;
 260         }
 261
 262         auto d = new ScreenDialog (GetParent(), _("Add Screen"));
 263         if (d->ShowModal () != wxID_OK) {
 264                 d->Destroy ();
 265                 return;
 266         }
 267
 268         for (auto screen: cinema->second->screens()) {
 269                 if (screen->name == d->name()) {
 270                         error_dialog (
 271                                 GetParent(),
 272                                 wxString::Format (
 273                                         _("You cannot add a screen called '%s' as the cinema already has a screen with this name."),
 274                                         std_to_wx(d->name()).data()
 275                                         )
 276                                 );
 277                         return;
 278                 }
 279         }
 280
 281         auto screen = std::make_shared<Screen>(d->name(), d->notes(), d->recipient(), d->recipient_file(), d->trusted_devices());
 282         cinema->second->add_screen (screen);
 283         auto id = add_screen (cinema->second, screen);
 284         if (id) {
 285                 _targets->Expand (id.get ());
 286         }
 287
 288         Config::instance()->changed (Config::CINEMAS);
 289
 290         d->Destroy ();
 291 }
 292
 293
 294 void
 295 ScreensPanel::edit_screen_clicked ()
 296 {
 297         if (_selected_screens.size() != 1) {
 298                 return;
 299         }
 300
 301         auto edit_screen = *_selected_screens.begin();
 302
 303         auto d = new ScreenDialog (
 304                 GetParent(), _("Edit screen"),
 305                 edit_screen.second->name,
 306                 edit_screen.second->notes,
 307                 edit_screen.second->recipient,
 308                 edit_screen.second->recipient_file,
 309                 edit_screen.second->trusted_devices
 310                 );
 311
 312         if (d->ShowModal() != wxID_OK) {
 313                 d->Destroy ();
 314                 return;
 315         }
 316
 317         auto cinema = edit_screen.second->cinema;
 318         for (auto screen: cinema->screens()) {
 319                 if (screen != edit_screen.second && screen->name == d->name()) {
 320                         error_dialog (
 321                                 GetParent(),
 322                                 wxString::Format (
 323                                         _("You cannot change this screen's name to '%s' as the cinema already has a screen with this name."),
 324                                         std_to_wx(d->name()).data()
 325                                         )
 326                                 );
 327                         return;
 328                 }
 329         }
 330
 331         edit_screen.second->name = d->name ();
 332         edit_screen.second->notes = d->notes ();
 333         edit_screen.second->recipient = d->recipient ();
 334         edit_screen.second->recipient_file = d->recipient_file ();
 335         edit_screen.second->trusted_devices = d->trusted_devices ();
 336         _targets->SetItemText (edit_screen.first, std_to_wx(d->name()));
 337         Config::instance()->changed (Config::CINEMAS);
 338
 339         d->Destroy ();
 340 }
 341
 342
 343 void
 344 ScreensPanel::remove_screen_clicked ()
 345 {
 346         if (_selected_screens.size() == 1) {
 347                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the screen '%s'?"), std_to_wx(_selected_screens.begin()->second->name)))) {
 348                         return;
 349                 }
 350         } else {
 351                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d screens?"), int(_selected_screens.size())))) {
 352                         return;
 353                 }
 354         }
 355
 356         for (auto const& i: _selected_screens) {
 357                 auto j = _cinemas.begin ();
 358                 while (j != _cinemas.end ()) {
 359                         auto sc = j->second->screens ();
 360                         if (find (sc.begin(), sc.end(), i.second) != sc.end ()) {
 361                                 break;
 362                         }
 363
 364                         ++j;
 365                 }
 366
 367                 if (j == _cinemas.end()) {
 368                         continue;
 369                 }
 370
 371                 j->second->remove_screen (i.second);
 372                 _targets->DeleteItem (i.first);
 373         }
 374
 375         Config::instance()->changed (Config::CINEMAS);
 376 }
 377
 378
 379 vector<shared_ptr<Screen>>
 380 ScreensPanel::screens () const
 381 {
 382         vector<shared_ptr<Screen>> output;
 383
 384         for (auto item = _targets->GetFirstItem(); item.IsOk(); item = _targets->GetNextItem(item)) {
 385                 if (_targets->GetCheckedState(item) == wxCHK_CHECKED) {
 386                         if (auto screen = item_to_screen(item)) {
 387                                 output.push_back (screen);
 388                         }
 389                 }
 390         }
 391
 392         return output;
 393 }
 394
 395
 396 void
 397 ScreensPanel::selection_changed_shim (wxTreeListEvent &)
 398 {
 399         selection_changed ();
 400 }
 401
 402
 403 void
 404 ScreensPanel::selection_changed ()
 405 {
 406         if (_ignore_selection_change) {
 407                 return;
 408         }
 409
 410         wxTreeListItems selection;
 411         _targets->GetSelections (selection);
 412
 413         _selected_cinemas.clear ();
 414         _selected_screens.clear ();
 415
 416         for (size_t i = 0; i < selection.size(); ++i) {
 417                 if (auto cinema = item_to_cinema(selection[i])) {
 418                         _selected_cinemas.push_back(make_pair(selection[i], cinema));
 419                 }
 420                 if (auto screen = item_to_screen(selection[i])) {
 421                         _selected_screens.push_back(make_pair(selection[i], screen));
 422                 }
 423         }
 424
 425         setup_sensitivity ();
 426 }
 427
 428
 429 void
 430 ScreensPanel::add_cinemas ()
 431 {
 432         for (auto cinema: Config::instance()->cinemas()) {
 433                 add_cinema (cinema);
 434         }
 435 }
 436
 437
 438 void
 439 ScreensPanel::search_changed ()
 440 {
 441         _targets->DeleteAllItems ();
 442         _cinemas.clear ();
 443         _screens.clear ();
 444
 445         _item_to_cinema.clear ();
 446         _cinema_to_item.clear ();
 447         _item_to_screen.clear ();
 448         _screen_to_item.clear ();
 449
 450         add_cinemas ();
 451
 452         _ignore_selection_change = true;
 453
 454         for (auto const& selection: _selected_cinemas) {
 455                 /* The wxTreeListItems will now be different, so we must search by cinema */
 456                 if (auto item = cinema_to_item(selection.second)) {
 457                         _targets->Select (*item);
 458                 }
 459         }
 460
 461         for (auto const& selection: _selected_screens) {
 462                 /* Likewise by screen */
 463                 if (auto item = screen_to_item(selection.second)) {
 464                         _targets->Select (*item);
 465                 }
 466         }
 467
 468         _ignore_selection_change = false;
 469 }
 470
 471
 472 void
 473 ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
 474 {
 475         if (item_to_cinema(ev.GetItem())) {
 476                 /* Cinema: check/uncheck all children */
 477                 auto const checked = _targets->GetCheckedState(ev.GetItem());
 478                 for (auto child = _targets->GetFirstChild(ev.GetItem()); child.IsOk(); child = _targets->GetNextSibling(child)) {
 479                         _targets->CheckItem(child, checked);
 480                 }
 481         } else {
 482                 /* Screen: set cinema to checked/unchecked/3state */
 483                 auto parent = _targets->GetItemParent(ev.GetItem());
 484                 DCPOMATIC_ASSERT (parent.IsOk());
 485                 int checked = 0;
 486                 int unchecked = 0;
 487                 for (auto child = _targets->GetFirstChild(parent); child.IsOk(); child = _targets->GetNextSibling(child)) {
 488                         if (_targets->GetCheckedState(child) == wxCHK_CHECKED) {
 489                             ++checked;
 490                         } else {
 491                             ++unchecked;
 492                         }
 493                 }
 494                 if (checked == 0) {
 495                         _targets->CheckItem(parent, wxCHK_UNCHECKED);
 496                 } else if (unchecked == 0) {
 497                         _targets->CheckItem(parent, wxCHK_CHECKED);
 498                 } else {
 499                         _targets->CheckItem(parent, wxCHK_UNDETERMINED);
 500                 }
 501         }
 502
 503         ScreensChanged ();
 504 }
 505
 506
 507 shared_ptr<Cinema>
 508 ScreensPanel::item_to_cinema (wxTreeListItem item) const
 509 {
 510         auto iter = _item_to_cinema.find (item);
 511         if (iter == _item_to_cinema.end()) {
 512                 return {};
 513         }
 514
 515         return iter->second;
 516 }
 517
 518
 519 shared_ptr<Screen>
 520 ScreensPanel::item_to_screen (wxTreeListItem item) const
 521 {
 522         auto iter = _item_to_screen.find (item);
 523         if (iter == _item_to_screen.end()) {
 524                 return {};
 525         }
 526
 527         return iter->second;
 528 }
 529
 530
 531 optional<wxTreeListItem>
 532 ScreensPanel::cinema_to_item (shared_ptr<Cinema> cinema) const
 533 {
 534         auto iter = _cinema_to_item.find (cinema);
 535         if (iter == _cinema_to_item.end()) {
 536                 return {};
 537         }
 538
 539         return iter->second;
 540 }
 541
 542
 543 optional<wxTreeListItem>
 544 ScreensPanel::screen_to_item (shared_ptr<Screen> screen) const
 545 {
 546         auto iter = _screen_to_item.find (screen);
 547         if (iter == _screen_to_item.end()) {
 548                 return {};
 549         }
 550
 551         return iter->second;
 552 }
 553
 554
 555 ScreensPanel::Comparator::Comparator ()
 556 {
 557         UErrorCode status = U_ZERO_ERROR;
 558         _collator = ucol_open(nullptr, &status);
 559         if (_collator) {
 560                 ucol_setAttribute(_collator, UCOL_NORMALIZATION_MODE, UCOL_ON, &status);
 561                 ucol_setAttribute(_collator, UCOL_STRENGTH, UCOL_PRIMARY, &status);
 562                 ucol_setAttribute(_collator, UCOL_ALTERNATE_HANDLING, UCOL_SHIFTED, &status);
 563         }
 564 }
 565
 566 ScreensPanel::Comparator::~Comparator ()
 567 {
 568         if (_collator) {
 569                 ucol_close (_collator);
 570         }
 571 }
 572
 573 int
 574 ScreensPanel::Comparator::Compare (wxTreeListCtrl* tree_list, unsigned, wxTreeListItem a, wxTreeListItem b)
 575 {
 576         auto utf8_a = wx_to_std(tree_list->GetItemText(a));
 577         auto utf8_b = wx_to_std(tree_list->GetItemText(b));
 578         if (_collator) {
 579                 UErrorCode error = U_ZERO_ERROR;
 580                 boost::scoped_array<uint16_t> utf16_a(new uint16_t[utf8_a.size() + 1]);
 581                 u_strFromUTF8(reinterpret_cast<UChar*>(utf16_a.get()), utf8_a.size() + 1, nullptr, utf8_a.c_str(), -1, &error);
 582                 boost::scoped_array<uint16_t> utf16_b(new uint16_t[utf8_b.size() + 1]);
 583                 u_strFromUTF8(reinterpret_cast<UChar*>(utf16_b.get()), utf8_b.size() + 1, nullptr, utf8_b.c_str(), -1, &error);
 584                 return ucol_strcoll(_collator, reinterpret_cast<UChar*>(utf16_a.get()), -1, reinterpret_cast<UChar*>(utf16_b.get()), -1);
 585         } else {
 586                 return strcoll(utf8_a.c_str(), utf8_b.c_str());
 587         }
 588 }