001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2007 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041:
042: package org.netbeans.test.umllib.customelements;
043:
044: import java.util.ArrayList;
045:
046: import org.netbeans.test.umllib.DiagramElementChooser;
047: import org.netbeans.test.umllib.DiagramElementOperator;
048: import org.netbeans.test.umllib.DiagramOperator;
049: import org.netbeans.test.umllib.LinkOperator;
050: import org.netbeans.test.umllib.exceptions.NotFoundException;
051: import org.netbeans.modules.uml.ui.support.viewfactorysupport.IETGraphObject;
052: import org.netbeans.modules.uml.core.metamodel.core.foundation.IPresentationElement;
053: import org.netbeans.modules.uml.core.support.umlutils.ETList;
054: import org.netbeans.modules.uml.ui.support.applicationmanager.IEdgePresentation;
055: import org.netbeans.modules.uml.ui.support.applicationmanager.MessageEdgePresentation;
056: import org.netbeans.modules.uml.ui.support.viewfactorysupport.IETLabel;
057: import org.netbeans.modules.uml.ui.support.viewfactorysupport.IETEdge;
058:
059: /**
060: *
061: * @author sunflower
062: */
063:
064: public class MessageOperator extends LinkOperator {
065:
066: public MessageOperator(DiagramOperator diagramOperator,
067: String operationSignature) throws NotFoundException {
068: this (diagramOperator, operationSignature, 0);
069: }
070:
071: public MessageOperator(DiagramOperator diagramOperator,
072: String operationSignature, int index)
073: throws NotFoundException {
074: super (diagramOperator, (IETEdge) (new DiagramElementOperator(
075: diagramOperator, new MesssageByLabelChooser(
076: operationSignature), index, true)
077: .getGraphObject()));
078: }
079:
080: public LifelineOperator getFromLifeline() {
081: return new LifelineOperator(getDiagramOperator(), getFrom());
082: }
083:
084: public LifelineOperator getToLifeline() {
085: return new LifelineOperator(getDiagramOperator(), getTo());
086: }
087:
088: public static class MesssageByLabelChooser implements
089: DiagramElementChooser {
090:
091: String name;
092:
093: public MesssageByLabelChooser(String name) {
094: this .name = name;
095: }
096:
097: public boolean checkElement(IETGraphObject graphObject) {
098:
099: if (graphObject instanceof IETEdge) {
100:
101: IETEdge edgeGraphObject = (IETEdge) graphObject;
102: ETList<IETLabel> labelList = edgeGraphObject
103: .getLabels();
104:
105: if (labelList != null) {
106:
107: for (IETLabel label : labelList) {
108: if (name.equals(label.getText())) {
109: return true;
110: }
111:
112: }
113: }
114:
115: }
116:
117: return false;
118:
119: }
120:
121: public String getDescription() {
122: return "Chooser for Message Element: " + name;
123: }
124:
125: }
126:
127: }
|