X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=CoreTools.hs;h=3c26793c8c804ce7f9dd7262851565398638195d;hb=d21c34b00b9041a146da89324e9dda6b22271b47;hp=444a4ba6c2ab7e469e164b030f77b5067c905459;hpb=e22ea2b99253fddd3846b2b219ca7a896a63d465;p=matthijs%2Fmaster-project%2Fc%CE%BBash.git diff --git a/CoreTools.hs b/CoreTools.hs index 444a4ba..3c26793 100644 --- a/CoreTools.hs +++ b/CoreTools.hs @@ -22,8 +22,10 @@ import qualified DynFlags import qualified SrcLoc import qualified CoreSyn import qualified Var +import qualified VarSet import qualified Unique import qualified CoreUtils +import qualified CoreFVs import GhcTools import HsTools @@ -61,6 +63,14 @@ sized_word_len ty = where (tycon, args) = Type.splitTyConApp ty [len] = args + +-- | Get the upperbound of a RangedWord type +ranged_word_bound :: Type.Type -> Int +ranged_word_bound ty = + eval_tfp_int len + where + (tycon, args) = Type.splitTyConApp ty + [len] = args -- | Evaluate a core Type representing type level int from the TypeLevel -- library to a real int. @@ -85,6 +95,13 @@ tfvec_len ty = where (tycon, args) = Type.splitTyConApp ty [len, el_ty] = args + +-- | Get the element type of a TFVec type +tfvec_elem :: Type.Type -> Type.Type +tfvec_elem ty = el_ty + where + (tycon, args) = Type.splitTyConApp ty + [len, el_ty] = args -- Is this a wild binder? is_wild :: CoreSyn.CoreBndr -> Bool @@ -117,3 +134,11 @@ is_var :: CoreSyn.CoreExpr -> Bool is_var (CoreSyn.Var _) = True is_var _ = False +-- Can the given core expression be applied to something? This is true for +-- applying to a value as well as a type. +is_applicable :: CoreSyn.CoreExpr -> Bool +is_applicable expr = is_fun expr || is_poly expr + +-- Does the given CoreExpr have any free type vars? +has_free_tyvars :: CoreSyn.CoreExpr -> Bool +has_free_tyvars = not . VarSet.isEmptyVarSet . (CoreFVs.exprSomeFreeVars Var.isTyVar)