You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
3.5 KiB
80 lines
3.5 KiB
// Copyright 2000-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
|
|
package org.jetbrains.java.decompiler.modules.decompiler;
|
|
|
|
import org.jetbrains.java.decompiler.code.CodeConstants;
|
|
import org.jetbrains.java.decompiler.main.ClassesProcessor;
|
|
import org.jetbrains.java.decompiler.main.DecompilerContext;
|
|
import org.jetbrains.java.decompiler.main.extern.IFernflowerLogger;
|
|
import org.jetbrains.java.decompiler.main.rels.MethodWrapper;
|
|
import org.jetbrains.java.decompiler.modules.decompiler.exps.*;
|
|
import org.jetbrains.java.decompiler.modules.decompiler.stats.SwitchStatement;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
public final class SwitchHelper {
|
|
public static void simplify(SwitchStatement switchStatement) {
|
|
SwitchExprent switchExprent = (SwitchExprent)switchStatement.getHeadexprent();
|
|
Exprent value = switchExprent.getValue();
|
|
if (isEnumArray(value)) {
|
|
List<List<Exprent>> caseValues = switchStatement.getCaseValues();
|
|
Map<Exprent, Exprent> mapping = new HashMap<>(caseValues.size());
|
|
ArrayExprent array = (ArrayExprent)value;
|
|
FieldExprent arrayField = (FieldExprent)array.getArray();
|
|
ClassesProcessor.ClassNode classNode =
|
|
DecompilerContext.getClassProcessor().getMapRootClasses().get(arrayField.getClassname());
|
|
if (classNode != null) {
|
|
MethodWrapper wrapper = classNode.getWrapper().getMethodWrapper(CodeConstants.CLINIT_NAME, "()V");
|
|
if (wrapper != null && wrapper.root != null) {
|
|
wrapper.getOrBuildGraph().iterateExprents(exprent -> {
|
|
if (exprent instanceof AssignmentExprent) {
|
|
AssignmentExprent assignment = (AssignmentExprent)exprent;
|
|
Exprent left = assignment.getLeft();
|
|
if (left.type == Exprent.EXPRENT_ARRAY && ((ArrayExprent)left).getArray().equals(arrayField)) {
|
|
mapping.put(assignment.getRight(), ((InvocationExprent)((ArrayExprent)left).getIndex()).getInstance());
|
|
}
|
|
}
|
|
return 0;
|
|
});
|
|
}
|
|
}
|
|
|
|
List<List<Exprent>> realCaseValues = new ArrayList<>(caseValues.size());
|
|
for (List<Exprent> caseValue : caseValues) {
|
|
List<Exprent> values = new ArrayList<>(caseValue.size());
|
|
realCaseValues.add(values);
|
|
for (Exprent exprent : caseValue) {
|
|
if (exprent == null) {
|
|
values.add(null);
|
|
}
|
|
else {
|
|
Exprent realConst = mapping.get(exprent);
|
|
if (realConst == null) {
|
|
DecompilerContext.getLogger()
|
|
.writeMessage("Unable to simplify switch on enum: " + exprent + " not found, available: " + mapping,
|
|
IFernflowerLogger.Severity.ERROR);
|
|
return;
|
|
}
|
|
values.add(realConst.copy());
|
|
}
|
|
}
|
|
}
|
|
caseValues.clear();
|
|
caseValues.addAll(realCaseValues);
|
|
switchExprent.replaceExprent(value, ((InvocationExprent)array.getIndex()).getInstance().copy());
|
|
}
|
|
}
|
|
|
|
private static boolean isEnumArray(Exprent exprent) {
|
|
if (exprent instanceof ArrayExprent) {
|
|
Exprent field = ((ArrayExprent)exprent).getArray();
|
|
Exprent index = ((ArrayExprent)exprent).getIndex();
|
|
return field instanceof FieldExprent &&
|
|
(((FieldExprent)field).getName().startsWith("$SwitchMap") ||
|
|
(index instanceof InvocationExprent && ((InvocationExprent)index).getName().equals("ordinal")));
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
|