svn commit: r13181 - trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java

Author: mvw
Date: 2007-07-29 01:15:03-0700
New Revision: 13181

Modified:
   trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java

Log:
Fix an exception in the new sequence diagram implementation.

Modified: trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java
Url: http://argouml.tigris.org/source/browse/argouml/trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java?view=diff&rev=13181&p1=trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java&p2=trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java&r1=13180&r2=13181
==============================================================================
--- trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java	(original)
+++ trunk/src/model-mdr/src/org/argouml/model/mdr/CoreHelperMDRImpl.java	2007-07-29 01:15:03-0700
@@ -1009,6 +1009,7 @@
     public Object getSource(Object relationship) {
         if (!(relationship instanceof Relationship)
                 && !(relationship instanceof Link)
+                && !(relationship instanceof Message)
                 && !(relationship instanceof AssociationEnd)) {
 
             throw new IllegalArgumentException("Argument "
@@ -1066,6 +1067,9 @@
             if (relationship instanceof AssociationEnd) {
                 return ((AssociationEnd) relationship).getAssociation();
             }
+            if (relationship instanceof Message) {
+                return ((Message) relationship).getSender();
+            }
         } catch (InvalidObjectException e) {
             throw new InvalidElementException(e);
         }
@@ -1076,6 +1080,7 @@
     public Object getDestination(Object relationship) {
         if (!(relationship instanceof Relationship)
                 && !(relationship instanceof Link)
+                && !(relationship instanceof Message)
                 && !(relationship instanceof AssociationEnd)) {
 
             throw new IllegalArgumentException("Argument is not "
@@ -1137,6 +1142,9 @@
             if (relationship instanceof AssociationEnd) {
                 return ((AssociationEnd) relationship).getParticipant();
             }
+            if (relationship instanceof Message) {
+                return ((Message) relationship).getReceiver();
+            }
         } catch (InvalidObjectException e) {
             throw new InvalidElementException(e);
         }