src/wx/audio_plot.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 <iostream>
  21 #include <boost/bind.hpp>
  22 #include <wx/graphics.h>
  23 #include "audio_plot.h"
  24 #include "lib/decoder_factory.h"
  25 #include "lib/audio_decoder.h"
  26 #include "lib/audio_analysis.h"
  27 #include "wx/wx_util.h"
  28
  29 using std::cout;
  30 using std::vector;
  31 using std::max;
  32 using std::min;
  33 using boost::bind;
  34 using boost::shared_ptr;
  35
  36 AudioPlot::AudioPlot (wxWindow* parent)
  37         : wxPanel (parent)
  38         , _channel (0)
  39         , _gain (0)
  40 {
  41         Connect (wxID_ANY, wxEVT_PAINT, wxPaintEventHandler (AudioPlot::paint), 0, this);
  42
  43         SetMinSize (wxSize (640, 512));
  44 }
  45
  46 void
  47 AudioPlot::set_analysis (shared_ptr<AudioAnalysis> a)
  48 {
  49         _analysis = a;
  50         _channel = 0;
  51         Refresh ();
  52 }
  53
  54 void
  55 AudioPlot::set_channel (int c)
  56 {
  57         _channel = c;
  58         Refresh ();
  59 }
  60
  61 void
  62 AudioPlot::paint (wxPaintEvent &)
  63 {
  64         wxPaintDC dc (this);
  65
  66         if (!_analysis) {
  67                 return;
  68         }
  69
  70         wxGraphicsContext* gc = wxGraphicsContext::Create (dc);
  71         if (!gc) {
  72                 return;
  73         }
  74
  75         int const width = GetSize().GetWidth();
  76         float const xs = width / float (_analysis->points (_channel));
  77         int const height = GetSize().GetHeight ();
  78         float const ys = height / 60;
  79
  80         wxGraphicsPath grid = gc->CreatePath ();
  81         gc->SetFont (gc->CreateFont (*wxSMALL_FONT));
  82         for (int i = -60; i <= 0; i += 10) {
  83                 int const y = height - (i + 60) * ys;
  84                 grid.MoveToPoint (0, y);
  85                 grid.AddLineToPoint (width, y);
  86                 gc->DrawText (std_to_wx (String::compose ("%1dB", i)), width - 32, y - 12);
  87         }
  88         gc->SetPen (*wxLIGHT_GREY_PEN);
  89         gc->StrokePath (grid);
  90
  91         wxGraphicsPath path[AudioPoint::COUNT];
  92
  93         for (int i = 0; i < AudioPoint::COUNT; ++i) {
  94                 path[i] = gc->CreatePath ();
  95                 path[i].MoveToPoint (0, height - (max (_analysis->get_point(_channel, 0)[i], -60.0f) + 60 + _gain) * ys);
  96         }
  97
  98         for (int i = 0; i < _analysis->points(_channel); ++i) {
  99                 for (int j = 0; j < AudioPoint::COUNT; ++j) {
 100                         path[j].AddLineToPoint (i * xs, height - (max (_analysis->get_point(_channel, i)[j], -60.0f) + 60 + _gain) * ys);
 101                 }
 102         }
 103
 104         gc->SetPen (*wxBLUE_PEN);
 105         gc->StrokePath (path[AudioPoint::RMS]);
 106
 107         gc->SetPen (*wxRED_PEN);
 108         gc->StrokePath (path[AudioPoint::PEAK]);
 109
 110         delete gc;
 111 }
 112
 113 void
 114 AudioPlot::set_gain (float g)
 115 {
 116         _gain = g;
 117         Refresh ();
 118 }