Merge pull request #222 from laubed/master

Added addAndReturn for easier Component Manipulation

ashley/src/com/badlogic/ashley/core/Entity.java

@@ -74,6 +74,15 @@ public Entity add (Component component) {
 		return this;
 	}
 
+	/**
+	 * Adds a {@link Component} to this Entity. If a {@link Component} of the same type already exists, it'll be replaced.
+	 * @return The Component for direct component manipulation (e.g. PooledComponent)
+	 */
+	public Component addAndReturn(Component component) {
+		add(component);
+		return component;
+	}
+
 	/**
 	 * Removes the {@link Component} of the specified type. Since there is only ever one component of one type, we don't need an
 	 * instance reference.

ashley/tests/com/badlogic/ashley/core/EntityTests.java

@@ -52,6 +52,18 @@ public void receive (Signal<Entity> signal, Entity object) {
 	private ComponentMapper<ComponentA> am = ComponentMapper.getFor(ComponentA.class);
 	private ComponentMapper<ComponentB> bm = ComponentMapper.getFor(ComponentB.class);
 
+	@Test
+	public void addAndReturnComponent(){
+		Entity entity = new Entity();
+		ComponentA componentA = new ComponentA();
+		ComponentB componentB = new ComponentB();
+
+		assertEquals(componentA, entity.addAndReturn(componentA));
+		assertEquals(componentB, entity.addAndReturn(componentB));
+
+		assertEquals(2, entity.getComponents().size());
+	}
+
 	@Test
 	public void noComponents () {
 		Entity entity = new Entity();