/* $Id: StructureDiagram2Renderer.java 18714 2010-08-31 16:43:47Z bobtarling $
*****************************************************************************
* Copyright (c) 2009 Contributors - see below
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* bobtarling
*****************************************************************************
*
* Some portions of this file was previously release using the BSD License:
*/
// Copyright (c) 1996-2008 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.diagram.uml2;
import java.util.Collection;
import java.util.Map;
import org.apache.log4j.Logger;
import org.argouml.model.CoreFactory;
import org.argouml.model.Model;
import org.argouml.uml.CommentEdge;
import org.argouml.uml.diagram.ArgoDiagram;
import org.argouml.uml.diagram.DiagramEdgeSettings;
import org.argouml.uml.diagram.DiagramSettings;
import org.argouml.uml.diagram.GraphChangeAdapter;
import org.argouml.uml.diagram.UmlDiagramRenderer;
import org.argouml.uml.diagram.ui.FigNodeModelElement;
import org.argouml.uml.diagram.ui.UMLDiagram;
import org.tigris.gef.base.Diagram;
import org.tigris.gef.base.Layer;
import org.tigris.gef.base.LayerPerspective;
import org.tigris.gef.graph.GraphModel;
import org.tigris.gef.presentation.FigEdge;
import org.tigris.gef.presentation.FigNode;
/**
* This class defines a renderer object for UML Class Diagrams. In a
* Class Diagram the following UML objects are displayed with the
* following Figs: <p>
*
* <pre>
* UML Object --- Fig
* ---------------------------------------
* Class --- FigClass
* Interface --- FigInterface
* Instance --- FigInstance
* Model --- FigModel
* Subsystem --- FigSubsystem
* Package --- FigPackage
* Comment --- FigComment
* (CommentEdge) --- FigEdgeNote
* Generalization --- FigGeneralization
* Realization --- FigRealization
* Permission --- FigPermission
* Usage --- FigUsage
* Dependency --- FigDependency
* Association --- FigAssociation
* AssociationClass --- FigAssociationClass
* Dependency --- FigDependency
* Link --- FigLink
* DataType --- FigDataType
* Stereotype --- FigStereotypeDeclaration
* </pre>
*
* @author jrobbins
*/
public class StructureDiagram2Renderer extends UmlDiagramRenderer {
/**
* The UID.
*/
static final long serialVersionUID = 675407719309039112L;
/**
* Logger.
*/
private static final Logger LOG =
Logger.getLogger(StructureDiagram2Renderer.class);
/*
* @see org.tigris.gef.graph.GraphNodeRenderer#getFigNodeFor(
* org.tigris.gef.graph.GraphModel,
* org.tigris.gef.base.Layer, java.lang.Object, java.util.Map)
*/
public FigNode getFigNodeFor(GraphModel gm, Layer lay,
Object node, Map styleAttributes) {
FigNodeModelElement figNode = null;
if (node == null) {
throw new IllegalArgumentException("A node must be supplied");
}
// Although not generally true for GEF, for Argo we know that the layer
// is a LayerPerspective which knows the associated diagram
Diagram diag = ((LayerPerspective) lay).getDiagram();
if (diag instanceof UMLDiagram
&& ((UMLDiagram) diag).doesAccept(node)) {
figNode = (FigNodeModelElement) ((UMLDiagram) diag)
.drop(node, null);
} else {
LOG.error("TODO: ClassDiagramRenderer getFigNodeFor " + node);
throw new IllegalArgumentException(
"Node is not a recognised type. Received "
+ node.getClass().getName());
}
lay.add(figNode);
figNode.setDiElement(
GraphChangeAdapter.getInstance().createElement(gm, node));
return figNode;
}
/**
* Return a Fig that can be used to represent the given edge.
* Throws IllegalArgumentException if the edge is not of an expected type.
* Throws IllegalStateException if the edge generated has no source
* or dest port.
* {@inheritDoc}
*/
public FigEdge getFigEdgeFor(GraphModel gm, Layer lay,
Object edge, Map styleAttribute) {
if (LOG.isDebugEnabled()) {
LOG.debug("making figedge for " + edge);
}
if (edge == null) {
throw new IllegalArgumentException("A model edge must be supplied");
}
assert lay instanceof LayerPerspective;
ArgoDiagram diag = (ArgoDiagram) ((LayerPerspective) lay).getDiagram();
DiagramSettings settings = diag.getDiagramSettings();
FigEdge newEdge = null;
if (Model.getFacade().isAAssociationClass(edge)) {
Object[] associationEnds =
Model.getFacade().getConnections(edge).toArray();
newEdge = new FigAssociationClass2(
new DiagramEdgeSettings(
edge,
associationEnds[0],
associationEnds[1]),
settings);
} else if (Model.getFacade().isAAssociationEnd(edge)) {
FigAssociationEnd2 asend = new FigAssociationEnd2(edge, settings);
Model.getFacade().getAssociation(edge);
FigNode associationFN =
(FigNode) lay.presentationFor(
Model.getFacade().getAssociation(edge));
FigNode classifierFN =
(FigNode) lay.presentationFor(Model.getFacade().getType(edge));
asend.setSourcePortFig(associationFN);
asend.setSourceFigNode(associationFN);
asend.setDestPortFig(classifierFN);
asend.setDestFigNode(classifierFN);
newEdge = asend;
} else if (Model.getFacade().isAAssociation(edge)) {
Object[] associationEnds =
Model.getFacade().getConnections(edge).toArray();
newEdge = new FigAssociation2(
new DiagramEdgeSettings(
edge,
associationEnds[0],
associationEnds[1]),
settings);
} else if (Model.getFacade().isALink(edge)) {
FigLink2 lnkFig = new FigLink2(edge, settings);
Collection linkEndsColn = Model.getFacade().getConnections(edge);
Object[] linkEnds = linkEndsColn.toArray();
Object fromInst = Model.getFacade().getInstance(linkEnds[0]);
Object toInst = Model.getFacade().getInstance(linkEnds[1]);
FigNode fromFN = (FigNode) lay.presentationFor(fromInst);
FigNode toFN = (FigNode) lay.presentationFor(toInst);
lnkFig.setSourcePortFig(fromFN);
lnkFig.setSourceFigNode(fromFN);
lnkFig.setDestPortFig(toFN);
lnkFig.setDestFigNode(toFN);
lnkFig.getFig().setLayer(lay);
newEdge = lnkFig;
} else if (Model.getFacade().isAGeneralization(edge)) {
newEdge = new FigGeneralization2(edge, settings);
} else if (Model.getFacade().isAPackageImport(edge)) {
newEdge = new FigPermission2(edge, settings);
} else if (Model.getFacade().isAUsage(edge)) {
newEdge = new FigUsage2(edge, settings);
} else if (Model.getFacade().isAAbstraction(edge)) {
newEdge = new FigAbstraction2(edge, settings);
} else if (Model.getFacade().isADependency(edge)) {
String name = "";
for (Object stereotype : Model.getFacade().getStereotypes(edge)) {
name = Model.getFacade().getName(stereotype);
if (CoreFactory.REALIZE_STEREOTYPE.equals(name)) {
break;
}
}
if (CoreFactory.REALIZE_STEREOTYPE.equals(name)) {
// TODO: This code doesn't look like it will get reached because
// any abstraction/realization is going to take the
// isAAbstraction leg of the if before it gets to this more
// general case. - tfm 20080508
FigAbstraction2 realFig = new FigAbstraction2(edge, settings);
Object supplier =
((Model.getFacade().getSuppliers(edge).toArray())[0]);
Object client =
((Model.getFacade().getClients(edge).toArray())[0]);
FigNode supFN = (FigNode) lay.presentationFor(supplier);
FigNode cliFN = (FigNode) lay.presentationFor(client);
realFig.setSourcePortFig(cliFN);
realFig.setSourceFigNode(cliFN);
realFig.setDestPortFig(supFN);
realFig.setDestFigNode(supFN);
realFig.getFig().setLayer(lay);
newEdge = realFig;
} else {
FigDependency2 depFig = new FigDependency2(edge, settings);
newEdge = depFig;
}
} else if (edge instanceof CommentEdge) {
newEdge = new FigEdgeNote2(edge, settings);
}
addEdge(lay, newEdge, edge);
return newEdge;
}
}