azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix & optimize DataFlowInfo.or()

Browse Source 
Get complete nullability & type info before creating a new data flow info.

Don't create unnecessary data flow infos
This commit is contained in:
Alexander Udalov
2012-11-15 22:34:15 +04:00
parent 6b65c44d45
commit d1de446ae0
1 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/frontend/src/org/jetbrains/jet/lang/resolve/calls/autocasts/DelegatingDataFlowInfo.java
+18 -19
View File
@@ -171,13 +171,6 @@ import static org.jetbrains.jet.lang.resolve.calls.autocasts.Nullability.NOT_NUL
return new DelegatingDataFlowInfo(this, newNullabilityInfo, newTypeInfo);
}
@NotNull
private ListMultimap<DataFlowValue, JetType> copyTypeInfo() {
ListMultimap<DataFlowValue, JetType> newTypeInfo = newTypeInfo();
newTypeInfo.putAll(typeInfo);
return newTypeInfo;
}
@NotNull
@Override
public DataFlowInfo and(@NotNull DataFlowInfo otherInfo) {
@@ -211,28 +204,30 @@ import static org.jetbrains.jet.lang.resolve.calls.autocasts.Nullability.NOT_NUL
@Override
public DataFlowInfo or(@NotNull DataFlowInfo otherInfo) {
if (otherInfo == EMPTY) return EMPTY;
if (this == EMPTY) return EMPTY;
if (this == otherInfo) return this;
assert otherInfo instanceof DelegatingDataFlowInfo : "Unknown DataFlowInfo type: " + otherInfo;
DelegatingDataFlowInfo other = (DelegatingDataFlowInfo) otherInfo;
Map<DataFlowValue, Nullability> builder = Maps.newHashMap(nullabilityInfo);
builder.keySet().retainAll(other.nullabilityInfo.keySet());
for (Map.Entry<DataFlowValue, Nullability> entry : builder.entrySet()) {
Map<DataFlowValue, Nullability> nullabilityMapBuilder = Maps.newHashMap();
for (Map.Entry<DataFlowValue, Nullability> entry : other.getCompleteNullabilityInfo().entrySet()) {
DataFlowValue key = entry.getKey();
Nullability thisFlags = entry.getValue();
Nullability otherFlags = other.nullabilityInfo.get(key);
assert (otherFlags != null);
builder.put(key, thisFlags.or(otherFlags));
Nullability otherFlags = entry.getValue();
Nullability thisFlags = getNullability(key);
nullabilityMapBuilder.put(key, thisFlags.or(otherFlags));
}
ListMultimap<DataFlowValue, JetType> myTypeInfo = getCompleteTypeInfo();
ListMultimap<DataFlowValue, JetType> otherTypeInfo = other.getCompleteTypeInfo();
ListMultimap<DataFlowValue, JetType> newTypeInfo = newTypeInfo();
Set<DataFlowValue> keys = Sets.newHashSet(typeInfo.keySet());
keys.retainAll(other.typeInfo.keySet());
Set<DataFlowValue> keys = Sets.newHashSet(myTypeInfo.keySet());
keys.retainAll(otherTypeInfo.keySet());
for (DataFlowValue key : keys) {
Collection<JetType> thisTypes = typeInfo.get(key);
Collection<JetType> otherTypes = other.typeInfo.get(key);
Collection<JetType> thisTypes = myTypeInfo.get(key);
Collection<JetType> otherTypes = otherTypeInfo.get(key);
Collection<JetType> newTypes = Sets.newHashSet(thisTypes);
newTypes.retainAll(otherTypes);
@@ -240,7 +235,11 @@ import static org.jetbrains.jet.lang.resolve.calls.autocasts.Nullability.NOT_NUL
newTypeInfo.putAll(key, newTypes);
}
return new DelegatingDataFlowInfo(null, ImmutableMap.copyOf(builder), newTypeInfo);
if (nullabilityMapBuilder.isEmpty() && newTypeInfo.isEmpty()) {
return EMPTY;
}
return new DelegatingDataFlowInfo(null, ImmutableMap.copyOf(nullabilityMapBuilder), newTypeInfo);
}
@NotNull