Define subterms paramorphically.

github · Mar 19, 2024 · 050188f · 050188f
1 parent 3861c38
commit 050188f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/semantic-analysis/src/Analysis/Syntax.hs b/semantic-analysis/src/Analysis/Syntax.hs
@@ -28,10 +28,8 @@ instance (forall t . Eq t => Eq (sig t), forall t. Ord t => Ord (sig t), Ord v)
   compare _          _        = GT
 
 
-subterms :: (forall t . Eq t => Eq (sig t), forall t . Ord t => Ord (sig t), Ord v, Foldable sig) => Term sig v -> Set.Set (Term sig v)
-subterms t = case t of
-  Var _  -> Set.singleton t
-  Term s -> Set.insert t (foldMap subterms s)
+subterms :: (forall t . Eq t => Eq (sig t), forall t . Ord t => Ord (sig t), Ord v, Foldable sig, Functor sig) => Term sig v -> Set.Set (Term sig v)
+subterms = paraTerm (Set.singleton . Var) (foldMap (uncurry Set.insert))
 
 foldTerm :: Functor sig => (v -> r) -> (sig r -> r) -> (Term sig v -> r)
 foldTerm var sig = go