software.coley.cafedude.tree.visitor.writer.FieldWriter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of cafedude-tree Show documentation

Tree module for CafeDude, containing a simplified intermediate model for classes

The newest version!

package software.coley.cafedude.tree.visitor.writer;

import software.coley.cafedude.classfile.AttributeConstants;
import software.coley.cafedude.classfile.Field;
import software.coley.cafedude.classfile.attribute.ConstantValueAttribute;
import software.coley.cafedude.classfile.constant.CpUtf8;
import software.coley.cafedude.tree.Constant;
import software.coley.cafedude.tree.visitor.FieldVisitor;

import javax.annotation.Nonnull;
import java.util.function.Consumer;

/**
 * Field visitor implementation to write back to a {@link Field}.
 *
 * @author Justus Garbe
 */
public class FieldWriter extends DeclarationWriter implements FieldVisitor {
	private final Field field;
	private final Consumer callback;

	FieldWriter(Symbols symbols, int access, CpUtf8 name, CpUtf8 descriptor, Consumer callback) {
		super(symbols);
		this.field = new Field(attributes, access, name, descriptor);
		this.callback = callback;
	}

	@Override
	public void visitConstantValue(@Nonnull Constant value) {
		attributes.add(new ConstantValueAttribute(
				symbols.newUtf8(AttributeConstants.CONSTANT_VALUE),
				symbols.newConstant(value)));
	}

	@Override
	public void visitFieldEnd() {
		super.visitDeclarationEnd();
		callback.accept(field);
	}
}