Reorganize blocks

1 files changed, 98 insertions, 0 deletions

diff --git a/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/PushbackBlockReader.java b/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/PushbackBlockReader.java
new file mode 100644
index 0000000..96906ae
--- /dev/null
+++ b/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/PushbackBlockReader.java
@@ -0,0 +1,98 @@
+package bjc.utils.ioutils.blocks;
+
+import java.io.IOException;
+import java.util.Deque;
+import java.util.LinkedList;
+
+/**
+ * A block reader that supports pushing blocks onto the input queue so that they
+ * are provided before blocks read from an input source.
+ * 
+ * @author bjculkin
+ *
+ */
+public class PushbackBlockReader implements BlockReader {
+	private BlockReader source;
+
+	private Deque<Block> waiting;
+
+	private Block curBlock;
+
+	private int blockNo;
+
+	/**
+	 * Create a new pushback block reader.
+	 * 
+	 * @param src
+	 *                The block reader to use when no blocks are queued.
+	 */
+	public PushbackBlockReader(BlockReader src) {
+		source = src;
+
+		waiting = new LinkedList<>();
+	}
+
+	@Override
+	public boolean hasNextBlock() {
+		return !waiting.isEmpty() || source.hasNextBlock();
+	}
+
+	@Override
+	public Block getBlock() {
+		return curBlock;
+	}
+
+	@Override
+	public boolean nextBlock() {
+		if (!waiting.isEmpty()) {
+			curBlock = waiting.pop();
+
+			blockNo += 1;
+
+			return true;
+		} else {
+			boolean succ = source.nextBlock();
+			curBlock = source.getBlock();
+
+			if (succ) blockNo += 1;
+
+			return succ;
+		}
+	}
+
+	@Override
+	public int getBlockCount() {
+		return blockNo;
+	}
+
+	@Override
+	public void close() throws IOException {
+		source.close();
+	}
+
+	/**
+	 * Insert a block at the back of the queue of pending blocks.
+	 * 
+	 * @param blk
+	 *                The block to put at the back.
+	 */
+	public void addBlock(Block blk) {
+		waiting.add(blk);
+	}
+
+	/**
+	 * Insert a block at the front of the queue of pending blocks.
+	 * 
+	 * @param blk
+	 *                The block to put at the front.
+	 */
+	public void pushBlock(Block blk) {
+		waiting.push(blk);
+	}
+
+	@Override
+	public String toString() {
+		return String.format("PushbackBlockReader [waiting=%s, curBlock=%s, blockNo=%s]", waiting, curBlock,
+				blockNo);
+	}
+}
\ No newline at end of file