X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Fscreens_panel.h;h=7c72f587362eb0544e269aa87f1eb45518ccab4c;hb=b16b0da184b349c9565a9e556b22164a2a158970;hp=bea9c2995672526ec59a2ca0edc17cea4a404b3a;hpb=943047c84a0cdc7fbcec392f512157d637b8ec71;p=dcpomatic.git

diff --git a/src/wx/screens_panel.h b/src/wx/screens_panel.h
index bea9c2995..7c72f5873 100644
--- a/src/wx/screens_panel.h
+++ b/src/wx/screens_panel.h
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2015-2016 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2015-2022 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -18,15 +18,18 @@
 
 */
 
-#include "lib/warnings.h"
-DCPOMATIC_DISABLE_WARNINGS
-#include <wx/wx.h>
-DCPOMATIC_ENABLE_WARNINGS
+
+#include <dcp/warnings.h>
+LIBDCP_DISABLE_WARNINGS
 #include <wx/srchctrl.h>
 #include <wx/treelist.h>
+#include <wx/wx.h>
+LIBDCP_ENABLE_WARNINGS
 #include <boost/signals2.hpp>
 #include <list>
 #include <map>
+#include <set>
+
 
 namespace dcpomatic {
 	class Screen;
@@ -34,25 +37,7 @@ namespace dcpomatic {
 
 
 class Cinema;
-
-
-/** Shim around wxTreeListCtrl so we can use strcoll() to compare things */
-class TreeListCtrl : public wxTreeListCtrl
-{
-public:
-	wxDECLARE_DYNAMIC_CLASS (TreeListCtrl);
-
-	TreeListCtrl () {}
-
-	TreeListCtrl (wxWindow* parent)
-		: wxTreeListCtrl (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTL_MULTIPLE | wxTL_3STATE | wxTL_NO_HEADER)
-	{}
-
-	virtual ~TreeListCtrl () {}
-
-private:
-	int OnCompareItems (wxTreeListItem const& a, wxTreeListItem const& b);
-};
+struct UCollator;
 
 
 class ScreensPanel : public wxPanel
@@ -68,7 +53,7 @@ public:
 
 private:
 	void add_cinemas ();
-	boost::optional<wxTreeListItem> add_cinema (std::shared_ptr<Cinema>);
+	boost::optional<wxTreeListItem> add_cinema (std::shared_ptr<Cinema>, wxTreeListItem previous);
 	boost::optional<wxTreeListItem> add_screen (std::shared_ptr<Cinema>, std::shared_ptr<dcpomatic::Screen>);
 	void add_cinema_clicked ();
 	void edit_cinema_clicked ();
@@ -80,23 +65,48 @@ private:
 	void selection_changed ();
 	void search_changed ();
 	void checkbox_changed (wxTreeListEvent& ev);
+	std::shared_ptr<Cinema> cinema_for_operation () const;
+	void set_screen_checked (wxTreeListItem item, bool checked);
+	void setup_cinema_checked_state (wxTreeListItem screen);
+	int compare (std::string const& utf8_a, std::string const& utf8_b);
+	void check_all ();
+	void uncheck_all ();
+	bool notify_cinemas_changed();
+
+	std::shared_ptr<Cinema> item_to_cinema (wxTreeListItem item) const;
+	std::shared_ptr<dcpomatic::Screen> item_to_screen (wxTreeListItem item) const;
+	boost::optional<wxTreeListItem> cinema_to_item (std::shared_ptr<Cinema> cinema) const;
+	boost::optional<wxTreeListItem> screen_to_item (std::shared_ptr<dcpomatic::Screen> screen) const;
 
 	wxSearchCtrl* _search;
-	TreeListCtrl* _targets;
+	wxTreeListCtrl* _targets;
 	wxButton* _add_cinema;
 	wxButton* _edit_cinema;
 	wxButton* _remove_cinema;
 	wxButton* _add_screen;
 	wxButton* _edit_screen;
 	wxButton* _remove_screen;
-
-	typedef std::map<wxTreeListItem, std::shared_ptr<Cinema>> CinemaMap;
-	typedef std::map<wxTreeListItem, std::shared_ptr<dcpomatic::Screen>> ScreenMap;
-
-	CinemaMap _cinemas;
-	ScreenMap _screens;
-	CinemaMap _selected_cinemas;
-	ScreenMap _selected_screens;
-
-	bool _ignore_selection_change;
+	wxButton* _check_all;
+	wxButton* _uncheck_all;
+
+	/* We want to be able to search (and so remove selected things from the view)
+	 * but not deselect them, so we maintain lists of selected cinemas and screens.
+	 */
+	std::vector<std::shared_ptr<Cinema>> _selected_cinemas;
+	std::vector<std::shared_ptr<dcpomatic::Screen>> _selected_screens;
+	/* Likewise with checked screens, except that we can work out which cinemas
+	 * are checked from which screens are checked, so we don't need to store the
+	 * cinemas.
+	 */
+	std::set<std::shared_ptr<dcpomatic::Screen>> _checked_screens;
+
+	std::map<wxTreeListItem, std::shared_ptr<Cinema>> _item_to_cinema;
+	std::map<wxTreeListItem, std::shared_ptr<dcpomatic::Screen>> _item_to_screen;
+	std::map<std::shared_ptr<Cinema>, wxTreeListItem> _cinema_to_item;
+	std::map<std::shared_ptr<dcpomatic::Screen>, wxTreeListItem> _screen_to_item;
+
+	bool _ignore_selection_change = false;
+	bool _ignore_check_change = false;
+
+	UCollator* _collator = nullptr;
 };