Rename Subtitle -> Text
[dcpomatic.git] / src / wx / subtitle_appearance_dialog.cc
index 0db55edfe1ce110453c728e12176abfe7e83d60d..f78ed9ab5ed4b585640e9e9cbd849661bb242f0e 100644 (file)
 /*
-    Copyright (C) 2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2015-2018 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
 #include "subtitle_appearance_dialog.h"
-#include "lib/subrip_content.h"
+#include "rgba_colour_picker.h"
+#include "lib/text_text_content.h"
+#include "lib/text_content.h"
+#include "lib/ffmpeg_subtitle_stream.h"
+#include "lib/ffmpeg_content.h"
 #include <wx/wx.h>
 #include <wx/clrpicker.h>
+#include <wx/spinctrl.h>
+#include <wx/gbsizer.h>
 
+using std::map;
 using boost::shared_ptr;
+using boost::bind;
+using boost::dynamic_pointer_cast;
+using boost::optional;
+
+int const SubtitleAppearanceDialog::NONE = 0;
+int const SubtitleAppearanceDialog::OUTLINE = 1;
+int const SubtitleAppearanceDialog::SHADOW = 2;
 
-SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr<SubRipContent> content)
-       : TableDialog (parent, _("Subtitle appearance"), 2, 1, true)
+SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr<Content> content)
+       : wxDialog (parent, wxID_ANY, _("Subtitle appearance"))
        , _content (content)
 {
-       add (_("Colour"), true);
-       _colour = new wxColourPickerCtrl (this, wxID_ANY);
-       add (_colour);
+       shared_ptr<FFmpegContent> ff = dynamic_pointer_cast<FFmpegContent> (content);
+       if (ff) {
+               _stream = ff->subtitle_stream ();
+       }
+
+       wxSizer* overall_sizer = new wxBoxSizer (wxVERTICAL);
+       SetSizer (overall_sizer);
+
+       _table = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
+
+       overall_sizer->Add (_table, 1, wxEXPAND | wxALL, DCPOMATIC_DIALOG_BORDER);
+
+       int r = 0;
+
+       add_label_to_sizer (_table, this, _("Colour"), true, wxGBPosition (r, 0));
+       _force_colour = set_to (_colour = new wxColourPickerCtrl (this, wxID_ANY), r);
+
+       add_label_to_sizer (_table, this, _("Effect"), true, wxGBPosition (r, 0));
+       _force_effect = set_to (_effect = new wxChoice (this, wxID_ANY), r);
+
+       add_label_to_sizer (_table, this, _("Effect colour"), true, wxGBPosition (r, 0));
+       _force_effect_colour = set_to (_effect_colour = new wxColourPickerCtrl (this, wxID_ANY), r);
+
+       add_label_to_sizer (_table, this, _("Outline width"), true, wxGBPosition (r, 0));
+       _outline_width = new wxSpinCtrl (this, wxID_ANY);
+       _table->Add (_outline_width, wxGBPosition (r, 1));
+       ++r;
+
+       add_label_to_sizer (_table, this, _("Fade in time"), true, wxGBPosition (r, 0));
+       _force_fade_in = set_to (_fade_in = new Timecode<ContentTime> (this), r);
+
+       add_label_to_sizer (_table, this, _("Fade out time"), true, wxGBPosition (r, 0));
+       _force_fade_out = set_to (_fade_out = new Timecode<ContentTime> (this), r);
+
+       if (_stream) {
+               wxScrolled<wxPanel>* colours_panel = new wxScrolled<wxPanel> (this);
+               colours_panel->EnableScrolling (false, true);
+               colours_panel->ShowScrollbars (wxSHOW_SB_NEVER, wxSHOW_SB_ALWAYS);
+               colours_panel->SetScrollRate (0, 16);
+
+               wxFlexGridSizer* table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
+               table->AddGrowableCol (1, 1);
+
+               map<RGBA, RGBA> colours = _stream->colours ();
+
+               wxStaticText* t = new wxStaticText (colours_panel, wxID_ANY, "");
+               t->SetLabelMarkup (_("<b>Original colour</b>"));
+               table->Add (t, 1, wxEXPAND);
+               t = new wxStaticText (colours_panel, wxID_ANY, "", wxDefaultPosition, wxDefaultSize, wxALIGN_CENTRE_HORIZONTAL);
+               t->SetLabelMarkup (_("<b>New colour</b>"));
+               table->Add (t, 1, wxALIGN_CENTER);
+
+               for (map<RGBA, RGBA>::const_iterator i = colours.begin(); i != colours.end(); ++i) {
+                       wxPanel* from = new wxPanel (colours_panel, wxID_ANY);
+                       from->SetBackgroundColour (wxColour (i->first.r, i->first.g, i->first.b, i->first.a));
+                       table->Add (from, 1, wxEXPAND);
+                       RGBAColourPicker* to = new RGBAColourPicker (colours_panel, i->second);
+                       table->Add (to, 1, wxEXPAND);
+                       _pickers[i->first] = to;
+               }
+
+               colours_panel->SetSizer (table);
+
+               overall_sizer->Add (colours_panel, 1, wxEXPAND | wxALL, DCPOMATIC_DIALOG_BORDER);
+
+               wxButton* restore = new wxButton (this, wxID_ANY, _("Restore to original colours"));
+               restore->Bind (wxEVT_BUTTON, bind (&SubtitleAppearanceDialog::restore, this));
+               overall_sizer->Add (restore, 0, wxALL, DCPOMATIC_SIZER_X_GAP);
+       }
+
+       wxSizer* buttons = CreateSeparatedButtonSizer (wxOK);
+       if (buttons) {
+               overall_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
+       }
 
-       _outline = new wxCheckBox (this, wxID_ANY, _("Outline"));
-       add (_outline);
-       add_spacer ();
+       overall_sizer->Layout ();
+       overall_sizer->SetSizeHints (this);
 
-       add (_("Outline colour"), true);
-       _outline_colour = new wxColourPickerCtrl (this, wxID_ANY);
-       add (_outline_colour);
+       /* Keep these Appends() up to date with NONE/OUTLINE/SHADOW variables */
+       _effect->Append (_("None"));
+       _effect->Append (_("Outline"));
+       _effect->Append (_("Shadow"));;
 
-       layout ();
+       optional<dcp::Colour> colour = _content->subtitle->colour();
+       _force_colour->SetValue (static_cast<bool>(colour));
+       if (colour) {
+               _colour->SetColour (wxColour (colour->r, colour->g, colour->b));
+       } else {
+               _colour->SetColour (wxColour (255, 255, 255));
+       }
 
-       _colour->SetColour (wxColour (_content->colour().r, _content->colour().g, _content->colour().b));
-       _outline->SetValue (_content->outline ());
-       _outline_colour->SetColour (wxColour (_content->outline_colour().r, _content->outline_colour().g, _content->outline_colour().b));
+       optional<dcp::Effect> effect = _content->subtitle->effect();
+       _force_effect->SetValue (static_cast<bool>(effect));
+       if (effect) {
+               switch (*effect) {
+               case dcp::NONE:
+                       _effect->SetSelection (NONE);
+                       break;
+               case dcp::BORDER:
+                       _effect->SetSelection (OUTLINE);
+                       break;
+               case dcp::SHADOW:
+                       _effect->SetSelection (SHADOW);
+                       break;
+               }
+       } else {
+               _effect->SetSelection (NONE);
+       }
+
+       optional<dcp::Colour> effect_colour = _content->subtitle->effect_colour();
+       _force_effect_colour->SetValue (static_cast<bool>(effect_colour));
+       if (effect_colour) {
+               _effect_colour->SetColour (wxColour (effect_colour->r, effect_colour->g, effect_colour->b));
+       } else {
+               _effect_colour->SetColour (wxColour (0, 0, 0));
+       }
+
+       optional<ContentTime> fade_in = _content->subtitle->fade_in();
+       _force_fade_in->SetValue (static_cast<bool>(fade_in));
+       if (fade_in) {
+               _fade_in->set (*fade_in, _content->active_video_frame_rate());
+       } else {
+               _fade_in->set (ContentTime(), _content->active_video_frame_rate());
+       }
+
+       optional<ContentTime> fade_out = _content->subtitle->fade_out();
+       _force_fade_out->SetValue (static_cast<bool>(fade_out));
+       if (fade_out) {
+               _fade_out->set (*fade_out, _content->active_video_frame_rate ());
+       } else {
+               _fade_out->set (ContentTime(), _content->active_video_frame_rate ());
+       }
+
+       _outline_width->SetValue (_content->subtitle->outline_width ());
+
+       _force_colour->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _force_effect_colour->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _force_effect->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _force_fade_in->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _force_fade_out->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _effect->Bind (wxEVT_CHOICE, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+       _content_connection = _content->Changed.connect (bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
+
+       setup_sensitivity ();
+}
+
+wxCheckBox*
+SubtitleAppearanceDialog::set_to (wxWindow* w, int& r)
+{
+       wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
+       wxCheckBox* set_to = new wxCheckBox (this, wxID_ANY, _("Set to"));
+       s->Add (set_to, 0, wxRIGHT | wxALIGN_CENTER_VERTICAL, 8);
+       s->Add (w, 0, wxALIGN_CENTER_VERTICAL);
+       _table->Add (s, wxGBPosition (r, 1));
+       ++r;
+       return set_to;
 }
 
 void
 SubtitleAppearanceDialog::apply ()
 {
-       wxColour const c = _colour->GetColour ();
-       _content->set_colour (dcp::Colour (c.Red(), c.Green(), c.Blue()));
-       _content->set_outline (_outline->GetValue ());
-       wxColour const oc = _outline_colour->GetColour ();
-       _content->set_outline_colour (dcp::Colour (oc.Red(), oc.Green(), oc.Blue()));
+       if (_force_colour->GetValue ()) {
+               wxColour const c = _colour->GetColour ();
+               _content->subtitle->set_colour (dcp::Colour (c.Red(), c.Green(), c.Blue()));
+       } else {
+               _content->subtitle->unset_colour ();
+       }
+       if (_force_effect->GetValue()) {
+               switch (_effect->GetSelection()) {
+               case NONE:
+                       _content->subtitle->set_effect (dcp::NONE);
+                       break;
+               case OUTLINE:
+                       _content->subtitle->set_effect (dcp::BORDER);
+                       break;
+               case SHADOW:
+                       _content->subtitle->set_effect (dcp::SHADOW);
+                       break;
+               }
+       } else {
+               _content->subtitle->unset_effect ();
+       }
+       if (_force_effect_colour->GetValue ()) {
+               wxColour const ec = _effect_colour->GetColour ();
+               _content->subtitle->set_effect_colour (dcp::Colour (ec.Red(), ec.Green(), ec.Blue()));
+       } else {
+               _content->subtitle->unset_effect_colour ();
+       }
+       if (_force_fade_in->GetValue ()) {
+               _content->subtitle->set_fade_in (_fade_in->get (_content->active_video_frame_rate ()));
+       } else {
+               _content->subtitle->unset_fade_in ();
+       }
+       if (_force_fade_out->GetValue ()) {
+               _content->subtitle->set_fade_out (_fade_out->get (_content->active_video_frame_rate ()));
+       } else {
+               _content->subtitle->unset_fade_out ();
+       }
+       _content->subtitle->set_outline_width (_outline_width->GetValue ());
+
+       if (_stream) {
+               for (map<RGBA, RGBAColourPicker*>::const_iterator i = _pickers.begin(); i != _pickers.end(); ++i) {
+                       _stream->set_colour (i->first, i->second->colour ());
+               }
+       }
+
+       shared_ptr<FFmpegContent> fc = dynamic_pointer_cast<FFmpegContent> (_content);
+       if (fc) {
+               fc->signal_subtitle_stream_changed ();
+       }
+}
+
+void
+SubtitleAppearanceDialog::restore ()
+{
+       for (map<RGBA, RGBAColourPicker*>::const_iterator i = _pickers.begin(); i != _pickers.end(); ++i) {
+               i->second->set (i->first);
+       }
+}
+
+void
+SubtitleAppearanceDialog::setup_sensitivity ()
+{
+       _colour->Enable (_force_colour->GetValue ());
+       _effect_colour->Enable (_force_effect_colour->GetValue ());
+       _effect->Enable (_force_effect->GetValue ());
+       _fade_in->Enable (_force_fade_in->GetValue ());
+       _fade_out->Enable (_force_fade_out->GetValue ());
+
+       bool const can_outline_width = _effect->GetSelection() == OUTLINE && _content->subtitle->burn ();
+       _outline_width->Enable (can_outline_width);
+       if (can_outline_width) {
+               _outline_width->UnsetToolTip ();
+       } else {
+               _outline_width->SetToolTip (_("Outline width cannot be set unless you are burning in subtitles"));
+       }
 }