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: import java.util.Iterator;
046: import org.netbeans.modules.uml.core.support.umlutils.ETList;
047: import org.netbeans.modules.uml.ui.products.ad.compartments.sequencediagram.ETLifelineNameCompartment;
048: import org.netbeans.modules.uml.ui.support.viewfactorysupport.ICompartment;
049: import org.netbeans.modules.uml.ui.support.viewfactorysupport.IETGraphObject;
050: import org.netbeans.test.umllib.DiagramElementChooser;
051: import org.netbeans.test.umllib.DiagramElementOperator;
052: import org.netbeans.test.umllib.DiagramOperator;
053: import org.netbeans.test.umllib.ElementTypes;
054: import org.netbeans.test.umllib.exceptions.NotFoundException;
055:
056: public class LifelineOperator extends DiagramElementOperator {
057:
058: public LifelineOperator(DiagramOperator diagramOperator,
059: IETGraphObject graphObject) throws NotFoundException {
060: super (diagramOperator, graphObject);
061: }
062:
063: public LifelineOperator(DiagramOperator diagramOperator,
064: String lifelineName, String classifierName)
065: throws NotFoundException {
066: super (diagramOperator, new LifelineByFullNameChooser(
067: lifelineName, classifierName), 0);
068: }
069:
070: public LifelineOperator(DiagramOperator diagramOperator,
071: String lifelineName, String classifierName, int index)
072: throws NotFoundException {
073: super (diagramOperator, new LifelineByFullNameChooser(
074: lifelineName, classifierName), index);
075: }
076:
077: /**
078: * Construct DiagramElementOperator by custom finder and index
079: * @param diagramOperator Diagram to look for element
080: * @param elementFinder custom finder
081: * @param index index
082: * @throws qa.uml.exceptions.NotFoundException when no suitable element found
083: */
084: public LifelineOperator(DiagramOperator diagramOperator,
085: DiagramElementChooser elementFinder, int index)
086: throws NotFoundException {
087: super (diagramOperator, waitForGraphObject(diagramOperator,
088: elementFinder, index));
089:
090: }
091:
092: public boolean equals(Object obj) {
093: return (obj instanceof LifelineOperator)
094: && (((LifelineOperator) obj).getGraphObject() == getGraphObject());
095: }
096:
097: public static class LifelineByFullNameChooser implements
098: DiagramElementChooser {
099: private String lineName = null;
100: private String classifierName = null;
101:
102: /**
103: *
104: * @param lineName
105: * @param classifierName
106: */
107: public LifelineByFullNameChooser(String lineName,
108: String classifierName) {
109: this .lineName = lineName;
110: this .classifierName = classifierName;
111: }
112:
113: /**
114: *
115: * @param vn
116: * @param elementType
117: */
118: /*
119: public ElementByVNChooser(String vn, String elementType){
120: this(vn, elementType, new Operator.DefaultStringComparator(true,true));
121: }
122: */
123: /**
124: *
125: * @param graphObject
126: * @return
127: */
128: public boolean checkElement(IETGraphObject graphObject) {
129:
130: if (!graphObject.getEngine().getElementType().equals(
131: ElementTypes.LIFELINE.toString())) {
132: return false;
133: }
134:
135: ETList<ICompartment> compartments = graphObject.getEngine()
136: .getCompartments();
137: if (compartments == null) {
138: return false;
139: }
140:
141: Iterator<ICompartment> it = compartments.iterator();
142: ArrayList<ICompartment> compartmentsFound = new ArrayList<ICompartment>();
143: while (it.hasNext()) {
144: ICompartment co = it.next();
145: if (co instanceof ETLifelineNameCompartment) {
146: String name = co.getName();
147: if (name.equals(lineName + " : " + classifierName)) {
148: return true;
149: } else {
150: return false;
151: }
152: }
153: }
154: return false;
155: }
156:
157: /**
158: *
159: * @return
160: */
161: public String getDescription() {
162: return "Chooser for description:" + lineName + " : "
163: + classifierName;
164: }
165:
166: }
167:
168: }
|