src/wx/screens_panel.cc

   1 /*
   2     Copyright (C) 2015-2018 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 #include "screens_panel.h"
  22 #include "wx_util.h"
  23 #include "cinema_dialog.h"
  24 #include "screen_dialog.h"
  25 #include "dcpomatic_button.h"
  26 #include "lib/config.h"
  27 #include "lib/cinema.h"
  28 #include "lib/screen.h"
  29 #include <boost/foreach.hpp>
  30
  31 using std::list;
  32 using std::pair;
  33 using std::cout;
  34 using std::map;
  35 using std::string;
  36 using std::make_pair;
  37 using boost::shared_ptr;
  38 using boost::optional;
  39 using namespace dcpomatic;
  40
  41 ScreensPanel::ScreensPanel (wxWindow* parent)
  42         : wxPanel (parent, wxID_ANY)
  43         , _ignore_selection_change (false)
  44 {
  45         wxBoxSizer* sizer = new wxBoxSizer (wxVERTICAL);
  46
  47 #ifdef __WXGTK3__
  48         int const height = 30;
  49 #else
  50         int const height = -1;
  51 #endif
  52
  53         _search = new wxSearchCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, height));
  54 #ifndef __WXGTK3__
  55         /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */
  56         _search->ShowCancelButton (true);
  57 #endif
  58         sizer->Add (_search, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP);
  59
  60         wxBoxSizer* targets = new wxBoxSizer (wxHORIZONTAL);
  61         _targets = new wxTreeCtrl (this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_HIDE_ROOT | wxTR_MULTIPLE | wxTR_HAS_BUTTONS | wxTR_LINES_AT_ROOT);
  62         targets->Add (_targets, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_GAP);
  63
  64         add_cinemas ();
  65
  66         wxBoxSizer* target_buttons = new wxBoxSizer (wxVERTICAL);
  67
  68         _add_cinema = new Button (this, _("Add Cinema..."));
  69         target_buttons->Add (_add_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  70         _edit_cinema = new Button (this, _("Edit Cinema..."));
  71         target_buttons->Add (_edit_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  72         _remove_cinema = new Button (this, _("Remove Cinema"));
  73         target_buttons->Add (_remove_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  74         _add_screen = new Button (this, _("Add Screen..."));
  75         target_buttons->Add (_add_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  76         _edit_screen = new Button (this, _("Edit Screen..."));
  77         target_buttons->Add (_edit_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  78         _remove_screen = new Button (this, _("Remove Screen"));
  79         target_buttons->Add (_remove_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
  80
  81         targets->Add (target_buttons, 0, 0);
  82
  83         sizer->Add (targets, 1, wxEXPAND);
  84
  85         _search->Bind        (wxEVT_TEXT, boost::bind (&ScreensPanel::search_changed, this));
  86         _targets->Bind       (wxEVT_TREE_SEL_CHANGED, &ScreensPanel::selection_changed_shim, this);
  87
  88         _add_cinema->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_cinema_clicked, this));
  89         _edit_cinema->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_cinema_clicked, this));
  90         _remove_cinema->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_cinema_clicked, this));
  91
  92         _add_screen->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_screen_clicked, this));
  93         _edit_screen->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_screen_clicked, this));
  94         _remove_screen->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_screen_clicked, this));
  95
  96         SetSizer (sizer);
  97 }
  98
  99 ScreensPanel::~ScreensPanel ()
 100 {
 101         _targets->Unbind (wxEVT_TREE_SEL_CHANGED, &ScreensPanel::selection_changed_shim, this);
 102 }
 103
 104 void
 105 ScreensPanel::setup_sensitivity ()
 106 {
 107         bool const sc = _selected_cinemas.size() == 1;
 108         bool const ss = _selected_screens.size() == 1;
 109
 110         _edit_cinema->Enable (sc);
 111         _remove_cinema->Enable (_selected_cinemas.size() >= 1);
 112
 113         _add_screen->Enable (sc);
 114         _edit_screen->Enable (ss);
 115         _remove_screen->Enable (_selected_screens.size() >= 1);
 116 }
 117
 118
 119 optional<wxTreeItemId>
 120 ScreensPanel::add_cinema (shared_ptr<Cinema> c)
 121 {
 122         string search = wx_to_std (_search->GetValue ());
 123         transform (search.begin(), search.end(), search.begin(), ::tolower);
 124
 125         if (!search.empty ()) {
 126                 string name = c->name;
 127                 transform (name.begin(), name.end(), name.begin(), ::tolower);
 128                 if (name.find (search) == string::npos) {
 129                         return optional<wxTreeItemId>();
 130                 }
 131         }
 132
 133         wxTreeItemId id = _targets->AppendItem(_root, std_to_wx(c->name));
 134
 135         _cinemas[id] = c;
 136
 137         BOOST_FOREACH (shared_ptr<Screen> i, c->screens()) {
 138                 add_screen (c, i);
 139         }
 140
 141         _targets->SortChildren (_root);
 142
 143         return id;
 144 }
 145
 146
 147 optional<wxTreeItemId>
 148 ScreensPanel::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
 149 {
 150         CinemaMap::const_iterator i = _cinemas.begin();
 151         while (i != _cinemas.end() && i->second != c) {
 152                 ++i;
 153         }
 154
 155         if (i == _cinemas.end()) {
 156                 return optional<wxTreeItemId> ();
 157         }
 158
 159         _screens[_targets->AppendItem (i->first, std_to_wx (s->name))] = s;
 160         return i->first;
 161 }
 162
 163 void
 164 ScreensPanel::add_cinema_clicked ()
 165 {
 166         CinemaDialog* d = new CinemaDialog (GetParent(), _("Add Cinema"));
 167         if (d->ShowModal () == wxID_OK) {
 168                 shared_ptr<Cinema> c (new Cinema (d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute()));
 169                 Config::instance()->add_cinema (c);
 170                 optional<wxTreeItemId> id = add_cinema (c);
 171                 if (id) {
 172                         _targets->Unselect ();
 173                         _targets->SelectItem (*id);
 174                 }
 175         }
 176
 177         d->Destroy ();
 178 }
 179
 180 void
 181 ScreensPanel::edit_cinema_clicked ()
 182 {
 183         if (_selected_cinemas.size() != 1) {
 184                 return;
 185         }
 186
 187         pair<wxTreeItemId, shared_ptr<Cinema> > c = *_selected_cinemas.begin();
 188
 189         CinemaDialog* d = new CinemaDialog (
 190                 GetParent(), _("Edit cinema"), c.second->name, c.second->emails, c.second->notes, c.second->utc_offset_hour(), c.second->utc_offset_minute()
 191                 );
 192
 193         if (d->ShowModal () == wxID_OK) {
 194                 c.second->name = d->name ();
 195                 c.second->emails = d->emails ();
 196                 c.second->notes = d->notes ();
 197                 c.second->set_utc_offset_hour (d->utc_offset_hour ());
 198                 c.second->set_utc_offset_minute (d->utc_offset_minute ());
 199                 _targets->SetItemText (c.first, std_to_wx (d->name()));
 200                 Config::instance()->changed (Config::CINEMAS);
 201         }
 202
 203         d->Destroy ();
 204 }
 205
 206 void
 207 ScreensPanel::remove_cinema_clicked ()
 208 {
 209         for (CinemaMap::iterator i = _selected_cinemas.begin(); i != _selected_cinemas.end(); ++i) {
 210                 Config::instance()->remove_cinema (i->second);
 211                 _targets->Delete (i->first);
 212         }
 213
 214         selection_changed ();
 215 }
 216
 217 void
 218 ScreensPanel::add_screen_clicked ()
 219 {
 220         if (_selected_cinemas.size() != 1) {
 221                 return;
 222         }
 223
 224         shared_ptr<Cinema> c = _selected_cinemas.begin()->second;
 225
 226         ScreenDialog* d = new ScreenDialog (GetParent(), _("Add Screen"));
 227         if (d->ShowModal () != wxID_OK) {
 228                 d->Destroy ();
 229                 return;
 230         }
 231
 232         BOOST_FOREACH (shared_ptr<Screen> i, c->screens ()) {
 233                 if (i->name == d->name()) {
 234                         error_dialog (
 235                                 GetParent(),
 236                                 wxString::Format (
 237                                         _("You cannot add a screen called '%s' as the cinema already has a screen with this name."),
 238                                         std_to_wx(d->name()).data()
 239                                         )
 240                                 );
 241                         return;
 242                 }
 243         }
 244
 245         shared_ptr<Screen> s (new Screen (d->name(), d->notes(), d->recipient(), d->trusted_devices()));
 246         c->add_screen (s);
 247         optional<wxTreeItemId> id = add_screen (c, s);
 248         if (id) {
 249                 _targets->Expand (id.get ());
 250         }
 251
 252         Config::instance()->changed (Config::CINEMAS);
 253
 254         d->Destroy ();
 255 }
 256
 257 void
 258 ScreensPanel::edit_screen_clicked ()
 259 {
 260         if (_selected_screens.size() != 1) {
 261                 return;
 262         }
 263
 264         pair<wxTreeItemId, shared_ptr<Screen> > s = *_selected_screens.begin();
 265
 266         ScreenDialog* d = new ScreenDialog (GetParent(), _("Edit screen"), s.second->name, s.second->notes, s.second->recipient, s.second->trusted_devices);
 267         if (d->ShowModal () != wxID_OK) {
 268                 d->Destroy ();
 269                 return;
 270         }
 271
 272         shared_ptr<Cinema> c = s.second->cinema;
 273         BOOST_FOREACH (shared_ptr<Screen> i, c->screens ()) {
 274                 if (i != s.second && i->name == d->name()) {
 275                         error_dialog (
 276                                 GetParent(),
 277                                 wxString::Format (
 278                                         _("You cannot change this screen's name to '%s' as the cinema already has a screen with this name."),
 279                                         std_to_wx(d->name()).data()
 280                                         )
 281                                 );
 282                         return;
 283                 }
 284         }
 285
 286         s.second->name = d->name ();
 287         s.second->notes = d->notes ();
 288         s.second->recipient = d->recipient ();
 289         s.second->trusted_devices = d->trusted_devices ();
 290         _targets->SetItemText (s.first, std_to_wx (d->name()));
 291         Config::instance()->changed (Config::CINEMAS);
 292
 293         d->Destroy ();
 294 }
 295
 296 void
 297 ScreensPanel::remove_screen_clicked ()
 298 {
 299         for (ScreenMap::iterator i = _selected_screens.begin(); i != _selected_screens.end(); ++i) {
 300                 CinemaMap::iterator j = _cinemas.begin ();
 301                 while (j != _cinemas.end ()) {
 302                         list<shared_ptr<Screen> > sc = j->second->screens ();
 303                         if (find (sc.begin(), sc.end(), i->second) != sc.end ()) {
 304                                 break;
 305                         }
 306
 307                         ++j;
 308                 }
 309
 310                 if (j == _cinemas.end()) {
 311                         continue;
 312                 }
 313
 314                 j->second->remove_screen (i->second);
 315                 _targets->Delete (i->first);
 316         }
 317
 318         Config::instance()->changed (Config::CINEMAS);
 319 }
 320
 321 list<shared_ptr<Screen> >
 322 ScreensPanel::screens () const
 323 {
 324         list<shared_ptr<Screen> > s;
 325
 326         for (CinemaMap::const_iterator i = _selected_cinemas.begin(); i != _selected_cinemas.end(); ++i) {
 327                 list<shared_ptr<Screen> > sc = i->second->screens ();
 328                 for (list<shared_ptr<Screen> >::const_iterator j = sc.begin(); j != sc.end(); ++j) {
 329                         s.push_back (*j);
 330                 }
 331         }
 332
 333         for (ScreenMap::const_iterator i = _selected_screens.begin(); i != _selected_screens.end(); ++i) {
 334                 s.push_back (i->second);
 335         }
 336
 337         s.sort ();
 338         s.unique ();
 339
 340         return s;
 341 }
 342
 343 void
 344 ScreensPanel::selection_changed_shim (wxTreeEvent &)
 345 {
 346         selection_changed ();
 347 }
 348
 349 void
 350 ScreensPanel::selection_changed ()
 351 {
 352         if (_ignore_selection_change) {
 353                 return;
 354         }
 355
 356         wxArrayTreeItemIds s;
 357         _targets->GetSelections (s);
 358
 359         _selected_cinemas.clear ();
 360         _selected_screens.clear ();
 361
 362         for (size_t i = 0; i < s.GetCount(); ++i) {
 363                 CinemaMap::const_iterator j = _cinemas.find (s[i]);
 364                 if (j != _cinemas.end ()) {
 365                         _selected_cinemas[j->first] = j->second;
 366                 }
 367                 ScreenMap::const_iterator k = _screens.find (s[i]);
 368                 if (k != _screens.end ()) {
 369                         _selected_screens[k->first] = k->second;
 370                 }
 371         }
 372
 373         setup_sensitivity ();
 374         ScreensChanged ();
 375 }
 376
 377 void
 378 ScreensPanel::add_cinemas ()
 379 {
 380         _root = _targets->AddRoot ("Foo");
 381
 382         BOOST_FOREACH (shared_ptr<Cinema> i, Config::instance()->cinemas ()) {
 383                 add_cinema (i);
 384         }
 385 }
 386
 387 void
 388 ScreensPanel::search_changed ()
 389 {
 390         _targets->DeleteAllItems ();
 391         _cinemas.clear ();
 392         _screens.clear ();
 393
 394         add_cinemas ();
 395
 396         _ignore_selection_change = true;
 397
 398         for (CinemaMap::const_iterator i = _selected_cinemas.begin(); i != _selected_cinemas.end(); ++i) {
 399                 /* The wxTreeItemIds will now be different, so we must search by cinema */
 400                 CinemaMap::const_iterator j = _cinemas.begin ();
 401                 while (j != _cinemas.end() && j->second != i->second) {
 402                         ++j;
 403                 }
 404
 405                 if (j != _cinemas.end ()) {
 406                         _targets->SelectItem (j->first);
 407                 }
 408         }
 409
 410         for (ScreenMap::const_iterator i = _selected_screens.begin(); i != _selected_screens.end(); ++i) {
 411                 ScreenMap::const_iterator j = _screens.begin ();
 412                 while (j != _screens.end() && j->second != i->second) {
 413                         ++j;
 414                 }
 415
 416                 if (j != _screens.end ()) {
 417                         _targets->SelectItem (j->first);
 418                 }
 419         }
 420
 421         _ignore_selection_change = false;
 422 }