package org.jf.dexlib2.writer;
import com.google.common.collect.Ordering;
import org.jf.dexlib2.ValueType;
import org.jf.dexlib2.base.BaseAnnotationElement;
import org.jf.dexlib2.iface.reference.FieldReference;
import org.jf.dexlib2.iface.reference.MethodHandleReference;
import org.jf.dexlib2.iface.reference.MethodReference;
import javax.annotation.Nonnull;
import java.io.IOException;
import java.util.Collection;
public abstract class EncodedValueWriter<StringKey, TypeKey, FieldRefKey extends FieldReference,
MethodRefKey extends MethodReference, AnnotationElement extends org.jf.dexlib2.iface.AnnotationElement,
ProtoRefKey, MethodHandleKey extends MethodHandleReference, EncodedValue> {
@Nonnull private final DexDataWriter writer;
@Nonnull private final StringSection<StringKey, ?> stringSection;
@Nonnull private final TypeSection<?, TypeKey, ?> typeSection;
@Nonnull private final FieldSection<?, ?, FieldRefKey, ?> fieldSection;
@Nonnull private final MethodSection<?, ?, ?, MethodRefKey, ?> methodSection;
@Nonnull private final ProtoSection<?, ?, ProtoRefKey, ?> protoSection;
@Nonnull private final MethodHandleSection<MethodHandleKey, ?, ?> methodHandleSection;
@Nonnull private final AnnotationSection<StringKey, TypeKey, ?, AnnotationElement, EncodedValue> annotationSection;
public EncodedValueWriter(
@Nonnull DexDataWriter writer,
@Nonnull StringSection<StringKey, ?> stringSection,
@Nonnull TypeSection<?, TypeKey, ?> typeSection,
@Nonnull FieldSection<?, ?, FieldRefKey, ?> fieldSection,
@Nonnull MethodSection<?, ?, ?, MethodRefKey, ?> methodSection,
ProtoSection<?, ?, ProtoRefKey, ?> protoSection,
MethodHandleSection<MethodHandleKey, ?, ?> methodHandleSection,
@Nonnull AnnotationSection<StringKey, TypeKey, ?, AnnotationElement, EncodedValue> annotationSection) {
this.writer = writer;
this.stringSection = stringSection;
this.typeSection = typeSection;
this.fieldSection = fieldSection;
this.methodSection = methodSection;
this.protoSection = protoSection;
this.methodHandleSection = methodHandleSection;
this.annotationSection = annotationSection;
}
protected abstract void writeEncodedValue(@Nonnull EncodedValue encodedValue) throws IOException;
public void writeAnnotation(TypeKey annotationType,
Collection<? extends AnnotationElement> elements) throws IOException {
writer.writeEncodedValueHeader(ValueType.ANNOTATION, 0);
writer.writeUleb128(typeSection.getItemIndex(annotationType));
writer.writeUleb128(elements.size());
Collection<? extends AnnotationElement> sortedElements = Ordering.from(BaseAnnotationElement.BY_NAME)
.immutableSortedCopy(elements);
for (AnnotationElement element: sortedElements) {
writer.writeUleb128(stringSection.getItemIndex(annotationSection.getElementName(element)));
writeEncodedValue(annotationSection.getElementValue(element));
}
}
public void writeArray(Collection<? extends EncodedValue> elements) throws IOException {
writer.writeEncodedValueHeader(ValueType.ARRAY, 0);
writer.writeUleb128(elements.size());
for (EncodedValue element: elements) {
writeEncodedValue(element);
}
}
public void writeBoolean(boolean value) throws IOException {
writer.writeEncodedValueHeader(ValueType.BOOLEAN, value ? 1 : 0);
}
public void writeByte(byte value) throws IOException {
writer.writeEncodedInt(ValueType.BYTE, value);
}
public void writeChar(char value) throws IOException {
writer.writeEncodedUint(ValueType.CHAR, value);
}
public void writeDouble(double value) throws IOException {
writer.writeEncodedDouble(ValueType.DOUBLE, value);
}
public void writeEnum(@Nonnull FieldRefKey value) throws IOException {
writer.writeEncodedUint(ValueType.ENUM, fieldSection.getItemIndex(value));
}
public void writeField(@Nonnull FieldRefKey value) throws IOException {
writer.writeEncodedUint(ValueType.FIELD, fieldSection.getItemIndex(value));
}
public void writeFloat(float value) throws IOException {
writer.writeEncodedFloat(ValueType.FLOAT, value);
}
public void writeInt(int value) throws IOException {
writer.writeEncodedInt(ValueType.INT, value);
}
public void writeLong(long value) throws IOException {
writer.writeEncodedLong(ValueType.LONG, value);
}
public void writeMethod(@Nonnull MethodRefKey value) throws IOException {
writer.writeEncodedUint(ValueType.METHOD, methodSection.getItemIndex(value));
}
public void writeNull() throws IOException {
writer.write(ValueType.NULL);
}
public void writeShort(int value) throws IOException {
writer.writeEncodedInt(ValueType.SHORT, value);
}
public void writeString(@Nonnull StringKey value) throws IOException {
writer.writeEncodedUint(ValueType.STRING, stringSection.getItemIndex(value));
}
public void writeType(@Nonnull TypeKey value) throws IOException {
writer.writeEncodedUint(ValueType.TYPE, typeSection.getItemIndex(value));
}
public void writeMethodType(@Nonnull ProtoRefKey value) throws IOException {
writer.writeEncodedUint(ValueType.METHOD_TYPE, protoSection.getItemIndex(value));
}
public void writeMethodHandle(@Nonnull MethodHandleKey value) throws IOException {
writer.writeEncodedUint(ValueType.METHOD_HANDLE, methodHandleSection.getItemIndex(value));
}
}