com.artemis.injection.ArtemisFieldResolver Maven / Gradle / Ivy

Go to download
package com.artemis.injection;

import com.artemis.BaseSystem;
import com.artemis.Component;
import com.artemis.ComponentMapper;
import com.artemis.Manager;
import com.artemis.World;
import com.artemis.utils.reflect.Field;

import java.util.IdentityHashMap;
import java.util.Map;

/**
 * Can resolve {@link com.artemis.ComponentMapper}, {@link com.artemis.BaseSystem} and
 * {@link com.artemis.Manager} types registerd in the {@link World}
 *
 * @author Snorre E. Brekke
 */
public class ArtemisFieldResolver implements FieldResolver, UseInjectionCache {

    private World world;
    private InjectionCache cache;

    private Map, Class> systems;
    private Map, Class> managers;

    public ArtemisFieldResolver() {
        systems = new IdentityHashMap, Class>();
        managers = new IdentityHashMap, Class>();
    }

    @Override
    public void initialize(World world) {
        this.world = world;

        for (BaseSystem es : world.getSystems()) {
            Class origin = es.getClass();
            Class clazz = origin;
            do {
                systems.put(clazz, origin);
            } while ((clazz = clazz.getSuperclass()) != Object.class);
        }

        for (Manager manager : world.getManagers()) {
            Class origin = manager.getClass();
            Class clazz = origin;
            do {
                managers.put(clazz, origin);
            } while ((clazz = clazz.getSuperclass()) != Object.class);
        }
    }

    @Override
    @SuppressWarnings("unchecked")
    public Object resolve(Class fieldType, Field field) {
        ClassType injectionType = cache.getFieldClassType(fieldType);
        switch (injectionType) {
            case MAPPER:
                return getComponentMapper(field);
            case SYSTEM:
                return world.getSystem((Class) systems.get(fieldType));
            case MANAGER:
                return world.getManager((Class) managers.get(fieldType));
            case FACTORY:
                return world.createFactory(fieldType);
            default:
                return null;

        }
    }

    @SuppressWarnings("unchecked")
    private ComponentMapper getComponentMapper(Field field) {
        Class mapperType = cache.getGenericType(field);
        return world.getMapper((Class) mapperType);

    }

    @Override
    public void setCache(InjectionCache cache) {
        this.cache = cache;
    }
}