Some more KDM hacking.

2 files changed, 46 insertions, 0 deletions

diff --git a/src/wx/kdm_dialog.cc b/src/wx/kdm_dialog.cc
index 34004902b..43d904d90 100644
--- a/src/wx/kdm_dialog.cc
+++ b/src/wx/kdm_dialog.cc
@@ -321,3 +321,44 @@ KDMDialog::remove_screen_clicked (wxCommandEvent &)
 
 	Config::instance()->write ();
 }
+
+list<shared_ptr<Screen> >
+KDMDialog::screens () const
+{
+	list<shared_ptr<Screen> > s;
+
+	list<pair<wxTreeItemId, shared_ptr<Cinema> > > cinemas = selected_cinemas ();
+	for (list<pair<wxTreeItemId, shared_ptr<Cinema> > >::iterator i = cinemas.begin(); i != cinemas.end(); ++i) {
+		for (list<Screen>::iterator j = i->second->screens.begin(); j != i->second->screens.end(); ++j) {
+			s.push_back (*j);
+		}
+	}
+
+	list<pair<wxTreeItemId, shared_ptr<Screen> > > screens = selected_screens ();
+	for (list<pair<wxTreeItemId, shared_ptr<Screen> > >::iterator i = screens.begin(); i != screens.end(); ++i) {
+		s.push_back (i->second);
+	}
+
+	s.sort ();
+	s.uniq ();
+
+	return s;
+}
+
+boost::locale::date_time
+KDMDialog::from () const
+{
+
+}
+
+boost::locale::date_time
+KDMDialog::until () const
+{
+
+}
+
+string
+KDMDialog::directory () const
+{
+	return wx_to_std (_folder->GetPath ());
+}
diff --git a/src/wx/kdm_dialog.h b/src/wx/kdm_dialog.h
index 99e260338..d161223a8 100644
--- a/src/wx/kdm_dialog.h
+++ b/src/wx/kdm_dialog.h
@@ -36,6 +36,11 @@ class KDMDialog : public wxDialog
 public:
 	KDMDialog (wxWindow *);
 
+	std::list<boost::shared_ptr<Screen> > screens () const;
+	boost::local::date_time from () const;
+	boost::local::date_time until () const;
+	std::string directory () const;
+
 private:
 	void add_cinema (boost::shared_ptr<Cinema>);
 	void add_screen (boost::shared_ptr<Cinema>, boost::shared_ptr<Screen>);