diff options
Diffstat (limited to 'BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java')
| -rw-r--r-- | BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java b/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java index a9cfa57..12fa848 100644 --- a/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java +++ b/BJC-Utils2/src/main/java/bjc/utils/ioutils/blocks/MappedBlockReader.java @@ -11,9 +11,13 @@ public class MappedBlockReader implements BlockReader { private UnaryOperator<Block> transform; + private int blockNo; + public MappedBlockReader(BlockReader source, UnaryOperator<Block> trans) { reader = source; transform = trans; + + blockNo = 0; } @Override @@ -29,7 +33,8 @@ public class MappedBlockReader implements BlockReader { @Override public boolean nextBlock() { if(hasNextBlock()) { - current = transform.apply(reader.next()); + current = transform.apply(reader.next()); + blockNo += 1; return true; } @@ -39,7 +44,7 @@ public class MappedBlockReader implements BlockReader { @Override public int getBlockCount() { - return reader.getBlockCount(); + return blockNo; } @Override |