94 lines
3.2 KiB
C++
94 lines
3.2 KiB
C++
|
/////////////////////////////////////////////////////////////////////
|
||
|
|
// File: ocrpara.cpp
|
||
|
|
// Description: OCR Paragraph Output Type
|
||
|
|
// Author: David Eger
|
||
|
|
//
|
||
|
|
// (C) Copyright 2010, 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 "ocrpara.h"
|
||
|
|
|
||
|
|
#include "host.h" // For NearlyEqual()
|
||
|
|
|
||
|
|
#include <cstdio>
|
||
|
|
|
||
|
|
namespace tesseract {
|
||
|
|
|
||
|
|
using tesseract::JUSTIFICATION_CENTER;
|
||
|
|
using tesseract::JUSTIFICATION_LEFT;
|
||
|
|
using tesseract::JUSTIFICATION_RIGHT;
|
||
|
|
using tesseract::JUSTIFICATION_UNKNOWN;
|
||
|
|
|
||
|
|
static const char *ParagraphJustificationToString(tesseract::ParagraphJustification justification) {
|
||
|
|
switch (justification) {
|
||
|
|
case JUSTIFICATION_LEFT:
|
||
|
|
return "LEFT";
|
||
|
|
case JUSTIFICATION_RIGHT:
|
||
|
|
return "RIGHT";
|
||
|
|
case JUSTIFICATION_CENTER:
|
||
|
|
return "CENTER";
|
||
|
|
default:
|
||
|
|
return "UNKNOWN";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ParagraphModel::ValidFirstLine(int lmargin, int lindent, int rindent, int rmargin) const {
|
||
|
|
switch (justification_) {
|
||
|
|
case JUSTIFICATION_LEFT:
|
||
|
|
return NearlyEqual(lmargin + lindent, margin_ + first_indent_, tolerance_);
|
||
|
|
case JUSTIFICATION_RIGHT:
|
||
|
|
return NearlyEqual(rmargin + rindent, margin_ + first_indent_, tolerance_);
|
||
|
|
case JUSTIFICATION_CENTER:
|
||
|
|
return NearlyEqual(lindent, rindent, tolerance_ * 2);
|
||
|
|
default:
|
||
|
|
// shouldn't happen
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ParagraphModel::ValidBodyLine(int lmargin, int lindent, int rindent, int rmargin) const {
|
||
|
|
switch (justification_) {
|
||
|
|
case JUSTIFICATION_LEFT:
|
||
|
|
return NearlyEqual(lmargin + lindent, margin_ + body_indent_, tolerance_);
|
||
|
|
case JUSTIFICATION_RIGHT:
|
||
|
|
return NearlyEqual(rmargin + rindent, margin_ + body_indent_, tolerance_);
|
||
|
|
case JUSTIFICATION_CENTER:
|
||
|
|
return NearlyEqual(lindent, rindent, tolerance_ * 2);
|
||
|
|
default:
|
||
|
|
// shouldn't happen
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ParagraphModel::Comparable(const ParagraphModel &other) const {
|
||
|
|
if (justification_ != other.justification_) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (justification_ == JUSTIFICATION_CENTER || justification_ == JUSTIFICATION_UNKNOWN) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
int tolerance = (tolerance_ + other.tolerance_) / 4;
|
||
|
|
return NearlyEqual(margin_ + first_indent_, other.margin_ + other.first_indent_, tolerance) &&
|
||
|
|
NearlyEqual(margin_ + body_indent_, other.margin_ + other.body_indent_, tolerance);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string ParagraphModel::ToString() const {
|
||
|
|
char buffer[200];
|
||
|
|
const char *alignment = ParagraphJustificationToString(justification_);
|
||
|
|
snprintf(buffer, sizeof(buffer), "margin: %d, first_indent: %d, body_indent: %d, alignment: %s",
|
||
|
|
margin_, first_indent_, body_indent_, alignment);
|
||
|
|
return std::string(buffer);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace tesseract
|