mirror of
https://github.com/mitb-archive/filebot
synced 2024-08-13 17:03:45 -04:00
* better account for large history files
This commit is contained in:
parent
f4070a4c07
commit
68c89ef01a
@ -1,7 +1,5 @@
|
||||
|
||||
package net.sourceforge.tuned;
|
||||
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
@ -9,142 +7,125 @@ import java.nio.ByteBuffer;
|
||||
import java.nio.channels.Channels;
|
||||
import java.nio.channels.ReadableByteChannel;
|
||||
|
||||
|
||||
public class ByteBufferOutputStream extends OutputStream {
|
||||
|
||||
|
||||
private ByteBuffer buffer;
|
||||
|
||||
|
||||
private final float loadFactor;
|
||||
|
||||
|
||||
public ByteBufferOutputStream(long initialCapacity) {
|
||||
this((int) initialCapacity);
|
||||
}
|
||||
|
||||
|
||||
public ByteBufferOutputStream(int initialCapacity) {
|
||||
this(initialCapacity, 1.0f);
|
||||
}
|
||||
|
||||
|
||||
public ByteBufferOutputStream(int initialCapacity, float loadFactor) {
|
||||
if (initialCapacity < 0)
|
||||
throw new IllegalArgumentException("initialCapacity must not be negative");
|
||||
|
||||
|
||||
if (loadFactor <= 0 || Float.isNaN(loadFactor))
|
||||
throw new IllegalArgumentException("loadFactor must be greater than 0");
|
||||
|
||||
this.buffer = ByteBuffer.allocate(initialCapacity);
|
||||
|
||||
this.buffer = ByteBuffer.allocate(initialCapacity + 1);
|
||||
this.loadFactor = loadFactor;
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void write(int b) throws IOException {
|
||||
ensureCapacity(buffer.position() + 1);
|
||||
buffer.put((byte) b);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void write(byte[] src) throws IOException {
|
||||
ensureCapacity(buffer.position() + src.length);
|
||||
buffer.put(src);
|
||||
}
|
||||
|
||||
|
||||
public void write(ByteBuffer src) throws IOException {
|
||||
ensureCapacity(buffer.position() + src.remaining());
|
||||
buffer.put(src);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void write(byte[] src, int offset, int length) throws IOException {
|
||||
ensureCapacity(buffer.position() + length);
|
||||
buffer.put(src, offset, length);
|
||||
}
|
||||
|
||||
|
||||
public void ensureCapacity(int minCapacity) {
|
||||
if (minCapacity <= buffer.capacity())
|
||||
return;
|
||||
|
||||
|
||||
// calculate new buffer size with load factor
|
||||
int newCapacity = (int) (buffer.capacity() * (1 + loadFactor));
|
||||
|
||||
|
||||
// ensure minCapacity
|
||||
if (newCapacity < minCapacity)
|
||||
newCapacity = minCapacity;
|
||||
|
||||
|
||||
// create new buffer with increased capacity
|
||||
ByteBuffer newBuffer = ByteBuffer.allocate(newCapacity);
|
||||
|
||||
|
||||
// copy current data to new buffer
|
||||
buffer.flip();
|
||||
newBuffer.put(buffer);
|
||||
|
||||
|
||||
buffer = newBuffer;
|
||||
}
|
||||
|
||||
|
||||
public ByteBuffer getByteBuffer() {
|
||||
ByteBuffer result = buffer.duplicate();
|
||||
|
||||
|
||||
// flip buffer so it can be read
|
||||
result.flip();
|
||||
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
public byte[] getByteArray() {
|
||||
ByteBuffer data = getByteBuffer();
|
||||
|
||||
|
||||
// copy data to byte array
|
||||
byte[] bytes = new byte[data.remaining()];
|
||||
data.get(bytes);
|
||||
|
||||
|
||||
return bytes;
|
||||
}
|
||||
|
||||
|
||||
public int transferFrom(ReadableByteChannel channel) throws IOException {
|
||||
// make sure buffer is not at its limit
|
||||
ensureCapacity(buffer.position() + 1);
|
||||
|
||||
|
||||
return channel.read(buffer);
|
||||
}
|
||||
|
||||
|
||||
public int transferFully(InputStream inputStream) throws IOException {
|
||||
return transferFully(Channels.newChannel(inputStream));
|
||||
}
|
||||
|
||||
|
||||
public int transferFully(ReadableByteChannel channel) throws IOException {
|
||||
int total = 0, read = 0;
|
||||
|
||||
|
||||
while ((read = transferFrom(channel)) >= 0) {
|
||||
total += read;
|
||||
}
|
||||
|
||||
|
||||
return total;
|
||||
}
|
||||
|
||||
|
||||
public int position() {
|
||||
return buffer.position();
|
||||
}
|
||||
|
||||
|
||||
public int capacity() {
|
||||
return buffer.capacity();
|
||||
}
|
||||
|
||||
|
||||
public void rewind() {
|
||||
buffer.rewind();
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user