package org.graalvm.compiler.lir;
import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.EnumSet;
import org.graalvm.compiler.core.common.FieldIntrospection;
import org.graalvm.compiler.core.common.Fields;
import org.graalvm.compiler.core.common.FieldsScanner;
import org.graalvm.compiler.debug.GraalError;
import org.graalvm.compiler.lir.CompositeValue.Component;
import org.graalvm.compiler.lir.LIRInstruction.OperandFlag;
import org.graalvm.compiler.lir.LIRIntrospection.LIRFieldsScanner;
import org.graalvm.compiler.lir.LIRIntrospection.OperandModeAnnotation;
import org.graalvm.compiler.lir.LIRIntrospection.Values;
public final class CompositeValueClass<T> extends FieldIntrospection<T> {
private static final ClassValue<CompositeValueClass<?>> compositeClass = new ClassValue<CompositeValueClass<?>>() {
@Override
protected CompositeValueClass<?> computeValue(Class<?> type) {
CompositeValueClass<?> compositeValueClass = new CompositeValueClass<>(type);
assert compositeValueClass.values.getDirectCount() == compositeValueClass.values.getCount() : "only direct fields are allowed in composites";
return compositeValueClass;
}
};
@SuppressWarnings("unchecked")
public static <T> CompositeValueClass<T> get(Class<T> type) {
return (CompositeValueClass<T>) compositeClass.get(type);
}
private final Values values;
private CompositeValueClass(Class<T> clazz) {
super(clazz);
CompositeValueFieldsScanner vfs = new CompositeValueFieldsScanner(new FieldsScanner.DefaultCalcOffset());
vfs.scan(clazz, CompositeValue.class, false);
values = new Values(vfs.valueAnnotations.get(CompositeValue.Component.class));
data = new Fields(vfs.data);
}
private static class CompositeValueFieldsScanner extends LIRFieldsScanner {
CompositeValueFieldsScanner(FieldsScanner.CalcOffset calc) {
super(calc);
valueAnnotations.put(CompositeValue.Component.class, new OperandModeAnnotation());
}
@Override
protected EnumSet<OperandFlag> getFlags(Field field) {
EnumSet<OperandFlag> result = EnumSet.noneOf(OperandFlag.class);
if (field.isAnnotationPresent(CompositeValue.Component.class)) {
result.addAll(Arrays.asList(field.getAnnotation(CompositeValue.Component.class).value()));
} else {
GraalError.shouldNotReachHere();
}
return result;
}
}
@Override
public Fields[] getAllFields() {
return new Fields[]{data, values};
}
@Override
public String toString() {
StringBuilder str = new StringBuilder();
str.append(getClass().getSimpleName()).append(" ").append(getClazz().getSimpleName()).append(" components[");
values.appendFields(str);
str.append("] data[");
data.appendFields(str);
str.append("]");
return str.toString();
}