Try to tidy up channel mapping a bit.
[dcpomatic.git] / src / wx / audio_dialog.cc
1 /*
2     Copyright (C) 2013 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/filesystem.hpp>
21 #include "audio_dialog.h"
22 #include "audio_plot.h"
23 #include "audio_analysis.h"
24 #include "film.h"
25 #include "wx_util.h"
26
27 using boost::shared_ptr;
28 using boost::bind;
29 using boost::optional;
30
31 AudioDialog::AudioDialog (wxWindow* parent)
32         : wxDialog (parent, wxID_ANY, _("Audio"), wxDefaultPosition, wxSize (640, 512), wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)
33         , _plot (0)
34 {
35         wxBoxSizer* sizer = new wxBoxSizer (wxHORIZONTAL);
36
37         _plot = new AudioPlot (this);
38         sizer->Add (_plot, 1);
39
40         wxFlexGridSizer* table = new wxFlexGridSizer (2, 6, 6);
41
42         for (int i = 0; i < MAX_AUDIO_CHANNELS; ++i) {
43                 _channel_checkbox[i] = new wxCheckBox (this, wxID_ANY, audio_channel_name (i));
44                 table->Add (_channel_checkbox[i], 1, wxEXPAND);
45                 table->AddSpacer (0);
46                 _channel_checkbox[i]->Connect (wxID_ANY, wxEVT_COMMAND_CHECKBOX_CLICKED, wxCommandEventHandler (AudioDialog::channel_clicked), 0, this);
47         }
48
49         table->AddSpacer (0);
50         table->AddSpacer (0);
51
52         wxString const types[] = {
53                 _("Peak"),
54                 _("RMS")
55         };
56
57         for (int i = 0; i < AudioPoint::COUNT; ++i) {
58                 _type_checkbox[i] = new wxCheckBox (this, wxID_ANY, types[i]);
59                 table->Add (_type_checkbox[i], 1, wxEXPAND);
60                 table->AddSpacer (0);
61                 _type_checkbox[i]->Connect (wxID_ANY, wxEVT_COMMAND_CHECKBOX_CLICKED, wxCommandEventHandler (AudioDialog::type_clicked), 0, this);
62         }
63
64         sizer->Add (table);
65
66         SetSizer (sizer);
67         sizer->Layout ();
68         sizer->SetSizeHints (this);
69 }
70
71 void
72 AudioDialog::set_film (boost::shared_ptr<Film> f)
73 {
74         _film_changed_connection.disconnect ();
75         _film_audio_analysis_finished_connection.disconnect ();
76         
77         _film = f;
78
79         try_to_load_analysis ();
80         setup_channels ();
81         _plot->set_gain (_film->audio_gain ());
82
83         _film_changed_connection = _film->Changed.connect (bind (&AudioDialog::film_changed, this, _1));
84         _film_audio_analysis_finished_connection = _film->AudioAnalysisFinished.connect (bind (&AudioDialog::try_to_load_analysis, this));
85 }
86
87 void
88 AudioDialog::setup_channels ()
89 {
90         if (!_film->audio_stream()) {
91                 return;
92         }
93
94         AudioMapping m (_film->audio_stream()->channels ());
95         
96         for (int i = 0; i < MAX_AUDIO_CHANNELS; ++i) {
97                 if (m.dcp_to_source(static_cast<libdcp::Channel>(i))) {
98                         _channel_checkbox[i]->Show ();
99                 } else {
100                         _channel_checkbox[i]->Hide ();
101                 }
102         }
103 }       
104
105 void
106 AudioDialog::try_to_load_analysis ()
107 {
108         shared_ptr<AudioAnalysis> a;
109
110         if (boost::filesystem::exists (_film->audio_analysis_path())) {
111                 a.reset (new AudioAnalysis (_film->audio_analysis_path ()));
112         } else {
113                 if (IsShown ()) {
114                         _film->analyse_audio ();
115                 }
116         }
117                 
118         _plot->set_analysis (a);
119
120         AudioMapping m (_film->audio_stream()->channels ());
121         optional<libdcp::Channel> c = m.source_to_dcp (0);
122         if (c) {
123                 _channel_checkbox[c.get()]->SetValue (true);
124                 _plot->set_channel_visible (0, true);
125         }
126
127         for (int i = 0; i < AudioPoint::COUNT; ++i) {
128                 _type_checkbox[i]->SetValue (true);
129                 _plot->set_type_visible (i, true);
130         }
131 }
132
133 void
134 AudioDialog::channel_clicked (wxCommandEvent& ev)
135 {
136         int c = 0;
137         while (c < MAX_AUDIO_CHANNELS && ev.GetEventObject() != _channel_checkbox[c]) {
138                 ++c;
139         }
140
141         assert (c < MAX_AUDIO_CHANNELS);
142
143         AudioMapping m (_film->audio_stream()->channels ());
144         optional<int> s = m.dcp_to_source (static_cast<libdcp::Channel> (c));
145         if (s) {
146                 _plot->set_channel_visible (s.get(), _channel_checkbox[c]->GetValue ());
147         }
148 }
149
150 void
151 AudioDialog::film_changed (Film::Property p)
152 {
153         switch (p) {
154         case Film::AUDIO_GAIN:
155                 _plot->set_gain (_film->audio_gain ());
156                 break;
157         case Film::CONTENT_AUDIO_STREAM:
158         case Film::EXTERNAL_AUDIO:
159         case Film::USE_CONTENT_AUDIO:
160                 setup_channels ();
161                 break;
162         default:
163                 break;
164         }
165 }
166
167 void
168 AudioDialog::type_clicked (wxCommandEvent& ev)
169 {
170         int t = 0;
171         while (t < AudioPoint::COUNT && ev.GetEventObject() != _type_checkbox[t]) {
172                 ++t;
173         }
174
175         assert (t < AudioPoint::COUNT);
176
177         _plot->set_type_visible (t, _type_checkbox[t]->GetValue ());
178 }