From 0862d87264173718f7fc8d38dd527f7d97f83af1 Mon Sep 17 00:00:00 2001
From: Marcus Gosselin <marcus13345@live.com>
Date: Thu, 18 Feb 2016 04:36:41 -0500
Subject: [PATCH] implemented getComponent

---
 src/spacewars/PlayerController.java | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/spacewars/PlayerController.java b/src/spacewars/PlayerController.java
index 73c23bf..67da86e 100644
--- a/src/spacewars/PlayerController.java
+++ b/src/spacewars/PlayerController.java
@@ -1,19 +1,23 @@
 package spacewars;
 
+import java.awt.Color;
 import java.awt.event.KeyEvent;
 
 import diveengine2d.DiveScript;
 import diveengine2d.Input;
+import diveengine2d.RectRenderer;
 import diveengine2d.RigidBody;
 
 public class PlayerController extends DiveScript {
 	public String message = null;
 	public float speed;
-	
+
 	private RigidBody rigidBody;
+	private RectRenderer renderer;
 	
 	public void create() {
-		rigidBody = (RigidBody) entity.components.get(2);
+		rigidBody = entity.getComponent(RigidBody.class);
+		renderer = entity.getComponent(RectRenderer.class);
 	}
 	
 	public void update() {
@@ -25,5 +29,6 @@ public class PlayerController extends DiveScript {
 			rigidBody.dx += -1 * speed;
 		if(Input.getKeyDown(KeyEvent.VK_D))
 			rigidBody.dx += 1 * speed;
+		renderer.color = new Color((int)(Math.random()*255), (int)(Math.random()*255), (int)(Math.random()*255));
 	}
 }