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
  31
  32 using std::cout;
  33 using std::make_pair;
  34 using std::make_shared;
  35 using std::map;
  36 using std::pair;
  37 using std::shared_ptr;
  38 using std::string;
  39 using std::vector;
  40 using boost::optional;
  41 using namespace dcpomatic;
  42
  43
  44 ScreensPanel::ScreensPanel (wxWindow* parent)
  45         : wxPanel (parent, wxID_ANY)
  46         , _ignore_selection_change (false)
  47 {
  48         auto sizer = new wxBoxSizer (wxVERTICAL);
  49
  50         _search = new wxSearchCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, search_ctrl_height()));
  51 #ifndef __WXGTK3__
  52         /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */
  53         _search->ShowCancelButton (true);
  54 #endif
  55         sizer->Add (_search, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP);
  56
  57         auto targets = new wxBoxSizer (wxHORIZONTAL);
  58         _targets = new TreeListCtrl (this);
  59         _targets->AppendColumn (wxT("foo"));
  60         targets->Add (_targets, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_GAP);
  61
  62         add_cinemas ();
  63
  64         auto target_buttons = new wxBoxSizer (wxVERTICAL);
  65
  66         _add_cinema = new Button (this, _("Add Cinema..."));
  67         target_buttons->Add (_add_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  68         _edit_cinema = new Button (this, _("Edit Cinema..."));
  69         target_buttons->Add (_edit_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  70         _remove_cinema = new Button (this, _("Remove Cinema"));
  71         target_buttons->Add (_remove_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  72         _add_screen = new Button (this, _("Add Screen..."));
  73         target_buttons->Add (_add_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  74         _edit_screen = new Button (this, _("Edit Screen..."));
  75         target_buttons->Add (_edit_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  76         _remove_screen = new Button (this, _("Remove Screen"));
  77         target_buttons->Add (_remove_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  78
  79         targets->Add (target_buttons, 0, 0);
  80
  81         sizer->Add (targets, 1, wxEXPAND);
  82
  83         _search->Bind        (wxEVT_TEXT, boost::bind (&ScreensPanel::search_changed, this));
  84         _targets->Bind       (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
  85         _targets->Bind       (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
  86
  87         _add_cinema->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_cinema_clicked, this));
  88         _edit_cinema->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_cinema_clicked, this));
  89         _remove_cinema->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_cinema_clicked, this));
  90
  91         _add_screen->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_screen_clicked, this));
  92         _edit_screen->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_screen_clicked, this));
  93         _remove_screen->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_screen_clicked, this));
  94
  95         SetSizer (sizer);
  96 }
  97
  98
  99 ScreensPanel::~ScreensPanel ()
 100 {
 101         _targets->Unbind (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
 102         _targets->Unbind (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
 103 }
 104
 105
 106 void
 107 ScreensPanel::setup_sensitivity ()
 108 {
 109         bool const sc = _selected_cinemas.size() == 1;
 110         bool const ss = _selected_screens.size() == 1;
 111
 112         _edit_cinema->Enable (sc || ss);
 113         _remove_cinema->Enable (_selected_cinemas.size() >= 1);
 114
 115         _add_screen->Enable (sc || ss);
 116         _edit_screen->Enable (ss);
 117         _remove_screen->Enable (_selected_screens.size() >= 1);
 118 }
 119
 120
 121 optional<wxTreeListItem>
 122 ScreensPanel::add_cinema (shared_ptr<Cinema> c)
 123 {
 124         auto search = wx_to_std (_search->GetValue ());
 125         transform (search.begin(), search.end(), search.begin(), ::tolower);
 126
 127         if (!search.empty ()) {
 128                 auto name = c->name;
 129                 transform (name.begin(), name.end(), name.begin(), ::tolower);
 130                 if (name.find (search) == string::npos) {
 131                         return {};
 132                 }
 133         }
 134
 135         auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(c->name));
 136
 137         _cinemas[id] = c;
 138
 139         for (auto i: c->screens()) {
 140                 add_screen (c, i);
 141         }
 142
 143         _targets->SetSortColumn (0);
 144
 145         return id;
 146 }
 147
 148
 149 optional<wxTreeListItem>
 150 ScreensPanel::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
 151 {
 152         auto i = _cinemas.begin();
 153         while (i != _cinemas.end() && i->second != c) {
 154                 ++i;
 155         }
 156
 157         if (i == _cinemas.end()) {
 158                 return {};
 159         }
 160
 161         _screens[_targets->AppendItem (i->first, std_to_wx (s->name))] = s;
 162         return i->first;
 163 }
 164
 165
 166 void
 167 ScreensPanel::add_cinema_clicked ()
 168 {
 169         auto d = new CinemaDialog (GetParent(), _("Add Cinema"));
 170         if (d->ShowModal () == wxID_OK) {
 171                 auto c = make_shared<Cinema>(d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute());
 172                 Config::instance()->add_cinema (c);
 173                 auto id = add_cinema (c);
 174                 if (id) {
 175                         _targets->UnselectAll ();
 176                         _targets->Select (*id);
 177                 }
 178         }
 179
 180         d->Destroy ();
 181 }
 182
 183
 184 optional<pair<wxTreeListItem, shared_ptr<Cinema>>>
 185 ScreensPanel::cinema_for_operation () const
 186 {
 187         if (_selected_cinemas.size() == 1) {
 188                 return make_pair(_selected_cinemas.begin()->first, _selected_cinemas.begin()->second);
 189         } else if (_selected_screens.size() == 1) {
 190                 return make_pair(_targets->GetItemParent(_selected_screens.begin()->first), _selected_screens.begin()->second->cinema);
 191         }
 192
 193         return {};
 194 }
 195
 196
 197 void
 198 ScreensPanel::edit_cinema_clicked ()
 199 {
 200         auto cinema = cinema_for_operation ();
 201         if (!cinema) {
 202                 return;
 203         }
 204
 205         auto d = new CinemaDialog (
 206                 GetParent(), _("Edit cinema"), cinema->second->name, cinema->second->emails, cinema->second->notes, cinema->second->utc_offset_hour(), cinema->second->utc_offset_minute()
 207                 );
 208
 209         if (d->ShowModal() == wxID_OK) {
 210                 cinema->second->name = d->name ();
 211                 cinema->second->emails = d->emails ();
 212                 cinema->second->notes = d->notes ();
 213                 cinema->second->set_utc_offset_hour (d->utc_offset_hour ());
 214                 cinema->second->set_utc_offset_minute (d->utc_offset_minute ());
 215                 _targets->SetItemText (cinema->first, std_to_wx(d->name()));
 216                 Config::instance()->changed (Config::CINEMAS);
 217         }
 218
 219         d->Destroy ();
 220 }
 221
 222
 223 void
 224 ScreensPanel::remove_cinema_clicked ()
 225 {
 226         if (_selected_cinemas.size() == 1) {
 227                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the cinema '%s'?"), std_to_wx(_selected_cinemas.begin()->second->name)))) {
 228                         return;
 229                 }
 230         } else {
 231                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d cinemas?"), int(_selected_cinemas.size())))) {
 232                         return;
 233                 }
 234         }
 235
 236         for (auto const& i: _selected_cinemas) {
 237                 Config::instance()->remove_cinema (i.second);
 238                 _targets->DeleteItem (i.first);
 239         }
 240
 241         selection_changed ();
 242 }
 243
 244
 245 void
 246 ScreensPanel::add_screen_clicked ()
 247 {
 248         auto cinema = cinema_for_operation ();
 249         if (!cinema) {
 250                 return;
 251         }
 252
 253         auto d = new ScreenDialog (GetParent(), _("Add Screen"));
 254         if (d->ShowModal () != wxID_OK) {
 255                 d->Destroy ();
 256                 return;
 257         }
 258
 259         for (auto i: cinema->second->screens()) {
 260                 if (i->name == d->name()) {
 261                         error_dialog (
 262                                 GetParent(),
 263                                 wxString::Format (
 264                                         _("You cannot add a screen called '%s' as the cinema already has a screen with this name."),
 265                                         std_to_wx(d->name()).data()
 266                                         )
 267                                 );
 268                         return;
 269                 }
 270         }
 271
 272         auto screen = std::make_shared<Screen>(d->name(), d->notes(), d->recipient(), d->recipient_file(), d->trusted_devices());
 273         cinema->second->add_screen (screen);
 274         auto id = add_screen (cinema->second, screen);
 275         if (id) {
 276                 _targets->Expand (id.get ());
 277         }
 278
 279         Config::instance()->changed (Config::CINEMAS);
 280
 281         d->Destroy ();
 282 }
 283
 284
 285 void
 286 ScreensPanel::edit_screen_clicked ()
 287 {
 288         if (_selected_screens.size() != 1) {
 289                 return;
 290         }
 291
 292         auto s = *_selected_screens.begin();
 293
 294         auto d = new ScreenDialog (GetParent(), _("Edit screen"), s.second->name, s.second->notes, s.second->recipient, s.second->recipient_file, s.second->trusted_devices);
 295         if (d->ShowModal() != wxID_OK) {
 296                 d->Destroy ();
 297                 return;
 298         }
 299
 300         auto c = s.second->cinema;
 301         for (auto i: c->screens ()) {
 302                 if (i != s.second && i->name == d->name()) {
 303                         error_dialog (
 304                                 GetParent(),
 305                                 wxString::Format (
 306                                         _("You cannot change this screen's name to '%s' as the cinema already has a screen with this name."),
 307                                         std_to_wx(d->name()).data()
 308                                         )
 309                                 );
 310                         return;
 311                 }
 312         }
 313
 314         s.second->name = d->name ();
 315         s.second->notes = d->notes ();
 316         s.second->recipient = d->recipient ();
 317         s.second->recipient_file = d->recipient_file ();
 318         s.second->trusted_devices = d->trusted_devices ();
 319         _targets->SetItemText (s.first, std_to_wx (d->name()));
 320         Config::instance()->changed (Config::CINEMAS);
 321
 322         d->Destroy ();
 323 }
 324
 325
 326 void
 327 ScreensPanel::remove_screen_clicked ()
 328 {
 329         if (_selected_screens.size() == 1) {
 330                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the screen '%s'?"), std_to_wx(_selected_screens.begin()->second->name)))) {
 331                         return;
 332                 }
 333         } else {
 334                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d screens?"), int(_selected_screens.size())))) {
 335                         return;
 336                 }
 337         }
 338
 339         for (auto const& i: _selected_screens) {
 340                 auto j = _cinemas.begin ();
 341                 while (j != _cinemas.end ()) {
 342                         auto sc = j->second->screens ();
 343                         if (find (sc.begin(), sc.end(), i.second) != sc.end ()) {
 344                                 break;
 345                         }
 346
 347                         ++j;
 348                 }
 349
 350                 if (j == _cinemas.end()) {
 351                         continue;
 352                 }
 353
 354                 j->second->remove_screen (i.second);
 355                 _targets->DeleteItem (i.first);
 356         }
 357
 358         Config::instance()->changed (Config::CINEMAS);
 359 }
 360
 361
 362 vector<shared_ptr<Screen>>
 363 ScreensPanel::screens () const
 364 {
 365         vector<shared_ptr<Screen>> output;
 366
 367         for (auto item = _targets->GetFirstItem(); item.IsOk(); item = _targets->GetNextItem(item)) {
 368                 if (_targets->GetCheckedState(item) == wxCHK_CHECKED) {
 369                         auto screen_iter = _screens.find(item);
 370                         if (screen_iter != _screens.end()) {
 371                                 output.push_back (screen_iter->second);
 372                         }
 373                 }
 374         }
 375
 376         return output;
 377 }
 378
 379
 380 void
 381 ScreensPanel::selection_changed_shim (wxTreeListEvent &)
 382 {
 383         selection_changed ();
 384 }
 385
 386
 387 void
 388 ScreensPanel::selection_changed ()
 389 {
 390         if (_ignore_selection_change) {
 391                 return;
 392         }
 393
 394         wxTreeListItems s;
 395         _targets->GetSelections (s);
 396
 397         _selected_cinemas.clear ();
 398         _selected_screens.clear ();
 399
 400         for (size_t i = 0; i < s.size(); ++i) {
 401                 auto j = _cinemas.find (s[i]);
 402                 if (j != _cinemas.end ()) {
 403                         _selected_cinemas[j->first] = j->second;
 404                 }
 405                 auto k = _screens.find (s[i]);
 406                 if (k != _screens.end ()) {
 407                         _selected_screens[k->first] = k->second;
 408                 }
 409         }
 410
 411         setup_sensitivity ();
 412 }
 413
 414
 415 void
 416 ScreensPanel::add_cinemas ()
 417 {
 418         for (auto i: Config::instance()->cinemas()) {
 419                 add_cinema (i);
 420         }
 421 }
 422
 423
 424 void
 425 ScreensPanel::search_changed ()
 426 {
 427         _targets->DeleteAllItems ();
 428         _cinemas.clear ();
 429         _screens.clear ();
 430
 431         add_cinemas ();
 432
 433         _ignore_selection_change = true;
 434
 435         for (auto const& i: _selected_cinemas) {
 436                 /* The wxTreeListItems will now be different, so we must search by cinema */
 437                 auto j = _cinemas.begin ();
 438                 while (j != _cinemas.end() && j->second != i.second) {
 439                         ++j;
 440                 }
 441
 442                 if (j != _cinemas.end()) {
 443                         _targets->Select (j->first);
 444                 }
 445         }
 446
 447         for (auto const& i: _selected_screens) {
 448                 auto j = _screens.begin ();
 449                 while (j != _screens.end() && j->second != i.second) {
 450                         ++j;
 451                 }
 452
 453                 if (j != _screens.end()) {
 454                         _targets->Select (j->first);
 455                 }
 456         }
 457
 458         _ignore_selection_change = false;
 459 }
 460
 461
 462 void
 463 ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
 464 {
 465         if (_cinemas.find(ev.GetItem()) != _cinemas.end()) {
 466                 /* Cinema: check/uncheck all children */
 467                 auto const checked = _targets->GetCheckedState(ev.GetItem());
 468                 for (auto child = _targets->GetFirstChild(ev.GetItem()); child.IsOk(); child = _targets->GetNextSibling(child)) {
 469                         _targets->CheckItem(child, checked);
 470                 }
 471         } else {
 472                 /* Screen: set cinema to checked/unchecked/3state */
 473                 auto parent = _targets->GetItemParent(ev.GetItem());
 474                 DCPOMATIC_ASSERT (parent.IsOk());
 475                 int checked = 0;
 476                 int unchecked = 0;
 477                 for (auto child = _targets->GetFirstChild(parent); child.IsOk(); child = _targets->GetNextSibling(child)) {
 478                         if (_targets->GetCheckedState(child) == wxCHK_CHECKED) {
 479                             ++checked;
 480                         } else {
 481                             ++unchecked;
 482                         }
 483                 }
 484                 if (checked == 0) {
 485                         _targets->CheckItem(parent, wxCHK_UNCHECKED);
 486                 } else if (unchecked == 0) {
 487                         _targets->CheckItem(parent, wxCHK_CHECKED);
 488                 } else {
 489                         _targets->CheckItem(parent, wxCHK_UNDETERMINED);
 490                 }
 491         }
 492
 493         ScreensChanged ();
 494 }
 495
 496
 497
 498 wxIMPLEMENT_DYNAMIC_CLASS (TreeListCtrl, wxTreeListCtrl);
 499
 500
 501 int
 502 TreeListCtrl::OnCompareItems (wxTreeListItem const& a, wxTreeListItem const& b)
 503 {
 504         return strcoll (wx_to_std(GetItemText(a)).c_str(), wx_to_std(GetItemText(b)).c_str());
 505 }
 506
 507