Added Watchdog to getAttributes

lf-lang · Jul 11, 2024 · 9e4355c · 9e4355c
1 parent 521a502
commit 9e4355c
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/core/src/main/java/org/lflang/AttributeUtils.java b/core/src/main/java/org/lflang/AttributeUtils.java
@@ -36,17 +36,7 @@
 import org.eclipse.xtext.nodemodel.util.NodeModelUtils;
 import org.eclipse.xtext.resource.XtextResource;
 import org.lflang.ast.ASTUtils;
-import org.lflang.lf.Action;
-import org.lflang.lf.AttrParm;
-import org.lflang.lf.Attribute;
-import org.lflang.lf.Input;
-import org.lflang.lf.Instantiation;
-import org.lflang.lf.Output;
-import org.lflang.lf.Parameter;
-import org.lflang.lf.Reaction;
-import org.lflang.lf.Reactor;
-import org.lflang.lf.StateVar;
-import org.lflang.lf.Timer;
+import org.lflang.lf.*;
 import org.lflang.util.StringUtil;
 
 /**
@@ -83,6 +73,8 @@ public static List<Attribute> getAttributes(EObject node) {
       return ((Output) node).getAttributes();
     } else if (node instanceof Instantiation) {
       return ((Instantiation) node).getAttributes();
+    } else if (node instanceof Watchdog) {
+      return ((Watchdog) node).getAttributes();
     }
     throw new IllegalArgumentException("Not annotatable: " + node);
   }