tesseract::ApplyBoxTest Class Reference

Inheritance diagram for tesseract::ApplyBoxTest:

Protected Member Functions
std::string	TestDataNameToPath (const std::string &name)
std::string	TessdataPath ()
	ApplyBoxTest ()
	~ApplyBoxTest () override
bool	SetImage (const char *filename)
void	VerifyBoxesAndText (const char *imagefile, const char *truth_str, const char *target_box_file, bool line_mode)
Protected Member Functions inherited from testing::Test
	Test ()
virtual void	SetUp ()
virtual void	TearDown ()

Protected Attributes
Image	src_pix_
std::string	ocr_text_
tesseract::TessBaseAPI	api_

Additional Inherited Members
Public Member Functions inherited from testing::Test
virtual	~Test ()
Static Public Member Functions inherited from testing::Test
static void	SetUpTestSuite ()
static void	TearDownTestSuite ()
static void	TearDownTestCase ()
static void	SetUpTestCase ()
static bool	HasFatalFailure ()
static bool	HasNonfatalFailure ()
static bool	IsSkipped ()
static bool	HasFailure ()
static void	RecordProperty (const std::string &key, const std::string &value)
static void	RecordProperty (const std::string &key, int value)

Detailed Description

Definition at line 27 of file applybox_test.cc.

Constructor & Destructor Documentation

ApplyBoxTest()

tesseract::ApplyBoxTest::ApplyBoxTest ( )

inlineprotected

Definition at line 36 of file applybox_test.cc.

                 {
    src_pix_ = nullptr;
  }

~ApplyBoxTest()

tesseract::ApplyBoxTest::~ApplyBoxTest ( )

inlineoverrideprotected

Definition at line 39 of file applybox_test.cc.

                           {
    src_pix_.destroy();
  }

Member Function Documentation

SetImage()

bool tesseract::ApplyBoxTest::SetImage ( const char *  filename )

inlineprotected

Definition at line 43 of file applybox_test.cc.

                                      {
    bool found = false;
    src_pix_.destroy();
    src_pix_ = pixRead(TestDataNameToPath(filename).c_str());
    if (api_.Init(TessdataPath().c_str(), "eng", tesseract::OEM_TESSERACT_ONLY) != -1) {
      api_.SetPageSegMode(tesseract::PSM_SINGLE_BLOCK);
      api_.SetImage(src_pix_);
      api_.SetVariable("tessedit_make_boxes_from_boxes", "1");
      api_.SetInputName(TestDataNameToPath(filename).c_str());
      found = true;
    }
    return found;
  }

TessdataPath()

std::string tesseract::ApplyBoxTest::TessdataPath ( )

inlineprotected

Definition at line 32 of file applybox_test.cc.

                           {
    return TESSDATA_DIR;
  }

TestDataNameToPath()

std::string tesseract::ApplyBoxTest::TestDataNameToPath ( const std::string &  name )

inlineprotected

Definition at line 29 of file applybox_test.cc.

                                                      {
    return file::JoinPath(TESTING_DIR, name);
  }

VerifyBoxesAndText()

void tesseract::ApplyBoxTest::VerifyBoxesAndText ( const char *  imagefile, const char *  truth_str, const char *  target_box_file, bool  line_mode  )

inlineprotected

Definition at line 62 of file applybox_test.cc.

                                          {
    if (!SetImage(imagefile)) {
      // eng.traineddata not found or other problem during Init.
      GTEST_SKIP();
      return;
    }
    if (line_mode) {
      api_.SetVariable("tessedit_resegment_from_line_boxes", "1");
    } else {
      api_.SetVariable("tessedit_resegment_from_boxes", "1");
    }
    api_.Recognize(nullptr);
    char *ocr_text = api_.GetUTF8Text();
    EXPECT_STREQ(truth_str, ocr_text);
    delete[] ocr_text;
    // Test the boxes by reading the target box file in parallel with the
    // bounding boxes in the ocr output.
    std::string box_filename = TestDataNameToPath(target_box_file);
    FILE *box_file = OpenBoxFile(box_filename.c_str());
    ASSERT_TRUE(box_file != nullptr);
    int height = pixGetHeight(src_pix_);
    ResultIterator *it = api_.GetIterator();
    do {
      int left, top, right, bottom;
      EXPECT_TRUE(it->BoundingBox(tesseract::RIL_SYMBOL, &left, &top, &right, &bottom));
      TBOX ocr_box(ICOORD(left, height - bottom), ICOORD(right, height - top));
      int line_number = 0;
      TBOX truth_box;
      std::string box_text;
      EXPECT_TRUE(ReadNextBox(0, &line_number, box_file, box_text, &truth_box));
      // Testing for major overlap is a bit weak, but if they all
      // major overlap successfully, then it has to be fairly close.
      EXPECT_TRUE(ocr_box.major_overlap(truth_box));
      // Also check that the symbol text matches the box text.
      char *symbol_text = it->GetUTF8Text(tesseract::RIL_SYMBOL);
      EXPECT_STREQ(box_text.c_str(), symbol_text);
      delete[] symbol_text;
    } while (it->Next(tesseract::RIL_SYMBOL));
    delete it;
  }

Member Data Documentation

api_

tesseract::TessBaseAPI tesseract::ApplyBoxTest::api_

protected

Definition at line 106 of file applybox_test.cc.

ocr_text_

std::string tesseract::ApplyBoxTest::ocr_text_

protected

Definition at line 105 of file applybox_test.cc.

src_pix_

Image tesseract::ApplyBoxTest::src_pix_

protected

Definition at line 104 of file applybox_test.cc.

---

The documentation for this class was generated from the following file:

• /media/home/debian/src/github/tesseract-ocr/tesseract/unittest/applybox_test.cc