private LevelResults compareZones(BxPage expected, BxPage actual) {
Map<BxChunk, BxZone> map = BxModelUtils.mapChunksToZones(actual);
LevelResults results = new LevelResults();
for (BxZone expectedZone : expected) {
if (ignoredLabels.contains(expectedZone.getLabel())) {
continue;
}
Set<BxZone> actualZones = new HashSet<BxZone>();
for (BxLine line : expectedZone) {
for (BxWord word : line) {
for (BxChunk chunk : word) {
actualZones.add(map.get(chunk));
}
}
}
if (actualZones.size() == 1) {
for (BxZone actualZone : actualZones) {
if (BxModelUtils.countChunks(actualZone) == BxModelUtils.countChunks(expectedZone)) {
results.matched++;
}
}
}
results.all++;
}
return results;
}
private LevelResults compareWords(BxPage expected, BxPage actual) {
Map<BxChunk, BxWord> map = BxModelUtils.mapChunksToWords(actual);
LevelResults results = new LevelResults();
for (BxZone expectedZone : expected) {
if (ignoredLabels.contains(expectedZone.getLabel())) {
continue;
}
for (BxLine expectedLine : expectedZone) {
for (BxWord expectedWord : expectedLine) {
Set<BxWord> actualWords = new HashSet<BxWord>();
for (BxChunk chunk : expectedWord) {
actualWords.add(map.get(chunk));
}
if (actualWords.size() == 1) {
for (BxWord actualWord : actualWords) {
if (actualWord.childrenCount() == expectedWord.childrenCount()) {
results.matched++;
}
}
}
results.all++;
}
}
}
return results;
}
@Override
protected BxDocument prepareActualDocument(BxDocument document) throws AnalysisException {
document = BxModelUtils.deepClone(document);
preprocessDocument(document);
return processDocument(document);
}
