From: Matthijs Kooijman <m.kooijman@student.utwente.nl>
Date: Sun, 21 Jun 2009 14:46:05 +0000 (+0200)
Subject: Add is_poly and is_var predicates.
X-Git-Url: https://git.stderr.nl/gitweb?a=commitdiff_plain;h=e22ea2b99253fddd3846b2b219ca7a896a63d465;p=matthijs%2Fmaster-project%2Fc%CE%BBash.git

Add is_poly and is_var predicates.
---

diff --git a/CoreTools.hs b/CoreTools.hs
index 0bfe58f..444a4ba 100644
--- a/CoreTools.hs
+++ b/CoreTools.hs
@@ -3,7 +3,10 @@
 -- Core and Haskell (it uses HsTools for this), but only the functions that
 -- know about various libraries and know which functions to call.
 module CoreTools where
-  
+
+--Standard modules
+import qualified Maybe
+
 -- GHC API
 import qualified GHC
 import qualified Type
@@ -101,3 +104,16 @@ is_fun :: CoreSyn.CoreExpr -> Bool
 -- Treat Type arguments differently, because exprType is not defined for them.
 is_fun (CoreSyn.Type _) = False
 is_fun expr = (Type.isFunTy . CoreUtils.exprType) expr
+
+-- Is the given core expression polymorphic (i.e., does it accept type
+-- arguments?).
+is_poly :: CoreSyn.CoreExpr -> Bool
+-- Treat Type arguments differently, because exprType is not defined for them.
+is_poly (CoreSyn.Type _) = False
+is_poly expr = (Maybe.isJust . Type.splitForAllTy_maybe . CoreUtils.exprType) expr
+
+-- Is the given core expression a variable reference?
+is_var :: CoreSyn.CoreExpr -> Bool
+is_var (CoreSyn.Var _) = True
+is_var _ = False
+