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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/page_image_annotation/core/page_annotator.h"
#include "base/functional/bind.h"
#include "base/observer_list.h"
namespace page_image_annotation {
namespace ia_mojom = image_annotation::mojom;
PageAnnotator::Observer::~Observer() = default;
PageAnnotator::PageAnnotator(mojo::PendingRemote<ia_mojom::Annotator> annotator)
: annotator_(std::move(annotator)) {}
PageAnnotator::~PageAnnotator() = default;
void PageAnnotator::ImageAddedOrPossiblyModified(
const ImageMetadata& metadata,
base::RepeatingCallback<SkBitmap()> pixels_callback) {
const auto lookup = images_.find(metadata.node_id);
if (lookup == images_.end()) {
// This is an image addition.
AddNewImage(metadata, std::move(pixels_callback));
for (Observer& observer : observers_) {
observer.OnImageAdded(metadata);
}
} else if (lookup->second.first.source_id != metadata.source_id) {
// We already have older data for this node ID; this is an update.
images_.erase(lookup);
AddNewImage(metadata, std::move(pixels_callback));
for (Observer& observer : observers_) {
observer.OnImageModified(metadata);
}
}
}
void PageAnnotator::ImageRemoved(const uint64_t node_id) {
images_.erase(node_id);
for (Observer& observer : observers_) {
observer.OnImageRemoved(node_id);
}
}
void PageAnnotator::AnnotateImage(Observer* const observer,
const uint64_t node_id) {
DCHECK(observers_.HasObserver(observer));
const auto lookup = images_.find(node_id);
if (lookup == images_.end())
return;
// TODO(crbug.com/41432474): get a user's preferred language and pass it here.
annotator_->AnnotateImage(
lookup->second.first.source_id,
std::string() /* description_language_tag */,
lookup->second.second.GetPendingRemote(),
base::BindOnce(&PageAnnotator::NotifyObserver, base::Unretained(this),
observer, node_id));
}
void PageAnnotator::AddObserver(Observer* const observer) {
observers_.AddObserver(observer);
// The new observer has not received any previous messages; inform them now of
// all existing images.
for (const auto& image : images_) {
observer->OnImageAdded(image.second.first);
}
}
void PageAnnotator::AddNewImage(
const ImageMetadata& metadata,
base::RepeatingCallback<SkBitmap()> pixels_callback) {
images_.emplace(std::piecewise_construct,
std::forward_as_tuple(metadata.node_id),
std::forward_as_tuple(metadata, std::move(pixels_callback)));
}
void PageAnnotator::NotifyObserver(Observer* const observer,
const uint64_t node_id,
ia_mojom::AnnotateImageResultPtr result) {
observer->OnImageAnnotated(node_id, std::move(result));
}
} // namespace page_image_annotation
|
