Merge 1.0 in.
[dcpomatic.git] / src / wx / kdm_dialog.cc
index 519317dcc9d07b804518585e5431822264471d88..a9f63cffce971e06dd65a8d1c9d6d2f0cbab2a29 100644 (file)
 #include <wx/datectrl.h>
 #include <wx/timectrl.h>
 #include "lib/cinema.h"
+#include "lib/config.h"
 #include "kdm_dialog.h"
-#include "new_cinema_dialog.h"
+#include "cinema_dialog.h"
+#include "screen_dialog.h"
 #include "wx_util.h"
 #ifdef __WXMSW__
 #include "dir_picker_ctrl.h"
 #include <wx/filepicker.h>
 #endif
 
-using namespace std;
+using std::string;
+using std::map;
+using std::list;
+using std::pair;
+using std::make_pair;
 using boost::shared_ptr;
 
 KDMDialog::KDMDialog (wxWindow* parent)
        : wxDialog (parent, wxID_ANY, _("Make KDMs"))
 {
        wxBoxSizer* vertical = new wxBoxSizer (wxVERTICAL);
-
-       add_label_to_sizer (vertical, this, "Make KDMs for");
-
        wxBoxSizer* targets = new wxBoxSizer (wxHORIZONTAL);
        
        _targets = new wxTreeCtrl (this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_HIDE_ROOT | wxTR_MULTIPLE | wxTR_HAS_BUTTONS);
        targets->Add (_targets, 1, wxEXPAND | wxALL, 6);
 
        _root = _targets->AddRoot ("Foo");
-       shared_ptr<Cinema> csy (new Cinema ("City Screen York"));
-       add_cinema (csy);
-       add_screen (csy, shared_ptr<Screen> (new Screen ("Screen 1")));
-       add_screen (csy, shared_ptr<Screen> (new Screen ("Screen 2")));
-       add_screen (csy, shared_ptr<Screen> (new Screen ("Screen 3")));
+
+       list<shared_ptr<Cinema> > c = Config::instance()->cinemas ();
+       for (list<shared_ptr<Cinema> >::iterator i = c.begin(); i != c.end(); ++i) {
+               add_cinema (*i);
+       }
+
        _targets->ExpandAll ();
 
        wxBoxSizer* target_buttons = new wxBoxSizer (wxVERTICAL);
 
-       _new_cinema = new wxButton (this, wxID_ANY, _("New Cinema..."));
-       target_buttons->Add (_new_cinema, 1, wxEXPAND | wxALL, 6);
-       _new_screen = new wxButton (this, wxID_ANY, _("New Screen..."));
-       target_buttons->Add (_new_screen, 1, wxEXPAND | wxALL, 6);
+       _add_cinema = new wxButton (this, wxID_ANY, _("Add Cinema..."));
+       target_buttons->Add (_add_cinema, 1, wxEXPAND, 6);
+       _edit_cinema = new wxButton (this, wxID_ANY, _("Edit Cinema..."));
+       target_buttons->Add (_edit_cinema, 1, wxEXPAND, 6);
+       _remove_cinema = new wxButton (this, wxID_ANY, _("Remove Cinema"));
+       target_buttons->Add (_remove_cinema, 1, wxEXPAND, 6);
+       
+       _add_screen = new wxButton (this, wxID_ANY, _("Add Screen..."));
+       target_buttons->Add (_add_screen, 1, wxEXPAND, 6);
+       _edit_screen = new wxButton (this, wxID_ANY, _("Edit Screen..."));
+       target_buttons->Add (_edit_screen, 1, wxEXPAND, 6);
+       _remove_screen = new wxButton (this, wxID_ANY, _("Remove Screen"));
+       target_buttons->Add (_remove_screen, 1, wxEXPAND, 6);
 
-       targets->Add (target_buttons, 0, wxEXPAND | wxALL, 6);
+       targets->Add (target_buttons, 0, 0, 6);
 
-       vertical->Add (targets, 0, wxEXPAND | wxALL, 6);
+       vertical->Add (targets, 1, wxEXPAND | wxALL, 6);
 
        wxFlexGridSizer* table = new wxFlexGridSizer (3, 2, 6);
-       add_label_to_sizer (table, this, "From");
+       add_label_to_sizer (table, this, "From", true);
        _from_date = new wxDatePickerCtrl (this, wxID_ANY);
        table->Add (_from_date, 1, wxEXPAND);
        _from_time = new wxTimePickerCtrl (this, wxID_ANY);
        table->Add (_from_time, 1, wxEXPAND);
        
-       add_label_to_sizer (table, this, "To");
-       _to_date = new wxDatePickerCtrl (this, wxID_ANY);
-       table->Add (_to_date, 1, wxEXPAND);
-       _to_time = new wxTimePickerCtrl (this, wxID_ANY);
-       table->Add (_to_time, 1, wxEXPAND);
+       add_label_to_sizer (table, this, "Until", true);
+       _until_date = new wxDatePickerCtrl (this, wxID_ANY);
+       table->Add (_until_date, 1, wxEXPAND);
+       _until_time = new wxTimePickerCtrl (this, wxID_ANY);
+       table->Add (_until_time, 1, wxEXPAND);
 
-       add_label_to_sizer (table, this, "Write to");
+       add_label_to_sizer (table, this, "Write to", true);
 
 #ifdef __WXMSW__
        _folder = new DirPickerCtrl (this);
@@ -87,44 +100,92 @@ KDMDialog::KDMDialog (wxWindow* parent)
 
        table->Add (_folder, 1, wxEXPAND);
        
-       vertical->Add (table, 1, wxEXPAND | wxALL, 6);
+       vertical->Add (table, 0, wxEXPAND | wxALL, 6);
 
-       wxSizer* buttons = CreateSeparatedButtonSizer (wxOK);
+       wxSizer* buttons = CreateSeparatedButtonSizer (wxOK | wxCANCEL);
        if (buttons) {
                vertical->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
        }
 
        _targets->Connect (wxID_ANY, wxEVT_COMMAND_TREE_SEL_CHANGED, wxCommandEventHandler (KDMDialog::targets_selection_changed), 0, this);
-       _new_cinema->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::new_cinema_clicked), 0, this);
-       _new_screen->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::new_screen_clicked), 0, this);
 
-       _new_screen->Enable (false);
+       _add_cinema->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::add_cinema_clicked), 0, this);
+       _edit_cinema->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::edit_cinema_clicked), 0, this);
+       _remove_cinema->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::remove_cinema_clicked), 0, this);
+
+       _add_screen->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::add_screen_clicked), 0, this);
+       _edit_screen->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::edit_screen_clicked), 0, this);
+       _remove_screen->Connect (wxID_ANY, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler (KDMDialog::remove_screen_clicked), 0, this);
+
+       setup_sensitivity ();
        
        SetSizer (vertical);
        vertical->Layout ();
        vertical->SetSizeHints (this);
 }
 
-void
-KDMDialog::targets_selection_changed (wxCommandEvent &)
+list<pair<wxTreeItemId, shared_ptr<Cinema> > >
+KDMDialog::selected_cinemas () const
 {
        wxArrayTreeItemIds s;
        _targets->GetSelections (s);
 
-       bool selected_cinema = false;
+       list<pair<wxTreeItemId, shared_ptr<Cinema> > > c;
        for (size_t i = 0; i < s.GetCount(); ++i) {
-               if (_cinemas.find (s[i]) != _cinemas.end()) {
-                       selected_cinema = true;
+               map<wxTreeItemId, shared_ptr<Cinema> >::const_iterator j = _cinemas.find (s[i]);
+               if (j != _cinemas.end ()) {
+                       c.push_back (make_pair (j->first, j->second));
                }
        }
+
+       return c;
+}
+
+list<pair<wxTreeItemId, shared_ptr<Screen> > >
+KDMDialog::selected_screens () const
+{
+       wxArrayTreeItemIds s;
+       _targets->GetSelections (s);
+
+       list<pair<wxTreeItemId, shared_ptr<Screen> > > c;
+       for (size_t i = 0; i < s.GetCount(); ++i) {
+               map<wxTreeItemId, shared_ptr<Screen> >::const_iterator j = _screens.find (s[i]);
+               if (j != _screens.end ()) {
+                       c.push_back (make_pair (j->first, j->second));
+               }
+       }
+
+       return c;
+}
+
+void
+KDMDialog::targets_selection_changed (wxCommandEvent &)
+{
+       setup_sensitivity ();
+}
+
+void
+KDMDialog::setup_sensitivity ()
+{
+       bool const sc = selected_cinemas().size() == 1;
+       bool const ss = selected_screens().size() == 1;
+       
+       _edit_cinema->Enable (sc);
+       _remove_cinema->Enable (sc);
        
-       _new_screen->Enable (selected_cinema);  
+       _add_screen->Enable (sc);
+       _edit_screen->Enable (ss);
+       _remove_screen->Enable (ss);
 }
 
 void
 KDMDialog::add_cinema (shared_ptr<Cinema> c)
 {
        _cinemas[_targets->AppendItem (_root, std_to_wx (c->name))] = c;
+
+       for (list<shared_ptr<Screen> >::iterator i = c->screens.begin(); i != c->screens.end(); ++i) {
+               add_screen (c, *i);
+       }
 }
 
 void
@@ -143,15 +204,170 @@ KDMDialog::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
 }
 
 void
-KDMDialog::new_cinema_clicked (wxCommandEvent &)
+KDMDialog::add_cinema_clicked (wxCommandEvent &)
+{
+       CinemaDialog* d = new CinemaDialog (this, "Add Cinema");
+       d->ShowModal ();
+
+       shared_ptr<Cinema> c (new Cinema (d->name(), d->email()));
+       Config::instance()->add_cinema (c);
+       add_cinema (c);
+
+       Config::instance()->write ();
+       
+       d->Destroy ();
+}
+
+void
+KDMDialog::edit_cinema_clicked (wxCommandEvent &)
+{
+       if (selected_cinemas().size() != 1) {
+               return;
+       }
+
+       pair<wxTreeItemId, shared_ptr<Cinema> > c = selected_cinemas().front();
+       
+       CinemaDialog* d = new CinemaDialog (this, "Edit cinema", c.second->name, c.second->email);
+       d->ShowModal ();
+
+       c.second->name = d->name ();
+       c.second->email = d->email ();
+       _targets->SetItemText (c.first, std_to_wx (d->name()));
+
+       Config::instance()->write ();
+
+       d->Destroy ();  
+}
+
+void
+KDMDialog::remove_cinema_clicked (wxCommandEvent &)
+{
+       if (selected_cinemas().size() != 1) {
+               return;
+       }
+
+       pair<wxTreeItemId, shared_ptr<Cinema> > c = selected_cinemas().front();
+
+       Config::instance()->remove_cinema (c.second);
+       _targets->Delete (c.first);
+
+       Config::instance()->write ();   
+}
+
+void
+KDMDialog::add_screen_clicked (wxCommandEvent &)
 {
-       NewCinemaDialog* d = new NewCinemaDialog (this);
+       if (selected_cinemas().size() != 1) {
+               return;
+       }
+
+       shared_ptr<Cinema> c = selected_cinemas().front().second;
+       
+       ScreenDialog* d = new ScreenDialog (this, "Add Screen");
        d->ShowModal ();
+
+       shared_ptr<Screen> s (new Screen (d->name(), d->certificate()));
+       c->screens.push_back (s);
+       add_screen (c, s);
+
+       Config::instance()->write ();
+
        d->Destroy ();
 }
 
 void
-KDMDialog::new_screen_clicked (wxCommandEvent &)
+KDMDialog::edit_screen_clicked (wxCommandEvent &)
+{
+       if (selected_screens().size() != 1) {
+               return;
+       }
+
+       pair<wxTreeItemId, shared_ptr<Screen> > s = selected_screens().front();
+       
+       ScreenDialog* d = new ScreenDialog (this, "Edit screen", s.second->name, s.second->certificate);
+       d->ShowModal ();
+
+       s.second->name = d->name ();
+       s.second->certificate = d->certificate ();
+       _targets->SetItemText (s.first, std_to_wx (d->name()));
+
+       Config::instance()->write ();
+
+       d->Destroy ();
+}
+
+void
+KDMDialog::remove_screen_clicked (wxCommandEvent &)
+{
+       if (selected_screens().size() != 1) {
+               return;
+       }
+
+       pair<wxTreeItemId, shared_ptr<Screen> > s = selected_screens().front();
+
+       map<wxTreeItemId, shared_ptr<Cinema> >::iterator i = _cinemas.begin ();
+       while (i != _cinemas.end() && find (i->second->screens.begin(), i->second->screens.end(), s.second) == i->second->screens.end()) {
+               ++i;
+       }
+
+       if (i == _cinemas.end()) {
+               return;
+       }
+
+       i->second->screens.remove (s.second);
+       _targets->Delete (s.first);
+
+       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<shared_ptr<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.unique ();
+
+       return s;
+}
+
+boost::posix_time::ptime
+KDMDialog::from () const
+{
+       return posix_time (_from_date, _from_time);
+}
+
+boost::posix_time::ptime
+KDMDialog::posix_time (wxDatePickerCtrl* date_picker, wxTimePickerCtrl* time_picker)
+{
+       wxDateTime const date = date_picker->GetValue ();
+       wxDateTime const time = time_picker->GetValue ();
+       return boost::posix_time::ptime (
+               boost::gregorian::date (date.GetYear(), date.GetMonth() + 1, date.GetDay()),
+               boost::posix_time::time_duration (time.GetHour(), time.GetMinute(), time.GetSecond())
+               );
+}
+
+boost::posix_time::ptime
+KDMDialog::until () const
+{
+       return posix_time (_until_date, _until_time);
+}
+
+string
+KDMDialog::directory () const
+{
+       return wx_to_std (_folder->GetPath ());
 }