use array.resize for setLength

Author: charlesisfeline

flixel/util/FlxArrayUtil.hx

@@ -11,25 +11,11 @@ class FlxArrayUtil
 	 * @param	array		The array.
 	 * @param	newLength	The length you want the array to have.
 	 */
-	@:generic
-	public static function setLength<T>(array:Array<T>, newLength:Int):Array<T>
+	public static inline function setLength<T>(array:Array<T>, newLength:Int):Array<T>
 	{
-		if (newLength < 0)
-			return array;
-
-		var oldLength:Int = array.length;
-		var diff:Int = newLength - oldLength;
-		if (diff >= 0)
-			return array;
-
-		#if flash
-		untyped array.length = newLength;
-		#else
-		diff = -diff;
-		for (i in 0...diff)
-			array.pop();
-		#end
-
+		if (newLength > 0)
+			array.resize(newLength);
+		
 		return array;
 	}
 

tests/unit/src/flixel/util/FlxArrayUtilTest.hx

@@ -102,4 +102,11 @@ class FlxArrayUtilTest
 		Assert.isTrue([0, 1, 2, 3, 4, 5].safeSwap(0, 2).equals([2, 1, 0, 3, 4, 5]));
 		Assert.isTrue([0, 1, 2, 3, 4, 5].safeSwap(1, 6).equals([0, 1, 2, 3, 4, 5]));
 	}
+
+	@Test
+	function testSetLength()
+	{
+		Assert.isTrue([0, 1, 2, 3, 4, 5].setLength(3).equals([0, 1, 2]));
+		Assert.isTrue([0, 1, 2].setLength(5).equals([0, 1, 2, 0, 0]));
+	}
 }