1 files changed, 14 insertions, 2 deletions
diff --git a/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/Entity.java b/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/Entity.java
index a9716146bb..0ae87d8816 100755
--- a/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/Entity.java
+++ b/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/Entity.java
@@ -12,14 +12,13 @@ import org.apache.tuscany.sca.impl.artifacts.Service;
 /**
  * Represents an unit (a component including its references, services, properties
  * and adjacent units) in the diagram.
- * @author nirmal
  *
  */
 public class Entity {
 	
 	private String componentName;
 	private int X, Y, level=-1, lane=-1, refHeight, serHeight, propLength;
-	
+	private final int height= Component.DEFAULT_HEIGHT, width= Component.DEFAULT_WIDTH;
 	public static final int defaultNoOfSers= Component.DEFAULT_HEIGHT / (Service.MAXIMUM_HEIGHT+Service.SPACING);
 	public static final int defaultNoOfRefs= Component.DEFAULT_HEIGHT / (Reference.MAXIMUM_HEIGHT+Reference.SPACING); //same value for defaultNoOfSers
 	public static final int defaultNoOfProps= Component.DEFAULT_WIDTH / (Property.MAXIMUM_HEIGHT+Property.SPACING); 
@@ -178,6 +177,7 @@ public class Entity {
 	public void setServices(ArrayList<String> services) {
 		this.services = services;
 	}
+	
 	public ArrayList<String> getServices() {
 		return services;
 	}
@@ -217,6 +217,10 @@ public class Entity {
 	public ArrayList<String> getRefProperties() {
 		return refProperties;
 	}
+	
+	public ArrayList<String> getReferences() {
+		return references;
+	}
 
 	public void setRefProperties(ArrayList<String> refProperties) {
 		this.refProperties = refProperties;
@@ -258,4 +262,12 @@ public class Entity {
 		return connectedEntities;
 	}
 
+	public int getHeight() {
+		return height;
+	}
+
+	public int getWidth() {
+		return width;
+	}
+
 }