diff --git a/src/diveengine2d/Entity.java b/src/diveengine2d/Entity.java index d9fd108..15b8ddf 100644 --- a/src/diveengine2d/Entity.java +++ b/src/diveengine2d/Entity.java @@ -10,6 +10,15 @@ public class Entity { public List components = new ArrayList(); + public T getComponent(Class componentType) { + for(DiveScript script : components) { + if(script.getClass().isAssignableFrom(componentType)) { + return (T) script; + } + } + return null; + } + public String toString() { return "name: " + name + "\n" + "GUID: " + GUID +