com.artemis.weaver.ConstructorInvocationVisitor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of artemis-odb-weaver Show documentation

There is a newer version: 2.3.0

package com.artemis.weaver;

import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;

import com.artemis.meta.ClassMetadata;

public class ConstructorInvocationVisitor extends MethodVisitor implements Opcodes {

	private final ClassMetadata meta;
	
	private boolean hasCalledSuper = false;

	public ConstructorInvocationVisitor(MethodVisitor mv, ClassMetadata meta) {
		super(ASM4, mv);
		this.meta = meta;
	}

	@Override
	public void visitMethodInsn(int opcode, String owner, String name, String desc, boolean itf) {
		if (!hasCalledSuper && INVOKESPECIAL == opcode && "".equals(name)) {
			mv.visitMethodInsn(opcode, owner(meta, owner), name, desc, itf);
			hasCalledSuper = true;
		} else {
			mv.visitMethodInsn(opcode, owner, name, desc, itf);
		}
	}

	private static String owner(ClassMetadata meta, String owner) {
		if (owner.equals(meta.type.getInternalName()))
			return owner;
		
		switch (meta.annotation) {
			case POOLED:
				return "com/artemis/PooledComponent";
			default:
				return "FailedTransformingSuperConstructorInvocation";
		}
	}
}