2009-07-11 10:39:56 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
|
|
// File: TabFind.cpp
|
|
|
|
// Description: Subclass of BBGrid to find vertically aligned blobs.
|
|
|
|
// Author: Ray Smith
|
|
|
|
// Created: Fri Mar 21 15:03:01 PST 2008
|
|
|
|
//
|
|
|
|
// (C) Copyright 2008, Google Inc.
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
//
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
|
|
#include "tabfind.h"
|
|
|
|
#include "alignedblob.h"
|
|
|
|
#include "blobbox.h"
|
2012-02-02 10:53:04 +08:00
|
|
|
#include "colpartitiongrid.h"
|
2009-07-11 10:39:56 +08:00
|
|
|
#include "detlinefit.h"
|
|
|
|
#include "linefind.h"
|
|
|
|
#include "ndminx.h"
|
|
|
|
|
2010-09-30 23:53:40 +08:00
|
|
|
// Include automatically generated configuration file if running autoconf.
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
#include "config_auto.h"
|
|
|
|
#endif
|
|
|
|
|
2009-07-11 10:39:56 +08:00
|
|
|
namespace tesseract {
|
|
|
|
|
|
|
|
// Multiple of box size to search for initial gaps.
|
|
|
|
const int kTabRadiusFactor = 5;
|
|
|
|
// Min and Max multiple of height to search vertically when extrapolating.
|
|
|
|
const int kMinVerticalSearch = 3;
|
|
|
|
const int kMaxVerticalSearch = 12;
|
|
|
|
const int kMaxRaggedSearch = 25;
|
|
|
|
// Minimum number of lines in a column width to make it interesting.
|
|
|
|
const int kMinLinesInColumn = 10;
|
|
|
|
// Minimum width of a column to be interesting.
|
|
|
|
const int kMinColumnWidth = 200;
|
|
|
|
// Minimum fraction of total column lines for a column to be interesting.
|
|
|
|
const double kMinFractionalLinesInColumn = 0.125;
|
|
|
|
// Fraction of height used as alignment tolerance for aligned tabs.
|
|
|
|
const double kAlignedFraction = 0.03125;
|
2010-11-24 02:34:14 +08:00
|
|
|
// Minimum gutter width in absolute inch (multiplied by resolution)
|
|
|
|
const double kMinGutterWidthAbsolute = 0.02;
|
|
|
|
// Maximum gutter width (in absolute inch) that we care about
|
|
|
|
const double kMaxGutterWidthAbsolute = 2.00;
|
2012-02-02 10:53:04 +08:00
|
|
|
// Multiplier of gridsize for min gutter width of TT_MAYBE_RAGGED blobs.
|
|
|
|
const int kRaggedGutterMultiple = 5;
|
2011-03-22 05:48:17 +08:00
|
|
|
// Min aspect ratio of tall objects to be considered a separator line.
|
|
|
|
// (These will be ignored in searching the gutter for obstructions.)
|
|
|
|
const double kLineFragmentAspectRatio = 10.0;
|
2009-07-11 10:39:56 +08:00
|
|
|
// Multiplier of new y positions in running average for skew estimation.
|
|
|
|
const double kSmoothFactor = 0.25;
|
|
|
|
// Min coverage for a good baseline between vectors
|
|
|
|
const double kMinBaselineCoverage = 0.5;
|
|
|
|
// Minimum overlap fraction when scanning text lines for column widths.
|
|
|
|
const double kCharVerticalOverlapFraction = 0.375;
|
|
|
|
// Maximum horizontal gap allowed when scanning for column widths
|
|
|
|
const double kMaxHorizontalGap = 3.0;
|
|
|
|
// Maximum upper quartile error allowed on a baseline fit as a fraction
|
|
|
|
// of height.
|
|
|
|
const double kMaxBaselineError = 0.4375;
|
|
|
|
// Min number of points to accept after evaluation.
|
|
|
|
const int kMinEvaluatedTabs = 3;
|
|
|
|
// Minimum aspect ratio of a textline to make a good textline blob with a
|
|
|
|
// single blob.
|
|
|
|
const int kMaxTextLineBlobRatio = 5;
|
|
|
|
// Minimum aspect ratio of a textline to make a good textline blob with
|
|
|
|
// multiple blobs. Target ratio varies according to number of blobs.
|
|
|
|
const int kMinTextLineBlobRatio = 3;
|
|
|
|
// Fraction of box area covered by image to make a blob image.
|
|
|
|
const double kMinImageArea = 0.5;
|
2010-11-24 02:34:14 +08:00
|
|
|
// Upto 30 degrees is allowed for rotations of diacritic blobs.
|
|
|
|
// Keep this value slightly larger than kCosSmallAngle in blobbox.cpp
|
|
|
|
// so that the assert there never fails.
|
|
|
|
const double kCosMaxSkewAngle = 0.866025;
|
2009-07-11 10:39:56 +08:00
|
|
|
|
|
|
|
BOOL_VAR(textord_tabfind_show_initialtabs, false, "Show tab candidates");
|
|
|
|
BOOL_VAR(textord_tabfind_show_finaltabs, false, "Show tab vectors");
|
2010-11-24 02:34:14 +08:00
|
|
|
double_VAR(textord_tabfind_aligned_gap_fraction, 0.75,
|
|
|
|
"Fraction of height used as a minimum gap for aligned blobs.");
|
2009-07-11 10:39:56 +08:00
|
|
|
|
|
|
|
TabFind::TabFind(int gridsize, const ICOORD& bleft, const ICOORD& tright,
|
2010-11-24 02:34:14 +08:00
|
|
|
TabVector_LIST* vlines, int vertical_x, int vertical_y,
|
|
|
|
int resolution)
|
2009-07-11 10:39:56 +08:00
|
|
|
: AlignedBlob(gridsize, bleft, tright),
|
2010-11-24 02:34:14 +08:00
|
|
|
resolution_(resolution),
|
2012-02-02 10:53:04 +08:00
|
|
|
image_origin_(0, tright.y() - 1) {
|
2009-07-11 10:39:56 +08:00
|
|
|
width_cb_ = NULL;
|
|
|
|
v_it_.set_to_list(&vectors_);
|
|
|
|
v_it_.add_list_after(vlines);
|
|
|
|
SetVerticalSkewAndParellelize(vertical_x, vertical_y);
|
2010-11-24 02:34:14 +08:00
|
|
|
width_cb_ = NewPermanentTessCallback(this, &TabFind::CommonWidth);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TabFind::~TabFind() {
|
|
|
|
if (width_cb_ != NULL)
|
|
|
|
delete width_cb_;
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////// PUBLIC functions (mostly used by TabVector). //////////////
|
|
|
|
|
|
|
|
// Insert a list of blobs into the given grid (not necessarily this).
|
|
|
|
// If take_ownership is true, then the blobs are removed from the source list.
|
|
|
|
// See InsertBlob for the other arguments.
|
2012-02-02 10:53:04 +08:00
|
|
|
// It would seem to make more sense to swap this and grid, but this way
|
|
|
|
// around allows grid to not be derived from TabFind, eg a ColPartitionGrid,
|
|
|
|
// while the grid that provides the tab stops(this) has to be derived from
|
|
|
|
// TabFind.
|
|
|
|
void TabFind::InsertBlobsToGrid(bool h_spread, bool v_spread,
|
|
|
|
BLOBNBOX_LIST* blobs,
|
|
|
|
BBGrid<BLOBNBOX, BLOBNBOX_CLIST,
|
|
|
|
BLOBNBOX_C_IT>* grid) {
|
2009-07-11 10:39:56 +08:00
|
|
|
BLOBNBOX_IT blob_it(blobs);
|
|
|
|
int b_count = 0;
|
|
|
|
int reject_count = 0;
|
|
|
|
for (blob_it.mark_cycle_pt(); !blob_it.cycled_list(); blob_it.forward()) {
|
|
|
|
BLOBNBOX* blob = blob_it.data();
|
2012-02-02 10:53:04 +08:00
|
|
|
// if (InsertBlob(true, true, blob, grid)) {
|
|
|
|
if (InsertBlob(h_spread, v_spread, blob, grid)) {
|
2009-07-11 10:39:56 +08:00
|
|
|
++b_count;
|
|
|
|
} else {
|
|
|
|
++reject_count;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (textord_debug_tabfind) {
|
2012-02-02 10:53:04 +08:00
|
|
|
tprintf("Inserted %d blobs into grid, %d rejected.\n",
|
|
|
|
b_count, reject_count);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Insert a single blob into the given grid (not necessarily this).
|
|
|
|
// If h_spread, then all cells covered horizontally by the box are
|
|
|
|
// used, otherwise, just the bottom-left. Similarly for v_spread.
|
|
|
|
// A side effect is that the left and right rule edges of the blob are
|
|
|
|
// set according to the tab vectors in this (not grid).
|
2012-02-02 10:53:04 +08:00
|
|
|
bool TabFind::InsertBlob(bool h_spread, bool v_spread, BLOBNBOX* blob,
|
2009-07-11 10:39:56 +08:00
|
|
|
BBGrid<BLOBNBOX, BLOBNBOX_CLIST,
|
|
|
|
BLOBNBOX_C_IT>* grid) {
|
|
|
|
TBOX box = blob->bounding_box();
|
|
|
|
blob->set_left_rule(LeftEdgeForBox(box, false, false));
|
|
|
|
blob->set_right_rule(RightEdgeForBox(box, false, false));
|
|
|
|
blob->set_left_crossing_rule(LeftEdgeForBox(box, true, false));
|
|
|
|
blob->set_right_crossing_rule(RightEdgeForBox(box, true, false));
|
2010-11-24 02:34:14 +08:00
|
|
|
if (blob->joined_to_prev())
|
|
|
|
return false;
|
2009-07-11 10:39:56 +08:00
|
|
|
grid->InsertBBox(h_spread, v_spread, blob);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2012-02-02 10:53:04 +08:00
|
|
|
// Calls SetBlobRuleEdges for all the blobs in the given block.
|
|
|
|
void TabFind::SetBlockRuleEdges(TO_BLOCK* block) {
|
|
|
|
SetBlobRuleEdges(&block->blobs);
|
|
|
|
SetBlobRuleEdges(&block->small_blobs);
|
|
|
|
SetBlobRuleEdges(&block->noise_blobs);
|
|
|
|
SetBlobRuleEdges(&block->large_blobs);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Sets the left and right rule and crossing_rules for the blobs in the given
|
|
|
|
// list by fiding the next outermost tabvectors for each blob.
|
|
|
|
void TabFind::SetBlobRuleEdges(BLOBNBOX_LIST* blobs) {
|
|
|
|
BLOBNBOX_IT blob_it(blobs);
|
|
|
|
for (blob_it.mark_cycle_pt(); !blob_it.cycled_list(); blob_it.forward()) {
|
|
|
|
BLOBNBOX* blob = blob_it.data();
|
|
|
|
TBOX box = blob->bounding_box();
|
|
|
|
blob->set_left_rule(LeftEdgeForBox(box, false, false));
|
|
|
|
blob->set_right_rule(RightEdgeForBox(box, false, false));
|
|
|
|
blob->set_left_crossing_rule(LeftEdgeForBox(box, true, false));
|
|
|
|
blob->set_right_crossing_rule(RightEdgeForBox(box, true, false));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-03-22 05:48:17 +08:00
|
|
|
// Returns the gutter width of the given TabVector between the given y limits.
|
|
|
|
// Also returns x-shift to be added to the vector to clear any intersecting
|
|
|
|
// blobs. The shift is deducted from the returned gutter.
|
2012-02-02 10:53:04 +08:00
|
|
|
// If ignore_unmergeables is true, then blobs of UnMergeableType are
|
|
|
|
// ignored as if they don't exist. (Used for text on image.)
|
|
|
|
// max_gutter_width is used as the maximum width worth searching for in case
|
|
|
|
// there is nothing near the TabVector.
|
2011-03-22 05:48:17 +08:00
|
|
|
int TabFind::GutterWidth(int bottom_y, int top_y, const TabVector& v,
|
2012-02-02 10:53:04 +08:00
|
|
|
bool ignore_unmergeables, int max_gutter_width,
|
2011-03-22 05:48:17 +08:00
|
|
|
int* required_shift) {
|
|
|
|
bool right_to_left = v.IsLeftTab();
|
|
|
|
int bottom_x = v.XAtY(bottom_y);
|
|
|
|
int top_x = v.XAtY(top_y);
|
|
|
|
int start_x = right_to_left ? MAX(top_x, bottom_x) : MIN(top_x, bottom_x);
|
|
|
|
BlobGridSearch sidesearch(this);
|
|
|
|
sidesearch.StartSideSearch(start_x, bottom_y, top_y);
|
2012-02-02 10:53:04 +08:00
|
|
|
int min_gap = max_gutter_width;
|
2011-03-22 05:48:17 +08:00
|
|
|
*required_shift = 0;
|
|
|
|
BLOBNBOX* blob = NULL;
|
|
|
|
while ((blob = sidesearch.NextSideSearch(right_to_left)) != NULL) {
|
|
|
|
const TBOX& box = blob->bounding_box();
|
|
|
|
if (box.bottom() >= top_y || box.top() <= bottom_y)
|
|
|
|
continue; // Doesn't overlap enough.
|
|
|
|
if (box.height() >= gridsize() * 2 &&
|
|
|
|
box.height() > box.width() * kLineFragmentAspectRatio) {
|
|
|
|
// Skip likely separator line residue.
|
|
|
|
continue;
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
if (ignore_unmergeables && BLOBNBOX::UnMergeableType(blob->region_type()))
|
|
|
|
continue; // Skip non-text if required.
|
2011-03-22 05:48:17 +08:00
|
|
|
int mid_y = (box.bottom() + box.top()) / 2;
|
|
|
|
// We use the x at the mid-y so that the required_shift guarantees
|
|
|
|
// to clear all the blobs on the tab-stop. If we use the min/max
|
|
|
|
// of x at top/bottom of the blob, then exactness would be required,
|
|
|
|
// which is not a good thing.
|
|
|
|
int tab_x = v.XAtY(mid_y);
|
|
|
|
int gap;
|
|
|
|
if (right_to_left) {
|
|
|
|
gap = tab_x - box.right();
|
|
|
|
if (gap < 0 && box.left() - tab_x < *required_shift)
|
|
|
|
*required_shift = box.left() - tab_x;
|
|
|
|
} else {
|
|
|
|
gap = box.left() - tab_x;
|
|
|
|
if (gap < 0 && box.right() - tab_x > *required_shift)
|
|
|
|
*required_shift = box.right() - tab_x;
|
|
|
|
}
|
|
|
|
if (gap > 0 && gap < min_gap)
|
|
|
|
min_gap = gap;
|
|
|
|
}
|
|
|
|
// Result may be negative, in which case, this is a really bad tabstop.
|
|
|
|
return min_gap - abs(*required_shift);
|
|
|
|
}
|
|
|
|
|
2009-07-11 10:39:56 +08:00
|
|
|
// Find the gutter width and distance to inner neighbour for the given blob.
|
|
|
|
void TabFind::GutterWidthAndNeighbourGap(int tab_x, int mean_height,
|
|
|
|
int max_gutter, bool left,
|
|
|
|
BLOBNBOX* bbox, int* gutter_width,
|
|
|
|
int* neighbour_gap ) {
|
|
|
|
const TBOX& box = bbox->bounding_box();
|
|
|
|
// The gutter and internal sides of the box.
|
|
|
|
int gutter_x = left ? box.left() : box.right();
|
|
|
|
int internal_x = left ? box.right() : box.left();
|
|
|
|
// On ragged edges, the gutter side of the box is away from the tabstop.
|
|
|
|
int tab_gap = left ? gutter_x - tab_x : tab_x - gutter_x;
|
|
|
|
*gutter_width = max_gutter;
|
|
|
|
// If the box is away from the tabstop, we need to increase
|
|
|
|
// the allowed gutter width.
|
|
|
|
if (tab_gap > 0)
|
|
|
|
*gutter_width += tab_gap;
|
2012-02-02 10:53:04 +08:00
|
|
|
bool debug = WithinTestRegion(2, box.left(), box.bottom());
|
|
|
|
if (debug)
|
|
|
|
tprintf("Looking in gutter\n");
|
2009-07-11 10:39:56 +08:00
|
|
|
// Find the nearest blob on the outside of the column.
|
2012-02-02 10:53:04 +08:00
|
|
|
BLOBNBOX* gutter_bbox = AdjacentBlob(bbox, left,
|
|
|
|
bbox->flow() == BTFT_TEXT_ON_IMAGE, 0.0,
|
|
|
|
*gutter_width, box.top(), box.bottom());
|
2009-07-11 10:39:56 +08:00
|
|
|
if (gutter_bbox != NULL) {
|
|
|
|
TBOX gutter_box = gutter_bbox->bounding_box();
|
|
|
|
*gutter_width = left ? tab_x - gutter_box.right()
|
|
|
|
: gutter_box.left() - tab_x;
|
|
|
|
}
|
|
|
|
if (*gutter_width >= max_gutter) {
|
|
|
|
// If there is no box because a tab was in the way, get the tab coord.
|
|
|
|
TBOX gutter_box(box);
|
|
|
|
if (left) {
|
|
|
|
gutter_box.set_left(tab_x - max_gutter - 1);
|
|
|
|
gutter_box.set_right(tab_x - max_gutter);
|
|
|
|
int tab_gutter = RightEdgeForBox(gutter_box, true, false);
|
|
|
|
if (tab_gutter < tab_x - 1)
|
|
|
|
*gutter_width = tab_x - tab_gutter;
|
|
|
|
} else {
|
|
|
|
gutter_box.set_left(tab_x + max_gutter);
|
|
|
|
gutter_box.set_right(tab_x + max_gutter + 1);
|
|
|
|
int tab_gutter = LeftEdgeForBox(gutter_box, true, false);
|
|
|
|
if (tab_gutter > tab_x + 1)
|
|
|
|
*gutter_width = tab_gutter - tab_x;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (*gutter_width > max_gutter)
|
|
|
|
*gutter_width = max_gutter;
|
|
|
|
// Now look for a neighbour on the inside.
|
2012-02-02 10:53:04 +08:00
|
|
|
if (debug)
|
|
|
|
tprintf("Looking for neighbour\n");
|
|
|
|
BLOBNBOX* neighbour = AdjacentBlob(bbox, !left,
|
|
|
|
bbox->flow() == BTFT_TEXT_ON_IMAGE, 0.0,
|
|
|
|
*gutter_width, box.top(), box.bottom());
|
2009-07-11 10:39:56 +08:00
|
|
|
int neighbour_edge = left ? RightEdgeForBox(box, true, false)
|
|
|
|
: LeftEdgeForBox(box, true, false);
|
|
|
|
if (neighbour != NULL) {
|
|
|
|
TBOX n_box = neighbour->bounding_box();
|
2012-02-02 10:53:04 +08:00
|
|
|
if (debug) {
|
|
|
|
tprintf("Found neighbour:");
|
|
|
|
n_box.print();
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
if (left && n_box.left() < neighbour_edge)
|
|
|
|
neighbour_edge = n_box.left();
|
|
|
|
else if (!left && n_box.right() > neighbour_edge)
|
|
|
|
neighbour_edge = n_box.right();
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
*neighbour_gap = left ? neighbour_edge - internal_x
|
|
|
|
: internal_x - neighbour_edge;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the x-coord that corresponds to the right edge for the given
|
|
|
|
// box. If there is a rule line to the right that vertically overlaps it,
|
|
|
|
// then return the x-coord of the rule line, otherwise return the right
|
|
|
|
// edge of the page. For details see RightTabForBox below.
|
|
|
|
int TabFind::RightEdgeForBox(const TBOX& box, bool crossing, bool extended) {
|
|
|
|
TabVector* v = RightTabForBox(box, crossing, extended);
|
|
|
|
return v == NULL ? tright_.x() : v->XAtY((box.top() + box.bottom()) / 2);
|
|
|
|
}
|
|
|
|
// As RightEdgeForBox, but finds the left Edge instead.
|
|
|
|
int TabFind::LeftEdgeForBox(const TBOX& box, bool crossing, bool extended) {
|
|
|
|
TabVector* v = LeftTabForBox(box, crossing, extended);
|
|
|
|
return v == NULL ? bleft_.x() : v->XAtY((box.top() + box.bottom()) / 2);
|
|
|
|
}
|
|
|
|
|
|
|
|
// This comment documents how this function works.
|
|
|
|
// For its purpose and arguments, see the comment in tabfind.h.
|
|
|
|
// TabVectors are stored sorted by perpendicular distance of middle from
|
|
|
|
// the global mean vertical vector. Since the individual vectors can have
|
|
|
|
// differing directions, their XAtY for a given y is not necessarily in the
|
|
|
|
// right order. Therefore the search has to be run with a margin.
|
|
|
|
// The middle of a vector that passes through (x,y) cannot be higher than
|
|
|
|
// halfway from y to the top, or lower than halfway from y to the bottom
|
|
|
|
// of the coordinate range; therefore, the search margin is the range of
|
|
|
|
// sort keys between these halfway points. Any vector with a sort key greater
|
|
|
|
// than the upper margin must be to the right of x at y, and likewise any
|
|
|
|
// vector with a sort key less than the lower margin must pass to the left
|
|
|
|
// of x at y.
|
|
|
|
TabVector* TabFind::RightTabForBox(const TBOX& box, bool crossing,
|
|
|
|
bool extended) {
|
|
|
|
if (v_it_.empty())
|
|
|
|
return NULL;
|
|
|
|
int top_y = box.top();
|
|
|
|
int bottom_y = box.bottom();
|
|
|
|
int mid_y = (top_y + bottom_y) / 2;
|
|
|
|
int right = crossing ? (box.left() + box.right()) / 2 : box.right();
|
|
|
|
int min_key, max_key;
|
|
|
|
SetupTabSearch(right, mid_y, &min_key, &max_key);
|
|
|
|
// Position the iterator at the first TabVector with sort_key >= min_key.
|
|
|
|
while (!v_it_.at_first() && v_it_.data()->sort_key() >= min_key)
|
|
|
|
v_it_.backward();
|
|
|
|
while (!v_it_.at_last() && v_it_.data()->sort_key() < min_key)
|
|
|
|
v_it_.forward();
|
|
|
|
// Find the leftmost tab vector that overlaps and has XAtY(mid_y) >= right.
|
|
|
|
TabVector* best_v = NULL;
|
|
|
|
int best_x = -1;
|
|
|
|
int key_limit = -1;
|
|
|
|
do {
|
|
|
|
TabVector* v = v_it_.data();
|
|
|
|
int x = v->XAtY(mid_y);
|
|
|
|
if (x >= right &&
|
|
|
|
(v->VOverlap(top_y, bottom_y) > 0 ||
|
|
|
|
(extended && v->ExtendedOverlap(top_y, bottom_y) > 0))) {
|
|
|
|
if (best_v == NULL || x < best_x) {
|
|
|
|
best_v = v;
|
|
|
|
best_x = x;
|
|
|
|
// We can guarantee that no better vector can be found if the
|
|
|
|
// sort key exceeds that of the best by max_key - min_key.
|
|
|
|
key_limit = v->sort_key() + max_key - min_key;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Break when the search is done to avoid wrapping the iterator and
|
|
|
|
// thereby potentially slowing the next search.
|
|
|
|
if (v_it_.at_last() ||
|
|
|
|
(best_v != NULL && v->sort_key() > key_limit))
|
|
|
|
break; // Prevent restarting list for next call.
|
|
|
|
v_it_.forward();
|
|
|
|
} while (!v_it_.at_first());
|
|
|
|
return best_v;
|
|
|
|
}
|
|
|
|
|
|
|
|
// As RightTabForBox, but finds the left TabVector instead.
|
|
|
|
TabVector* TabFind::LeftTabForBox(const TBOX& box, bool crossing,
|
|
|
|
bool extended) {
|
|
|
|
if (v_it_.empty())
|
|
|
|
return NULL;
|
|
|
|
int top_y = box.top();
|
|
|
|
int bottom_y = box.bottom();
|
|
|
|
int mid_y = (top_y + bottom_y) / 2;
|
|
|
|
int left = crossing ? (box.left() + box.right()) / 2 : box.left();
|
|
|
|
int min_key, max_key;
|
|
|
|
SetupTabSearch(left, mid_y, &min_key, &max_key);
|
|
|
|
// Position the iterator at the last TabVector with sort_key <= max_key.
|
|
|
|
while (!v_it_.at_last() && v_it_.data()->sort_key() <= max_key)
|
|
|
|
v_it_.forward();
|
|
|
|
while (!v_it_.at_first() && v_it_.data()->sort_key() > max_key) {
|
|
|
|
v_it_.backward();
|
|
|
|
}
|
|
|
|
// Find the rightmost tab vector that overlaps and has XAtY(mid_y) <= left.
|
|
|
|
TabVector* best_v = NULL;
|
|
|
|
int best_x = -1;
|
|
|
|
int key_limit = -1;
|
|
|
|
do {
|
|
|
|
TabVector* v = v_it_.data();
|
|
|
|
int x = v->XAtY(mid_y);
|
|
|
|
if (x <= left &&
|
|
|
|
(v->VOverlap(top_y, bottom_y) > 0 ||
|
|
|
|
(extended && v->ExtendedOverlap(top_y, bottom_y) > 0))) {
|
|
|
|
if (best_v == NULL || x > best_x) {
|
|
|
|
best_v = v;
|
|
|
|
best_x = x;
|
|
|
|
// We can guarantee that no better vector can be found if the
|
|
|
|
// sort key is less than that of the best by max_key - min_key.
|
|
|
|
key_limit = v->sort_key() - (max_key - min_key);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Break when the search is done to avoid wrapping the iterator and
|
|
|
|
// thereby potentially slowing the next search.
|
|
|
|
if (v_it_.at_first() ||
|
|
|
|
(best_v != NULL && v->sort_key() < key_limit))
|
|
|
|
break; // Prevent restarting list for next call.
|
|
|
|
v_it_.backward();
|
|
|
|
} while (!v_it_.at_last());
|
|
|
|
return best_v;
|
|
|
|
}
|
|
|
|
|
2010-11-24 02:34:14 +08:00
|
|
|
// Return true if the given width is close to one of the common
|
|
|
|
// widths in column_widths_.
|
|
|
|
bool TabFind::CommonWidth(int width) {
|
|
|
|
width /= kColumnWidthFactor;
|
|
|
|
ICOORDELT_IT it(&column_widths_);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
ICOORDELT* w = it.data();
|
|
|
|
if (NearlyEqual<int>(width, w->x(), 1))
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return true if the sizes are more than a
|
|
|
|
// factor of 2 different.
|
|
|
|
bool TabFind::DifferentSizes(int size1, int size2) {
|
|
|
|
return size1 > size2 * 2 || size2 > size1 * 2;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return true if the sizes are more than a
|
|
|
|
// factor of 5 different.
|
|
|
|
bool TabFind::VeryDifferentSizes(int size1, int size2) {
|
|
|
|
return size1 > size2 * 5 || size2 > size1 * 5;
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////// PROTECTED functions (used by ColumnFinder). //////////////
|
|
|
|
|
|
|
|
// Top-level function to find TabVectors in an input page block.
|
|
|
|
// Returns false if the detected skew angle is impossible.
|
2012-02-02 10:53:04 +08:00
|
|
|
// Applies the detected skew angle to deskew the tabs, blobs and part_grid.
|
2010-11-24 02:34:14 +08:00
|
|
|
bool TabFind::FindTabVectors(TabVector_LIST* hlines,
|
|
|
|
BLOBNBOX_LIST* image_blobs, TO_BLOCK* block,
|
|
|
|
int min_gutter_width,
|
2012-02-02 10:53:04 +08:00
|
|
|
ColPartitionGrid* part_grid,
|
2010-11-24 02:34:14 +08:00
|
|
|
FCOORD* deskew, FCOORD* reskew) {
|
2011-03-22 05:48:17 +08:00
|
|
|
ScrollView* tab_win = FindInitialTabVectors(image_blobs, min_gutter_width,
|
|
|
|
block);
|
2012-02-02 10:53:04 +08:00
|
|
|
ComputeColumnWidths(tab_win, part_grid);
|
2010-11-24 02:34:14 +08:00
|
|
|
TabVector::MergeSimilarTabVectors(vertical_skew_, &vectors_, this);
|
|
|
|
SortVectors();
|
|
|
|
CleanupTabs();
|
|
|
|
if (!Deskew(hlines, image_blobs, block, deskew, reskew))
|
|
|
|
return false; // Skew angle is too large.
|
2012-02-02 10:53:04 +08:00
|
|
|
part_grid->Deskew(*deskew);
|
2010-11-24 02:34:14 +08:00
|
|
|
ApplyTabConstraints();
|
2012-03-02 06:43:51 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2010-11-24 02:34:14 +08:00
|
|
|
if (textord_tabfind_show_finaltabs) {
|
2011-03-22 05:48:17 +08:00
|
|
|
tab_win = MakeWindow(640, 50, "FinalTabs");
|
2010-11-24 02:34:14 +08:00
|
|
|
if (textord_debug_images) {
|
|
|
|
tab_win->Image(AlignedBlob::textord_debug_pix().string(),
|
|
|
|
image_origin_.x(), image_origin_.y());
|
|
|
|
} else {
|
|
|
|
DisplayBoxes(tab_win);
|
|
|
|
DisplayTabs("FinalTabs", tab_win);
|
|
|
|
}
|
|
|
|
tab_win = DisplayTabVectors(tab_win);
|
|
|
|
}
|
2012-03-02 06:43:51 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2010-11-24 02:34:14 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Top-level function to not find TabVectors in an input page block,
|
|
|
|
// but setup for single column mode.
|
|
|
|
void TabFind::DontFindTabVectors(BLOBNBOX_LIST* image_blobs, TO_BLOCK* block,
|
|
|
|
FCOORD* deskew, FCOORD* reskew) {
|
2012-02-02 10:53:04 +08:00
|
|
|
InsertBlobsToGrid(false, false, image_blobs, this);
|
|
|
|
InsertBlobsToGrid(true, false, &block->blobs, this);
|
2010-11-24 02:34:14 +08:00
|
|
|
deskew->set_x(1.0f);
|
|
|
|
deskew->set_y(0.0f);
|
|
|
|
reskew->set_x(1.0f);
|
|
|
|
reskew->set_y(0.0f);
|
|
|
|
}
|
|
|
|
|
2012-02-02 10:53:04 +08:00
|
|
|
// Cleans up the lists of blobs in the block ready for use by TabFind.
|
|
|
|
// Large blobs that look like text are moved to the main blobs list.
|
|
|
|
// Main blobs that are superseded by the image blobs are deleted.
|
|
|
|
void TabFind::TidyBlobs(TO_BLOCK* block) {
|
|
|
|
BLOBNBOX_IT large_it = &block->large_blobs;
|
|
|
|
BLOBNBOX_IT blob_it = &block->blobs;
|
|
|
|
int b_count = 0;
|
|
|
|
for (large_it.mark_cycle_pt(); !large_it.cycled_list(); large_it.forward()) {
|
|
|
|
BLOBNBOX* large_blob = large_it.data();
|
|
|
|
if (large_blob->owner() != NULL) {
|
|
|
|
blob_it.add_to_end(large_it.extract());
|
|
|
|
++b_count;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (textord_debug_tabfind) {
|
|
|
|
tprintf("Moved %d large blobs to normal list\n",
|
|
|
|
b_count);
|
2012-03-02 06:43:51 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2012-02-02 10:53:04 +08:00
|
|
|
ScrollView* rej_win = MakeWindow(500, 300, "Image blobs");
|
|
|
|
block->plot_graded_blobs(rej_win);
|
|
|
|
block->plot_noise_blobs(rej_win);
|
|
|
|
rej_win->Update();
|
2012-03-02 06:43:51 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2012-02-02 10:53:04 +08:00
|
|
|
}
|
|
|
|
block->DeleteUnownedNoise();
|
|
|
|
}
|
|
|
|
|
2009-07-11 10:39:56 +08:00
|
|
|
// Helper function to setup search limits for *TabForBox.
|
|
|
|
void TabFind::SetupTabSearch(int x, int y, int* min_key, int* max_key) {
|
|
|
|
int key1 = TabVector::SortKey(vertical_skew_, x, (y + tright_.y()) / 2);
|
|
|
|
int key2 = TabVector::SortKey(vertical_skew_, x, (y + bleft_.y()) / 2);
|
|
|
|
*min_key = MIN(key1, key2);
|
|
|
|
*max_key = MAX(key1, key2);
|
|
|
|
}
|
|
|
|
|
|
|
|
ScrollView* TabFind::DisplayTabVectors(ScrollView* tab_win) {
|
|
|
|
#ifndef GRAPHICS_DISABLED
|
|
|
|
// For every vector, display it.
|
|
|
|
TabVector_IT it(&vectors_);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* vector = it.data();
|
|
|
|
vector->Display(tab_win);
|
|
|
|
}
|
|
|
|
tab_win->Update();
|
|
|
|
#endif
|
|
|
|
return tab_win;
|
|
|
|
}
|
|
|
|
|
|
|
|
// PRIVATE CODE.
|
|
|
|
//
|
|
|
|
// First part of FindTabVectors, which may be used twice if the text
|
|
|
|
// is mostly of vertical alignment.
|
2011-03-22 05:48:17 +08:00
|
|
|
ScrollView* TabFind::FindInitialTabVectors(BLOBNBOX_LIST* image_blobs,
|
|
|
|
int min_gutter_width,
|
|
|
|
TO_BLOCK* block) {
|
2009-07-11 10:39:56 +08:00
|
|
|
if (textord_tabfind_show_initialtabs) {
|
|
|
|
ScrollView* line_win = MakeWindow(0, 0, "VerticalLines");
|
|
|
|
line_win = DisplayTabVectors(line_win);
|
|
|
|
}
|
|
|
|
// Prepare the grid.
|
2012-02-02 10:53:04 +08:00
|
|
|
if (image_blobs != NULL)
|
|
|
|
InsertBlobsToGrid(true, false, image_blobs, this);
|
|
|
|
InsertBlobsToGrid(true, false, &block->blobs, this);
|
2010-11-24 02:34:14 +08:00
|
|
|
ScrollView* initial_win = FindTabBoxes(min_gutter_width);
|
2011-03-22 05:48:17 +08:00
|
|
|
FindAllTabVectors(min_gutter_width);
|
2009-07-11 10:39:56 +08:00
|
|
|
|
|
|
|
TabVector::MergeSimilarTabVectors(vertical_skew_, &vectors_, this);
|
|
|
|
SortVectors();
|
|
|
|
EvaluateTabs();
|
2012-02-02 10:53:04 +08:00
|
|
|
if (textord_tabfind_show_initialtabs && initial_win != NULL)
|
2011-03-22 05:48:17 +08:00
|
|
|
initial_win = DisplayTabVectors(initial_win);
|
2010-11-24 02:34:14 +08:00
|
|
|
MarkVerticalText();
|
2011-03-22 05:48:17 +08:00
|
|
|
return initial_win;
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
|
2012-02-02 10:53:04 +08:00
|
|
|
// Helper displays all the boxes in the given vector on the given window.
|
2014-01-10 01:38:53 +08:00
|
|
|
static void DisplayBoxVector(const GenericVector<BLOBNBOX*>& boxes,
|
2012-02-02 10:53:04 +08:00
|
|
|
ScrollView* win) {
|
2012-03-02 06:43:51 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2012-02-02 10:53:04 +08:00
|
|
|
for (int i = 0; i < boxes.size(); ++i) {
|
|
|
|
TBOX box = boxes[i]->bounding_box();
|
|
|
|
int left_x = box.left();
|
|
|
|
int right_x = box.right();
|
|
|
|
int top_y = box.top();
|
|
|
|
int bottom_y = box.bottom();
|
|
|
|
ScrollView::Color box_color = boxes[i]->BoxColor();
|
|
|
|
win->Pen(box_color);
|
|
|
|
win->Rectangle(left_x, bottom_y, right_x, top_y);
|
|
|
|
}
|
|
|
|
win->Update();
|
2012-03-02 06:43:51 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2012-02-02 10:53:04 +08:00
|
|
|
}
|
|
|
|
|
2009-07-11 10:39:56 +08:00
|
|
|
// For each box in the grid, decide whether it is a candidate tab-stop,
|
2012-02-02 10:53:04 +08:00
|
|
|
// and if so add it to the left/right tab boxes.
|
2010-11-24 02:34:14 +08:00
|
|
|
ScrollView* TabFind::FindTabBoxes(int min_gutter_width) {
|
2012-02-02 10:53:04 +08:00
|
|
|
left_tab_boxes_.clear();
|
|
|
|
right_tab_boxes_.clear();
|
2009-07-11 10:39:56 +08:00
|
|
|
// For every bbox in the grid, determine whether it uses a tab on an edge.
|
|
|
|
GridSearch<BLOBNBOX, BLOBNBOX_CLIST, BLOBNBOX_C_IT> gsearch(this);
|
|
|
|
gsearch.StartFullSearch();
|
|
|
|
BLOBNBOX* bbox;
|
|
|
|
while ((bbox = gsearch.NextFullSearch()) != NULL) {
|
2010-11-24 02:34:14 +08:00
|
|
|
if (TestBoxForTabs(bbox, min_gutter_width)) {
|
2012-02-02 10:53:04 +08:00
|
|
|
// If it is any kind of tab, insert it into the vectors.
|
|
|
|
if (bbox->left_tab_type() != TT_NONE)
|
|
|
|
left_tab_boxes_.push_back(bbox);
|
|
|
|
if (bbox->right_tab_type() != TT_NONE)
|
|
|
|
right_tab_boxes_.push_back(bbox);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
// Sort left tabs by left and right by right to see the outermost one first
|
|
|
|
// on a ragged tab.
|
|
|
|
left_tab_boxes_.sort(SortByBoxLeft<BLOBNBOX>);
|
|
|
|
right_tab_boxes_.sort(SortRightToLeft<BLOBNBOX>);
|
2009-07-11 10:39:56 +08:00
|
|
|
ScrollView* tab_win = NULL;
|
2012-03-02 06:43:51 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2009-07-11 10:39:56 +08:00
|
|
|
if (textord_tabfind_show_initialtabs) {
|
2012-02-02 10:53:04 +08:00
|
|
|
tab_win = MakeWindow(0, 100, "InitialTabs");
|
|
|
|
tab_win->Pen(ScrollView::BLUE);
|
|
|
|
tab_win->Brush(ScrollView::NONE);
|
|
|
|
// Display the left and right tab boxes.
|
|
|
|
DisplayBoxVector(left_tab_boxes_, tab_win);
|
|
|
|
DisplayBoxVector(right_tab_boxes_, tab_win);
|
2009-07-11 10:39:56 +08:00
|
|
|
tab_win = DisplayTabs("Tabs", tab_win);
|
|
|
|
}
|
2012-03-02 06:43:51 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2009-07-11 10:39:56 +08:00
|
|
|
return tab_win;
|
|
|
|
}
|
|
|
|
|
2010-11-24 02:34:14 +08:00
|
|
|
bool TabFind::TestBoxForTabs(BLOBNBOX* bbox, int min_gutter_width) {
|
2009-07-11 10:39:56 +08:00
|
|
|
GridSearch<BLOBNBOX, BLOBNBOX_CLIST, BLOBNBOX_C_IT> radsearch(this);
|
|
|
|
TBOX box = bbox->bounding_box();
|
|
|
|
// If there are separator lines, get the column edges.
|
|
|
|
int left_column_edge = bbox->left_rule();
|
|
|
|
int right_column_edge = bbox->right_rule();
|
|
|
|
// The edges of the bounding box of the blob being processed.
|
|
|
|
int left_x = box.left();
|
|
|
|
int right_x = box.right();
|
|
|
|
int top_y = box.top();
|
|
|
|
int bottom_y = box.bottom();
|
|
|
|
int height = box.height();
|
|
|
|
bool debug = WithinTestRegion(3, left_x, top_y);
|
|
|
|
if (debug) {
|
|
|
|
tprintf("Column edges for blob at (%d,%d)->(%d,%d) are [%d, %d]\n",
|
|
|
|
left_x, top_y, right_x, bottom_y,
|
|
|
|
left_column_edge, right_column_edge);
|
|
|
|
}
|
|
|
|
// Compute a search radius based on a multiple of the height.
|
|
|
|
int radius = (height * kTabRadiusFactor + gridsize_ - 1) / gridsize_;
|
|
|
|
radsearch.StartRadSearch((left_x + right_x)/2, (top_y + bottom_y)/2, radius);
|
2010-11-24 02:34:14 +08:00
|
|
|
// In Vertical Page mode, once we have an estimate of the vertical line
|
|
|
|
// spacing, the minimum amount of gutter space before a possible tab is
|
|
|
|
// increased under the assumption that column partition is always larger
|
|
|
|
// than line spacing.
|
|
|
|
int min_spacing =
|
|
|
|
static_cast<int>(height * textord_tabfind_aligned_gap_fraction);
|
|
|
|
if (min_gutter_width > min_spacing)
|
|
|
|
min_spacing = min_gutter_width;
|
2012-02-02 10:53:04 +08:00
|
|
|
int min_ragged_gutter = kRaggedGutterMultiple * gridsize();
|
|
|
|
if (min_gutter_width > min_ragged_gutter)
|
|
|
|
min_ragged_gutter = min_gutter_width;
|
2010-11-24 02:34:14 +08:00
|
|
|
int target_right = left_x - min_spacing;
|
|
|
|
int target_left = right_x + min_spacing;
|
2009-07-11 10:39:56 +08:00
|
|
|
// We will be evaluating whether the left edge could be a left tab, and
|
|
|
|
// whether the right edge could be a right tab.
|
|
|
|
// A box can be a tab if its bool is_(left/right)_tab remains true, meaning
|
|
|
|
// that no blobs have been found in the gutter during the radial search.
|
|
|
|
// A box can also be a tab if there are objects in the gutter only above
|
|
|
|
// or only below, and there are aligned objects on the opposite side, but
|
|
|
|
// not too many unaligned objects. The maybe_(left/right)_tab_up counts
|
|
|
|
// aligned objects above and negatively counts unaligned objects above,
|
|
|
|
// and is set to -MAX_INT32 if a gutter object is found above.
|
|
|
|
// The other 3 maybe ints work similarly for the other sides.
|
2012-02-02 10:53:04 +08:00
|
|
|
// These conditions are very strict, to minimize false positives, and really
|
|
|
|
// only aligned tabs and outermost ragged tab blobs will qualify, so we
|
|
|
|
// also have maybe_ragged_left/right with less stringent rules.
|
|
|
|
// A blob that is maybe_ragged_left/right will be further qualified later,
|
|
|
|
// using the min_ragged_gutter.
|
2009-07-11 10:39:56 +08:00
|
|
|
bool is_left_tab = true;
|
|
|
|
bool is_right_tab = true;
|
2012-02-02 10:53:04 +08:00
|
|
|
bool maybe_ragged_left = true;
|
|
|
|
bool maybe_ragged_right = true;
|
2009-07-11 10:39:56 +08:00
|
|
|
int maybe_left_tab_up = 0;
|
|
|
|
int maybe_right_tab_up = 0;
|
|
|
|
int maybe_left_tab_down = 0;
|
|
|
|
int maybe_right_tab_down = 0;
|
2010-11-24 02:34:14 +08:00
|
|
|
if (bbox->leader_on_left()) {
|
|
|
|
is_left_tab = false;
|
2012-02-02 10:53:04 +08:00
|
|
|
maybe_ragged_left = false;
|
2010-11-24 02:34:14 +08:00
|
|
|
maybe_left_tab_up = -MAX_INT32;
|
|
|
|
maybe_left_tab_down = -MAX_INT32;
|
|
|
|
}
|
|
|
|
if (bbox->leader_on_right()) {
|
|
|
|
is_right_tab = false;
|
2012-02-02 10:53:04 +08:00
|
|
|
maybe_ragged_right = false;
|
2010-11-24 02:34:14 +08:00
|
|
|
maybe_right_tab_up = -MAX_INT32;
|
|
|
|
maybe_right_tab_down = -MAX_INT32;
|
|
|
|
}
|
2009-07-11 10:39:56 +08:00
|
|
|
int alignment_tolerance = static_cast<int>(resolution_ * kAlignedFraction);
|
|
|
|
BLOBNBOX* neighbour = NULL;
|
|
|
|
while ((neighbour = radsearch.NextRadSearch()) != NULL) {
|
|
|
|
if (neighbour == bbox)
|
|
|
|
continue;
|
|
|
|
TBOX nbox = neighbour->bounding_box();
|
|
|
|
int n_left = nbox.left();
|
|
|
|
int n_right = nbox.right();
|
|
|
|
if (debug)
|
|
|
|
tprintf("Neighbour at (%d,%d)->(%d,%d)\n",
|
|
|
|
n_left, nbox.bottom(), n_right, nbox.top());
|
|
|
|
// If the neighbouring blob is the wrong side of a separator line, then it
|
|
|
|
// "doesn't exist" as far as we are concerned.
|
|
|
|
if (n_right > right_column_edge || n_left < left_column_edge ||
|
|
|
|
left_x < neighbour->left_rule() || right_x > neighbour->right_rule())
|
|
|
|
continue; // Separator line in the way.
|
|
|
|
int n_mid_x = (n_left + n_right) / 2;
|
|
|
|
int n_mid_y = (nbox.top() + nbox.bottom()) / 2;
|
|
|
|
if (n_mid_x <= left_x && n_right >= target_right) {
|
|
|
|
if (debug)
|
|
|
|
tprintf("Not a left tab\n");
|
|
|
|
is_left_tab = false;
|
|
|
|
if (n_mid_y < top_y)
|
|
|
|
maybe_left_tab_down = -MAX_INT32;
|
|
|
|
if (n_mid_y > bottom_y)
|
|
|
|
maybe_left_tab_up = -MAX_INT32;
|
|
|
|
} else if (NearlyEqual(left_x, n_left, alignment_tolerance)) {
|
|
|
|
if (debug)
|
|
|
|
tprintf("Maybe a left tab\n");
|
|
|
|
if (n_mid_y > top_y && maybe_left_tab_up > -MAX_INT32)
|
|
|
|
++maybe_left_tab_up;
|
|
|
|
if (n_mid_y < bottom_y && maybe_left_tab_down > -MAX_INT32)
|
|
|
|
++maybe_left_tab_down;
|
|
|
|
} else if (n_left < left_x && n_right >= left_x) {
|
|
|
|
// Overlaps but not aligned so negative points on a maybe.
|
|
|
|
if (debug)
|
|
|
|
tprintf("Maybe Not a left tab\n");
|
|
|
|
if (n_mid_y > top_y && maybe_left_tab_up > -MAX_INT32)
|
|
|
|
--maybe_left_tab_up;
|
|
|
|
if (n_mid_y < bottom_y && maybe_left_tab_down > -MAX_INT32)
|
|
|
|
--maybe_left_tab_down;
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
if (n_left < left_x && nbox.y_overlap(box) && n_right >= target_right) {
|
|
|
|
maybe_ragged_left = false;
|
|
|
|
if (debug)
|
|
|
|
tprintf("Not a ragged left\n");
|
|
|
|
}
|
2009-07-11 10:39:56 +08:00
|
|
|
if (n_mid_x >= right_x && n_left <= target_left) {
|
|
|
|
if (debug)
|
|
|
|
tprintf("Not a right tab\n");
|
|
|
|
is_right_tab = false;
|
|
|
|
if (n_mid_y < top_y)
|
|
|
|
maybe_right_tab_down = -MAX_INT32;
|
|
|
|
if (n_mid_y > bottom_y)
|
|
|
|
maybe_right_tab_up = -MAX_INT32;
|
|
|
|
} else if (NearlyEqual(right_x, n_right, alignment_tolerance)) {
|
|
|
|
if (debug)
|
|
|
|
tprintf("Maybe a right tab\n");
|
|
|
|
if (n_mid_y > top_y && maybe_right_tab_up > -MAX_INT32)
|
|
|
|
++maybe_right_tab_up;
|
|
|
|
if (n_mid_y < bottom_y && maybe_right_tab_down > -MAX_INT32)
|
|
|
|
++maybe_right_tab_down;
|
|
|
|
} else if (n_right > right_x && n_left <= right_x) {
|
|
|
|
// Overlaps but not aligned so negative points on a maybe.
|
|
|
|
if (debug)
|
|
|
|
tprintf("Maybe Not a right tab\n");
|
|
|
|
if (n_mid_y > top_y && maybe_right_tab_up > -MAX_INT32)
|
|
|
|
--maybe_right_tab_up;
|
|
|
|
if (n_mid_y < bottom_y && maybe_right_tab_down > -MAX_INT32)
|
|
|
|
--maybe_right_tab_down;
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
if (n_right > right_x && nbox.y_overlap(box) && n_left <= target_left) {
|
|
|
|
maybe_ragged_right = false;
|
|
|
|
if (debug)
|
|
|
|
tprintf("Not a ragged right\n");
|
|
|
|
}
|
2009-07-11 10:39:56 +08:00
|
|
|
if (maybe_left_tab_down == -MAX_INT32 && maybe_left_tab_up == -MAX_INT32 &&
|
|
|
|
maybe_right_tab_down == -MAX_INT32 && maybe_right_tab_up == -MAX_INT32)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (is_left_tab || maybe_left_tab_up > 1 || maybe_left_tab_down > 1) {
|
2012-02-02 10:53:04 +08:00
|
|
|
bbox->set_left_tab_type(TT_MAYBE_ALIGNED);
|
|
|
|
} else if (maybe_ragged_left && ConfirmRaggedLeft(bbox, min_ragged_gutter)) {
|
|
|
|
bbox->set_left_tab_type(TT_MAYBE_RAGGED);
|
|
|
|
} else {
|
|
|
|
bbox->set_left_tab_type(TT_NONE);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
if (is_right_tab || maybe_right_tab_up > 1 || maybe_right_tab_down > 1) {
|
2012-02-02 10:53:04 +08:00
|
|
|
bbox->set_right_tab_type(TT_MAYBE_ALIGNED);
|
|
|
|
} else if (maybe_ragged_right &&
|
|
|
|
ConfirmRaggedRight(bbox, min_ragged_gutter)) {
|
|
|
|
bbox->set_right_tab_type(TT_MAYBE_RAGGED);
|
|
|
|
} else {
|
|
|
|
bbox->set_right_tab_type(TT_NONE);
|
|
|
|
}
|
|
|
|
if (debug) {
|
|
|
|
tprintf("Left result = %s, Right result=%s\n",
|
|
|
|
bbox->left_tab_type() == TT_MAYBE_ALIGNED ? "Aligned" :
|
|
|
|
(bbox->left_tab_type() == TT_MAYBE_RAGGED ? "Ragged" : "None"),
|
|
|
|
bbox->right_tab_type() == TT_MAYBE_ALIGNED ? "Aligned" :
|
|
|
|
(bbox->right_tab_type() == TT_MAYBE_RAGGED ? "Ragged" : "None"));
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
return bbox->left_tab_type() != TT_NONE || bbox->right_tab_type() != TT_NONE;
|
|
|
|
}
|
|
|
|
|
2012-02-02 10:53:04 +08:00
|
|
|
// Returns true if there is nothing in the rectangle of width min_gutter to
|
|
|
|
// the left of bbox.
|
|
|
|
bool TabFind::ConfirmRaggedLeft(BLOBNBOX* bbox, int min_gutter) {
|
|
|
|
TBOX search_box(bbox->bounding_box());
|
|
|
|
search_box.set_right(search_box.left());
|
|
|
|
search_box.set_left(search_box.left() - min_gutter);
|
|
|
|
return NothingYOverlapsInBox(search_box, bbox->bounding_box());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if there is nothing in the rectangle of width min_gutter to
|
|
|
|
// the right of bbox.
|
|
|
|
bool TabFind::ConfirmRaggedRight(BLOBNBOX* bbox, int min_gutter) {
|
|
|
|
TBOX search_box(bbox->bounding_box());
|
|
|
|
search_box.set_left(search_box.right());
|
|
|
|
search_box.set_right(search_box.right() + min_gutter);
|
|
|
|
return NothingYOverlapsInBox(search_box, bbox->bounding_box());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if there is nothing in the given search_box that vertically
|
|
|
|
// overlaps target_box other than target_box itself.
|
|
|
|
bool TabFind::NothingYOverlapsInBox(const TBOX& search_box,
|
|
|
|
const TBOX& target_box) {
|
|
|
|
BlobGridSearch rsearch(this);
|
|
|
|
rsearch.StartRectSearch(search_box);
|
|
|
|
BLOBNBOX* blob;
|
|
|
|
while ((blob = rsearch.NextRectSearch()) != NULL) {
|
|
|
|
const TBOX& box = blob->bounding_box();
|
|
|
|
if (box.y_overlap(target_box) && !(box == target_box))
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2011-03-22 05:48:17 +08:00
|
|
|
void TabFind::FindAllTabVectors(int min_gutter_width) {
|
2009-07-11 10:39:56 +08:00
|
|
|
// A list of vectors that will be created in estimating the skew.
|
|
|
|
TabVector_LIST dummy_vectors;
|
|
|
|
// An estimate of the vertical direction, revised as more lines are added.
|
|
|
|
int vertical_x = 0;
|
|
|
|
int vertical_y = 1;
|
|
|
|
// Find an estimate of the vertical direction by finding some tab vectors.
|
|
|
|
// Slowly up the search size until we get some vectors.
|
|
|
|
for (int search_size = kMinVerticalSearch; search_size < kMaxVerticalSearch;
|
|
|
|
search_size += kMinVerticalSearch) {
|
|
|
|
int vector_count = FindTabVectors(search_size, TA_LEFT_ALIGNED,
|
2011-03-22 05:48:17 +08:00
|
|
|
min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors,
|
|
|
|
&vertical_x, &vertical_y);
|
|
|
|
vector_count += FindTabVectors(search_size, TA_RIGHT_ALIGNED,
|
2011-03-22 05:48:17 +08:00
|
|
|
min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors,
|
|
|
|
&vertical_x, &vertical_y);
|
|
|
|
if (vector_count > 0)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
// Get rid of the test vectors and reset the types of the tabs.
|
|
|
|
dummy_vectors.clear();
|
2012-02-02 10:53:04 +08:00
|
|
|
for (int i = 0; i < left_tab_boxes_.size(); ++i) {
|
|
|
|
BLOBNBOX* bbox = left_tab_boxes_[i];
|
2009-07-11 10:39:56 +08:00
|
|
|
if (bbox->left_tab_type() == TT_CONFIRMED)
|
2012-02-02 10:53:04 +08:00
|
|
|
bbox->set_left_tab_type(TT_MAYBE_ALIGNED);
|
|
|
|
}
|
|
|
|
for (int i = 0; i < right_tab_boxes_.size(); ++i) {
|
|
|
|
BLOBNBOX* bbox = right_tab_boxes_[i];
|
2009-07-11 10:39:56 +08:00
|
|
|
if (bbox->right_tab_type() == TT_CONFIRMED)
|
2012-02-02 10:53:04 +08:00
|
|
|
bbox->set_right_tab_type(TT_MAYBE_ALIGNED);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
if (textord_debug_tabfind) {
|
|
|
|
tprintf("Beginning real tab search with vertical = %d,%d...\n",
|
|
|
|
vertical_x, vertical_y);
|
|
|
|
}
|
|
|
|
// Now do the real thing ,but keep the vectors in the dummy_vectors list
|
|
|
|
// until they are all done, so we don't get the tab vectors confused with
|
|
|
|
// the rule line vectors.
|
2011-03-22 05:48:17 +08:00
|
|
|
FindTabVectors(kMaxVerticalSearch, TA_LEFT_ALIGNED, min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors, &vertical_x, &vertical_y);
|
2011-03-22 05:48:17 +08:00
|
|
|
FindTabVectors(kMaxVerticalSearch, TA_RIGHT_ALIGNED, min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors, &vertical_x, &vertical_y);
|
2011-03-22 05:48:17 +08:00
|
|
|
FindTabVectors(kMaxRaggedSearch, TA_LEFT_RAGGED, min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors, &vertical_x, &vertical_y);
|
2011-03-22 05:48:17 +08:00
|
|
|
FindTabVectors(kMaxRaggedSearch, TA_RIGHT_RAGGED, min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
&dummy_vectors, &vertical_x, &vertical_y);
|
|
|
|
// Now add the vectors to the vectors_ list.
|
|
|
|
TabVector_IT v_it(&vectors_);
|
|
|
|
v_it.add_list_after(&dummy_vectors);
|
|
|
|
// Now use the summed (mean) vertical vector as the direction for everything.
|
|
|
|
SetVerticalSkewAndParellelize(vertical_x, vertical_y);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Helper for FindAllTabVectors finds the vectors of a particular type.
|
|
|
|
int TabFind::FindTabVectors(int search_size_multiple, TabAlignment alignment,
|
2011-03-22 05:48:17 +08:00
|
|
|
int min_gutter_width, TabVector_LIST* vectors,
|
2009-07-11 10:39:56 +08:00
|
|
|
int* vertical_x, int* vertical_y) {
|
|
|
|
TabVector_IT vector_it(vectors);
|
|
|
|
int vector_count = 0;
|
2012-02-02 10:53:04 +08:00
|
|
|
// Search the right or left tab boxes, looking for tab vectors.
|
2009-07-11 10:39:56 +08:00
|
|
|
bool right = alignment == TA_RIGHT_ALIGNED || alignment == TA_RIGHT_RAGGED;
|
2012-02-02 10:53:04 +08:00
|
|
|
const GenericVector<BLOBNBOX*>& boxes = right ? right_tab_boxes_
|
|
|
|
: left_tab_boxes_;
|
|
|
|
for (int i = 0; i < boxes.size(); ++i) {
|
|
|
|
BLOBNBOX* bbox = boxes[i];
|
|
|
|
if ((!right && bbox->left_tab_type() == TT_MAYBE_ALIGNED) ||
|
|
|
|
(right && bbox->right_tab_type() == TT_MAYBE_ALIGNED)) {
|
2011-03-22 05:48:17 +08:00
|
|
|
TabVector* vector = FindTabVector(search_size_multiple, min_gutter_width,
|
|
|
|
alignment,
|
2009-07-11 10:39:56 +08:00
|
|
|
bbox, vertical_x, vertical_y);
|
|
|
|
if (vector != NULL) {
|
|
|
|
++vector_count;
|
|
|
|
vector_it.add_to_end(vector);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return vector_count;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Finds a vector corresponding to a tabstop running through the
|
|
|
|
// given box of the given alignment type.
|
|
|
|
// search_size_multiple is a multiple of height used to control
|
|
|
|
// the size of the search.
|
|
|
|
// vertical_x and y are updated with an estimate of the real
|
|
|
|
// vertical direction. (skew finding.)
|
|
|
|
// Returns NULL if no decent tabstop can be found.
|
|
|
|
TabVector* TabFind::FindTabVector(int search_size_multiple,
|
2011-03-22 05:48:17 +08:00
|
|
|
int min_gutter_width,
|
2009-07-11 10:39:56 +08:00
|
|
|
TabAlignment alignment,
|
|
|
|
BLOBNBOX* bbox,
|
|
|
|
int* vertical_x, int* vertical_y) {
|
2012-02-02 10:53:04 +08:00
|
|
|
int height = MAX(bbox->bounding_box().height(), gridsize());
|
2009-07-11 10:39:56 +08:00
|
|
|
AlignedBlobParams align_params(*vertical_x, *vertical_y,
|
2012-02-02 10:53:04 +08:00
|
|
|
height,
|
2011-03-22 05:48:17 +08:00
|
|
|
search_size_multiple, min_gutter_width,
|
|
|
|
resolution_, alignment);
|
2009-07-11 10:39:56 +08:00
|
|
|
// FindVerticalAlignment is in the parent (AlignedBlob) class.
|
|
|
|
return FindVerticalAlignment(align_params, bbox, vertical_x, vertical_y);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the vertical_skew_ member from the given vector and refit
|
|
|
|
// all vectors parallel to the skew vector.
|
|
|
|
void TabFind::SetVerticalSkewAndParellelize(int vertical_x, int vertical_y) {
|
|
|
|
// Fit the vertical vector into an ICOORD, which is 16 bit.
|
|
|
|
vertical_skew_.set_with_shrink(vertical_x, vertical_y);
|
|
|
|
if (textord_debug_tabfind)
|
|
|
|
tprintf("Vertical skew vector=(%d,%d)\n",
|
|
|
|
vertical_skew_.x(), vertical_skew_.y());
|
|
|
|
v_it_.set_to_list(&vectors_);
|
|
|
|
for (v_it_.mark_cycle_pt(); !v_it_.cycled_list(); v_it_.forward()) {
|
|
|
|
TabVector* v = v_it_.data();
|
|
|
|
v->Fit(vertical_skew_, true);
|
|
|
|
}
|
|
|
|
// Now sort the vectors as their direction has potentially changed.
|
|
|
|
SortVectors();
|
|
|
|
}
|
|
|
|
|
|
|
|
// Sort all the current vectors using the given vertical direction vector.
|
|
|
|
void TabFind::SortVectors() {
|
|
|
|
vectors_.sort(TabVector::SortVectorsByKey);
|
|
|
|
v_it_.set_to_list(&vectors_);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Evaluate all the current tab vectors.
|
|
|
|
void TabFind::EvaluateTabs() {
|
|
|
|
TabVector_IT rule_it(&vectors_);
|
|
|
|
for (rule_it.mark_cycle_pt(); !rule_it.cycled_list(); rule_it.forward()) {
|
|
|
|
TabVector* tab = rule_it.data();
|
|
|
|
if (!tab->IsSeparator()) {
|
|
|
|
tab->Evaluate(vertical_skew_, this);
|
|
|
|
if (tab->BoxCount() < kMinEvaluatedTabs) {
|
|
|
|
if (textord_debug_tabfind > 2)
|
|
|
|
tab->Print("Too few boxes");
|
|
|
|
delete rule_it.extract();
|
|
|
|
v_it_.set_to_list(&vectors_);
|
|
|
|
} else if (WithinTestRegion(3, tab->startpt().x(), tab->startpt().y())) {
|
|
|
|
tab->Print("Evaluated tab");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Trace textlines from one side to the other of each tab vector, saving
|
|
|
|
// the most frequent column widths found in a list so that a given width
|
|
|
|
// can be tested for being a common width with a simple callback function.
|
2012-02-02 10:53:04 +08:00
|
|
|
void TabFind::ComputeColumnWidths(ScrollView* tab_win,
|
|
|
|
ColPartitionGrid* part_grid) {
|
2012-03-02 06:43:51 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2009-07-11 10:39:56 +08:00
|
|
|
if (tab_win != NULL)
|
|
|
|
tab_win->Pen(ScrollView::WHITE);
|
2012-09-21 23:24:15 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2009-07-11 10:39:56 +08:00
|
|
|
// Accumulate column sections into a STATS
|
2012-02-02 10:53:04 +08:00
|
|
|
int col_widths_size = (tright_.x() - bleft_.x()) / kColumnWidthFactor;
|
2009-07-11 10:39:56 +08:00
|
|
|
STATS col_widths(0, col_widths_size + 1);
|
2012-02-02 10:53:04 +08:00
|
|
|
ApplyPartitionsToColumnWidths(part_grid, &col_widths);
|
2012-09-21 23:24:15 +08:00
|
|
|
#ifndef GRAPHICS_DISABLED
|
2009-07-11 10:39:56 +08:00
|
|
|
if (tab_win != NULL) {
|
|
|
|
tab_win->Update();
|
|
|
|
}
|
2012-09-21 23:24:15 +08:00
|
|
|
#endif // GRAPHICS_DISABLED
|
2012-02-02 10:53:04 +08:00
|
|
|
if (textord_debug_tabfind > 1)
|
|
|
|
col_widths.print();
|
2009-07-11 10:39:56 +08:00
|
|
|
// Now make a list of column widths.
|
2012-09-21 23:24:15 +08:00
|
|
|
MakeColumnWidths(col_widths_size, &col_widths);
|
2012-02-02 10:53:04 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Find column width and pair-up tab vectors with existing ColPartitions.
|
|
|
|
void TabFind::ApplyPartitionsToColumnWidths(ColPartitionGrid* part_grid,
|
|
|
|
STATS* col_widths) {
|
|
|
|
// For every ColPartition in the part_grid, add partners to the tabvectors
|
|
|
|
// and accumulate the column widths.
|
|
|
|
ColPartitionGridSearch gsearch(part_grid);
|
|
|
|
gsearch.StartFullSearch();
|
|
|
|
ColPartition* part;
|
|
|
|
while ((part = gsearch.NextFullSearch()) != NULL) {
|
|
|
|
BLOBNBOX_C_IT blob_it(part->boxes());
|
|
|
|
if (blob_it.empty())
|
|
|
|
continue;
|
|
|
|
BLOBNBOX* left_blob = blob_it.data();
|
|
|
|
blob_it.move_to_last();
|
|
|
|
BLOBNBOX* right_blob = blob_it.data();
|
|
|
|
TabVector* left_vector = LeftTabForBox(left_blob->bounding_box(),
|
|
|
|
true, false);
|
|
|
|
if (left_vector == NULL || left_vector->IsRightTab())
|
|
|
|
continue;
|
|
|
|
TabVector* right_vector = RightTabForBox(right_blob->bounding_box(),
|
|
|
|
true, false);
|
|
|
|
if (right_vector == NULL || right_vector->IsLeftTab())
|
|
|
|
continue;
|
|
|
|
|
|
|
|
AddPartnerVector(left_blob, right_blob, left_vector, right_vector);
|
|
|
|
int line_left = left_vector->XAtY(left_blob->bounding_box().bottom());
|
|
|
|
int line_right = right_vector->XAtY(right_blob->bounding_box().bottom());
|
|
|
|
// Add to STATS of measurements if the width is significant.
|
|
|
|
int width = line_right - line_left;
|
|
|
|
if (width >= kMinColumnWidth)
|
|
|
|
col_widths->add(width / kColumnWidthFactor, 1);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Helper makes the list of common column widths in column_widths_ from the
|
|
|
|
// input col_widths. Destroys the content of col_widths by repeatedly
|
|
|
|
// finding the mode and erasing the peak.
|
|
|
|
void TabFind::MakeColumnWidths(int col_widths_size, STATS* col_widths) {
|
2009-07-11 10:39:56 +08:00
|
|
|
ICOORDELT_IT w_it(&column_widths_);
|
2012-02-02 10:53:04 +08:00
|
|
|
int total_col_count = col_widths->get_total();
|
|
|
|
while (col_widths->get_total() > 0) {
|
|
|
|
int width = col_widths->mode();
|
|
|
|
int col_count = col_widths->pile_count(width);
|
|
|
|
col_widths->add(width, -col_count);
|
2009-07-11 10:39:56 +08:00
|
|
|
// Get the entire peak.
|
|
|
|
for (int left = width - 1; left > 0 &&
|
2012-02-02 10:53:04 +08:00
|
|
|
col_widths->pile_count(left) > 0;
|
2009-07-11 10:39:56 +08:00
|
|
|
--left) {
|
2012-02-02 10:53:04 +08:00
|
|
|
int new_count = col_widths->pile_count(left);
|
2009-07-11 10:39:56 +08:00
|
|
|
col_count += new_count;
|
2012-02-02 10:53:04 +08:00
|
|
|
col_widths->add(left, -new_count);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
for (int right = width + 1; right < col_widths_size &&
|
2012-02-02 10:53:04 +08:00
|
|
|
col_widths->pile_count(right) > 0;
|
2009-07-11 10:39:56 +08:00
|
|
|
++right) {
|
2012-02-02 10:53:04 +08:00
|
|
|
int new_count = col_widths->pile_count(right);
|
2009-07-11 10:39:56 +08:00
|
|
|
col_count += new_count;
|
2012-02-02 10:53:04 +08:00
|
|
|
col_widths->add(right, -new_count);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
if (col_count > kMinLinesInColumn &&
|
|
|
|
col_count > kMinFractionalLinesInColumn * total_col_count) {
|
|
|
|
ICOORDELT* w = new ICOORDELT(width, col_count);
|
|
|
|
w_it.add_after_then_move(w);
|
|
|
|
if (textord_debug_tabfind)
|
|
|
|
tprintf("Column of width %d has %d = %.2f%% lines\n",
|
|
|
|
width * kColumnWidthFactor, col_count,
|
|
|
|
100.0 * col_count / total_col_count);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Mark blobs as being in a vertical text line where that is the case.
|
|
|
|
// Returns true if the majority of the image is vertical text lines.
|
|
|
|
void TabFind::MarkVerticalText() {
|
2010-11-24 02:34:14 +08:00
|
|
|
if (textord_debug_tabfind)
|
|
|
|
tprintf("Checking for vertical lines\n");
|
|
|
|
BlobGridSearch gsearch(this);
|
|
|
|
gsearch.StartFullSearch();
|
|
|
|
BLOBNBOX* blob = NULL;
|
|
|
|
while ((blob = gsearch.NextFullSearch()) != NULL) {
|
|
|
|
if (blob->region_type() < BRT_UNKNOWN)
|
|
|
|
continue;
|
|
|
|
if (blob->UniquelyVertical()) {
|
|
|
|
blob->set_region_type(BRT_VERT_TEXT);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2010-11-24 02:34:14 +08:00
|
|
|
int TabFind::FindMedianGutterWidth(TabVector_LIST *lines) {
|
|
|
|
TabVector_IT it(lines);
|
|
|
|
int prev_right = -1;
|
|
|
|
int max_gap = static_cast<int>(kMaxGutterWidthAbsolute * resolution_);
|
|
|
|
STATS gaps(0, max_gap);
|
|
|
|
STATS heights(0, max_gap);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
TabVector* partner = v->GetSinglePartner();
|
|
|
|
if (!v->IsLeftTab() || v->IsSeparator() || !partner) continue;
|
|
|
|
heights.add(partner->startpt().x() - v->startpt().x(), 1);
|
|
|
|
if (prev_right > 0 && v->startpt().x() > prev_right) {
|
|
|
|
gaps.add(v->startpt().x() - prev_right, 1);
|
|
|
|
}
|
|
|
|
prev_right = partner->startpt().x();
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
2010-11-24 02:34:14 +08:00
|
|
|
if (textord_debug_tabfind)
|
|
|
|
tprintf("TabGutter total %d median_gap %.2f median_hgt %.2f\n",
|
|
|
|
gaps.get_total(), gaps.median(), heights.median());
|
|
|
|
if (gaps.get_total() < kMinLinesInColumn) return 0;
|
|
|
|
return static_cast<int>(gaps.median());
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
|
2012-02-02 10:53:04 +08:00
|
|
|
// Find the next adjacent (looking to the left or right) blob on this text
|
|
|
|
// line, with the constraint that it must vertically significantly overlap
|
2009-07-11 10:39:56 +08:00
|
|
|
// the [top_y, bottom_y] range.
|
|
|
|
// If ignore_images is true, then blobs with aligned_text() < 0 are treated
|
|
|
|
// as if they do not exist.
|
|
|
|
BLOBNBOX* TabFind::AdjacentBlob(const BLOBNBOX* bbox,
|
2012-02-02 10:53:04 +08:00
|
|
|
bool look_left, bool ignore_images,
|
|
|
|
double min_overlap_fraction,
|
2009-07-11 10:39:56 +08:00
|
|
|
int gap_limit, int top_y, int bottom_y) {
|
|
|
|
GridSearch<BLOBNBOX, BLOBNBOX_CLIST, BLOBNBOX_C_IT> sidesearch(this);
|
|
|
|
const TBOX& box = bbox->bounding_box();
|
|
|
|
int left = box.left();
|
|
|
|
int right = box.right();
|
|
|
|
int mid_x = (left + right) / 2;
|
|
|
|
sidesearch.StartSideSearch(mid_x, bottom_y, top_y);
|
|
|
|
int best_gap = 0;
|
2012-02-02 10:53:04 +08:00
|
|
|
bool debug = WithinTestRegion(3, left, bottom_y);
|
2009-07-11 10:39:56 +08:00
|
|
|
BLOBNBOX* result = NULL;
|
|
|
|
BLOBNBOX* neighbour = NULL;
|
2012-02-02 10:53:04 +08:00
|
|
|
while ((neighbour = sidesearch.NextSideSearch(look_left)) != NULL) {
|
|
|
|
if (debug) {
|
|
|
|
tprintf("Adjacent blob: considering box:");
|
|
|
|
neighbour->bounding_box().print();
|
|
|
|
}
|
2009-07-11 10:39:56 +08:00
|
|
|
if (neighbour == bbox ||
|
|
|
|
(ignore_images && neighbour->region_type() < BRT_UNKNOWN))
|
|
|
|
continue;
|
|
|
|
const TBOX& nbox = neighbour->bounding_box();
|
|
|
|
int n_top_y = nbox.top();
|
|
|
|
int n_bottom_y = nbox.bottom();
|
|
|
|
int v_overlap = MIN(n_top_y, top_y) - MAX(n_bottom_y, bottom_y);
|
|
|
|
int height = top_y - bottom_y;
|
|
|
|
int n_height = n_top_y - n_bottom_y;
|
2012-02-02 10:53:04 +08:00
|
|
|
if (v_overlap > min_overlap_fraction * MIN(height, n_height) &&
|
|
|
|
(min_overlap_fraction == 0.0 || !DifferentSizes(height, n_height))) {
|
2009-07-11 10:39:56 +08:00
|
|
|
int n_left = nbox.left();
|
|
|
|
int n_right = nbox.right();
|
|
|
|
int h_gap = MAX(n_left, left) - MIN(n_right, right);
|
|
|
|
int n_mid_x = (n_left + n_right) / 2;
|
2012-02-02 10:53:04 +08:00
|
|
|
if (look_left == (n_mid_x < mid_x) && n_mid_x != mid_x) {
|
2009-07-11 10:39:56 +08:00
|
|
|
if (h_gap > gap_limit) {
|
|
|
|
// Hit a big gap before next tab so don't return anything.
|
2012-02-02 10:53:04 +08:00
|
|
|
if (debug)
|
2009-07-11 10:39:56 +08:00
|
|
|
tprintf("Giving up due to big gap = %d vs %d\n",
|
|
|
|
h_gap, gap_limit);
|
|
|
|
return result;
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
if (h_gap > 0 && (look_left ? neighbour->right_tab_type()
|
|
|
|
: neighbour->left_tab_type()) >= TT_CONFIRMED) {
|
2009-07-11 10:39:56 +08:00
|
|
|
// Hit a tab facing the wrong way. Stop in case we are crossing
|
|
|
|
// the column boundary.
|
2012-02-02 10:53:04 +08:00
|
|
|
if (debug)
|
2009-07-11 10:39:56 +08:00
|
|
|
tprintf("Collision with like tab of type %d at %d,%d\n",
|
2012-02-02 10:53:04 +08:00
|
|
|
look_left ? neighbour->right_tab_type()
|
2009-07-11 10:39:56 +08:00
|
|
|
: neighbour->left_tab_type(),
|
|
|
|
n_left, nbox.bottom());
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
// This is a good fit to the line. Continue with this
|
|
|
|
// neighbour as the bbox if the best gap.
|
|
|
|
if (result == NULL || h_gap < best_gap) {
|
2012-02-02 10:53:04 +08:00
|
|
|
if (debug)
|
2009-07-11 10:39:56 +08:00
|
|
|
tprintf("Good result\n");
|
|
|
|
result = neighbour;
|
|
|
|
best_gap = h_gap;
|
|
|
|
} else {
|
|
|
|
// The new one is worse, so we probably already have the best result.
|
|
|
|
return result;
|
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
} else if (debug) {
|
|
|
|
tprintf("Wrong way\n");
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
2012-02-02 10:53:04 +08:00
|
|
|
} else if (debug) {
|
|
|
|
tprintf("Insufficient overlap\n");
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if (WithinTestRegion(3, left, box.top()))
|
|
|
|
tprintf("Giving up due to end of search\n");
|
|
|
|
return result; // Hit the edge and found nothing.
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add a bi-directional partner relationship between the left
|
|
|
|
// and the right. If one (or both) of the vectors is a separator,
|
|
|
|
// extend a nearby extendable vector or create a new one of the
|
|
|
|
// correct type, using the given left or right blob as a guide.
|
|
|
|
void TabFind::AddPartnerVector(BLOBNBOX* left_blob, BLOBNBOX* right_blob,
|
|
|
|
TabVector* left, TabVector* right) {
|
|
|
|
const TBOX& left_box = left_blob->bounding_box();
|
|
|
|
const TBOX& right_box = right_blob->bounding_box();
|
|
|
|
if (left->IsSeparator()) {
|
|
|
|
// Try to find a nearby left edge to extend.
|
|
|
|
TabVector* v = LeftTabForBox(left_box, true, true);
|
|
|
|
if (v != NULL && v != left && v->IsLeftTab() &&
|
|
|
|
v->XAtY(left_box.top()) > left->XAtY(left_box.top())) {
|
|
|
|
left = v; // Found a good replacement.
|
|
|
|
left->ExtendToBox(left_blob);
|
|
|
|
} else {
|
|
|
|
// Fake a vector.
|
|
|
|
left = new TabVector(*left, TA_LEFT_RAGGED, vertical_skew_, left_blob);
|
|
|
|
vectors_.add_sorted(TabVector::SortVectorsByKey, left);
|
|
|
|
v_it_.move_to_first();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (right->IsSeparator()) {
|
|
|
|
// Try to find a nearby left edge to extend.
|
|
|
|
if (WithinTestRegion(3, right_box.right(), right_box.bottom())) {
|
|
|
|
tprintf("Box edge (%d,%d-%d)",
|
|
|
|
right_box.right(), right_box.bottom(), right_box.top());
|
|
|
|
right->Print(" looking for improvement for");
|
|
|
|
}
|
|
|
|
TabVector* v = RightTabForBox(right_box, true, true);
|
|
|
|
if (v != NULL && v != right && v->IsRightTab() &&
|
|
|
|
v->XAtY(right_box.top()) < right->XAtY(right_box.top())) {
|
|
|
|
right = v; // Found a good replacement.
|
|
|
|
right->ExtendToBox(right_blob);
|
|
|
|
if (WithinTestRegion(3, right_box.right(), right_box.bottom())) {
|
|
|
|
right->Print("Extended vector");
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Fake a vector.
|
|
|
|
right = new TabVector(*right, TA_RIGHT_RAGGED, vertical_skew_,
|
|
|
|
right_blob);
|
|
|
|
vectors_.add_sorted(TabVector::SortVectorsByKey, right);
|
|
|
|
v_it_.move_to_first();
|
|
|
|
if (WithinTestRegion(3, right_box.right(), right_box.bottom())) {
|
|
|
|
right->Print("Created new vector");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
left->AddPartner(right);
|
|
|
|
right->AddPartner(left);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove separators and unused tabs from the main vectors_ list
|
|
|
|
// to the dead_vectors_ list.
|
|
|
|
void TabFind::CleanupTabs() {
|
|
|
|
// TODO(rays) Before getting rid of separators and unused vectors, it
|
|
|
|
// would be useful to try moving ragged vectors outwards to see if this
|
|
|
|
// allows useful extension. Could be combined with checking ends of partners.
|
|
|
|
TabVector_IT it(&vectors_);
|
|
|
|
TabVector_IT dead_it(&dead_vectors_);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
if (v->IsSeparator() || v->Partnerless()) {
|
|
|
|
dead_it.add_after_then_move(it.extract());
|
|
|
|
v_it_.set_to_list(&vectors_);
|
|
|
|
} else {
|
|
|
|
v->FitAndEvaluateIfNeeded(vertical_skew_, this);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2010-11-24 02:34:14 +08:00
|
|
|
// Apply the given rotation to the given list of blobs.
|
|
|
|
void TabFind::RotateBlobList(const FCOORD& rotation, BLOBNBOX_LIST* blobs) {
|
2009-07-11 10:39:56 +08:00
|
|
|
BLOBNBOX_IT it(blobs);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
it.data()->rotate_box(rotation);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Recreate the grid with deskewed BLOBNBOXes.
|
2010-11-24 02:34:14 +08:00
|
|
|
// Returns false if the detected skew angle is impossible.
|
|
|
|
bool TabFind::Deskew(TabVector_LIST* hlines, BLOBNBOX_LIST* image_blobs,
|
|
|
|
TO_BLOCK* block, FCOORD* deskew, FCOORD* reskew) {
|
|
|
|
ComputeDeskewVectors(deskew, reskew);
|
|
|
|
if (deskew->x() < kCosMaxSkewAngle)
|
|
|
|
return false;
|
|
|
|
RotateBlobList(*deskew, image_blobs);
|
|
|
|
RotateBlobList(*deskew, &block->blobs);
|
|
|
|
RotateBlobList(*deskew, &block->small_blobs);
|
|
|
|
RotateBlobList(*deskew, &block->noise_blobs);
|
2009-07-11 10:39:56 +08:00
|
|
|
if (textord_debug_images) {
|
|
|
|
// Rotate the debug pix and arrange for it to be drawn at the correct
|
|
|
|
// pixel offset.
|
|
|
|
Pix* pix_grey = pixRead(AlignedBlob::textord_debug_pix().string());
|
|
|
|
int width = pixGetWidth(pix_grey);
|
|
|
|
int height = pixGetHeight(pix_grey);
|
2010-11-24 02:34:14 +08:00
|
|
|
float angle = atan2(deskew->y(), deskew->x());
|
2009-07-11 10:39:56 +08:00
|
|
|
// Positive angle is clockwise to pixRotate.
|
|
|
|
Pix* pix_rot = pixRotate(pix_grey, -angle, L_ROTATE_AREA_MAP,
|
|
|
|
L_BRING_IN_WHITE, width, height);
|
|
|
|
// The image must be translated by the rotation of its center, since it
|
|
|
|
// has just been rotated about its center.
|
|
|
|
ICOORD center_offset(width / 2, height / 2);
|
|
|
|
ICOORD new_center_offset(center_offset);
|
2010-11-24 02:34:14 +08:00
|
|
|
new_center_offset.rotate(*deskew);
|
2009-07-11 10:39:56 +08:00
|
|
|
image_origin_ += new_center_offset - center_offset;
|
|
|
|
// The image grew as it was rotated, so offset the (top/left) origin
|
|
|
|
// by half the change in size. y is opposite to x because it is drawn
|
|
|
|
// at ist top/left, not bottom/left.
|
|
|
|
ICOORD corner_offset((width - pixGetWidth(pix_rot)) / 2,
|
|
|
|
(pixGetHeight(pix_rot) - height) / 2);
|
|
|
|
image_origin_ += corner_offset;
|
|
|
|
pixWrite(AlignedBlob::textord_debug_pix().string(), pix_rot, IFF_PNG);
|
|
|
|
pixDestroy(&pix_grey);
|
|
|
|
pixDestroy(&pix_rot);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Rotate the horizontal vectors. The vertical vectors don't need
|
|
|
|
// rotating as they can just be refitted.
|
|
|
|
TabVector_IT h_it(hlines);
|
|
|
|
for (h_it.mark_cycle_pt(); !h_it.cycled_list(); h_it.forward()) {
|
|
|
|
TabVector* h = h_it.data();
|
2010-11-24 02:34:14 +08:00
|
|
|
h->Rotate(*deskew);
|
|
|
|
}
|
|
|
|
TabVector_IT d_it(&dead_vectors_);
|
|
|
|
for (d_it.mark_cycle_pt(); !d_it.cycled_list(); d_it.forward()) {
|
|
|
|
TabVector* d = d_it.data();
|
|
|
|
d->Rotate(*deskew);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
SetVerticalSkewAndParellelize(0, 1);
|
|
|
|
// Rebuild the grid to the new size.
|
|
|
|
TBOX grid_box(bleft_, tright_);
|
2010-11-24 02:34:14 +08:00
|
|
|
grid_box.rotate_large(*deskew);
|
2009-07-11 10:39:56 +08:00
|
|
|
Init(gridsize(), grid_box.botleft(), grid_box.topright());
|
2012-02-02 10:53:04 +08:00
|
|
|
InsertBlobsToGrid(false, false, image_blobs, this);
|
|
|
|
InsertBlobsToGrid(true, false, &block->blobs, this);
|
2010-11-24 02:34:14 +08:00
|
|
|
return true;
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
|
2010-11-24 02:34:14 +08:00
|
|
|
// Flip the vertical and horizontal lines and rotate the grid ready
|
|
|
|
// for working on the rotated image.
|
|
|
|
// This also makes parameter adjustments for FindInitialTabVectors().
|
|
|
|
void TabFind::ResetForVerticalText(const FCOORD& rotate, const FCOORD& rerotate,
|
|
|
|
TabVector_LIST* horizontal_lines,
|
|
|
|
int* min_gutter_width) {
|
2009-07-11 10:39:56 +08:00
|
|
|
// Rotate the horizontal and vertical vectors and swap them over.
|
2010-11-24 02:34:14 +08:00
|
|
|
// Only the separators are kept and rotated; other tabs are used
|
|
|
|
// to estimate the gutter width then thrown away.
|
2009-07-11 10:39:56 +08:00
|
|
|
TabVector_LIST ex_verticals;
|
|
|
|
TabVector_IT ex_v_it(&ex_verticals);
|
2010-11-24 02:34:14 +08:00
|
|
|
TabVector_LIST vlines;
|
|
|
|
TabVector_IT v_it(&vlines);
|
2009-07-11 10:39:56 +08:00
|
|
|
while (!v_it_.empty()) {
|
|
|
|
TabVector* v = v_it_.extract();
|
|
|
|
if (v->IsSeparator()) {
|
2010-11-24 02:34:14 +08:00
|
|
|
v->Rotate(rotate);
|
2009-07-11 10:39:56 +08:00
|
|
|
ex_v_it.add_after_then_move(v);
|
|
|
|
} else {
|
2010-11-24 02:34:14 +08:00
|
|
|
v_it.add_after_then_move(v);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
v_it_.forward();
|
|
|
|
}
|
2010-11-24 02:34:14 +08:00
|
|
|
|
|
|
|
// Adjust the min gutter width for better tabbox selection
|
|
|
|
// in 2nd call to FindInitialTabVectors().
|
|
|
|
int median_gutter = FindMedianGutterWidth(&vlines);
|
|
|
|
if (median_gutter > *min_gutter_width)
|
|
|
|
*min_gutter_width = median_gutter;
|
|
|
|
|
|
|
|
TabVector_IT h_it(horizontal_lines);
|
2009-07-11 10:39:56 +08:00
|
|
|
for (h_it.mark_cycle_pt(); !h_it.cycled_list(); h_it.forward()) {
|
|
|
|
TabVector* h = h_it.data();
|
2010-11-24 02:34:14 +08:00
|
|
|
h->Rotate(rotate);
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
2010-11-24 02:34:14 +08:00
|
|
|
v_it_.add_list_after(horizontal_lines);
|
2009-07-11 10:39:56 +08:00
|
|
|
v_it_.move_to_first();
|
2010-11-24 02:34:14 +08:00
|
|
|
h_it.set_to_list(horizontal_lines);
|
2009-07-11 10:39:56 +08:00
|
|
|
h_it.add_list_after(&ex_verticals);
|
|
|
|
|
|
|
|
// Rebuild the grid to the new size.
|
2010-11-24 02:34:14 +08:00
|
|
|
TBOX grid_box(bleft(), tright());
|
|
|
|
grid_box.rotate_large(rotate);
|
2009-07-11 10:39:56 +08:00
|
|
|
Init(gridsize(), grid_box.botleft(), grid_box.topright());
|
2012-02-02 10:53:04 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Clear the grid and get rid of the tab vectors, but not separators,
|
|
|
|
// ready to start again.
|
|
|
|
void TabFind::Reset() {
|
|
|
|
v_it_.move_to_first();
|
|
|
|
for (v_it_.mark_cycle_pt(); !v_it_.cycled_list(); v_it_.forward()) {
|
|
|
|
if (!v_it_.data()->IsSeparator())
|
|
|
|
delete v_it_.extract();
|
|
|
|
}
|
|
|
|
Clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
// Reflect the separator tab vectors and the grids in the y-axis.
|
|
|
|
// Can only be called after Reset!
|
|
|
|
void TabFind::ReflectInYAxis() {
|
|
|
|
TabVector_LIST temp_list;
|
|
|
|
TabVector_IT temp_it(&temp_list);
|
|
|
|
v_it_.move_to_first();
|
|
|
|
// The TabVector list only contains vertical lines, but they need to be
|
|
|
|
// reflected and the list needs to be reversed, so they are still in
|
|
|
|
// sort_key order.
|
|
|
|
while (!v_it_.empty()) {
|
|
|
|
TabVector* v = v_it_.extract();
|
|
|
|
v_it_.forward();
|
|
|
|
v->ReflectInYAxis();
|
|
|
|
temp_it.add_before_then_move(v);
|
|
|
|
}
|
|
|
|
v_it_.add_list_after(&temp_list);
|
|
|
|
v_it_.move_to_first();
|
|
|
|
// Reset this grid with reflected bounding boxes.
|
|
|
|
TBOX grid_box(bleft(), tright());
|
|
|
|
int tmp = grid_box.left();
|
|
|
|
grid_box.set_left(-grid_box.right());
|
|
|
|
grid_box.set_right(-tmp);
|
|
|
|
Init(gridsize(), grid_box.botleft(), grid_box.topright());
|
2009-07-11 10:39:56 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Compute the rotation required to deskew, and its inverse rotation.
|
|
|
|
void TabFind::ComputeDeskewVectors(FCOORD* deskew, FCOORD* reskew) {
|
|
|
|
double length = vertical_skew_ % vertical_skew_;
|
|
|
|
length = sqrt(length);
|
2010-11-24 02:34:14 +08:00
|
|
|
deskew->set_x(static_cast<float>(vertical_skew_.y() / length));
|
|
|
|
deskew->set_y(static_cast<float>(vertical_skew_.x() / length));
|
2009-07-11 10:39:56 +08:00
|
|
|
reskew->set_x(deskew->x());
|
|
|
|
reskew->set_y(-deskew->y());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Compute and apply constraints to the end positions of TabVectors so
|
|
|
|
// that where possible partners end at the same y coordinate.
|
|
|
|
void TabFind::ApplyTabConstraints() {
|
|
|
|
TabVector_IT it(&vectors_);
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
v->SetupConstraints();
|
|
|
|
}
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
// With the first and last partner, we want a common bottom and top,
|
|
|
|
// respectively, and for each change of partner, we want a common
|
|
|
|
// top of first with bottom of next.
|
|
|
|
v->SetupPartnerConstraints();
|
|
|
|
}
|
|
|
|
// TODO(rays) The back-to-back pairs should really be done like the
|
|
|
|
// front-to-front pairs, but there is no convenient way of producing the
|
|
|
|
// list of partners like there is with the front-to-front.
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
if (!v->IsRightTab())
|
|
|
|
continue;
|
|
|
|
// For each back-to-back pair of vectors, try for common top and bottom.
|
|
|
|
TabVector_IT partner_it(it);
|
|
|
|
for (partner_it.forward(); !partner_it.at_first(); partner_it.forward()) {
|
|
|
|
TabVector* partner = partner_it.data();
|
|
|
|
if (!partner->IsLeftTab() || !v->VOverlap(*partner))
|
|
|
|
continue;
|
|
|
|
v->SetupPartnerConstraints(partner);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Now actually apply the constraints to get common start/end points.
|
|
|
|
for (it.mark_cycle_pt(); !it.cycled_list(); it.forward()) {
|
|
|
|
TabVector* v = it.data();
|
|
|
|
if (!v->IsSeparator())
|
|
|
|
v->ApplyConstraints();
|
|
|
|
}
|
|
|
|
// TODO(rays) Where constraint application fails, it would be good to try
|
|
|
|
// checking the ends to see if they really should be moved.
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace tesseract.
|