Clean up ResolveHeatCoolAttribute

runtimeverification · Oct 24, 2023 · 08fbf2b · 08fbf2b
1 parent 143996f
commit 08fbf2b
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/kernel/src/main/java/org/kframework/compile/ResolveHeatCoolAttribute.java b/kernel/src/main/java/org/kframework/compile/ResolveHeatCoolAttribute.java
@@ -2,7 +2,6 @@
 package org.kframework.compile;
 
 import org.kframework.attributes.Att;
-import org.kframework.attributes.Att.Key;
 import org.kframework.builtin.BooleanUtils;
 import org.kframework.definition.Context;
 import org.kframework.definition.Module;
@@ -20,13 +19,7 @@
 import static org.kframework.definition.Constructors.*;
 import static org.kframework.kore.KORE.*;
 
-public class ResolveHeatCoolAttribute {
-
-    private final Set<String> unrestrictedRules;
-
-    public ResolveHeatCoolAttribute(Set<String> unrestrictedRules) {
-        this.unrestrictedRules = unrestrictedRules;
-    }
+public record ResolveHeatCoolAttribute(Set<String> unrestrictedRules) {
 
     private Rule resolve(Module m, Rule rule) {
         return Rule(
@@ -70,10 +63,10 @@ public Sentence resolve(Module m, Sentence s) {
         if (!s.att().contains(Att.HEAT()) && !s.att().contains(Att.COOL())) {
             return s;
         }
-        if (s instanceof Rule) {
-            return resolve(m, (Rule) s);
-        } else if (s instanceof Context) {
-            return resolve(m, (Context) s);
+        if (s instanceof Rule r) {
+            return resolve(m, r);
+        } else if (s instanceof Context c) {
+            return resolve(m, c);
         } else {
             return s;
         }