1 files changed, 97 insertions, 0 deletions
diff --git a/src/main/java/se/lublin/humla/audio/encoder/PreprocessingEncoder.java b/src/main/java/se/lublin/humla/audio/encoder/PreprocessingEncoder.java
new file mode 100644
index 0000000..e914105
--- /dev/null
+++ b/src/main/java/se/lublin/humla/audio/encoder/PreprocessingEncoder.java
@@ -0,0 +1,97 @@
+/*
+ * Copyright (C) 2014 Andrew Comminos
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package se.lublin.humla.audio.encoder;
+
+import com.googlecode.javacpp.IntPointer;
+import se.lublin.humla.audio.javacpp.Speex;
+import se.lublin.humla.exception.NativeAudioException;
+import se.lublin.humla.net.PacketBuffer;
+
+import java.nio.BufferUnderflowException;
+
+/**
+ * Wrapper performing preprocessing options on the nested encoder.
+ * Uses Speex preprocessor.
+ * Created by andrew on 17/04/14.
+ */
+public class PreprocessingEncoder implements IEncoder {
+    private IEncoder mEncoder;
+    private Speex.SpeexPreprocessState mPreprocessor;
+
+    public PreprocessingEncoder(IEncoder encoder, int frameSize, int sampleRate) {
+        mEncoder = encoder;
+        mPreprocessor = new Speex.SpeexPreprocessState(frameSize, sampleRate);
+
+        IntPointer arg = new IntPointer(1);
+
+        arg.put(0);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_SET_VAD, arg);
+        arg.put(1);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_SET_AGC, arg);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_SET_DENOISE, arg);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_SET_DEREVERB, arg);
+
+        arg.put(30000);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_SET_AGC_TARGET, arg);
+
+        // TODO AGC max gain, decrement, noise suppress, echo
+
+        // Increase VAD difficulty
+        arg.put(99);
+        mPreprocessor.control(Speex.SpeexPreprocessState.SPEEX_PREPROCESS_GET_PROB_START, arg);
+    }
+
+    @Override
+    public int encode(short[] input, int inputSize) throws NativeAudioException {
+        mPreprocessor.preprocess(input);
+        return mEncoder.encode(input, inputSize);
+    }
+
+    @Override
+    public int getBufferedFrames() {
+        return mEncoder.getBufferedFrames();
+    }
+
+    @Override
+    public boolean isReady() {
+        return mEncoder.isReady();
+    }
+
+    @Override
+    public void getEncodedData(PacketBuffer packetBuffer) throws BufferUnderflowException {
+        mEncoder.getEncodedData(packetBuffer);
+    }
+
+    @Override
+    public void terminate() throws NativeAudioException {
+        mEncoder.terminate();
+    }
+
+    public void setEncoder(IEncoder encoder) {
+        if(mEncoder != null) mEncoder.destroy();
+        mEncoder = encoder;
+    }
+
+    @Override
+    public void destroy() {
+        mPreprocessor.destroy();
+        mEncoder.destroy();
+        mPreprocessor = null;
+        mEncoder = null;
+    }
+}