src/wx/subtitle_panel.cc

   1 /*
   2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <boost/lexical_cast.hpp>
  21 #include <wx/spinctrl.h>
  22 #include "lib/ffmpeg_content.h"
  23 #include "lib/subrip_content.h"
  24 #include "lib/ffmpeg_subtitle_stream.h"
  25 #include "lib/dcp_subtitle_content.h"
  26 #include "lib/subrip_decoder.h"
  27 #include "lib/dcp_subtitle_decoder.h"
  28 #include "subtitle_panel.h"
  29 #include "film_editor.h"
  30 #include "wx_util.h"
  31 #include "subtitle_view.h"
  32 #include "content_panel.h"
  33
  34 using std::vector;
  35 using std::string;
  36 using boost::shared_ptr;
  37 using boost::lexical_cast;
  38 using boost::dynamic_pointer_cast;
  39
  40 SubtitlePanel::SubtitlePanel (ContentPanel* p)
  41         : ContentSubPanel (p, _("Subtitles"))
  42         , _view (0)
  43 {
  44         wxFlexGridSizer* grid = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
  45         _sizer->Add (grid, 0, wxALL, 8);
  46
  47         _use = new wxCheckBox (this, wxID_ANY, _("Use subtitles"));
  48         grid->Add (_use);
  49         grid->AddSpacer (0);
  50
  51         {
  52                 add_label_to_sizer (grid, this, _("X Offset"), true);
  53                 wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
  54                 _x_offset = new wxSpinCtrl (this);
  55                 s->Add (_x_offset);
  56                 add_label_to_sizer (s, this, _("%"), false);
  57                 grid->Add (s);
  58         }
  59
  60         {
  61                 add_label_to_sizer (grid, this, _("Y Offset"), true);
  62                 wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
  63                 _y_offset = new wxSpinCtrl (this);
  64                 s->Add (_y_offset);
  65                 add_label_to_sizer (s, this, _("%"), false);
  66                 grid->Add (s);
  67         }
  68
  69         {
  70                 add_label_to_sizer (grid, this, _("X Scale"), true);
  71                 wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
  72                 _x_scale = new wxSpinCtrl (this);
  73                 s->Add (_x_scale);
  74                 add_label_to_sizer (s, this, _("%"), false);
  75                 grid->Add (s);
  76         }
  77
  78         {
  79                 add_label_to_sizer (grid, this, _("Y Scale"), true);
  80                 wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
  81                 _y_scale = new wxSpinCtrl (this);
  82                 s->Add (_y_scale);
  83                 add_label_to_sizer (s, this, _("%"), false);
  84                 grid->Add (s);
  85         }
  86
  87         add_label_to_sizer (grid, this, _("Language"), true);
  88         _language = new wxTextCtrl (this, wxID_ANY);
  89         grid->Add (_language, 1, wxEXPAND);
  90
  91         add_label_to_sizer (grid, this, _("Stream"), true);
  92         _stream = new wxChoice (this, wxID_ANY);
  93         grid->Add (_stream, 1, wxEXPAND);
  94
  95         _view_button = new wxButton (this, wxID_ANY, _("View..."));
  96         grid->Add (_view_button);
  97
  98         _x_offset->SetRange (-100, 100);
  99         _y_offset->SetRange (-100, 100);
 100         _x_scale->SetRange (10, 1000);
 101         _y_scale->SetRange (10, 1000);
 102
 103         _use->Bind         (wxEVT_COMMAND_CHECKBOX_CLICKED, boost::bind (&SubtitlePanel::use_toggled, this));
 104         _x_offset->Bind    (wxEVT_COMMAND_SPINCTRL_UPDATED, boost::bind (&SubtitlePanel::x_offset_changed, this));
 105         _y_offset->Bind    (wxEVT_COMMAND_SPINCTRL_UPDATED, boost::bind (&SubtitlePanel::y_offset_changed, this));
 106         _x_scale->Bind     (wxEVT_COMMAND_SPINCTRL_UPDATED, boost::bind (&SubtitlePanel::x_scale_changed, this));
 107         _y_scale->Bind     (wxEVT_COMMAND_SPINCTRL_UPDATED, boost::bind (&SubtitlePanel::y_scale_changed, this));
 108         _language->Bind    (wxEVT_COMMAND_TEXT_UPDATED,     boost::bind (&SubtitlePanel::language_changed, this));
 109         _stream->Bind      (wxEVT_COMMAND_CHOICE_SELECTED,  boost::bind (&SubtitlePanel::stream_changed, this));
 110         _view_button->Bind (wxEVT_COMMAND_BUTTON_CLICKED,   boost::bind (&SubtitlePanel::view_clicked, this));
 111 }
 112
 113 void
 114 SubtitlePanel::film_changed (Film::Property property)
 115 {
 116         if (property == Film::CONTENT) {
 117                 setup_sensitivity ();
 118         }
 119 }
 120
 121 void
 122 SubtitlePanel::film_content_changed (int property)
 123 {
 124         FFmpegContentList fc = _parent->selected_ffmpeg ();
 125         SubtitleContentList sc = _parent->selected_subtitle ();
 126
 127         shared_ptr<FFmpegContent> fcs;
 128         if (fc.size() == 1) {
 129                 fcs = fc.front ();
 130         }
 131
 132         shared_ptr<SubtitleContent> scs;
 133         if (sc.size() == 1) {
 134                 scs = sc.front ();
 135         }
 136
 137         if (property == FFmpegContentProperty::SUBTITLE_STREAMS) {
 138                 _stream->Clear ();
 139                 if (fcs) {
 140                         vector<shared_ptr<FFmpegSubtitleStream> > s = fcs->subtitle_streams ();
 141                         for (vector<shared_ptr<FFmpegSubtitleStream> >::iterator i = s.begin(); i != s.end(); ++i) {
 142                                 _stream->Append (std_to_wx ((*i)->name), new wxStringClientData (std_to_wx ((*i)->identifier ())));
 143                         }
 144
 145                         if (fcs->subtitle_stream()) {
 146                                 checked_set (_stream, fcs->subtitle_stream()->identifier ());
 147                         } else {
 148                                 _stream->SetSelection (wxNOT_FOUND);
 149                         }
 150                 }
 151                 setup_sensitivity ();
 152         } else if (property == SubtitleContentProperty::USE_SUBTITLES) {
 153                 checked_set (_use, scs ? scs->use_subtitles() : false);
 154                 setup_sensitivity ();
 155         } else if (property == SubtitleContentProperty::SUBTITLE_X_OFFSET) {
 156                 checked_set (_x_offset, scs ? (scs->subtitle_x_offset() * 100) : 0);
 157         } else if (property == SubtitleContentProperty::SUBTITLE_Y_OFFSET) {
 158                 checked_set (_y_offset, scs ? (scs->subtitle_y_offset() * 100) : 0);
 159         } else if (property == SubtitleContentProperty::SUBTITLE_X_SCALE) {
 160                 checked_set (_x_scale, scs ? int (rint (scs->subtitle_x_scale() * 100)) : 100);
 161         } else if (property == SubtitleContentProperty::SUBTITLE_Y_SCALE) {
 162                 checked_set (_y_scale, scs ? int (rint (scs->subtitle_y_scale() * 100)) : 100);
 163         }
 164 }
 165
 166 void
 167 SubtitlePanel::use_toggled ()
 168 {
 169         SubtitleContentList c = _parent->selected_subtitle ();
 170         for (SubtitleContentList::iterator i = c.begin(); i != c.end(); ++i) {
 171                 (*i)->set_use_subtitles (_use->GetValue());
 172         }
 173 }
 174
 175 void
 176 SubtitlePanel::setup_sensitivity ()
 177 {
 178         int any_subs = 0;
 179         int ffmpeg_subs = 0;
 180         int subrip_or_dcp_subs = 0;
 181         SubtitleContentList c = _parent->selected_subtitle ();
 182         for (SubtitleContentList::const_iterator i = c.begin(); i != c.end(); ++i) {
 183                 shared_ptr<const FFmpegContent> fc = boost::dynamic_pointer_cast<const FFmpegContent> (*i);
 184                 shared_ptr<const SubRipContent> sc = boost::dynamic_pointer_cast<const SubRipContent> (*i);
 185                 shared_ptr<const DCPSubtitleContent> dsc = boost::dynamic_pointer_cast<const DCPSubtitleContent> (*i);
 186                 if (fc) {
 187                         if (fc->has_subtitles ()) {
 188                                 ++ffmpeg_subs;
 189                                 ++any_subs;
 190                         }
 191                 } else if (sc || dsc) {
 192                         ++subrip_or_dcp_subs;
 193                         ++any_subs;
 194                 } else {
 195                         ++any_subs;
 196                 }
 197         }
 198
 199         _use->Enable (any_subs > 0);
 200         bool const use = _use->GetValue ();
 201
 202         _x_offset->Enable (any_subs > 0 && use);
 203         _y_offset->Enable (any_subs > 0 && use);
 204         _x_scale->Enable (any_subs > 0 && use);
 205         _y_scale->Enable (any_subs > 0 && use);
 206         _stream->Enable (ffmpeg_subs == 1);
 207         _view_button->Enable (subrip_or_dcp_subs == 1);
 208 }
 209
 210 void
 211 SubtitlePanel::stream_changed ()
 212 {
 213         FFmpegContentList fc = _parent->selected_ffmpeg ();
 214         if (fc.size() != 1) {
 215                 return;
 216         }
 217
 218         shared_ptr<FFmpegContent> fcs = fc.front ();
 219
 220         vector<shared_ptr<FFmpegSubtitleStream> > a = fcs->subtitle_streams ();
 221         vector<shared_ptr<FFmpegSubtitleStream> >::iterator i = a.begin ();
 222         string const s = string_client_data (_stream->GetClientObject (_stream->GetSelection ()));
 223         while (i != a.end() && (*i)->identifier () != s) {
 224                 ++i;
 225         }
 226
 227         if (i != a.end ()) {
 228                 fcs->set_subtitle_stream (*i);
 229         }
 230 }
 231
 232 void
 233 SubtitlePanel::x_offset_changed ()
 234 {
 235         SubtitleContentList c = _parent->selected_subtitle ();
 236         for (SubtitleContentList::iterator i = c.begin(); i != c.end(); ++i) {
 237                 (*i)->set_subtitle_x_offset (_x_offset->GetValue() / 100.0);
 238         }
 239 }
 240
 241 void
 242 SubtitlePanel::y_offset_changed ()
 243 {
 244         SubtitleContentList c = _parent->selected_subtitle ();
 245         for (SubtitleContentList::iterator i = c.begin(); i != c.end(); ++i) {
 246                 (*i)->set_subtitle_y_offset (_y_offset->GetValue() / 100.0);
 247         }
 248 }
 249
 250 void
 251 SubtitlePanel::x_scale_changed ()
 252 {
 253         SubtitleContentList c = _parent->selected_subtitle ();
 254         if (c.size() == 1) {
 255                 c.front()->set_subtitle_x_scale (_x_scale->GetValue() / 100.0);
 256         }
 257 }
 258
 259 void
 260 SubtitlePanel::y_scale_changed ()
 261 {
 262         SubtitleContentList c = _parent->selected_subtitle ();
 263         for (SubtitleContentList::iterator i = c.begin(); i != c.end(); ++i) {
 264                 (*i)->set_subtitle_y_scale (_y_scale->GetValue() / 100.0);
 265         }
 266 }
 267
 268 void
 269 SubtitlePanel::language_changed ()
 270 {
 271         SubtitleContentList c = _parent->selected_subtitle ();
 272         for (SubtitleContentList::iterator i = c.begin(); i != c.end(); ++i) {
 273                 (*i)->set_subtitle_language (wx_to_std (_language->GetValue()));
 274         }
 275 }
 276
 277 void
 278 SubtitlePanel::content_selection_changed ()
 279 {
 280         film_content_changed (FFmpegContentProperty::SUBTITLE_STREAMS);
 281         film_content_changed (SubtitleContentProperty::USE_SUBTITLES);
 282         film_content_changed (SubtitleContentProperty::SUBTITLE_X_OFFSET);
 283         film_content_changed (SubtitleContentProperty::SUBTITLE_Y_OFFSET);
 284         film_content_changed (SubtitleContentProperty::SUBTITLE_X_SCALE);
 285         film_content_changed (SubtitleContentProperty::SUBTITLE_Y_SCALE);
 286         film_content_changed (SubtitleContentProperty::SUBTITLE_LANGUAGE);
 287 }
 288
 289 void
 290 SubtitlePanel::view_clicked ()
 291 {
 292         if (_view) {
 293                 _view->Destroy ();
 294                 _view = 0;
 295         }
 296
 297         SubtitleContentList c = _parent->selected_subtitle ();
 298         assert (c.size() == 1);
 299
 300         shared_ptr<SubtitleDecoder> decoder;
 301
 302         shared_ptr<SubRipContent> sr = dynamic_pointer_cast<SubRipContent> (c.front ());
 303         if (sr) {
 304                 decoder.reset (new SubRipDecoder (sr));
 305         }
 306
 307         shared_ptr<DCPSubtitleContent> dc = dynamic_pointer_cast<DCPSubtitleContent> (c.front ());
 308         if (dc) {
 309                 decoder.reset (new DCPSubtitleDecoder (dc));
 310         }
 311
 312         if (decoder) {
 313                 _view = new SubtitleView (this, _parent->film(), decoder, c.front()->position ());
 314                 _view->Show ();
 315         }
 316 }