ecbe1cb58a530161b0870cb7bce3c69dc25bece6
[sonitus.git] / src / main / java / net / pterodactylus / sonitus / data / filter / SoxResampleFilter.java
1 /*
2  * Sonitus - ResampleFilter.java - Copyright © 2013 David Roden
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 3 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, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sonitus.data.filter;
19
20 import static com.google.common.base.Preconditions.checkArgument;
21 import static com.google.common.base.Preconditions.checkNotNull;
22
23 import java.io.IOException;
24
25 import net.pterodactylus.sonitus.data.Metadata;
26
27 import com.google.common.collect.ImmutableList;
28 import com.google.common.eventbus.EventBus;
29
30 /**
31  * {@link net.pterodactylus.sonitus.data.Filter} implementation that uses {@code
32  * sox} to resample a PCM-encoded source.
33  *
34  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
35  */
36 public class SoxResampleFilter extends ExternalFilter {
37
38         /** The location of the binary. */
39         private final String binary;
40
41         /** The final sampling rate. */
42         private final int rate;
43
44         /**
45          * Creates a new resample filter.
46          *
47          * @param eventBus
48          *              The event bus
49          * @param binary
50          *              The location of the binary
51          * @param rate
52          *              The new sampling rate
53          */
54         public SoxResampleFilter(EventBus eventBus, String binary, int rate) {
55                 super(eventBus, String.format("Resample to %s kHz", rate / 1000.0));
56                 this.binary = binary;
57                 this.rate = rate;
58         }
59
60         //
61         // FILTER METHODS
62         //
63
64         @Override
65         public Metadata metadata() {
66                 return super.metadata().frequency(rate);
67         }
68
69         @Override
70         public void open(Metadata metadata) throws IOException {
71                 checkNotNull(metadata, "metadata must not be null");
72                 checkArgument(metadata.encoding().equalsIgnoreCase("PCM"), "source must be PCM-encoded");
73
74                 super.open(metadata);
75         }
76
77         //
78         // EXTERNALFILTER METHODS
79         //
80
81         @Override
82         protected String binary(Metadata metadata) {
83                 return binary;
84         }
85
86         @Override
87         protected Iterable<String> parameters(Metadata metadata) {
88                 ImmutableList.Builder<String> parameters = ImmutableList.builder();
89                 /* the input file. */
90                 parameters.add("--bits").add("16");
91                 parameters.add("--channels").add(String.valueOf(metadata.channels()));
92                 parameters.add("--encoding").add("signed-integer");
93                 parameters.add("--rate").add(String.valueOf(metadata.frequency()));
94                 parameters.add("--type").add("raw");
95                 parameters.add("--endian").add("little");
96                 parameters.add("-");
97                 /* the output file. */
98                 parameters.add("--bits").add("16");
99                 parameters.add("--channels").add(String.valueOf(metadata.channels()));
100                 parameters.add("--encoding").add("signed-integer");
101                 parameters.add("--type").add("raw");
102                 parameters.add("--endian").add("little");
103                 parameters.add("-");
104                 /* rate effect. */
105                 parameters.add("rate").add("-h").add(String.valueOf(rate));
106                 return parameters.build();
107         }
108
109 }