857db66a8b396c13de0f8c2ff273ba30d6491b0b
[dcpomatic.git] / src / wx / closed_captions_dialog.cc
1 /*
2     Copyright (C) 2018 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "closed_captions_dialog.h"
22 #include "lib/string_text.h"
23 #include <boost/bind.hpp>
24
25 using std::list;
26 using std::cout;
27 using std::make_pair;
28 using boost::shared_ptr;
29 using boost::weak_ptr;
30
31 int const ClosedCaptionsDialog::_num_lines = 3;
32 int const ClosedCaptionsDialog::_num_chars_per_line = 30;
33
34 ClosedCaptionsDialog::ClosedCaptionsDialog (wxWindow* parent)
35         : wxDialog (parent, wxID_ANY, _("Closed captions"), wxDefaultPosition, wxDefaultSize,
36 #ifdef DCPOMATIC_OSX
37                 /* I can't get wxFRAME_FLOAT_ON_PARENT to work on OS X, and although wxSTAY_ON_TOP keeps
38                    the window above all others (and not just our own) it's better than nothing for now.
39                 */
40                 wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxSTAY_ON_TOP
41 #else
42                 wxDEFAULT_FRAME_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE | wxFRAME_FLOAT_ON_PARENT
43 #endif
44                 )
45
46 {
47         _lines.resize (_num_lines);
48         Bind (wxEVT_PAINT, boost::bind (&ClosedCaptionsDialog::paint, this));
49 }
50
51 void
52 ClosedCaptionsDialog::paint ()
53 {
54         wxPaintDC dc (this);
55         dc.SetBackground (*wxBLACK_BRUSH);
56         dc.Clear ();
57         dc.SetTextForeground (*wxWHITE);
58
59         /* Choose a font which fits vertically */
60         int const line_height = dc.GetSize().GetHeight() / _num_lines;
61         wxFont font (*wxNORMAL_FONT);
62         font.SetPixelSize (wxSize (0, line_height * 0.8));
63         dc.SetFont (font);
64
65         for (int i = 0; i < _num_lines; ++i) {
66                 wxString const good = _lines[i].Left (_num_chars_per_line);
67                 dc.DrawText (good, 8, line_height * i);
68                 if (_lines[i].Length() > _num_chars_per_line) {
69                         wxString const bad = _lines[i].Right (_lines[i].Length() - _num_chars_per_line);
70                         wxSize size = dc.GetTextExtent (good);
71                         dc.SetTextForeground (*wxRED);
72                         dc.DrawText (bad, 8 + size.GetWidth(), line_height * i);
73                         dc.SetTextForeground (*wxWHITE);
74                 }
75         }
76 }
77
78 class ClosedCaptionSorter
79 {
80 public:
81         bool operator() (StringText const & a, StringText const & b)
82         {
83                 return from_top(a) < from_top(b);
84         }
85
86 private:
87         float from_top (StringText const & c) const
88         {
89                 switch (c.v_align()) {
90                 case dcp::VALIGN_TOP:
91                         return c.v_position();
92                 case dcp::VALIGN_CENTER:
93                         return c.v_position() + 0.5;
94                 case dcp::VALIGN_BOTTOM:
95                         return 1.0 - c.v_position();
96                 }
97                 DCPOMATIC_ASSERT (false);
98                 return 0;
99         }
100 };
101
102 void
103 ClosedCaptionsDialog::update (DCPTime time)
104 {
105         shared_ptr<Player> player = _player.lock ();
106         DCPOMATIC_ASSERT (player);
107         list<StringText> to_show;
108         BOOST_FOREACH (PlayerText i, player->closed_captions_for_frame(time)) {
109                 BOOST_FOREACH (StringText j, i.text) {
110                         to_show.push_back (j);
111                 }
112         }
113
114         for (int j = 0; j < _num_lines; ++j) {
115                 _lines[j] = "";
116         }
117
118         to_show.sort (ClosedCaptionSorter());
119
120         list<StringText>::const_iterator j = to_show.begin();
121         int k = 0;
122         while (j != to_show.end() && k < _num_lines) {
123                 _lines[k] = j->text();
124                 ++j;
125                 ++k;
126         }
127
128         Refresh ();
129 }
130
131 void
132 ClosedCaptionsDialog::clear ()
133 {
134         Refresh ();
135 }
136
137 void
138 ClosedCaptionsDialog::set_player (weak_ptr<Player> player)
139 {
140         _player = player;
141 }