plantuml/src/net/sourceforge/plantuml/code/deflate/ByteHistory.java

93 lines
2.6 KiB
Java

package net.sourceforge.plantuml.code.deflate;
/*
* Simple DEFLATE decompressor
* Copyright (c) Project Nayuki
*
* https://www.nayuki.io/page/simple-deflate-decompressor
* https://github.com/nayuki/Simple-DEFLATE-decompressor
*/
import java.io.IOException;
import java.util.Objects;
/**
* Stores a finite recent history of a byte stream. Useful as an implicit
* dictionary for Lempel-Ziv schemes. Mutable and not thread-safe.
*/
final class ByteHistory {
/*---- Fields ----*/
// Circular buffer of byte data.
private byte[] data;
// Index of next byte to write to, always in the range [0, data.length).
private int index;
/*---- Constructor ----*/
/**
* Constructs a byte history of the specified size, initialized to zeros.
* @param size the size, which must be positive
* @throws IllegalArgumentException if size is zero or negative
*/
public ByteHistory(int size) {
if (size < 1)
throw new IllegalArgumentException("Size must be positive");
data = new byte[size];
index = 0;
}
/*---- Methods ----*/
/**
* Appends the specified byte to this history.
* This overwrites the byte value at {@code size} positions ago.
* @param b the byte value to append
*/
public void append(int b) {
if (index < 0 || index >= data.length)
throw new IllegalStateException();
data[index] = (byte)b;
index = (index + 1) % data.length;
}
/**
* Copies {@code len} bytes starting at {@code dist} bytes ago to
* the specified output stream and also back into this buffer itself.
* <p>Note that if the length exceeds the distance, then some of the output
* data will be a copy of data that was copied earlier in the process.</p>
* @param dist the distance to go back, in the range [1, size]
* @param len the length to copy, which must be at least 0
* @param out the output stream to write to (not {@code null})
* @throws NullPointerException if the output stream is {@code null}
* @throws IllegalArgumentException if the length is negative,
* distance is not positive, or distance is greater than the buffer size
* @throws IOException if an I/O exception occurs
*/
public void copy(int dist, int len, OutputStreamProtected out) throws IOException {
Objects.requireNonNull(out);
if (len < 0 || dist < 1 || dist > data.length)
throw new IllegalArgumentException();
int readIndex = (index - dist + data.length) % data.length;
if (readIndex < 0 || readIndex >= data.length)
throw new IllegalStateException();
for (int i = 0; i < len; i++) {
byte b = data[readIndex];
readIndex = (readIndex + 1) % data.length;
out.write(b);
append(b);
}
}
}