701c8263a48b27e8a8391da28865eae1fb516dc1
[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, wxALL, 12);
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, 0, wxALL, 12);
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         SetTitle (String::compose ("DVD-o-matic audio - %1", _film->name()));
87 }
88
89 void
90 AudioDialog::setup_channels ()
91 {
92         if (!_film->audio_stream()) {
93                 return;
94         }
95
96         AudioMapping m (_film->audio_stream()->channels ());
97         
98         for (int i = 0; i < MAX_AUDIO_CHANNELS; ++i) {
99                 if (m.dcp_to_source(static_cast<libdcp::Channel>(i))) {
100                         _channel_checkbox[i]->Show ();
101                 } else {
102                         _channel_checkbox[i]->Hide ();
103                 }
104         }
105 }       
106
107 void
108 AudioDialog::try_to_load_analysis ()
109 {
110         shared_ptr<AudioAnalysis> a;
111
112         if (boost::filesystem::exists (_film->audio_analysis_path())) {
113                 a.reset (new AudioAnalysis (_film->audio_analysis_path ()));
114         } else {
115                 if (IsShown ()) {
116                         _film->analyse_audio ();
117                 }
118         }
119                 
120         _plot->set_analysis (a);
121
122         AudioMapping m (_film->audio_stream()->channels ());
123         optional<libdcp::Channel> c = m.source_to_dcp (0);
124         if (c) {
125                 _channel_checkbox[c.get()]->SetValue (true);
126                 _plot->set_channel_visible (0, true);
127         }
128
129         for (int i = 0; i < AudioPoint::COUNT; ++i) {
130                 _type_checkbox[i]->SetValue (true);
131                 _plot->set_type_visible (i, true);
132         }
133 }
134
135 void
136 AudioDialog::channel_clicked (wxCommandEvent& ev)
137 {
138         int c = 0;
139         while (c < MAX_AUDIO_CHANNELS && ev.GetEventObject() != _channel_checkbox[c]) {
140                 ++c;
141         }
142
143         assert (c < MAX_AUDIO_CHANNELS);
144
145         AudioMapping m (_film->audio_stream()->channels ());
146         optional<int> s = m.dcp_to_source (static_cast<libdcp::Channel> (c));
147         if (s) {
148                 _plot->set_channel_visible (s.get(), _channel_checkbox[c]->GetValue ());
149         }
150 }
151
152 void
153 AudioDialog::film_changed (Film::Property p)
154 {
155         switch (p) {
156         case Film::AUDIO_GAIN:
157                 _plot->set_gain (_film->audio_gain ());
158                 break;
159         case Film::CONTENT_AUDIO_STREAM:
160         case Film::EXTERNAL_AUDIO:
161         case Film::USE_CONTENT_AUDIO:
162                 setup_channels ();
163                 break;
164         default:
165                 break;
166         }
167 }
168
169 void
170 AudioDialog::type_clicked (wxCommandEvent& ev)
171 {
172         int t = 0;
173         while (t < AudioPoint::COUNT && ev.GetEventObject() != _type_checkbox[t]) {
174                 ++t;
175         }
176
177         assert (t < AudioPoint::COUNT);
178
179         _plot->set_type_visible (t, _type_checkbox[t]->GetValue ());
180 }