1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
// 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 "modules/shapedetection/TextDetector.h"
#include "core/dom/DOMException.h"
#include "core/dom/DOMRect.h"
#include "core/frame/LocalFrame.h"
#include "core/html/canvas/CanvasImageSource.h"
#include "modules/shapedetection/DetectedText.h"
#include "public/platform/InterfaceProvider.h"
namespace blink {
TextDetector* TextDetector::create(Document& document) {
return new TextDetector(*document.frame());
}
TextDetector::TextDetector(LocalFrame& frame) : ShapeDetector(frame) {
frame.interfaceProvider()->getInterface(mojo::MakeRequest(&m_textService));
m_textService.set_connection_error_handler(convertToBaseCallback(WTF::bind(
&TextDetector::onTextServiceConnectionError, wrapWeakPersistent(this))));
}
ScriptPromise TextDetector::doDetect(
ScriptPromiseResolver* resolver,
mojo::ScopedSharedBufferHandle sharedBufferHandle,
int imageWidth,
int imageHeight) {
ScriptPromise promise = resolver->promise();
if (!m_textService) {
resolver->reject(DOMException::create(
NotSupportedError, "Text detection service unavailable."));
return promise;
}
m_textServiceRequests.add(resolver);
m_textService->Detect(std::move(sharedBufferHandle), imageWidth, imageHeight,
convertToBaseCallback(WTF::bind(
&TextDetector::onDetectText, wrapPersistent(this),
wrapPersistent(resolver))));
return promise;
}
void TextDetector::onDetectText(
ScriptPromiseResolver* resolver,
Vector<shape_detection::mojom::blink::TextDetectionResultPtr>
textDetectionResults) {
DCHECK(m_textServiceRequests.contains(resolver));
m_textServiceRequests.remove(resolver);
HeapVector<Member<DetectedText>> detectedText;
for (const auto& text : textDetectionResults) {
detectedText.push_back(DetectedText::create(
text->raw_value,
DOMRect::create(text->bounding_box->x, text->bounding_box->y,
text->bounding_box->width,
text->bounding_box->height)));
}
resolver->resolve(detectedText);
}
void TextDetector::onTextServiceConnectionError() {
for (const auto& request : m_textServiceRequests) {
request->reject(DOMException::create(NotSupportedError,
"Text Detection not implemented."));
}
m_textServiceRequests.clear();
m_textService.reset();
}
DEFINE_TRACE(TextDetector) {
ShapeDetector::trace(visitor);
visitor->trace(m_textServiceRequests);
}
} // namespace blink
|
