Skip to content

Commit

Permalink
ALTO renderer: use proper BlockTypes
Browse files Browse the repository at this point in the history 
- use TextBlock, Illustration, GraphicalElement (not just TextBlock),
  as appropriate for the internal block types
- do not enter RIL_TEXTLINE, RIL_WORD, RIL_SYMBOL and ChoiceIterator
  on anything other than TextBlocks
- refactor loop to make it more readable
  • Loading branch information
@bertsky
bertsky committed Dec 18, 2019
1 parent caff12e commit 1dab4e0
Showing 1 changed file with 95 additions and 83 deletions.
178 changes: 95 additions & 83 deletions src/api/altorenderer.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -164,97 +164,109 @@ char* TessBaseAPI::GetAltoText(ETEXT_DESC* monitor, int page_number) {
<< " HEIGHT=\"" << rect_height_ << "\">\n";

ResultIterator* res_it = GetIterator();
while (!res_it->Empty(RIL_BLOCK)) {
if (res_it->Empty(RIL_WORD)) {
res_it->Next(RIL_WORD);
continue;
for (; !res_it->Empty(RIL_BLOCK); res_it->Next(RIL_BLOCK)) {
alto_str << "\t\t\t\t<ComposedBlock ID=\"cblock_" << bcnt << "\"";
AddBoxToAlto(res_it, RIL_BLOCK, alto_str);
alto_str << "\n";

const char* block_type;
switch (res_it->BlockType()) {
case PT_FLOWING_TEXT:
case PT_HEADING_TEXT:
case PT_PULLOUT_TEXT:
case PT_CAPTION_TEXT:
case PT_VERTICAL_TEXT:
case PT_TABLE: // nothing special here
case PT_EQUATION:
case PT_INLINE_EQUATION:
block_type = "TextBlock";
break;
case PT_FLOWING_IMAGE:
case PT_HEADING_IMAGE:
case PT_PULLOUT_IMAGE:
block_type = "Illustration";
break;
case PT_HORZ_LINE:
case PT_VERT_LINE:
block_type = "GraphicalElement";
break;
default:
block_type = "ComposedBlock";
}

if (res_it->IsAtBeginningOf(RIL_BLOCK)) {
alto_str << "\t\t\t\t<ComposedBlock ID=\"cblock_" << bcnt << "\"";
AddBoxToAlto(res_it, RIL_BLOCK, alto_str);
alto_str << "\n";
}

if (res_it->IsAtBeginningOf(RIL_PARA)) {
alto_str << "\t\t\t\t\t<TextBlock ID=\"block_" << tcnt << "\"";
for (; !res_it->Empty(RIL_PARA); res_it->Next(RIL_PARA)) {
alto_str << "\t\t\t\t\t<" << block_type << " ID=\"block_" << tcnt << "\"";
AddBoxToAlto(res_it, RIL_PARA, alto_str);
alto_str << "\n";
}

if (res_it->IsAtBeginningOf(RIL_TEXTLINE)) {
alto_str << "\t\t\t\t\t\t<TextLine ID=\"line_" << lcnt << "\"";
AddBoxToAlto(res_it, RIL_TEXTLINE, alto_str);
alto_str << "\n";
}

alto_str << "\t\t\t\t\t\t\t<String ID=\"string_" << wcnt << "\"";
AddBoxToAlto(res_it, RIL_WORD, alto_str);
alto_str << " CONTENT=\"" << HOcrEscape(res_it->GetUTF8Text(RIL_WORD)).c_str() << "\">";

bool last_word_in_line = res_it->IsAtFinalElement(RIL_TEXTLINE, RIL_WORD);
bool last_word_in_tblock = res_it->IsAtFinalElement(RIL_PARA, RIL_WORD);
bool last_word_in_cblock = res_it->IsAtFinalElement(RIL_BLOCK, RIL_WORD);


int left, top, right, bottom;
res_it->BoundingBox(RIL_WORD, &left, &top, &right, &bottom);

do {
alto_str << "\n\t\t\t\t\t\t\t\t<Glyph ID=\"glyph_" << scnt << "\"";
AddBoxToAlto(res_it, RIL_SYMBOL, alto_str);
alto_str << " CONTENT=\"";
const std::unique_ptr<const char[]> grapheme(
res_it->GetUTF8Text(RIL_SYMBOL));
if (grapheme && grapheme[0] != 0) {
alto_str << HOcrEscape(grapheme.get()).c_str();
if (strcmp(block_type, "TextBlock") == 0) {
for (; !res_it->Empty(RIL_TEXTLINE); res_it->Next(RIL_TEXTLINE)) {
alto_str << "\t\t\t\t\t\t<TextLine ID=\"line_" << lcnt << "\"";
AddBoxToAlto(res_it, RIL_TEXTLINE, alto_str);
alto_str << "\n";

for (; !res_it->Empty(RIL_WORD); res_it->Next(RIL_WORD)) {
int left = 0, top = 0, right = 0, bottom = 0;
if (!res_it->IsAtBeginningOf(RIL_TEXTLINE)) {
int hpos = right;
int vpos = top;
res_it->BoundingBox(RIL_WORD, &left, &top, &right, &bottom);
int width = left - hpos;
int height = bottom - top;
alto_str << "\n\t\t\t\t\t\t\t<SP";
alto_str << " HPOS=\"" << hpos << "\"";
alto_str << " VPOS=\"" << vpos << "\"";
alto_str << " WIDTH=\"" << width << "\"";
alto_str << " HEIGHT=\"" << height << "\"/>\n";
}
res_it->BoundingBox(RIL_WORD, &left, &top, &right, &bottom);

alto_str << "\t\t\t\t\t\t\t<String ID=\"string_" << wcnt << "\"";
AddBoxToAlto(res_it, RIL_WORD, alto_str);
alto_str << " CONTENT=\"" << HOcrEscape(res_it->GetUTF8Text(RIL_WORD)).c_str() << "\">";

for (; !res_it->Empty(RIL_SYMBOL); res_it->Next(RIL_SYMBOL)) {
alto_str << "\n\t\t\t\t\t\t\t\t<Glyph ID=\"glyph_" << scnt << "\"";
AddBoxToAlto(res_it, RIL_SYMBOL, alto_str);
alto_str << " CONTENT=\"";
const std::unique_ptr<const char[]> grapheme(res_it->GetUTF8Text(RIL_SYMBOL));
if (grapheme && grapheme[0] != 0)
alto_str << HOcrEscape(grapheme.get()).c_str();
alto_str << "\">";

ChoiceIterator choice_it(*res_it);
do {
int vc = choice_it.Confidence();
alto_str << "\n\t\t\t\t\t\t\t\t\t<Variant VC=\"0." << vc << "\"";
alto_str << " CONTENT=\"";
const char* variant = choice_it.GetUTF8Text();
if (variant && variant[0] != 0)
alto_str << HOcrEscape(variant).c_str();
alto_str << "\"/>";
} while (choice_it.Next());
alto_str << "\n\t\t\t\t\t\t\t\t</Glyph>";
scnt++;
if (res_it->IsAtFinalElement(RIL_WORD, RIL_SYMBOL))
break;
}
alto_str << "\n\t\t\t\t\t\t\t</String>";
wcnt++;
if (res_it->IsAtFinalElement(RIL_TEXTLINE, RIL_WORD))
break;
}
alto_str << "\n\t\t\t\t\t\t</TextLine>\n";
lcnt++;
if (res_it->IsAtFinalElement(RIL_PARA, RIL_TEXTLINE))
break;
}
}
alto_str << "\">";
ChoiceIterator choice_it(*res_it);
do {
int vc = choice_it.Confidence();
alto_str << "\n\t\t\t\t\t\t\t\t\t<Variant VC=\"0." << vc << "\"";
alto_str << " CONTENT=\"";
const char* variant = choice_it.GetUTF8Text();
if (variant && variant[0] != 0)
alto_str << HOcrEscape(variant).c_str();
alto_str << "\"/>";
} while (choice_it.Next());
alto_str << "\n\t\t\t\t\t\t\t\t</Glyph>";
res_it->Next(RIL_SYMBOL);

scnt++;
} while (!res_it->Empty(RIL_BLOCK) && !res_it->IsAtBeginningOf(RIL_WORD));

alto_str << "\n\t\t\t\t\t\t\t</String>";

wcnt++;

if (last_word_in_line) {
alto_str << "\n\t\t\t\t\t\t</TextLine>\n";
lcnt++;
} else {
int hpos = right;
int vpos = top;
res_it->BoundingBox(RIL_WORD, &left, &top, &right, &bottom);
int width = left - hpos;
int height = bottom - top;
alto_str << "\n\t\t\t\t\t\t\t<SP";
alto_str << " HPOS=\"" << hpos << "\"";
alto_str << " VPOS=\"" << vpos << "\"";
alto_str << " WIDTH=\"" << width << "\"";
alto_str << " HEIGHT=\"" << height << "\"/>\n";
}

if (last_word_in_tblock) {
alto_str << "\t\t\t\t\t</TextBlock>\n";
alto_str << "\t\t\t\t\t</" << block_type << ">\n";
tcnt++;
if (res_it->IsAtFinalElement(RIL_BLOCK, RIL_PARA))
break;
}

if (last_word_in_cblock) {
alto_str << "\t\t\t\t</ComposedBlock>\n";
bcnt++;
}
alto_str << "\t\t\t\t</ComposedBlock>\n";
bcnt++;
}

alto_str << "\t\t\t</PrintSpace>\n"
Expand Down

0 comments on commit 1dab4e0

Please sign in to comment.