EntityFactory.java

package com.uwsoft.editor.renderer.factory;

import box2dLight.RayHandler;

import com.badlogic.ashley.core.Component;
import com.badlogic.ashley.core.ComponentMapper;
import com.badlogic.ashley.core.Engine;
import com.badlogic.ashley.core.Entity;
import com.badlogic.gdx.Gdx;
import com.badlogic.gdx.physics.box2d.World;
import com.badlogic.gdx.utils.viewport.Viewport;
import com.uwsoft.editor.renderer.commons.IExternalItemType;
import com.uwsoft.editor.renderer.components.CompositeTransformComponent;
import com.uwsoft.editor.renderer.components.MainItemComponent;
import com.uwsoft.editor.renderer.components.TransformComponent;
import com.uwsoft.editor.renderer.components.ViewPortComponent;
import com.uwsoft.editor.renderer.data.*;
import com.uwsoft.editor.renderer.factory.component.*;
import com.uwsoft.editor.renderer.resources.IResourceRetriever;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;

public class EntityFactory {
	
	public static final int IMAGE_TYPE 		= 1;
	public static final int LABEL_TYPE 		= 2;
	public static final int SPRITE_TYPE 	= 3;
	public static final int SPINE_TYPE 		= 4;
	public static final int SPRITER_TYPE 	= 5;
	public static final int COMPOSITE_TYPE 	= 6;
	public static final int PARTICLE_TYPE 	= 7;
	public static final int LIGHT_TYPE 		= 8;
	public static final int NINE_PATCH 		= 9;
	public static final int COLOR_PRIMITIVE = 10;
	
	public RayHandler rayHandler;
	public World world;
	public IResourceRetriever rm = null;

	public EntityFactory( RayHandler rayHandler, World world, IResourceRetriever rm ) {

		this.rayHandler = rayHandler;
		this.world = world;
		this.rm = rm;

		compositeComponentFactory = new CompositeComponentFactory(rayHandler, world, rm);
		lightComponentFactory = new LightComponentFactory(rayHandler, world, rm);
		particleEffectComponentFactory = new ParticleEffectComponentFactory(rayHandler, world, rm);
		simpleImageComponentFactory = new SimpleImageComponentFactory(rayHandler, world, rm);
		spriteComponentFactory = new SpriteComponentFactory(rayHandler, world, rm);
		spriterComponentFactory = new SpriterComponentFactory(rayHandler, world, rm);
		labelComponentFactory = new LabelComponentFactory(rayHandler, world, rm);
		ninePatchComponentFactory = new NinePatchComponentFactory(rayHandler, world, rm);
		colorPrimitiveFactory = new ColorPrimitiveComponentFactory(rayHandler, world, rm);

	}

	protected ComponentFactory compositeComponentFactory, lightComponentFactory, particleEffectComponentFactory,
			simpleImageComponentFactory, spriteComponentFactory, spriterComponentFactory, labelComponentFactory, ninePatchComponentFactory, colorPrimitiveFactory;

	private HashMap<Integer, ComponentFactory> externalFactories = new HashMap<Integer, ComponentFactory>();

	private HashMap<Integer, Entity> entities = new HashMap<Integer, Entity>();

	public ComponentFactory getCompositeComponentFactory() {
		return compositeComponentFactory;
	}

    public SpriteComponentFactory getSpriteComponentFactory() {
        return (SpriteComponentFactory) spriteComponentFactory;
    }

	public void addExternalFactory(IExternalItemType itemType) {
		externalFactories.put(itemType.getTypeId(), itemType.getComponentFactory());
	}

	public Entity createEntity(Entity root, SimpleImageVO vo){

		Entity entity = new Entity();

		simpleImageComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, Image9patchVO vo){

		Entity entity = new Entity();

		ninePatchComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, LabelVO vo) {
		
		Entity entity = new Entity();
		
		labelComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);

		return entity;
	}
	
	public Entity createEntity(Entity root, ParticleEffectVO vo){

		Entity entity = new Entity();
		
		particleEffectComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, LightVO vo){

		Entity entity = new Entity();

		lightComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, SpineVO vo){

		Entity entity = new Entity();

		ComponentFactory factory = externalFactories.get(SPINE_TYPE);
		if(factory != null) {
			factory.createComponents(root, entity, vo);
			postProcessEntity(entity);
		}
		
		return entity;
	}
	
	public Entity createEntity(Entity root, SpriteAnimationVO vo){

		Entity entity = new Entity();

		spriteComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, SpriterVO vo){

		Entity entity = new Entity();

		spriterComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}
	
	public Entity createEntity(Entity root, CompositeItemVO vo){

		Entity entity = new Entity();

		compositeComponentFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);
		
		return entity;
	}

	public Entity createEntity(Entity root, ColorPrimitiveVO vo){

		Entity entity = new Entity();

		colorPrimitiveFactory.createComponents(root, entity, vo);

		postProcessEntity(entity);

		return entity;
	}

	public Entity createRootEntity(CompositeVO compositeVo, Viewport viewport){

		CompositeItemVO vo = new CompositeItemVO();
		vo.composite = compositeVo;

		Entity entity = new Entity();

		compositeComponentFactory.createComponents(null, entity, vo);
//		CompositeTransformComponent compositeTransform = new CompositeTransformComponent();
		TransformComponent transform = new TransformComponent();

		ViewPortComponent viewPortComponent = new ViewPortComponent();
		viewPortComponent.viewPort = viewport;

		//TODO: not sure if this line is okay
        //without this line the editor won't load
		viewPortComponent.viewPort.update(Gdx.graphics.getWidth(), Gdx.graphics.getHeight(), true);

		entity.add(transform);
		entity.add(viewPortComponent);

		postProcessEntity(entity);

		return entity;
	}

	public Integer postProcessEntity(Entity entity) {
		ComponentMapper<MainItemComponent> mainItemComponentComponentMapper = ComponentMapper.getFor(MainItemComponent.class);
		MainItemComponent mainItemComponent = mainItemComponentComponentMapper.get(entity);
		if(mainItemComponent.uniqueId == -1) mainItemComponent.uniqueId = getFreeId();
		entities.put(mainItemComponent.uniqueId, entity);

		return mainItemComponent.uniqueId;
	}

	private int getFreeId() {
		if(entities == null || entities.size() == 0) return 1;
		ArrayList<Integer> ids = new ArrayList<Integer>(entities.keySet());
		Collections.sort(ids);
		for(int i = 1; i < ids.size(); i++) {
			if(ids.get(i)-ids.get(i-1) > 1) {
				return ids.get(i-1)+1;
			}
		}
		return ids.get(ids.size()-1)+1;
	}

	public Integer updateMap(Entity entity) {
		ComponentMapper<MainItemComponent> mainItemComponentComponentMapper = ComponentMapper.getFor(MainItemComponent.class);
		MainItemComponent mainItemComponent = mainItemComponentComponentMapper.get(entity);
		entities.put(mainItemComponent.uniqueId, entity);

		return mainItemComponent.uniqueId;
	}

	public void initAllChildren(Engine engine, Entity entity, CompositeVO vo) {
		for (int i = 0; i < vo.sImages.size(); i++) {
			Entity child = createEntity(entity, vo.sImages.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sImage9patchs.size(); i++) {
			Entity child = createEntity(entity, vo.sImage9patchs.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sLabels.size(); i++) {
			Entity child = createEntity(entity, vo.sLabels.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sParticleEffects.size(); i++) {
			Entity child = createEntity(entity, vo.sParticleEffects.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sLights.size(); i++) {
			Entity child = createEntity(entity, vo.sLights.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sSpineAnimations.size(); i++) {
			Entity child = createEntity(entity, vo.sSpineAnimations.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sSpriteAnimations.size(); i++) {
			Entity child = createEntity(entity, vo.sSpriteAnimations.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sSpriterAnimations.size(); i++) {
			Entity child = createEntity(entity, vo.sSpriterAnimations.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sColorPrimitives.size(); i++) {
			Entity child = createEntity(entity, vo.sColorPrimitives.get(i));
			engine.addEntity(child);
		}

		for (int i = 0; i < vo.sComposites.size(); i++) {
			Entity child = createEntity(entity, vo.sComposites.get(i));
			engine.addEntity(child);
			initAllChildren(engine, child, vo.sComposites.get(i).composite);
		}
	}

	public Entity getEntityByUniqueId(Integer id) {
		return entities.get(id);
	}


	public void clean() {
		entities.clear();
	}
}