blob: 6b868bb690bf67440df79db6fcdf40305678eb7c (
plain)
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
package bjc.utils.ioutils;
import java.io.IOException;
import java.io.LineNumberReader;
import java.io.Reader;
import java.util.NoSuchElementException;
import java.util.Scanner;
import java.util.regex.Pattern;
/**
* Simple implementation of {@link BlockReader}
*
* NOTE: The EOF marker is always treated as a delimiter. You are expected to
* handle blocks that may be shorter than you expect.
*
* @author EVE
*
*/
public class SimpleBlockReader implements BlockReader {
/*
* I/O source for blocks.
*/
private LineNumberReader lnReader;
private Scanner blockReader;
/*
* The current block.
*/
private Block currBlock;
private int blockNo;
/**
* Create a new block reader.
*
* @param blockDelim
* The pattern that separates blocks. Note that the end
* of file is always considered to end a block.
*
* @param source
* The source to read blocks from.
*/
public SimpleBlockReader(String blockDelim, Reader source) {
lnReader = new LineNumberReader(source);
blockReader = new Scanner(lnReader);
String pattern = String.format("(?:%s)|\\Z", blockDelim);
Pattern pt = Pattern.compile(pattern, Pattern.MULTILINE);
blockReader.useDelimiter(pt);
}
@Override
public boolean hasNextBlock() {
return blockReader.hasNext();
}
@Override
public Block getBlock() {
return currBlock;
}
@Override
public boolean nextBlock() {
try {
int blockStartLine = lnReader.getLineNumber();
String blockContents = blockReader.next();
int blockEndLine = lnReader.getLineNumber();
blockNo += 1;
currBlock = new Block(blockNo, blockContents, blockStartLine, blockEndLine);
return true;
} catch (NoSuchElementException nseex) {
currBlock = null;
return false;
}
}
@Override
public int getBlockCount() {
return blockNo;
}
@Override
public void close() throws IOException {
blockReader.close();
lnReader.close();
}
/**
* Set the delimiter used to separate blocks.
*
* @param delim
* The delimiter used to separate blocks.
*/
public void setDelimiter(String delim) {
blockReader.useDelimiter(delim);
}
@Override
public String toString() {
return String.format("SimpleBlockReader [currBlock=%s, blockNo=%s]", currBlock, blockNo);
}
}
|
