parent
3f5ed0af08
commit
f64ed119a7
@ -0,0 +1,138 @@ |
|||||||
|
/* |
||||||
|
* To change this license header, choose License Headers in Project Properties. |
||||||
|
* To change this template file, choose Tools | Templates |
||||||
|
* and open the template in the editor. |
||||||
|
*/ |
||||||
|
package moe.nekojimi.chords; |
||||||
|
|
||||||
|
import java.io.Closeable; |
||||||
|
import java.io.IOException; |
||||||
|
import java.nio.ByteBuffer; |
||||||
|
import java.util.logging.Level; |
||||||
|
import java.util.logging.Logger; |
||||||
|
import javax.sound.sampled.AudioFormat; |
||||||
|
import javax.sound.sampled.AudioInputStream; |
||||||
|
import javax.sound.sampled.AudioSystem; |
||||||
|
import javax.sound.sampled.UnsupportedAudioFileException; |
||||||
|
import net.dv8tion.jda.api.audio.AudioSendHandler; |
||||||
|
import org.apache.commons.io.input.buffer.CircularByteBuffer; |
||||||
|
|
||||||
|
/** |
||||||
|
* |
||||||
|
* @author jimj316 |
||||||
|
*/ |
||||||
|
public class TrackPlayer implements Closeable |
||||||
|
{ |
||||||
|
|
||||||
|
private static final int DESIRED_BUFFER_SIZE = 3840 * 500; |
||||||
|
private static final int MAX_READ_FAILS = 100; |
||||||
|
|
||||||
|
private final CircularByteBuffer audioBuffer = new CircularByteBuffer(3840 * 1024); |
||||||
|
|
||||||
|
private final AudioInputStream input; |
||||||
|
|
||||||
|
private boolean arrayErr = false; // supresses ArrayIndexOutOfBoundsException after the first time, to prevent spam
|
||||||
|
|
||||||
|
public TrackPlayer(Song song) throws UnsupportedAudioFileException, IOException |
||||||
|
{ |
||||||
|
AudioInputStream in = AudioSystem.getAudioInputStream(song.getLocation()); |
||||||
|
AudioFormat decodedFormat = AudioSendHandler.INPUT_FORMAT; |
||||||
|
input = AudioSystem.getAudioInputStream(decodedFormat, in); |
||||||
|
fillBuffer(false); |
||||||
|
} |
||||||
|
|
||||||
|
public TrackPlayer(AudioInputStream input) |
||||||
|
{ |
||||||
|
this.input = input; |
||||||
|
fillBuffer(false); |
||||||
|
} |
||||||
|
|
||||||
|
boolean has(int byteCount) |
||||||
|
{ |
||||||
|
return audioBuffer.getCurrentNumberOfBytes() > byteCount; |
||||||
|
} |
||||||
|
|
||||||
|
public ByteBuffer read(int length) |
||||||
|
{ |
||||||
|
boolean filled = fillBuffer(true); |
||||||
|
if (!filled) |
||||||
|
throw new OutOfInputException(); |
||||||
|
|
||||||
|
byte[] data = new byte[length]; |
||||||
|
audioBuffer.read(data, 0, data.length); |
||||||
|
// byte[] data = queue.poll();
|
||||||
|
return ByteBuffer.wrap(data); // Wrap this in a java.nio.ByteBuffer
|
||||||
|
} |
||||||
|
|
||||||
|
private boolean fillBuffer(boolean canSkip) |
||||||
|
{ |
||||||
|
int fails = 0; |
||||||
|
// use what we have in our buffer to send audio as PCM
|
||||||
|
while (audioBuffer.getCurrentNumberOfBytes() < DESIRED_BUFFER_SIZE) |
||||||
|
{ |
||||||
|
boolean read = readData(); |
||||||
|
if (!read && !canSkip) |
||||||
|
return false; |
||||||
|
else if (fails < MAX_READ_FAILS) |
||||||
|
fails++; |
||||||
|
else |
||||||
|
return false; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
private boolean readData() |
||||||
|
{ |
||||||
|
if (input == null) |
||||||
|
return false; |
||||||
|
try |
||||||
|
{ |
||||||
|
// if (din.available() == 0)
|
||||||
|
// return false;
|
||||||
|
int bytesToRead = DESIRED_BUFFER_SIZE - audioBuffer.getCurrentNumberOfBytes(); |
||||||
|
int space = audioBuffer.getSpace(); |
||||||
|
if (input.available() > 0 && input.available() < bytesToRead) |
||||||
|
bytesToRead = input.available(); |
||||||
|
if (bytesToRead > space) |
||||||
|
bytesToRead = space; |
||||||
|
if (bytesToRead == 0) |
||||||
|
return false; |
||||||
|
byte[] bytes = new byte[bytesToRead]; |
||||||
|
// byte[] bytes = din.readNBytes(bytesToRead);
|
||||||
|
int read = input.read(bytes); |
||||||
|
// System.out.println("Wanted: " + byteCount + " Space:" + space + " Available: " + din.available() + " To read: " + bytesToRead + " Read: " + read);
|
||||||
|
if (read < 0) |
||||||
|
return false; |
||||||
|
// queue.add(bytes);
|
||||||
|
|
||||||
|
audioBuffer.add(bytes, 0, read); |
||||||
|
} catch (IOException ex) |
||||||
|
{ |
||||||
|
Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); |
||||||
|
return false; |
||||||
|
} catch (ArrayIndexOutOfBoundsException ex) |
||||||
|
{ |
||||||
|
if (!arrayErr) |
||||||
|
arrayErr = true; |
||||||
|
else |
||||||
|
{ |
||||||
|
Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex); |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void close() throws IOException |
||||||
|
{ |
||||||
|
input.close(); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
public static class OutOfInputException extends RuntimeException |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
} |
Loading…
Reference in new issue