Remove unused LOG_DEBUG_DECODE and re-use it to add some 3D shuffler logging.
[dcpomatic.git] / src / wx / closed_captions_dialog.cc
index be520057b81a4b46cf543442b52aef7b30411daa..afcf5fa876f97b35d0cf66bf9765cf379e0afa13 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2018 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2018-2019 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
 
 */
 
-#include "closed_captions_view.h"
+#include "closed_captions_dialog.h"
+#include "wx_util.h"
+#include "film_viewer.h"
+#include "lib/string_text.h"
+#include "lib/butler.h"
+#include "lib/text_content.h"
+#include "lib/compose.hpp"
 #include <boost/bind.hpp>
 
 using std::list;
+using std::max;
 using std::cout;
+using std::pair;
 using std::make_pair;
+using boost::shared_ptr;
+using boost::weak_ptr;
+using boost::optional;
+using namespace dcpomatic;
 
-int const ClosedCaptionsDialog::_num_lines = 3;
-int const ClosedCaptionsDialog::_num_chars_per_line = 30;
-
-ClosedCaptionsDialog::ClosedCaptionsDialog (wxWindow* parent)
+ClosedCaptionsDialog::ClosedCaptionsDialog (wxWindow* parent, FilmViewer* viewer)
        : wxDialog (parent, wxID_ANY, _("Closed captions"), wxDefaultPosition, wxDefaultSize,
 #ifdef DCPOMATIC_OSX
-               /* I can't get wxFRAME_FLOAT_ON_PARENT to work on OS X, and although wxSTAY_ON_TOP keeps
-                  the window above all others (and not just our own) it's better than nothing for now.
-               */
-               wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxSTAY_ON_TOP
+                   /* I can't get wxFRAME_FLOAT_ON_PARENT to work on OS X, and although wxSTAY_ON_TOP keeps
+                      the window above all others (and not just our own) it's better than nothing for now.
+                   */
+                   wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxSTAY_ON_TOP
 #else
-               wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxFRAME_FLOAT_ON_PARENT
+                   wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxFRAME_FLOAT_ON_PARENT
 #endif
                )
+       , _viewer (viewer)
+         /* XXX: empirical and probably unhelpful default size here; needs to be related to font metrics */
+        , _display (new wxPanel(this, wxID_ANY, wxDefaultPosition, wxSize(640, (640 / 10) + 64)))
+       , _track (new wxChoice(this, wxID_ANY))
+       , _current_in_lines (false)
+       , _timer (this)
+{
+       _lines.resize (CLOSED_CAPTION_LINES);
+
+       wxBoxSizer* sizer = new wxBoxSizer (wxVERTICAL);
+
+       wxBoxSizer* track_sizer = new wxBoxSizer (wxHORIZONTAL);
+       add_label_to_sizer (track_sizer, this, _("Track"), true);
+       track_sizer->Add (_track, 0, wxEXPAND | wxLEFT, DCPOMATIC_SIZER_X_GAP);
+
+       sizer->Add (track_sizer, 0, wxALL, DCPOMATIC_SIZER_GAP);
+       sizer->Add (_display, 1, wxEXPAND);
+
+       Bind (wxEVT_SHOW, boost::bind(&ClosedCaptionsDialog::shown, this, _1));
+       Bind (wxEVT_TIMER, boost::bind(&ClosedCaptionsDialog::update, this));
+       _display->Bind (wxEVT_PAINT, boost::bind(&ClosedCaptionsDialog::paint, this));
+       _track->Bind (wxEVT_CHOICE, boost::bind(&ClosedCaptionsDialog::track_selected, this));
 
+       SetSizerAndFit (sizer);
+}
+
+void
+ClosedCaptionsDialog::shown (wxShowEvent ev)
 {
-       _lines.resize (_num_lines);
-       Bind (wxEVT_PAINT, boost::bind (&ClosedCaptionsDialog::paint, this));
+       if (ev.IsShown ()) {
+               _timer.Start (40);
+       } else {
+               _timer.Stop ();
+       }
+}
+
+void
+ClosedCaptionsDialog::track_selected ()
+{
+       _current = optional<TextRingBuffers::Data> ();
+       _viewer->slow_refresh ();
+       update ();
 }
 
 void
 ClosedCaptionsDialog::paint ()
 {
-       wxPaintDC dc (this);
+       wxPaintDC dc (_display);
        dc.SetBackground (*wxBLACK_BRUSH);
        dc.Clear ();
        dc.SetTextForeground (*wxWHITE);
 
        /* Choose a font which fits vertically */
-       int const line_height = dc.GetSize().GetHeight() / _num_lines;
+       int const line_height = max (8, dc.GetSize().GetHeight() / CLOSED_CAPTION_LINES);
        wxFont font (*wxNORMAL_FONT);
        font.SetPixelSize (wxSize (0, line_height * 0.8));
        dc.SetFont (font);
 
-       for (int i = 0; i < _num_lines; ++i) {
-               if (_lines[i].IsEmpty()) {
-                       dc.DrawText (wxString::Format("Line %d", i + 1), 8, line_height * i);
-               } else {
-                       dc.DrawText (_lines[i], 8, line_height * i);
+       for (int i = 0; i < CLOSED_CAPTION_LINES; ++i) {
+               wxString const good = _lines[i].Left (CLOSED_CAPTION_LENGTH);
+               dc.DrawText (good, 8, line_height * i);
+               if (_lines[i].Length() > CLOSED_CAPTION_LENGTH) {
+                       wxString const bad = _lines[i].Right (_lines[i].Length() - CLOSED_CAPTION_LENGTH);
+                       wxSize size = dc.GetTextExtent (good);
+                       dc.SetTextForeground (*wxRED);
+                       dc.DrawText (bad, 8 + size.GetWidth(), line_height * i);
+                       dc.SetTextForeground (*wxWHITE);
                }
        }
 }
@@ -71,13 +122,13 @@ ClosedCaptionsDialog::paint ()
 class ClosedCaptionSorter
 {
 public:
-       bool operator() (TextCaption const & a, TextCaption const & b)
+       bool operator() (StringText const & a, StringText const & b)
        {
                return from_top(a) < from_top(b);
        }
 
 private:
-       float from_top (TextCaption const & c) const
+       float from_top (StringText const & c) const
        {
                switch (c.v_align()) {
                case dcp::VALIGN_TOP:
@@ -93,51 +144,100 @@ private:
 };
 
 void
-ClosedCaptionsDialog::refresh (DCPTime time)
+ClosedCaptionsDialog::update ()
 {
-       list<TextCaption> to_show;
-       list<Caption>::iterator i = _captions.begin ();
-       while (i != _captions.end ()) {
-               if (time > i->second.to) {
-                       list<Caption>::iterator tmp = i;
-                       ++i;
-                       _captions.erase (tmp);
-               } else if (i->second.contains (time)) {
-                       BOOST_FOREACH (TextCaption j, i->first.text) {
-                               to_show.push_back (j);
-                       }
-                       ++i;
-               } else {
-                       ++i;
-               }
+       DCPTime const time = _viewer->time ();
+
+       if (_current_in_lines && _current && _current->period.to > time) {
+               /* Current one is fine */
+               return;
        }
 
-       for (int j = 0; j < _num_lines; ++j) {
-               _lines[j] = "";
+       if (_current && _current->period.to < time) {
+               /* Current one has finished; clear out */
+               for (int j = 0; j < CLOSED_CAPTION_LINES; ++j) {
+                       _lines[j] = "";
+               }
+               Refresh ();
+               _current = optional<TextRingBuffers::Data>();
        }
 
-       to_show.sort (ClosedCaptionSorter());
+       if (!_current && !_tracks.empty()) {
+               /* We have no current one: get another */
+               shared_ptr<Butler> butler = _butler.lock ();
+               DCPOMATIC_ASSERT (butler);
+               DCPOMATIC_ASSERT (_track->GetSelection() >= 0);
+               DCPOMATIC_ASSERT (_track->GetSelection() < int(_tracks.size()));
+               DCPTextTrack track = _tracks[_track->GetSelection()];
+               while (true) {
+                       optional<TextRingBuffers::Data> d = butler->get_closed_caption ();
+                       if (!d) {
+                               break;
+                       }
+                       if (d->track == track) {
+                               _current = d;
+                               break;
+                       }
+               }
 
-       list<TextCaption>::const_iterator j = to_show.begin();
-       int k = 0;
-       while (j != to_show.end() && k < _num_lines) {
-               _lines[k] = j->text();
-               ++j;
-               ++k;
+               _current_in_lines = false;
        }
 
-       Refresh ();
+       if (_current && _current->period.contains(time)) {
+               /* We need to set this new one up */
+
+               list<StringText> to_show = _current->text.string;
+
+               for (int j = 0; j < CLOSED_CAPTION_LINES; ++j) {
+                       _lines[j] = "";
+               }
+
+               to_show.sort (ClosedCaptionSorter());
+
+               list<StringText>::const_iterator j = to_show.begin();
+               int k = 0;
+               while (j != to_show.end() && k < CLOSED_CAPTION_LINES) {
+                       _lines[k] = j->text();
+                       ++j;
+                       ++k;
+               }
+
+               Refresh ();
+               _current_in_lines = true;
+       }
 }
 
 void
-ClosedCaptionsDialog::caption (PlayerCaption caption, DCPTimePeriod period)
+ClosedCaptionsDialog::clear ()
 {
-       _captions.push_back (make_pair (caption, period));
+       _current = optional<TextRingBuffers::Data>();
+       _current_in_lines = false;
+       Refresh ();
 }
 
 void
-ClosedCaptionsDialog::clear ()
+ClosedCaptionsDialog::set_film_and_butler (shared_ptr<Film> film, weak_ptr<Butler> butler)
 {
-       _captions.clear ();
-       Refresh ();
+       _tracks.clear ();
+
+       BOOST_FOREACH (shared_ptr<Content> i, film->content()) {
+               BOOST_FOREACH (shared_ptr<TextContent> j, i->text) {
+                       if (j->use() && j->type() == TEXT_CLOSED_CAPTION && j->dcp_track()) {
+                               if (find(_tracks.begin(), _tracks.end(), j->dcp_track()) == _tracks.end()) {
+                                       _tracks.push_back (*j->dcp_track());
+                               }
+                       }
+               }
+       }
+
+       _track->Clear ();
+       BOOST_FOREACH (DCPTextTrack const & i, _tracks) {
+               _track->Append (std_to_wx(String::compose("%1 (%2)", i.name, i.language)));
+       }
+
+       if (_track->GetCount() > 0) {
+               _track->SetSelection (0);
+       }
+
+       _butler = butler;
 }