AnalyseAudioJob does not need an explicit playlist.
[dcpomatic.git] / src / wx / audio_dialog.cc
1 /*
2     Copyright (C) 2013-2015 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 "audio_dialog.h"
21 #include "audio_plot.h"
22 #include "wx_util.h"
23 #include "lib/audio_analysis.h"
24 #include "lib/film.h"
25 #include "lib/analyse_audio_job.h"
26 #include "lib/job_manager.h"
27 #include "lib/playlist.h"
28 #include <boost/filesystem.hpp>
29
30 using boost::shared_ptr;
31 using boost::bind;
32 using boost::optional;
33
34 AudioDialog::AudioDialog (wxWindow* parent, shared_ptr<Film> film)
35         : wxDialog (parent, wxID_ANY, _("Audio"), wxDefaultPosition, wxSize (640, 512), wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE)
36         , _film (film)
37         , _plot (0)
38 {
39         wxFont subheading_font (*wxNORMAL_FONT);
40         subheading_font.SetWeight (wxFONTWEIGHT_BOLD);
41
42         wxBoxSizer* overall_sizer = new wxBoxSizer (wxVERTICAL);
43         wxBoxSizer* lr_sizer = new wxBoxSizer (wxHORIZONTAL);
44         
45         wxBoxSizer* left = new wxBoxSizer (wxVERTICAL);
46
47         _plot = new AudioPlot (this);
48         left->Add (_plot, 1, wxALL | wxEXPAND, 12);
49         _peak_time = new wxStaticText (this, wxID_ANY, wxT (""));
50         left->Add (_peak_time, 0, wxALL, 12);
51
52         lr_sizer->Add (left, 1, wxALL, 12);
53
54         wxBoxSizer* right = new wxBoxSizer (wxVERTICAL);
55
56         {
57                 wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Channels"));
58                 m->SetFont (subheading_font);
59                 right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM, 16);
60         }
61
62         for (int i = 0; i < MAX_DCP_AUDIO_CHANNELS; ++i) {
63                 _channel_checkbox[i] = new wxCheckBox (this, wxID_ANY, std_to_wx (audio_channel_name (i)));
64                 right->Add (_channel_checkbox[i], 0, wxEXPAND | wxALL, 3);
65                 _channel_checkbox[i]->Bind (wxEVT_COMMAND_CHECKBOX_CLICKED, boost::bind (&AudioDialog::channel_clicked, this, _1));
66         }
67
68         {
69                 wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Type"));
70                 m->SetFont (subheading_font);
71                 right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP, 16);
72         }
73         
74         wxString const types[] = {
75                 _("Peak"),
76                 _("RMS")
77         };
78
79         for (int i = 0; i < AudioPoint::COUNT; ++i) {
80                 _type_checkbox[i] = new wxCheckBox (this, wxID_ANY, types[i]);
81                 right->Add (_type_checkbox[i], 0, wxEXPAND | wxALL, 3);
82                 _type_checkbox[i]->Bind (wxEVT_COMMAND_CHECKBOX_CLICKED, boost::bind (&AudioDialog::type_clicked, this, _1));
83         }
84
85         {
86                 wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Smoothing"));
87                 m->SetFont (subheading_font);
88                 right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP, 16);
89         }
90         
91         _smoothing = new wxSlider (this, wxID_ANY, AudioPlot::max_smoothing / 2, 1, AudioPlot::max_smoothing);
92         _smoothing->Bind (wxEVT_SCROLL_THUMBTRACK, boost::bind (&AudioDialog::smoothing_changed, this));
93         right->Add (_smoothing, 0, wxEXPAND);
94
95         lr_sizer->Add (right, 0, wxALL, 12);
96
97         overall_sizer->Add (lr_sizer);
98
99 #ifdef DCPOMATIC_LINUX  
100         wxSizer* buttons = CreateSeparatedButtonSizer (wxCLOSE);
101         if (buttons) {
102                 overall_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
103         }
104 #endif  
105
106         SetSizer (overall_sizer);
107         overall_sizer->Layout ();
108         overall_sizer->SetSizeHints (this);
109 }
110
111 void
112 AudioDialog::set_playlist (shared_ptr<const Playlist> p)
113 {
114         _playlist_connection.disconnect ();
115         _playlist = p;
116         _playlist_connection = _playlist->ContentChanged.connect (boost::bind (&AudioDialog::try_to_load_analysis, this));
117         try_to_load_analysis ();
118         SetTitle (_("DCP-o-matic audio"));
119 }
120
121 void
122 AudioDialog::try_to_load_analysis ()
123 {
124         if (!IsShown ()) {
125                 return;
126         }
127
128         shared_ptr<const Film> film = _film.lock ();
129         DCPOMATIC_ASSERT (film);
130
131         boost::filesystem::path path = film->audio_analysis_path ();
132
133         if (!boost::filesystem::exists (path)) {
134                 _plot->set_analysis (shared_ptr<AudioAnalysis> ());
135                 _analysis.reset ();
136                 shared_ptr<AnalyseAudioJob> job (new AnalyseAudioJob (film));
137                 _analysis_finished_connection = job->Finished.connect (bind (&AudioDialog::analysis_finished, this));
138                 JobManager::instance()->add (job);
139                 return;
140         }
141
142         try {
143                 _analysis.reset (new AudioAnalysis (path));
144         } catch (xmlpp::exception& e) {
145                 /* Probably an old-style analysis file: recreate it */
146                 shared_ptr<AnalyseAudioJob> job (new AnalyseAudioJob (film));
147                 _analysis_finished_connection = job->Finished.connect (bind (&AudioDialog::analysis_finished, this));
148                 JobManager::instance()->add (job);
149                 return;
150         }
151         
152         _plot->set_analysis (_analysis);
153         setup_peak_time ();
154
155         /* Set up some defaults if no check boxes are checked */
156         
157         int i = 0;
158         while (i < MAX_DCP_AUDIO_CHANNELS && (!_channel_checkbox[i] || !_channel_checkbox[i]->GetValue ())) {
159                 ++i;
160         }
161
162         if (i == MAX_DCP_AUDIO_CHANNELS && _channel_checkbox[0]) {
163                 _channel_checkbox[0]->SetValue (true);
164                 _plot->set_channel_visible (0, true);
165         }
166
167         i = 0;
168         while (i < AudioPoint::COUNT && !_type_checkbox[i]->GetValue ()) {
169                 i++;
170         }
171
172         if (i == AudioPoint::COUNT) {
173                 for (int i = 0; i < AudioPoint::COUNT; ++i) {
174                         _type_checkbox[i]->SetValue (true);
175                         _plot->set_type_visible (i, true);
176                 }
177         }
178
179         Refresh ();
180 }
181
182 void
183 AudioDialog::analysis_finished ()
184 {
185         shared_ptr<const Film> film = _film.lock ();
186         DCPOMATIC_ASSERT (film);
187         
188         if (!boost::filesystem::exists (film->audio_analysis_path ())) {
189                 /* We analysed and still nothing showed up, so maybe it was cancelled or it failed.
190                    Give up.
191                 */
192                 _plot->set_message (_("Could not analyse audio."));
193                 return;
194         }
195
196         try_to_load_analysis ();
197 }
198
199 void
200 AudioDialog::channel_clicked (wxCommandEvent& ev)
201 {
202         int c = 0;
203         while (c < MAX_DCP_AUDIO_CHANNELS && ev.GetEventObject() != _channel_checkbox[c]) {
204                 ++c;
205         }
206
207         DCPOMATIC_ASSERT (c < MAX_DCP_AUDIO_CHANNELS);
208
209         _plot->set_channel_visible (c, _channel_checkbox[c]->GetValue ());
210 }
211
212 void
213 AudioDialog::content_changed (int p)
214 {
215         if (p == AudioContentProperty::AUDIO_GAIN || p == AudioContentProperty::AUDIO_STREAMS) {
216                 try_to_load_analysis ();
217         }
218 }
219
220 void
221 AudioDialog::type_clicked (wxCommandEvent& ev)
222 {
223         int t = 0;
224         while (t < AudioPoint::COUNT && ev.GetEventObject() != _type_checkbox[t]) {
225                 ++t;
226         }
227
228         DCPOMATIC_ASSERT (t < AudioPoint::COUNT);
229
230         _plot->set_type_visible (t, _type_checkbox[t]->GetValue ());
231 }
232
233 void
234 AudioDialog::smoothing_changed ()
235 {
236         _plot->set_smoothing (_smoothing->GetValue ());
237 }
238
239 void
240 AudioDialog::setup_peak_time ()
241 {
242         if (!_analysis || !_analysis->peak ()) {
243                 return;
244         }
245         
246         shared_ptr<Film> film = _film.lock ();
247         if (!film) {
248                 return;
249         }
250         
251         float peak_dB = 20 * log10 (_analysis->peak().get());
252         
253         _peak_time->SetLabel (
254                 wxString::Format (
255                         _("Peak is %.2fdB at %s"),
256                         peak_dB,
257                         time_to_timecode (_analysis->peak_time().get(), film->video_frame_rate ()).data ()
258                         )
259                 );
260         
261         if (peak_dB > -3) {
262                 _peak_time->SetForegroundColour (wxColour (255, 0, 0));
263         } else {
264                 _peak_time->SetForegroundColour (wxColour (0, 0, 0));
265         }
266 }