BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

package bjc.utils.ioutils.blocks;

import java.io.IOException;

import java.util.function.UnaryOperator;

public class MappedBlockReader implements BlockReader {
	private BlockReader reader;

	private Block current;

	private UnaryOperator<Block> transform;

	public MappedBlockReader(BlockReader source, UnaryOperator<Block> trans) {
		reader    = source;
		transform = trans;
	}

	@Override
	public boolean hasNextBlock() {
		return reader.hasNextBlock();
	}

	@Override
	public Block getBlock() {
		return current;
	}

	@Override	
	public boolean nextBlock() {
		if(hasNextBlock()) {
			current = trans.apply(reader.next());

			return true;
		}

		return false;
	}

	@Override
	public int getBlockCount() {
		return reader.getBlockCount();
	}
}