blob: e7094251c0a824230c9d81ab292913d461ce789d [file] [log] [blame]
// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "services/shape_detection/text_detection_impl_mac.h"
#include "base/bind.h"
#include "base/command_line.h"
#include "base/mac/mac_util.h"
#include "base/mac/scoped_cftyperef.h"
#include "base/mac/scoped_nsobject.h"
#include "base/mac/sdk_forward_declarations.h"
#include "base/run_loop.h"
#include "base/test/scoped_task_environment.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/skia/include/utils/mac/SkCGUtils.h"
#include "ui/gl/gl_switches.h"
namespace shape_detection {
ACTION_P(RunClosure, closure) {
closure.Run();
}
class TextDetectionImplMacTest : public ::testing::Test {
public:
~TextDetectionImplMacTest() override = default;
void DetectCallback(std::vector<mojom::TextDetectionResultPtr> results) {
// CIDetectorTypeText doesn't return the decoded text, juts bounding boxes.
Detection(results.size());
}
MOCK_METHOD1(Detection, void(size_t));
API_AVAILABLE(macosx(10.11)) std::unique_ptr<TextDetectionImplMac> impl_;
base::test::TaskEnvironment task_environment_;
};
TEST_F(TextDetectionImplMacTest, CreateAndDestroy) {}
// This test generates an image with a single text line and scans it back.
TEST_F(TextDetectionImplMacTest, ScanOnce) {
// Text detection needs at least MAC OS X 10.11, and GPU infrastructure.
if (!base::CommandLine::ForCurrentProcess()->HasSwitch(
switches::kUseGpuInTests) ||
!base::mac::IsAtLeastOS10_11()) {
return;
}
if (@available(macOS 10.11, *)) {
impl_.reset(new TextDetectionImplMac);
base::ScopedCFTypeRef<CGColorSpaceRef> rgb_colorspace(
CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB));
const int width = 200;
const int height = 50;
base::ScopedCFTypeRef<CGContextRef> context(CGBitmapContextCreate(
nullptr, width, height, 8 /* bitsPerComponent */,
width * 4 /* rowBytes */, rgb_colorspace,
kCGImageAlphaPremultipliedFirst | kCGBitmapByteOrder32Host));
// Draw a white background.
CGContextSetRGBFillColor(context, 1.0, 1.0, 1.0, 1.0);
CGContextFillRect(context, CGRectMake(0.0, 0.0, width, height));
// Create a line of Helvetica 16 text, and draw it in the |context|.
base::scoped_nsobject<NSFont> helvetica(
[NSFont fontWithName:@"Helvetica" size:16]);
NSDictionary* attributes = [NSDictionary
dictionaryWithObjectsAndKeys:helvetica, kCTFontAttributeName, nil];
base::scoped_nsobject<NSAttributedString> info([[NSAttributedString alloc]
initWithString:@"https://www.chromium.org"
attributes:attributes]);
base::ScopedCFTypeRef<CTLineRef> line(
CTLineCreateWithAttributedString((CFAttributedStringRef)info.get()));
CGContextSetTextPosition(context, 10.0, height / 2.0);
CTLineDraw(line, context);
// Extract a CGImage and its raw pixels from |context|.
base::ScopedCFTypeRef<CGImageRef> cg_image(
CGBitmapContextCreateImage(context));
EXPECT_EQ(static_cast<size_t>(width), CGImageGetWidth(cg_image));
EXPECT_EQ(static_cast<size_t>(height), CGImageGetHeight(cg_image));
SkBitmap bitmap;
ASSERT_TRUE(SkCreateBitmapFromCGImage(&bitmap, cg_image));
base::RunLoop run_loop;
base::Closure quit_closure = run_loop.QuitClosure();
// Send the image to Detect() and expect the response in callback.
EXPECT_CALL(*this, Detection(1)).WillOnce(RunClosure(quit_closure));
impl_->Detect(bitmap, base::Bind(&TextDetectionImplMacTest::DetectCallback,
base::Unretained(this)));
run_loop.Run();
}
}
} // shape_detection namespace