cb3cc786ff7d00d679ac0e834d60c41c72d198ee
[dcpomatic.git] / src / wx / audio_mapping_view.cc
1 /*
2     Copyright (C) 2013-2016 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 /** @file  src/wx/audio_mapping_view.cc
22  *  @brief AudioMappingView class and helpers.
23  */
24
25 #include "audio_mapping_view.h"
26 #include "wx_util.h"
27 #include "audio_gain_dialog.h"
28 #include "lib/audio_mapping.h"
29 #include "lib/util.h"
30 #include <dcp/locale_convert.h>
31 #include <dcp/types.h>
32 #include <wx/wx.h>
33 #include <wx/renderer.h>
34 #include <wx/grid.h>
35 #include <wx/graphics.h>
36 #include <boost/foreach.hpp>
37 #include <iostream>
38
39 using std::cout;
40 using std::list;
41 using std::string;
42 using std::min;
43 using std::max;
44 using std::vector;
45 using std::pair;
46 using std::make_pair;
47 using boost::shared_ptr;
48 using dcp::locale_convert;
49
50 #define INDICATOR_SIZE 16
51 #define LEFT_WIDTH 48
52
53 enum {
54         ID_off = 1,
55         ID_full = 2,
56         ID_minus6dB = 3,
57         ID_edit = 4
58 };
59
60 class NoSelectionStringRenderer : public wxGridCellStringRenderer
61 {
62 public:
63         void Draw (wxGrid& grid, wxGridCellAttr& attr, wxDC& dc, const wxRect& rect, int row, int col, bool)
64         {
65                 wxGridCellStringRenderer::Draw (grid, attr, dc, rect, row, col, false);
66         }
67 };
68
69 /** @class ValueRenderer
70  *  @brief wxGridCellRenderer for a gain value.
71  */
72 class ValueRenderer : public wxGridCellRenderer
73 {
74 public:
75
76         void Draw (wxGrid& grid, wxGridCellAttr &, wxDC& dc, const wxRect& rect, int row, int col, bool)
77         {
78                 dc.SetPen (*wxThePenList->FindOrCreatePen (wxColour (255, 255, 255), 1, wxPENSTYLE_SOLID));
79                 dc.SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (255, 255, 255), wxBRUSHSTYLE_SOLID));
80                 dc.DrawRectangle (rect);
81
82                 int const xo = (rect.GetWidth() - INDICATOR_SIZE) / 2;
83                 int const yo = (rect.GetHeight() - INDICATOR_SIZE) / 2;
84
85                 dc.SetPen (*wxThePenList->FindOrCreatePen (wxColour (0, 0, 0), 1, wxPENSTYLE_SOLID));
86                 dc.SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (255, 255, 255), wxBRUSHSTYLE_SOLID));
87                 dc.DrawRectangle (wxRect (rect.GetLeft() + xo, rect.GetTop() + yo, INDICATOR_SIZE, INDICATOR_SIZE));
88
89                 float const value = locale_convert<float> (wx_to_std (grid.GetCellValue (row, col)));
90                 float const value_dB = 20 * log10 (value);
91                 int const range = 18;
92                 int height = 0;
93                 if (value_dB > -range) {
94                         height = INDICATOR_SIZE * (1 + value_dB / range);
95                 }
96
97                 height = max (0, height);
98
99                 if (value > 0) {
100                         /* Make sure we get a little bit of the marker if there is any gain */
101                         height = max (3, height);
102                 }
103
104                 dc.SetBrush (*wxTheBrushList->FindOrCreateBrush (wxColour (0, 255, 0), wxBRUSHSTYLE_SOLID));
105                 dc.DrawRectangle (wxRect (rect.GetLeft() + xo, rect.GetTop() + yo + INDICATOR_SIZE - height, INDICATOR_SIZE, height));
106         }
107
108         wxSize GetBestSize (wxGrid &, wxGridCellAttr &, wxDC &, int, int)
109         {
110                 return wxSize (INDICATOR_SIZE + 4, INDICATOR_SIZE + 4);
111         }
112
113         wxGridCellRenderer* Clone () const
114         {
115                 return new ValueRenderer;
116         }
117 };
118
119
120 AudioMappingView::AudioMappingView (wxWindow* parent)
121         : wxPanel (parent, wxID_ANY)
122         , _menu_row (0)
123         , _menu_column (1)
124         , _last_tooltip_row (0)
125         , _last_tooltip_column (0)
126 {
127         _left_labels = new wxPanel (this, wxID_ANY);
128         _left_labels->Bind (wxEVT_PAINT, boost::bind (&AudioMappingView::paint_left_labels, this));
129         _top_labels = new wxPanel (this, wxID_ANY);
130         _top_labels->Bind (wxEVT_PAINT, boost::bind (&AudioMappingView::paint_top_labels, this));
131
132         _grid = new wxGrid (this, wxID_ANY);
133
134         _grid->CreateGrid (0, MAX_DCP_AUDIO_CHANNELS + 1);
135         _grid->HideRowLabels ();
136         _grid->DisableDragRowSize ();
137         _grid->DisableDragColSize ();
138         _grid->EnableEditing (false);
139         _grid->SetCellHighlightPenWidth (0);
140         _grid->SetDefaultRenderer (new NoSelectionStringRenderer);
141         _grid->AutoSize ();
142
143         wxSizer* vertical_sizer = new wxBoxSizer (wxVERTICAL);
144         vertical_sizer->Add (_top_labels);
145         wxSizer* horizontal_sizer = new wxBoxSizer (wxHORIZONTAL);
146         horizontal_sizer->Add (_left_labels);
147         horizontal_sizer->Add (_grid, 1, wxEXPAND | wxALL);
148         vertical_sizer->Add (horizontal_sizer);
149         SetSizerAndFit (vertical_sizer);
150
151         Bind (wxEVT_GRID_CELL_LEFT_CLICK, boost::bind (&AudioMappingView::left_click, this, _1));
152         Bind (wxEVT_GRID_CELL_RIGHT_CLICK, boost::bind (&AudioMappingView::right_click, this, _1));
153         _grid->GetGridWindow()->Bind (wxEVT_MOTION, boost::bind (&AudioMappingView::mouse_moved_grid, this, _1));
154         _grid->Bind (wxEVT_SCROLLWIN_TOP, boost::bind (&AudioMappingView::grid_scrolled, this));
155         _grid->Bind (wxEVT_SCROLLWIN_BOTTOM, boost::bind (&AudioMappingView::grid_scrolled, this));
156         _grid->Bind (wxEVT_SCROLLWIN_LINEUP, boost::bind (&AudioMappingView::grid_scrolled, this));
157         _grid->Bind (wxEVT_SCROLLWIN_LINEDOWN, boost::bind (&AudioMappingView::grid_scrolled, this));
158         _grid->Bind (wxEVT_SCROLLWIN_PAGEUP, boost::bind (&AudioMappingView::grid_scrolled, this));
159         _grid->Bind (wxEVT_SCROLLWIN_PAGEDOWN, boost::bind (&AudioMappingView::grid_scrolled, this));
160         _grid->Bind (wxEVT_SCROLLWIN_THUMBTRACK, boost::bind (&AudioMappingView::grid_scrolled, this));
161         _grid->Bind (wxEVT_SCROLLWIN_THUMBRELEASE, boost::bind (&AudioMappingView::grid_scrolled, this));
162         Bind (wxEVT_SIZE, boost::bind (&AudioMappingView::sized, this, _1));
163
164         _menu = new wxMenu;
165         _menu->Append (ID_off, _("Off"));
166         _menu->Append (ID_full, _("Full"));
167         _menu->Append (ID_minus6dB, _("-6dB"));
168         _menu->Append (ID_edit, _("Edit..."));
169
170         Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&AudioMappingView::off, this), ID_off);
171         Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&AudioMappingView::full, this), ID_full);
172         Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&AudioMappingView::minus6dB, this), ID_minus6dB);
173         Bind (wxEVT_COMMAND_MENU_SELECTED, boost::bind (&AudioMappingView::edit, this), ID_edit);
174
175         _left_labels->Bind (wxEVT_MOTION, bind (&AudioMappingView::mouse_moved_left_labels, this, _1));
176 }
177
178 /** Called when any gain value has changed */
179 void
180 AudioMappingView::map_values_changed ()
181 {
182         update_cells ();
183         Changed (_map);
184         _last_tooltip_column = -1;
185 }
186
187 void
188 AudioMappingView::left_click (wxGridEvent& ev)
189 {
190         if (ev.GetCol() == 0) {
191                 return;
192         }
193
194         int const d = ev.GetCol() - 1;
195
196         if (_map.get (ev.GetRow(), d) > 0) {
197                 _map.set (ev.GetRow(), d, 0);
198         } else {
199                 _map.set (ev.GetRow(), d, 1);
200         }
201
202         map_values_changed ();
203 }
204
205 void
206 AudioMappingView::right_click (wxGridEvent& ev)
207 {
208         if (ev.GetCol() == 0) {
209                 return;
210         }
211
212         _menu_row = ev.GetRow ();
213         _menu_column = ev.GetCol ();
214         PopupMenu (_menu, ev.GetPosition ());
215 }
216
217 void
218 AudioMappingView::off ()
219 {
220         _map.set (_menu_row, _menu_column - 1, 0);
221         map_values_changed ();
222 }
223
224 void
225 AudioMappingView::full ()
226 {
227         _map.set (_menu_row, _menu_column - 1, 1);
228         map_values_changed ();
229 }
230
231 void
232 AudioMappingView::minus6dB ()
233 {
234         _map.set (_menu_row, _menu_column - 1, pow (10, -6.0 / 20));
235         map_values_changed ();
236 }
237
238 void
239 AudioMappingView::edit ()
240 {
241         int const d = _menu_column - 1;
242
243         AudioGainDialog* dialog = new AudioGainDialog (this, _menu_row, _menu_column - 1, _map.get (_menu_row, d));
244         if (dialog->ShowModal () == wxID_OK) {
245                 _map.set (_menu_row, d, dialog->value ());
246                 map_values_changed ();
247         }
248
249         dialog->Destroy ();
250 }
251
252 void
253 AudioMappingView::set (AudioMapping map)
254 {
255         _map = map;
256         update_cells ();
257 }
258
259 void
260 AudioMappingView::set_input_channels (vector<string> const & names)
261 {
262         for (int i = 0; i < _grid->GetNumberRows(); ++i) {
263                 _grid->SetCellValue (i, 0, std_to_wx (names[i]));
264         }
265 }
266
267 void
268 AudioMappingView::set_output_channels (vector<string> const & names)
269 {
270         int const o = names.size() + 1;
271         if (o < _grid->GetNumberCols ()) {
272                 _grid->DeleteCols (o, _grid->GetNumberCols() - o);
273         } else if (o > _grid->GetNumberCols ()) {
274                 _grid->InsertCols (_grid->GetNumberCols(), o - _grid->GetNumberCols());
275         }
276
277         _grid->SetColLabelValue (0, wxT (""));
278
279         for (size_t i = 0; i < names.size(); ++i) {
280                 _grid->SetColLabelValue (i + 1, std_to_wx (names[i]));
281         }
282
283         update_cells ();
284         setup_sizes ();
285 }
286
287 void
288 AudioMappingView::update_cells ()
289 {
290         vector<string> row_names;
291         for (int i = 0; i < _grid->GetNumberRows (); ++i) {
292                 row_names.push_back (wx_to_std (_grid->GetCellValue (i, 0)));
293         }
294
295         if (_grid->GetNumberRows ()) {
296                 _grid->DeleteRows (0, _grid->GetNumberRows ());
297         }
298
299         _grid->InsertRows (0, _map.input_channels ());
300
301         for (int i = 0; i < _map.input_channels(); ++i) {
302                 for (int j = 0; j < _map.output_channels(); ++j) {
303                         _grid->SetCellRenderer (i, j + 1, new ValueRenderer);
304                 }
305         }
306
307         for (int i = 0; i < _map.input_channels(); ++i) {
308                 if (i < int (row_names.size ())) {
309                         _grid->SetCellValue (i, 0, std_to_wx (row_names[i]));
310                 }
311                 for (int j = 1; j < _grid->GetNumberCols(); ++j) {
312                         _grid->SetCellValue (i, j, std_to_wx (locale_convert<string> (_map.get (i, j - 1))));
313                 }
314         }
315
316         _grid->AutoSize ();
317 }
318
319 void
320 AudioMappingView::mouse_moved_grid (wxMouseEvent& ev)
321 {
322         int xx;
323         int yy;
324         _grid->CalcUnscrolledPosition (ev.GetX(), ev.GetY(), &xx, &yy);
325
326         int const row = _grid->YToRow (yy);
327         int const column = _grid->XToCol (xx);
328
329         if (row < 0 || column < 1) {
330                 _grid->GetGridWindow()->SetToolTip ("");
331                 _last_tooltip_row = row;
332                 _last_tooltip_column = column;
333         }
334
335         if (row != _last_tooltip_row || column != _last_tooltip_column) {
336
337                 wxString s;
338                 float const gain = _map.get (row, column - 1);
339                 if (gain == 0) {
340                         s = wxString::Format (_("No audio will be passed from content channel %d to DCP channel %d."), row + 1, column);
341                 } else if (gain == 1) {
342                         s = wxString::Format (_("Audio will be passed from content channel %d to DCP channel %d unaltered."), row + 1, column);
343                 } else {
344                         float const dB = 20 * log10 (gain);
345                         s = wxString::Format (_("Audio will be passed from content channel %d to DCP channel %d with gain %.1fdB."), row + 1, column, dB);
346                 }
347
348                 _grid->GetGridWindow()->SetToolTip (s + " " + _("Right click to change gain."));
349                 _last_tooltip_row = row;
350                 _last_tooltip_column = column;
351         }
352
353         ev.Skip ();
354 }
355
356 void
357 AudioMappingView::sized (wxSizeEvent& ev)
358 {
359         setup_sizes ();
360         ev.Skip ();
361 }
362
363 void
364 AudioMappingView::setup_sizes ()
365 {
366         int const top_height = 24;
367
368         _grid->AutoSize ();
369         _left_labels->SetMinSize (wxSize (LEFT_WIDTH, _grid->GetSize().GetHeight()));
370         _top_labels->SetMinSize (wxSize (_grid->GetSize().GetWidth() + LEFT_WIDTH, top_height));
371         /* Try to make the _top_labels 'actual' size respect the minimum we just set */
372         _top_labels->Fit ();
373         _left_labels->Refresh ();
374         _top_labels->Refresh ();
375 }
376
377 void
378 AudioMappingView::paint_left_labels ()
379 {
380         wxPaintDC dc (_left_labels);
381
382         wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
383         if (!gc) {
384                 return;
385         }
386
387         int sx, sy;
388         _grid->CalcUnscrolledPosition (0, 0, &sx, &sy);
389         gc->Translate (0, -sy);
390
391         wxSize const size = dc.GetSize();
392         int const half = size.GetWidth() / 2;
393
394         gc->SetPen (wxPen (wxColour (0, 0, 0)));
395         gc->SetAntialiasMode (wxANTIALIAS_DEFAULT);
396
397         wxGraphicsPath lines = gc->CreatePath();
398
399         vector<pair<int, int> >::const_iterator i = _input_group_positions.begin();
400         if (i != _input_group_positions.end()) {
401                 lines.MoveToPoint (half, i->first);
402                 lines.AddLineToPoint (size.GetWidth(), i->first);
403         }
404
405         vector<Group>::const_iterator j = _input_groups.begin();
406         while (i != _input_group_positions.end() && j != _input_groups.end()) {
407
408                 dc.SetClippingRegion (0, i->first + 2, size.GetWidth(), i->second - 4);
409
410                 dc.SetFont (*wxSWISS_FONT);
411                 wxCoord label_width;
412                 wxCoord label_height;
413                 dc.GetTextExtent (std_to_wx (j->name), &label_width, &label_height);
414
415                 dc.DrawRotatedText (
416                         j->name,
417                         half + (half - label_height) / 2,
418                         min (i->second, (i->second + i->first + label_width) / 2) - sy,
419                         90
420                         );
421
422                 dc.DestroyClippingRegion ();
423
424                 lines.MoveToPoint (half, i->second);
425                 lines.AddLineToPoint (size.GetWidth(), i->second);
426
427                 gc->StrokePath (lines);
428
429                 ++i;
430                 ++j;
431         }
432
433         /* Overall label */
434         dc.SetFont (wxSWISS_FONT->Bold());
435         wxCoord overall_label_width;
436         wxCoord overall_label_height;
437         dc.GetTextExtent (_("Content"), &overall_label_width, &overall_label_height);
438         dc.DrawRotatedText (
439                 _("Content"),
440                 (half - overall_label_height) / 2,
441                 min (size.GetHeight(), (size.GetHeight() + _grid->GetColLabelSize() + overall_label_width) / 2 - sy),
442                 90
443                 );
444
445         delete gc;
446 }
447
448 void
449 AudioMappingView::paint_top_labels ()
450 {
451         wxPaintDC dc (_top_labels);
452         if (_grid->GetNumberCols() == 0) {
453                 return;
454         }
455
456         wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
457         if (!gc) {
458                 return;
459         }
460
461         int sx, sy;
462         _grid->CalcUnscrolledPosition (0, 0, &sx, &sy);
463         gc->Translate (-sx, 0);
464
465         wxSize const size = dc.GetSize();
466
467         gc->SetAntialiasMode (wxANTIALIAS_DEFAULT);
468
469         dc.SetFont (wxSWISS_FONT->Bold());
470         wxCoord label_width;
471         wxCoord label_height;
472         dc.GetTextExtent (_("DCP"), &label_width, &label_height);
473
474         dc.DrawText (_("DCP"), (size.GetWidth() + _grid->GetColSize(0) + LEFT_WIDTH - label_width) / 2 - sx, (size.GetHeight() - label_height) / 2);
475
476         gc->SetPen (wxPen (wxColour (0, 0, 0)));
477         wxGraphicsPath lines = gc->CreatePath();
478         lines.MoveToPoint (LEFT_WIDTH + _grid->GetColSize(0) - 1, 0);
479         lines.AddLineToPoint (LEFT_WIDTH + _grid->GetColSize(0) - 1, size.GetHeight());
480         lines.MoveToPoint (size.GetWidth() - 1, 0);
481         lines.AddLineToPoint (size.GetWidth() - 1, size.GetHeight());
482         gc->StrokePath (lines);
483
484         delete gc;
485 }
486
487 void
488 AudioMappingView::set_input_groups (vector<Group> const & groups)
489 {
490         if (_grid->GetNumberRows() == 0) {
491                 return;
492         }
493
494         _input_groups = groups;
495         _input_group_positions.clear ();
496
497         int ypos = _grid->GetColLabelSize() - 1;
498         BOOST_FOREACH (Group const & i, _input_groups) {
499                 int const old_ypos = ypos;
500                 ypos += (i.to - i.from + 1) * _grid->GetRowSize(0);
501                 _input_group_positions.push_back (make_pair (old_ypos, ypos));
502         }
503 }
504
505 void
506 AudioMappingView::mouse_moved_left_labels (wxMouseEvent& event)
507 {
508         bool done = false;
509         for (size_t i = 0; i < _input_group_positions.size(); ++i) {
510                 if (_input_group_positions[i].first <= event.GetY() && event.GetY() < _input_group_positions[i].second) {
511                         _left_labels->SetToolTip (_input_groups[i].name);
512                         done = true;
513                 }
514         }
515
516         if (!done) {
517                 _left_labels->SetToolTip ("");
518         }
519 }
520
521 void
522 AudioMappingView::grid_scrolled ()
523 {
524         _left_labels->Refresh ();
525         _left_labels->Update ();
526         _top_labels->Refresh ();
527         _top_labels->Update ();
528 }