/
CaptureFields.java
67 lines (59 loc) · 2.24 KB
/
CaptureFields.java
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
/*
* 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.google.devtools.cyclefinder;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.ListMultimap;
import com.google.devtools.j2objc.ast.AnonymousClassDeclaration;
import com.google.devtools.j2objc.ast.CompilationUnit;
import com.google.devtools.j2objc.ast.TreeVisitor;
import com.google.devtools.j2objc.ast.TypeDeclaration;
import com.google.devtools.j2objc.util.CaptureInfo;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;
/**
* Collects implicit capture fields from populated CaptureInfo instances.
*/
public class CaptureFields {
private final Set<TypeElement> haveOuterReference = new HashSet<>();
private final ListMultimap<TypeElement, VariableElement> captureFields =
ArrayListMultimap.create();
public void collect(CompilationUnit unit) {
unit.accept(new TreeVisitor() {
@Override
public void endVisit(TypeDeclaration node) {
collect(unit, node.getTypeElement());
}
@Override
public void endVisit(AnonymousClassDeclaration node) {
collect(unit, node.getTypeElement());
}
});
}
private void collect(CompilationUnit unit, TypeElement type) {
CaptureInfo captureInfo = unit.getEnv().captureInfo();
if (captureInfo.getOuterField(type) != null) {
haveOuterReference.add(type);
}
captureFields.replaceValues(type, captureInfo.getCaptureFields(type));
}
public boolean hasOuterReference(TypeElement type) {
return haveOuterReference.contains(type);
}
public List<VariableElement> getCaptureFields(TypeElement type) {
return captureFields.get(type);
}
}