EntitySystemBuilder.java

/*
 * Copyright 2012 Benjamin Glatzel <benjamin.glatzel@me.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.terasology.game.bootstrap;

import org.reflections.Reflections;
import org.terasology.asset.AssetType;
import org.terasology.audio.Sound;
import org.terasology.entitySystem.*;
import org.terasology.entitySystem.metadata.ComponentLibrary;
import org.terasology.entitySystem.metadata.ComponentLibraryImpl;
import org.terasology.entitySystem.metadata.extension.*;
import org.terasology.entitySystem.pojo.PojoEntityManager;
import org.terasology.entitySystem.pojo.PojoEventSystem;
import org.terasology.entitySystem.pojo.PojoPrefabManager;
import org.terasology.game.CoreRegistry;
import org.terasology.math.Vector3i;
import org.terasology.model.blocks.BlockFamily;
import org.terasology.rendering.assets.Material;
import org.terasology.rendering.primitives.Mesh;

import javax.vecmath.Color4f;
import javax.vecmath.Quat4f;
import javax.vecmath.Vector2f;
import javax.vecmath.Vector3f;
import java.util.Set;

/**
 * @author Immortius
 */
public class EntitySystemBuilder {

    public PersistableEntityManager build() {
        ComponentLibrary library = new ComponentLibraryImpl();
        registerTypeHandlers(library);
        CoreRegistry.put(ComponentLibrary.class, library);

        PrefabManager prefabManager = new PojoPrefabManager(library);
        CoreRegistry.put(PrefabManager.class, prefabManager);

        PersistableEntityManager entityManager = new PojoEntityManager(library, prefabManager);
        entityManager.setEventSystem(new PojoEventSystem(entityManager));
        CoreRegistry.put(EntityManager.class, entityManager);
        CoreRegistry.put(EventSystem.class, entityManager.getEventSystem());

        registerComponents(library);
        registerEvents(entityManager.getEventSystem());
        return entityManager;
    }

    private void registerTypeHandlers(ComponentLibrary library) {
        library.registerTypeHandler(Color4f.class, new Color4fTypeHandler());
        library.registerTypeHandler(Quat4f.class, new Quat4fTypeHandler());
        library.registerTypeHandler(Mesh.class, new AssetTypeHandler(AssetType.MESH, Mesh.class));
        library.registerTypeHandler(Sound.class, new AssetTypeHandler(AssetType.SOUND, Sound.class));
        library.registerTypeHandler(Material.class, new AssetTypeHandler(AssetType.MATERIAL, Material.class));
        library.registerTypeHandler(Vector3f.class, new Vector3fTypeHandler());
        library.registerTypeHandler(Vector2f.class, new Vector2fTypeHandler());
        library.registerTypeHandler(Vector3i.class, new Vector3iTypeHandler());
    }

    private void registerComponents(ComponentLibrary library) {
        Reflections reflections = new Reflections("org.terasology");
        Set<Class<? extends Component>> componentTypes = reflections.getSubTypesOf(Component.class);
        for (Class<? extends Component> componentType : componentTypes) {
            library.registerComponentClass(componentType);
        }
    }

    private void registerEvents(EventSystem eventSystem) {
        Reflections reflections = new Reflections("org.terasology");
        Set<Class<? extends Event>> eventTypes = reflections.getSubTypesOf(Event.class);
        for (Class<? extends Event> eventType : eventTypes) {
            eventSystem.registerEvent("engine:" + eventType.getSimpleName(), eventType);
        }
    }
}