DGP - added isCompatible checks for all Rocksim components

[debian/openrocket] / src / net / sf / openrocket / file / rocksim / MassObjectHandler.java

diff --git a/src/net/sf/openrocket/file/rocksim/MassObjectHandler.java b/src/net/sf/openrocket/file/rocksim/MassObjectHandler.java
index 31ddef7059657e3522fdf8e64c49beaa3a022b9c..479242d954529b4a254e18ce7079644502d1c8d8 100644 (file)
--- a/src/net/sf/openrocket/file/rocksim/MassObjectHandler.java
+++ b/src/net/sf/openrocket/file/rocksim/MassObjectHandler.java
@@ -36,15 +36,18 @@ class MassObjectHandler extends PositionDependentHandler<MassComponent> {
      * Constructor.
      *l
      * @param c the parent component
+     * @param warnings  the warning set
      * 
      * @throws IllegalArgumentException  thrown if <code>c</code> is null
      */
-    public MassObjectHandler(RocketComponent c) throws IllegalArgumentException {
+    public MassObjectHandler(RocketComponent c, WarningSet warnings) throws IllegalArgumentException {
         if (c == null) {
             throw new IllegalArgumentException("The parent component of a mass component may not be null.");
         }
         mass = new MassComponent();
-        c.addChild(mass);
+        if (isCompatible(c, MassComponent.class, warnings)) {
+            c.addChild(mass);
+        }
     }
 
     @Override