src/lib/upmixer_a.cc

   1 /*
   2     Copyright (C) 2014-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 "upmixer_a.h"
  21 #include "audio_buffers.h"
  22 #include "audio_mapping.h"
  23
  24 #include "i18n.h"
  25
  26 using std::string;
  27 using std::min;
  28 using std::vector;
  29 using boost::shared_ptr;
  30
  31 UpmixerA::UpmixerA (int sampling_rate)
  32         : _left (0.02, 1900.0 / sampling_rate, 4800.0 / sampling_rate)
  33         , _right (0.02, 1900.0 / sampling_rate, 4800.0 / sampling_rate)
  34         , _centre (0.02, 150.0 / sampling_rate, 1900.0 / sampling_rate)
  35         , _lfe (0.02, 20.0 / sampling_rate, 150.0 / sampling_rate)
  36         , _ls (0.02, 4800.0 / sampling_rate, 20000.0 / sampling_rate)
  37         , _rs (0.02, 4800.0 / sampling_rate, 20000.0 / sampling_rate)
  38 {
  39
  40 }
  41
  42 string
  43 UpmixerA::name () const
  44 {
  45         return _("Stereo to 5.1 up-mixer A");
  46 }
  47
  48
  49 string
  50 UpmixerA::id () const
  51 {
  52         return N_("stereo-5.1-upmix-a");
  53 }
  54
  55 ChannelCount
  56 UpmixerA::in_channels () const
  57 {
  58         return ChannelCount (2);
  59 }
  60
  61 int
  62 UpmixerA::out_channels () const
  63 {
  64         return 6;
  65 }
  66
  67 shared_ptr<AudioProcessor>
  68 UpmixerA::clone (int sampling_rate) const
  69 {
  70         return shared_ptr<AudioProcessor> (new UpmixerA (sampling_rate));
  71 }
  72
  73 shared_ptr<AudioBuffers>
  74 UpmixerA::run (shared_ptr<const AudioBuffers> in, int channels)
  75 {
  76         /* Input L and R */
  77         shared_ptr<AudioBuffers> in_L = in->channel (0);
  78         shared_ptr<AudioBuffers> in_R = in->channel (1);
  79
  80         /* Mix of L and R */
  81         shared_ptr<AudioBuffers> in_LR = in_L->clone ();
  82         in_LR->accumulate_frames (in_R.get(), 0, 0, in_R->frames ());
  83         in_LR->apply_gain (0.5);
  84
  85         /* Run filters */
  86         vector<shared_ptr<AudioBuffers> > all_out;
  87         all_out.push_back (_left.run (in_L));
  88         all_out.push_back (_right.run (in_R));
  89         all_out.push_back (_centre.run (in_LR));
  90         all_out.push_back (_lfe.run (in_LR));
  91         all_out.push_back (_ls.run (in_L));
  92         all_out.push_back (_rs.run (in_R));
  93
  94         shared_ptr<AudioBuffers> out (new AudioBuffers (channels, in->frames ()));
  95         int const N = min (channels, 6);
  96
  97         for (int i = 0; i < N; ++i) {
  98                 out->copy_channel_from (all_out[i].get(), 0, i);
  99         }
 100
 101         for (int i = N; i < channels; ++i) {
 102                 out->make_silent (i);
 103         }
 104
 105         return out;
 106 }
 107
 108 void
 109 UpmixerA::flush ()
 110 {
 111         _left.flush ();
 112         _right.flush ();
 113         _centre.flush ();
 114         _lfe.flush ();
 115         _ls.flush ();
 116         _rs.flush ();
 117 }
 118
 119 void
 120 UpmixerA::make_audio_mapping_default (AudioMapping& mapping) const
 121 {
 122         /* Just map the first two input channels to our L/R */
 123         mapping.make_zero ();
 124         for (int i = 0; i < min (2, mapping.input_channels()); ++i) {
 125                 mapping.set (i, i, 1);
 126         }
 127 }
 128
 129 vector<string>
 130 UpmixerA::input_names () const
 131 {
 132         vector<string> n;
 133         n.push_back (_("Upmix L"));
 134         n.push_back (_("Upmix R"));
 135         return n;
 136 }