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