com.artemis.model.scan.ArtemisScanner Maven / Gradle / Ivy

Go to download

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

There is a newer version: 0.8.1

package com.artemis.model.scan;


import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.FieldVisitor;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;

class ArtemisScanner extends ClassVisitor {
	
	private static final Type COMPONENT_MAPPER = Type.getType("Lcom/artemis/ComponentMapper;");
	
	private final ArtemisTypeData config;
	private final ConfigurationResolver resolver;

	ArtemisScanner(ArtemisTypeData config, ConfigurationResolver configurationResolver) {
		super(Opcodes.ASM4);
		this.config = config;
		this.resolver = configurationResolver;
	}
	
	@Override
	public FieldVisitor visitField(int access, String name, String desc, String signature, Object value) {
		if (!desc.endsWith(";"))
			return super.visitField(access, name, desc, signature, value);
		
		Type type = Type.getType(desc);
		if (resolver.systems.contains(type)) {
			config.systems.add(type);
		} else if (resolver.managers.contains(type)) {
			config.managers.add(type);
		} else if (COMPONENT_MAPPER.equals(type)) {
			String componentDesc = signature.substring(signature.indexOf('<') + 1, signature.indexOf('>'));
			config.optional.add(Type.getType(componentDesc));
		}
		
		return super.visitField(access, name, desc, signature, value);
	}
	
	@Override
	public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
		if ("".equals(name)) {
			return new ConstructorScanner(config, resolver);
		} else {
			return new MethodScanner(config, resolver);
		}
	}
}