Preserve checked status during search (#2212).
[dcpomatic.git] / src / wx / screens_panel.h
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 "lib/warnings.h"
23 DCPOMATIC_DISABLE_WARNINGS
24 #include <wx/wx.h>
25 DCPOMATIC_ENABLE_WARNINGS
26 #include <wx/srchctrl.h>
27 #include <wx/treelist.h>
28 #include <boost/signals2.hpp>
29 #include <list>
30 #include <map>
31
32
33 namespace dcpomatic {
34         class Screen;
35 }
36
37
38 class Cinema;
39 class UCollator;
40
41
42 class ScreensPanel : public wxPanel
43 {
44 public:
45         explicit ScreensPanel (wxWindow* parent);
46         ~ScreensPanel ();
47
48         std::vector<std::shared_ptr<dcpomatic::Screen>> screens () const;
49         void setup_sensitivity ();
50
51         boost::signals2::signal<void ()> ScreensChanged;
52
53 private:
54         void add_cinemas ();
55         boost::optional<wxTreeListItem> add_cinema (std::shared_ptr<Cinema>);
56         boost::optional<wxTreeListItem> add_screen (std::shared_ptr<Cinema>, std::shared_ptr<dcpomatic::Screen>);
57         void add_cinema_clicked ();
58         void edit_cinema_clicked ();
59         void remove_cinema_clicked ();
60         void add_screen_clicked ();
61         void edit_screen_clicked ();
62         void remove_screen_clicked ();
63         void selection_changed_shim (wxTreeListEvent &);
64         void selection_changed ();
65         void search_changed ();
66         void checkbox_changed (wxTreeListEvent& ev);
67         boost::optional<std::pair<wxTreeListItem, std::shared_ptr<Cinema>>> cinema_for_operation () const;
68         void set_screen_checked (wxTreeListItem item, bool checked);
69         void setup_cinema_checked_state (wxTreeListItem screen);
70
71         typedef std::vector<std::pair<wxTreeListItem, std::shared_ptr<Cinema>>> Cinemas;
72         typedef std::vector<std::pair<wxTreeListItem, std::shared_ptr<dcpomatic::Screen>>> Screens;
73
74         std::shared_ptr<Cinema> item_to_cinema (wxTreeListItem item) const;
75         std::shared_ptr<dcpomatic::Screen> item_to_screen (wxTreeListItem item) const;
76         boost::optional<wxTreeListItem> cinema_to_item (std::shared_ptr<Cinema> cinema) const;
77         boost::optional<wxTreeListItem> screen_to_item (std::shared_ptr<dcpomatic::Screen> screen) const;
78
79         wxSearchCtrl* _search;
80         wxTreeListCtrl* _targets;
81         wxButton* _add_cinema;
82         wxButton* _edit_cinema;
83         wxButton* _remove_cinema;
84         wxButton* _add_screen;
85         wxButton* _edit_screen;
86         wxButton* _remove_screen;
87
88         Cinemas _cinemas;
89         Screens _screens;
90         /* We want to be able to search (and so remove selected things from the view)
91          * but not deselect them, so we maintain lists of selected cinemas and screens.
92          */
93         Cinemas _selected_cinemas;
94         Screens _selected_screens;
95         /* Likewise with checked screens, except that we can work out which cinemas
96          * are checked from which screens are checked, so we don't need to store the
97          * cinemas.
98          */
99         Screens _checked_screens;
100
101         std::map<wxTreeListItem, std::shared_ptr<Cinema>> _item_to_cinema;
102         std::map<wxTreeListItem, std::shared_ptr<dcpomatic::Screen>> _item_to_screen;
103         std::map<std::shared_ptr<Cinema>, wxTreeListItem> _cinema_to_item;
104         std::map<std::shared_ptr<dcpomatic::Screen>, wxTreeListItem> _screen_to_item;
105
106         bool _ignore_selection_change = false;
107         bool _ignore_check_change = false;
108
109         class Comparator : public wxTreeListItemComparator
110         {
111         public:
112                 Comparator ();
113                 ~Comparator ();
114
115                 int Compare (wxTreeListCtrl* tree_list, unsigned, wxTreeListItem a, wxTreeListItem b) override;
116
117         private:
118                 UCollator* _collator;
119         };
120
121         Comparator _comparator;
122 };