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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/*
* Copyright (C) 2012 The Android Open Source Project
*
* 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.
*/
package com.android.uiautomator.tree;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class UiNode extends BasicTreeNode {
private static final Pattern BOUNDS_PATTERN = Pattern
.compile("\\[-?(\\d+),-?(\\d+)\\]\\[-?(\\d+),-?(\\d+)\\]");
// use LinkedHashMap to preserve the order of the attributes
private final Map<String, String> mAttributes = new LinkedHashMap<String, String>();
private String mDisplayName = "ShouldNotSeeMe";
private Object[] mCachedAttributesArray;
public void addAtrribute(String key, String value) {
mAttributes.put(key, value);
updateDisplayName();
if ("bounds".equals(key)) {
updateBounds(value);
}
}
public Map<String, String> getAttributes() {
return Collections.unmodifiableMap(mAttributes);
}
/**
* Builds the display name based on attributes of the node
*/
private void updateDisplayName() {
String className = mAttributes.get("class");
if (className == null)
return;
String text = mAttributes.get("text");
if (text == null)
return;
String contentDescription = mAttributes.get("content-desc");
if (contentDescription == null)
return;
String index = mAttributes.get("index");
if (index == null)
return;
String bounds = mAttributes.get("bounds");
if (bounds == null) {
return;
}
// shorten the standard class names, otherwise it takes up too much space on UI
className = className.replace("android.widget.", "");
className = className.replace("android.view.", "");
StringBuilder builder = new StringBuilder();
builder.append('(');
builder.append(index);
builder.append(") ");
builder.append(className);
if (!text.isEmpty()) {
builder.append(':');
builder.append(text);
}
if (!contentDescription.isEmpty()) {
builder.append(" {");
builder.append(contentDescription);
builder.append('}');
}
builder.append(' ');
builder.append(bounds);
mDisplayName = builder.toString();
}
private void updateBounds(String bounds) {
Matcher m = BOUNDS_PATTERN.matcher(bounds);
if (m.matches()) {
x = Integer.parseInt(m.group(1));
y = Integer.parseInt(m.group(2));
width = Integer.parseInt(m.group(3)) - x;
height = Integer.parseInt(m.group(4)) - y;
mHasBounds = true;
} else {
throw new RuntimeException("Invalid bounds: " + bounds);
}
}
@Override
public String toString() {
return mDisplayName;
}
public String getAttribute(String key) {
return mAttributes.get(key);
}
@Override
public Object[] getAttributesArray() {
// this approach means we do not handle the situation where an attribute is added
// after this function is first called. This is currently not a concern because the
// tree is supposed to be readonly
if (mCachedAttributesArray == null) {
mCachedAttributesArray = new Object[mAttributes.size()];
int i = 0;
for (String attr : mAttributes.keySet()) {
mCachedAttributesArray[i++] = new AttributePair(attr, mAttributes.get(attr));
}
}
return mCachedAttributesArray;
}
}
|
