package com.c2kernel.collection;

import com.c2kernel.common.ObjectNotFoundException;
import com.c2kernel.graph.model.GraphModel;
import com.c2kernel.graph.model.GraphPoint;
import com.c2kernel.graph.model.TypeNameAndConstructionInfo;
import com.c2kernel.lookup.ItemPath;
import com.c2kernel.persistency.ClusterStorage;
import com.c2kernel.process.Gateway;
import com.c2kernel.utils.CastorHashMap;
import com.c2kernel.utils.Language;
import com.c2kernel.utils.Logger;

/**
 * @version $Revision: 1.59 $ $Date: 2004/08/10 07:56:08 $
 * @author  $Author: abranson $
 */

abstract public class Aggregation extends Collection<AggregationMember>
{

    protected GraphModel mLayout = new GraphModel(new AggregationVertexOutlineCreator());

    private final TypeNameAndConstructionInfo[] mVertexTypeNameAndConstructionInfo = {
            new TypeNameAndConstructionInfo(Language.translate("Slot"), "AggregationMember")
    };

    public Aggregation()
    {
    	setName("Aggregation");
    }


    public GraphModel getLayout()
    {
        return mLayout;
    }

    public void setLayout(GraphModel layout)
    {
        mLayout = layout;
        layout.setVertexOutlineCreator(new AggregationVertexOutlineCreator());
    }

    public TypeNameAndConstructionInfo[] getVertexTypeNameAndConstructionInfo()
    {
        return mVertexTypeNameAndConstructionInfo;
    }

    public boolean exists(ItemPath itemPath)
    {
    	for (int i=0; i<size(); i++)
    	{
    		AggregationMember element = mMembers.list.get(i);
            if (element.getItemPath().equals(itemPath))
                return true;
    	}
    	return false;
    }

	public AggregationMember getMemberPair(int vertexID)
	{
    	for (int i=0; i<size(); i++)
    	{
    		AggregationMember element = mMembers.list.get(i);
            if (element.getID() == vertexID)
                return element;
    	}
    	return null;
	}

    public AggregationMember addSlot(CastorHashMap props, String classProps, GraphPoint location, int w, int h)
    {

		// Default geometry if not present
        if (location == null) location = new GraphPoint(100,100*getCounter());
        if (w<0) w = 20;
        if (h<0) h = 20;

        // Create new member object
        AggregationMember aggMem = new AggregationMember();
        aggMem.setProperties(props);
        aggMem.setClassProps(classProps);
        // create vertex
        com.c2kernel.graph.model.Vertex vertex = new com.c2kernel.graph.model.Vertex();
        vertex.setHeight(h); vertex.setWidth(w);
        mLayout.addVertexAndCreateId(vertex,location);
        aggMem.setCollection(this);
        aggMem.setID(vertex.getID());
        aggMem.setIsLayoutable(true);
        
        mMembers.list.add(aggMem);
        Logger.msg(8, "AggregationDescription::addSlot new slot linked to vertexid " + vertex.getID());
        return aggMem;
    }	
    
    public AggregationMember addMember(ItemPath itemPath, CastorHashMap props, String classProps, GraphPoint location, int w, int h)
        throws MembershipException
    {
    	AggregationMember aggMem = addSlot(props, classProps, location, w, h);
        if (itemPath != null) { // some clients use this method when not setting a member
        	aggMem.assignItem(itemPath);
        	aggMem.setIsComposite( getIsComposite(itemPath, getName())  );
        }
        Logger.msg(8, "AggregationDescription::addMember(" + itemPath + ") assigned to new slot " + aggMem.getID());
        return aggMem;
    }


    @Override
	public AggregationMember addMember(ItemPath itemPath, CastorHashMap props, String classProps) throws MembershipException
    {
        return addMember(itemPath, props, classProps, null, -1, -1);
    }
    
	public AggregationMember addSlot(CastorHashMap props, String classProps)
    {
        return addSlot(props, classProps, null, -1, -1);
    }

    @Override
	public AggregationMember addMember(ItemPath itemPath) throws MembershipException {
        throw new MembershipException("Aggregations cannot accept arbitrary members without type info");

    }

    @Override
	public void removeMember(int memberId) throws MembershipException {
        for (AggregationMember element : mMembers.list) {
            if (element.getID() == memberId) {
                element.clearItem();
                mLayout.removeVertex(getLayout().getVertexById(memberId));
                return;
            }
        }
        throw new MembershipException("Member "+memberId+" not found");
    }
    
	static public boolean getIsComposite(ItemPath itemPath, String name)
	{
        if (itemPath == null) return false;
       	try {
			for(String collName: Gateway.getProxyManager().getProxy(itemPath).getContents(ClusterStorage.COLLECTION) )
					if (name == null || name.equals(collName)) return true;
		} catch (ObjectNotFoundException e) {
			return false;
		}
		return false;
	}

}