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
96
97
98
99
100
content / renderer / accessibility / annotations / ax_annotators_manager.cc [blame]
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/renderer/accessibility/annotations/ax_annotators_manager.h"
#include <utility>
#include "build/build_config.h"
#include "content/renderer/accessibility/annotations/ax_image_annotator.h"
#include "content/renderer/accessibility/render_accessibility_impl.h"
#include "content/renderer/render_frame_impl.h"
#include "services/screen_ai/buildflags/buildflags.h"
#include "third_party/blink/public/web/web_document.h"
#if BUILDFLAG(ENABLE_SCREEN_AI_SERVICE)
#include "content/renderer/accessibility/annotations/ax_main_node_annotator.h"
#include "ui/accessibility/accessibility_features.h"
#endif
namespace content {
AXAnnotatorsManager::AXAnnotatorsManager(
RenderAccessibilityImpl* const render_accessibility)
: render_accessibility_(render_accessibility) {
DCHECK(render_accessibility_);
ax_annotators_.emplace_back(
std::make_unique<AXImageAnnotator>(render_accessibility_));
#if BUILDFLAG(ENABLE_SCREEN_AI_SERVICE)
if (features::IsMainNodeAnnotationsEnabled()) {
ax_annotators_.emplace_back(
std::make_unique<AXMainNodeAnnotator>(render_accessibility_));
}
#endif
}
AXAnnotatorsManager::~AXAnnotatorsManager() {}
void AXAnnotatorsManager::Annotate(const blink::WebDocument& document,
ui::AXTreeUpdate* update,
bool load_complete) {
for (const auto& annotator : ax_annotators_) {
annotator->Annotate(document, update, load_complete);
}
}
void AXAnnotatorsManager::AccessibilityModeChanged(ui::AXMode old_mode,
ui::AXMode new_mode) {
for (const auto& annotator : ax_annotators_) {
uint32_t flag = annotator->GetAXModeToEnableAnnotations();
if (!old_mode.has_mode(flag) && new_mode.has_mode(flag)) {
annotator->EnableAnnotations();
} else if (old_mode.has_mode(flag) && !new_mode.has_mode(flag)) {
annotator->CancelAnnotations();
}
}
}
void AXAnnotatorsManager::CancelAnnotations() {
for (const auto& annotator : ax_annotators_) {
annotator->CancelAnnotations();
}
}
void AXAnnotatorsManager::PerformAction(ax::mojom::Action action) {
bool applied_annotations = false;
for (const auto& annotator : ax_annotators_) {
if (!annotator->HasAXActionToEnableAnnotations() ||
action != annotator->GetAXActionToEnableAnnotations()) {
continue;
}
applied_annotations = true;
annotator->EnableAnnotations();
}
if (!applied_annotations) {
return;
}
// Rebuild the document tree so that annotations are applied.
DCHECK(render_accessibility_->GetAXContext());
render_accessibility_->GetAXContext()->MarkDocumentDirty();
}
void AXAnnotatorsManager::AddDebuggingAttributes(
const std::vector<ui::AXTreeUpdate>& updates) {
for (const auto& annotator : ax_annotators_) {
annotator->AddDebuggingAttributes(updates);
}
}
void AXAnnotatorsManager::AddAnnotatorForTesting(
std::unique_ptr<AXAnnotator> annotator) {
ax_annotators_.push_back(std::move(annotator));
}
void AXAnnotatorsManager::ClearAnnotatorsForTesting() {
ax_annotators_.clear();
}
} // namespace content