import java.io.File;
import java.nio.ByteBuffer;
import java.nio.IntBuffer;
import java.nio.ShortBuffer;
import java.util.ArrayList;
import java.util.List;
import javax.sound.sampled.AudioFormat;
import javax.sound.sampled.AudioSystem;
import javax.sound.sampled.DataLine;
import javax.sound.sampled.LineUnavailableException;
import javax.sound.sampled.SourceDataLine;
import javax.sound.sampled.TargetDataLine;
import com.sun.jna.Native;
import com.sun.jna.ptr.PointerByReference;
import net.tomp2p.opuswrapper.*;
public class OpusTest {
static {
try {
System.loadLibrary("opus");
} catch (UnsatisfiedLinkError e1) {
try {
File f = Native.extractFromResourcePath("opus");
System.load(f.getAbsolutePath());
} catch (Exception e2) {
e1.printStackTrace();
e2.printStackTrace();
}
}
}
public void testCodec() throws LineUnavailableException {
AudioFormat format = new AudioFormat(8000.0f, 16, 1, true, true);
ShortBuffer dataFromMic = recordFromMicrophone(format, 5000);
List<ByteBuffer> packets = encode(dataFromMic);
// packets go over network
ShortBuffer decodedFromNetwork = decode(packets);
playBack(format, decodedFromNetwork);
}
private ShortBuffer decode(List<ByteBuffer> packets) {
IntBuffer error = IntBuffer.allocate(4);
PointerByReference opusDecoder = Opus.INSTANCE.opus_decoder_create(8000, 1, error);
ShortBuffer shortBuffer = ShortBuffer.allocate(1024 * 1024);
for (ByteBuffer dataBuffer : packets) {
byte[] transferedBytes = new byte[dataBuffer.remaining()];
dataBuffer.get(transferedBytes);
int decoded = Opus.INSTANCE.opus_decode(opusDecoder, transferedBytes, transferedBytes.length,
shortBuffer, 80, 0);
shortBuffer.position(shortBuffer.position() + decoded);
}
shortBuffer.flip();
Opus.INSTANCE.opus_decoder_destroy(opusDecoder);
return shortBuffer;
}
private List<ByteBuffer> encode(ShortBuffer shortBuffer) {
IntBuffer error = IntBuffer.allocate(4);
PointerByReference opusEncoder = Opus.INSTANCE.opus_encoder_create(8000, 1,
Opus.OPUS_APPLICATION_RESTRICTED_LOWDELAY, error);
int read = 0;
List<ByteBuffer> list = new ArrayList<>();
while (shortBuffer.hasRemaining()) {
ByteBuffer dataBuffer = ByteBuffer.allocate(1024);
int toRead = Math.min(shortBuffer.remaining(), dataBuffer.remaining());
read = Opus.INSTANCE.opus_encode(opusEncoder, shortBuffer, 80, dataBuffer, toRead);
dataBuffer.position(dataBuffer.position() + read);
dataBuffer.flip();
list.add(dataBuffer);
shortBuffer.position(shortBuffer.position() + 80);
}
Opus.INSTANCE.opus_encoder_destroy(opusEncoder);
// used for debugging
shortBuffer.flip();
return list;
}
private void playBack(AudioFormat format, ShortBuffer shortBuffer) throws LineUnavailableException {
SourceDataLine speaker = AudioSystem.getSourceDataLine(format);
speaker.open(format);
speaker.start();
short[] shortAudioBuffer = new short[shortBuffer.remaining()];
shortBuffer.get(shortAudioBuffer);
byte[] audio = ShortToByte_Twiddle_Method(shortAudioBuffer);
speaker.write(audio, 0, audio.length);
}
private ShortBuffer recordFromMicrophone(AudioFormat format, int lengthMillis)
throws LineUnavailableException {
DataLine.Info info = new DataLine.Info(TargetDataLine.class, format);
if (!AudioSystem.isLineSupported(info)) {
throw new LineUnavailableException("not supported");
}
TargetDataLine microphone = AudioSystem.getTargetDataLine(format);
// Obtain and open the line.
microphone.open(format);
// Assume that the TargetDataLine, line, has already been obtained and
// opened.
byte[] data = new byte[microphone.getBufferSize() / 5];
// Begin audio capture.
microphone.start();
// probably way too big
ShortBuffer shortBuffer = ShortBuffer.allocate(1024 * 1024);
// Here, stopped is a global boolean set by another thread.
long start = System.currentTimeMillis();
int numBytesRead;
while (System.currentTimeMillis() - start < lengthMillis) {
// Read the next chunk of data from the TargetDataLine.
numBytesRead = microphone.read(data, 0, data.length);
// Save this chunk of data.
for (int i = 0; i < numBytesRead; i += 2) {
int b1 = data[i + 1] & 0xff;
int b2 = data[i] << 8;
shortBuffer.put((short) (b1 | b2));
}
}
shortBuffer.flip();
return shortBuffer;
}
private byte[] ShortToByte_Twiddle_Method(final short[] input) {
final int len = input.length;
final byte[] buffer = new byte[len * 2];
for (int i = 0; i < len; i++) {
buffer[(i * 2) + 1] = (byte) (input[i]);
buffer[(i * 2)] = (byte) (input[i] >> 8);
}
return buffer;
}
}
