Peak value of audio into the audio tab.
[dcpomatic.git] / src / wx / audio_dialog.cc
index 32864ca15974a9c400bb88da7c27efcbc75a17a0..27b34540611dd76cf94a0af46b1649022e1efe20 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
 
 #include "audio_dialog.h"
 #include "audio_plot.h"
-#include "audio_analysis.h"
-#include "film.h"
 #include "wx_util.h"
+#include "lib/audio_analysis.h"
+#include "lib/film.h"
+#include "lib/analyse_audio_job.h"
+#include "lib/audio_content.h"
+#include "lib/job_manager.h"
+#include <libxml++/libxml++.h>
+#include <boost/filesystem.hpp>
 
+using std::cout;
 using boost::shared_ptr;
+using boost::bind;
+using boost::optional;
+using boost::const_pointer_cast;
+using boost::dynamic_pointer_cast;
 
-AudioDialog::AudioDialog (wxWindow* parent, boost::shared_ptr<Film> film)
-       : wxDialog (parent, wxID_ANY, _("Audio"), wxDefaultPosition, wxSize (640, 512), wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)
+/** @param content Content to analyse, or 0 to analyse all of the film's audio */
+AudioDialog::AudioDialog (wxWindow* parent, shared_ptr<Film> film, shared_ptr<AudioContent> content)
+       : wxDialog (parent, wxID_ANY, _("Audio"), wxDefaultPosition, wxSize (640, 512), wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER | wxFULL_REPAINT_ON_RESIZE)
+       , _film (film)
        , _plot (0)
 {
-       wxBoxSizer* sizer = new wxBoxSizer (wxHORIZONTAL);
+       wxFont subheading_font (*wxNORMAL_FONT);
+       subheading_font.SetWeight (wxFONTWEIGHT_BOLD);
+
+       wxBoxSizer* overall_sizer = new wxBoxSizer (wxVERTICAL);
+       wxBoxSizer* lr_sizer = new wxBoxSizer (wxHORIZONTAL);
+
+       wxBoxSizer* left = new wxBoxSizer (wxVERTICAL);
 
        _plot = new AudioPlot (this);
-       sizer->Add (_plot, 1);
+       left->Add (_plot, 1, wxALL | wxEXPAND, 12);
+       _peak_time = new wxStaticText (this, wxID_ANY, wxT (""));
+       left->Add (_peak_time, 0, wxALL, 12);
+
+       lr_sizer->Add (left, 1, wxALL, 12);
+
+       wxBoxSizer* right = new wxBoxSizer (wxVERTICAL);
+
+       {
+               wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Channels"));
+               m->SetFont (subheading_font);
+               right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM, 16);
+       }
+
+       for (int i = 0; i < MAX_DCP_AUDIO_CHANNELS; ++i) {
+               _channel_checkbox[i] = new wxCheckBox (this, wxID_ANY, std_to_wx (audio_channel_name (i)));
+               right->Add (_channel_checkbox[i], 0, wxEXPAND | wxALL, 3);
+               _channel_checkbox[i]->Bind (wxEVT_COMMAND_CHECKBOX_CLICKED, boost::bind (&AudioDialog::channel_clicked, this, _1));
+       }
 
-       wxFlexGridSizer* table = new wxFlexGridSizer (2, 6, 6);
+       {
+               wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Type"));
+               m->SetFont (subheading_font);
+               right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP, 16);
+       }
+
+       wxString const types[] = {
+               _("Peak"),
+               _("RMS")
+       };
+
+       for (int i = 0; i < AudioPoint::COUNT; ++i) {
+               _type_checkbox[i] = new wxCheckBox (this, wxID_ANY, types[i]);
+               right->Add (_type_checkbox[i], 0, wxEXPAND | wxALL, 3);
+               _type_checkbox[i]->Bind (wxEVT_COMMAND_CHECKBOX_CLICKED, boost::bind (&AudioDialog::type_clicked, this, _1));
+       }
 
-       add_label_to_sizer (table, this, _("Channel"));
-       _channel = new wxChoice (this, wxID_ANY);
-       table->Add (_channel, 1, wxEXPAND | wxALL, 6);
+       {
+               wxStaticText* m = new wxStaticText (this, wxID_ANY, _("Smoothing"));
+               m->SetFont (subheading_font);
+               right->Add (m, 1, wxALIGN_CENTER_VERTICAL | wxTOP, 16);
+       }
+
+       _smoothing = new wxSlider (this, wxID_ANY, AudioPlot::max_smoothing / 2, 1, AudioPlot::max_smoothing);
+       _smoothing->Bind (wxEVT_SCROLL_THUMBTRACK, boost::bind (&AudioDialog::smoothing_changed, this));
+       right->Add (_smoothing, 0, wxEXPAND);
 
-       sizer->Add (table);
+       lr_sizer->Add (right, 0, wxALL, 12);
 
-       set_film (film);
+       overall_sizer->Add (lr_sizer);
+
+#ifdef DCPOMATIC_LINUX
+       wxSizer* buttons = CreateSeparatedButtonSizer (wxCLOSE);
+       if (buttons) {
+               overall_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
+       }
+#endif
 
-       SetSizer (sizer);
-       sizer->Layout ();
-       sizer->SetSizeHints (this);
+       SetSizer (overall_sizer);
+       overall_sizer->Layout ();
+       overall_sizer->SetSizeHints (this);
 
-       _channel->Connect (wxID_ANY, wxEVT_COMMAND_CHOICE_SELECTED, wxCommandEventHandler (AudioDialog::channel_changed), 0, this);
+       _film_connection = film->ContentChanged.connect (boost::bind (&AudioDialog::content_changed, this, _2));
+       SetTitle (_("DCP-o-matic audio"));
+
+       if (content) {
+               _playlist.reset (new Playlist ());
+               const_pointer_cast<Playlist> (_playlist)->add (content);
+       } else {
+               _playlist = film->playlist ();
+       }
 }
 
 void
-AudioDialog::set_film (boost::shared_ptr<Film> f)
+AudioDialog::try_to_load_analysis ()
 {
-       _film_connection.disconnect ();
-       _film = f;
-       
-       shared_ptr<AudioAnalysis> a;
+       if (!IsShown ()) {
+               return;
+       }
+
+       shared_ptr<const Film> film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+
+       boost::filesystem::path const path = film->audio_analysis_path (_playlist);
+       if (!boost::filesystem::exists (path)) {
+               _plot->set_analysis (shared_ptr<AudioAnalysis> ());
+               _analysis.reset ();
+               JobManager::instance()->analyse_audio (film, _playlist, _analysis_finished_connection, bind (&AudioDialog::analysis_finished, this));
+               return;
+       }
 
        try {
-               a.reset (new AudioAnalysis (f->audio_analysis_path ()));
-       } catch (...) {
+               _analysis.reset (new AudioAnalysis (path));
+       } catch (xmlpp::exception& e) {
+               /* Probably an old-style analysis file: recreate it */
+               JobManager::instance()->analyse_audio (film, _playlist, _analysis_finished_connection, bind (&AudioDialog::analysis_finished, this));
+               return;
+        }
+
+       _plot->set_analysis (_analysis);
+       _plot->set_gain_correction (_analysis->gain_correction (_playlist));
+       setup_peak_time ();
+
+       /* Set up some defaults if no check boxes are checked */
+
+       int i = 0;
+       while (i < MAX_DCP_AUDIO_CHANNELS && (!_channel_checkbox[i] || !_channel_checkbox[i]->GetValue ())) {
+               ++i;
+       }
+
+       if (i == MAX_DCP_AUDIO_CHANNELS && _channel_checkbox[0]) {
+               _channel_checkbox[0]->SetValue (true);
+               _plot->set_channel_visible (0, true);
+       }
+
+       i = 0;
+       while (i < AudioPoint::COUNT && !_type_checkbox[i]->GetValue ()) {
+               i++;
+       }
+
+       if (i == AudioPoint::COUNT) {
+               for (int i = 0; i < AudioPoint::COUNT; ++i) {
+                       _type_checkbox[i]->SetValue (true);
+                       _plot->set_type_visible (i, true);
+               }
+       }
+
+       Refresh ();
+}
+
+void
+AudioDialog::analysis_finished ()
+{
+       shared_ptr<const Film> film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
 
+       if (!boost::filesystem::exists (film->audio_analysis_path (_playlist))) {
+               /* We analysed and still nothing showed up, so maybe it was cancelled or it failed.
+                  Give up.
+               */
+               _plot->set_message (_("Could not analyse audio."));
+               return;
        }
-               
-       _plot->set_analysis (a);
 
-       _channel->Clear ();
-       for (int i = 0; i < f->audio_stream()->channels(); ++i) {
-               _channel->Append (audio_channel_name (i));
+       try_to_load_analysis ();
+}
+
+void
+AudioDialog::channel_clicked (wxCommandEvent& ev)
+{
+       int c = 0;
+       while (c < MAX_DCP_AUDIO_CHANNELS && ev.GetEventObject() != _channel_checkbox[c]) {
+               ++c;
        }
 
-       _channel->SetSelection (0);
+       DCPOMATIC_ASSERT (c < MAX_DCP_AUDIO_CHANNELS);
 
-       _film_connection = f->Changed.connect (bind (&AudioDialog::film_changed, this, _1));
+       _plot->set_channel_visible (c, _channel_checkbox[c]->GetValue ());
 }
 
 void
-AudioDialog::channel_changed (wxCommandEvent &)
+AudioDialog::content_changed (int p)
 {
-       _plot->set_channel (_channel->GetSelection ());
+       if (p == AudioContentProperty::AUDIO_STREAMS) {
+               try_to_load_analysis ();
+       } else if (p == AudioContentProperty::AUDIO_GAIN) {
+               if (_playlist->content().size() == 1) {
+                       /* We can use a short-cut to render the effect of this
+                          change, rather than recalculating everything.
+                       */
+                       _plot->set_gain_correction (_analysis->gain_correction (_playlist));
+                       setup_peak_time ();
+               } else {
+                       try_to_load_analysis ();
+               }
+       }
 }
 
 void
-AudioDialog::film_changed (Film::Property p)
+AudioDialog::type_clicked (wxCommandEvent& ev)
 {
-       if (p == Film::AUDIO_GAIN) {
-               _plot->set_gain (_film->audio_gain ());
+       int t = 0;
+       while (t < AudioPoint::COUNT && ev.GetEventObject() != _type_checkbox[t]) {
+               ++t;
        }
+
+       DCPOMATIC_ASSERT (t < AudioPoint::COUNT);
+
+       _plot->set_type_visible (t, _type_checkbox[t]->GetValue ());
+}
+
+void
+AudioDialog::smoothing_changed ()
+{
+       _plot->set_smoothing (_smoothing->GetValue ());
+}
+
+void
+AudioDialog::setup_peak_time ()
+{
+       if (!_analysis || !_analysis->peak ()) {
+               return;
+       }
+
+       shared_ptr<Film> film = _film.lock ();
+       if (!film) {
+               return;
+       }
+
+       float const peak_dB = 20 * log10 (_analysis->peak().get()) + _analysis->gain_correction (_playlist);
+
+       _peak_time->SetLabel (
+               wxString::Format (
+                       _("Peak is %.2fdB at %s"),
+                       peak_dB,
+                       time_to_timecode (_analysis->peak_time().get(), film->video_frame_rate ()).data ()
+                       )
+               );
+
+       if (peak_dB > -3) {
+               _peak_time->SetForegroundColour (wxColour (255, 0, 0));
+       } else {
+               _peak_time->SetForegroundColour (wxColour (0, 0, 0));
+       }
+}
+
+bool
+AudioDialog::Show (bool show)
+{
+       bool const r = wxDialog::Show (show);
+       try_to_load_analysis ();
+       return r;
 }