tesseract/unittest/osd_test.cc

///////////////////////////////////////////////////////////////////////
// File:        osd_test.cc
// Description: OSD Tests for Tesseract.
// Author:      ShreeDevi Kumar
//
// 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.
///////////////////////////////////////////////////////////////////////

// based on https://gist.github.com/amitdo/7c7a522004dd79b398340c9595b377e1

// expects clones of tessdata, tessdata_fast and tessdata_best repos

//#include "log.h"
#include <allheaders.h>
#include <tesseract/baseapi.h>
#include <iostream>
#include <memory> // std::unique_ptr
#include <string>
#include "include_gunit.h"
#include "image.h"

namespace tesseract {

class TestClass : public testing::Test {
protected:
};

#ifndef DISABLED_LEGACY_ENGINE
static void OSDTester(int expected_deg, const char *imgname, const char *tessdatadir) {
  // log.info() << tessdatadir << " for image: " << imgname << std::endl;
  auto api = std::make_unique<tesseract::TessBaseAPI>();
  ASSERT_FALSE(api->Init(tessdatadir, "osd")) << "Could not initialize tesseract.";
  Image image = pixRead(imgname);
  ASSERT_TRUE(image != nullptr) << "Failed to read test image.";
  api->SetImage(image);
  int orient_deg;
  float orient_conf;
  const char *script_name;
  float script_conf;
  bool detected =
      api->DetectOrientationScript(&orient_deg, &orient_conf, &script_name, &script_conf);
  ASSERT_FALSE(!detected) << "Failed to detect OSD.";
  printf(
      "************ Orientation in degrees: %d, Orientation confidence: %.2f\n"
      "             Script: %s, Script confidence: %.2f\n",
      orient_deg, orient_conf, script_name, script_conf);
  EXPECT_EQ(expected_deg, orient_deg);
  api->End();
  image.destroy();
}
#endif

class OSDTest : public TestClass,
                public ::testing::WithParamInterface<std::tuple<int, const char *, const char *>> {
};

TEST_P(OSDTest, MatchOrientationDegrees) {
#ifdef DISABLED_LEGACY_ENGINE
  // Skip test because TessBaseAPI::DetectOrientationScript is missing.
  GTEST_SKIP();
#else
  OSDTester(std::get<0>(GetParam()), std::get<1>(GetParam()), std::get<2>(GetParam()));
#endif
}

INSTANTIATE_TEST_SUITE_P(TessdataEngEuroHebrew, OSDTest,
                         ::testing::Combine(::testing::Values(0),
                                            ::testing::Values(TESTING_DIR "/phototest.tif",
                                                              TESTING_DIR "/eurotext.tif",
                                                              TESTING_DIR "/hebrew.png"),
                                            ::testing::Values(TESSDATA_DIR)));

INSTANTIATE_TEST_SUITE_P(TessdataBestEngEuroHebrew, OSDTest,
                         ::testing::Combine(::testing::Values(0),
                                            ::testing::Values(TESTING_DIR "/phototest.tif",
                                                              TESTING_DIR "/eurotext.tif",
                                                              TESTING_DIR "/hebrew.png"),
                                            ::testing::Values(TESSDATA_DIR "_best")));

INSTANTIATE_TEST_SUITE_P(TessdataFastEngEuroHebrew, OSDTest,
                         ::testing::Combine(::testing::Values(0),
                                            ::testing::Values(TESTING_DIR "/phototest.tif",
                                                              TESTING_DIR "/eurotext.tif",
                                                              TESTING_DIR "/hebrew.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

INSTANTIATE_TEST_SUITE_P(TessdataFastRotated90, OSDTest,
                         ::testing::Combine(::testing::Values(90),
                                            ::testing::Values(TESTING_DIR
                                                              "/phototest-rotated-R.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

INSTANTIATE_TEST_SUITE_P(TessdataFastRotated180, OSDTest,
                         ::testing::Combine(::testing::Values(180),
                                            ::testing::Values(TESTING_DIR
                                                              "/phototest-rotated-180.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

INSTANTIATE_TEST_SUITE_P(TessdataFastRotated270, OSDTest,
                         ::testing::Combine(::testing::Values(270),
                                            ::testing::Values(TESTING_DIR
                                                              "/phototest-rotated-L.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

INSTANTIATE_TEST_SUITE_P(TessdataFastDevaRotated270, OSDTest,
                         ::testing::Combine(::testing::Values(270),
                                            ::testing::Values(TESTING_DIR
                                                              "/devatest-rotated-270.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

INSTANTIATE_TEST_SUITE_P(TessdataFastDeva, OSDTest,
                         ::testing::Combine(::testing::Values(0),
                                            ::testing::Values(TESTING_DIR "/devatest.png"),
                                            ::testing::Values(TESSDATA_DIR "_fast")));

} // namespace tesseract