tesseract::LayoutTest Class Reference

Inheritance diagram for tesseract::LayoutTest:

Protected Member Functions
std::string	TestDataNameToPath (const std::string &name)
std::string	TessdataPath ()
	LayoutTest ()
	~LayoutTest () override
void	SetImage (const char *filename, const char *lang)
void	VerifyBlockTextOrder (const char *strings[], const PolyBlockType *blocks, ResultIterator *it)
void	VerifyRoughBlockOrder (bool right_to_left, ResultIterator *it)
void	VerifyTotalContainment (int winding_target, MutableIterator *it)
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 55 of file layout_test.cc.

Constructor & Destructor Documentation

LayoutTest()

tesseract::LayoutTest::LayoutTest ( )

inlineprotected

Definition at line 64 of file layout_test.cc.

               {
    src_pix_ = nullptr;
  }

~LayoutTest()

tesseract::LayoutTest::~LayoutTest ( )

inlineoverrideprotected

Definition at line 67 of file layout_test.cc.

                         {
    src_pix_.destroy();
  }

Member Function Documentation

SetImage()

void tesseract::LayoutTest::SetImage ( const char *  filename, const char *  lang  )

inlineprotected

Definition at line 71 of file layout_test.cc.

                                                        {
    src_pix_.destroy();
    src_pix_ = pixRead(TestDataNameToPath(filename).c_str());
    api_.Init(TessdataPath().c_str(), lang, tesseract::OEM_TESSERACT_ONLY);
    api_.SetPageSegMode(tesseract::PSM_AUTO);
    api_.SetImage(src_pix_);
  }

TessdataPath()

std::string tesseract::LayoutTest::TessdataPath ( )

inlineprotected

Definition at line 60 of file layout_test.cc.

                           {
    return file::JoinPath(TESSDATA_DIR, "");
  }

TestDataNameToPath()

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

inlineprotected

Definition at line 57 of file layout_test.cc.

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

VerifyBlockTextOrder()

void tesseract::LayoutTest::VerifyBlockTextOrder ( const char *  strings[], const PolyBlockType *  blocks, ResultIterator *  it  )

inlineprotected

Definition at line 84 of file layout_test.cc.

                                                {
    it->Begin();
    int string_index = 0;
    int block_index = 0;
    do {
      char *block_text = it->GetUTF8Text(tesseract::RIL_BLOCK);
      if (block_text != nullptr && it->BlockType() == blocks[string_index] &&
          strstr(block_text, strings[string_index]) != nullptr) {
        LOG(INFO) << "Found string " << strings[string_index] << " in block " << block_index
                  << " of type " << kPolyBlockNames[blocks[string_index]] << "\n";
        // Found this one.
        ++string_index;
      } else if (it->BlockType() == blocks[string_index] && block_text == nullptr &&
                 strings[string_index][0] == '\0') {
        LOG(INFO) << "Found block of type " << kPolyBlockNames[blocks[string_index]] << " at block "
                  << block_index << "\n";
        // Found this one.
        ++string_index;
      } else {
        LOG(INFO) << "No match found in block with text:\n" << block_text;
      }
      delete[] block_text;
      ++block_index;
      if (strings[string_index] == nullptr) {
        break;
      }
    } while (it->Next(tesseract::RIL_BLOCK));
    EXPECT_TRUE(strings[string_index] == nullptr);
  }

VerifyRoughBlockOrder()

void tesseract::LayoutTest::VerifyRoughBlockOrder ( bool  right_to_left, ResultIterator *  it  )

inlineprotected

Definition at line 120 of file layout_test.cc.

                                                                     {
    int prev_left = 0;
    int prev_right = 0;
    int prev_bottom = 0;
    it->Begin();
    do {
      int left, top, right, bottom;
      if (it->BoundingBox(tesseract::RIL_BLOCK, &left, &top, &right, &bottom) &&
          PTIsTextType(it->BlockType()) && right - left > 800 && bottom - top > 200) {
        if (prev_right > prev_left) {
          if (std::min(right, prev_right) > std::max(left, prev_left)) {
            EXPECT_GE(top, prev_bottom) << "Overlapping block should be below";
          } else if (top < prev_bottom) {
            if (right_to_left) {
              EXPECT_GE(prev_left, right) << "Block should be to the left";
            } else {
              EXPECT_GE(left, prev_right) << "Block should be to the right";
            }
          }
        }
        prev_left = left;
        prev_right = right;
        prev_bottom = bottom;
      }
    } while (it->Next(tesseract::RIL_BLOCK));
  }

VerifyTotalContainment()

void tesseract::LayoutTest::VerifyTotalContainment ( int  winding_target, MutableIterator *  it  )

inlineprotected

Definition at line 150 of file layout_test.cc.

                                                                       {
    it->Begin();
    do {
      int left, top, right, bottom;
      if (it->BoundingBox(tesseract::RIL_BLOCK, &left, &top, &right, &bottom) &&
          PTIsTextType(it->BlockType()) && right - left > 800 && bottom - top > 200) {
        const PAGE_RES_IT *pr_it = it->PageResIt();
        POLY_BLOCK *pb = pr_it->block()->block->pdblk.poly_block();
        CHECK(pb != nullptr);
        FCOORD skew = pr_it->block()->block->skew();
        EXPECT_GT(skew.x(), 0.0f);
        EXPECT_GT(skew.y(), 0.0f);
        // Iterate the words in the block.
        MutableIterator word_it = *it;
        do {
          const PAGE_RES_IT *w_it = word_it.PageResIt();
          // Iterate the blobs in the word.
          C_BLOB_IT b_it(w_it->word()->word->cblob_list());
          for (b_it.mark_cycle_pt(); !b_it.cycled_list(); b_it.forward()) {
            C_BLOB *blob = b_it.data();
            // Iterate the outlines in the blob.
            C_OUTLINE_IT ol_it(blob->out_list());
            for (ol_it.mark_cycle_pt(); !ol_it.cycled_list(); ol_it.forward()) {
              C_OUTLINE *ol = ol_it.data();
              TBOX box = ol->bounding_box();
              ICOORD middle((box.left() + box.right()) / 2, (box.top() + box.bottom()) / 2);
              EXPECT_EQ(winding_target, pb->winding_number(middle));
            }
          }
        } while (word_it.Next(tesseract::RIL_WORD) &&
                 !word_it.IsAtBeginningOf(tesseract::RIL_BLOCK));
      }
    } while (it->Next(tesseract::RIL_BLOCK));
  }

Member Data Documentation

api_

tesseract::TessBaseAPI tesseract::LayoutTest::api_

protected

Definition at line 187 of file layout_test.cc.

ocr_text_

std::string tesseract::LayoutTest::ocr_text_

protected

Definition at line 186 of file layout_test.cc.

src_pix_

Image tesseract::LayoutTest::src_pix_

protected

Definition at line 185 of file layout_test.cc.

---

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

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