Move _state_timer into VideoView.
[dcpomatic.git] / src / wx / subtitle_appearance_dialog.cc
index 831afe757d880832ff517e4a16765089bca35112..76e8d8b16d00f1df23698d07257278779f43f14d 100644 (file)
 
 #include "subtitle_appearance_dialog.h"
 #include "rgba_colour_picker.h"
-#include "lib/text_subtitle_content.h"
-#include "lib/subtitle_content.h"
+#include "static_text.h"
+#include "check_box.h"
+#include "dcpomatic_button.h"
+#include "lib/string_text_file_content.h"
+#include "lib/text_content.h"
 #include "lib/ffmpeg_subtitle_stream.h"
 #include "lib/ffmpeg_content.h"
+#include "lib/examine_ffmpeg_subtitles_job.h"
+#include "lib/job_manager.h"
 #include <wx/wx.h>
 #include <wx/clrpicker.h>
 #include <wx/spinctrl.h>
 #include <wx/gbsizer.h>
 
 using std::map;
+using std::string;
 using boost::shared_ptr;
 using boost::bind;
 using boost::dynamic_pointer_cast;
 using boost::optional;
+using namespace dcpomatic;
 
 int const SubtitleAppearanceDialog::NONE = 0;
 int const SubtitleAppearanceDialog::OUTLINE = 1;
 int const SubtitleAppearanceDialog::SHADOW = 2;
 
-SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr<Content> content)
-       : wxDialog (parent, wxID_ANY, _("Subtitle appearance"))
+SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr<const Film> film, shared_ptr<Content> content, shared_ptr<TextContent> caption)
+       : wxDialog (parent, wxID_ANY, _("Caption appearance"))
+       , _film (film)
+       , _finding (0)
        , _content (content)
+       , _caption (caption)
 {
        shared_ptr<FFmpegContent> ff = dynamic_pointer_cast<FFmpegContent> (content);
        if (ff) {
                _stream = ff->subtitle_stream ();
+               /* XXX: assuming that all FFmpeg streams have bitmap subs */
+               if (_stream->colours().empty()) {
+                       _job_manager_connection = JobManager::instance()->ActiveJobsChanged.connect(boost::bind(&SubtitleAppearanceDialog::active_jobs_changed, this, _1));
+                       _job = JobManager::instance()->add(shared_ptr<Job>(new ExamineFFmpegSubtitlesJob(film, ff)));
+               }
        }
 
-       wxSizer* overall_sizer = new wxBoxSizer (wxVERTICAL);
-       SetSizer (overall_sizer);
+       _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);
+       _overall_sizer->Add (_table, 1, wxEXPAND | wxALL, DCPOMATIC_DIALOG_BORDER);
 
        int r = 0;
 
@@ -78,63 +93,64 @@ SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr
        _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);
+               _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);
 
-               map<RGBA, RGBA> colours = _stream->colours ();
+               _colour_table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
+               _colour_table->AddGrowableCol (1, 1);
 
-               wxStaticText* t = new wxStaticText (colours_panel, wxID_ANY, "");
+               wxStaticText* t = new StaticText (_colours_panel, "");
                t->SetLabelMarkup (_("<b>Original colour</b>"));
-               table->Add (t, 1, wxEXPAND);
-               t = new wxStaticText (colours_panel, wxID_ANY, "", wxDefaultPosition, wxDefaultSize, wxALIGN_CENTRE_HORIZONTAL);
+               _colour_table->Add (t, 1, wxEXPAND);
+               t = new StaticText (_colours_panel, "", 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;
+               _colour_table->Add (t, 1, wxALIGN_CENTER);
+
+               shared_ptr<Job> job = _job.lock ();
+               if (!job || job->finished()) {
+                       add_colours ();
                }
 
-               colours_panel->SetSizer (table);
+               _colours_panel->SetSizer (_colour_table);
 
-               overall_sizer->Add (colours_panel, 1, wxEXPAND | wxALL, DCPOMATIC_DIALOG_BORDER);
+               /* XXX: still assuming that all FFmpeg streams have bitmap subs */
+               if (_stream->colours().empty()) {
+                       _finding = new wxStaticText(this, wxID_ANY, _("Finding the colours in these subtitles..."));
+                       _overall_sizer->Add (_finding, 0, wxALL, DCPOMATIC_DIALOG_BORDER);
+                       _colours_panel->Show (false);
+               }
 
-               wxButton* restore = new wxButton (this, wxID_ANY, _("Restore to original colours"));
+               _overall_sizer->Add (_colours_panel, 1, wxEXPAND | wxALL, DCPOMATIC_DIALOG_BORDER);
+
+               wxButton* restore = new Button (this, _("Restore to original colours"));
                restore->Bind (wxEVT_BUTTON, bind (&SubtitleAppearanceDialog::restore, this));
-               overall_sizer->Add (restore, 0, wxALL, DCPOMATIC_SIZER_X_GAP);
+               _overall_sizer->Add (restore, 0, wxALL, DCPOMATIC_SIZER_X_GAP);
        }
 
        wxSizer* buttons = CreateSeparatedButtonSizer (wxOK);
        if (buttons) {
-               overall_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
+               _overall_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
        }
 
-       overall_sizer->Layout ();
-       overall_sizer->SetSizeHints (this);
+       _overall_sizer->Layout ();
+       _overall_sizer->SetSizeHints (this);
 
        /* Keep these Appends() up to date with NONE/OUTLINE/SHADOW variables */
        _effect->Append (_("None"));
        _effect->Append (_("Outline"));
        _effect->Append (_("Shadow"));;
 
-       optional<dcp::Colour> colour = _content->subtitle->colour();
-       _force_effect_colour->SetValue (static_cast<bool>(colour));
+       optional<dcp::Colour> colour = _caption->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));
        }
 
-       optional<dcp::Effect> effect = _content->subtitle->effect();
+       optional<dcp::Effect> effect = _caption->effect();
        _force_effect->SetValue (static_cast<bool>(effect));
        if (effect) {
                switch (*effect) {
@@ -152,31 +168,31 @@ SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr
                _effect->SetSelection (NONE);
        }
 
-       optional<dcp::Colour> effect_colour = _content->subtitle->effect_colour();
-       _force_effect_colour->SetValue (static_cast<bool>(_force_effect_colour));
+       optional<dcp::Colour> effect_colour = _caption->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();
+       optional<ContentTime> fade_in = _caption->fade_in();
        _force_fade_in->SetValue (static_cast<bool>(fade_in));
        if (fade_in) {
-               _fade_in->set (*fade_in, _content->active_video_frame_rate());
+               _fade_in->set (*fade_in, _content->active_video_frame_rate(film));
        } else {
-               _fade_in->set (ContentTime(), _content->active_video_frame_rate());
+               _fade_in->set (ContentTime(), _content->active_video_frame_rate(film));
        }
 
-       optional<ContentTime> fade_out = _content->subtitle->fade_out();
+       optional<ContentTime> fade_out = _caption->fade_out();
        _force_fade_out->SetValue (static_cast<bool>(fade_out));
        if (fade_out) {
-               _fade_out->set (*fade_out, _content->active_video_frame_rate ());
+               _fade_out->set (*fade_out, _content->active_video_frame_rate(film));
        } else {
-               _fade_out->set (ContentTime(), _content->active_video_frame_rate ());
+               _fade_out->set (ContentTime(), _content->active_video_frame_rate(film));
        }
 
-       _outline_width->SetValue (_content->subtitle->outline_width ());
+       _outline_width->SetValue (_caption->outline_width ());
 
        _force_colour->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
        _force_effect_colour->Bind (wxEVT_CHECKBOX, bind (&SubtitleAppearanceDialog::setup_sensitivity, this));
@@ -184,16 +200,24 @@ SubtitleAppearanceDialog::SubtitleAppearanceDialog (wxWindow* parent, shared_ptr
        _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));
+       _content_connection = _content->Change.connect (bind (&SubtitleAppearanceDialog::content_change, this, _1));
 
        setup_sensitivity ();
 }
 
+void
+SubtitleAppearanceDialog::content_change (ChangeType type)
+{
+       if (type == CHANGE_TYPE_DONE) {
+               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"));
+       wxCheckBox* set_to = new CheckBox (this, _("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));
@@ -204,40 +228,46 @@ SubtitleAppearanceDialog::set_to (wxWindow* w, int& r)
 void
 SubtitleAppearanceDialog::apply ()
 {
+       shared_ptr<const Film> film = _film.lock ();
+
        if (_force_colour->GetValue ()) {
                wxColour const c = _colour->GetColour ();
-               _content->subtitle->set_colour (dcp::Colour (c.Red(), c.Green(), c.Blue()));
+               _caption->set_colour (dcp::Colour (c.Red(), c.Green(), c.Blue()));
        } else {
-               _content->subtitle->unset_colour ();
+               _caption->unset_colour ();
        }
-       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;
+       if (_force_effect->GetValue()) {
+               switch (_effect->GetSelection()) {
+               case NONE:
+                       _caption->set_effect (dcp::NONE);
+                       break;
+               case OUTLINE:
+                       _caption->set_effect (dcp::BORDER);
+                       break;
+               case SHADOW:
+                       _caption->set_effect (dcp::SHADOW);
+                       break;
+               }
+       } else {
+               _caption->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()));
+               _caption->set_effect_colour (dcp::Colour (ec.Red(), ec.Green(), ec.Blue()));
        } else {
-               _content->subtitle->unset_effect_colour ();
+               _caption->unset_effect_colour ();
        }
        if (_force_fade_in->GetValue ()) {
-               _content->subtitle->set_fade_in (_fade_in->get (_content->active_video_frame_rate ()));
+               _caption->set_fade_in (_fade_in->get(_content->active_video_frame_rate(film)));
        } else {
-               _content->subtitle->unset_fade_in ();
+               _caption->unset_fade_in ();
        }
        if (_force_fade_out->GetValue ()) {
-               _content->subtitle->set_fade_out (_fade_out->get (_content->active_video_frame_rate ()));
+               _caption->set_fade_out (_fade_out->get(_content->active_video_frame_rate(film)));
        } else {
-               _content->subtitle->unset_fade_out ();
+               _caption->unset_fade_out ();
        }
-       _content->subtitle->set_outline_width (_outline_width->GetValue ());
+       _caption->set_outline_width (_outline_width->GetValue ());
 
        if (_stream) {
                for (map<RGBA, RGBAColourPicker*>::const_iterator i = _pickers.begin(); i != _pickers.end(); ++i) {
@@ -268,11 +298,39 @@ SubtitleAppearanceDialog::setup_sensitivity ()
        _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 ();
+       bool const can_outline_width = _effect->GetSelection() == OUTLINE && _caption->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"));
+               _outline_width->SetToolTip (_("Outline width cannot be set unless you are burning in captions"));
+       }
+}
+
+void
+SubtitleAppearanceDialog::active_jobs_changed (optional<string> last)
+{
+       if (last && *last == "examine_subtitles") {
+               _colours_panel->Show (true);
+               if (_finding) {
+                       _finding->Show (false);
+               }
+               add_colours ();
+               _overall_sizer->Layout ();
+               _overall_sizer->SetSizeHints (this);
+       }
+}
+
+void
+SubtitleAppearanceDialog::add_colours ()
+{
+       map<RGBA, RGBA> colours = _stream->colours ();
+       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));
+               _colour_table->Add (from, 1, wxEXPAND);
+               RGBAColourPicker* to = new RGBAColourPicker (_colours_panel, i->second);
+               _colour_table->Add (to, 1, wxEXPAND);
+               _pickers[i->first] = to;
        }
 }