// $Id: DeploymentDiagramRenderer.java,v 1.14 2004/08/28 15:24:29 mvw Exp $
// Copyright (c) 2003 The Regents of the University of California. All
// Rights Reserved. Permission to use, copy, modify, and distribute this
// software and its documentation without fee, and without a written
// agreement is hereby granted, provided that the above copyright notice
// and this paragraph appear in all copies.  This software program and
// documentation are copyrighted by The Regents of the University of
// California. The software program and documentation are supplied "AS
// IS", without any accompanying services from The Regents. The Regents
// does not warrant that the operation of the program will be
// uninterrupted or error-free. The end-user understands that the program
// was developed for research purposes and is advised not to rely
// exclusively on the program for any reason.  IN NO EVENT SHALL THE
// UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT,
// SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS,
// ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF
// THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF
// SUCH DAMAGE. THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY
// WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE
// PROVIDED HEREUNDER IS ON AN "AS IS" BASIS, AND THE UNIVERSITY OF
// CALIFORNIA HAS NO OBLIGATIONS TO PROVIDE MAINTENANCE, SUPPORT,
// UPDATES, ENHANCEMENTS, OR MODIFICATIONS.

package org.argouml.uml.diagram.deployment.ui;

import java.util.Collection;

import org.apache.log4j.Logger;

import org.argouml.model.ModelFacade;
import org.argouml.uml.diagram.static_structure.ui.CommentEdge;
import org.argouml.uml.diagram.static_structure.ui.FigClass;
import org.argouml.uml.diagram.static_structure.ui.FigComment;
import org.argouml.uml.diagram.static_structure.ui.FigEdgeNote;
import org.argouml.uml.diagram.static_structure.ui.FigInterface;
import org.argouml.uml.diagram.static_structure.ui.FigLink;
import org.argouml.uml.diagram.ui.FigAssociation;
import org.argouml.uml.diagram.ui.FigDependency;

import org.tigris.gef.base.Layer;
import org.tigris.gef.graph.GraphEdgeRenderer;
import org.tigris.gef.graph.GraphModel;
import org.tigris.gef.graph.GraphNodeRenderer;
import org.tigris.gef.presentation.FigEdge;
import org.tigris.gef.presentation.FigNode;

/**
 * This class defines a renderer object for UML Deployment Diagrams. 
 *
 */
public class DeploymentDiagramRenderer
    implements GraphNodeRenderer, GraphEdgeRenderer 
{
    private static final Logger LOG =
	Logger.getLogger(DeploymentDiagramRenderer.class);

    /** 
     * Return a Fig that can be used to represent the given node. 
     * 
     * @see org.tigris.gef.graph.GraphNodeRenderer#getFigNodeFor(
     * org.tigris.gef.graph.GraphModel, org.tigris.gef.base.Layer, 
     * java.lang.Object)
     */
    public FigNode getFigNodeFor(GraphModel gm, Layer lay, Object node) {
	if (ModelFacade.isANode(node)) return new FigMNode(gm, node);
	else if (ModelFacade.isANodeInstance(node))
	    return new FigMNodeInstance(gm, node);
	else if (ModelFacade.isAComponent(node))
	    return new FigComponent(gm, node); 
	else if (ModelFacade.isAComponentInstance(node))
	    return new FigComponentInstance(gm, node);
	else if (ModelFacade.isAClass(node)) return new FigClass(gm, node); 
	else if (ModelFacade.isAInterface(node))
	    return new FigInterface(gm, node); 
	else if (ModelFacade.isAObject(node)) return new FigObject(gm, node);
	else if (ModelFacade.isAComment(node)) {
	            return new FigComment(gm, node);
	}
	LOG.debug("TODO: DeploymentDiagramRenderer getFigNodeFor");
	return null;
    }

    /** 
     * Return a Fig that can be used to represent the given edge.
     * 
     * @see org.tigris.gef.graph.GraphEdgeRenderer#getFigEdgeFor(
     * org.tigris.gef.graph.GraphModel, org.tigris.gef.base.Layer, 
     * java.lang.Object)
     */
    public FigEdge getFigEdgeFor(GraphModel gm, Layer lay, Object edge) {

	if (ModelFacade.isAAssociation(edge)) {
	    Object asc = /*(MAssociation)*/ edge;
	    FigAssociation ascFig = new FigAssociation(asc, lay);
	    return ascFig;
	}
	if (ModelFacade.isALink(edge)) {
	    Object lnk = /*(MLink)*/ edge;
	    FigLink lnkFig = new FigLink(lnk);
	    Collection linkEnds = ModelFacade.getConnections(lnk);
	    if (linkEnds == null) LOG.debug("null linkRoles....");
	    Object[] leArray = linkEnds.toArray();
	    Object fromEnd = leArray[0];
	    Object fromInst = ModelFacade.getInstance(fromEnd);
	    Object toEnd = leArray[1];
	    Object toInst = ModelFacade.getInstance(toEnd);
	    FigNode fromFN = (FigNode) lay.presentationFor(fromInst);
	    FigNode toFN = (FigNode) lay.presentationFor(toInst);
	    lnkFig.setSourcePortFig(fromFN);
	    lnkFig.setSourceFigNode(fromFN);
	    lnkFig.setDestPortFig(toFN);
	    lnkFig.setDestFigNode(toFN);
	    return lnkFig;
	}
	if (ModelFacade.isADependency(edge)) {
	    Object dep = /*(MDependency)*/ edge;
	    FigDependency depFig = new FigDependency(dep);

	    Object supplier = /*(MModelElement)*/
		 ((ModelFacade.getSuppliers(dep).toArray())[0]);
	    Object client = /*(MModelElement)*/
		 ((ModelFacade.getClients(dep).toArray())[0]);

	    FigNode supFN = (FigNode) lay.presentationFor(supplier);
	    FigNode cliFN = (FigNode) lay.presentationFor(client);

	    depFig.setSourcePortFig(cliFN);
	    depFig.setSourceFigNode(cliFN);
	    depFig.setDestPortFig(supFN);
	    depFig.setDestFigNode(supFN);
	    depFig.getFig().setDashed(true);
	    return depFig;
	}
	if (edge instanceof CommentEdge) {
            return new FigEdgeNote(edge, lay);
        }

	return null;
    }

    static final long serialVersionUID = 8002278834226522224L;

}