1 files changed, 18 insertions, 0 deletions
diff --git a/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/CompositeEntity.java b/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/CompositeEntity.java
index 0adffcc44e..810c0a8f09 100755
--- a/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/CompositeEntity.java
+++ b/collaboration/GSoC-2011-Nirmal/CompositeDiagramGeneratorUsingBatik/src/main/java/org/apache/tuscany/sca/impl/layout/CompositeEntity.java
@@ -43,12 +43,14 @@ public class CompositeEntity extends Entity{
 //	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); 
 	
+	private final String fileNameSuffix = "_diagram.svg";
 	private int maxInternalLevel=0;
 	private int maxInternalLane=0;
 	private ComponentEntity[] componentList;
 	private int[][] connections;
 	private HashMap<String, String> promoteAService = new HashMap<String, String>();
 	private HashMap<String, String> promoteAReference = new HashMap<String, String>();
+	private ArrayList<String> includedComposites = new ArrayList<String>();
 	//private HashSet<String> connectedEntities = new HashSet<String>();
 	
 	
@@ -201,6 +203,13 @@ public class CompositeEntity extends Entity{
 		promoteAReference.put(compositeRef, componentRef);
 		return true;
 	}
+	
+	public boolean addToIncludedComposites(String composite){
+		
+		includedComposites.add(composite);
+		
+		return true;
+	}
 
 	public void setPromoteAReference(HashMap<String, String> promoteAReference) {
 		this.promoteAReference = promoteAReference;
@@ -252,6 +261,15 @@ public class CompositeEntity extends Entity{
 	}
 
 	
+	public ArrayList<String> getIncludedComposites() {
+		return includedComposites;
+	}
+
+	public String getFileNameSuffix() {
+		return fileNameSuffix;
+	}
+
+	
 	
 //	public int getNoOfRefs(){
 //		return references.size();