Output of PlantUML

Author: kevinlano

Association.java

@@ -5806,9 +5806,11 @@ else if (card2 == ONE)
     String opp = ""; 
     if (role1 != null && role1.length() > 0)
     { opp = " oppositeOf " + role1; } 
+    else if (initialExpression != null)
+    { opp = " := " + initialExpression; } 
 
-	if (isQualified())
-	{ return "    reference " + role2 + mult + " qualified " + ord + agg + ": " + entity2 + opp + ";"; }
+    if (isQualified())
+    { return "    reference " + role2 + mult + " qualified " + ord + agg + ": " + entity2 + opp + ";"; }
 
     return "    reference " + role2 + mult + " " + ord + agg + ": " + entity2 + opp + ";";  
   } 
@@ -5850,6 +5852,36 @@ else if (card2 == ZEROONE)
     out.println(res + "/>");  
   } // ordering? 
 
+  public void saveAsPlantUML(PrintWriter out)
+  { String c1 = ""; 
+
+    if (card1 == ONE) 
+    { c1 = "1"; } 
+    else if (card1 == ZEROONE) 
+    { c1 = "0..1"; }
+    else if (card1 == MANY)
+    { c1 = "*"; }
+
+    String c2 = ""; 
+
+    if (card2 == ONE) 
+    { c2 = "1"; } 
+    else if (card2 == ZEROONE) 
+    { c2 = "0..1"; }
+    else if (card2 == MANY)
+    { c2 = "*"; }
+ 
+    String arrow = " --> "; 
+    if (aggregation)
+    { arrow = " *--> "; } 
+
+    if (ordered)
+    { c2 = c2 + " {ordered}"; }
+
+    String res = " " + entity1 + " \"" + c1 + "\"" + arrow + " \"" + c2 + "\" " + entity2 + " : " + role2; 
+    out.println(res + "\n");  
+  } // ordering? 
+
   public String xmiSaveModel(String domain) 
   { String res = ""; 
     String e1name = entity1.getName(); 

BasicExpression.java

@@ -3631,10 +3631,27 @@ else if (objectRef.type != null &&
                objectRef.type.isEntity(entities)) 
       { context.add(0,objectRef.type.getEntity(entities)); }
       else 
-      { System.err.println("! Warning: Cannot locate class of " + objectRef); 
-        /* JOptionPane.showMessageDialog(null, 
-           "Warning!: Cannot locate class of " + objectRef, 
-           "Semantic error", JOptionPane.ERROR_MESSAGE); */ 
+      { System.out.println(">>> Searching for class of " + objectRef);
+
+        // Search for a class with data. 
+
+        for (int i = 0; i < entities.size(); i++) 
+        { Entity ex = (Entity) entities.get(i); 
+          if (ex.hasAttribute(data)) 
+          { JOptionPane.showMessageDialog(null, 
+              "Located possible class " + ex + " of " + objectRef, 
+              "", JOptionPane.INFORMATION_MESSAGE);
+            objectRef.type = new Type(ex);
+
+            vartypes.put("" + objectRef, objectRef.type); 
+
+            Attribute attx = ex.getAttribute(data); 
+            type = attx.getType(); 
+            elementType = attx.getElementType();  
+
+            vartypes.put(data, type); 
+          } 
+        }
       } 
     }
     else // objectRef == null

BehaviouralFeature.java

@@ -5120,6 +5120,61 @@ public String saveAsUSEData()
     return res; 
   }  // multiple pre, post, and stereotypes
 
+  public String saveAsPlantUML()
+  { if (derived) { return ""; } 
+   
+    String res = ""; 
+   
+    if (instanceScope) 
+    { if (isAbstract()) 
+      { res = res + "{abstract} "; }
+    }  
+    else 
+    { res = res + "{static} "; } 
+
+    res = res + getName() + "(";
+
+    for (int i = 0; i < parameters.size(); i++)
+    { Attribute par = (Attribute) parameters.get(i);
+      String pnme = par.getName();
+      String tnme = par.getType().getUMLName();
+      String elemt = ""; 
+      if ("Set".equals(tnme) || "Sequence".equals(tnme))
+      { if (par.getElementType() != null) 
+        { elemt = "(" + par.getElementType().getUMLName() + ")"; } 
+      } 
+      res = res + pnme + " : " + tnme + elemt;
+      if (i < parameters.size() - 1) { res = res + ", "; } 
+    }
+    res = res + ") "; 
+
+    if (resultType != null)
+    { String rtname = resultType.getName(); 
+      if (rtname.equals("Set") || rtname.equals("Sequence"))
+      { if (elementType != null) 
+        { res = res + " : " + resultType.getUMLName() + "(" + elementType.getUMLName() + ")"; } 
+        else 
+        { res = res + " : " + resultType.getUMLName() + "()"; } 
+      } 
+      else 
+      { res = res + " : " + resultType.getUMLName(); }
+    } 
+    res = res + "\n"; 
+    
+  /*  java.util.Map env = new java.util.HashMap(); 
+
+    if (pre == null) 
+    { pre = new BasicExpression("true"); }  
+    
+    if (post == null) 
+    { post = new BasicExpression("true"); } 
+
+    res = res + "pre pre1:\n  " + pre.toOcl(env,true) + "\n";
+    res = res + "post post1:\n   " + post.toOcl(env,true); 
+    res = res + "\n\n"; */ 
+
+    return res; 
+  }  // multiple pre, post, and stereotypes
 
   public static Expression wpc(Expression pst, String op, String feat, 
                                String val)

Entity.java

@@ -11794,6 +11794,39 @@ public String saveAsUSEData()
     return res + "end\n\n";
   }   // for each op, save name, type, parameters, pre, post
 
+  public String saveAsPlantUML() 
+  { String nme = getName(); 
+
+    String inheritstring = ""; 
+
+    if (superclass != null) 
+    { inheritstring = superclass.getName() + " <|-- " + nme + "\n\n"; } 
+
+    for (int i = 0; i < interfaces.size(); i++) 
+    { Entity intf = (Entity) interfaces.get(i); 
+
+      inheritstring = inheritstring + 
+          intf.getName() + " <|.. " + nme + "\n\n"; 
+    }    
+
+    String res =  inheritstring + 
+                  "class " + nme + " {\n";
+    
+    for (int i = 0; i < attributes.size(); i++)
+    { Attribute att = (Attribute) attributes.get(i);
+      String aname = att.getName();
+      Type t = att.getType();
+      res = res + aname + " : " + t.getUMLName() + "\n";
+    }
+
+    for (int i = 0; i < operations.size(); i++)
+    { BehaviouralFeature op = (BehaviouralFeature) operations.get(i);
+      res = res + op.saveAsPlantUML();
+    }
+
+    return res + "}\n\n";
+  }   // for each op, save name, type, parameters, pre, post
+
   public String saveAsZ3Data() 
   { String ename = getName(); 
     String res =  "(declare-sort " + ename + ")\n";

UCDArea.java

@@ -12880,7 +12880,7 @@ public void generateRESTWebService()
       out.close(); 
     } 
     catch (IOException e) 
-    { System.out.println("!!! Error saving data"); } 
+    { System.err.println("!!! Error saving data"); } 
 
     for (int i = 0; i < useCases.size(); i++)
     { Object obj = useCases.get(i); 
@@ -12919,7 +12919,7 @@ public void generateSOAPWebService()
       out.close(); 
     } 
     catch (IOException e) 
-    { System.out.println("Error saving data"); } 
+    { System.err.println("!! Error saving data"); } 
   } 
 
 
@@ -12963,7 +12963,7 @@ public void saveDataToFile()
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving data"); } 
+      { System.err.println("!! Error saving data"); } 
     }
   }
 
@@ -13002,7 +13002,7 @@ public void saveDataToFile(String f)
       out.close(); 
     }
     catch (IOException e) 
-    { System.out.println("Error saving data"); } 
+    { System.err.println("!! Error saving data"); } 
   }
 
   public void saveEMFToFile()
@@ -13021,7 +13021,7 @@ public void saveEMFToFile()
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving EMF"); } 
+      { System.err.println("!! Error saving EMF"); } 
     }
   }
 
@@ -13041,7 +13041,7 @@ public void saveKM3ToFile()
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving KM3"); } 
+      { System.err.println("!! Error saving KM3"); } 
     }
   }
 
@@ -13061,7 +13061,7 @@ public void saveSimpleKM3ToFile()
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving KM3"); } 
+      { System.err.println("!! Error saving KM3"); } 
     }
   }
 
@@ -13082,7 +13082,7 @@ public void saveEcoreToFile()
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving Ecore"); } 
+      { System.err.println("!! Error saving Ecore"); } 
     }
   }
 
@@ -13104,7 +13104,7 @@ public void saveInterfaceDescription(String f)
       out.close(); 
     }
     catch (IOException e) 
-    { System.out.println("Error saving model"); } 
+    { System.err.println("!! Error saving model"); } 
   }
 
 
@@ -13143,7 +13143,7 @@ else if (uc instanceof UseCase)
         out.close(); 
       }
       catch (IOException e) 
-      { System.out.println("Error saving model"); } 
+      { System.err.println("!! Error saving model"); } 
     }
   }
 
@@ -13176,7 +13176,43 @@ public void saveUSEDataToFile(String f)
       out.close(); 
     }
     catch (IOException e) 
-    { System.out.println("Error saving data"); } 
+    { System.err.println("!! Error saving data"); } 
+  }
+
+  public void savePlantUMLToFile(String f)
+  { File file = new File("output/" + f);
+    // Vector locals = new Vector(); 
+
+    try
+    { PrintWriter out =
+          new PrintWriter(
+            new BufferedWriter(new FileWriter(file)));
+      out.println("@startuml"); 
+
+      for (int p = 0; p < entities.size(); p++) 
+      { Entity ent = (Entity) entities.get(p); 
+        out.println(ent.saveAsPlantUML()); 
+        // locals.addAll(ent.getInvariants()); 
+      } 
+
+      for (int i = 0; i < associations.size(); i++) 
+      { Association ast = (Association) associations.get(i); 
+        ast.saveAsPlantUML(out); 
+      } 
+
+      /* if (locals.size() > 0)
+      { out.println("constraints\n\n"); } 
+
+      for (int q = 0; q < locals.size(); q++)
+      { Constraint cc = (Constraint) locals.get(q); 
+        out.println(cc.saveAsUSEData()); 
+      } */ 
+
+      out.println("@enduml"); 
+      out.close(); 
+    }
+    catch (IOException e) 
+    { System.err.println("!! Error saving data"); } 
   }
 
 

UmlTool.java

@@ -459,6 +459,10 @@ public UmlTool()
     saveUSEMI.addActionListener(this); 
     saveMenu.add(saveUSEMI); 
 
+    JMenuItem saveplantMI = new JMenuItem("Save as PlantUML"); 
+    saveplantMI.addActionListener(this); 
+    saveMenu.add(saveplantMI); 
+
 
     JMenuItem saveCSVMI = new JMenuItem("Save as CSV"); 
     saveCSVMI.addActionListener(this); 
@@ -1706,6 +1710,10 @@ else if (label.equals("Save as USE"))
       { ucdArea.saveUSEDataToFile("mm.use"); 
         thisLabel.setText("Model saved to output/mm.use"); 
       } 
+      else if (label.equals("Save as PlantUML"))
+      { ucdArea.savePlantUMLToFile("mm.plantuml"); 
+        thisLabel.setText("Model saved to output/mm.plantuml"); 
+      } 
       else if (label.equals("Save as CSV"))
       { ucdArea.saveCSV(); } 
       else if (label.equals("Map TL to CSTL"))