src/wx/audio_plot.h

   1 /*
   2     Copyright (C) 2012-2020 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 #include "lib/util.h"
  22 #include "lib/audio_analysis.h"
  23 #include "lib/warnings.h"
  24 DCPOMATIC_DISABLE_WARNINGS
  25 #include <wx/wx.h>
  26 DCPOMATIC_ENABLE_WARNINGS
  27 #include <boost/shared_ptr.hpp>
  28 #include <boost/signals2.hpp>
  29 #include <vector>
  30
  31 struct Metrics;
  32 class FilmViewer;
  33
  34 class AudioPlot : public wxPanel
  35 {
  36 public:
  37         explicit AudioPlot (wxWindow *, boost::weak_ptr<FilmViewer> viewer);
  38
  39         void set_analysis (boost::shared_ptr<AudioAnalysis>);
  40         void set_channel_visible (int c, bool v);
  41         void set_type_visible (int t, bool v);
  42         void set_smoothing (int);
  43         void set_message (wxString);
  44         void set_gain_correction (double gain);
  45
  46         wxColour colour (int n) const;
  47
  48         boost::signals2::signal<void (boost::optional<dcpomatic::DCPTime>, boost::optional<float>)> Cursor;
  49
  50         static const int max_smoothing;
  51
  52 private:
  53
  54         struct Point {
  55                 Point ()
  56                         : db(0)
  57                 {}
  58
  59                 Point (wxPoint draw_, dcpomatic::DCPTime time_, float db_)
  60                         : draw(draw_)
  61                         , time(time_)
  62                         , db(db_)
  63                 {}
  64
  65                 wxPoint draw;
  66                 dcpomatic::DCPTime time;
  67                 float db;
  68         };
  69
  70         typedef std::vector<Point> PointList;
  71
  72         void paint ();
  73         void plot_peak (wxGraphicsPath &, int, Metrics const &) const;
  74         void plot_rms (wxGraphicsPath &, int, Metrics const &) const;
  75         float y_for_linear (float, Metrics const &) const;
  76         AudioPoint get_point (int channel, int point) const;
  77         void left_down ();
  78         void mouse_moved (wxMouseEvent& ev);
  79         void mouse_leave (wxMouseEvent& ev);
  80         void search (std::map<int, PointList> const & search, wxMouseEvent const & ev, double& min_dist, Point& min_point) const;
  81
  82         boost::weak_ptr<FilmViewer> _viewer;
  83         boost::shared_ptr<AudioAnalysis> _analysis;
  84         bool _channel_visible[MAX_DCP_AUDIO_CHANNELS];
  85         bool _type_visible[AudioPoint::COUNT];
  86         int _smoothing;
  87         std::vector<wxColour> _colours;
  88         wxString _message;
  89         float _gain_correction;
  90
  91         mutable std::map<int, PointList> _peak;
  92         mutable std::map<int, PointList> _rms;
  93
  94         boost::optional<Point> _cursor;
  95
  96         static const int _minimum;
  97         static const int _cursor_size;
  98 };